socket.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_SOCKET_HH
#define SPONGE_LIBSPONGE_SOCKET_HH
 
#include "address.hh"
#include "file_descriptor.hh"
 
#include <cstdint>
#include <functional>
#include <string>
#include <sys/socket.h>
 
class Socket : public FileDescriptor {
  private:
    Address get_address(const std::string &name_of_function,
                        const std::function<int(int, sockaddr *, socklen_t *)> &function) const;
 
  protected:
    Socket(const int domain, const int type);
 
    Socket(FileDescriptor &&fd, const int domain, const int type);
 
    template <typename option_type>
    void setsockopt(const int level, const int option, const option_type &option_value);
 
  public:
    void bind(const Address &address);
 
    void connect(const Address &address);
 
    void shutdown(const int how);
 
    Address local_address() const;
    Address peer_address() const;
 
    void set_reuseaddr();
};
 
class UDPSocket : public Socket {
  protected:
    explicit UDPSocket(FileDescriptor &&fd) : Socket(std::move(fd), AF_INET, SOCK_DGRAM) {}
 
  public:
    UDPSocket() : Socket(AF_INET, SOCK_DGRAM) {}
 
    struct received_datagram {
        Address source_address;  
        std::string payload;     
    };
 
    received_datagram recv(const size_t mtu = 65536);
 
    void recv(received_datagram &datagram, const size_t mtu = 65536);
 
    void sendto(const Address &destination, const BufferViewList &payload);
 
    void send(const BufferViewList &payload);
};
 
 
class TCPSocket : public Socket {
  private:
    explicit TCPSocket(FileDescriptor &&fd) : Socket(std::move(fd), AF_INET, SOCK_STREAM) {}
 
  public:
    TCPSocket() : Socket(AF_INET, SOCK_STREAM) {}
 
    void listen(const int backlog = 16);
 
    TCPSocket accept();
};
 
 
class LocalStreamSocket : public Socket {
  public:
    explicit LocalStreamSocket(FileDescriptor &&fd) : Socket(std::move(fd), AF_UNIX, SOCK_STREAM) {}
};
 
 
#endif  // SPONGE_LIBSPONGE_SOCKET_HH