X-Git-Url: https://git.quassel-irc.org/?p=quassel.git;a=blobdiff_plain;f=src%2Fcommon%2Fauthhandler.cpp;h=679473013f312fda32c771d7bc3958cb1163d8f8;hp=9877145e8795c6a6a955b90d666e03a3a0ea0357;hb=158443f71d48215eea8b47b836b61afd77654b78;hpb=64cf9f9b8a737dad5f29447805d4004cfd03c454

diff --git a/src/common/authhandler.cpp b/src/common/authhandler.cpp
index 9877145e..67947301 100644
--- a/src/common/authhandler.cpp
+++ b/src/common/authhandler.cpp
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-2013 by the Quassel Project                        *
+ *   Copyright (C) 2005-2018 by the Quassel Project                        *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
@@ -18,50 +18,62 @@
  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  ***************************************************************************/
 
+#include <QHostAddress>
+
 #include "authhandler.h"
 
 AuthHandler::AuthHandler(QObject *parent)
-    : QObject(parent),
-    _state(UnconnectedState),
-    _socket(0)
+    : QObject(parent)
 {
 
 }
 
 
-AuthHandler::State AuthHandler::state() const
+QTcpSocket *AuthHandler::socket() const
 {
-    return _state;
+    return _socket;
 }
 
 
-void AuthHandler::setState(AuthHandler::State state)
+void AuthHandler::setSocket(QTcpSocket *socket)
 {
-    if (_state != state) {
-        _state = state;
-        emit stateChanged(state);
-    }
+    _socket = socket;
+    connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(onSocketError(QAbstractSocket::SocketError)));
+    connect(socket, SIGNAL(disconnected()), SLOT(onSocketDisconnected()));
 }
 
 
-QTcpSocket *AuthHandler::socket() const
+bool AuthHandler::isLocal() const
 {
-    return _socket;
+    if (socket()) {
+        if (socket()->peerAddress() == QHostAddress::LocalHost || socket()->peerAddress() == QHostAddress::LocalHostIPv6)
+            return true;
+    }
+    return false;
 }
 
 
-void AuthHandler::setSocket(QTcpSocket *socket)
+// Some errors (e.g. connection refused) don't trigger a disconnected() from the socket, so send this explicitly
+// (but make sure it's only sent once!)
+void AuthHandler::onSocketError(QAbstractSocket::SocketError error)
 {
-    _socket = socket;
-    connect(socket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SIGNAL(socketStateChanged(QAbstractSocket::SocketState)));
-    connect(socket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(socketError(QAbstractSocket::SocketError)));
-    connect(socket, SIGNAL(disconnected()), SIGNAL(disconnected()));
+    emit socketError(error, _socket->errorString());
+
+    if (!socket()->isOpen() || !socket()->isValid()) {
+        if (!_disconnectedSent) {
+            _disconnectedSent = true;
+            emit disconnected();
+        }
+    }
 }
 
 
-void AuthHandler::socketError(QAbstractSocket::SocketError error)
+void AuthHandler::onSocketDisconnected()
 {
-    emit socketError(error, _socket->errorString());
+    if (!_disconnectedSent) {
+        _disconnectedSent = true;
+        emit disconnected();
+    }
 }