From 286f34cb3f350c9884a96d5ea3e20e2a3bbf0c37 Mon Sep 17 00:00:00 2001 From: Tatsuhiro Tsujikawa Date: Sat, 22 Dec 2007 03:57:55 +0000 Subject: 2007-12-22 Tatsuhiro Tsujikawa Added uTorrent compatible Peer Exchange. * src/BencodeVisitor.{h, cc} * test/BencodeVisitorTest.cc * src/BtConstants.h * src/BtContext.h: Added 'private' flag. * src/BtExtendedMessage.{h, cc} * test/BtExtendedMessageTest.cc * src/BtHandshakeMessage.{h, cc}: Set extended messaging bit in reserved field. * test/BtHandshakeMessageTest.cc * src/BtMessageFactory.h * src/BtRegistry.h * src/BtRuntime.h: This class holds default extension message IDs for aria2. By default, aria2 uses ID 8 for ut_pex. * src/DefaultBtContext.cc * src/DefaultBtInteractive.{h, cc}: This class holds _utPexEnabled. When it is true, aria2 enables ut_pex. This value is set by PeerInteractionCommand. * src/DefaultBtMessageFactory.{h, cc} * test/DefaultBtMessageFactoryTest.cc * src/DefaultBtMessageReceiver.cc: Moved the code of fast extension handling to DefaultBtInteractive class. * src/DefaultExtensionMessageFactory.{h, cc} * test/DefaultExtensionMessageFactoryTest.cc * src/DefaultPeerStorage.cc: Returns false if a peer is already in the container(peers and incomingPeers. The equality is determined by Peer::id). * test/DefaultPeerStorageTest.cc * src/ExtensionMessage.h * test/MockExtensionMessage.h * src/ExtensionMessageFactory.h * test/MockExtensionMessageFactory.h * src/HandshakeExtensionMessage.{h, cc} * test/HandshakeExtensionMessageTest.cc * src/MetaEntry.h * src/Peer.{h, cc} * src/PeerInteractionCommand.cc * src/PeerReceiveHandshakeCommand.cc: Evaluate the return value of addIncomingPeer. * src/PeerMessageUtil.{h, cc} * src/PeerObject.h * src/UTPexExtensionMessage.{h, cc} * test/UTPexExtensionMessageTest.cc * src/message.h * src/prefs.h Fixed the bug that returns incomplete data when it contains null character. A convenient constructor was also added. * src/Data.{h, cc} Rewritten. * src/CompactPeerListProcessor.cc Fixed typos. * src/message.h * src/MetaFileUtil.cc --- src/BencodeVisitor.cc | 84 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 84 insertions(+) create mode 100644 src/BencodeVisitor.cc (limited to 'src/BencodeVisitor.cc') diff --git a/src/BencodeVisitor.cc b/src/BencodeVisitor.cc new file mode 100644 index 00000000..32a11308 --- /dev/null +++ b/src/BencodeVisitor.cc @@ -0,0 +1,84 @@ +/* */ +#include "BencodeVisitor.h" +#include "Data.h" +#include "List.h" +#include "Dictionary.h" +#include "Util.h" + +BencodeVisitor::BencodeVisitor() {} + +BencodeVisitor::~BencodeVisitor() {} + +void BencodeVisitor::visit(const Data* d) +{ + if(d->isNumber()) { + _bencodedData += "i"+d->toString()+"e"; + } else { + _bencodedData += Util::itos(d->getLen())+":"+d->toString(); + } +} + +void BencodeVisitor::visit(const List* l) +{ + _bencodedData += "l"; + for_each(l->getList().begin(), l->getList().end(), + bind2nd(mem_fun(&MetaEntry::accept), this)); + _bencodedData += "e"; +} + +void BencodeVisitor::visit(const Dictionary* d) +{ + _bencodedData += "d"; + + for(Order::const_iterator itr = d->getOrder().begin(); itr != d->getOrder().end(); ++itr) { + _bencodedData += Util::itos((int32_t)(*itr).size()); + _bencodedData += ":"; + _bencodedData += *itr; + d->get(*itr)->accept(this); + } + _bencodedData += "e"; +} + +void BencodeVisitor::visit(const MetaEntry* e) +{ + if(dynamic_cast(e) != 0) { + visit((const Data*)e); + } else if(dynamic_cast(e) != 0) { + visit((const List*)e); + } else if(dynamic_cast(e) != 0) { + visit((const Dictionary*)e); + } +} -- cgit