mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
2.5 KiB
C++
115 lines
2.5 KiB
C++
#include <link_message_parser.hpp>
|
|
|
|
#include <router_contact.hpp>
|
|
#include <util/buffer.hpp>
|
|
#include <util/logger.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
InboundMessageParser::InboundMessageParser(Router* _router) : router(_router)
|
|
{
|
|
}
|
|
|
|
bool
|
|
InboundMessageParser::OnKey(dict_reader* r, llarp_buffer_t* key)
|
|
{
|
|
InboundMessageParser* handler =
|
|
static_cast< InboundMessageParser* >(r->user);
|
|
|
|
// we are reading the first key
|
|
if(handler->firstkey)
|
|
{
|
|
llarp_buffer_t strbuf;
|
|
// check for empty dict
|
|
if(!key)
|
|
return false;
|
|
// we are expecting the first key to be 'a'
|
|
if(!llarp_buffer_eq(*key, "a"))
|
|
{
|
|
llarp::LogWarn("message has no message type");
|
|
return false;
|
|
}
|
|
|
|
if(!bencode_read_string(r->buffer, &strbuf))
|
|
{
|
|
llarp::LogWarn("could not read value of message type");
|
|
return false;
|
|
}
|
|
// bad key size
|
|
if(strbuf.sz != 1)
|
|
{
|
|
llarp::LogWarn("bad mesage type size: ", strbuf.sz);
|
|
return false;
|
|
}
|
|
// create the message to parse based off message type
|
|
llarp::LogDebug("inbound message ", *strbuf.cur);
|
|
switch(*strbuf.cur)
|
|
{
|
|
case 'i':
|
|
handler->msg = &handler->holder.i;
|
|
break;
|
|
case 'd':
|
|
handler->msg = &handler->holder.d;
|
|
break;
|
|
case 'u':
|
|
handler->msg = &handler->holder.u;
|
|
break;
|
|
case 'm':
|
|
handler->msg = &handler->holder.m;
|
|
break;
|
|
case 'c':
|
|
handler->msg = &handler->holder.c;
|
|
break;
|
|
case 'x':
|
|
handler->msg = &handler->holder.x;
|
|
break;
|
|
default:
|
|
return false;
|
|
}
|
|
handler->msg->session = handler->from;
|
|
handler->firstkey = false;
|
|
return true;
|
|
}
|
|
// check for last element
|
|
if(!key)
|
|
return handler->MessageDone();
|
|
|
|
return handler->msg->DecodeKey(*key, r->buffer);
|
|
}
|
|
|
|
bool
|
|
InboundMessageParser::MessageDone()
|
|
{
|
|
bool result = false;
|
|
if(msg)
|
|
{
|
|
result = msg->HandleMessage(router);
|
|
}
|
|
Reset();
|
|
return result;
|
|
}
|
|
|
|
bool
|
|
InboundMessageParser::ProcessFrom(ILinkSession* src, llarp_buffer_t buf)
|
|
{
|
|
if(!src)
|
|
{
|
|
llarp::LogWarn("no link session");
|
|
return false;
|
|
}
|
|
reader.user = this;
|
|
reader.on_key = &OnKey;
|
|
from = src;
|
|
firstkey = true;
|
|
return bencode_read_dict(&buf, &reader);
|
|
}
|
|
|
|
void
|
|
InboundMessageParser::Reset()
|
|
{
|
|
if(msg)
|
|
msg->Clear();
|
|
msg = nullptr;
|
|
}
|
|
} // namespace llarp
|