mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-17 15:25:35 +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.
126 lines
3.4 KiB
C++
126 lines
3.4 KiB
C++
#include "gotintro.hpp"
|
|
|
|
#include <llarp/service/intro.hpp>
|
|
#include <llarp/dht/context.hpp>
|
|
#include <memory>
|
|
#include <llarp/path/path_context.hpp>
|
|
#include <llarp/router/abstractrouter.hpp>
|
|
#include <llarp/routing/dht_message.hpp>
|
|
#include <llarp/tooling/dht_event.hpp>
|
|
#include <utility>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace dht
|
|
{
|
|
GotIntroMessage::GotIntroMessage(std::vector<service::EncryptedIntroSet> results, uint64_t tx)
|
|
: IMessage({}), found(std::move(results)), txid(tx)
|
|
{}
|
|
|
|
bool
|
|
GotIntroMessage::HandleMessage(
|
|
llarp_dht_context* ctx, std::vector<std::unique_ptr<IMessage>>& /*replies*/) const
|
|
{
|
|
auto& dht = *ctx->impl;
|
|
auto* router = dht.GetRouter();
|
|
|
|
router->NotifyRouterEvent<tooling::GotIntroReceivedEvent>(
|
|
router->pubkey(),
|
|
Key_t(From.data()),
|
|
(found.size() > 0 ? found[0] : llarp::service::EncryptedIntroSet{}),
|
|
txid);
|
|
|
|
for (const auto& introset : found)
|
|
{
|
|
if (!introset.Verify(dht.Now()))
|
|
{
|
|
LogWarn(
|
|
"Invalid introset while handling direct GotIntro "
|
|
"from ",
|
|
From);
|
|
return false;
|
|
}
|
|
}
|
|
TXOwner owner(From, txid);
|
|
|
|
auto serviceLookup = dht.pendingIntrosetLookups().GetPendingLookupFrom(owner);
|
|
if (serviceLookup)
|
|
{
|
|
if (not found.empty())
|
|
{
|
|
dht.pendingIntrosetLookups().Found(owner, serviceLookup->target, found);
|
|
}
|
|
else
|
|
{
|
|
dht.pendingIntrosetLookups().NotFound(owner, nullptr);
|
|
}
|
|
return true;
|
|
}
|
|
LogError("no pending TX for GIM from ", From, " txid=", txid);
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
RelayedGotIntroMessage::HandleMessage(
|
|
llarp_dht_context* ctx,
|
|
[[maybe_unused]] std::vector<std::unique_ptr<IMessage>>& replies) const
|
|
{
|
|
// TODO: implement me better?
|
|
auto pathset = ctx->impl->GetRouter()->pathContext().GetLocalPathSet(pathID);
|
|
if (pathset)
|
|
{
|
|
auto copy = std::make_shared<const RelayedGotIntroMessage>(*this);
|
|
return pathset->HandleGotIntroMessage(copy);
|
|
}
|
|
LogWarn("No path for got intro message pathid=", pathID);
|
|
return false;
|
|
}
|
|
|
|
bool
|
|
GotIntroMessage::DecodeKey(const llarp_buffer_t& key, llarp_buffer_t* buf)
|
|
{
|
|
if (key == "I")
|
|
{
|
|
return BEncodeReadList(found, buf);
|
|
}
|
|
if (key == "K")
|
|
{
|
|
if (closer) // duplicate key?
|
|
return false;
|
|
dht::Key_t K;
|
|
if (not K.BDecode(buf))
|
|
return false;
|
|
closer = K;
|
|
return true;
|
|
}
|
|
bool read = false;
|
|
if (!BEncodeMaybeReadDictInt("T", txid, read, key, buf))
|
|
return false;
|
|
if (!BEncodeMaybeReadDictInt("V", version, read, key, buf))
|
|
return false;
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
GotIntroMessage::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
if (!bencode_start_dict(buf))
|
|
return false;
|
|
if (!BEncodeWriteDictMsgType(buf, "A", "G"))
|
|
return false;
|
|
if (!BEncodeWriteDictList("I", found, buf))
|
|
return false;
|
|
if (closer)
|
|
{
|
|
if (!BEncodeWriteDictEntry("K", *closer, buf))
|
|
return false;
|
|
}
|
|
if (!BEncodeWriteDictInt("T", txid, buf))
|
|
return false;
|
|
if (!BEncodeWriteDictInt("V", version, buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
} // namespace dht
|
|
} // namespace llarp
|