address.hpp

Go to the documentation of this file.

#ifndef IV_SRC_COMMS_SOCKET_ADDRESS_HPP_
#define IV_SRC_COMMS_SOCKET_ADDRESS_HPP_
 
#include <memory>
#include <string>
#include <string_view>
#include <stdexcept>
#include <sys/socket.h>
#include <arpa/inet.h>
 
namespace iv::comms::socket
{
template <typename T>
class SocketAddress
{
protected:
    std::unique_ptr<T> m_sockaddr;
    sa_family_t m_family;
    std::string m_address;
    uint16_t m_port;
 
public:
    SocketAddress(std::string_view address, uint16_t port)
    {
        m_sockaddr = std::make_unique<T>();
        m_address = address;
        m_port = port;
 
        if constexpr (std::is_same<T, sockaddr_in>::value)
        {
            m_sockaddr->sin_family = AF_INET;
            m_sockaddr->sin_port = htons(port);
            if (inet_pton(AF_INET, address, &m_sockaddr->sin_addr) <= 0)
            {
                throw std::runtime_error("Invalid address");
            }
        }
        else if constexpr (std::is_same<T, sockaddr_in6>::value)
        {
            m_sockaddr->sin6_family = AF_INET6;
            m_sockaddr->sin6_port = htons(port);
            if (inet_pton(AF_INET6, address, &m_sockaddr->sin6_addr) <= 0)
            {
                throw std::runtime_error("Invalid address");
            }
        }
        else
        {
            throw std::runtime_error("Invalid address family");
        }
    }
 
    ~SocketAddress() = default;
 
 
    sockaddr *getSocketAddress() const
    {
        return reinterpret_cast<sockaddr *>(m_sockaddr.get());
    }
 
    std::string getAddress() const
    {
        return m_address;
    }
 
    uint16_t getPort() const
    {
        return m_port;
    }
 
    socklen_t size() const
    {
        return sizeof(*m_sockaddr);
    }
 
    sa_family_t getFamily() const
    {
        return m_family;
    }
 
};
 
using IPv4Address = SocketAddress<sockaddr_in>;
using IPv6Address = SocketAddress<sockaddr_in6>;
 
}
 
 
#endif//IV_SRC_COMMS_SOCKET_ADDRESS_HPP_