keepassxc-browser

src/proxy/CMakeLists.txt

@@ -0,0 +1,34 @@
+#  Copyright (C) 2017 Sami Vänttinen <sami.vanttinen@protonmail.com>
+#  Copyright (C) 2017 KeePassXC Team <team@keepassxc.org>
+#
+#  This program is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+if(WITH_XC_BROWSER)
+    include_directories(${BROWSER_SOURCE_DIR})
+
+    set(proxy_SOURCES
+        keepassxc-proxy.cpp
+        ${BROWSER_SOURCE_DIR}/NativeMessagingBase.cpp
+        NativeMessagingHost.cpp)
+
+    add_library(proxy STATIC ${proxy_SOURCES})
+    target_link_libraries(proxy Qt5::Core Qt5::Network)
+    add_executable(keepassxc-proxy keepassxc-proxy.cpp)
+    target_link_libraries(keepassxc-proxy proxy)
+
+    install(TARGETS keepassxc-proxy
+            BUNDLE DESTINATION . COMPONENT Runtime
+            RUNTIME DESTINATION ${PROXY_INSTALL_DIR} COMPONENT Runtime)
+
+endif()

src/proxy/NativeMessagingHost.cpp

@@ -0,0 +1,95 @@
+/*
+*  Copyright (C) 2017 KeePassXC Team <team@keepassxc.org>
+*
+*  This program is free software: you can redistribute it and/or modify
+*  it under the terms of the GNU General Public License as published by
+*  the Free Software Foundation, either version 3 of the License, or
+*  (at your option) any later version.
+*
+*  This program is distributed in the hope that it will be useful,
+*  but WITHOUT ANY WARRANTY; without even the implied warranty of
+*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*  GNU General Public License for more details.
+*
+*  You should have received a copy of the GNU General Public License
+*  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <QCoreApplication>
+#include "NativeMessagingHost.h"
+
+NativeMessagingHost::NativeMessagingHost() : NativeMessagingBase()
+{
+    m_localSocket = new QLocalSocket();
+    m_localSocket->connectToServer(getLocalServerPath());
+#ifdef Q_OS_WIN
+    m_running.store(true);
+    m_future = QtConcurrent::run(this, static_cast<void(NativeMessagingHost::*)()>(&NativeMessagingHost::readNativeMessages));
+#endif
+    connect(m_localSocket, SIGNAL(readyRead()), this, SLOT(newLocalMessage()));
+    connect(m_localSocket, SIGNAL(disconnected()), this, SLOT(deleteSocket()));
+    connect(m_localSocket, SIGNAL(stateChanged(QLocalSocket::LocalSocketState)), this, SLOT(socketStateChanged(QLocalSocket::LocalSocketState)));
+}
+
+NativeMessagingHost::~NativeMessagingHost()
+{
+#ifdef Q_OS_WIN
+    m_future.waitForFinished();
+#endif
+}
+
+void NativeMessagingHost::readLength()
+{
+    quint32 length = 0;
+    std::cin.read(reinterpret_cast<char*>(&length), 4);
+    if (!std::cin.eof() && length > 0) {
+        readStdIn(length);
+    } else {
+    	QCoreApplication::quit();
+    }
+}
+
+void NativeMessagingHost::readStdIn(const quint32 length)
+{
+    if (length > 0) {
+        QByteArray arr;
+        arr.reserve(length);
+
+        for (quint32 i = 0; i < length; ++i) {
+            arr.append(getchar());
+        }
+
+        if (arr.length() > 0 && m_localSocket && m_localSocket->state() == QLocalSocket::ConnectedState) {
+            m_localSocket->write(arr.constData(), arr.length());
+            m_localSocket->flush();
+        }
+    }
+}
+
+void NativeMessagingHost::newLocalMessage()
+{
+    if (!m_localSocket || m_localSocket->bytesAvailable() <= 0) {
+        return;
+    }
+
+    QByteArray arr = m_localSocket->readAll();
+    if (!arr.isEmpty()) {
+       sendReply(arr);
+    }
+}
+
+void NativeMessagingHost::deleteSocket()
+{
+    if (m_notifier) {
+        m_notifier->setEnabled(false);
+    }
+    m_localSocket->deleteLater();
+    QCoreApplication::quit();
+}
+
+void NativeMessagingHost::socketStateChanged(QLocalSocket::LocalSocketState socketState)
+{
+    if (socketState == QLocalSocket::UnconnectedState || socketState == QLocalSocket::ClosingState) {
+        m_running.testAndSetOrdered(true, false);
+    }
+}

src/proxy/NativeMessagingHost.h

@@ -0,0 +1,43 @@
+/*
+*  Copyright (C) 2017 KeePassXC Team <team@keepassxc.org>
+*
+*  This program is free software: you can redistribute it and/or modify
+*  it under the terms of the GNU General Public License as published by
+*  the Free Software Foundation, either version 3 of the License, or
+*  (at your option) any later version.
+*
+*  This program is distributed in the hope that it will be useful,
+*  but WITHOUT ANY WARRANTY; without even the implied warranty of
+*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*  GNU General Public License for more details.
+*
+*  You should have received a copy of the GNU General Public License
+*  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef NATIVEMESSAGINGHOST_H
+#define NATIVEMESSAGINGHOST_H
+
+#include "NativeMessagingBase.h"
+
+class NativeMessagingHost : public NativeMessagingBase
+{
+    Q_OBJECT
+public:
+    NativeMessagingHost();
+    ~NativeMessagingHost();
+
+public slots:
+    void newLocalMessage();
+    void deleteSocket();
+    void socketStateChanged(QLocalSocket::LocalSocketState socketState);
+
+private:
+    void readLength();
+    void readStdIn(const quint32 length);
+
+private:
+    QLocalSocket*                           m_localSocket;
+};
+
+#endif // NATIVEMESSAGINGHOST_H

src/proxy/keepassxc-proxy.cpp

@@ -0,0 +1,66 @@
+/*
+*  Copyright (C) 2017 Sami Vänttinen <sami.vanttinen@protonmail.com>
+*  Copyright (C) 2017 KeePassXC Team <team@keepassxc.org>
+*
+*  This program is free software: you can redistribute it and/or modify
+*  it under the terms of the GNU General Public License as published by
+*  the Free Software Foundation, either version 3 of the License, or
+*  (at your option) any later version.
+*
+*  This program is distributed in the hope that it will be useful,
+*  but WITHOUT ANY WARRANTY; without even the implied warranty of
+*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+*  GNU General Public License for more details.
+*
+*  You should have received a copy of the GNU General Public License
+*  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <QCoreApplication>
+#include <iostream>
+#include "NativeMessagingHost.h"
+
+#ifndef Q_OS_WIN
+#include <initializer_list>
+#include <signal.h>
+#include <unistd.h>
+
+// (C) Gist: https://gist.github.com/azadkuh/a2ac6869661ebd3f8588
+void ignoreUnixSignals(std::initializer_list<int> ignoreSignals) {
+    for (int sig : ignoreSignals) {
+        signal(sig, SIG_IGN);
+    }
+}
+
+void catchUnixSignals(std::initializer_list<int> quitSignals) {
+    auto handler = [](int sig) -> void {
+        std::cerr << sig;
+        QCoreApplication::quit();
+    };
+
+    sigset_t blocking_mask;
+    sigemptyset(&blocking_mask);
+    for (auto sig : quitSignals) {
+        sigaddset(&blocking_mask, sig);
+    }
+
+    struct sigaction sa;
+    sa.sa_handler = handler;
+    sa.sa_mask    = blocking_mask;
+    sa.sa_flags   = 0;
+
+    for (auto sig : quitSignals) {
+        sigaction(sig, &sa, nullptr);
+    }
+}
+#endif
+
+int main(int argc, char *argv[])
+{
+    QCoreApplication a(argc, argv);
+#if defined(Q_OS_UNIX) || defined(Q_OS_LINUX)
+    catchUnixSignals({SIGQUIT, SIGINT, SIGTERM, SIGHUP});
+#endif
+    NativeMessagingHost host;
+    return a.exec();
+}