From 0f133632298e31dab0efe6fd36f00c84d2175f33 Mon Sep 17 00:00:00 2001 From: Tatsuhiro Tsujikawa Date: Fri, 1 Feb 2008 17:36:33 +0000 Subject: 2008-02-01 Tatsuhiro Tsujikawa Added DHT functionality, compatible with mainline. DHT is disabled by default. To enable it, give --enable-dht to aria2c. You may need to specify entry point to DHT network using --dht-entry-point. DHT uses UDP port to listen incoming message. Use --dht-listen-port to specify port number. Make sure that your firewall configuration can pass through UDP traffic to the port. The routing table is saved in $HOME/.aria2/dht.dat. * src/DHT* * src/BNode.{h, cc} * src/PeerInteractionCommand.cc: enable DHT functionality for a particular torrent. * src/Data.cc: Rewritten ctor. * src/OptionHandlerFactory.cc: Added --enable-dht, --dht-listen-port, --dht-entry-point. * src/DefaultBtInteractive.cc: Send port message if dht is enabled. * src/RequestGroup.cc: Initialize DHT functionality. When download ends, remove BtContext from DHTPeerAnnounceStorage. * src/BtPortMessage.{h, cc}: Rewritten. * src/message.h * src/OptionHandlerImpl.cc * src/option_processing.cc: Added --enable-dht, --dht-listen-port, --dht-entry-point. * src/Dictionary.{h, cc} (remove): New function. * src/prefs.h * src/DefaultBtMessageFactory.h * src/BtHandshakeMessage.cc * src/ActivePeerConnectionCommand.cc * src/SocketCore.{h, cc}: Added datagram socket support. * src/DefaultBtMessageFactory.cc * src/BtSetup.cc: Add BtContext to DHTPeerAnnounceStorage here. Create DHT commands. * src/BtMessageFactory.h * src/PeerMessageUtil.{h, cc} --- src/DHTMessageFactory.h | 107 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 107 insertions(+) create mode 100644 src/DHTMessageFactory.h (limited to 'src/DHTMessageFactory.h') diff --git a/src/DHTMessageFactory.h b/src/DHTMessageFactory.h new file mode 100644 index 00000000..59472164 --- /dev/null +++ b/src/DHTMessageFactory.h @@ -0,0 +1,107 @@ +/* */ +#ifndef _D_DHT_MESSAGE_FACTORY_H_ +#define _D_DHT_MESSAGE_FACTORY_H_ + +#include "common.h" +#include "DHTMessageDecl.h" +#include "DHTNodeDecl.h" +#include "PeerDecl.h" + +class Dictionary; + +class DHTMessageFactory { +public: + virtual ~DHTMessageFactory() {} + + virtual DHTMessageHandle + createQueryMessage(const Dictionary* d, + const string& ipaddr, uint16_t port) = 0; + + virtual DHTMessageHandle + createResponseMessage(const string& messageType, + const Dictionary* d, + const DHTNodeHandle& remoteNode) = 0; + + virtual DHTMessageHandle + createPingMessage(const DHTNodeHandle& remoteNode, + const string& transactionID = "") = 0; + + virtual DHTMessageHandle + createPingReplyMessage(const DHTNodeHandle& remoteNode, + const unsigned char* id, + const string& transactionID) = 0; + + virtual DHTMessageHandle + createFindNodeMessage(const DHTNodeHandle& remoteNode, + const unsigned char* targetNodeID, + const string& transactionID = "") = 0; + + virtual DHTMessageHandle + createFindNodeReplyMessage(const DHTNodeHandle& remoteNode, + const DHTNodes& closestKNodes, + const string& transactionID) = 0; + + virtual DHTMessageHandle + createGetPeersMessage(const DHTNodeHandle& remoteNode, + const unsigned char* infoHash, + const string& transactionID = "") = 0; + + virtual DHTMessageHandle + createGetPeersReplyMessage(const DHTNodeHandle& remoteNode, + const DHTNodes& closestKNodes, + const string& token, + const string& transactionID) = 0; + + virtual DHTMessageHandle + createGetPeersReplyMessage(const DHTNodeHandle& remoteNode, + const Peers& peers, + const string& token, + const string& transactionID) = 0; + + virtual DHTMessageHandle + createAnnouncePeerMessage(const DHTNodeHandle& remoteNode, + const unsigned char* infoHash, + uint16_t tcpPort, + const string& token, + const string& transactionID = "") = 0; + + virtual DHTMessageHandle + createAnnouncePeerReplyMessage(const DHTNodeHandle& remoteNode, + const string& transactionID) = 0; + +}; + +#endif // _D_DHT_MESSAGE_FACTORY_H_ -- cgit