Refactor SignalProxy, network and protocol code

[quassel.git] / src / common / remoteconnection.cpp

diff --git a/src/common/remoteconnection.cpp b/src/common/remoteconnection.cpp
new file mode 100644 (file)
index 0000000..4b73768
--- /dev/null
+++ b/src/common/remoteconnection.cpp
@@ -0,0 +1,190 @@
+/***************************************************************************
+ *   Copyright (C) 2005-2012 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 <QHostAddress>
+#include <QTimer>
+
+#ifdef HAVE_SSL
+#  include <QSslSocket>
+#endif
+
+#include "remoteconnection.h"
+
+RemoteConnection::RemoteConnection(QTcpSocket *socket, QObject *parent)
+    : SignalProxy::AbstractPeer(parent),
+    _socket(socket),
+    _signalProxy(0),
+    _heartBeatTimer(new QTimer(this)),
+    _heartBeatCount(0),
+    _lag(0)
+{
+    socket->setParent(this);
+    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);
+    if (sslSocket)
+        connect(sslSocket, SIGNAL(encrypted()), SIGNAL(secureStateChanged()));
+#endif
+
+    connect(_heartBeatTimer, SIGNAL(timeout()), SLOT(sendHeartBeat()));
+}
+
+
+QString RemoteConnection::description() const
+{
+    if (socket())
+        return socket()->peerAddress().toString();
+
+    return QString();
+}
+
+
+SignalProxy *RemoteConnection::signalProxy() const
+{
+    return _signalProxy;
+}
+
+
+void RemoteConnection::setSignalProxy(SignalProxy *proxy)
+{
+    if (proxy == _signalProxy)
+        return;
+
+    if (!proxy) {
+        _heartBeatTimer->stop();
+        disconnect(signalProxy(), 0, this, 0);
+        _signalProxy = 0;
+        if (isOpen())
+            close();
+    }
+    else {
+        if (signalProxy()) {
+            qWarning() << Q_FUNC_INFO << "Setting another SignalProxy not supported, ignoring!";
+            return;
+        }
+        _signalProxy = proxy;
+        connect(proxy, SIGNAL(heartBeatIntervalChanged(int)), SLOT(changeHeartBeatInterval(int)));
+        _heartBeatTimer->setInterval(proxy->heartBeatInterval() * 1000);
+        _heartBeatTimer->start();
+    }
+}
+
+
+void RemoteConnection::changeHeartBeatInterval(int secs)
+{
+    if(secs <= 0)
+        _heartBeatTimer->stop();
+    else {
+        _heartBeatTimer->setInterval(secs * 1000);
+        _heartBeatTimer->start();
+    }
+}
+
+
+int RemoteConnection::lag() const
+{
+    return _lag;
+}
+
+
+QTcpSocket *RemoteConnection::socket() const
+{
+    return _socket;
+}
+
+
+bool RemoteConnection::isSecure() const
+{
+    if (socket()) {
+        if (isLocal())
+            return true;
+#ifdef HAVE_SSL
+        QSslSocket *sslSocket = qobject_cast<QSslSocket *>(socket());
+        if (sslSocket && sslSocket->isEncrypted())
+            return true;
+#endif
+    }
+    return false;
+}
+
+
+bool RemoteConnection::isLocal() const
+{
+    if (socket()) {
+        if (socket()->peerAddress() == QHostAddress::LocalHost || socket()->peerAddress() == QHostAddress::LocalHostIPv6)
+            return true;
+    }
+    return false;
+}
+
+
+bool RemoteConnection::isOpen() const
+{
+    return socket() && socket()->state() == QTcpSocket::ConnectedState;
+}
+
+
+void RemoteConnection::close(const QString &reason)
+{
+    if (!reason.isEmpty()) {
+        qWarning() << "Disconnecting:" << reason;
+    }
+
+    if (socket() && socket()->state() != QTcpSocket::UnconnectedState) {
+        socket()->disconnectFromHost();
+    }
+}
+
+
+void RemoteConnection::handle(const HeartBeat &heartBeat)
+{
+    dispatch(HeartBeatReply(heartBeat.timestamp()));
+}
+
+
+void RemoteConnection::handle(const HeartBeatReply &heartBeatReply)
+{
+    _heartBeatCount = 0;
+    emit lagUpdated(heartBeatReply.timestamp().msecsTo(QDateTime::currentDateTimeUtc()));
+}
+
+
+void RemoteConnection::sendHeartBeat()
+{
+    if (signalProxy()->maxHeartBeatCount() > 0 && _heartBeatCount >= signalProxy()->maxHeartBeatCount()) {
+        qWarning() << "Disconnecting peer:" << description()
+                   << "(didn't receive a heartbeat for over" << _heartBeatCount *_heartBeatTimer->interval() / 1000 << "seconds)";
+        socket()->close();
+        _heartBeatTimer->stop();
+        return;
+    }
+
+    if (_heartBeatCount > 0) {
+        _lag = _heartBeatCount * _heartBeatTimer->interval();
+        emit lagUpdated(_lag);
+    }
+
+    dispatch(HeartBeat(QDateTime::currentDateTimeUtc()));
+    ++_heartBeatCount;
+}
+
+