lokinet/llarp/messages/dht_immediate.cpp

87 lines
1.8 KiB
C++

#include <messages/dht_immediate.hpp>
#include <router/router.hpp>
namespace llarp
{
DHTImmediateMessage::~DHTImmediateMessage()
{
}
void
DHTImmediateMessage::Clear()
{
msgs.clear();
}
bool
DHTImmediateMessage::DecodeKey(const llarp_buffer_t &key, llarp_buffer_t *buf)
{
if(llarp_buffer_eq(key, "m"))
return llarp::dht::DecodeMesssageList(dht::Key_t(session->GetPubKey()),
buf, msgs);
if(llarp_buffer_eq(key, "v"))
{
if(!bencode_read_integer(buf, &version))
return false;
return version == LLARP_PROTO_VERSION;
}
// bad key
return false;
}
bool
DHTImmediateMessage::BEncode(llarp_buffer_t *buf) const
{
if(!bencode_start_dict(buf))
return false;
// message type
if(!bencode_write_bytestring(buf, "a", 1))
return false;
if(!bencode_write_bytestring(buf, "m", 1))
return false;
// dht messages
if(!bencode_write_bytestring(buf, "m", 1))
return false;
// begin list
if(!bencode_start_list(buf))
return false;
for(const auto &msg : msgs)
{
if(!msg->BEncode(buf))
return false;
}
// end list
if(!bencode_end(buf))
return false;
// protocol version
if(!bencode_write_version_entry(buf))
return false;
return bencode_end(buf);
}
bool
DHTImmediateMessage::HandleMessage(llarp::Router *router) const
{
DHTImmediateMessage reply;
reply.session = session;
bool result = true;
for(auto &msg : msgs)
{
result &= msg->HandleMessage(router->dht(), reply.msgs);
}
if(reply.msgs.size())
{
if(result)
{
result = router->SendToOrQueue(session->GetPubKey(), &reply);
}
}
return result;
}
} // namespace llarp