mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
126 lines
2.7 KiB
C++
126 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include <llarp/crypto/types.hpp>
|
|
#include <llarp/net/ip_packet.hpp>
|
|
#include <llarp/path/abstracthophandler.hpp>
|
|
#include <llarp/service/protocol_type.hpp>
|
|
#include <llarp/util/time.hpp>
|
|
|
|
#include <queue>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace handlers
|
|
{
|
|
// forward declare
|
|
struct ExitEndpoint;
|
|
} // namespace handlers
|
|
|
|
namespace exit
|
|
{
|
|
/// persistant exit state for 1 identity on the exit node
|
|
struct Endpoint
|
|
{
|
|
static constexpr size_t MaxUpstreamQueueSize = 256;
|
|
|
|
explicit Endpoint(
|
|
const llarp::PubKey& remoteIdent,
|
|
const llarp::path::HopHandler_ptr& path,
|
|
bool rewriteIP,
|
|
huint128_t ip,
|
|
llarp::handlers::ExitEndpoint* parent);
|
|
|
|
~Endpoint();
|
|
|
|
/// close ourselves
|
|
void
|
|
Close();
|
|
|
|
/// implement istateful
|
|
util::StatusObject
|
|
ExtractStatus() const;
|
|
|
|
/// return true if we are expired right now
|
|
bool
|
|
IsExpired(llarp_time_t now) const;
|
|
|
|
bool
|
|
ExpiresSoon(llarp_time_t now, llarp_time_t dlt = 5s) const;
|
|
|
|
/// return true if this endpoint looks dead right now
|
|
bool
|
|
LooksDead(llarp_time_t now, llarp_time_t timeout = 10s) const;
|
|
|
|
/// tick ourself, reset tx/rx rates
|
|
void
|
|
Tick(llarp_time_t now);
|
|
|
|
/// queue traffic from service node / internet to be transmitted
|
|
bool
|
|
QueueInboundTraffic(std::vector<byte_t> data, service::ProtocolType t);
|
|
|
|
/// flush inbound and outbound traffic queues
|
|
bool
|
|
Flush();
|
|
|
|
/// queue outbound traffic
|
|
/// does ip rewrite here
|
|
// bool
|
|
// QueueOutboundTraffic(
|
|
// PathID_t txid, std::vector<byte_t> data, uint64_t counter, service::ProtocolType t);
|
|
|
|
/// update local path id and cascade information to parent
|
|
/// return true if success
|
|
bool
|
|
UpdateLocalPath(const llarp::PathID_t& nextPath);
|
|
|
|
llarp::path::HopHandler_ptr
|
|
GetCurrentPath() const
|
|
{
|
|
return current_path;
|
|
}
|
|
|
|
const llarp::PubKey&
|
|
PubKey() const
|
|
{
|
|
return remote_signkey;
|
|
}
|
|
|
|
RouterID
|
|
router_id() const
|
|
{
|
|
return remote_signkey.data();
|
|
}
|
|
|
|
uint64_t
|
|
TxRate() const
|
|
{
|
|
return tx_rate;
|
|
}
|
|
|
|
uint64_t
|
|
RxRate() const
|
|
{
|
|
return rx_rate;
|
|
}
|
|
|
|
huint128_t
|
|
LocalIP() const
|
|
{
|
|
return IP;
|
|
}
|
|
|
|
const llarp_time_t createdAt;
|
|
|
|
private:
|
|
llarp::handlers::ExitEndpoint* parent;
|
|
llarp::PubKey remote_signkey;
|
|
llarp::path::HopHandler_ptr current_path;
|
|
llarp::huint128_t IP;
|
|
uint64_t tx_rate, rx_rate;
|
|
llarp_time_t last_active;
|
|
bool rewrite_source;
|
|
};
|
|
} // namespace exit
|
|
} // namespace llarp
|