mirror of
https://github.com/aria2/aria2.git
synced 2025-01-09 03:21:33 +00:00
389f770008
Made socket for dht connections non-blocking * src/DHTAbstractMessage.cc * src/DHTAbstractMessage.h * src/DHTConnection.h * src/DHTConnectionImpl.cc * src/DHTConnectionImpl.h * src/DHTMessage.h * src/DHTMessageDispatcherImpl.cc * src/DHTMessageDispatcherImpl.h * src/DHTUnknownMessage.cc * src/DHTUnknownMessage.h * src/SocketCore.cc * src/SocketCore.h * test/MockDHTMessage.h
51 lines
1.1 KiB
C++
51 lines
1.1 KiB
C++
#ifndef _D_MOCK_DHT_MESSAGE_H_
|
|
#define _D_MOCK_DHT_MESSAGE_H_
|
|
|
|
#include "DHTMessage.h"
|
|
|
|
#include <deque>
|
|
|
|
#include "DHTNode.h"
|
|
#include "Peer.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class MockDHTMessage:public DHTMessage {
|
|
public:
|
|
bool _isReply;
|
|
|
|
std::string _messageType;
|
|
|
|
std::deque<SharedHandle<DHTNode> > _nodes;
|
|
|
|
std::deque<SharedHandle<Peer> > _peers;
|
|
|
|
std::string _token;
|
|
public:
|
|
MockDHTMessage(const SharedHandle<DHTNode>& localNode,
|
|
const SharedHandle<DHTNode>& remoteNode,
|
|
const std::string& messageType = "mock",
|
|
const std::string& transactionID = ""):
|
|
DHTMessage(localNode, remoteNode, transactionID), _isReply(false), _messageType(messageType) {}
|
|
|
|
virtual ~MockDHTMessage() {}
|
|
|
|
virtual void doReceivedAction() {}
|
|
|
|
virtual bool send() { return true; }
|
|
|
|
virtual bool isReply() const { return _isReply; }
|
|
|
|
void setReply(bool f) { _isReply = f; }
|
|
|
|
virtual std::string getMessageType() const { return _messageType; }
|
|
|
|
virtual std::string toString() const { return "MockDHTMessage"; }
|
|
|
|
virtual void validate() const {}
|
|
};
|
|
|
|
} // namespace aria2
|
|
|
|
#endif // _D_MOCK_DHT_MESSAGE_H_
|