mirror of https://github.com/aria2/aria2
122 lines
3.8 KiB
C++
122 lines
3.8 KiB
C++
#ifndef D_MOCK_DHT_MESSAGE_FACTORY_H
|
|
#define D_MOCK_DHT_MESSAGE_FACTORY_H
|
|
|
|
#include "DHTMessageFactory.h"
|
|
#include "DHTNode.h"
|
|
#include "MockDHTMessage.h"
|
|
#include "DHTPingMessage.h"
|
|
#include "DHTPingReplyMessage.h"
|
|
#include "DHTFindNodeMessage.h"
|
|
#include "DHTFindNodeReplyMessage.h"
|
|
#include "DHTGetPeersMessage.h"
|
|
#include "DHTGetPeersReplyMessage.h"
|
|
#include "DHTAnnouncePeerMessage.h"
|
|
#include "DHTAnnouncePeerReplyMessage.h"
|
|
#include "DHTUnknownMessage.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MockDHTMessageFactory : public DHTMessageFactory {
|
|
protected:
|
|
std::shared_ptr<DHTNode> localNode_;
|
|
|
|
public:
|
|
MockDHTMessageFactory() {}
|
|
|
|
virtual std::unique_ptr<DHTQueryMessage>
|
|
createQueryMessage(const Dict* dict, const std::string& ipaddr,
|
|
uint16_t port) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTResponseMessage>
|
|
createResponseMessage(const std::string& messageType, const Dict* dict,
|
|
const std::string& ipaddr, uint16_t port) CXX11_OVERRIDE
|
|
{
|
|
auto remoteNode = std::make_shared<DHTNode>();
|
|
// TODO At this point, removeNode's ID is random.
|
|
remoteNode->setIPAddress(ipaddr);
|
|
remoteNode->setPort(port);
|
|
return make_unique<MockDHTResponseMessage>(
|
|
localNode_, remoteNode, downcast<String>(dict->get("t"))->s());
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTPingMessage>
|
|
createPingMessage(const std::shared_ptr<DHTNode>& remoteNode,
|
|
const std::string& transactionID = "") CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTPingReplyMessage>
|
|
createPingReplyMessage(const std::shared_ptr<DHTNode>& remoteNode,
|
|
const unsigned char* remoteNodeID,
|
|
const std::string& transactionID) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTFindNodeMessage>
|
|
createFindNodeMessage(const std::shared_ptr<DHTNode>& remoteNode,
|
|
const unsigned char* targetNodeID,
|
|
const std::string& transactionID = "") CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTFindNodeReplyMessage> createFindNodeReplyMessage(
|
|
const std::shared_ptr<DHTNode>& remoteNode,
|
|
std::vector<std::shared_ptr<DHTNode>> closestKNodes,
|
|
const std::string& transactionID) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTGetPeersMessage>
|
|
createGetPeersMessage(const std::shared_ptr<DHTNode>& remoteNode,
|
|
const unsigned char* infoHash,
|
|
const std::string& transactionID) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTGetPeersReplyMessage> createGetPeersReplyMessage(
|
|
const std::shared_ptr<DHTNode>& remoteNode,
|
|
std::vector<std::shared_ptr<DHTNode>> closestKNodes,
|
|
std::vector<std::shared_ptr<Peer>> peers, const std::string& token,
|
|
const std::string& transactionID) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTAnnouncePeerMessage> createAnnouncePeerMessage(
|
|
const std::shared_ptr<DHTNode>& remoteNode, const unsigned char* infoHash,
|
|
uint16_t tcpPort, const std::string& token,
|
|
const std::string& transactionID = "") CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTAnnouncePeerReplyMessage>
|
|
createAnnouncePeerReplyMessage(const std::shared_ptr<DHTNode>& remoteNode,
|
|
const std::string& transactionID)
|
|
CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual std::unique_ptr<DHTUnknownMessage>
|
|
createUnknownMessage(const unsigned char* data, size_t length,
|
|
const std::string& ipaddr, uint16_t port) CXX11_OVERRIDE
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
void setLocalNode(const std::shared_ptr<DHTNode>& node) { localNode_ = node; }
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // D_MOCK_DHT_MESSAGE_FACTORY_H
|