mirror of https://github.com/vacp2p/nim-libp2p.git
700 lines
28 KiB
Nim
700 lines
28 KiB
Nim
## Nim-LibP2P
|
|
## Copyright (c) 2018 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import endians, strutils
|
|
import chronos
|
|
export chronos
|
|
|
|
type
|
|
IpMask* = object
|
|
case family*: AddressFamily
|
|
of AddressFamily.None, AddressFamily.Unix:
|
|
discard
|
|
of AddressFamily.IPv4:
|
|
mask4*: uint32
|
|
of AddressFamily.IPv6:
|
|
mask6*: array[2, uint64]
|
|
|
|
IpNet* = object
|
|
host*: TransportAddress
|
|
mask*: IpMask
|
|
|
|
proc toNetworkOrder(mask: IpMask): IpMask {.inline.} =
|
|
## Converts ``mask`` from host order (which can be big/little-endian) to
|
|
## network order (which is big-endian) representation.
|
|
result = IpMask(family: mask.family)
|
|
if mask.family == AddressFamily.IPv4:
|
|
bigEndian32(cast[pointer](addr result.mask4),
|
|
cast[pointer](unsafeAddr mask.mask4))
|
|
elif mask.family == AddressFamily.IPv6:
|
|
bigEndian64(cast[pointer](addr result.mask6[0]),
|
|
cast[pointer](unsafeAddr mask.mask6[0]))
|
|
bigEndian64(cast[pointer](addr result.mask6[1]),
|
|
cast[pointer](unsafeAddr mask.mask6[1]))
|
|
|
|
proc toHostOrder(mask: IpMask): IpMask {.inline.} =
|
|
## Converts ``mask`` from network order (which is big-endian) back to
|
|
## host representation (which can be big/little-endian).
|
|
when system.cpuEndian == bigEndian:
|
|
result = mask
|
|
else:
|
|
result = IpMask(family: mask.family)
|
|
if mask.family == AddressFamily.IPv4:
|
|
swapEndian32(cast[pointer](addr result.mask4),
|
|
cast[pointer](unsafeAddr mask.mask4))
|
|
elif mask.family == AddressFamily.IPv6:
|
|
swapEndian64(cast[pointer](addr result.mask6[0]),
|
|
cast[pointer](unsafeAddr mask.mask6[0]))
|
|
swapEndian64(cast[pointer](addr result.mask6[1]),
|
|
cast[pointer](unsafeAddr mask.mask6[1]))
|
|
|
|
proc `==`*(m1, m2: IpMask): bool {.inline.} =
|
|
## Returns ``true`` if masks ``m1`` and ``m2`` are equal in IP family and
|
|
## by value.
|
|
if m1.family == m2.family:
|
|
if m1.family == AddressFamily.IPv4:
|
|
result = (m1.mask4 == m2.mask4)
|
|
elif m1.family == AddressFamily.IPv6:
|
|
result = ((m1.mask6[0] == m2.mask6[0]) and (m1.mask6[1] == m2.mask6[1]))
|
|
|
|
proc init*(t: typedesc[IpMask], family: AddressFamily, prefix: int): IpMask =
|
|
## Initialize mask of IP family ``family`` from prefix length ``prefix``.
|
|
##
|
|
## For IPv4 mask, if ``prefix`` is ``0`` or more then ``32`` it will be set
|
|
## to ``32``.
|
|
##
|
|
## For IPv6 mask, if ``prefix`` is ``0`` or more then ``128`` it will be set
|
|
## to ``128``.
|
|
if family == AddressFamily.IPv4:
|
|
result = IpMask(family: AddressFamily.IPv4)
|
|
result.mask4 = 0xFFFF_FFFF'u32
|
|
if prefix > 0 and prefix < 32:
|
|
result.mask4 = 0xFFFF_FFFF'u32
|
|
result.mask4 = cast[uint32](result.mask4 shl (32 - prefix))
|
|
elif family == AddressFamily.IPv6:
|
|
result = IpMask(family: AddressFamily.IPv6)
|
|
if prefix >= 128 or prefix <= 0:
|
|
result.mask6[0] = 0xFFFF_FFFF_FFFF_FFFF'u64
|
|
result.mask6[1] = 0xFFFF_FFFF_FFFF_FFFF'u64
|
|
else:
|
|
result.mask6[0] = 0xFFFF_FFFF_FFFF_FFFF'u64
|
|
if prefix > 64:
|
|
result.mask6[1] = 0xFFFF_FFFF_FFFF_FFFF'u64
|
|
result.mask6[1] = result.mask6[1] shl (128 - prefix)
|
|
elif prefix == 64:
|
|
result.mask6[1] = 0x00'u64
|
|
else:
|
|
result.mask6[0] = result.mask6[0] shl (64 - prefix)
|
|
result.mask6[1] = 0x00'u64
|
|
result = result.toNetworkOrder()
|
|
|
|
proc init*(t: typedesc[IpMask], netmask: TransportAddress): IpMask =
|
|
## Initialize network mask using address ``netmask``.
|
|
if netmask.family == AddressFamily.IPv4:
|
|
result.family = netmask.family
|
|
result.mask4 = cast[ptr uint32](unsafeAddr netmask.address_v4[0])[]
|
|
elif netmask.family == AddressFamily.IPv6:
|
|
result.family = netmask.family
|
|
result.mask6[0] = cast[ptr uint64](unsafeAddr netmask.address_v6[0])[]
|
|
result.mask6[1] = cast[ptr uint64](unsafeAddr netmask.address_v6[8])[]
|
|
|
|
proc initIp*(t: typedesc[IpMask], netmask: string): IpMask =
|
|
## Initialize network mask using IPv4 or IPv6 address in text representation
|
|
## ``netmask``.
|
|
##
|
|
## If ``netmask`` address string is invalid, result IpMask.family will be
|
|
## set to ``AddressFamily.None``.
|
|
try:
|
|
var ip = parseIpAddress(netmask)
|
|
var tip = initTAddress(ip, Port(0))
|
|
result = t.init(tip)
|
|
except ValueError:
|
|
discard
|
|
|
|
proc init*(t: typedesc[IpMask], netmask: string): IpMask =
|
|
## Initialize network mask using hexadecimal string representation
|
|
## ``netmask``.
|
|
##
|
|
## If ``netmask`` mask is invalid, result IpMask.family will be set to
|
|
## ``AddressFamily.None``.
|
|
const
|
|
hexNumbers = {'0'..'9'}
|
|
hexCapitals = {'A'..'F'}
|
|
hexLowers = {'a'..'f'}
|
|
let length = len(netmask)
|
|
if length == 8 or length == (2 + 8):
|
|
## IPv4 mask
|
|
var offset = 0
|
|
if length == 2 + 8:
|
|
offset = 2
|
|
var res = IpMask(family: AddressFamily.IPv4)
|
|
var r, v: uint32
|
|
for i in 0..<8:
|
|
if netmask[offset + i] in hexNumbers:
|
|
v = cast[uint32](ord(netmask[offset + i]) - ord('0'))
|
|
elif netmask[offset + i] in hexCapitals:
|
|
v = cast[uint32](ord(netmask[offset + i]) - ord('A') + 10)
|
|
elif netmask[offset + i] in hexLowers:
|
|
v = cast[uint32](ord(netmask[offset + i]) - ord('a') + 10)
|
|
else:
|
|
return
|
|
r = (r shl 4) or v
|
|
bigEndian32(addr res.mask4, addr r)
|
|
result = res
|
|
elif length == 32 or length == (2 + 32):
|
|
## IPv6 mask
|
|
var offset = 0
|
|
if length == 2 + 32:
|
|
offset = 2
|
|
var res = IpMask(family: AddressFamily.IPv6)
|
|
for i in 0..1:
|
|
var r, v: uint64
|
|
for i in 0..<16:
|
|
if netmask[offset + i] in hexNumbers:
|
|
v = cast[uint64](ord(netmask[offset + i]) - ord('0'))
|
|
elif netmask[offset + i] in hexCapitals:
|
|
v = cast[uint64](ord(netmask[offset + i]) - ord('A') + 10)
|
|
elif netmask[offset + i] in hexLowers:
|
|
v = cast[uint64](ord(netmask[offset + i]) - ord('a') + 10)
|
|
else:
|
|
return
|
|
r = (r shl 4) or v
|
|
offset += 16
|
|
bigEndian64(addr res.mask6[i], addr r)
|
|
result = res
|
|
|
|
proc toIPv6*(address: TransportAddress): TransportAddress =
|
|
## Map IPv4 ``address`` to IPv6 address.
|
|
##
|
|
## If ``address`` is IPv4 address then it will be mapped as:
|
|
## <80 bits of zeros> + <16 bits of ones> + <32-bit IPv4 address>.
|
|
##
|
|
## If ``address`` is IPv6 address it will be returned without any changes.
|
|
if address.family == AddressFamily.IPv4:
|
|
result = TransportAddress(family: AddressFamily.IPv6)
|
|
result.address_v6[10] = 0xFF'u8
|
|
result.address_v6[11] = 0xFF'u8
|
|
let data = cast[ptr uint32](unsafeAddr address.address_v4[0])[]
|
|
cast[ptr uint32](addr result.address_v6[12])[] = data
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = address
|
|
|
|
proc isV4Mapped*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is (IPv4 to IPv6) mapped address, e.g.
|
|
## 0000:0000:0000:0000:0000:FFFF:xxxx:xxxx
|
|
##
|
|
## Procedure returns ``false`` if ``address`` family is IPv4.
|
|
if address.family == AddressFamily.IPv6:
|
|
let data0 = cast[ptr uint64](unsafeAddr address.address_v6[0])[]
|
|
let data1 = cast[ptr uint16](unsafeAddr address.address_v6[8])[]
|
|
let data2 = cast[ptr uint16](unsafeAddr address.address_v6[10])[]
|
|
result = (data0 == 0'u64) and (data1 == 0x00'u16) and (data2 == 0xFFFF'u16)
|
|
|
|
proc toIPv4*(address: TransportAddress): TransportAddress =
|
|
## Get IPv4 from (IPv4 to IPv6) mapped address.
|
|
##
|
|
## If ``address`` is IPv4 address it will be returned without any changes.
|
|
##
|
|
## If ``address`` is not IPv4 to IPv6 mapped address, then result family will
|
|
## be set to AddressFamily.None.
|
|
if address.family == AddressFamily.IPv6:
|
|
if isV4Mapped(address):
|
|
result = TransportAddress(family: AddressFamily.IPv4)
|
|
let data = cast[ptr uint32](unsafeAddr address.address_v6[12])[]
|
|
cast[ptr uint32](addr result.address_v4[0])[] = data
|
|
elif address.family == AddressFamily.IPv4:
|
|
result = address
|
|
|
|
proc mask*(a: TransportAddress, m: IpMask): TransportAddress =
|
|
## Apply IP mask ``m`` to address ``a`` and return result address.
|
|
##
|
|
## If ``a`` family is IPv4 and ``m`` family is IPv6, masking is still
|
|
## possible when ``m`` has ``FFFF:FFFF:FFFF:FFFF:FFFF:FFFF`` prefix. Returned
|
|
## value will be IPv4 address.
|
|
##
|
|
## If ``a`` family is IPv6 and ``m`` family is IPv4, masking is still
|
|
## possible when ``a`` holds (IPv4 to IPv6) mapped address. Returned value
|
|
## will be IPv6 address.
|
|
##
|
|
## If ``a`` family is IPv4 and ``m`` family is IPv4, returned value will be
|
|
## IPv4 address.
|
|
##
|
|
## If ``a`` family is IPv6 and ``m`` family is IPv6, returned value will be
|
|
## IPv6 address.
|
|
##
|
|
## In all other cases returned address will have ``AddressFamily.None``.
|
|
if a.family == AddressFamily.IPv4 and m.family == AddressFamily.IPv6:
|
|
if (m.mask6[0] == 0xFFFF_FFFF_FFFF_FFFF'u64) and
|
|
(m.mask6[1] and 0xFFFF_FFFF'u64) == 0xFFFF_FFFF'u64:
|
|
result = TransportAddress(family: a.family)
|
|
let mask = cast[uint32](m.mask6[1] shr 32)
|
|
let data = cast[ptr uint32](unsafeAddr a.address_v4[0])[]
|
|
cast[ptr uint32](addr result.address_v4[0])[] = data and mask
|
|
result.port = a.port
|
|
elif a.family == AddressFamily.IPv6 and m.family == AddressFamily.IPv4:
|
|
var ip = a.toIPv4()
|
|
if ip.family == AddressFamily.IPv4:
|
|
let data = cast[ptr uint32](addr ip.address_v4[0])[]
|
|
cast[ptr uint32](addr ip.address_v4[0])[] = data and m.mask4
|
|
result = ip.toIPv6()
|
|
result.port = a.port
|
|
elif a.family == AddressFamily.IPv4 and m.family == AddressFamily.IPv4:
|
|
result = TransportAddress(family: AddressFamily.IPv4)
|
|
let data = cast[ptr uint32](unsafeAddr a.address_v4[0])[]
|
|
cast[ptr uint32](addr result.address_v4[0])[] = data and m.mask4
|
|
result.port = a.port
|
|
elif a.family == AddressFamily.IPv6 and m.family == AddressFamily.IPv6:
|
|
result = TransportAddress(family: AddressFamily.IPv6)
|
|
let data0 = cast[ptr uint64](unsafeAddr a.address_v6[0])[]
|
|
let data1 = cast[ptr uint64](unsafeAddr a.address_v6[8])[]
|
|
cast[ptr uint64](addr result.address_v6[0])[] = data0 and m.mask6[0]
|
|
cast[ptr uint64](addr result.address_v6[8])[] = data1 and m.mask6[1]
|
|
result.port = a.port
|
|
|
|
proc prefix*(mask: IpMask): int =
|
|
## Returns number of bits set `1` in IP mask ``mask``.
|
|
##
|
|
## Procedure returns ``-1`` if mask is not canonical, e.g. has holes with
|
|
## ``0`` bits between ``1`` bits.
|
|
var hmask = mask.toHostOrder()
|
|
if hmask.family == AddressFamily.IPv4:
|
|
var n = hmask.mask4
|
|
while n != 0:
|
|
if (n and 0x8000_0000'u32) == 0'u32:
|
|
result = -1
|
|
return
|
|
n = n shl 1
|
|
inc(result)
|
|
elif hmask.family == AddressFamily.IPv6:
|
|
if hmask.mask6[0] == 0xFFFF_FFFF_FFFF_FFFF'u64:
|
|
result += 64
|
|
if hmask.mask6[1] == 0xFFFF_FFFF_FFFF_FFFF'u64:
|
|
result += 64:
|
|
else:
|
|
var n = hmask.mask6[1]
|
|
while n != 0:
|
|
if (n and 0x8000_0000_0000_0000'u64) == 0'u64:
|
|
result = -1
|
|
return
|
|
n = n shl 1
|
|
inc(result)
|
|
else:
|
|
var n = hmask.mask6[0]
|
|
while n != 0:
|
|
if (n and 0x8000_0000_0000_0000'u64) == 0'u64:
|
|
result = -1
|
|
return
|
|
n = n shl 1
|
|
inc(result)
|
|
if hmask.mask6[1] != 0x00'u64:
|
|
result = -1
|
|
|
|
proc subnetMask*(mask: IpMask): TransportAddress =
|
|
## Returns TransportAddress representation of IP mask ``mask``.
|
|
result = TransportAddress(family: mask.family)
|
|
if mask.family == AddressFamily.IPv4:
|
|
cast[ptr uint32](addr result.address_v4[0])[] = mask.mask4
|
|
elif mask.family == AddressFamily.IPv6:
|
|
cast[ptr uint64](addr result.address_v6[0])[] = mask.mask6[0]
|
|
cast[ptr uint64](addr result.address_v6[8])[] = mask.mask6[1]
|
|
|
|
proc `$`*(mask: IpMask, include0x = false): string =
|
|
## Returns hexadecimal string representation of IP mask ``mask``.
|
|
var host = mask.toHostOrder()
|
|
result = ""
|
|
if host.family == AddressFamily.IPv4:
|
|
result = if include0x: "0x" else: ""
|
|
var n = 32
|
|
var m = host.mask4
|
|
while n > 0:
|
|
n -= 4
|
|
var c = cast[int]((m shr n) and 0x0F)
|
|
if c < 10:
|
|
result.add(chr(ord('0') + c))
|
|
else:
|
|
result.add(chr(ord('A') + (c - 10)))
|
|
elif host.family == AddressFamily.IPv6:
|
|
result = if include0x: "0x" else: ""
|
|
for i in 0..1:
|
|
var n = 64
|
|
var m = host.mask6[i]
|
|
while n > 0:
|
|
n -= 4
|
|
var c = cast[int]((m shr n) and 0x0F)
|
|
if c < 10:
|
|
result.add(chr(ord('0') + c))
|
|
else:
|
|
result.add(chr(ord('A') + (c - 10)))
|
|
else:
|
|
raise newException(ValueError, "Invalid mask")
|
|
|
|
proc ip*(mask: IpMask): string =
|
|
## Returns IP address text representation of IP mask ``mask``.
|
|
if mask.family == AddressFamily.IPv4:
|
|
var ip = IpAddress(family: IpAddressFamily.IPv4)
|
|
copyMem(addr ip.address_v4[0], unsafeAddr mask.mask4, sizeof(uint32))
|
|
result = $ip
|
|
elif mask.family == AddressFamily.IPv6:
|
|
var ip = IpAddress(family: IpAddressFamily.IPv6)
|
|
copyMem(addr ip.address_v6[0], unsafeAddr mask.mask6[0], 16)
|
|
result = $ip
|
|
else:
|
|
raise newException(ValueError, "Invalid mask")
|
|
|
|
proc init*(t: typedesc[IpNet], host: TransportAddress,
|
|
prefix: int): IpNet {.inline.} =
|
|
## Initialize IP Network using host address ``host`` and prefix length
|
|
## ``prefix``.
|
|
result.mask = IpMask.init(host.family, prefix)
|
|
result.host = host
|
|
|
|
proc init*(t: typedesc[IpNet], host, mask: TransportAddress): IpNet {.inline.} =
|
|
## Initialize IP Network using host address ``host`` and network mask
|
|
## address ``mask``.
|
|
##
|
|
## Note that ``host`` and ``mask`` must be from the same IP family.
|
|
if host.family == mask.family:
|
|
result.mask = IpMask.init(mask)
|
|
result.host = host
|
|
|
|
proc init*(t: typedesc[IpNet], host: TransportAddress,
|
|
mask: IpMask): IpNet {.inline.} =
|
|
## Initialize IP Network using host address ``host`` and network mask
|
|
## ``mask``.
|
|
result.mask = mask
|
|
result.host = host
|
|
|
|
proc init*(t: typedesc[IpNet], network: string): IpNet =
|
|
## Initialize IP Network from string representation in format
|
|
## <address>/<prefix length> or <address>/<netmask address>.
|
|
var parts = network.rsplit("/", maxsplit = 1)
|
|
var host, mhost: TransportAddress
|
|
var ipaddr: IpAddress
|
|
var mask: IpMask
|
|
var prefix: int
|
|
try:
|
|
ipaddr = parseIpAddress(parts[0])
|
|
if ipaddr.family == IpAddressFamily.IPv4:
|
|
host = TransportAddress(family: AddressFamily.IPv4)
|
|
host.address_v4 = ipaddr.address_v4
|
|
prefix = 32
|
|
elif ipaddr.family == IpAddressFamily.IPv6:
|
|
host = TransportAddress(family: AddressFamily.IPv6)
|
|
host.address_v6 = ipaddr.address_v6
|
|
prefix = 128
|
|
if len(parts) > 1:
|
|
try:
|
|
prefix = parseInt(parts[1])
|
|
except:
|
|
prefix = -1
|
|
if prefix == -1:
|
|
ipaddr = parseIpAddress(parts[1])
|
|
if ipaddr.family == IpAddressFamily.IPv4:
|
|
mhost = TransportAddress(family: AddressFamily.IPv4)
|
|
mhost.address_v4 = ipaddr.address_v4
|
|
elif ipaddr.family == IpAddressFamily.IPv6:
|
|
mhost = TransportAddress(family: AddressFamily.IPv6)
|
|
mhost.address_v6 = ipaddr.address_v6
|
|
mask = IpMask.init(mhost)
|
|
if mask.family != host.family:
|
|
raise newException(TransportAddressError,
|
|
"Incorrect network address!")
|
|
else:
|
|
if (ipaddr.family == IpAddressFamily.IPv4 and
|
|
(prefix <= 0 or prefix > 32)) or
|
|
(ipaddr.family == IpAddressFamily.IPv6 and
|
|
(prefix <= 0 or prefix > 128)):
|
|
raise newException(TransportAddressError,
|
|
"Incorrect network address!")
|
|
if prefix == -1:
|
|
result = t.init(host, mask)
|
|
else:
|
|
result = t.init(host, prefix)
|
|
except:
|
|
raise newException(TransportAddressError, "Incorrect network address!")
|
|
|
|
proc `==`*(n1, n2: IpNet): bool {.inline.} =
|
|
## Returns ``true`` if networks ``n1`` and ``n2`` are equal in IP family and
|
|
## by value.
|
|
if n1.host.family == n2.host.family:
|
|
if n1.host.family == AddressFamily.IPv4:
|
|
result = (n1.host.address_v4 == n2.host.address_v4) and
|
|
(n1.mask == n2.mask)
|
|
elif n1.host.family == AddressFamily.IPv6:
|
|
result = (n1.host.address_v6 == n2.host.address_v6) and
|
|
(n1.mask == n2.mask)
|
|
|
|
proc contains*(net: IpNet, address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` belongs to IP Network ``net``
|
|
if net.host.family == address.family:
|
|
var host1 = mask(address, net.mask)
|
|
var host2 = mask(net.host, net.mask)
|
|
host2.port = host1.port
|
|
result = (host1 == host2)
|
|
|
|
proc broadcast*(net: IpNet): TransportAddress =
|
|
## Returns broadcast address for IP Network ``net``.
|
|
result = TransportAddress(family: net.host.family)
|
|
if result.family == AddressFamily.IPv4:
|
|
let address = cast[ptr uint32](unsafeAddr net.host.address_v4[0])[]
|
|
let mask = cast[ptr uint32](unsafeAddr net.mask.mask4)[]
|
|
cast[ptr uint32](addr result.address_v4[0])[] = address or (not(mask))
|
|
elif result.family == AddressFamily.IPv6:
|
|
let address0 = cast[ptr uint64](unsafeAddr net.host.address_v6[0])[]
|
|
let address1 = cast[ptr uint64](unsafeAddr net.host.address_v6[8])[]
|
|
let data0 = cast[ptr uint64](unsafeAddr net.mask.mask6[0])[]
|
|
let data1 = cast[ptr uint64](unsafeAddr net.mask.mask6[1])[]
|
|
cast[ptr uint64](addr result.address_v6[0])[] = address0 or (not(data0))
|
|
cast[ptr uint64](addr result.address_v6[8])[] = address1 or (not(data1))
|
|
|
|
proc subnetMask*(net: IpNet): TransportAddress =
|
|
## Returns netmask address for IP Network ``net``.
|
|
result = TransportAddress(family: net.host.family)
|
|
if result.family == AddressFamily.IPv4:
|
|
let address = cast[ptr uint32](unsafeAddr net.mask.mask4)[]
|
|
cast[ptr uint32](addr result.address_v4[0])[] = address
|
|
elif result.family == AddressFamily.IPv6:
|
|
let address0 = cast[ptr uint64](unsafeAddr net.mask.mask6[0])[]
|
|
let address1 = cast[ptr uint64](unsafeAddr net.mask.mask6[1])[]
|
|
cast[ptr uint64](addr result.address_v6[0])[] = address0
|
|
cast[ptr uint64](addr result.address_v6[8])[] = address1
|
|
|
|
proc network*(net: IpNet): TransportAddress {.inline.} =
|
|
## Returns network address (host address masked with network mask) for
|
|
## IP Network ``net``.
|
|
result = mask(net.host, net.mask)
|
|
|
|
proc `and`*(address1, address2: TransportAddress): TransportAddress =
|
|
## Bitwise ``and`` operation for ``address1 and address2``.
|
|
##
|
|
## Note only IPv4 and IPv6 addresses are supported. ``address1`` and
|
|
## ``address2`` must be in equal IP family
|
|
if address1.family == address2.family:
|
|
if address1.family == AddressFamily.IPv4:
|
|
let data1 = cast[ptr uint32](unsafeAddr address1.address_v4[0])[]
|
|
let data2 = cast[ptr uint32](unsafeAddr address2.address_v4[0])[]
|
|
result = TransportAddress(family: address1.family)
|
|
cast[ptr uint32](addr result.address_v4[0])[] = data1 and data2
|
|
elif address1.family == AddressFamily.IPv6:
|
|
let data1 = cast[ptr uint64](unsafeAddr address1.address_v6[0])[]
|
|
let data2 = cast[ptr uint64](unsafeAddr address1.address_v6[8])[]
|
|
let data3 = cast[ptr uint64](unsafeAddr address2.address_v6[0])[]
|
|
let data4 = cast[ptr uint64](unsafeAddr address2.address_v6[8])[]
|
|
result = TransportAddress(family: address1.family)
|
|
cast[ptr uint64](addr result.address_v6[0])[] = data1 and data3
|
|
cast[ptr uint64](addr result.address_v6[8])[] = data2 and data4
|
|
|
|
proc `or`*(address1, address2: TransportAddress): TransportAddress =
|
|
## Bitwise ``or`` operation for ``address1 or address2``.
|
|
##
|
|
## Note only IPv4 and IPv6 addresses are supported. ``address1`` and
|
|
## ``address2`` must be in equal IP family
|
|
if address1.family == address2.family:
|
|
if address1.family == AddressFamily.IPv4:
|
|
let data1 = cast[ptr uint32](unsafeAddr address1.address_v4[0])[]
|
|
let data2 = cast[ptr uint32](unsafeAddr address2.address_v4[0])[]
|
|
result = TransportAddress(family: address1.family)
|
|
cast[ptr uint32](addr result.address_v4[0])[] = data1 or data2
|
|
elif address1.family == AddressFamily.IPv6:
|
|
let data1 = cast[ptr uint64](unsafeAddr address1.address_v6[0])[]
|
|
let data2 = cast[ptr uint64](unsafeAddr address1.address_v6[8])[]
|
|
let data3 = cast[ptr uint64](unsafeAddr address2.address_v6[0])[]
|
|
let data4 = cast[ptr uint64](unsafeAddr address2.address_v6[8])[]
|
|
result = TransportAddress(family: address1.family)
|
|
cast[ptr uint64](addr result.address_v6[0])[] = data1 or data3
|
|
cast[ptr uint64](addr result.address_v6[8])[] = data2 or data4
|
|
|
|
proc `not`*(address: TransportAddress): TransportAddress =
|
|
## Bitwise ``not`` operation for ``address``.
|
|
if address.family == AddressFamily.IPv4:
|
|
let data = cast[ptr uint32](unsafeAddr address.address_v4[0])[]
|
|
result = TransportAddress(family: address.family)
|
|
cast[ptr uint32](addr result.address_v4[0])[] = not(data)
|
|
elif address.family == AddressFamily.IPv6:
|
|
let data1 = cast[ptr uint64](unsafeAddr address.address_v6[0])[]
|
|
let data2 = cast[ptr uint64](unsafeAddr address.address_v6[8])[]
|
|
result = TransportAddress(family: address.family)
|
|
cast[ptr uint64](addr result.address_v6[0])[] = not(data1)
|
|
cast[ptr uint64](addr result.address_v6[8])[] = not(data2)
|
|
|
|
proc `+`*(address: TransportAddress, v: uint): TransportAddress =
|
|
## Add to IPv4/IPv6 transport ``address`` unsigned integer ``v``.
|
|
result = TransportAddress(family: address.family)
|
|
if address.family == AddressFamily.IPv4:
|
|
var a: uint64
|
|
let data = cast[ptr uint32](unsafeAddr address.address_v4[0])
|
|
when system.cpuEndian == bigEndian:
|
|
a = data
|
|
else:
|
|
swapEndian32(addr a, data)
|
|
a = a + v
|
|
bigEndian32(cast[pointer](addr result.address_v4[0]), addr a)
|
|
elif address.family == AddressFamily.IPv6:
|
|
var a1, a2: uint64
|
|
let data1 = cast[ptr uint64](unsafeAddr address.address_v6[0])
|
|
let data2 = cast[ptr uint64](unsafeAddr address.address_v6[8])
|
|
when system.cpuEndian == bigEndian:
|
|
a1 = data1
|
|
a2 = data2
|
|
else:
|
|
swapEndian64(addr a1, data1)
|
|
swapEndian64(addr a2, data2)
|
|
var a3 = a2 + v
|
|
if a3 < a2:
|
|
## Overflow
|
|
a1 = a1 + 1
|
|
bigEndian64(cast[pointer](addr result.address_v6[0]), addr a1)
|
|
bigEndian64(cast[pointer](addr result.address_v6[8]), addr a3)
|
|
|
|
proc inc*(address: var TransportAddress, v: uint = 1'u) =
|
|
## Increment IPv4/IPv6 transport ``address`` by unsigned integer ``v``.
|
|
address = address + v
|
|
|
|
proc `$`*(net: IpNet): string =
|
|
## Return string representation of IP network in format:
|
|
## <IPv4 or IPv6 address>/<prefix length>.
|
|
if net.host.family == AddressFamily.IPv4:
|
|
var a = IpAddress(family: IpAddressFamily.IPv4,
|
|
address_v4: net.host.address_v4)
|
|
result = $a
|
|
result.add("/")
|
|
let prefix = net.mask.prefix()
|
|
if prefix == -1:
|
|
result.add(net.mask.ip())
|
|
else:
|
|
result.add($prefix)
|
|
elif net.host.family == AddressFamily.IPv6:
|
|
var a = IpAddress(family: IpAddressFamily.IPv6,
|
|
address_v6: net.host.address_v6)
|
|
result = $a
|
|
result.add("/")
|
|
let prefix = net.mask.prefix()
|
|
if prefix == -1:
|
|
result.add(net.mask.ip())
|
|
else:
|
|
result.add($prefix)
|
|
|
|
proc isUnspecified*(address: TransportAddress): bool {.inline.} =
|
|
## Returns ``true`` if ``address`` is not specified yet, e.g. its ``family``
|
|
## field is not set or equal to ``AddressFamily.None``.
|
|
if address.family == AddressFamily.None:
|
|
result = true
|
|
|
|
proc isMulticast*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is a multicast address.
|
|
##
|
|
## ``IPv4``: 224.0.0.0 - 239.255.255.255
|
|
##
|
|
## ``IPv6``: FF00:: - FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF
|
|
if address.family == AddressFamily.IPv4:
|
|
result = ((address.address_v4[0] and 0xF0'u8) == 0xE0'u8)
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFF'u8)
|
|
|
|
proc isInterfaceLocalMulticast*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is interface local multicast address.
|
|
##
|
|
## ``IPv4``: N/A (always returns ``false``)
|
|
if address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFF'u8) and
|
|
((address.address_v6[1] and 0x0F'u8) == 0x01'u8)
|
|
|
|
proc isLinkLocalMulticast*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address` is link local multicast address.
|
|
##
|
|
## ``IPv4``: 224.0.0.0 - 224.0.0.255
|
|
if address.family == AddressFamily.IPv4:
|
|
result = (address.address_v4[0] == 224'u8) and
|
|
(address.address_v4[1] == 0'u8) and
|
|
(address.address_v4[2] == 0'u8)
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFF'u8) and
|
|
((address.address_v6[1] and 0x0F'u8) == 0x02'u8)
|
|
|
|
proc isLoopback*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is loopback address.
|
|
##
|
|
## ``IPv4``: 127.0.0.0 - 127.255.255.255
|
|
##
|
|
## ``IPv6``: ::1
|
|
if address.family == AddressFamily.IPv4:
|
|
result = (address.address_v4[0] == 127'u8)
|
|
elif address.family == AddressFamily.IPv6:
|
|
var test = 0
|
|
for i in 0..<(len(address.address_v6) - 1):
|
|
test = test or cast[int](address.address_v6[i])
|
|
result = (test == 0) and (address.address_v6[15] == 1'u8)
|
|
|
|
proc isAnyLocal*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is a wildcard address.
|
|
##
|
|
## ``IPv4``: 0.0.0.0
|
|
##
|
|
## ``IPv6``: ::
|
|
if address.family == AddressFamily.IPv4:
|
|
let data = cast[ptr uint32](unsafeAddr address.address_v4[0])[]
|
|
result = (data == 0'u32)
|
|
elif address.family == AddressFamily.IPv6:
|
|
let data1 = cast[ptr uint32](unsafeAddr address.address_v6[0])[]
|
|
let data2 = cast[ptr uint32](unsafeAddr address.address_v6[4])[]
|
|
let data3 = cast[ptr uint32](unsafeAddr address.address_v6[8])[]
|
|
let data4 = cast[ptr uint32](unsafeAddr address.address_v6[12])[]
|
|
result = ((data1 or data2 or data3 or data4) == 0'u32)
|
|
|
|
proc isLinkLocal*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is link local address.
|
|
##
|
|
## ``IPv4``: 169.254.0.0 - 169.254.255.255
|
|
##
|
|
## ``IPv6``: FE80:: - FEBF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF
|
|
if address.family == AddressFamily.IPv4:
|
|
result = (address.address_v4[0] == 169'u8) and
|
|
(address.address_v4[1] == 254'u8)
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFE'u8) and
|
|
((address.address_v6[1] and 0xC0'u8) == 0x80'u8)
|
|
|
|
proc isLinkLocalUnicast*(address: TransportAddress): bool {.inline.} =
|
|
result = isLinkLocal(address)
|
|
|
|
proc isSiteLocal*(address: TransportAddress): bool =
|
|
## Returns ``true`` if ``address`` is site local address.
|
|
##
|
|
## ``IPv4``: 10.0.0.0 - 10.255.255.255, 172.16.0.0 - 172.31.255.255,
|
|
## 192.168.0.0 - 192.168.255.255
|
|
##
|
|
## ``IPv6``: FEC0:: - FEFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF
|
|
if address.family == AddressFamily.IPv4:
|
|
result = (address.address_v4[0] == 10'u8) or
|
|
((address.address_v4[0] == 172'u8) and
|
|
((address.address_v4[1] and 0xF0) == 16)) or
|
|
((address.address_v4[0] == 192'u8) and
|
|
((address.address_v4[1] == 168'u8)))
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFE'u8) and
|
|
((address.address_v6[1] and 0xC0'u8) == 0xC0'u8)
|
|
|
|
proc isGlobalMulticast*(address: TransportAddress): bool =
|
|
## Returns ``true`` if the multicast address has global scope.
|
|
##
|
|
## ``IPv4``: 224.0.1.0 - 238.255.255.255
|
|
##
|
|
## ``IPv6``: FF0E:: - FFFE:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF
|
|
if address.family == AddressFamily.IPv4:
|
|
result = (address.address_v4[0] >= 224'u8) and
|
|
(address.address_v4[0] <= 238'u8) and
|
|
not(
|
|
(address.address_v4[0] == 224'u8) and
|
|
(address.address_v4[1] == 0'u8) and
|
|
(address.address_v4[2] == 0'u8)
|
|
)
|
|
elif address.family == AddressFamily.IPv6:
|
|
result = (address.address_v6[0] == 0xFF'u8) and
|
|
((address.address_v6[1] and 0x0F'u8) == 0x0E'u8)
|