tcp_sponge_socket.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_TCP_SPONGE_SOCKET_HH
#define SPONGE_LIBSPONGE_TCP_SPONGE_SOCKET_HH
 
#include "byte_stream.hh"
#include "eventloop.hh"
#include "fd_adapter.hh"
#include "file_descriptor.hh"
#include "tcp_config.hh"
#include "tcp_connection.hh"
#include "tcp_over_ip.hh"
#include "tuntap_adapter.hh"
 
#include <atomic>
#include <cstdint>
#include <optional>
#include <thread>
#include <vector>
 
template <typename AdaptT>
class TCPSpongeSocket : public LocalStreamSocket {
  private:
    LocalStreamSocket _thread_data;
 
    AdaptT _datagram_adapter;
 
    void _initialize_TCP(const TCPConfig &config);
 
    std::optional<TCPConnection> _tcp{};
 
    EventLoop _eventloop{};
 
    void _tcp_loop(const std::function<bool()> &condition);
 
    void _tcp_main();
 
    std::thread _tcp_thread{};
 
    TCPSpongeSocket(std::pair<FileDescriptor, FileDescriptor> data_socket_pair, AdaptT &&datagram_interface);
 
    std::atomic_bool _abort{false};  
 
    bool _inbound_shutdown{false};  
 
    bool _outbound_shutdown{false};  
 
    bool _fully_acked{false};  
 
  public:
    explicit TCPSpongeSocket(AdaptT &&datagram_interface);
 
    void wait_until_closed();
 
    void connect(const TCPConfig &c_tcp, const FdAdapterConfig &c_ad);
 
    void listen_and_accept(const TCPConfig &c_tcp, const FdAdapterConfig &c_ad);
 
    ~TCPSpongeSocket();
 
 
    TCPSpongeSocket(const TCPSpongeSocket &) = delete;
    TCPSpongeSocket(TCPSpongeSocket &&) = delete;
    TCPSpongeSocket &operator=(const TCPSpongeSocket &) = delete;
    TCPSpongeSocket &operator=(TCPSpongeSocket &&) = delete;
 
 
    void bind(const Address &address) = delete;
    Address local_address() const = delete;
    Address peer_address() const = delete;
    void set_reuseaddr() = delete;
};
 
using TCPOverUDPSpongeSocket = TCPSpongeSocket<TCPOverUDPSocketAdapter>;
using TCPOverIPv4SpongeSocket = TCPSpongeSocket<TCPOverIPv4OverTunFdAdapter>;
using LossyTCPOverUDPSpongeSocket = TCPSpongeSocket<LossyTCPOverUDPSocketAdapter>;
using LossyTCPOverIPv4SpongeSocket = TCPSpongeSocket<LossyTCPOverIPv4OverTunFdAdapter>;
 
 
class CS144TCPSocket : public TCPOverIPv4SpongeSocket {
  public:
    CS144TCPSocket();
    void connect(const Address &address);
};
 
#endif  // SPONGE_LIBSPONGE_TCP_SPONGE_SOCKET_HH