220 lines
8.0 KiB
Nim
220 lines
8.0 KiB
Nim
|
import unittest, strutils
|
||
|
|
import ../libp2p/ipnet/ipnet
|
||
|
|
|
||
|
|
const MaskVectors = [
|
||
|
|
["192.168.1.127:1024", "255.255.255.128", "192.168.1.0:1024"],
|
||
|
|
["192.168.1.127:1024", "255.255.255.192", "192.168.1.64:1024"],
|
||
|
|
["192.168.1.127:1024", "255.255.255.224", "192.168.1.96:1024"],
|
||
|
|
["192.168.1.127:1024", "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ff80",
|
||
|
|
"192.168.1.0:1024"],
|
||
|
|
["192.168.1.127:1024", "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffc0",
|
||
|
|
"192.168.1.64:1024"],
|
||
|
|
["192.168.1.127:1024", "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffe0",
|
||
|
|
"192.168.1.96:1024"],
|
||
|
|
["192.168.1.127:1024", "255.0.255.0", "192.0.1.0:1024"],
|
||
|
|
["[2001:db8::1]:1024", "ffff:ff80::", "[2001:d80::]:1024"],
|
||
|
|
["[2001:db8::1]:1024", "f0f0:0f0f::", "[2000:d08::]:1024"]
|
||
|
|
]
|
||
|
|
|
||
|
|
const NonCanonicalMasks = [
|
||
|
|
["ip", "0.255.255.255", "-1"],
|
||
|
|
["ip", "255.0.255.255", "-1"],
|
||
|
|
["ip", "255.255.0.255", "-1"],
|
||
|
|
["ip", "255.255.255.0", "24"],
|
||
|
|
["ms", "0FFFFFFF", "-1"],
|
||
|
|
["ms", "F0FFFFFF", "-1"],
|
||
|
|
["ms", "FF0FFFFF", "-1"],
|
||
|
|
["ms", "FFF0FFFF", "-1"],
|
||
|
|
["ms", "FFFF0FFF", "-1"],
|
||
|
|
["ms", "FFFFF0FF", "-1"],
|
||
|
|
["ms", "FFFFFF0F", "-1"],
|
||
|
|
["ms", "FFFFFFF0", "28"],
|
||
|
|
["ip", "00FF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FF00:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:00FF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FF00:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:00FF:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FF00:FFFF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:00FF:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FF00:FFFF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:00FF:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FF00:FFFF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:00FF:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:FF00:FFFF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:00FF:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FF00:FFFF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:00FF", "-1"],
|
||
|
|
["ip", "FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FFFF:FF00", "120"],
|
||
|
|
["ms", "0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "F0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FF0FFFFFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFF0FFFFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFF0FFFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFF0FFFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFF0FFFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFF0FFFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFF0FFFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFF0FFFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFF0FFFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFF0FFFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFF0FFFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFF0FFFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFF0FFFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFF0FFFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFF0FFFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFF0FFFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFF0FFFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFF0FFFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFF0FFFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFF0FFFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFF0FFFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFF0FFFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFF0FFFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFF0FFFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFF0FFFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFFF0FFFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFFFF0FFF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFFFFF0FF", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF0F", "-1"],
|
||
|
|
["ms", "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF0", "124"],
|
||
|
|
]
|
||
|
|
|
||
|
|
const NetworkVectors = [
|
||
|
|
["135.104.0.0/32", "true", "135.104.0.0:0", "FFFFFFFF"],
|
||
|
|
["0.0.0.0/24", "true", "0.0.0.0:0", "FFFFFF00"],
|
||
|
|
["135.104.0.0/24", "true", "135.104.0.0:0", "FFFFFF00"],
|
||
|
|
["135.104.0.1/32", "true", "135.104.0.1:0", "FFFFFFFF"],
|
||
|
|
["135.104.0.1/24", "true", "135.104.0.1:0", "FFFFFF00"],
|
||
|
|
["::1/128", "true", "[::1]:0", "FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"],
|
||
|
|
["abcd:2345::/127", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFE"],
|
||
|
|
["abcd:2345::/65", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFFFFFFFFFF8000000000000000"],
|
||
|
|
["abcd:2345::/64", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFFFFFFFFFF0000000000000000"],
|
||
|
|
["abcd:2345::/63", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFFFFFFFFFE0000000000000000"],
|
||
|
|
["abcd:2345::/33", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFF800000000000000000000000"],
|
||
|
|
["abcd:2345::/32", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFFFF000000000000000000000000"],
|
||
|
|
["abcd:2344::/31", "true", "[abcd:2344::]:0",
|
||
|
|
"FFFFFFFE000000000000000000000000"],
|
||
|
|
["abcd:2300::/24", "true", "[abcd:2300::]:0",
|
||
|
|
"FFFFFF00000000000000000000000000"],
|
||
|
|
["abcd:2345::/24", "true", "[abcd:2345::]:0",
|
||
|
|
"FFFFFF00000000000000000000000000"],
|
||
|
|
["2001:db8::/48", "true", "[2001:db8::]:0",
|
||
|
|
"FFFFFFFFFFFF00000000000000000000"],
|
||
|
|
["2001:db8::1/48", "true", "[2001:db8::1]:0",
|
||
|
|
"FFFFFFFFFFFF00000000000000000000"],
|
||
|
|
["192.168.1.1/255.255.255.0", "true", "192.168.1.1:0", "FFFFFF00"],
|
||
|
|
["192.168.1.1/35", "false", "", ""],
|
||
|
|
["2001:db8::1/-1", "false", "", ""],
|
||
|
|
["2001:db8::1/-0", "false", "", ""],
|
||
|
|
["-0.0.0.0/32", "false", "", ""],
|
||
|
|
["0.-1.0.0/32", "false", "", ""],
|
||
|
|
["0.0.-2.0/32", "false", "", ""],
|
||
|
|
["0.0.0.-3/32", "false", "", ""],
|
||
|
|
["0.0.0.0/-0", "false", "", ""],
|
||
|
|
["", "false", "", ""]
|
||
|
|
]
|
||
|
|
|
||
|
|
const NetworkContainsVectors = [
|
||
|
|
["172.16.1.1:1024", "172.16.0.0/12", "true"],
|
||
|
|
["172.24.0.1:1024", "172.16.0.0/13", "false"],
|
||
|
|
["192.168.0.3:1024", "192.168.0.0/0.0.255.252", "true"],
|
||
|
|
["192.168.0.4:1024", "192.168.0.0/0.255.0.252", "false"],
|
||
|
|
["[2001:db8:1:2::1]:1024", "2001:db8:1::/47", "true"],
|
||
|
|
["[2001:db8:1:2::1]:1024", "2001:db8:2::/47", "false"],
|
||
|
|
["[2001:db8:1:2::1]:1024", "2001:db8:1::/ffff:0:ffff::", "true"],
|
||
|
|
["[2001:db8:1:2::1]:1024", "2001:db8:1::/0:0:0:ffff::", "false"]
|
||
|
|
]
|
||
|
|
|
||
|
|
suite "IP network utilities test suite":
|
||
|
|
|
||
|
|
test "IpMask test vectors":
|
||
|
|
for item in MaskVectors:
|
||
|
|
var a = initTAddress(item[0])
|
||
|
|
var m = IpMask.initIp(item[1])
|
||
|
|
var r = a.mask(m)
|
||
|
|
check $r == item[2]
|
||
|
|
|
||
|
|
test "IpMask serialization/deserialization test":
|
||
|
|
for i in 1..32:
|
||
|
|
var m = IpMask.init(AddressFamily.IPv4, i)
|
||
|
|
check m.prefix() == i
|
||
|
|
var s0x = `$`(m, true)
|
||
|
|
var s = $m
|
||
|
|
var sip = m.ip()
|
||
|
|
var m1 = IpMask.init(s0x)
|
||
|
|
var m2 = IpMask.init(s)
|
||
|
|
var m3 = IpMask.initIp(sip)
|
||
|
|
check:
|
||
|
|
m == m1
|
||
|
|
m == m2
|
||
|
|
m == m3
|
||
|
|
for i in 1..128:
|
||
|
|
var m = IpMask.init(AddressFamily.IPv6, i)
|
||
|
|
check m.prefix() == i
|
||
|
|
var s0x = `$`(m, true)
|
||
|
|
var s = $m
|
||
|
|
var sip = m.ip()
|
||
|
|
var m1 = IpMask.init(s0x)
|
||
|
|
var m2 = IpMask.init(s)
|
||
|
|
var m3 = IpMask.initIp(sip)
|
||
|
|
check:
|
||
|
|
m == m1
|
||
|
|
m == m2
|
||
|
|
m == m3
|
||
|
|
|
||
|
|
test "IpMask non-canonical masks":
|
||
|
|
for item in NonCanonicalMasks:
|
||
|
|
var m: IpMask
|
||
|
|
if item[0] == "ip":
|
||
|
|
m = IpMask.initIp(item[1])
|
||
|
|
elif item[0] == "ms":
|
||
|
|
m = IpMask.init(item[1])
|
||
|
|
var c = $(m.prefix())
|
||
|
|
check:
|
||
|
|
c == item[2]
|
||
|
|
|
||
|
|
test "IpNet test vectors":
|
||
|
|
for item in NetworkVectors:
|
||
|
|
var res: bool
|
||
|
|
var inet: IpNet
|
||
|
|
try:
|
||
|
|
inet = IpNet.init(item[0])
|
||
|
|
res = true
|
||
|
|
except:
|
||
|
|
res = false
|
||
|
|
check:
|
||
|
|
$res == item[1]
|
||
|
|
if res:
|
||
|
|
check:
|
||
|
|
$inet.host == item[2]
|
||
|
|
$inet.mask == $item[3]
|
||
|
|
|
||
|
|
test "IpNet contains test vectors":
|
||
|
|
for item in NetworkContainsVectors:
|
||
|
|
var a = initTAddress(item[0])
|
||
|
|
var n = IpNet.init(item[1])
|
||
|
|
var res = a in n
|
||
|
|
check:
|
||
|
|
$res == item[2]
|
||
|
|
|
||
|
|
test "IpNet serialization/deserialization test":
|
||
|
|
var ip4 = initTAddress("192.168.1.0:1024")
|
||
|
|
for i in 1..32:
|
||
|
|
var net = IpNet.init(ip4, i)
|
||
|
|
var s1 = $net
|
||
|
|
var net2 = IpNet.init(s1)
|
||
|
|
check net == net2
|
||
|
|
|
||
|
|
var ip6 = initTAddress("[8000:f123:f456:cafe::]:1024")
|
||
|
|
for i in 1..128:
|
||
|
|
var net = IpNet.init(ip6, i)
|
||
|
|
var s1 = $net
|
||
|
|
var net2 = IpNet.init(s1)
|
||
|
|
check net == net2
|