lokinet/llarp/dht/messages/gotrouter.cpp

126 lines
2.8 KiB
C++
Raw Normal View History

2019-01-16 00:24:16 +00:00
#include <dht/context.hpp>
2018-12-15 16:21:52 +00:00
#include <dht/messages/gotrouter.hpp>
2019-01-16 00:24:16 +00:00
#include <path/path.hpp>
#include <router/abstractrouter.hpp>
2019-01-16 00:24:16 +00:00
namespace llarp
{
namespace dht
{
GotRouterMessage::~GotRouterMessage()
{
}
bool
GotRouterMessage::BEncode(llarp_buffer_t *buf) const
{
if(!bencode_start_dict(buf))
return false;
// message type
if(!BEncodeWriteDictMsgType(buf, "A", "S"))
return false;
if(K)
{
if(!BEncodeWriteDictEntry("K", *K.get(), buf))
return false;
}
// near
if(N.size())
{
if(!BEncodeWriteDictList("N", N, buf))
return false;
}
if(!BEncodeWriteDictList("R", R, buf))
return false;
// txid
if(!BEncodeWriteDictInt("T", txid, buf))
return false;
// version
if(!BEncodeWriteDictInt("V", version, buf))
return false;
return bencode_end(buf);
}
bool
GotRouterMessage::DecodeKey(const llarp_buffer_t &key, llarp_buffer_t *val)
2019-01-16 00:24:16 +00:00
{
if(key == "K")
2019-01-16 00:24:16 +00:00
{
if(K) // duplicate key?
return false;
K.reset(new dht::Key_t());
return K->BDecode(val);
}
if(key == "N")
2019-01-16 00:24:16 +00:00
{
return BEncodeReadList(N, val);
}
if(key == "R")
2019-01-16 00:24:16 +00:00
{
return BEncodeReadList(R, val);
}
if(key == "T")
2019-01-16 00:24:16 +00:00
{
return bencode_read_integer(val, &txid);
}
bool read = false;
if(!BEncodeMaybeReadVersion("V", version, LLARP_PROTO_VERSION, read, key,
val))
return false;
return read;
}
bool
GotRouterMessage::HandleMessage(
llarp_dht_context *ctx,
__attribute__((unused))
std::vector< std::unique_ptr< IMessage > > &replies) const
{
auto &dht = *ctx->impl;
2019-01-16 00:24:16 +00:00
if(relayed)
{
auto pathset =
ctx->impl->GetRouter()->pathContext().GetLocalPathSet(pathID);
2019-01-16 00:24:16 +00:00
return pathset && pathset->HandleGotRouterMessage(this);
}
// not relayed
TXOwner owner(From, txid);
if(dht.pendingExploreLookups().HasPendingLookupFrom(owner))
2019-01-16 00:24:16 +00:00
{
if(N.size() == 0)
dht.pendingExploreLookups().NotFound(owner, K);
2019-01-16 00:24:16 +00:00
else
{
dht.pendingExploreLookups().Found(owner, From.as_array(), N);
2019-01-16 00:24:16 +00:00
}
return true;
}
// not explore lookup
if(!dht.pendingRouterLookups().HasPendingLookupFrom(owner))
2019-01-16 00:24:16 +00:00
{
llarp::LogWarn("Unwarranted GRM from ", From, " txid=", txid);
2019-01-16 00:24:16 +00:00
return false;
}
// no pending lookup
llarp::LogInfo("DHT no pending lookup");
if(R.size() == 1)
dht.pendingRouterLookups().Found(owner, R[0].pubkey, {R[0]});
2019-01-16 00:24:16 +00:00
else
dht.pendingRouterLookups().NotFound(owner, K);
2019-01-16 00:24:16 +00:00
return true;
}
} // namespace dht
} // namespace llarp