src/common/authhandler.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include <QHostAddress>
  22
  23 #include "authhandler.h"
  24 #include "util.h"
  25
  26 AuthHandler::AuthHandler(QObject *parent)
  27     : QObject(parent)
  28 {
  29
  30 }
  31
  32
  33 QTcpSocket *AuthHandler::socket() const
  34 {
  35     return _socket;
  36 }
  37
  38
  39 void AuthHandler::setSocket(QTcpSocket *socket)
  40 {
  41     _socket = socket;
  42     connect(socket, selectOverload<QAbstractSocket::SocketError>(&QTcpSocket::error), this, &AuthHandler::onSocketError);
  43     connect(socket, &QAbstractSocket::disconnected, this, &AuthHandler::onSocketDisconnected);
  44 }
  45
  46
  47 bool AuthHandler::isLocal() const
  48 {
  49     if (socket()) {
  50         if (socket()->peerAddress() == QHostAddress::LocalHost || socket()->peerAddress() == QHostAddress::LocalHostIPv6)
  51             return true;
  52     }
  53     return false;
  54 }
  55
  56
  57 // Some errors (e.g. connection refused) don't trigger a disconnected() from the socket, so send this explicitly
  58 // (but make sure it's only sent once!)
  59 void AuthHandler::onSocketError(QAbstractSocket::SocketError error)
  60 {
  61     emit socketError(error, _socket->errorString());
  62
  63     if (!socket()->isOpen() || !socket()->isValid()) {
  64         if (!_disconnectedSent) {
  65             _disconnectedSent = true;
  66             emit disconnected();
  67         }
  68     }
  69 }
  70
  71
  72 void AuthHandler::onSocketDisconnected()
  73 {
  74     if (!_disconnectedSent) {
  75         _disconnectedSent = true;
  76         emit disconnected();
  77     }
  78 }
  79
  80
  81 void AuthHandler::invalidMessage()
  82 {
  83     qWarning() << Q_FUNC_INFO << "No handler for message!";
  84 }
  85
  86
  87 void AuthHandler::close()
  88 {
  89     if (_socket && _socket->isOpen())
  90         _socket->close();
  91 }