src/common/authhandler.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2020 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 "authhandler.h"
  22
  23 #include <QHostAddress>
  24
  25 #include "util.h"
  26
  27 AuthHandler::AuthHandler(QObject* parent)
  28     : QObject(parent)
  29 {}
  30
  31 QTcpSocket* AuthHandler::socket() const
  32 {
  33     return _socket;
  34 }
  35
  36 void AuthHandler::setSocket(QTcpSocket* socket)
  37 {
  38     _socket = socket;
  39     connect(socket, selectOverload<QAbstractSocket::SocketError>(&QTcpSocket::error), this, &AuthHandler::onSocketError);
  40     connect(socket, &QAbstractSocket::disconnected, this, &AuthHandler::onSocketDisconnected);
  41 }
  42
  43 bool AuthHandler::isLocal() const
  44 {
  45     if (socket()) {
  46         if (socket()->peerAddress() == QHostAddress::LocalHost || socket()->peerAddress() == QHostAddress::LocalHostIPv6)
  47             return true;
  48     }
  49     return false;
  50 }
  51
  52 // Some errors (e.g. connection refused) don't trigger a disconnected() from the socket, so send this explicitly
  53 // (but make sure it's only sent once!)
  54 void AuthHandler::onSocketError(QAbstractSocket::SocketError error)
  55 {
  56     emit socketError(error, _socket->errorString());
  57
  58     if (!socket()->isOpen() || !socket()->isValid()) {
  59         if (!_disconnectedSent) {
  60             _disconnectedSent = true;
  61             emit disconnected();
  62         }
  63     }
  64 }
  65
  66 void AuthHandler::onSocketDisconnected()
  67 {
  68     if (!_disconnectedSent) {
  69         _disconnectedSent = true;
  70         emit disconnected();
  71     }
  72 }
  73
  74 void AuthHandler::invalidMessage()
  75 {
  76     qWarning() << Q_FUNC_INFO << "No handler for message!";
  77 }
  78
  79 void AuthHandler::close()
  80 {
  81     if (_socket && _socket->isOpen())
  82         _socket->close();
  83 }