WindowsSocket.hpp

#pragma once
 
#include <winsock2.h>
#include <ws2tcpip.h>
#include <EasySockets/Api/SocketTypes.hpp>
#include <EasySockets/Api/Addresses.hpp>
#include <EasySockets/Api/Documentation.hpp>
 
namespace es
{
    ES_API_DOC(socket)
    class WindowsSocket
    {
    public:
        ES_API_DOC(socket_default_constructor)
        WindowsSocket();
 
        ES_API_DOC(socket_constructor)
        WindowsSocket(IpVersion ip_version, Protocol protocol);
 
        ~WindowsSocket();
 
        WindowsSocket(const WindowsSocket&) = delete;
 
        WindowsSocket& operator=(const WindowsSocket&) = delete;
 
        WindowsSocket(WindowsSocket&& other) noexcept;
 
        WindowsSocket& operator=(WindowsSocket&& other) noexcept;
 
        ES_API_DOC(socket_bind_to)
        void bind_to(const EndPoint& end_point);
 
        ES_API_DOC(socket_connect_to)
        void connect_to(const EndPoint& end_point);
 
        ES_API_DOC(socket_listen_for_connections)
        void listen_for_connections(int backlog);
 
        ES_API_DOC(socket_accept_connection)
        WindowsSocket accept_connection();
 
        ES_API_DOC(socket_receive_data)
        int64_t receive_data(char* buffer, int buffer_size);
 
        ES_API_DOC(socket_receive_data_from)
        int64_t receive_data_from(char* buffer, int buffer_size, EndPoint& sender_end_point);
 
        ES_API_DOC(socket_send_data)
        int64_t send_data(const char* buffer, int buffer_size);
 
        ES_API_DOC(socket_send_data_to)
        int64_t send_data_to(const char* buffer, int buffer_size, const EndPoint& end_point);
 
        ES_API_DOC(socket_close)
        void close();
 
        ES_API_DOC(make_connected_tcp)
        static WindowsSocket make_connected_tcp(const EndPoint& end_point);
 
        ES_API_DOC(make_connected_udp)
        static WindowsSocket make_connected_udp(const EndPoint& end_point);
 
        ES_API_DOC(make_bound_tcp)
        static WindowsSocket make_bound_tcp(Port port, IpVersion ip_version = IpVersion::DUAL_STACK);
 
        ES_API_DOC(make_bound_udp)
        static WindowsSocket make_bound_udp(Port port, IpVersion ip_version = IpVersion::DUAL_STACK);
 
    private:
        // data that winsock needs for the socket
        struct WinsockData
        {
            int af;
            int type;
            int protocol;
        };
 
        // must free with freeaddrinfo
        static addrinfo* resolve_address(const WinsockData& winsock_data, const EndPoint& end_point, int flags);
 
        SOCKET m_socket;
        WinsockData m_winsock_data;
    };
}