mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-31 09:20:21 +00:00
355 lines
8.0 KiB
C++
355 lines
8.0 KiB
C++
#pragma once
|
|
|
|
#include <llarp/config/config.hpp>
|
|
#include <llarp/config/key_manager.hpp>
|
|
#include <memory>
|
|
#include <llarp/util/types.hpp>
|
|
#include <llarp/util/status.hpp>
|
|
#include "i_outbound_message_handler.hpp"
|
|
#include <vector>
|
|
#include <llarp/ev/ev.hpp>
|
|
#include <functional>
|
|
#include <llarp/router_contact.hpp>
|
|
#include <llarp/tooling/router_event.hpp>
|
|
#include <llarp/peerstats/peer_db.hpp>
|
|
#include <llarp/consensus/reachability_testing.hpp>
|
|
|
|
#include <optional>
|
|
|
|
#ifdef LOKINET_HIVE
|
|
#include <llarp/tooling/router_event.hpp>
|
|
#endif
|
|
|
|
struct llarp_buffer_t;
|
|
struct llarp_dht_context;
|
|
|
|
namespace oxenmq
|
|
{
|
|
class OxenMQ;
|
|
}
|
|
|
|
namespace llarp
|
|
{
|
|
class NodeDB;
|
|
struct Config;
|
|
struct RouterID;
|
|
struct ILinkMessage;
|
|
struct ILinkSession;
|
|
struct PathID_t;
|
|
struct Profiling;
|
|
struct SecretKey;
|
|
struct Signature;
|
|
struct IOutboundMessageHandler;
|
|
struct IOutboundSessionMaker;
|
|
struct ILinkManager;
|
|
struct I_RCLookupHandler;
|
|
struct RoutePoker;
|
|
|
|
namespace exit
|
|
{
|
|
struct Context;
|
|
}
|
|
|
|
namespace rpc
|
|
{
|
|
struct LokidRpcClient;
|
|
}
|
|
|
|
namespace path
|
|
{
|
|
struct PathContext;
|
|
}
|
|
|
|
namespace routing
|
|
{
|
|
struct IMessageHandler;
|
|
}
|
|
|
|
namespace service
|
|
{
|
|
struct Context;
|
|
}
|
|
|
|
namespace thread
|
|
{
|
|
class ThreadPool;
|
|
}
|
|
|
|
namespace vpn
|
|
{
|
|
class Platform;
|
|
}
|
|
|
|
using LMQ_ptr = std::shared_ptr<oxenmq::OxenMQ>;
|
|
|
|
struct AbstractRouter : public std::enable_shared_from_this<AbstractRouter>
|
|
{
|
|
#ifdef LOKINET_HIVE
|
|
tooling::RouterHive* hive = nullptr;
|
|
#endif
|
|
|
|
virtual ~AbstractRouter() = default;
|
|
|
|
virtual bool
|
|
HandleRecvLinkMessageBuffer(ILinkSession* from, const llarp_buffer_t& msg) = 0;
|
|
|
|
virtual const LMQ_ptr&
|
|
lmq() const = 0;
|
|
|
|
virtual vpn::Platform*
|
|
GetVPNPlatform() const = 0;
|
|
|
|
virtual const std::shared_ptr<rpc::LokidRpcClient>&
|
|
RpcClient() const = 0;
|
|
|
|
virtual llarp_dht_context*
|
|
dht() const = 0;
|
|
|
|
virtual const std::shared_ptr<NodeDB>&
|
|
nodedb() const = 0;
|
|
|
|
virtual const path::PathContext&
|
|
pathContext() const = 0;
|
|
|
|
virtual path::PathContext&
|
|
pathContext() = 0;
|
|
|
|
virtual const RouterContact&
|
|
rc() const = 0;
|
|
|
|
/// modify our rc
|
|
/// modify returns nullopt if unmodified otherwise it returns the new rc to be sigend and
|
|
/// published out
|
|
virtual void
|
|
ModifyOurRC(std::function<std::optional<RouterContact>(RouterContact)> modify) = 0;
|
|
|
|
virtual exit::Context&
|
|
exitContext() = 0;
|
|
|
|
virtual const std::shared_ptr<KeyManager>&
|
|
keyManager() const = 0;
|
|
|
|
virtual const SecretKey&
|
|
identity() const = 0;
|
|
|
|
virtual const SecretKey&
|
|
encryption() const = 0;
|
|
|
|
virtual Profiling&
|
|
routerProfiling() = 0;
|
|
|
|
virtual const EventLoop_ptr&
|
|
loop() const = 0;
|
|
|
|
/// call function in crypto worker
|
|
virtual void QueueWork(std::function<void(void)>) = 0;
|
|
|
|
/// call function in disk io thread
|
|
virtual void QueueDiskIO(std::function<void(void)>) = 0;
|
|
|
|
virtual std::shared_ptr<Config>
|
|
GetConfig() const
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
virtual service::Context&
|
|
hiddenServiceContext() = 0;
|
|
|
|
virtual const service::Context&
|
|
hiddenServiceContext() const = 0;
|
|
|
|
virtual IOutboundMessageHandler&
|
|
outboundMessageHandler() = 0;
|
|
|
|
virtual IOutboundSessionMaker&
|
|
outboundSessionMaker() = 0;
|
|
|
|
virtual ILinkManager&
|
|
linkManager() = 0;
|
|
|
|
virtual RoutePoker&
|
|
routePoker() = 0;
|
|
|
|
virtual I_RCLookupHandler&
|
|
rcLookupHandler() = 0;
|
|
|
|
virtual std::shared_ptr<PeerDb>
|
|
peerDb() = 0;
|
|
|
|
virtual bool
|
|
Sign(Signature& sig, const llarp_buffer_t& buf) const = 0;
|
|
|
|
virtual bool
|
|
Configure(std::shared_ptr<Config> conf, bool isSNode, std::shared_ptr<NodeDB> nodedb) = 0;
|
|
|
|
virtual bool
|
|
IsServiceNode() const = 0;
|
|
|
|
virtual bool
|
|
StartRpcServer() = 0;
|
|
|
|
virtual bool
|
|
Run() = 0;
|
|
|
|
virtual bool
|
|
IsRunning() const = 0;
|
|
|
|
virtual bool
|
|
LooksAlive() const = 0;
|
|
|
|
/// stop running the router logic gracefully
|
|
virtual void
|
|
Stop() = 0;
|
|
|
|
/// thaw from long sleep or network changed event
|
|
virtual void
|
|
Thaw() = 0;
|
|
|
|
/// non gracefully stop the router
|
|
virtual void
|
|
Die() = 0;
|
|
|
|
/// pump low level links
|
|
virtual void
|
|
PumpLL() = 0;
|
|
|
|
virtual bool
|
|
IsBootstrapNode(RouterID r) const = 0;
|
|
|
|
virtual const byte_t*
|
|
pubkey() const = 0;
|
|
|
|
/// connect to N random routers
|
|
virtual void
|
|
ConnectToRandomRouters(int N) = 0;
|
|
|
|
virtual bool
|
|
TryConnectAsync(RouterContact rc, uint16_t tries) = 0;
|
|
|
|
/// called by link when a remote session has no more sessions open
|
|
virtual void
|
|
SessionClosed(RouterID remote) = 0;
|
|
|
|
/// returns system clock milliseconds since epoch
|
|
virtual llarp_time_t
|
|
Now() const = 0;
|
|
|
|
/// returns milliseconds since started
|
|
virtual llarp_time_t
|
|
Uptime() const = 0;
|
|
|
|
virtual bool
|
|
GetRandomGoodRouter(RouterID& r) = 0;
|
|
|
|
virtual bool
|
|
SendToOrQueue(
|
|
const RouterID& remote, const ILinkMessage& msg, SendStatusHandler handler = nullptr) = 0;
|
|
|
|
virtual void
|
|
PersistSessionUntil(const RouterID& remote, llarp_time_t until) = 0;
|
|
|
|
virtual bool
|
|
ParseRoutingMessageBuffer(
|
|
const llarp_buffer_t& buf, routing::IMessageHandler* h, const PathID_t& rxid) = 0;
|
|
|
|
/// count the number of service nodes we are connected to
|
|
virtual size_t
|
|
NumberOfConnectedRouters() const = 0;
|
|
|
|
/// count the number of clients that are connected to us
|
|
virtual size_t
|
|
NumberOfConnectedClients() const = 0;
|
|
|
|
virtual bool
|
|
GetRandomConnectedRouter(RouterContact& result) const = 0;
|
|
|
|
virtual void
|
|
HandleDHTLookupForExplore(RouterID remote, const std::vector<RouterContact>& results) = 0;
|
|
|
|
virtual void SetDownHook(std::function<void(void)>){};
|
|
|
|
/// lookup router by pubkey
|
|
/// if we are a service node this is done direct otherwise it's done via
|
|
/// path
|
|
virtual void
|
|
LookupRouter(RouterID remote, RouterLookupHandler resultHandler) = 0;
|
|
|
|
/// check if newRc matches oldRC and update local rc for this remote contact
|
|
/// if valid
|
|
/// returns true on valid and updated
|
|
/// returns false otherwise
|
|
virtual bool
|
|
CheckRenegotiateValid(RouterContact newRc, RouterContact oldRC) = 0;
|
|
|
|
/// set router's service node whitelist
|
|
virtual void
|
|
SetRouterWhitelist(
|
|
const std::vector<RouterID>& whitelist, const std::vector<RouterID>& greylist) = 0;
|
|
|
|
virtual std::unordered_set<RouterID>
|
|
GetRouterWhitelist() const = 0;
|
|
|
|
/// visit each connected link session
|
|
virtual void
|
|
ForEachPeer(std::function<void(const ILinkSession*, bool)> visit, bool randomize) const = 0;
|
|
|
|
virtual bool
|
|
SessionToRouterAllowed(const RouterID& router) const = 0;
|
|
|
|
virtual bool
|
|
PathToRouterAllowed(const RouterID& router) const = 0;
|
|
|
|
/// return true if we have an exit as a client
|
|
virtual bool
|
|
HasClientExit() const
|
|
{
|
|
return false;
|
|
};
|
|
|
|
virtual path::BuildLimiter&
|
|
pathBuildLimiter() = 0;
|
|
|
|
/// return true if we have at least 1 session to this router in either
|
|
/// direction
|
|
virtual bool
|
|
HasSessionTo(const RouterID& router) const = 0;
|
|
|
|
virtual uint32_t
|
|
NextPathBuildNumber() = 0;
|
|
|
|
virtual std::string
|
|
ShortName() const = 0;
|
|
|
|
virtual util::StatusObject
|
|
ExtractStatus() const = 0;
|
|
|
|
/// gossip an rc if required
|
|
virtual void
|
|
GossipRCIfNeeded(const RouterContact rc) = 0;
|
|
|
|
/// Templated convenience function to generate a RouterHive event and
|
|
/// delegate to non-templated (and overridable) function for handling.
|
|
template <class EventType, class... Params>
|
|
void
|
|
NotifyRouterEvent([[maybe_unused]] Params&&... args) const
|
|
{
|
|
// TODO: no-op when appropriate
|
|
auto event = std::make_unique<EventType>(args...);
|
|
HandleRouterEvent(std::move(event));
|
|
}
|
|
|
|
#if defined(ANDROID)
|
|
virtual int
|
|
GetOutboundUDPSocket() const = 0;
|
|
#endif
|
|
|
|
protected:
|
|
/// Virtual function to handle RouterEvent. HiveRouter overrides this in
|
|
/// order to inject the event. The default implementation in Router simply
|
|
/// logs it.
|
|
virtual void
|
|
HandleRouterEvent(tooling::RouterEventPtr event) const = 0;
|
|
};
|
|
} // namespace llarp
|