#ifndef LLARP_HANDLERS_TUN_HPP #define LLARP_HANDLERS_TUN_HPP #include #include #include #include #include #include // for relay #include // for lookup #include // for tracker namespace llarp { namespace handlers { static const int DefaultTunNetmask = 16; static const char DefaultTunIfname[] = "lokinet0"; static const char DefaultTunDstAddr[] = "10.10.0.1"; static const char DefaultTunSrcAddr[] = "10.10.0.2"; struct TunEndpoint : public service::Endpoint { TunEndpoint(const std::string& nickname, llarp_router* r); ~TunEndpoint(); bool SetOption(const std::string& k, const std::string& v); void Tick(llarp_time_t now); void TickTun(llarp_time_t now); bool MapAddress(const service::Address& remote, uint32_t ip); bool Start(); /// set up tun interface, blocking bool SetupTun(); /// overrides Endpoint bool SetupNetworking(); /// overrides Endpoint /// handle inbound traffic bool ProcessDataMessage(service::ProtocolMessage* msg); #ifndef _MINGW32_NO_THREADS /// overrides Endpount bool IsolationFailed() { m_TunSetupResult.set_value(false); return false; } #endif llarp_tun_io tunif; /// called before writing to tun interface static void tunifBeforeWrite(llarp_tun_io* t); /// handle user to network send buffer flush /// called in router logic thread static void handleNetSend(void*); /// called every time we wish to read a packet from the tun interface static void tunifRecvPkt(llarp_tun_io* t, const void* pkt, ssize_t sz); /// called in the endpoint logic thread static void handleTickTun(void* u); protected: typedef llarp::util::CoDelQueue< net::IPv4Packet, net::IPv4Packet::GetTime, net::IPv4Packet::PutTime, net::IPv4Packet::CompareOrder > PacketQueue_t; /// queue for sending packets over the network from us PacketQueue_t m_UserToNetworkPktQueue; /// queue for sending packets to user from network PacketQueue_t m_NetworkToUserPktQueue; /// return true if we have a remote loki address for this ip address bool HasRemoteForIP(const uint32_t& ipv4) const; /// get ip address for service address unconditionally uint32_t ObtainIPForAddr(const service::Address& addr); /// mark this address as active void MarkIPActive(uint32_t ip); /// mark this address as active forever void MarkIPActiveForever(uint32_t ip); void FlushSend(); private: #ifndef _MINGW32_NO_THREADS /// handles setup, given value true on success and false on failure to set /// up interface std::promise< bool > m_TunSetupResult; #endif /// DNS server per tun struct dnsd_context dnsd; /// DNS loki lookup subsystem configuration (also holds optional iptracker /// for netns) struct dotLokiLookup dll; /// maps ip to service address (host byte order) std::unordered_map< uint32_t, service::Address > m_IPToAddr; /// maps service address to ip (host byte order) std::unordered_map< service::Address, uint32_t, service::Address::Hash > m_AddrToIP; /// maps ip address to timestamp last active std::unordered_map< uint32_t, llarp_time_t > m_IPActivity; /// our ip address (host byte order) uint32_t m_OurIP; /// next ip address to allocate (host byte order) uint32_t m_NextIP; /// highest ip address to allocate uint32_t m_MaxIP; }; } // namespace handlers } // namespace llarp #endif