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