lokinet/llarp/link_message.cpp

118 lines
2.9 KiB
C++
Raw Normal View History

#include <llarp/router_contact.h>
2018-06-08 13:12:17 +00:00
#include <llarp/messages.hpp>
#include "buffer.hpp"
#include "logger.hpp"
#include "router.hpp"
namespace llarp
{
2018-06-01 18:05:40 +00:00
ILinkMessage::ILinkMessage(const RouterID& id) : remote(id)
{
}
2018-06-01 14:08:54 +00:00
InboundMessageParser::InboundMessageParser(llarp_router* _router)
: router(_router)
{
reader.user = this;
reader.on_key = &OnKey;
}
bool
2018-06-01 14:08:54 +00:00
InboundMessageParser::OnKey(dict_reader* r, llarp_buffer_t* key)
{
2018-06-01 14:08:54 +00:00
InboundMessageParser* handler =
static_cast< InboundMessageParser* >(r->user);
llarp_buffer_t strbuf;
// we are reading the first key
if(handler->firstkey)
{
// check for empty dict
if(!key)
return false;
// we are expecting the first key to be 'a'
if(!llarp_buffer_eq(*key, "a"))
{
2018-06-11 13:25:10 +00:00
llarp::Warn("message has no message type");
return false;
}
if(!bencode_read_string(r->buffer, &strbuf))
{
2018-06-11 13:25:10 +00:00
llarp::Warn("could not read value of message type");
return false;
}
// bad key size
if(strbuf.sz != 1)
{
2018-06-11 13:25:10 +00:00
llarp::Warn("bad mesage type size: ", strbuf.sz);
return false;
}
2018-06-08 13:12:17 +00:00
// create the message to parse based off message type
2018-06-20 12:34:48 +00:00
llarp::Debug("inbound message ", *strbuf.cur);
2018-06-01 14:08:54 +00:00
switch(*strbuf.cur)
{
case 'i':
handler->msg = new LinkIntroMessage(
handler->from->get_remote_router(handler->from));
break;
2018-06-08 13:12:17 +00:00
case 'd':
handler->msg = new RelayDownstreamMessage(handler->GetCurrentFrom());
break;
case 'u':
handler->msg = new RelayUpstreamMessage(handler->GetCurrentFrom());
break;
2018-06-01 14:08:54 +00:00
case 'm':
handler->msg = new DHTImmeidateMessage(handler->GetCurrentFrom());
2018-06-08 13:12:17 +00:00
break;
case 'a':
handler->msg = new LR_AckMessage(handler->GetCurrentFrom());
2018-06-01 14:08:54 +00:00
break;
case 'c':
handler->msg = new LR_CommitMessage(handler->GetCurrentFrom());
break;
2018-06-03 13:04:51 +00:00
case 'z':
handler->msg = new DiscardMessage(handler->GetCurrentFrom());
2018-06-03 15:04:46 +00:00
break;
default:
return false;
2018-06-01 14:08:54 +00:00
}
handler->firstkey = false;
2018-06-01 14:08:54 +00:00
return handler->msg != nullptr;
}
// check for last element
if(!key)
return handler->MessageDone();
2018-06-01 14:08:54 +00:00
return handler->msg->DecodeKey(*key, r->buffer);
}
2018-06-01 14:08:54 +00:00
RouterID
InboundMessageParser::GetCurrentFrom()
{
auto rc = from->get_remote_router(from);
return rc->pubkey;
}
bool
2018-06-01 14:08:54 +00:00
InboundMessageParser::MessageDone()
{
2018-06-01 14:08:54 +00:00
bool result = false;
if(msg)
{
2018-06-01 14:08:54 +00:00
result = msg->HandleMessage(router);
delete msg;
msg = nullptr;
}
2018-06-01 14:08:54 +00:00
return result;
}
bool
2018-06-01 14:08:54 +00:00
InboundMessageParser::ProcessFrom(llarp_link_session* src, llarp_buffer_t buf)
{
from = src;
firstkey = true;
return bencode_read_dict(&buf, &reader);
}
2018-06-20 12:34:48 +00:00
} // namespace llarp