mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
f168b7cf72
It didn't do equality, it did "does the remaining space start with the argument" (and so the replacement in the previous commit was broken). This renames it to avoid the confusion and restores to what it was doing on dev.
66 lines
1.8 KiB
C++
66 lines
1.8 KiB
C++
#include "findname.hpp"
|
|
#include <oxenc/bt_serialize.h>
|
|
#include <llarp/dht/context.hpp>
|
|
#include "gotname.hpp"
|
|
#include <llarp/router/abstractrouter.hpp>
|
|
#include <llarp/rpc/lokid_rpc_client.hpp>
|
|
#include <llarp/path/path_context.hpp>
|
|
#include <llarp/routing/dht_message.hpp>
|
|
|
|
namespace llarp::dht
|
|
{
|
|
FindNameMessage::FindNameMessage(const Key_t& from, Key_t namehash, uint64_t txid)
|
|
: IMessage(from), NameHash(std::move(namehash)), TxID(txid)
|
|
{}
|
|
|
|
bool
|
|
FindNameMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
const auto data = oxenc::bt_serialize(oxenc::bt_dict{
|
|
{"A", "N"sv},
|
|
{"H", std::string_view{(char*)NameHash.data(), NameHash.size()}},
|
|
{"T", TxID}});
|
|
return buf->write(data.begin(), data.end());
|
|
}
|
|
|
|
bool
|
|
FindNameMessage::DecodeKey(const llarp_buffer_t& key, llarp_buffer_t* val)
|
|
{
|
|
if (key.startswith("H"))
|
|
{
|
|
return NameHash.BDecode(val);
|
|
}
|
|
if (key.startswith("T"))
|
|
{
|
|
return bencode_read_integer(val, &TxID);
|
|
}
|
|
return bencode_discard(val);
|
|
}
|
|
|
|
bool
|
|
FindNameMessage::HandleMessage(struct llarp_dht_context* dht, std::vector<Ptr_t>& replies) const
|
|
{
|
|
(void)replies;
|
|
auto r = dht->impl->GetRouter();
|
|
if (pathID.IsZero() or not r->IsServiceNode())
|
|
return false;
|
|
r->RpcClient()->LookupLNSNameHash(NameHash, [r, pathID = pathID, TxID = TxID](auto maybe) {
|
|
auto path = r->pathContext().GetPathForTransfer(pathID);
|
|
if (path == nullptr)
|
|
return;
|
|
routing::DHTMessage msg;
|
|
if (maybe.has_value())
|
|
{
|
|
msg.M.emplace_back(new GotNameMessage(dht::Key_t{}, TxID, *maybe));
|
|
}
|
|
else
|
|
{
|
|
msg.M.emplace_back(new GotNameMessage(dht::Key_t{}, TxID, service::EncryptedName{}));
|
|
}
|
|
path->SendRoutingMessage(msg, r);
|
|
});
|
|
return true;
|
|
}
|
|
|
|
} // namespace llarp::dht
|