1 files changed, 73 insertions, 0 deletions
diff --git a/src/sshagent/SSHAgent.h b/src/sshagent/SSHAgent.h
new file mode 100644
index 000000000..078ff7b0d
--- /dev/null
+++ b/src/sshagent/SSHAgent.h
@@ -0,0 +1,73 @@
+/*
+ *  Copyright (C) 2017 Toni Spets <toni.spets@iki.fi>
+ *  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 2 or (at your option)
+ *  version 3 of the License.
+ *
+ *  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 AGENTCLIENT_H
+#define AGENTCLIENT_H
+
+#include <QtCore>
+#include <QList>
+#include "OpenSSHKey.h"
+
+#include "gui/DatabaseWidget.h"
+
+class SSHAgent : public QObject
+{
+    Q_OBJECT
+
+public:
+    static SSHAgent* instance();
+    static void init(QObject* parent);
+
+    bool isAgentRunning() const;
+    bool addIdentity(OpenSSHKey& key, quint32 lifetime = 0, bool confirm = false) const;
+    bool removeIdentity(OpenSSHKey& key) const;
+    void removeIdentityAtLock(const OpenSSHKey& key, const Uuid& uuid);
+
+public slots:
+    void databaseModeChanged(DatabaseWidget::Mode mode = DatabaseWidget::LockedMode);
+
+private:
+    const quint8 SSH_AGENT_FAILURE              = 5;
+    const quint8 SSH_AGENT_SUCCESS              = 6;
+    const quint8 SSH_AGENTC_REQUEST_IDENTITIES  = 11;
+    const quint8 SSH_AGENT_IDENTITIES_ANSWER    = 12;
+    const quint8 SSH_AGENTC_ADD_IDENTITY        = 17;
+    const quint8 SSH_AGENTC_REMOVE_IDENTITY     = 18;
+    const quint8 SSH_AGENTC_ADD_ID_CONSTRAINED  = 25;
+
+    const quint8 SSH_AGENT_CONSTRAIN_LIFETIME   = 1;
+    const quint8 SSH_AGENT_CONSTRAIN_CONFIRM    = 2;
+
+    explicit SSHAgent(QObject* parent = nullptr);
+    ~SSHAgent();
+
+    bool sendMessage(const QByteArray& in, QByteArray& out) const;
+
+    static SSHAgent* m_instance;
+
+#ifndef Q_OS_WIN
+    QString m_socketPath;
+#else
+    const quint32 AGENT_MAX_MSGLEN = 8192;
+    const quint32 AGENT_COPYDATA_ID = 0x804e50ba;
+#endif
+
+    QMap<QString, QSet<OpenSSHKey>> m_keys;
+};
+
+#endif // AGENTCLIENT_H