lokinet/llarp/net/net_int.hpp
Jeff Becker e060082441
hook every dns for .loki and .snode when applicable
make {n,h}uint{32,16}_t templated type.
2019-05-01 09:40:10 -04:00

211 lines
3.4 KiB
C++

#ifndef LLARP_NET_INT_HPP
#define LLARP_NET_INT_HPP
// for addrinfo
#ifndef _WIN32
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#else
#include <winsock2.h>
#include <ws2tcpip.h>
#define inet_aton(x, y) inet_pton(AF_INET, x, y)
#endif
#include <net/net.h>
#include <stdlib.h> // for itoa
#include <iostream>
#include <util/endian.hpp>
#include <vector>
namespace llarp
{
template < typename UInt_t >
struct huint_t
{
UInt_t h;
constexpr huint_t operator&(huint_t x) const
{
return huint_t{UInt_t(h & x.h)};
}
constexpr huint_t
operator|(huint_t x) const
{
return huint_t{UInt_t(h | x.h)};
}
constexpr huint_t
operator^(huint_t x) const
{
return huint_t{UInt_t(h ^ x.h)};
}
constexpr huint_t
operator~() const
{
return huint_t{UInt_t(~h)};
}
inline huint_t
operator++()
{
++h;
return *this;
}
inline huint_t
operator--()
{
--h;
return *this;
}
constexpr bool
operator<(huint_t x) const
{
return h < x.h;
}
constexpr bool
operator==(huint_t x) const
{
return h == x.h;
}
struct Hash
{
inline size_t
operator()(huint_t x) const
{
return std::hash< UInt_t >{}(x.h);
}
};
using V6Container = std::vector< uint8_t >;
void
ToV6(V6Container& c);
std::string
ToString() const;
friend std::ostream&
operator<<(std::ostream& out, const huint_t& i)
{
return out << i.ToString();
}
};
using huint128_t = huint_t< __uint128_t >;
using huint32_t = huint_t< uint32_t >;
using huint16_t = huint_t< uint16_t >;
template < typename UInt_t >
struct nuint_t
{
UInt_t n;
constexpr nuint_t operator&(nuint_t x) const
{
return nuint_t{UInt_t(n & x.n)};
}
constexpr nuint_t
operator|(nuint_t x) const
{
return nuint_t{UInt_t(n | x.n)};
}
constexpr nuint_t
operator^(nuint_t x) const
{
return nuint_t{UInt_t(n ^ x.n)};
}
constexpr nuint_t
operator~() const
{
return nuint_t{UInt_t(~n)};
}
inline nuint_t
operator++()
{
++n;
return *this;
}
inline nuint_t
operator--()
{
--n;
return *this;
}
constexpr bool
operator<(nuint_t x) const
{
return n < x.n;
}
constexpr bool
operator==(nuint_t x) const
{
return n == x.n;
}
struct Hash
{
inline size_t
operator()(nuint_t x) const
{
return std::hash< UInt_t >{}(x.n);
}
};
using V6Container = std::vector< uint8_t >;
void
ToV6(V6Container& c);
std::string
ToString() const;
friend std::ostream&
operator<<(std::ostream& out, const nuint_t& i)
{
return out << i.ToString();
}
};
using nuint128_t = nuint_t< __uint128_t >;
using nuint32_t = nuint_t< uint32_t >;
using nuint16_t = nuint_t< uint16_t >;
static inline nuint32_t
xhtonl(huint32_t x)
{
return nuint32_t{htonl(x.h)};
}
static inline huint32_t
xntohl(nuint32_t x)
{
return huint32_t{ntohl(x.n)};
}
static inline nuint16_t
xhtons(huint16_t x)
{
return nuint16_t{htons(x.h)};
}
static inline huint16_t
xntohs(nuint16_t x)
{
return huint16_t{ntohs(x.n)};
}
} // namespace llarp
#endif