network_connection.hpp

#pragma once

#include "global.hpp"

#include "string.hpp"

namespace Cvb
{
CVB_BEGIN_INLINE_NS


struct NetworkConnection
{
  typedef String IPAddress;
  typedef std::uint32_t IPAddressUInt;
  typedef std::uint16_t Port;

  IPAddressUInt ipAddressUInt_{0u};
  IPAddress ipAddress_{CVB_LIT("0.0.0.0")};
  Port port_{0u};

  NetworkConnection() noexcept {}

  NetworkConnection(const IPAddress &ip, const Port &port) : ipAddress_(ip), port_(port)
  {
    ipAddressUInt_ = ToIPAddressUInt(ipAddress_);
  }

  NetworkConnection(const IPAddressUInt &ipUInt, const Port &port) : ipAddressUInt_(ipUInt), port_(port)
  {
    ipAddress_ = ToIPAddress(ipAddressUInt_);
  }

  static IPAddress ToIPAddress(const IPAddressUInt &ipAddressInt)
  {
    unsigned char bytes[4];
    bytes[3] = ipAddressInt & 0xFF;
    bytes[2] = (ipAddressInt >> 8) & 0xFF;
    bytes[1] = (ipAddressInt >> 16) & 0xFF;
    bytes[0] = (ipAddressInt >> 24) & 0xFF;

    std::string str = std::to_string((int)bytes[0]);
    str += "." + std::to_string(static_cast<int>(bytes[1]));
    str += "." + std::to_string(static_cast<int>(bytes[2]));
    str += "." + std::to_string(static_cast<int>(bytes[3]));

    return IPAddress(str.begin(), str.end());
  }

  static IPAddressUInt ToIPAddressUInt(const IPAddress &ip)
  {
    std::string _ip (Internal::CastToAscii(ip));
    char *runner = const_cast<char *>(_ip.c_str());
    unsigned long factor = 1;
    IPAddressUInt temp = 0;
    unsigned int reset = 0;

    while (*runner)
      runner++;

    while ((_ip.c_str() - runner--))
    {
      if (*runner == '.')
      {
        factor <<= 8;
        reset = 0;
        runner--;
      }
      temp += (factor * static_cast<unsigned long>(*(runner) - '0') * (reset == 1 ? 10 : (reset == 2 ? 100 : 1)));
      reset++;
    }
    return temp;
  }
};

CVB_END_INLINE_NS
}