/***************************************************************************
- * Copyright (C) 2005-2015 by the Quassel Project *
+ * Copyright (C) 2005-2020 by the Quassel Project *
* devel@quassel-irc.org *
* *
* This program is free software; you can redistribute it and/or modify *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
-#include <QtEndian>
+#include "coretransfer.h"
#include <QCoreApplication>
#include <QTcpSocket>
+#include <QtEndian>
-#include "coretransfer.h"
+#include "util.h"
const qint64 chunkSize = 16 * 1024;
-INIT_SYNCABLE_OBJECT(CoreTransfer)
-
-CoreTransfer::CoreTransfer(Direction direction, const QString &nick, const QString &fileName, const QHostAddress &address, quint16 port, quint64 fileSize, QObject *parent)
- : Transfer(direction, nick, fileName, address, port, fileSize, parent),
- _socket(0),
- _pos(0),
- _reading(false)
+CoreTransfer::CoreTransfer(Direction direction,
+ const QString& nick,
+ const QString& fileName,
+ const QHostAddress& address,
+ quint16 port,
+ quint64 fileSize,
+ QObject* parent)
+ : Transfer(direction, nick, fileName, address, port, fileSize, parent)
+ , _socket(nullptr)
+ , _pos(0)
+ , _reading(false)
+{}
+
+quint64 CoreTransfer::transferred() const
{
-
+ return _pos;
}
-
void CoreTransfer::cleanUp()
{
if (_socket) {
_socket->close();
_socket->deleteLater();
- _socket = 0;
+ _socket = nullptr;
}
_buffer.clear();
_reading = false;
}
-
void CoreTransfer::onSocketDisconnected()
{
- if (state() == Connecting || state() == Transferring) {
+ if (status() == Status::Connecting || status() == Status::Transferring) {
setError(tr("Socket closed while still transferring!"));
}
else
cleanUp();
}
-
void CoreTransfer::onSocketError(QAbstractSocket::SocketError error)
{
Q_UNUSED(error)
- if (state() == Connecting || state() == Transferring) {
+ if (status() == Status::Connecting || status() == Status::Transferring) {
setError(tr("DCC connection error: %1").arg(_socket->errorString()));
}
}
-
void CoreTransfer::requestAccepted(PeerPtr peer)
{
- if (_peer || !peer || state() != New)
- return; // transfer was already accepted
+ if (_peer || !peer || status() != Status::New)
+ return; // transfer was already accepted
_peer = peer;
- setState(Pending);
+ setStatus(Status::Pending);
emit accepted(peer);
start();
}
-
void CoreTransfer::requestRejected(PeerPtr peer)
{
- if (_peer || state() != New)
+ if (_peer || status() != Status::New)
return;
_peer = peer;
- setState(Rejected);
+ setStatus(Status::Rejected);
emit rejected(peer);
}
-
void CoreTransfer::start()
{
- if (!_peer || state() != Pending || direction() != Receive)
+ if (!_peer || status() != Status::Pending || direction() != Direction::Receive)
return;
setupConnectionForReceive();
}
-
void CoreTransfer::setupConnectionForReceive()
{
if (port() == 0) {
return;
}
- setState(Connecting);
+ setStatus(Status::Connecting);
_socket = new QTcpSocket(this);
- connect(_socket, SIGNAL(connected()), SLOT(startReceiving()));
- connect(_socket, SIGNAL(disconnected()), SLOT(onSocketDisconnected()));
- connect(_socket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(onSocketError(QAbstractSocket::SocketError)));
- connect(_socket, SIGNAL(readyRead()), SLOT(onDataReceived()));
+ connect(_socket, &QAbstractSocket::connected, this, &CoreTransfer::startReceiving);
+ connect(_socket, &QAbstractSocket::disconnected, this, &CoreTransfer::onSocketDisconnected);
+ connect(_socket, selectOverload<QAbstractSocket::SocketError>(&QAbstractSocket::error), this, &CoreTransfer::onSocketError);
+ connect(_socket, &QIODevice::readyRead, this, &CoreTransfer::onDataReceived);
_socket->connectToHost(address(), port());
}
-
void CoreTransfer::startReceiving()
{
- setState(Transferring);
+ setStatus(Status::Transferring);
}
-
void CoreTransfer::onDataReceived()
{
- if (_reading) // since we're spinning the event loop, we may get another readyRead() and thus reentrancy
+ if (_reading) // since we're spinning the event loop, we may get another readyRead() and thus reentrancy
return;
_reading = true;
while (_socket->bytesAvailable()) {
QByteArray data = _socket->read(chunkSize);
_pos += data.size();
+ emit transferredChanged(transferred());
if (!relayData(data, true))
return;
QCoreApplication::processEvents(); // don't block the rest of the core/client communication
- if (!_socket) // just in case something happened during spinning the event loop that killed our socket
+ if (!_socket) // just in case something happened during spinning the event loop that killed our socket
return;
}
// Send ack to sender. The DCC protocol only specifies 32 bit values, but modern clients (i.e. those who can send files
// larger than 4 GB) will ignore this anyway...
- quint32 ack = qToBigEndian((quint32)_pos);// qDebug() << Q_FUNC_INFO << _pos;
- _socket->write((char *)&ack, 4);
+ quint32 ack = qToBigEndian((quint32)_pos); // qDebug() << Q_FUNC_INFO << _pos;
+ _socket->write((char*)&ack, 4);
if (_pos > fileSize()) {
qWarning() << "DCC Receive: Got more data than expected!";
}
else if (_pos == fileSize()) {
qDebug() << "DCC Receive: Transfer finished";
- if (relayData(QByteArray(), false)) // empty buffer
- setState(Completed);
+ if (relayData(QByteArray(), false)) // empty buffer
+ setStatus(Status::Completed);
}
_reading = false;
}
-
-bool CoreTransfer::relayData(const QByteArray &data, bool requireChunkSize)
+bool CoreTransfer::relayData(const QByteArray& data, bool requireChunkSize)
{
// safeguard against a disconnecting quasselclient
if (!_peer) {
// we only want to send data to the client once we have reached the chunksize
if (_buffer.size() > 0 && (_buffer.size() >= chunkSize || !requireChunkSize)) {
- SYNC_OTHER(dataReceived, ARG(_peer), ARG(_buffer));
+ Peer* p = _peer.data();
+ SYNC_OTHER(dataReceived, ARG(p), ARG(_buffer));
_buffer.clear();
}