lokinet/include/llarp/messages/discard.hpp

111 lines
2.4 KiB
C++
Raw Normal View History

2018-09-06 20:31:58 +00:00
#ifndef LLARP_MESSAGES_DISCARD_HPP
#define LLARP_MESSAGES_DISCARD_HPP
2018-12-12 02:04:32 +00:00
2018-12-12 01:32:10 +00:00
#include <link_message.hpp>
2018-09-06 20:31:58 +00:00
#include <llarp/bencode.hpp>
2018-12-12 02:04:32 +00:00
#include <routing/handler.hpp>
#include <routing/message.hpp>
2018-09-06 20:31:58 +00:00
namespace llarp
{
struct DiscardMessage final : public ILinkMessage
2018-09-06 20:31:58 +00:00
{
/// who did this message come from or is going to
DiscardMessage() : ILinkMessage(nullptr)
{
}
DiscardMessage(ILinkSession* from) : ILinkMessage(from)
{
}
~DiscardMessage()
{
}
bool
BEncode(llarp_buffer_t* buf) const override
2018-09-06 20:31:58 +00:00
{
if(!bencode_start_dict(buf))
return false;
if(!bencode_write_bytestring(buf, "a", 1))
return false;
if(!bencode_write_bytestring(buf, "x", 1))
return false;
return bencode_end(buf);
}
bool
DecodeKey(__attribute__((unused)) llarp_buffer_t key,
__attribute__((unused)) llarp_buffer_t* buf) override
2018-09-06 20:31:58 +00:00
{
return false;
}
bool
HandleMessage(__attribute__((unused)) llarp::Router* router) const override
2018-09-06 20:31:58 +00:00
{
return true;
}
};
namespace routing
{
struct DataDiscardMessage final : public IMessage
{
PathID_t P;
DataDiscardMessage() : IMessage()
{
}
2018-09-11 15:34:12 +00:00
DataDiscardMessage(const PathID_t& dst, uint64_t s) : P(dst)
{
S = s;
version = LLARP_PROTO_VERSION;
}
bool
HandleMessage(IMessageHandler* h, llarp::Router* r) const override
{
return h->HandleDataDiscardMessage(this, r);
}
bool
DecodeKey(llarp_buffer_t k, llarp_buffer_t* buf) override
{
bool read = false;
if(!BEncodeMaybeReadDictEntry("P", P, read, k, buf))
return false;
if(!BEncodeMaybeReadDictInt("S", S, read, k, buf))
return false;
if(!BEncodeMaybeReadDictInt("V", version, read, k, buf))
return false;
return read;
}
bool
BEncode(llarp_buffer_t* buf) const override
{
if(!bencode_start_dict(buf))
return false;
if(!BEncodeWriteDictMsgType(buf, "A", "D"))
return false;
if(!BEncodeWriteDictEntry("P", P, buf))
return false;
if(!BEncodeWriteDictInt("S", S, buf))
return false;
if(!BEncodeWriteDictInt("V", version, buf))
return false;
return bencode_end(buf);
}
};
} // namespace routing
2018-09-06 20:31:58 +00:00
} // namespace llarp
#endif