mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-29 11:05:43 +00:00
124 lines
3.1 KiB
C++
124 lines
3.1 KiB
C++
#ifndef LLARP_SERVICE_PROTOCOL_HPP
|
|
#define LLARP_SERVICE_PROTOCOL_HPP
|
|
#include <llarp/time.hpp>
|
|
#include <llarp/bencode.hpp>
|
|
#include <llarp/crypto.hpp>
|
|
#include <llarp/encrypted.hpp>
|
|
#include <llarp/routing/message.hpp>
|
|
#include <llarp/service/Identity.hpp>
|
|
#include <llarp/service/Info.hpp>
|
|
#include <llarp/service/Intro.hpp>
|
|
#include <llarp/service/handler.hpp>
|
|
#include <vector>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace service
|
|
{
|
|
constexpr std::size_t MAX_PROTOCOL_MESSAGE_SIZE = 2048 * 2;
|
|
|
|
typedef uint64_t ProtocolType;
|
|
|
|
constexpr ProtocolType eProtocolText = 0UL;
|
|
constexpr ProtocolType eProtocolTraffic = 1UL;
|
|
|
|
/// inner message
|
|
struct ProtocolMessage final : public IBEncodeMessage
|
|
{
|
|
ProtocolMessage(const ConvoTag& tag);
|
|
ProtocolMessage();
|
|
~ProtocolMessage();
|
|
ProtocolType proto = eProtocolTraffic;
|
|
llarp_time_t queued = 0;
|
|
std::vector< byte_t > payload;
|
|
Introduction introReply;
|
|
ServiceInfo sender;
|
|
IDataHandler* handler = nullptr;
|
|
/// local path we got this message from
|
|
PathID_t srcPath;
|
|
ConvoTag tag;
|
|
|
|
bool
|
|
DecodeKey(llarp_buffer_t key, llarp_buffer_t* val) override;
|
|
|
|
bool
|
|
BEncode(llarp_buffer_t* buf) const override;
|
|
|
|
void
|
|
PutBuffer(llarp_buffer_t payload);
|
|
|
|
static void
|
|
ProcessAsync(void* user);
|
|
};
|
|
|
|
/// outer message
|
|
struct ProtocolFrame final : public llarp::routing::IMessage
|
|
{
|
|
llarp::PQCipherBlock C;
|
|
llarp::Encrypted D;
|
|
llarp::KeyExchangeNonce N;
|
|
llarp::Signature Z;
|
|
llarp::service::ConvoTag T;
|
|
|
|
ProtocolFrame(const ProtocolFrame& other)
|
|
: llarp::routing::IMessage()
|
|
, C(other.C)
|
|
, D(other.D)
|
|
, N(other.N)
|
|
, Z(other.Z)
|
|
, T(other.T)
|
|
{
|
|
S = other.S;
|
|
version = other.version;
|
|
}
|
|
|
|
ProtocolFrame() : llarp::routing::IMessage()
|
|
{
|
|
}
|
|
|
|
~ProtocolFrame();
|
|
|
|
bool
|
|
operator==(const ProtocolFrame& other) const;
|
|
|
|
bool
|
|
operator!=(const ProtocolFrame& other) const
|
|
{
|
|
return !(*this == other);
|
|
}
|
|
|
|
ProtocolFrame&
|
|
operator=(const ProtocolFrame& other);
|
|
|
|
bool
|
|
EncryptAndSign(llarp_crypto* c, const ProtocolMessage& msg,
|
|
const byte_t* sharedkey, const Identity& localIdent);
|
|
|
|
bool
|
|
AsyncDecryptAndVerify(llarp_logic* logic, llarp_crypto* c,
|
|
const PathID_t& srcpath, llarp_threadpool* worker,
|
|
const Identity& localIdent,
|
|
IDataHandler* handler) const;
|
|
|
|
bool
|
|
DecryptPayloadInto(llarp_crypto* c, const byte_t* sharedkey,
|
|
ProtocolMessage& into) const;
|
|
|
|
bool
|
|
DecodeKey(llarp_buffer_t key, llarp_buffer_t* val) override;
|
|
|
|
bool
|
|
BEncode(llarp_buffer_t* buf) const override;
|
|
|
|
bool
|
|
Verify(llarp_crypto* c, const ServiceInfo& from) const;
|
|
|
|
bool
|
|
HandleMessage(llarp::routing::IMessageHandler* h,
|
|
llarp_router* r) const override;
|
|
};
|
|
} // namespace service
|
|
} // namespace llarp
|
|
|
|
#endif
|