-/***************************************************************************
- * Copyright (C) 2005-2014 by the Quassel Project *
- * devel@quassel-irc.org *
- * *
- * This program is free software; you can redistribute it and/or modify *
- * it under the terms of the GNU General Public License as published by *
- * the Free Software Foundation; either version 2 of the License, or *
- * (at your option) version 3. *
- * *
- * This program is distributed in the hope that it will be useful, *
- * but WITHOUT ANY WARRANTY; without even the implied warranty of *
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
- * GNU General Public License for more details. *
- * *
- * You should have received a copy of the GNU General Public License *
- * along with this program; if not, write to the *
- * Free Software Foundation, Inc., *
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
- ***************************************************************************/
-
-#include <QtEndian>
-
-#include <QCoreApplication>
-#include <QTcpSocket>
-
-#include "coretransfer.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)
-{
-
-}
-
-
-void CoreTransfer::cleanUp()
-{
- if (_socket) {
- _socket->close();
- _socket->deleteLater();
- _socket = 0;
- }
-
- _buffer.clear();
- _reading = false;
-}
-
-
-void CoreTransfer::onSocketDisconnected()
-{
- if (state() == Connecting || state() == Transferring) {
- setError(tr("Socket closed while still transferring!"));
- }
- else
- cleanUp();
-}
-
-
-void CoreTransfer::onSocketError(QAbstractSocket::SocketError error)
-{
- Q_UNUSED(error)
-
- if (state() == Connecting || state() == 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
-
- _peer = peer;
- setState(Pending);
-
- emit accepted(peer);
-
- // FIXME temporary until we have queueing
- start();
-}
-
-
-void CoreTransfer::requestRejected(PeerPtr peer)
-{
- if (_peer || state() != New)
- return;
-
- _peer = peer;
- setState(Rejected);
-
- emit rejected(peer);
-}
-
-
-void CoreTransfer::start()
-{
- if (!_peer || state() != Pending || direction() != Receive)
- return;
-
- setupConnectionForReceive();
-}
-
-
-void CoreTransfer::setupConnectionForReceive()
-{
- if (port() == 0) {
- setError(tr("Reverse DCC not supported yet!"));
- return;
- }
-
- setState(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()));
-
- _socket->connectToHost(address(), port());
-}
-
-
-void CoreTransfer::startReceiving()
-{
- setState(Transferring);
-}
-
-
-void CoreTransfer::onDataReceived()
-{
- 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();
- 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
- 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);
-
- if (_pos > fileSize()) {
- qWarning() << "DCC Receive: Got more data than expected!";
- setError(tr("DCC Receive: Got more data than expected!"));
- }
- else if (_pos == fileSize()) {
- qDebug() << "DCC Receive: Transfer finished";
- if (relayData(QByteArray(), false)) // empty buffer
- setState(Completed);
- }
-
- _reading = false;
-}
-
-
-bool CoreTransfer::relayData(const QByteArray &data, bool requireChunkSize)
-{
- // safeguard against a disconnecting quasselclient
- if (!_peer) {
- setError(tr("DCC Receive: Quassel Client disconnected during transfer!"));
- return false;
- }
- _buffer.append(data);
-
- // 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));
- _buffer.clear();
- }
-
- return true;
-}