mirror of
https://github.com/oxen-io/lokinet
synced 2023-12-14 06:53:00 +01: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.
63 lines
1.8 KiB
C++
63 lines
1.8 KiB
C++
#include "gotname.hpp"
|
|
#include <oxenc/bt_serialize.h>
|
|
#include <llarp/dht/context.hpp>
|
|
#include <llarp/router/abstractrouter.hpp>
|
|
#include <llarp/path/path_context.hpp>
|
|
|
|
namespace llarp::dht
|
|
{
|
|
constexpr size_t NameSizeLimit = 128;
|
|
|
|
GotNameMessage::GotNameMessage(const Key_t& from, uint64_t txid, service::EncryptedName data)
|
|
: IMessage(from), result(std::move(data)), TxID(txid)
|
|
{
|
|
if (result.ciphertext.size() > NameSizeLimit)
|
|
throw std::invalid_argument("name data too big");
|
|
}
|
|
|
|
bool
|
|
GotNameMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
const std::string nonce((const char*)result.nonce.data(), result.nonce.size());
|
|
const auto data = oxenc::bt_serialize(
|
|
oxenc::bt_dict{{"A", "M"sv}, {"D", result.ciphertext}, {"N", nonce}, {"T", TxID}});
|
|
return buf->write(data.begin(), data.end());
|
|
}
|
|
|
|
bool
|
|
GotNameMessage::DecodeKey(const llarp_buffer_t& key, llarp_buffer_t* val)
|
|
{
|
|
if (key.startswith("D"))
|
|
{
|
|
llarp_buffer_t str{};
|
|
if (not bencode_read_string(val, &str))
|
|
return false;
|
|
if (str.sz > NameSizeLimit)
|
|
return false;
|
|
result.ciphertext.resize(str.sz);
|
|
std::copy_n(str.cur, str.sz, result.ciphertext.data());
|
|
return true;
|
|
}
|
|
if (key.startswith("N"))
|
|
{
|
|
return result.nonce.BDecode(val);
|
|
}
|
|
if (key.startswith("T"))
|
|
{
|
|
return bencode_read_integer(val, &TxID);
|
|
}
|
|
return bencode_discard(val);
|
|
}
|
|
|
|
bool
|
|
GotNameMessage::HandleMessage(struct llarp_dht_context* ctx, std::vector<Ptr_t>&) const
|
|
{
|
|
auto pathset = ctx->impl->GetRouter()->pathContext().GetLocalPathSet(pathID);
|
|
if (pathset == nullptr)
|
|
return false;
|
|
auto copy = std::make_shared<const GotNameMessage>(*this);
|
|
return pathset->HandleGotNameMessage(copy);
|
|
}
|
|
|
|
} // namespace llarp::dht
|