mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-29 11:05:43 +00:00
18b50f4a74
* update dht docs * update other docs * start working on dht for hidden services * fix up unit tests for dht * update makefile and other build files
61 lines
1.3 KiB
C++
61 lines
1.3 KiB
C++
#include <llarp/messages/dht.hpp>
|
|
#include "../router.hpp"
|
|
|
|
namespace llarp
|
|
{
|
|
namespace routing
|
|
{
|
|
DHTMessage::~DHTMessage()
|
|
{
|
|
for(auto& msg : M)
|
|
delete msg;
|
|
}
|
|
|
|
bool
|
|
DHTMessage::DecodeKey(llarp_buffer_t key, llarp_buffer_t* val)
|
|
{
|
|
llarp::dht::Key_t from;
|
|
from.Zero();
|
|
if(llarp_buffer_eq(key, "M"))
|
|
{
|
|
return llarp::dht::DecodeMesssageList(from, val, M);
|
|
}
|
|
else if(llarp_buffer_eq(key, "V"))
|
|
{
|
|
return bencode_read_integer(val, &V);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
DHTMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
|
|
if(!BEncodeWriteDictMsgType(buf, "A", "M"))
|
|
return false;
|
|
if(!BEncodeWriteDictBEncodeList("M", M, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictInt(buf, "V", LLARP_PROTO_VERSION))
|
|
return false;
|
|
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
bool
|
|
DHTMessage::HandleMessage(IMessageHandler* h, llarp_router* r) const
|
|
{
|
|
// set source as us
|
|
llarp::dht::Key_t us = r->pubkey();
|
|
for(auto& msg : M)
|
|
{
|
|
msg->From = us;
|
|
if(!r->dht->impl.RelayRequestForPath(from, msg))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // namespace routing
|
|
} // namespace llarp
|