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.
70 lines
1.8 KiB
C++
70 lines
1.8 KiB
C++
#include <llarp/messages/transfer_traffic.hpp>
|
|
#include <llarp/routing/handler.hpp>
|
|
#include <llarp/endian.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace routing
|
|
{
|
|
TransferTrafficMessage&
|
|
TransferTrafficMessage::operator=(const TransferTrafficMessage& other)
|
|
{
|
|
S = other.S;
|
|
version = other.version;
|
|
X = other.X;
|
|
return *this;
|
|
}
|
|
|
|
bool
|
|
TransferTrafficMessage::PutBuffer(llarp_buffer_t buf, uint64_t counter)
|
|
{
|
|
if(buf.sz > MaxExitMTU)
|
|
return false;
|
|
X.emplace_back(buf.sz + 8);
|
|
byte_t * ptr = X.back().data();
|
|
htobe64buf(ptr, counter);
|
|
ptr += 8;
|
|
memcpy(ptr, buf.base, buf.sz);
|
|
// 8 bytes encoding overhead and 8 bytes counter
|
|
_size += buf.sz + 16;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
TransferTrafficMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
if(!BEncodeWriteDictMsgType(buf, "A", "I"))
|
|
return false;
|
|
if(!BEncodeWriteDictInt("S", S, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictInt("V", version, buf))
|
|
return false;
|
|
if(!BEncodeWriteDictList("X", X, buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
bool
|
|
TransferTrafficMessage::DecodeKey(llarp_buffer_t key, llarp_buffer_t* buf)
|
|
{
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictInt("S", S, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictInt("V", version, read, key, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictList("X", X, read, key, buf))
|
|
return false;
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
TransferTrafficMessage::HandleMessage(IMessageHandler* h,
|
|
llarp_router* r) const
|
|
{
|
|
return h->HandleTransferTrafficMessage(this, r);
|
|
}
|
|
|
|
} // namespace routing
|
|
} // namespace llarp
|