diff options
| author | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-02-01 17:36:33 +0000 |
|---|---|---|
| committer | Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com> | 2008-02-01 17:36:33 +0000 |
| commit | 0f133632298e31dab0efe6fd36f00c84d2175f33 (patch) | |
| tree | 0e9effc84f05b47e3a6c6ec4a9f7b38391d8be86 /src/DHTNode.h | |
| parent | 0bce06348e3be57c97e246d2a7af71683dc76ede (diff) | |
2008-02-01 Tatsuhiro Tsujikawa <tujikawa at rednoah dot com>
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}
Diffstat (limited to 'src/DHTNode.h')
| -rw-r--r-- | src/DHTNode.h | 123 |
1 files changed, 123 insertions, 0 deletions
diff --git a/src/DHTNode.h b/src/DHTNode.h new file mode 100644 index 00000000..c6af39aa --- /dev/null +++ b/src/DHTNode.h @@ -0,0 +1,123 @@ +/* <!-- copyright */ +/* + * aria2 - The high speed download utility + * + * Copyright (C) 2006 Tatsuhiro Tsujikawa + * + * 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 + * + * In addition, as a special exception, the copyright holders give + * permission to link the code of portions of this program with the + * OpenSSL library under certain conditions as described in each + * individual source file, and distribute linked combinations + * including the two. + * You must obey the GNU General Public License in all respects + * for all of the code used other than OpenSSL. If you modify + * file(s) with this exception, you may extend this exception to your + * version of the file(s), but you are not obligated to do so. If you + * do not wish to do so, delete this exception statement from your + * version. If you delete this exception statement from all source + * files in the program, then also delete it here. + */ +/* copyright --> */ +#ifndef _D_DHT_NODE_H_ +#define _D_DHT_NODE_H_ + +#include "common.h" +#include "DHTConstants.h" +#include "DHTNodeDecl.h" +#include "TimeA2.h" + +class DHTNode { +private: + unsigned char _id[DHT_ID_LENGTH]; + + string _ipaddr; + + uint16_t _port; + + // in milli sec + int32_t _rtt; + + uint32_t _condition; + + Time _lastContact; +public: + DHTNode(); + + /** + * id must be 20 bytes length + */ + DHTNode(const unsigned char* id); + + void generateID(); + + const unsigned char* getID() const + { + return _id; + } + + void updateRTT(int32_t millisec) + { + _rtt = millisec; + } + + string getIPAddress() const + { + return _ipaddr; + } + + void setIPAddress(const string& ipaddr) + { + _ipaddr = ipaddr; + } + + void setID(const unsigned char* id) + { + memcpy(_id, id, DHT_ID_LENGTH); + } + + uint16_t getPort() const + { + return _port; + } + + void setPort(uint16_t port) + { + _port = port; + } + + bool isGood() const; + + bool isBad() const; + + bool isQuestionable() const; + + void updateLastContact(); + + void markGood(); + + void markBad(); + + void timeout(); + + bool operator==(const DHTNode& node) const; + + bool operator<(const DHTNode& node) const; + + string toString() const; +}; + +#endif // _D_DHT_NODE_H_ |
