1 /***************************************************************************
2 * Copyright (C) 2005-2018 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 ***************************************************************************/
23 #include <QCoreApplication>
26 #include "coretransfer.h"
29 const qint64 chunkSize = 16 * 1024;
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),
41 quint64 CoreTransfer::transferred() const
47 void CoreTransfer::cleanUp()
51 _socket->deleteLater();
60 void CoreTransfer::onSocketDisconnected()
62 if (status() == Status::Connecting || status() == Status::Transferring) {
63 setError(tr("Socket closed while still transferring!"));
70 void CoreTransfer::onSocketError(QAbstractSocket::SocketError error)
74 if (status() == Status::Connecting || status() == Status::Transferring) {
75 setError(tr("DCC connection error: %1").arg(_socket->errorString()));
80 void CoreTransfer::requestAccepted(PeerPtr peer)
82 if (_peer || !peer || status() != Status::New)
83 return; // transfer was already accepted
86 setStatus(Status::Pending);
90 // FIXME temporary until we have queueing
95 void CoreTransfer::requestRejected(PeerPtr peer)
97 if (_peer || status() != Status::New)
101 setStatus(Status::Rejected);
107 void CoreTransfer::start()
109 if (!_peer || status() != Status::Pending || direction() != Direction::Receive)
112 setupConnectionForReceive();
116 void CoreTransfer::setupConnectionForReceive()
119 setError(tr("Reverse DCC not supported yet!"));
123 setStatus(Status::Connecting);
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);
131 _socket->connectToHost(address(), port());
135 void CoreTransfer::startReceiving()
137 setStatus(Status::Transferring);
141 void CoreTransfer::onDataReceived()
143 if (_reading) // since we're spinning the event loop, we may get another readyRead() and thus reentrancy
147 while (_socket->bytesAvailable()) {
148 QByteArray data = _socket->read(chunkSize);
150 emit transferredChanged(transferred());
151 if (!relayData(data, true))
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
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);
164 if (_pos > fileSize()) {
165 qWarning() << "DCC Receive: Got more data than expected!";
166 setError(tr("DCC Receive: Got more data than expected!"));
168 else if (_pos == fileSize()) {
169 qDebug() << "DCC Receive: Transfer finished";
170 if (relayData(QByteArray(), false)) // empty buffer
171 setStatus(Status::Completed);
178 bool CoreTransfer::relayData(const QByteArray &data, bool requireChunkSize)
180 // safeguard against a disconnecting quasselclient
182 setError(tr("DCC Receive: Quassel Client disconnected during transfer!"));
185 _buffer.append(data);
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));