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.
ktorrent/libktorrent/kademlia/dht.h

138 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 DHTDHT_H
#define DHTDHT_H
#include <tqtimer.h>
#include <tqstring.h>
#include <tqmap.h>
#include <util/constants.h>
#include <util/timer.h>
#include "key.h"
#include "dhtbase.h"
namespace bt
{
class SHA1Hash;
}
namespace KNetwork
{
class KInetSocketAddress;
}
namespace dht
{
class Node;
class RPCServer;
class PingReq;
class FindNodeReq;
class FindValueReq;
class StoreValueReq;
class GetPeersReq;
class MsgBase;
class ErrMsg;
class MsgBase;
class AnnounceReq;
class Database;
class TaskManager;
class Task;
class AnnounceTask;
class NodeLookup;
class KBucket;
/**
@author Joris Guisson <joris.guisson@gmail.com>
*/
class DHT : public DHTBase
{
Q_OBJECT
public:
DHT();
virtual ~DHT();
void ping(PingReq* r);
void findNode(FindNodeReq* r);
void response(MsgBase* r);
void getPeers(GetPeersReq* r);
void announce(AnnounceReq* r);
void error(ErrMsg* r);
void timeout(const MsgBase* r);
/**
* A Peer has received a PORT message, and uses this function to alert the DHT of it.
* @param ip The IP of the peer
* @param port The port in the PORT message
*/
void portRecieved(const TQString & ip,bt::Uint16 port);
/**
* Do an announce on the DHT network
* @param info_hash The info_hash
* @param port The port
* @return The task which handles this
*/
AnnounceTask* announce(const bt::SHA1Hash & info_hash,bt::Uint16 port);
/**
* Refresh a bucket using a find node task.
* @param id The id
* @param bucket The bucket to refresh
*/
NodeLookup* refreshBucket(const dht::Key & id,KBucket & bucket);
/**
* Do a NodeLookup.
* @param id The id of the key to search
*/
NodeLookup* findNode(const dht::Key & id);
/// See if it is possible to start a task
bool canStartTask() const;
void start(const TQString & table,const TQString & key_file,bt::Uint16 port);
void stop();
void addDHTNode(const TQString & host,bt::Uint16 hport);
/**
* Returns maxNodes number of <IP address, port> nodes
* that are closest to ourselves and are good.
* @param maxNodes maximum nr of nodes in TQMap to return.
*/
TQMap<TQString, int> getClosestGoodNodes(int maxNodes);
private slots:
void update();
private:
Node* node;
RPCServer* srv;
Database* db;
TaskManager* tman;
bt::Timer expire_timer;
TQString table_file;
TQTimer update_timer;
};
}
#endif