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.
141 lines
4.1 KiB
141 lines
4.1 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 BTUDPTRACKERSOCKET_H
|
|
#define BTUDPTRACKERSOCKET_H
|
|
|
|
#include <tqobject.h>
|
|
#include <tqmap.h>
|
|
#include <tqcstring.h>
|
|
#include <util/constants.h>
|
|
|
|
|
|
namespace KNetwork
|
|
{
|
|
class KDatagramSocket;
|
|
class TDESocketAddress;
|
|
}
|
|
|
|
namespace bt
|
|
{
|
|
|
|
|
|
enum Action
|
|
{
|
|
CONNECT = 0,
|
|
ANNOUNCE = 1,
|
|
SCRAPE = 2,
|
|
ERROR = 3
|
|
};
|
|
|
|
|
|
|
|
/**
|
|
* @author Joris Guisson
|
|
*
|
|
* Class which handles communication with one or more UDP trackers.
|
|
*/
|
|
class UDPTrackerSocket : public TQObject
|
|
{
|
|
TQ_OBJECT
|
|
|
|
public:
|
|
UDPTrackerSocket();
|
|
virtual ~UDPTrackerSocket();
|
|
|
|
/**
|
|
* Send a connect message. As a response to this, the connectRecieved
|
|
* signal will be emitted, classes recieving this signal should check if
|
|
* the transaction_id is the same.
|
|
* @param tid The transaction_id
|
|
* @param addr The address to send to
|
|
*/
|
|
void sendConnect(Int32 tid,const KNetwork::TDESocketAddress & addr);
|
|
|
|
/**
|
|
* Send an announce message. As a response to this, the announceRecieved
|
|
* signal will be emitted, classes recieving this signal should check if
|
|
* the transaction_id is the same.
|
|
* @param tid The transaction_id
|
|
* @param data The data to send (connect input structure, in UDP Tracker specifaction)
|
|
* @param addr The address to send to
|
|
*/
|
|
void sendAnnounce(Int32 tid,const Uint8* data,const KNetwork::TDESocketAddress & addr);
|
|
|
|
/**
|
|
* If a transaction times out, this should be used to cancel it.
|
|
* @param tid
|
|
*/
|
|
void cancelTransaction(Int32 tid);
|
|
|
|
|
|
/**
|
|
* Compute a free transaction_id.
|
|
* @return A free transaction_id
|
|
*/
|
|
Int32 newTransactionID();
|
|
|
|
/**
|
|
* Set the port ot use.
|
|
* @param p The port
|
|
*/
|
|
static void setPort(Uint16 p);
|
|
|
|
/// Get the port in use.
|
|
static Uint16 getPort();
|
|
private slots:
|
|
void dataReceived();
|
|
|
|
signals:
|
|
/**
|
|
* Emitted when a connect message is received.
|
|
* @param tid The transaction_id
|
|
* @param connection_id The connection_id returned
|
|
*/
|
|
void connectRecieved(Int32 tid,Int64 connection_id);
|
|
|
|
/**
|
|
* Emitted when an announce message is received.
|
|
* @param tid The transaction_id
|
|
* @param buf The data
|
|
*/
|
|
void announceRecieved(Int32 tid,const TQByteArray & buf);
|
|
|
|
/**
|
|
* Signal emitted, when an error occurs during a transaction.
|
|
* @param tid The transaction_id
|
|
* @param error_string Potential error string
|
|
*/
|
|
void error(Int32 tid,const TQString & error_string);
|
|
|
|
private:
|
|
void handleConnect(const TQByteArray & buf);
|
|
void handleAnnounce(const TQByteArray & buf);
|
|
void handleError(const TQByteArray & buf);
|
|
|
|
private:
|
|
Uint16 udp_port;
|
|
KNetwork::KDatagramSocket* sock;
|
|
TQMap<Int32,Action> transactions;
|
|
static Uint16 port;
|
|
};
|
|
}
|
|
|
|
#endif
|