src/common/authhandler.cpp

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