lokinet/llarp/net/net_int.cpp

167 lines
2.9 KiB
C++

#include "net_int.hpp"
#include "ip.hpp"
#include <string>
#include <variant>
#include <oxenc/endian.h>
namespace llarp
{
namespace net
{
huint16_t
ToHost(port_t x)
{
return huint16_t{oxenc::big_to_host(x.n)};
}
huint32_t
ToHost(ipv4addr_t x)
{
return huint32_t{oxenc::big_to_host(x.n)};
}
huint128_t
ToHost(ipv6addr_t x)
{
return {ntoh128(x.n)};
}
port_t
ToNet(huint16_t x)
{
return port_t{oxenc::host_to_big(x.h)};
}
ipv4addr_t
ToNet(huint32_t x)
{
return ipv4addr_t{oxenc::host_to_big(x.h)};
}
ipv6addr_t
ToNet(huint128_t x)
{
return ipv6addr_t{hton128(x.h)};
}
} // namespace net
template <>
void
huint32_t::ToV6(V6Container& c)
{
c.resize(16);
std::fill(c.begin(), c.end(), 0);
oxenc::write_host_as_big(h, c.data() + 12);
c[11] = 0xff;
c[10] = 0xff;
}
template <>
void
huint128_t::ToV6(V6Container& c)
{
c.resize(16);
const in6_addr addr = net::HUIntToIn6(*this);
std::copy_n(addr.s6_addr, 16, c.begin());
}
template <>
std::string
huint32_t::ToString() const
{
uint32_t n = htonl(h);
char tmp[INET_ADDRSTRLEN] = {0};
if (!inet_ntop(AF_INET, (void*)&n, tmp, sizeof(tmp)))
return "";
return tmp;
}
template <>
std::string
huint128_t::ToString() const
{
auto addr = ntoh128(h);
char tmp[INET6_ADDRSTRLEN] = {0};
if (!inet_ntop(AF_INET6, (void*)&addr, tmp, sizeof(tmp)))
return "";
return tmp;
}
template <>
bool
huint16_t::FromString(const std::string& str)
{
if (auto val = std::atoi(str.c_str()); val >= 0)
{
h = val;
return true;
}
else
return false;
}
template <>
bool
huint32_t::FromString(const std::string& str)
{
uint32_t n;
if (!inet_pton(AF_INET, str.c_str(), &n))
return false;
h = ntohl(n);
return true;
}
template <>
bool
huint128_t::FromString(const std::string& str)
{
llarp::uint128_t i;
if (!inet_pton(AF_INET6, str.c_str(), &i))
return false;
h = ntoh128(i);
return true;
}
template <>
std::string
nuint32_t::ToString() const
{
char tmp[INET_ADDRSTRLEN] = {0};
if (!inet_ntop(AF_INET, (void*)&n, tmp, sizeof(tmp)))
return "";
return tmp;
}
template <>
std::string
nuint128_t::ToString() const
{
char tmp[INET6_ADDRSTRLEN] = {0};
if (!inet_ntop(AF_INET6, (void*)&n, tmp, sizeof(tmp)))
return "";
return tmp;
}
template <>
std::string
huint16_t::ToString() const
{
return std::to_string(h);
}
template <>
std::string
nuint16_t::ToString() const
{
return std::to_string(ntohs(n));
}
std::string
net::ToString(const ipaddr_t& ipaddr)
{
return std::visit([](const auto& ip) { return ip.ToString(); }, ipaddr);
}
} // namespace llarp