/***************************************************************************
- * Copyright (C) 2005-2013 by the Quassel Project *
+ * Copyright (C) 2005-2014 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 <QHostAddress>
#include <QTimer>
#ifdef HAVE_SSL
# include <QSslSocket>
+#else
+# include <QTcpSocket>
#endif
#include "remotepeer.h"
using namespace Protocol;
-RemotePeer::RemotePeer(QTcpSocket *socket, QObject *parent)
- : Peer(parent),
+const quint32 maxMessageSize = 64 * 1024 * 1024; // This is uncompressed size. 64 MB should be enough for any sort of initData or backlog chunk
+
+RemotePeer::RemotePeer(::AuthHandler *authHandler, QTcpSocket *socket, QObject *parent)
+ : Peer(authHandler, parent),
_socket(socket),
_signalProxy(0),
_heartBeatTimer(new QTimer(this)),
_heartBeatCount(0),
- _lag(0)
+ _lag(0),
+ _msgSize(0)
{
socket->setParent(this);
+ connect(socket, SIGNAL(readyRead()), SLOT(onReadyRead()));
+ connect(socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SLOT(onSocketStateChanged(QAbstractSocket::SocketState)));
+ connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(onSocketError(QAbstractSocket::SocketError)));
connect(socket, SIGNAL(disconnected()), SIGNAL(disconnected()));
- connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), SIGNAL(error(QAbstractSocket::SocketError)));
#ifdef HAVE_SSL
QSslSocket *sslSocket = qobject_cast<QSslSocket *>(socket);
#endif
connect(_heartBeatTimer, SIGNAL(timeout()), SLOT(sendHeartBeat()));
+
+ // It's possible that more data has already arrived during the handshake, so readyRead() wouldn't be triggered.
+ // However, we can't call a virtual function from the ctor, so let's do it asynchronously.
+ if (socket->bytesAvailable())
+ QTimer::singleShot(0, this, SLOT(onReadyRead()));
+}
+
+
+void RemotePeer::onSocketStateChanged(QAbstractSocket::SocketState state)
+{
+ if (state == QAbstractSocket::ClosingState) {
+ emit statusMessage(tr("Disconnecting..."));
+ }
+}
+
+
+void RemotePeer::onSocketError(QAbstractSocket::SocketError error)
+{
+ emit socketError(error, socket()->errorString());
}
}
+void RemotePeer::onReadyRead()
+{
+ // don't try to read more data if we're already closing
+ if (socket()->state() != QAbstractSocket::ConnectedState)
+ return;
+
+ QByteArray msg;
+ while (readMessage(msg))
+ processMessage(msg);
+}
+
+
+bool RemotePeer::readMessage(QByteArray &msg)
+{
+ if (_msgSize == 0) {
+ if (socket()->bytesAvailable() < 4)
+ return false;
+ socket()->read((char*)&_msgSize, 4);
+ _msgSize = qFromBigEndian<quint32>(_msgSize);
+
+ if (_msgSize > maxMessageSize) {
+ close("Peer tried to send package larger than max package size!");
+ return false;
+ }
+
+ if (_msgSize == 0) {
+ close("Peer tried to send an empty message!");
+ return false;
+ }
+ }
+
+ if (socket()->bytesAvailable() < _msgSize) {
+ emit transferProgress(socket()->bytesAvailable(), _msgSize);
+ return false;
+ }
+
+ emit transferProgress(_msgSize, _msgSize);
+
+ msg.resize(_msgSize);
+ qint64 bytesRead = socket()->read(msg.data(), _msgSize);
+ if (bytesRead != _msgSize) {
+ close("Premature end of data stream!");
+ return false;
+ }
+
+ _msgSize = 0;
+ return true;
+}
+
+
+void RemotePeer::writeMessage(const QByteArray &msg)
+{
+ quint32 size = qToBigEndian<quint32>(msg.size());
+ socket()->write((const char*)&size, 4);
+ socket()->write(msg.constData(), msg.size());
+}
+
+
void RemotePeer::handle(const HeartBeat &heartBeat)
{
- dispatch(HeartBeatReply(heartBeat.timestamp()));
+ dispatch(HeartBeatReply(heartBeat.timestamp));
}
void RemotePeer::handle(const HeartBeatReply &heartBeatReply)
{
_heartBeatCount = 0;
- emit lagUpdated(heartBeatReply.timestamp().msecsTo(QDateTime::currentDateTime().toUTC()) / 2);
+#if QT_VERSION >= 0x040700
+ emit lagUpdated(heartBeatReply.timestamp.msecsTo(QDateTime::currentDateTime().toUTC()) / 2);
+#else
+ emit lagUpdated(heartBeatReply.timestamp.time().msecsTo(QDateTime::currentDateTime().toUTC().time()) / 2);
+#endif
}