tuntap_adapter.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_TUNFD_ADAPTER_HH
#define SPONGE_LIBSPONGE_TUNFD_ADAPTER_HH
 
#include "ethernet_header.hh"
#include "network_interface.hh"
#include "tun.hh"
 
#include <optional>
#include <unordered_map>
#include <utility>
 
class TCPOverIPv4OverTunFdAdapter : public TCPOverIPv4Adapter {
  private:
    TunFD _tun;
 
  public:
    explicit TCPOverIPv4OverTunFdAdapter(TunFD &&tun) : _tun(std::move(tun)) {}
 
    std::optional<TCPSegment> read() {
        InternetDatagram ip_dgram;
        if (ip_dgram.parse(_tun.read()) != ParseResult::NoError) {
            return {};
        }
        return unwrap_tcp_in_ip(ip_dgram);
    }
 
    void write(TCPSegment &seg) { _tun.write(wrap_tcp_in_ip(seg).serialize()); }
 
    operator TunFD &() { return _tun; }
 
    operator const TunFD &() const { return _tun; }
};
 
using LossyTCPOverIPv4OverTunFdAdapter = LossyFdAdapter<TCPOverIPv4OverTunFdAdapter>;
 
class TCPOverIPv4OverEthernetAdapter : public TCPOverIPv4Adapter {
  private:
    TapFD _tap;  
 
    NetworkInterface _interface;  
 
    Address _next_hop;  
 
    void send_pending();  
 
  public:
    explicit TCPOverIPv4OverEthernetAdapter(TapFD &&tap,
                                            const EthernetAddress &eth_address,
                                            const Address &ip_address,
                                            const Address &next_hop);
    std::optional<TCPSegment> read();
 
    void write(TCPSegment &seg);
 
    void tick(const size_t ms_since_last_tick);
 
    operator TapFD &() { return _tap; }
 
    operator const TapFD &() const { return _tap; }
};
 
#endif  // SPONGE_LIBSPONGE_TUNFD_ADAPTER_HH