mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-02 03:40:12 +00:00
7caa87862e
All #ifndef guards on headers have been removed, I think, in favor of #pragma once Headers are now included as `#include "filename"` if the included file resides in the same directory as the file including it, or any subdirectory therein. Otherwise they are included as `#include <project/top/dir/relative/path/filename>` The above does not include system/os headers.
118 lines
2.9 KiB
C++
118 lines
2.9 KiB
C++
#include "relay.hpp"
|
|
|
|
#include <llarp/path/path_context.hpp>
|
|
#include <llarp/router/abstractrouter.hpp>
|
|
#include <llarp/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
|