src/core/coretransfer.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 CORETRANSFER_H
  22 #define CORETRANSFER_H
  23
  24 #include <QPointer>
  25
  26 #include "transfer.h"
  27 #include "peer.h"
  28
  29 class QTcpSocket;
  30
  31 class CoreTransfer : public Transfer
  32 {
  33     Q_OBJECT
  34     SYNCABLE_OBJECT
  35
  36 public:
  37     CoreTransfer(Direction direction, const QString &nick, const QString &fileName, const QHostAddress &address, quint16 port, quint64 size = 0, QObject *parent = 0);
  38
  39 public slots:
  40     void start();
  41
  42     // called through sync calls
  43     void requestAccepted(PeerPtr peer);
  44     void requestRejected(PeerPtr peer);
  45
  46 private slots:
  47     void startReceiving();
  48     void onDataReceived();
  49     void onSocketDisconnected();
  50     void onSocketError(QAbstractSocket::SocketError error);
  51
  52 private:
  53     void setupConnectionForReceive();
  54     bool relayData(const QByteArray &data, bool requireChunkSize);
  55     virtual void cleanUp();
  56
  57     QPointer<Peer> _peer;
  58     QTcpSocket *_socket;
  59     quint64 _pos;
  60     QByteArray _buffer;
  61     bool _reading;
  62 };
  63
  64 #endif