mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
273270916e
This commit reflects changes to clang-format rules. Unfortunately, these rule changes create a massive change to the codebase, which causes an apparent rewrite of git history. Git blame's --ignore-rev flag can be used to ignore this commit when attempting to `git blame` some code.
118 lines
2.9 KiB
C++
118 lines
2.9 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();
|
|
version = 0;
|
|
}
|
|
|
|
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 (!BEncodeMaybeVerifyVersion("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();
|
|
version = 0;
|
|
}
|
|
|
|
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 (!BEncodeMaybeVerifyVersion("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 id=", pathid);
|
|
return false;
|
|
}
|
|
} // namespace llarp
|