src/core/coretransfer.cpp

   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 #include <QtEndian>
  22
  23 #include <QCoreApplication>
  24 #include <QTcpSocket>
  25
  26 #include "coretransfer.h"
  27 #include "util.h"
  28
  29 const qint64 chunkSize = 16 * 1024;
  30
  31 CoreTransfer::CoreTransfer(Direction direction, const QString &nick, const QString &fileName, const QHostAddress &address, quint16 port, quint64 fileSize, QObject *parent)
  32     : Transfer(direction, nick, fileName, address, port, fileSize, parent),
  33     _socket(nullptr),
  34     _pos(0),
  35     _reading(false)
  36 {
  37
  38 }
  39
  40
  41 quint64 CoreTransfer::transferred() const
  42 {
  43     return _pos;
  44 }
  45
  46
  47 void CoreTransfer::cleanUp()
  48 {
  49     if (_socket) {
  50         _socket->close();
  51         _socket->deleteLater();
  52         _socket = nullptr;
  53     }
  54
  55     _buffer.clear();
  56     _reading = false;
  57 }
  58
  59
  60 void CoreTransfer::onSocketDisconnected()
  61 {
  62     if (status() == Status::Connecting || status() == Status::Transferring) {
  63         setError(tr("Socket closed while still transferring!"));
  64     }
  65     else
  66         cleanUp();
  67 }
  68
  69
  70 void CoreTransfer::onSocketError(QAbstractSocket::SocketError error)
  71 {
  72     Q_UNUSED(error)
  73
  74     if (status() == Status::Connecting || status() == Status::Transferring) {
  75         setError(tr("DCC connection error: %1").arg(_socket->errorString()));
  76     }
  77 }
  78
  79
  80 void CoreTransfer::requestAccepted(PeerPtr peer)
  81 {
  82     if (_peer || !peer || status() != Status::New)
  83         return; // transfer was already accepted
  84
  85     _peer = peer;
  86     setStatus(Status::Pending);
  87
  88     emit accepted(peer);
  89
  90     // FIXME temporary until we have queueing
  91     start();
  92 }
  93
  94
  95 void CoreTransfer::requestRejected(PeerPtr peer)
  96 {
  97     if (_peer || status() != Status::New)
  98         return;
  99
 100     _peer = peer;
 101     setStatus(Status::Rejected);
 102
 103     emit rejected(peer);
 104 }
 105
 106
 107 void CoreTransfer::start()
 108 {
 109     if (!_peer || status() != Status::Pending || direction() != Direction::Receive)
 110         return;
 111
 112     setupConnectionForReceive();
 113 }
 114
 115
 116 void CoreTransfer::setupConnectionForReceive()
 117 {
 118     if (port() == 0) {
 119         setError(tr("Reverse DCC not supported yet!"));
 120         return;
 121     }
 122
 123     setStatus(Status::Connecting);
 124
 125     _socket = new QTcpSocket(this);
 126     connect(_socket, &QAbstractSocket::connected, this, &CoreTransfer::startReceiving);
 127     connect(_socket, &QAbstractSocket::disconnected, this, &CoreTransfer::onSocketDisconnected);
 128     connect(_socket, selectOverload<QAbstractSocket::SocketError>(&QAbstractSocket::error), this, &CoreTransfer::onSocketError);
 129     connect(_socket, &QIODevice::readyRead, this, &CoreTransfer::onDataReceived);
 130
 131     _socket->connectToHost(address(), port());
 132 }
 133
 134
 135 void CoreTransfer::startReceiving()
 136 {
 137     setStatus(Status::Transferring);
 138 }
 139
 140
 141 void CoreTransfer::onDataReceived()
 142 {
 143     if (_reading) // since we're spinning the event loop, we may get another readyRead() and thus reentrancy
 144         return;
 145     _reading = true;
 146
 147     while (_socket->bytesAvailable()) {
 148         QByteArray data = _socket->read(chunkSize);
 149         _pos += data.size();
 150         emit transferredChanged(transferred());
 151         if (!relayData(data, true))
 152             return;
 153
 154         QCoreApplication::processEvents();  // don't block the rest of the core/client communication
 155         if (!_socket)  // just in case something happened during spinning the event loop that killed our socket
 156             return;
 157     }
 158
 159     // Send ack to sender. The DCC protocol only specifies 32 bit values, but modern clients (i.e. those who can send files
 160     // larger than 4 GB) will ignore this anyway...
 161     quint32 ack = qToBigEndian((quint32)_pos);// qDebug() << Q_FUNC_INFO << _pos;
 162     _socket->write((char *)&ack, 4);
 163
 164     if (_pos > fileSize()) {
 165         qWarning() << "DCC Receive: Got more data than expected!";
 166         setError(tr("DCC Receive: Got more data than expected!"));
 167     }
 168     else if (_pos == fileSize()) {
 169         qDebug() << "DCC Receive: Transfer finished";
 170         if (relayData(QByteArray(), false)) // empty buffer
 171             setStatus(Status::Completed);
 172     }
 173
 174     _reading = false;
 175 }
 176
 177
 178 bool CoreTransfer::relayData(const QByteArray &data, bool requireChunkSize)
 179 {
 180     // safeguard against a disconnecting quasselclient
 181     if (!_peer) {
 182         setError(tr("DCC Receive: Quassel Client disconnected during transfer!"));
 183         return false;
 184     }
 185     _buffer.append(data);
 186
 187     // we only want to send data to the client once we have reached the chunksize
 188     if (_buffer.size() > 0 && (_buffer.size() >= chunkSize || !requireChunkSize)) {
 189         Peer *p = _peer.data();
 190         SYNC_OTHER(dataReceived, ARG(p), ARG(_buffer));
 191         _buffer.clear();
 192     }
 193
 194     return true;
 195 }