Move all socket handling into RemotePeer

[quassel.git] / src / common / remotepeer.h

diff --git a/src/common/remotepeer.h b/src/common/remotepeer.h
index e58d140..b1c5bf4 100644 (file)
--- a/src/common/remotepeer.h
+++ b/src/common/remotepeer.h
@@ -72,6 +72,9 @@ signals:
 protected:
     SignalProxy *signalProxy() const;
 
+    void writeMessage(const QByteArray &msg);
+    virtual void processMessage(const QByteArray &msg) = 0;
+
     // These protocol messages get handled internally and won't reach SignalProxy
     void handle(const Protocol::HeartBeat &heartBeat);
     void handle(const Protocol::HeartBeatReply &heartBeatReply);
@@ -79,20 +82,25 @@ protected:
     virtual void dispatch(const Protocol::HeartBeatReply &msg) = 0;
 
 protected slots:
-    virtual void onSocketDataAvailable() = 0;
     virtual void onSocketStateChanged(QAbstractSocket::SocketState state);
     virtual void onSocketError(QAbstractSocket::SocketError error);
 
 private slots:
+    void onReadyRead();
+
     void sendHeartBeat();
     void changeHeartBeatInterval(int secs);
 
+private:
+    bool readMessage(QByteArray &msg);
+
 private:
     QTcpSocket *_socket;
     SignalProxy *_signalProxy;
     QTimer *_heartBeatTimer;
     int _heartBeatCount;
     int _lag;
+    quint32 _msgSize;
 };
 
 #endif