lokinet/llarp/exit/endpoint.hpp

#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