map/das-dn/third_party/AdsLib/Sockets.h

// SPDX-License-Identifier: MIT
/**
   Copyright (c) 2015 - 2022 Beckhoff Automation GmbH & Co. KG
 */

#pragma once

#include "Frame.h"
#include "wrap_socket.h"
#include <stdexcept>
#include <string>

namespace Beckhoff
{
using AddressList = std::unique_ptr<struct addrinfo, void (*)(struct addrinfo*)>;
/**
 * Splits the provided host string into host and port. If no port was found
 * in the host string, the provided default port is used.
 */
AddressList GetHostAddresses(const std::string& host, const std::string& port = {}, bool* found = nullptr);

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);
    size_t read(uint8_t* buffer, size_t maxBytes, timeval* timeout);
    size_t write(const Frame& frame);
    void Shutdown();
    bool IsValid() const;
    bool IsConnected() const;
    int GetError() const;

    struct TimeoutEx : std::runtime_error {
        TimeoutEx(const char* _Message) : std::runtime_error(_Message)
        {}
    };

    /**
     * Confirm if this Socket is connected to one of the target addresses.
     * @param[in] targetAddresses pointer to a previously allocated list of
     *           "struct addrinfo" returned by getaddrinfo(3).
     * @return true, this connection can be used to reach one of the targetAddresses.
     */
    bool IsConnectedTo(const struct addrinfo* targetAddresses) const;

protected:
    bool m_Connected;
    int m_WSAInitialized;
    int m_LastError;
    SOCKET m_Socket;
    sockaddr_storage m_SockAddress;
    const sockaddr* const m_DestAddr;
    socklen_t m_DestAddrLen;
    sockaddr_in m_HostAddr;
    int m_type;

    Socket(const struct addrinfo* host, int type);
    ~Socket();
    bool Select(timeval* timeout);
};

struct TcpSocket : Socket {
    TcpSocket(const struct addrinfo* host);
    uint32_t GetLocalSockAddr() const;
    uint32_t GetHostSockAddr() const;
};

struct UdpSocket : Socket {
    UdpSocket(const struct addrinfo* host);
};
}