1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
/***************************************************************************
* Copyright (C) 2005 by Joris Guisson *
* joris.guisson@gmail.com *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#ifndef BTAUTHENTICATEBASE_H
#define BTAUTHENTICATEBASE_H
#include <tqobject.h>
#include <tqsocket.h>
#include <tqtimer.h>
#include <util/constants.h>
namespace mse
{
class StreamSocket;
}
namespace bt
{
class SHA1Hash;
class PeerID;
/**
* @author Joris Guisson
*
* Base class for authentication classes. This class just groups
* some common stuff between Authenticate and ServerAuthentciate.
* It has a socket, handles the timing out, provides a function to send
* the handshake.
*/
class AuthenticateBase : public TQObject
{
TQ_OBJECT
public:
AuthenticateBase(mse::StreamSocket* s = 0);
virtual ~AuthenticateBase();
/// Set wether this is a local peer
void setLocal(bool loc) {local = loc;}
/// Is this a local peer
bool isLocal() const {return local;}
/// See if the authentication is finished
bool isFinished() const {return finished;}
/// Flags indicating which extensions are supported
Uint32 supportedExtensions() const {return ext_support;}
/// get teh socket
const mse::StreamSocket* getSocket() const {return sock;}
/// We can read from the socket
virtual void onReadyRead();
/// We can write to the socket (used to detect a succesfull connection)
virtual void onReadyWrite();
int getPollIndex() const {return poll_index;}
void setPollIndex(int pi) {poll_index = pi;}
protected:
/**
* Send a handshake
* @param info_hash The info_hash to include
* @param our_peer_id Our PeerID
*/
void sendHandshake(const SHA1Hash & info_hash,const PeerID & our_peer_id);
/**
* Authentication finished.
* @param succes Succes or not
*/
virtual void onFinish(bool succes) = 0;
/**
* The other side send a handshake. The first 20 bytes
* of the handshake will already have been checked.
* @param full Indicates wether we have a full handshake
* if this is not full, we should just send our own
*/
virtual void handshakeRecieved(bool full) = 0;
/**
* Fill in the handshake in a buffer.
*/
void makeHandshake(bt::Uint8* buf,const SHA1Hash & info_hash,const PeerID & our_peer_id);
protected slots:
void onTimeout();
void onError(int err);
protected:
mse::StreamSocket* sock;
TQTimer timer;
bool finished;
Uint8 handshake[68];
Uint32 bytes_of_handshake_recieved;
Uint32 ext_support;
bool local;
int poll_index;
};
}
#endif
|
