mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-07 15:20:31 +00:00
7caa87862e
All #ifndef guards on headers have been removed, I think, in favor of #pragma once Headers are now included as `#include "filename"` if the included file resides in the same directory as the file including it, or any subdirectory therein. Otherwise they are included as `#include <project/top/dir/relative/path/filename>` The above does not include system/os headers.
66 lines
1.8 KiB
C++
66 lines
1.8 KiB
C++
#include "findname.hpp"
|
|
#include <oxenmq/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 = oxenmq::bt_serialize(oxenmq::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 == "H")
|
|
{
|
|
return NameHash.BDecode(val);
|
|
}
|
|
if (key == "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
|