lokinet/include/llarp/service/protocol.hpp

123 lines
3.0 KiB
C++
Raw Normal View History

2018-07-19 04:58:39 +00:00
#ifndef LLARP_SERVICE_PROTOCOL_HPP
#define LLARP_SERVICE_PROTOCOL_HPP
#include <llarp/time.h>
#include <llarp/bencode.hpp>
#include <llarp/crypto.hpp>
2018-07-22 23:14:29 +00:00
#include <llarp/encrypted.hpp>
#include <llarp/routing/message.hpp>
#include <llarp/service/Identity.hpp>
2018-07-22 23:14:29 +00:00
#include <llarp/service/Info.hpp>
#include <llarp/service/Intro.hpp>
2018-08-09 19:02:17 +00:00
#include <llarp/service/handler.hpp>
2018-07-19 04:58:39 +00:00
#include <vector>
namespace llarp
{
namespace service
{
2018-07-22 23:14:29 +00:00
constexpr std::size_t MAX_PROTOCOL_MESSAGE_SIZE = 2048;
2018-08-09 19:02:17 +00:00
typedef uint64_t ProtocolType;
constexpr ProtocolType eProtocolText = 0UL;
constexpr ProtocolType eProtocolTraffic = 1UL;
2018-07-19 04:58:39 +00:00
2018-07-22 23:14:29 +00:00
/// inner message
2018-08-09 19:02:17 +00:00
struct ProtocolMessage : public IBEncodeMessage
2018-07-19 04:58:39 +00:00
{
2018-08-09 19:02:17 +00:00
ProtocolMessage(const ConvoTag& tag);
2018-07-22 23:14:29 +00:00
ProtocolMessage();
2018-07-19 04:58:39 +00:00
~ProtocolMessage();
2018-09-10 18:04:30 +00:00
ProtocolType proto = eProtocolTraffic;
2018-07-19 04:58:39 +00:00
llarp_time_t queued = 0;
std::vector< byte_t > payload;
2018-07-22 23:14:29 +00:00
Introduction introReply;
ServiceInfo sender;
2018-08-09 19:02:17 +00:00
IDataHandler* handler = nullptr;
/// local path we got this message from
PathID_t srcPath;
2018-08-09 19:02:17 +00:00
ConvoTag tag;
2018-07-19 04:58:39 +00:00
bool
DecodeKey(llarp_buffer_t key, llarp_buffer_t* val);
2018-08-09 19:02:17 +00:00
2018-07-19 04:58:39 +00:00
bool
BEncode(llarp_buffer_t* buf) const;
void
PutBuffer(llarp_buffer_t payload);
2018-08-09 19:02:17 +00:00
static void
ProcessAsync(void* user);
2018-07-22 23:14:29 +00:00
};
/// outer message
struct ProtocolFrame : public llarp::routing::IMessage
2018-07-22 23:14:29 +00:00
{
llarp::PQCipherBlock C;
2018-07-22 23:14:29 +00:00
llarp::Encrypted D;
llarp::KeyExchangeNonce N;
llarp::Signature Z;
2018-08-09 19:02:17 +00:00
llarp::service::ConvoTag T;
2018-09-17 13:28:26 +00:00
ProtocolFrame(const ProtocolFrame& other)
2018-09-17 15:32:37 +00:00
: llarp::routing::IMessage()
, C(other.C)
, D(other.D)
, N(other.N)
, Z(other.Z)
, T(other.T)
2018-09-17 13:28:26 +00:00
{
S = other.S;
version = other.version;
}
2018-09-17 16:12:42 +00:00
ProtocolFrame() : llarp::routing::IMessage()
{
}
2018-07-22 23:14:29 +00:00
~ProtocolFrame();
2018-09-17 15:32:37 +00:00
bool
operator==(const ProtocolFrame& other) const;
bool
operator!=(const ProtocolFrame& other) const
{
return !(*this == other);
}
2018-08-14 21:17:18 +00:00
ProtocolFrame&
operator=(const ProtocolFrame& other);
2018-07-22 23:14:29 +00:00
bool
EncryptAndSign(llarp_crypto* c, const ProtocolMessage& msg,
const byte_t* sharedkey, const Identity& localIdent);
2018-07-22 23:14:29 +00:00
2018-08-09 19:02:17 +00:00
bool
AsyncDecryptAndVerify(llarp_logic* logic, llarp_crypto* c,
llarp_threadpool* worker,
const Identity& localIdent,
2018-08-09 19:02:17 +00:00
IDataHandler* handler) const;
2018-07-22 23:14:29 +00:00
bool
DecryptPayloadInto(llarp_crypto* c, const byte_t* sharedkey,
ProtocolMessage& into) const;
2018-07-19 04:58:39 +00:00
2018-07-22 23:14:29 +00:00
bool
DecodeKey(llarp_buffer_t key, llarp_buffer_t* val);
2018-07-20 04:50:28 +00:00
2018-07-22 23:14:29 +00:00
bool
BEncode(llarp_buffer_t* buf) const;
2018-07-19 04:58:39 +00:00
2018-07-22 23:14:29 +00:00
bool
2018-08-10 21:34:11 +00:00
Verify(llarp_crypto* c, const ServiceInfo& from) const;
bool
HandleMessage(llarp::routing::IMessageHandler* h, llarp_router* r) const;
2018-07-19 04:58:39 +00:00
};
} // namespace service
} // namespace llarp
2018-09-10 18:04:30 +00:00
#endif