1 files changed, 68 insertions, 0 deletions
diff --git a/src/sshagent/BinaryStream.h b/src/sshagent/BinaryStream.h
new file mode 100644
index 000000000..c61010180
--- /dev/null
+++ b/src/sshagent/BinaryStream.h
@@ -0,0 +1,68 @@
+/*
+ *  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 BINARYSTREAM_H
+#define BINARYSTREAM_H
+
+#include <QObject>
+#include <QIODevice>
+#include <QBuffer>
+
+class BinaryStream : QObject
+{
+    Q_OBJECT
+public:
+    BinaryStream(QObject* parent = nullptr);
+    BinaryStream(QIODevice* device);
+    BinaryStream(QByteArray* ba, QObject* parent = nullptr);
+    ~BinaryStream();
+
+    const QString errorString() const;
+    QIODevice* device() const;
+    void setDevice(QIODevice* device);
+    void setData(QByteArray* ba);
+    void setTimeout(int timeout);
+
+    bool read(QByteArray& ba);
+    bool read(quint32& i);
+    bool read(quint16& i);
+    bool read(quint8& i);
+    bool readString(QByteArray& ba);
+    bool readString(QString& s);
+
+    bool write(const QByteArray& ba);
+    bool write(quint32 i);
+    bool write(quint16 i);
+    bool write(quint8 i);
+    bool writeString(const QByteArray& ba);
+    bool writeString(const QString& s);
+
+    bool flush();
+
+protected:
+    bool read(char* ptr, qint64 len);
+    bool write(const char* ptr, qint64 len);
+
+private:
+    int m_timeout;
+    QString m_error;
+    QIODevice* m_device;
+    QScopedPointer<QBuffer> m_buffer;
+};
+
+#endif // BINARYSTREAM_H