Sockets.h

// SPDX-License-Identifier: MIT
#pragma once
 
#include "Frame.h"
#include "wrap_socket.h"
#include <stdexcept>
#include <string>
 
namespace bhf
{
namespace ads
{
using AddressList = std::unique_ptr<struct addrinfo, void (*)(struct addrinfo*)>;
AddressList GetListOfAddresses(const std::string& hostPort, const std::string& defaultPort = {});
}
}
 
struct IpV4 {
    const uint32_t value;
    IpV4(const std::string& addr);
    IpV4(uint32_t __val);
    bool operator<(const IpV4& ref) const;
    bool operator==(const IpV4& ref) const;
};
 
struct Socket {
    Frame& read(Frame& frame, timeval* timeout) const;
    size_t read(uint8_t* buffer, size_t maxBytes, timeval* timeout) const;
    size_t write(const Frame& frame) const;
    void Shutdown();
 
    struct TimeoutEx : std::runtime_error {
        TimeoutEx(const char* _Message) : std::runtime_error(_Message)
        {}
    };
 
protected:
    int m_WSAInitialized;
    SOCKET m_Socket;
    sockaddr_storage m_SockAddress;
    const sockaddr* const m_DestAddr;
    socklen_t m_DestAddrLen;
 
    Socket(const struct addrinfo* host, int type);
    ~Socket();
    bool Select(timeval* timeout) const;
};
 
struct TcpSocket : Socket {
    TcpSocket(const struct addrinfo* host);
    uint32_t Connect() const;
 
    bool IsConnectedTo(const struct addrinfo* targetAddresses) const;
};
 
struct UdpSocket : Socket {
    UdpSocket(const struct addrinfo* host);
};