lokinet/llarp/service/protocol.hpp

144 lines
3.3 KiB
C++
Raw Normal View History

2018-07-19 04:58:39 +00:00
#ifndef LLARP_SERVICE_PROTOCOL_HPP
#define LLARP_SERVICE_PROTOCOL_HPP
2018-12-12 02:04:32 +00:00
#include <crypto/encrypted.hpp>
#include <crypto/types.hpp>
2018-12-12 00:48:54 +00:00
#include <dht/message.hpp>
2018-12-12 02:04:32 +00:00
#include <routing/message.hpp>
2018-12-12 02:15:08 +00:00
#include <service/Identity.hpp>
#include <service/Info.hpp>
#include <service/Intro.hpp>
#include <service/handler.hpp>
#include <util/bencode.hpp>
#include <util/time.hpp>
2018-12-12 02:04:32 +00:00
2018-07-19 04:58:39 +00:00
#include <vector>
struct llarp_threadpool;
2018-07-19 04:58:39 +00:00
namespace llarp
{
struct Crypto;
class Logic;
2018-07-19 04:58:39 +00:00
namespace service
{
constexpr std::size_t MAX_PROTOCOL_MESSAGE_SIZE = 2048 * 2;
2018-07-22 23:14:29 +00:00
using ProtocolType = uint64_t;
2018-08-09 19:02:17 +00:00
constexpr ProtocolType eProtocolControl = 0UL;
2018-08-09 19:02:17 +00:00
constexpr ProtocolType eProtocolTraffic = 1UL;
2018-07-19 04:58:39 +00:00
2018-07-22 23:14:29 +00:00
/// inner message
struct ProtocolMessage final : 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(const llarp_buffer_t& key, llarp_buffer_t* val) override;
2018-08-09 19:02:17 +00:00
2018-07-19 04:58:39 +00:00
bool
BEncode(llarp_buffer_t* buf) const override;
2018-07-19 04:58:39 +00:00
void
PutBuffer(const 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 final : public routing::IMessage
2018-07-22 23:14:29 +00:00
{
using Encrypted_t = Encrypted< 2048 >;
PQCipherBlock C;
2018-12-20 16:49:05 +00:00
Encrypted_t D;
KeyExchangeNonce N;
Signature Z;
service::ConvoTag T;
2018-08-09 19:02:17 +00:00
2018-09-17 13:28:26 +00:00
ProtocolFrame(const ProtocolFrame& other)
: routing::IMessage()
2018-09-17 15:32:37 +00:00
, 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;
}
ProtocolFrame() : routing::IMessage()
2018-09-17 16:12:42 +00:00
{
2018-12-29 15:44:25 +00:00
Clear();
2018-09-17 16:12:42 +00:00
}
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(Crypto* c, const ProtocolMessage& msg,
const SharedSecret& sharedkey, const Identity& localIdent);
2018-07-22 23:14:29 +00:00
2018-08-09 19:02:17 +00:00
bool
AsyncDecryptAndVerify(Logic* logic, Crypto* c, const PathID_t& srcpath,
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(Crypto* c, const SharedSecret& sharedkey,
ProtocolMessage& into) const;
2018-07-19 04:58:39 +00:00
2018-07-22 23:14:29 +00:00
bool
DecodeKey(const llarp_buffer_t& key, llarp_buffer_t* val) override;
2018-07-20 04:50:28 +00:00
2018-07-22 23:14:29 +00:00
bool
BEncode(llarp_buffer_t* buf) const override;
2018-07-19 04:58:39 +00:00
void
Clear() override
{
2018-12-29 15:44:25 +00:00
C.Zero();
D.Clear();
2018-12-29 15:44:25 +00:00
T.Zero();
N.Zero();
Z.Zero();
}
2018-07-22 23:14:29 +00:00
bool
Verify(Crypto* c, const ServiceInfo& from) const;
bool
2019-02-18 23:58:12 +00:00
HandleMessage(routing::IMessageHandler* h,
AbstractRouter* r) const override;
2018-07-19 04:58:39 +00:00
};
} // namespace service
} // namespace llarp
2018-09-10 18:04:30 +00:00
#endif