lokinet/llarp/routing/message_parser.cpp

101 lines
2.6 KiB
C++
Raw Normal View History

#include <llarp/messages/dht.hpp>
#include <llarp/messages/path_confirm.hpp>
2018-06-26 16:23:43 +00:00
#include <llarp/messages/path_latency.hpp>
#include <llarp/messages/path_transfer.hpp>
#include <llarp/messages/discard.hpp>
#include <llarp/routing/message.hpp>
#include "mem.hpp"
namespace llarp
{
namespace routing
{
InboundMessageParser::InboundMessageParser()
{
reader.user = this;
reader.on_key = &OnKey;
firstKey = false;
}
bool
InboundMessageParser::OnKey(dict_reader* r, llarp_buffer_t* key)
{
InboundMessageParser* self =
static_cast< InboundMessageParser* >(r->user);
if(key == nullptr && self->firstKey)
{
// empty dict
return false;
}
if(!key)
return true;
if(self->firstKey)
{
llarp_buffer_t strbuf;
if(!llarp_buffer_eq(*key, "A"))
return false;
if(!bencode_read_string(r->buffer, &strbuf))
return false;
if(strbuf.sz != 1)
return false;
2018-07-17 06:17:13 +00:00
self->key = *strbuf.cur;
switch(self->key)
{
case 'D':
self->msg = new DataDiscardMessage();
break;
2018-06-26 16:23:43 +00:00
case 'L':
self->msg = new PathLatencyMessage();
2018-06-26 16:23:43 +00:00
break;
case 'M':
self->msg = new DHTMessage();
break;
case 'P':
self->msg = new PathConfirmMessage();
break;
case 'T':
self->msg = new PathTransferMessage();
break;
2018-08-12 17:22:29 +00:00
case 'H':
self->msg = new service::ProtocolFrame();
break;
default:
llarp::LogError("invalid routing message id: ", *strbuf.cur);
}
self->firstKey = false;
return self->msg != nullptr;
}
else
{
return self->msg->DecodeKey(*key, r->buffer);
}
}
bool
InboundMessageParser::ParseMessageBuffer(llarp_buffer_t buf,
2018-06-26 16:23:43 +00:00
IMessageHandler* h,
const PathID_t& from,
2018-06-26 16:23:43 +00:00
llarp_router* r)
{
bool result = false;
msg = nullptr;
firstKey = true;
if(bencode_read_dict(&buf, &reader))
{
msg->from = from;
result = msg->HandleMessage(h, r);
if(!result)
2018-07-17 06:17:13 +00:00
llarp::LogWarn("Failed to handle inbound routing message ", key);
delete msg;
}
else
{
llarp::LogError("read dict failed in routing layer");
llarp::DumpBuffer< llarp_buffer_t, 128 >(buf);
}
return result;
}
} // namespace routing
} // namespace llarp