X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcore%2Fcoretransfer.h;h=c8b83d4958e22b4bcb7d84c3c859f11451b1ac36;hp=9d57db37921c52fb6daf8a8a807e3580564f407a;hb=e3c42d072b2b8f39c8c9ea44dfc3bded87ae43b0;hpb=584085c95a70e921f401b387bb73e66e2d71a714;ds=sidebyside

diff --git a/src/core/coretransfer.h b/src/core/coretransfer.h
index 9d57db37..c8b83d49 100644
--- a/src/core/coretransfer.h
+++ b/src/core/coretransfer.h
@@ -21,7 +21,12 @@
 #ifndef CORETRANSFER_H
 #define CORETRANSFER_H
 
+#include <QPointer>
+
 #include "transfer.h"
+#include "peer.h"
+
+class QTcpSocket;
 
 class CoreTransfer : public Transfer
 {
@@ -32,10 +37,28 @@ public:
     CoreTransfer(Direction direction, const QString &nick, const QString &fileName, const QHostAddress &address, quint16 port, quint64 size = 0, QObject *parent = 0);
 
 public slots:
+    void start();
+
     // called through sync calls
     void requestAccepted(PeerPtr peer);
     void requestRejected(PeerPtr peer);
 
+private slots:
+    void startReceiving();
+    void onDataReceived();
+    void onSocketDisconnected();
+    void onSocketError(QAbstractSocket::SocketError error);
+
+private:
+    void setupConnectionForReceive();
+    bool relayData(const QByteArray &data, bool requireChunkSize);
+    virtual void cleanUp();
+
+    QPointer<Peer> _peer;
+    QTcpSocket *_socket;
+    quint64 _pos;
+    QByteArray _buffer;
+    bool _reading;
 };
 
 #endif