src/common/protocols/datastream/datastreampeer.h

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #ifndef DATASTREAMPEER_H
  22 #define DATASTREAMPEER_H
  23
  24 #include "../../remotepeer.h"
  25
  26 class QDataStream;
  27
  28 class DataStreamPeer : public RemotePeer
  29 {
  30     Q_OBJECT
  31
  32 public:
  33     enum RequestType
  34     {
  35         Sync = 1,
  36         RpcCall,
  37         InitRequest,
  38         InitData,
  39         HeartBeat,
  40         HeartBeatReply
  41     };
  42
  43     DataStreamPeer(AuthHandler* authHandler, QTcpSocket* socket, quint16 features, Compressor::CompressionLevel level, QObject* parent = nullptr);
  44
  45     Protocol::Type protocol() const override { return Protocol::DataStreamProtocol; }
  46     QString protocolName() const override { return "the DataStream protocol"; }
  47
  48     static quint16 supportedFeatures();
  49     static bool acceptsFeatures(quint16 peerFeatures);
  50     quint16 enabledFeatures() const override;
  51
  52     void dispatch(const Protocol::RegisterClient& msg) override;
  53     void dispatch(const Protocol::ClientDenied& msg) override;
  54     void dispatch(const Protocol::ClientRegistered& msg) override;
  55     void dispatch(const Protocol::SetupData& msg) override;
  56     void dispatch(const Protocol::SetupFailed& msg) override;
  57     void dispatch(const Protocol::SetupDone& msg) override;
  58     void dispatch(const Protocol::Login& msg) override;
  59     void dispatch(const Protocol::LoginFailed& msg) override;
  60     void dispatch(const Protocol::LoginSuccess& msg) override;
  61     void dispatch(const Protocol::SessionState& msg) override;
  62
  63     void dispatch(const Protocol::SyncMessage& msg) override;
  64     void dispatch(const Protocol::RpcCall& msg) override;
  65     void dispatch(const Protocol::InitRequest& msg) override;
  66     void dispatch(const Protocol::InitData& msg) override;
  67
  68     void dispatch(const Protocol::HeartBeat& msg) override;
  69     void dispatch(const Protocol::HeartBeatReply& msg) override;
  70
  71 signals:
  72     void protocolError(const QString& errorString);
  73
  74 private:
  75     using RemotePeer::writeMessage;
  76     void writeMessage(const QVariantMap& handshakeMsg);
  77     void writeMessage(const QVariantList& sigProxyMsg);
  78     void processMessage(const QByteArray& msg) override;
  79
  80     void handleHandshakeMessage(const QVariantList& mapData);
  81     void handlePackedFunc(const QVariantList& packedFunc);
  82     void dispatchPackedFunc(const QVariantList& packedFunc);
  83 };
  84
  85 #endif