mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
65 lines
1.6 KiB
C++
65 lines
1.6 KiB
C++
#ifndef LLARP_DHT_MESSAGES_FIND_ROUTER_HPP
|
|
#define LLARP_DHT_MESSAGES_FIND_ROUTER_HPP
|
|
#include <dht/message.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace dht
|
|
{
|
|
struct FindRouterMessage : public IMessage
|
|
{
|
|
// inbound parsing
|
|
FindRouterMessage(const Key_t& from) : IMessage(from)
|
|
{
|
|
}
|
|
|
|
// find by routerid
|
|
FindRouterMessage(uint64_t id, const RouterID& target)
|
|
: IMessage({}), K(target), txid(id)
|
|
{
|
|
}
|
|
|
|
// exploritory
|
|
FindRouterMessage(uint64_t id) : IMessage({}), exploritory(true), txid(id)
|
|
{
|
|
K.Randomize();
|
|
}
|
|
|
|
~FindRouterMessage() override;
|
|
|
|
bool
|
|
BEncode(llarp_buffer_t* buf) const override;
|
|
|
|
bool
|
|
DecodeKey(const llarp_buffer_t& key, llarp_buffer_t* val) override;
|
|
|
|
bool
|
|
HandleMessage(
|
|
llarp_dht_context* ctx,
|
|
std::vector< std::unique_ptr< IMessage > >& replies) const override;
|
|
|
|
RouterID K;
|
|
bool iterative = false;
|
|
bool exploritory = false;
|
|
uint64_t txid = 0;
|
|
uint64_t version = 0;
|
|
};
|
|
|
|
/// variant of FindRouterMessage relayed via path
|
|
struct RelayedFindRouterMessage final : public FindRouterMessage
|
|
{
|
|
RelayedFindRouterMessage(const Key_t& from) : FindRouterMessage(from)
|
|
{
|
|
}
|
|
|
|
/// handle a relayed FindRouterMessage, do a lookup on the dht and inform
|
|
/// the path of the result
|
|
/// TODO: smart path expiration logic needs to be implemented
|
|
bool
|
|
HandleMessage(llarp_dht_context* ctx,
|
|
std::vector< IMessage::Ptr_t >& replies) const override;
|
|
};
|
|
} // namespace dht
|
|
} // namespace llarp
|
|
#endif
|