src/common/protocols/datastream/datastreampeer.h

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2014 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 <QDataStream>
  25
  26 #include "../../remotepeer.h"
  27
  28 class QDataStream;
  29
  30 class DataStreamPeer : public RemotePeer
  31 {
  32     Q_OBJECT
  33
  34 public:
  35     enum RequestType {
  36         Sync = 1,
  37         RpcCall,
  38         InitRequest,
  39         InitData,
  40         HeartBeat,
  41         HeartBeatReply
  42     };
  43
  44     DataStreamPeer(AuthHandler *authHandler, QTcpSocket *socket, quint16 features, QObject *parent = 0);
  45
  46     Protocol::Type protocol() const { return Protocol::DataStreamProtocol; }
  47
  48     static quint16 supportedFeatures();
  49     static bool acceptsFeatures(quint16 peerFeatures);
  50     quint16 enabledFeatures() const;
  51
  52     void dispatch(const Protocol::RegisterClient &msg);
  53     void dispatch(const Protocol::ClientDenied &msg);
  54     void dispatch(const Protocol::ClientRegistered &msg);
  55     void dispatch(const Protocol::SetupData &msg);
  56     void dispatch(const Protocol::SetupFailed &msg);
  57     void dispatch(const Protocol::SetupDone &msg);
  58     void dispatch(const Protocol::Login &msg);
  59     void dispatch(const Protocol::LoginFailed &msg);
  60     void dispatch(const Protocol::LoginSuccess &msg);
  61     void dispatch(const Protocol::SessionState &msg);
  62
  63     void dispatch(const Protocol::SyncMessage &msg);
  64     void dispatch(const Protocol::RpcCall &msg);
  65     void dispatch(const Protocol::InitRequest &msg);
  66     void dispatch(const Protocol::InitData &msg);
  67
  68     void dispatch(const Protocol::HeartBeat &msg);
  69     void dispatch(const Protocol::HeartBeatReply &msg);
  70
  71 signals:
  72     void protocolError(const QString &errorString);
  73
  74 protected slots:
  75     void onSocketDataAvailable();
  76
  77 private:
  78     bool readSocketData(QByteArray &data);
  79     void writeSocketData(const QVariantList &list);
  80     void writeSocketData(const QVariantMap &handshakeMsg);
  81
  82     void handleHandshakeMessage(const QVariantList &mapData);
  83     void handlePackedFunc(const QVariantList &packedFunc);
  84     void dispatchPackedFunc(const QVariantList &packedFunc);
  85
  86     QDataStream _stream;
  87     quint32 _blockSize;
  88 };
  89
  90 #endif