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 | #ifndef AUTHHANDLER_H |
22 | #define AUTHHANDLER_H |
23 | |
24 | #include <QTcpSocket> |
25 | |
26 | #include "protocol.h" |
27 | |
28 | class Peer; |
29 | |
30 | class AuthHandler : public QObject |
31 | { |
32 | Q_OBJECT |
33 | |
34 | public: |
35 | AuthHandler(QObject *parent = 0); |
36 | |
37 | QTcpSocket *socket() const; |
38 | |
39 | bool isLocal() const; |
40 | |
41 | virtual void handle(const Protocol::RegisterClient &) { invalidMessage(); } |
42 | virtual void handle(const Protocol::ClientDenied &) { invalidMessage(); } |
43 | virtual void handle(const Protocol::ClientRegistered &) { invalidMessage(); } |
44 | virtual void handle(const Protocol::SetupData &) { invalidMessage(); } |
45 | virtual void handle(const Protocol::SetupFailed &) { invalidMessage(); } |
46 | virtual void handle(const Protocol::SetupDone &) { invalidMessage(); } |
47 | virtual void handle(const Protocol::Login &) { invalidMessage(); } |
48 | virtual void handle(const Protocol::LoginFailed &) { invalidMessage(); } |
49 | virtual void handle(const Protocol::LoginSuccess &) { invalidMessage(); } |
50 | virtual void handle(const Protocol::SessionState &) { invalidMessage(); } |
51 | |
52 | // fallback for unknown types, will trigger an error |
53 | template<class T> |
54 | void handle(const T &) { invalidMessage(); } |
55 | |
56 | public slots: |
57 | void close(); |
58 | |
59 | signals: |
60 | void disconnected(); |
61 | void socketError(QAbstractSocket::SocketError error, const QString &errorString); |
62 | |
63 | protected: |
64 | void setSocket(QTcpSocket *socket); |
65 | |
66 | protected slots: |
67 | virtual void onSocketError(QAbstractSocket::SocketError error); |
68 | virtual void onSocketDisconnected(); |
69 | |
70 | private: |
71 | void invalidMessage(); |
72 | |
73 | QTcpSocket *_socket; // FIXME: should be a QSharedPointer? -> premature disconnect before the peer has taken over |
74 | bool _disconnectedSent; |
75 | }; |
76 | |
77 | #endif |
78 | |