diff options
Diffstat (limited to 'libktorrent/torrent/authenticatebase.h')
| -rw-r--r-- | libktorrent/torrent/authenticatebase.h | 125 |
1 files changed, 125 insertions, 0 deletions
diff --git a/libktorrent/torrent/authenticatebase.h b/libktorrent/torrent/authenticatebase.h new file mode 100644 index 0000000..fdab158 --- /dev/null +++ b/libktorrent/torrent/authenticatebase.h @@ -0,0 +1,125 @@ +/*************************************************************************** + * 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 <qobject.h> +#include <qsocket.h> +#include <qtimer.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 QObject + { + Q_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; + QTimer timer; + bool finished; + Uint8 handshake[68]; + Uint32 bytes_of_handshake_recieved; + Uint32 ext_support; + bool local; + int poll_index; + }; + +} + +#endif |