mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-03 23:15:52 +00:00
56cbef6086
- Remove SIIT from method names & comments because we're doing IPv4 mapped addresses (::ffff:0:0/96) rather than actual SIIT (::ffff:0:0:0/96). - add constructor taking a string+numeric port (and then don't allow a port in the string). - simplify IP string parsing by using parse_int() - replace addrIsV4 with call to ipv6_is_mapped_ipv4 (this also fixes a bug where addrIsV4 was not checking for leading 0s and so could return true for a public IPv6 that happened to have ffff in the wrong spot).
120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
#pragma once
|
|
#include <ostream>
|
|
#include <net/ip.hpp>
|
|
#include <net/net_bits.hpp>
|
|
#include <util/bits.hpp>
|
|
#include <util/types.hpp>
|
|
#include <string>
|
|
|
|
namespace llarp
|
|
{
|
|
struct IPRange
|
|
{
|
|
using Addr_t = huint128_t;
|
|
huint128_t addr = {0};
|
|
huint128_t netmask_bits = {0};
|
|
|
|
static constexpr IPRange
|
|
FromIPv4(byte_t a, byte_t b, byte_t c, byte_t d, byte_t mask)
|
|
{
|
|
return IPRange{net::ExpandV4(ipaddr_ipv4_bits(a, b, c, d)), netmask_ipv6_bits(mask + 96)};
|
|
}
|
|
|
|
/// return true if this iprange is in the IPv4 mapping range for containing ipv4 addresses
|
|
constexpr bool
|
|
IsV4() const
|
|
{
|
|
constexpr auto ipv4_map = IPRange{huint128_t{0x0000'ffff'0000'0000UL}, netmask_ipv6_bits(96)};
|
|
return ipv4_map.Contains(addr);
|
|
}
|
|
|
|
/// return the number of bits set in the hostmask
|
|
constexpr int
|
|
HostmaskBits() const
|
|
{
|
|
if (IsV4())
|
|
{
|
|
return bits::count_bits(net::TruncateV6(netmask_bits));
|
|
}
|
|
return bits::count_bits(netmask_bits);
|
|
}
|
|
|
|
/// return true if the other range is inside our range
|
|
constexpr bool
|
|
Contains(const IPRange& other) const
|
|
{
|
|
return Contains(other.addr) and Contains(other.HighestAddr());
|
|
}
|
|
|
|
/// return true if ip is contained in this ip range
|
|
constexpr bool
|
|
Contains(const Addr_t& ip) const
|
|
{
|
|
return (addr & netmask_bits) == (ip & netmask_bits);
|
|
}
|
|
|
|
/// return true if we are a ipv4 range and contains this ip
|
|
constexpr bool
|
|
Contains(const huint32_t& ip) const
|
|
{
|
|
if (not IsV4())
|
|
return false;
|
|
return Contains(net::ExpandV4(ip));
|
|
}
|
|
|
|
friend std::ostream&
|
|
operator<<(std::ostream& out, const IPRange& a)
|
|
{
|
|
return out << a.ToString();
|
|
}
|
|
|
|
/// get the highest address on this range
|
|
constexpr huint128_t
|
|
HighestAddr() const
|
|
{
|
|
return (addr & netmask_bits) + (huint128_t{1} << (128 - bits::count_bits_128(netmask_bits.h)))
|
|
- huint128_t{1};
|
|
}
|
|
|
|
bool
|
|
operator<(const IPRange& other) const
|
|
{
|
|
return (this->addr & this->netmask_bits) < (other.addr & other.netmask_bits)
|
|
|| this->netmask_bits < other.netmask_bits;
|
|
}
|
|
|
|
bool
|
|
operator==(const IPRange& other) const
|
|
{
|
|
return addr == other.addr and netmask_bits == other.netmask_bits;
|
|
}
|
|
|
|
std::string
|
|
ToString() const
|
|
{
|
|
return BaseAddressString() + "/" + std::to_string(HostmaskBits());
|
|
}
|
|
|
|
std::string
|
|
BaseAddressString() const;
|
|
|
|
bool
|
|
FromString(std::string str);
|
|
};
|
|
|
|
} // namespace llarp
|
|
|
|
namespace std
|
|
{
|
|
template <>
|
|
struct hash<llarp::IPRange>
|
|
{
|
|
size_t
|
|
operator()(const llarp::IPRange& range) const
|
|
{
|
|
const auto str = range.ToString();
|
|
return std::hash<std::string>{}(str);
|
|
}
|
|
};
|
|
} // namespace std
|