mirror of
https://github.com/aria2/aria2.git
synced 2025-01-24 10:41:18 +00:00
286f34cb3f
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
52 lines
1.1 KiB
C++
52 lines
1.1 KiB
C++
#ifndef _D_MOCK_EXTENSION_MESSAGE_H_
|
|
#define _D_MOCK_EXTENSION_MESSAGE_H_
|
|
|
|
#include "ExtensionMessage.h"
|
|
|
|
class MockExtensionMessage:public ExtensionMessage {
|
|
public:
|
|
string _extensionName;
|
|
uint8_t _extensionMessageID;
|
|
string _data;
|
|
bool _doReceivedActionCalled;
|
|
public:
|
|
MockExtensionMessage(const string& extensionName,
|
|
uint8_t extensionMessageID,
|
|
const char* data,
|
|
size_t length):_extensionName(extensionName),
|
|
_extensionMessageID(extensionMessageID),
|
|
_data(&data[0], &data[length]),
|
|
_doReceivedActionCalled(false) {}
|
|
|
|
virtual ~MockExtensionMessage() {}
|
|
|
|
virtual string getBencodedData()
|
|
{
|
|
return _data;
|
|
}
|
|
|
|
virtual uint8_t getExtensionMessageID()
|
|
{
|
|
return _extensionMessageID;
|
|
}
|
|
|
|
virtual const string& getExtensionName() const
|
|
{
|
|
return _extensionName;
|
|
}
|
|
|
|
virtual string toString() const
|
|
{
|
|
return _extensionName;
|
|
}
|
|
|
|
virtual void doReceivedAction()
|
|
{
|
|
_doReceivedActionCalled = true;
|
|
}
|
|
};
|
|
|
|
typedef SharedHandle<MockExtensionMessage> MockExtensionMessageHandle;
|
|
|
|
#endif // _D_MOCK_EXTENSION_MESSAGE_H_
|