AddressInfo.hxx

#pragma once
 
// C++
#include <optional>
 
// Linux
#include <netdb.h>
 
// cosmos
#include <cosmos/net/IPAddress.hxx>
#include <cosmos/net/types.hxx>
 
namespace cosmos {
 
class AddressInfo :
                protected addrinfo {
public: // functions
 
        SocketFamily family() const {
                return SocketFamily{ai_family};
        }
 
        bool isV4() const { return this->family() == SocketFamily::INET; }
        bool isV6() const { return this->family() == SocketFamily::INET6; }
 
        SocketType type() const {
                return SocketType{ai_socktype};
        }
 
        SocketProtocol protocol() const {
                return SocketProtocol{ai_protocol};
        }
 
        bool hasCanonName() const {
                return ai_canonname != nullptr;
        }
 
        SysString canonName() const {
                if (ai_canonname) {
                        return SysString{ai_canonname};
                } else {
                        return {};
                }
        }
 
        std::optional<IP4Address> asIP4() const {
                if (!ai_addr || family() != SocketFamily::INET)
                        return std::nullopt;
 
                return IP4Address{*reinterpret_cast<sockaddr_in*>(ai_addr)};
        }
 
        std::optional<IP6Address> asIP6() const {
                if (!ai_addr || family() != SocketFamily::INET6)
                        return std::nullopt;
 
                return IP6Address{*reinterpret_cast<sockaddr_in6*>(ai_addr)};
        }
 
protected: // data
 
        friend struct AddressInfoIterator;
 
        bool hasNext() const {
                return ai_next != nullptr;
        }
 
        const AddressInfo* next() const {
                return reinterpret_cast<const AddressInfo*>(ai_next);
        }
 
        // this should only be constructed for special purposes, otherwise this
        // type is just used for casting libc allocated struct addrinfo into.
        AddressInfo() = delete;
};
 
} // end ns