diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 09:34:30 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 09:34:30 +0000 |
commit | 4fc2f55f761d71aae1f145d5aa94ba929cc39676 (patch) | |
tree | 5c1e1db3b46dd4edbe11f612d93cb94b96891ce3 /test-iputils_hh.cc | |
parent | Initial commit. (diff) | |
download | dnsdist-upstream.tar.xz dnsdist-upstream.zip |
Adding upstream version 1.7.3.upstream/1.7.3upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | test-iputils_hh.cc | 855 |
1 files changed, 855 insertions, 0 deletions
diff --git a/test-iputils_hh.cc b/test-iputils_hh.cc new file mode 100644 index 0000000..36e6577 --- /dev/null +++ b/test-iputils_hh.cc @@ -0,0 +1,855 @@ +#define BOOST_TEST_DYN_LINK +#define BOOST_TEST_NO_MAIN +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#include <boost/test/unit_test.hpp> +#include <bitset> +#include "iputils.hh" + +using namespace boost; + +BOOST_AUTO_TEST_SUITE(test_iputils_hh) + +BOOST_AUTO_TEST_CASE(test_ComboAddress) { + ComboAddress local("127.0.0.1", 53); + BOOST_CHECK(local==local); + BOOST_CHECK_EQUAL(local.sin4.sin_family, AF_INET); + BOOST_CHECK_EQUAL(local.sin4.sin_port, htons(53)); + BOOST_CHECK_EQUAL(local.sin4.sin_addr.s_addr, htonl(0x7f000001UL)); + + ComboAddress remote("130.161.33.15", 53); + BOOST_CHECK(!(local == remote)); + BOOST_CHECK_EQUAL(remote.sin4.sin_port, htons(53)); + + ComboAddress withport("213.244.168.210:53"); + BOOST_CHECK_EQUAL(withport.sin4.sin_port, htons(53)); + + ComboAddress withportO("213.244.168.210:53", 5300); + BOOST_CHECK_EQUAL(withportO.sin4.sin_port, htons(53)); + + withport = ComboAddress("[::]:53"); + BOOST_CHECK_EQUAL(withport.sin4.sin_port, htons(53)); + + withport = ComboAddress("[::]:5300", 53); + BOOST_CHECK_EQUAL(withport.sin4.sin_port, htons(5300)); + + ComboAddress defaultport("213.244.168.210"); + BOOST_CHECK_EQUAL(defaultport.sin4.sin_port, htons(0)); + + defaultport = ComboAddress("[::1]"); + BOOST_CHECK_EQUAL(defaultport.sin4.sin_port, htons(0)); + + defaultport = ComboAddress("::1"); + BOOST_CHECK_EQUAL(defaultport.sin4.sin_port, htons(0)); + + // Verify that 2 'empty' ComboAddresses are equal, used in syncres.hh to + // signal auth-zones + ComboAddress a = ComboAddress(); + ComboAddress b = ComboAddress(); + BOOST_CHECK(a == b); + + // Verify that 2 ComboAddresses are not the same + ComboAddress c = ComboAddress("127.0.0.1:53"); + ComboAddress d = ComboAddress("127.0.0.1:52"); + ComboAddress e = ComboAddress("127.0.0.2:53"); + + BOOST_CHECK(a != c); + BOOST_CHECK(c != d); + BOOST_CHECK(c != e); + BOOST_CHECK(d != e); + BOOST_CHECK(!(a != b)); + + // Verify that we don't allow invalid port numbers + BOOST_CHECK_THROW(ComboAddress("127.0.0.1:70000"), PDNSException); // Port no. too high + BOOST_CHECK_THROW(ComboAddress("127.0.0.1:-6"), PDNSException); // Port no. too low + BOOST_CHECK_THROW(ComboAddress("[::1]:70000"), PDNSException); // Port no. too high + BOOST_CHECK_THROW(ComboAddress("[::1]:-6"), PDNSException); // Port no. too low +} + +BOOST_AUTO_TEST_CASE(test_ComboAddressCompare) { + ComboAddress a, b; + a.reset(); + b.reset(); + BOOST_CHECK(!(a<b)); + BOOST_CHECK(!(a>b)); +} + +BOOST_AUTO_TEST_CASE(test_ComboAddressTruncate) { + ComboAddress ca4("130.161.252.29"); + ca4.truncate(24); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.0"); + ca4.truncate(16); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.0.0"); + + + + ca4 = ComboAddress("130.161.252.29"); + ComboAddress orig(ca4); + for(int n=32; n; --n) { + ca4.truncate(n); + + uint32_t p; + memcpy(&p, (char*)&ca4.sin4.sin_addr.s_addr, 4); + std::bitset<32> result(htonl(p)); + + memcpy(&p, (char*)&orig.sin4.sin_addr.s_addr, 4); + std::bitset<32> manual(htonl(p)); + + auto tokill=32-n; + for(int i =0; i< tokill; ++i) + manual.set(i, 0); + + BOOST_CHECK_EQUAL(result, manual); + } + + ca4 = ComboAddress("130.161.252.29"); + ca4.truncate(31); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.28"); + + ca4.truncate(30); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.28"); + + ca4.truncate(29); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.24"); + + ca4.truncate(23); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.0"); + + ca4.truncate(22); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.252.0"); + + ca4.truncate(21); + BOOST_CHECK_EQUAL(ca4.toString(), "130.161.248.0"); + + ComboAddress ca6("2001:888:2000:1d::2"); + ca6.truncate(120); + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888:2000:1d::"); + ca6.truncate(64); + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888:2000:1d::"); + ca6.truncate(72); // 0102 304 0506 78 + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888:2000:1d::"); + ca6.truncate(56); + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888:2000::"); + ca6.truncate(48); + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888:2000::"); + ca6.truncate(32); + BOOST_CHECK_EQUAL(ca6.toString(), "2001:888::"); + ca6.truncate(16); + BOOST_CHECK_EQUAL(ca6.toString(), "2001::"); + ca6.truncate(8); + BOOST_CHECK_EQUAL(ca6.toString(), "2000::"); + + + orig=ca6=ComboAddress("2001:888:2000:1d::2"); + for(int n=128; n; --n) { + ca6.truncate(n); + + std::bitset<128> result, manual; + for(int i=0; i < 16; ++i) { + result<<=8; + result|= std::bitset<128>(*((unsigned char*)&ca6.sin6.sin6_addr.s6_addr + i)); + + manual<<=8; + manual|= std::bitset<128>(*((unsigned char*)&orig.sin6.sin6_addr.s6_addr + i)); + } + + auto tokill=128-n; + for(int i =0; i< tokill; ++i) + manual.set(i, 0); + + BOOST_CHECK_EQUAL(result, manual); + } +} + +BOOST_AUTO_TEST_CASE(test_Mapping) +{ + ComboAddress lh("::1"); + BOOST_CHECK_EQUAL(lh.toString(), "::1"); +} + +BOOST_AUTO_TEST_CASE(test_Netmask) { + ComboAddress local("127.0.0.1", 53); + ComboAddress remote("130.161.252.29", 53); + + Netmask nm("127.0.0.1/24"); + BOOST_CHECK(nm.getBits() == 24); + BOOST_CHECK(nm.match(local)); + BOOST_CHECK(!nm.match(remote)); + BOOST_CHECK(nm.isIPv4()); + BOOST_CHECK(!nm.isIPv6()); + + Netmask nm6("fe80::92fb:a6ff:fe4a:51da/64"); + BOOST_CHECK(nm6.getBits() == 64); + BOOST_CHECK(nm6.match("fe80::92fb:a6ff:fe4a:51db")); + BOOST_CHECK(!nm6.match("fe81::92fb:a6ff:fe4a:51db")); + BOOST_CHECK(!nm6.isIPv4()); + BOOST_CHECK(nm6.isIPv6()); + + Netmask nmp("130.161.252.29/32"); + BOOST_CHECK(nmp.match(remote)); + + Netmask nmp6("fe80::92fb:a6ff:fe4a:51da/128"); + BOOST_CHECK(nmp6.match("fe80::92fb:a6ff:fe4a:51da")); + BOOST_CHECK(!nmp6.match("fe81::92fb:a6ff:fe4a:51db")); + + Netmask all("0.0.0.0/0"); + BOOST_CHECK(all.match(local) && all.match(remote)); + + Netmask all6("::/0"); + BOOST_CHECK(all6.match("::1") && all6.match("fe80::92fb:a6ff:fe4a:51da")); + + + Netmask fromCombo1(ComboAddress("192.0.2.1:53"), 32); + Netmask fromCombo2(ComboAddress("192.0.2.1:54"), 32); + BOOST_CHECK(fromCombo1 == fromCombo2); + BOOST_CHECK(fromCombo1.match("192.0.2.1")); + BOOST_CHECK(fromCombo1.match(ComboAddress("192.0.2.1:80"))); + BOOST_CHECK(fromCombo1.getNetwork() == ComboAddress("192.0.2.1")); + BOOST_CHECK(fromCombo1.getMaskedNetwork() == ComboAddress("192.0.2.1")); + + Netmask nm25("192.0.2.255/25"); + BOOST_CHECK(nm25.getBits() == 25); + BOOST_CHECK(nm25.getNetwork() == ComboAddress("192.0.2.128")); + BOOST_CHECK(nm25.getMaskedNetwork() == ComboAddress("192.0.2.128")); + + /* Make sure that more specific Netmasks are lesser than less specific ones, + as this is very useful when matching. */ + Netmask specific32("192.0.0.0/32"); + Netmask specific24("192.0.0.0/24"); + Netmask specific16("192.0.0.0/16"); + BOOST_CHECK(specific32 < specific24); + BOOST_CHECK(specific24 > specific32); + BOOST_CHECK(specific24 < specific16); + BOOST_CHECK(specific16 > specific24); + + Netmask sameMask1("192.0.0.0/16"); + Netmask sameMask2("192.0.0.1/16"); + BOOST_CHECK(!(sameMask1 < sameMask2)); + BOOST_CHECK(!(sameMask2 > sameMask1)); + BOOST_CHECK(sameMask1 == sameMask2); + + Netmask nm1921("192.1.255.255/16"); + Netmask nm1922("192.2.255.255/16"); + BOOST_CHECK(!(nm1921 == nm1922)); + BOOST_CHECK(nm1921 < nm1922); + BOOST_CHECK(nm1922 > nm1921); + + /* An empty Netmask should be larger than + every others. */ + Netmask empty = Netmask(); + Netmask full("255.255.255.255/32"); + BOOST_CHECK(empty > all); + BOOST_CHECK(all < empty); + BOOST_CHECK(empty > full); + BOOST_CHECK(full < empty); +} + +static std::string NMGOutputToSorted(const std::string& str) +{ + std::vector<std::string> vect; + stringtok(vect, str, ", "); + std::sort(vect.begin(), vect.end()); + std::string result; + for (const auto& entry : vect) { + if (!result.empty()) { + result += " "; + } + result += entry; + } + + return result; +} + +BOOST_AUTO_TEST_CASE(test_NetmaskGroup) { + + { + NetmaskGroup ng; + BOOST_CHECK_EQUAL(ng.empty(), true); + BOOST_CHECK_EQUAL(ng.size(), 0U); + ng.addMask("10.0.1.0"); + BOOST_CHECK_EQUAL(ng.empty(), false); + BOOST_CHECK_EQUAL(ng.size(), 1U); + BOOST_CHECK(ng.match(ComboAddress("10.0.1.0"))); + ng.toMasks("127.0.0.0/8, 10.0.0.0/24"); + BOOST_CHECK_EQUAL(ng.size(), 3U); + BOOST_CHECK(ng.match(ComboAddress("127.0.0.1"))); + BOOST_CHECK(ng.match(ComboAddress("10.0.0.3"))); + BOOST_CHECK(ng.match(ComboAddress("10.0.1.0"))); + BOOST_CHECK(!ng.match(ComboAddress("128.1.2.3"))); + BOOST_CHECK(!ng.match(ComboAddress("10.0.1.1"))); + BOOST_CHECK(!ng.match(ComboAddress("::1"))); + ng.addMask("::1"); + BOOST_CHECK_EQUAL(ng.size(), 4U); + BOOST_CHECK(ng.match(ComboAddress("::1"))); + BOOST_CHECK(!ng.match(ComboAddress("::2"))); + ng.addMask("fe80::/16"); + BOOST_CHECK_EQUAL(ng.size(), 5U); + BOOST_CHECK(ng.match(ComboAddress("fe80::1"))); + BOOST_CHECK(!ng.match(ComboAddress("fe81::1"))); + BOOST_CHECK_EQUAL(NMGOutputToSorted(ng.toString()), NMGOutputToSorted("10.0.1.0/32, 127.0.0.0/8, 10.0.0.0/24, ::1/128, fe80::/16")); + + /* negative entries using the explicit flag */ + ng.addMask("172.16.0.0/16", true); + BOOST_CHECK_EQUAL(ng.size(), 6U); + BOOST_CHECK(ng.match(ComboAddress("172.16.1.1"))); + BOOST_CHECK(ng.match(ComboAddress("172.16.4.50"))); + ng.addMask("172.16.4.0/24", false); + BOOST_CHECK_EQUAL(ng.size(), 7U); + BOOST_CHECK(ng.match(ComboAddress("172.16.1.1"))); + BOOST_CHECK(!ng.match(ComboAddress("172.16.4.50"))); + ng.addMask("fe80::/24", false); + BOOST_CHECK_EQUAL(ng.size(), 8U); + BOOST_CHECK(!ng.match(ComboAddress("fe80::1"))); + BOOST_CHECK(!ng.match(ComboAddress("fe81::1"))); + /* not in fe80::/24 but in fe80::/16, should match */ + BOOST_CHECK(ng.match(ComboAddress("fe80:0100::1"))); + + /* negative entries using '!' */ + BOOST_CHECK(ng.match(ComboAddress("172.16.10.80"))); + ng.addMask("!172.16.10.0/24"); + BOOST_CHECK_EQUAL(ng.size(), 9U); + BOOST_CHECK(!ng.match(ComboAddress("172.16.10.80"))); + ng.addMask("2001:db8::/32"); + BOOST_CHECK_EQUAL(ng.size(), 10U); + ng.addMask("!2001:db8::/64"); + BOOST_CHECK_EQUAL(ng.size(), 11U); + BOOST_CHECK(!ng.match(ComboAddress("2001:db8::1"))); + /* not in 2001:db8::/64 but in 2001:db8::/32, should match */ + BOOST_CHECK(ng.match(ComboAddress("2001:db8:1::1"))); + + BOOST_CHECK_EQUAL(NMGOutputToSorted(ng.toString()), NMGOutputToSorted("10.0.1.0/32, 127.0.0.0/8, 10.0.0.0/24, ::1/128, fe80::/16, 172.16.0.0/16, !172.16.4.0/24, !fe80::/24, !172.16.10.0/24, 2001:db8::/32, !2001:db8::/64")); + } + + { + /* this time using Netmask objects instead of strings */ + NetmaskGroup ng; + BOOST_CHECK_EQUAL(ng.empty(), true); + BOOST_CHECK_EQUAL(ng.size(), 0U); + ng.addMask(Netmask("10.0.1.0")); + BOOST_CHECK_EQUAL(ng.empty(), false); + BOOST_CHECK_EQUAL(ng.size(), 1U); + BOOST_CHECK(ng.match(ComboAddress("10.0.1.0"))); + ng.addMask(Netmask("127.0.0.0/8")); + BOOST_CHECK_EQUAL(ng.size(), 2U); + ng.addMask(Netmask("10.0.0.0/24")); + BOOST_CHECK_EQUAL(ng.size(), 3U); + BOOST_CHECK(ng.match(ComboAddress("127.0.0.1"))); + BOOST_CHECK(ng.match(ComboAddress("10.0.0.3"))); + BOOST_CHECK(ng.match(ComboAddress("10.0.1.0"))); + BOOST_CHECK(!ng.match(ComboAddress("128.1.2.3"))); + BOOST_CHECK(!ng.match(ComboAddress("10.0.1.1"))); + BOOST_CHECK(!ng.match(ComboAddress("::1"))); + ng.addMask(Netmask("::1")); + BOOST_CHECK_EQUAL(ng.size(), 4U); + BOOST_CHECK(ng.match(ComboAddress("::1"))); + BOOST_CHECK(!ng.match(ComboAddress("::2"))); + ng.addMask(Netmask("fe80::/16")); + BOOST_CHECK_EQUAL(ng.size(), 5U); + BOOST_CHECK(ng.match(ComboAddress("fe80::1"))); + BOOST_CHECK(!ng.match(ComboAddress("fe81::1"))); + BOOST_CHECK_EQUAL(NMGOutputToSorted(ng.toString()), NMGOutputToSorted("10.0.1.0/32, 127.0.0.0/8, 10.0.0.0/24, ::1/128, fe80::/16")); + + /* negative entries using the explicit flag */ + ng.addMask(Netmask("172.16.0.0/16"), true); + BOOST_CHECK_EQUAL(ng.size(), 6U); + BOOST_CHECK(ng.match(ComboAddress("172.16.1.1"))); + BOOST_CHECK(ng.match(ComboAddress("172.16.4.50"))); + ng.addMask(Netmask("172.16.4.0/24"), false); + BOOST_CHECK_EQUAL(ng.size(), 7U); + BOOST_CHECK(ng.match(ComboAddress("172.16.1.1"))); + BOOST_CHECK(!ng.match(ComboAddress("172.16.4.50"))); + ng.addMask("fe80::/24", false); + BOOST_CHECK_EQUAL(ng.size(), 8U); + BOOST_CHECK(!ng.match(ComboAddress("fe80::1"))); + BOOST_CHECK(!ng.match(ComboAddress("fe81::1"))); + /* not in fe80::/24 but in fe80::/16, should match */ + BOOST_CHECK(ng.match(ComboAddress("fe80:0100::1"))); + + BOOST_CHECK_EQUAL(NMGOutputToSorted(ng.toString()), NMGOutputToSorted("10.0.1.0/32, 127.0.0.0/8, 10.0.0.0/24, ::1/128, fe80::/16, 172.16.0.0/16, !172.16.4.0/24, !fe80::/24")); + } +} + +BOOST_AUTO_TEST_CASE(test_NetmaskTree) { + NetmaskTree<int> nmt; + BOOST_CHECK_EQUAL(nmt.empty(), true); + BOOST_CHECK_EQUAL(nmt.size(), 0U); + nmt.insert(Netmask("130.161.252.0/24")).second=0; + BOOST_CHECK_EQUAL(nmt.empty(), false); + BOOST_CHECK_EQUAL(nmt.size(), 1U); + nmt.insert(Netmask("130.161.0.0/16")).second=1; + BOOST_CHECK_EQUAL(nmt.size(), 2U); + nmt.insert(Netmask("130.0.0.0/8")).second=2; + BOOST_CHECK_EQUAL(nmt.size(), 3U); + + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("213.244.168.210")), (void*)0); + auto found=nmt.lookup(ComboAddress("130.161.252.29")); + BOOST_CHECK(found); + BOOST_CHECK_EQUAL(found->second, 0); + found=nmt.lookup(ComboAddress("130.161.180.1")); + BOOST_CHECK(found); + BOOST_CHECK_EQUAL(found->second, 1); + + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("130.255.255.255"))->second, 2); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("130.161.252.255"))->second, 0); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("130.161.253.255"))->second, 1); + + found=nmt.lookup(ComboAddress("130.145.180.1")); + BOOST_CHECK(found); + BOOST_CHECK_EQUAL(found->second, 2); + + nmt.insert(Netmask("0.0.0.0/0")).second=3; + BOOST_CHECK_EQUAL(nmt.size(), 4U); + nmt.insert(Netmask("0.0.0.0/7")).second=4; + BOOST_CHECK_EQUAL(nmt.size(), 5U); + nmt.insert(Netmask("0.0.0.0/15")).second=5; + BOOST_CHECK_EQUAL(nmt.size(), 6U); + BOOST_CHECK_EQUAL(nmt.lookup(Netmask("0.0.0.0/0"))->second, 3); + BOOST_CHECK_EQUAL(nmt.lookup(Netmask("0.0.0.0/7"))->second, 4); + BOOST_CHECK_EQUAL(nmt.lookup(Netmask("0.0.0.0/15"))->second, 5); + + nmt.clear(); + BOOST_CHECK_EQUAL(nmt.empty(), true); + BOOST_CHECK_EQUAL(nmt.size(), 0U); + BOOST_CHECK(!nmt.lookup(ComboAddress("130.161.180.1"))); + + nmt.insert(Netmask("::1")).second=1; + BOOST_CHECK_EQUAL(nmt.empty(), false); + BOOST_CHECK_EQUAL(nmt.size(), 1U); + nmt.insert(Netmask("::/0")).second=0; + BOOST_CHECK_EQUAL(nmt.size(), 2U); + nmt.insert(Netmask("fe80::/16")).second=2; + BOOST_CHECK_EQUAL(nmt.size(), 3U); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("130.161.253.255")), (void*)0); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("::2"))->second, 0); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("::ffff"))->second, 0); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("::1"))->second, 1); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("fe80::1"))->second, 2); +} + +BOOST_AUTO_TEST_CASE(test_single) { + NetmaskTree<bool> nmt; + BOOST_CHECK_EQUAL(nmt.empty(), true); + BOOST_CHECK_EQUAL(nmt.size(), 0U); + nmt.insert(Netmask("127.0.0.0/8")).second=1; + BOOST_CHECK_EQUAL(nmt.empty(), false); + BOOST_CHECK_EQUAL(nmt.size(), 1U); + BOOST_CHECK_EQUAL(nmt.lookup(ComboAddress("127.0.0.1"))->second, 1); +} + +BOOST_AUTO_TEST_CASE(test_scale) { + string start="192.168."; + NetmaskTree<int> works; + BOOST_CHECK_EQUAL(works.size(), 0U); + for(size_t i=0; i < 256; ++i) { + for(size_t j=0; j < 256; ++j) { + works.insert(Netmask(start+std::to_string(i)+"."+std::to_string(j))).second=i*j; + BOOST_CHECK_EQUAL(works.size(), i*256 + j + 1); + } + } + + for(int i=0; i < 256; ++i) { + for(int j=0; j < 256; ++j) { + BOOST_CHECK_EQUAL(works.lookup(ComboAddress(start+std::to_string(i)+"."+std::to_string(j)))->second, i*j); + } + } + + start="130.161."; + for(int i=0; i < 256; ++i) { + for(int j=0; j < 256; ++j) { + BOOST_CHECK_EQUAL(works.lookup(ComboAddress(start+std::to_string(i)+"."+std::to_string(j))), (void*)0); + } + } + + start="2000:123:"; + for(size_t i=0; i < 256; ++i) { + for(size_t j=0; j < 256; ++j) { + works.insert(Netmask(start+std::to_string(i)+":"+std::to_string(j)+"::/64")).second=i*j; + BOOST_CHECK_EQUAL(works.size(), (256*256) + i*256 + j + 1); + } + } + + for(int i=0; i < 256; ++i) { + for(int j=0; j < 256; ++j) { + BOOST_CHECK_EQUAL(works.lookup(ComboAddress(start+std::to_string(i)+":"+std::to_string(j)+"::"+std::to_string(i)+":"+std::to_string(j)))->second, i*j); + } + } + + start="2001:123:"; + for(int i=0; i < 256; ++i) { + for(int j=0; j < 256; ++j) { + BOOST_CHECK_EQUAL(works.lookup(ComboAddress(start+std::to_string(i)+":"+std::to_string(j)+"::"+std::to_string(i)+":"+std::to_string(j))), (void*)0); + } + } +} + +BOOST_AUTO_TEST_CASE(test_removal) { + std::string prefix = "192."; + NetmaskTree<int> nmt; + BOOST_CHECK(nmt.empty()); + BOOST_CHECK_EQUAL(nmt.size(), 0U); + + size_t count = 0; + for(unsigned int i = 0; i < 256; ++i) { + for(unsigned int j = 16; j <= 32; ++j) { + nmt.insert(Netmask(prefix + std::to_string(i) +".127.255/"+std::to_string(j))).second = j; + count++; + BOOST_CHECK_EQUAL(nmt.size(), count); + } + } + + for(unsigned int i = 0; i < 256; ++i) { + ComboAddress key(prefix + std::to_string(i) + ".127.255"); + const auto result = nmt.lookup(key); + BOOST_CHECK_EQUAL(result->first.getBits(), 32); + BOOST_CHECK_EQUAL(result->first.getMaskedNetwork().toString(), key.toString()); + BOOST_CHECK_EQUAL(result->second, 32); + } + + for(int i = 0; i < 256; ++i) { + for(int j = 32; j >= 16; --j) { + ComboAddress key(prefix + std::to_string(i) + ".127.255"); + nmt.erase(Netmask(key, j)); + count--; + BOOST_CHECK_EQUAL(nmt.size(), count); + const auto result = nmt.lookup(key); + + if (j > 16) { + BOOST_REQUIRE(result != nullptr); + BOOST_CHECK_EQUAL(result->first.getBits(), j-1); + BOOST_CHECK_EQUAL(result->first.getMaskedNetwork().toString(), Netmask(key, j-1).getMaskedNetwork().toString()); + BOOST_CHECK_EQUAL(result->second, j - 1); + } + else { + BOOST_CHECK(result == nullptr); + } + } + } + + BOOST_CHECK_EQUAL(nmt.size(), 0U); + BOOST_CHECK(nmt.empty()); +} + +BOOST_AUTO_TEST_CASE(test_iterator) { + NetmaskTree<int> masks_set1; + std::set<Netmask> masks_set2; + + // create sets. the std::set entries are normalized to match internal behavior + // of NetmaskTree + for(int i=0; i < 256; ++i) { + std::stringstream ss; + Netmask mask; + + ss << i << "." << i << "." << i << "." << i; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << (255-i) << "." << (i/2) << "." << (i/3) << "." << (i/5); + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << (i/5) << "." << (i/3) << "." << (i/2) << "." << (255-i); + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << (i/2) << "." << (i/4) << "." << (255-i) << ".0/" << (i%24); + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << std::hex << "2001:" << i << i << ":" << i << i << "::/64"; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << std::hex << "2001:" << (i/5) << (i/3) << ":" << (i/2) << (255-i) << "::/64"; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << std::hex << "2001:" << (255-i) << (i/2) << ":" << (i/3) << (i/5) << "::/64"; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << std::hex << "20" << i/2 << ":" << i/3 << i/7 << "::" << i << (i > 0 ? i-1 : i + 1); + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << std::hex << "20" << i << ":" << i << i << "::/" << std::dec << (i%48); + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + } + for(int i=0; i <= 32; ++i) { + std::stringstream ss; + Netmask mask; + + ss << "85.85.85.85/" << i; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << "170.170.170.170/" << i; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + } + for(int i=0; i <= 128; ++i) { + std::stringstream ss; + Netmask mask; + + ss << "5555:5555:5555:5555:5555:5555:5555:5555/" << i; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + + ss.str(""); + ss << "aaaa:aaaa:aaaa:aaaa:aaaa:aaaa:aaaa:aaaa/" << i; + mask = Netmask(ss.str()); + masks_set1.insert(mask).second=i; + masks_set2.insert(mask.getNormalized()); + } + + + // check set equality using iterators + BOOST_CHECK_EQUAL(masks_set1.size(), masks_set2.size()); + BOOST_CHECK_EQUAL((size_t)std::distance(masks_set1.begin(), masks_set1.end()), + (size_t)std::distance(masks_set2.begin(), masks_set2.end())); + for (auto entry: masks_set1) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + BOOST_CHECK(masks_set1.lookup(mask) != nullptr); + } + + // create a copy of the NetmaskTree (check copy by assignment) + NetmaskTree<int> masks_set1_cp1 = masks_set1; + + // taint the old version + masks_set1.insert("1.2.3.4"); + masks_set1.erase("1.1.1.1"); + + // check set equality using iterators + BOOST_CHECK_EQUAL(masks_set1_cp1.size(), masks_set2.size()); + BOOST_CHECK_EQUAL((size_t)std::distance(masks_set1_cp1.begin(), masks_set1_cp1.end()), + (size_t)std::distance(masks_set2.begin(), masks_set2.end())); + for (auto entry: masks_set1_cp1) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + BOOST_CHECK(masks_set1_cp1.lookup(mask) != nullptr); + } + + // create a copy of the NetmaskTree (check copy constructor) + NetmaskTree<int> masks_set1_cp2(masks_set1_cp1); + + // taint the old version + masks_set1_cp1.insert("2.3.4.5"); + masks_set1_cp1.erase("2.2.2.2"); + + // check set equality using iterators + BOOST_CHECK_EQUAL(masks_set1_cp2.size(), masks_set2.size()); + BOOST_CHECK_EQUAL((size_t)std::distance(masks_set1_cp2.begin(), masks_set1_cp2.end()), + (size_t)std::distance(masks_set2.begin(), masks_set2.end())); + for (auto entry: masks_set1_cp2) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + BOOST_CHECK(masks_set1_cp2.lookup(mask) != nullptr); + } + + // swap contents of the NetmaskTree + NetmaskTree<int> masks_set1_cp3; + masks_set1_cp3.swap(masks_set1_cp2); + + // taint the old version + masks_set1_cp2.insert("3.4.5.6"); + masks_set1_cp2.erase("3.3.3.3"); + + // check set equality using iterators + BOOST_CHECK_EQUAL(masks_set1_cp3.size(), masks_set2.size()); + BOOST_CHECK_EQUAL((size_t)std::distance(masks_set1_cp3.begin(), masks_set1_cp3.end()), + (size_t)std::distance(masks_set2.begin(), masks_set2.end())); + for (auto entry: masks_set1_cp3) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + BOOST_CHECK(masks_set1_cp3.lookup(mask) != nullptr); + } + + // copy contents to an std::set + std::set<NetmaskTree<int>::node_type> masks_set1_cp4(masks_set1_cp3.begin(), masks_set1_cp3.end()); + + // check set equality + BOOST_CHECK_EQUAL(masks_set1_cp4.size(), masks_set2.size()); + for (auto entry: masks_set1_cp4) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + Netmask maskl = mask.getNormalized(); + bool found = false; + for (auto entry: masks_set1_cp4) { + Netmask maskr = entry.first.getNormalized(); + + if (maskl == maskr) + found = true; + } + BOOST_CHECK(found); + } + + // create a copy of the NetmaskTree + NetmaskTree<int> masks_set1_cp5(masks_set1_cp3); + + // erase select values + { + Netmask mask; + + mask = Netmask("16.16.16.16"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("223.16.10.6"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("12.21.32.191"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("64.32.127.0/8"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("2001:ffff:ffff::/64"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("2001:192a:407f::/64"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("2001:bf20:15c::/64"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("2010:a4::201f"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("2010:1010::/16"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("85.85.85.85"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("170.170.170.170"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("5555:5555:5555:5555:5555:5555:5555:5555"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + + mask = Netmask("aaaa:aaaa:aaaa:aaaa:aaaa:aaaa:aaaa:aaaa"); + masks_set1_cp5.erase(mask); + masks_set2.erase(mask.getNormalized()); + } + + // check set equality using iterators + BOOST_CHECK_EQUAL(masks_set1_cp5.size(), masks_set2.size()); + BOOST_CHECK_EQUAL((size_t)std::distance(masks_set1_cp5.begin(), masks_set1_cp5.end()), + (size_t)std::distance(masks_set2.begin(), masks_set2.end())); + for (auto entry: masks_set1_cp5) { + Netmask mask = entry.first.getNormalized(); + + BOOST_CHECK(masks_set2.find(mask) != masks_set2.end()); + } + for (const Netmask& mask: masks_set2) { + BOOST_CHECK(masks_set1_cp5.lookup(mask) != nullptr); + } +} + +BOOST_AUTO_TEST_CASE(test_ComboAddress_caContainerToString) { + ComboAddress ca1("192.0.2.1:53"); + ComboAddress ca2("192.0.2.2:5300"); + ComboAddress ca3("[2001:db8:53::3]:53"); + ComboAddress ca4("[2001:db8:53::4]:5300"); + + set<ComboAddress> caSet({ca1, ca2, ca3, ca4}); + vector<ComboAddress> caVector({ca1, ca2, ca3, ca4}); + + string caSetStr = ComboAddress::caContainerToString(caSet, false); + string caVectorStr = ComboAddress::caContainerToString(caVector, false); + BOOST_CHECK_EQUAL(caSetStr, "192.0.2.1,192.0.2.2,2001:db8:53::3,2001:db8:53::4"); + BOOST_CHECK_EQUAL(caVectorStr, "192.0.2.1,192.0.2.2,2001:db8:53::3,2001:db8:53::4"); + + caSetStr = ComboAddress::caContainerToString(caSet, true); + caVectorStr = ComboAddress::caContainerToString(caVector, true); + BOOST_CHECK_EQUAL(caSetStr, "192.0.2.1,192.0.2.2:5300,2001:db8:53::3,[2001:db8:53::4]:5300"); + BOOST_CHECK_EQUAL(caVectorStr, "192.0.2.1,192.0.2.2:5300,2001:db8:53::3,[2001:db8:53::4]:5300"); + + caSetStr = ComboAddress::caContainerToString(caSet, true, 0); + caVectorStr = ComboAddress::caContainerToString(caVector, true, 0); + BOOST_CHECK_EQUAL(caSetStr, "192.0.2.1:53,192.0.2.2:5300,[2001:db8:53::3]:53,[2001:db8:53::4]:5300"); + BOOST_CHECK_EQUAL(caVectorStr, "192.0.2.1:53,192.0.2.2:5300,[2001:db8:53::3]:53,[2001:db8:53::4]:5300"); +} + +BOOST_AUTO_TEST_CASE(test_parseIPAndPort) +{ + struct { + std::string str; + uint16_t port; + std::string result; + bool ex; + } tests[] = { + { "", 0, "", true }, + { "1.2.3.a", 53, "", true }, + { "1::g3", 99, "", true }, + { "1.2.3.4", 0, "1.2.3.4:0", false }, + { "1.2.3.4", 999, "1.2.3.4:999", false }, + { "1::", 999, "[1::]:999", false }, + { "1::33:99", 0, "[1::33:99]", false }, + { "[1::33]:99", 0, "[1::33]:99", false }, + { "1:33::99", 0, "1:33::99", false }, + { "[1:33::]:99", 0, "[1:33::]:99", false }, + { "2003:1234::f561", 53, "[2003:1234::f561]:53", false }, + { "2003:1234::f561:53", 54, "[2003:1234::f561:53]:54", false }, + }; + + for (const auto& t : tests) { + if (t.ex) { + BOOST_CHECK_THROW(parseIPAndPort(t.str, t.port), PDNSException); + } else { + ComboAddress a = parseIPAndPort(t.str, t.port); + BOOST_CHECK_EQUAL(a.toStringWithPort(), ComboAddress(t.result).toStringWithPort()); + } + } +} + +BOOST_AUTO_TEST_SUITE_END() |