Address.h

#pragma once
 
#if !defined( EMSCRIPTEN )
 
#include "../FoundationBase.h"
 
#include <cstdint>
#include <cstring>
#include <string>
 
#if defined( _WIN32 )
  #include <Ws2tcpip.h>
  #undef GetObject
#else
  #include <netinet/ip.h>
#endif
 
namespace Cogs::Network{
 
#ifdef _WIN32
  typedef ADDRESS_FAMILY Family;
#else
  typedef sa_family_t Family;
#endif
 
  /*
   * Addr / IP
   */
  class COGSFOUNDATION_API Addr{
  public:
    virtual Family family() const = 0;
    virtual std::string string() const = 0;
    virtual bool is_multicast() const = 0;
  };
 
  /*
   * Addr IPv4
   */
  class COGSFOUNDATION_API AddrIn: public Addr{
  public:
    AddrIn(): addr({}) {}
    AddrIn(in_addr addr): addr(addr) {}
    AddrIn(uint8_t ip[4]) : AddrIn(ip[0], ip[1], ip[2], ip[3]) {}
    AddrIn(uint32_t ip)
    {
      addr.s_addr = htonl(ip);
    }
    AddrIn(uint8_t a, uint8_t b, uint8_t c, uint8_t d)
    {
      addr.s_addr = htonl((uint32_t)a<<24 | (uint32_t)b<<16 | (uint32_t)c<<8 | (uint32_t)d<<0);
    }
    AddrIn(const char* str);
    AddrIn(const std::string& str);
 
    virtual Family family() const override { return AF_INET; }
    virtual std::string string() const override;
    virtual bool is_multicast() const override { return (addr.s_addr&0xf0u) == 0xe0u; }
    uint8_t operator[](size_t i) const { return (addr.s_addr>>(i*8))&0xffu;}
    bool operator ==(const AddrIn& rhs) const { return !std::memcmp(&addr, &rhs.addr, sizeof(addr)); }
 
    in_addr addr;
  };
 
  /*
   * Addr IPv6
   */
  class COGSFOUNDATION_API AddrIn6: public Addr{
  public:
    AddrIn6(): addr({}) { }
    AddrIn6(in6_addr addr): addr(addr) { }
    AddrIn6(const char *str);
 
    virtual Family family() const override { return AF_INET6; }
    virtual std::string string() const override;
    virtual bool is_multicast() const override { return addr.s6_addr[0] == 0xffu; }
    uint8_t operator[](size_t i) const { return addr.s6_addr[i]; }
 
    in6_addr addr;
  };
 
  /*
   * Sockaddr / IP + Port
   */
  class COGSFOUNDATION_API Sockaddr{
  public:
    virtual Family family() const = 0;
    virtual uint16_t port() const = 0;
    virtual sockaddr *getPtr() = 0;
    virtual const sockaddr *getPtr() const = 0;
    virtual int size() const = 0;
  };
 
  /*
   * Sockaddr IPv4
   */
  class COGSFOUNDATION_API SockaddrIn: public Sockaddr{
  public:
    SockaddrIn(): addr({}) { addr.sin_family = family(); }
    SockaddrIn(sockaddr_in addr): addr(addr) {}
    SockaddrIn(AddrIn ip, uint16_t port)
    {
      addr = {};
      addr.sin_family = family();
      addr.sin_port = htons(port);
      addr.sin_addr = ip.addr;
    }
 
    virtual Family family() const override { return AF_INET; }
    AddrIn ip() const { return AddrIn(addr.sin_addr); }
    virtual uint16_t port() const override { return ntohs(addr.sin_port); }
    virtual sockaddr *getPtr() override { return reinterpret_cast<sockaddr*>(&addr); }
    virtual const sockaddr *getPtr() const override { return reinterpret_cast<const sockaddr *>(&addr); }
    virtual int size() const override { return sizeof(addr); }
 
    sockaddr_in addr;
  };
 
  /*
   * Sockaddr IPv6
   */
  class COGSFOUNDATION_API SockaddrIn6: public Sockaddr{
  public:
    SockaddrIn6(): addr({}) { addr.sin6_family = family(); }
    SockaddrIn6(sockaddr_in6 addr): addr(addr) {}
 
    virtual Family family() const override { return AF_INET6; }
    AddrIn6 ip() const { return AddrIn6(addr.sin6_addr); }
    virtual uint16_t port() const override { return ntohs(addr.sin6_port); }
    virtual sockaddr *getPtr() override { return reinterpret_cast<sockaddr*>(&addr); }
    virtual const sockaddr *getPtr() const override { return reinterpret_cast<const sockaddr *>(&addr); }
    virtual int size() const override { return sizeof(addr); }
 
    sockaddr_in6 addr;
  };
 
}// namespace ...
 
#endif