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.
120 lines
3.0 KiB
C++
120 lines
3.0 KiB
C++
#include <messages/relay.hpp>
|
|
|
|
#include <path/path_context.hpp>
|
|
#include <router/abstractrouter.hpp>
|
|
#include <util/bencode.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
void
|
|
RelayUpstreamMessage::Clear()
|
|
{
|
|
pathid.Zero();
|
|
X.Clear();
|
|
Y.Zero();
|
|
}
|
|
|
|
bool
|
|
RelayUpstreamMessage::BEncode(llarp_buffer_t *buf) const
|
|
{
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
if(!BEncodeWriteDictMsgType(buf, "a", "u"))
|
|
return false;
|
|
|
|
if(!BEncodeWriteDictEntry("p", pathid, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictInt("v", LLARP_PROTO_VERSION, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictEntry("x", X, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictEntry("y", Y, buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
bool
|
|
RelayUpstreamMessage::DecodeKey(const llarp_buffer_t &key,
|
|
llarp_buffer_t *buf)
|
|
{
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictEntry("p", pathid, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadVersion("v", version, LLARP_PROTO_VERSION, read, key,
|
|
buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictEntry("x", X, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictEntry("y", Y, read, key, buf))
|
|
return false;
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
RelayUpstreamMessage::HandleMessage(AbstractRouter *r) const
|
|
{
|
|
auto path = r->pathContext().GetByDownstream(session->GetPubKey(), pathid);
|
|
if(path)
|
|
{
|
|
return path->HandleUpstream(llarp_buffer_t(X), Y, r);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void
|
|
RelayDownstreamMessage::Clear()
|
|
{
|
|
pathid.Zero();
|
|
X.Clear();
|
|
Y.Zero();
|
|
}
|
|
|
|
bool
|
|
RelayDownstreamMessage::BEncode(llarp_buffer_t *buf) const
|
|
{
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
if(!BEncodeWriteDictMsgType(buf, "a", "d"))
|
|
return false;
|
|
|
|
if(!BEncodeWriteDictEntry("p", pathid, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictInt("v", LLARP_PROTO_VERSION, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictEntry("x", X, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictEntry("y", Y, buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
bool
|
|
RelayDownstreamMessage::DecodeKey(const llarp_buffer_t &key,
|
|
llarp_buffer_t *buf)
|
|
{
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictEntry("p", pathid, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadVersion("v", version, LLARP_PROTO_VERSION, read, key,
|
|
buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictEntry("x", X, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictEntry("y", Y, read, key, buf))
|
|
return false;
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
RelayDownstreamMessage::HandleMessage(AbstractRouter *r) const
|
|
{
|
|
auto path = r->pathContext().GetByUpstream(session->GetPubKey(), pathid);
|
|
if(path)
|
|
{
|
|
return path->HandleDownstream(llarp_buffer_t(X), Y, r);
|
|
}
|
|
llarp::LogWarn("unhandled downstream message");
|
|
return false;
|
|
}
|
|
} // namespace llarp
|