You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
3.8 KiB
127 lines
3.8 KiB
/***************************************************************************
|
|
* 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
|