diff options
Diffstat (limited to 'src/common/ip_test.cpp')
-rw-r--r-- | src/common/ip_test.cpp | 332 |
1 files changed, 332 insertions, 0 deletions
diff --git a/src/common/ip_test.cpp b/src/common/ip_test.cpp new file mode 100644 index 0000000..64a537f --- /dev/null +++ b/src/common/ip_test.cpp @@ -0,0 +1,332 @@ +#include "ip.hpp" + +#include <gtest/gtest.h> + +#include "cxxstdio.hpp" + +#define CB(X) (std::integral_constant<bool, (X)>::value) +TEST(ip4addr, cmp) +{ + constexpr static + IP4Address a = IP4_LOCALHOST; + constexpr static + IP4Address b = IP4_BROADCAST; + + EXPECT_FALSE(CB(a < a)); + EXPECT_TRUE (CB(a < b)); + EXPECT_FALSE(CB(b < a)); + EXPECT_FALSE(CB(b < b)); + + EXPECT_FALSE(CB(a > a)); + EXPECT_FALSE(CB(a > b)); + EXPECT_TRUE (CB(b > a)); + EXPECT_FALSE(CB(b > b)); + + EXPECT_TRUE (CB(a <= a)); + EXPECT_TRUE (CB(a <= b)); + EXPECT_FALSE(CB(b <= a)); + EXPECT_TRUE (CB(b <= b)); + + EXPECT_TRUE (CB(a >= a)); + EXPECT_FALSE(CB(a >= b)); + EXPECT_TRUE (CB(b >= a)); + EXPECT_TRUE (CB(b >= b)); + + EXPECT_TRUE (CB(a == a)); + EXPECT_FALSE(CB(a == b)); + EXPECT_FALSE(CB(b == a)); + EXPECT_TRUE (CB(b == b)); + + EXPECT_FALSE(CB(a != a)); + EXPECT_TRUE (CB(a != b)); + EXPECT_TRUE (CB(b != a)); + EXPECT_FALSE(CB(b != b)); +} + +TEST(ip4addr, str) +{ + IP4Address a; + EXPECT_EQ("0.0.0.0", STRNPRINTF(17, "%s", a)); + EXPECT_EQ("127.0.0.1", STRNPRINTF(17, "%s", IP4_LOCALHOST)); + EXPECT_EQ("255.255.255.255", STRNPRINTF(17, "%s", IP4_BROADCAST)); +} + +TEST(ip4addr, extract) +{ + IP4Address a; + EXPECT_TRUE(extract("0.0.0.0", &a)); + EXPECT_EQ("0.0.0.0", STRNPRINTF(16, "%s", a)); + EXPECT_TRUE(extract("127.0.0.1", &a)); + EXPECT_EQ("127.0.0.1", STRNPRINTF(16, "%s", a)); + EXPECT_TRUE(extract("255.255.255.255", &a)); + EXPECT_EQ("255.255.255.255", STRNPRINTF(16, "%s", a)); + EXPECT_TRUE(extract("1.2.3.4", &a)); + EXPECT_EQ("1.2.3.4", STRNPRINTF(16, "%s", a)); + + EXPECT_FALSE(extract("1.2.3.4.5", &a)); + EXPECT_FALSE(extract("1.2.3.4.", &a)); + EXPECT_FALSE(extract("1.2.3.", &a)); + EXPECT_FALSE(extract("1.2.3", &a)); + EXPECT_FALSE(extract("1.2.", &a)); + EXPECT_FALSE(extract("1.2", &a)); + EXPECT_FALSE(extract("1.", &a)); + EXPECT_FALSE(extract("1", &a)); + EXPECT_FALSE(extract("", &a)); +} + + +TEST(ip4mask, body) +{ + IP4Mask m; + EXPECT_EQ(IP4Address(), m.addr()); + EXPECT_EQ(IP4Address(), m.mask()); + m = IP4Mask(IP4_LOCALHOST, IP4_BROADCAST); + EXPECT_EQ(IP4_LOCALHOST, m.addr()); + EXPECT_EQ(IP4_BROADCAST, m.mask()); +} + +TEST(ip4mask, str) +{ + IP4Mask m; + EXPECT_EQ("0.0.0.0/0.0.0.0", STRNPRINTF(33, "%s", m)); + m = IP4Mask(IP4_LOCALHOST, IP4_BROADCAST); + EXPECT_EQ("127.0.0.1/255.255.255.255", STRNPRINTF(33, "%s", m)); +} + +TEST(ip4mask, extract) +{ + IP4Mask m; + EXPECT_FALSE(extract("9.8.7.6/33", &m)); + EXPECT_FALSE(extract("9.8.7.6.5", &m)); + EXPECT_FALSE(extract("9.8.7.6/", &m)); + EXPECT_FALSE(extract("9.8.7", &m)); + EXPECT_FALSE(extract("9.8", &m)); + EXPECT_FALSE(extract("9", &m)); + + EXPECT_TRUE(extract("127.0.0.1", &m)); + EXPECT_EQ("127.0.0.1/255.255.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("127.0.0.1.", &m)); + EXPECT_EQ("127.0.0.1/255.255.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("127.0.0.", &m)); + EXPECT_EQ("127.0.0.0/255.255.255.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("127.0.", &m)); + EXPECT_EQ("127.0.0.0/255.255.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("127.", &m)); + EXPECT_EQ("127.0.0.0/255.0.0.0", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("1.2.3.4/255.255.255.255", &m)); + EXPECT_EQ("1.2.3.4/255.255.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.2.3.0/255.255.255.0", &m)); + EXPECT_EQ("1.2.3.0/255.255.255.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.2.0.4/255.255.0.255", &m)); + EXPECT_EQ("1.2.0.4/255.255.0.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.2.0.0/255.255.0.0", &m)); + EXPECT_EQ("1.2.0.0/255.255.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.0.3.4/255.0.255.255", &m)); + EXPECT_EQ("1.0.3.4/255.0.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.0.3.0/255.0.255.0", &m)); + EXPECT_EQ("1.0.3.0/255.0.255.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.0.0.4/255.0.0.255", &m)); + EXPECT_EQ("1.0.0.4/255.0.0.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("1.0.0.0/255.0.0.0", &m)); + EXPECT_EQ("1.0.0.0/255.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.2.3.4/0.255.255.255", &m)); + EXPECT_EQ("0.2.3.4/0.255.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.2.3.0/0.255.255.0", &m)); + EXPECT_EQ("0.2.3.0/0.255.255.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.2.0.4/0.255.0.255", &m)); + EXPECT_EQ("0.2.0.4/0.255.0.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.2.0.0/0.255.0.0", &m)); + EXPECT_EQ("0.2.0.0/0.255.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.3.4/0.0.255.255", &m)); + EXPECT_EQ("0.0.3.4/0.0.255.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.3.0/0.0.255.0", &m)); + EXPECT_EQ("0.0.3.0/0.0.255.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.4/0.0.0.255", &m)); + EXPECT_EQ("0.0.0.4/0.0.0.255", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/0.0.0.0", &m)); + EXPECT_EQ("0.0.0.0/0.0.0.0", STRNPRINTF(32, "%s", m)); + + // please don't do this + EXPECT_TRUE(extract("120.248.200.217/89.57.126.5", &m)); + EXPECT_EQ("88.56.72.1/89.57.126.5", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("0.0.0.0/32", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.255", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("0.0.0.0/31", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.254", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/30", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.252", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/29", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.248", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/28", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.240", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/27", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.224", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/26", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.192", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/25", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.128", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/24", &m)); + EXPECT_EQ("0.0.0.0/255.255.255.0", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("0.0.0.0/23", &m)); + EXPECT_EQ("0.0.0.0/255.255.254.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/22", &m)); + EXPECT_EQ("0.0.0.0/255.255.252.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/21", &m)); + EXPECT_EQ("0.0.0.0/255.255.248.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/20", &m)); + EXPECT_EQ("0.0.0.0/255.255.240.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/19", &m)); + EXPECT_EQ("0.0.0.0/255.255.224.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/18", &m)); + EXPECT_EQ("0.0.0.0/255.255.192.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/17", &m)); + EXPECT_EQ("0.0.0.0/255.255.128.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/16", &m)); + EXPECT_EQ("0.0.0.0/255.255.0.0", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("0.0.0.0/15", &m)); + EXPECT_EQ("0.0.0.0/255.254.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/14", &m)); + EXPECT_EQ("0.0.0.0/255.252.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/13", &m)); + EXPECT_EQ("0.0.0.0/255.248.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/12", &m)); + EXPECT_EQ("0.0.0.0/255.240.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/11", &m)); + EXPECT_EQ("0.0.0.0/255.224.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/10", &m)); + EXPECT_EQ("0.0.0.0/255.192.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/9", &m)); + EXPECT_EQ("0.0.0.0/255.128.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/8", &m)); + EXPECT_EQ("0.0.0.0/255.0.0.0", STRNPRINTF(32, "%s", m)); + + EXPECT_TRUE(extract("0.0.0.0/7", &m)); + EXPECT_EQ("0.0.0.0/254.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/6", &m)); + EXPECT_EQ("0.0.0.0/252.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/5", &m)); + EXPECT_EQ("0.0.0.0/248.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/4", &m)); + EXPECT_EQ("0.0.0.0/240.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/3", &m)); + EXPECT_EQ("0.0.0.0/224.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/2", &m)); + EXPECT_EQ("0.0.0.0/192.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/1", &m)); + EXPECT_EQ("0.0.0.0/128.0.0.0", STRNPRINTF(32, "%s", m)); + EXPECT_TRUE(extract("0.0.0.0/0", &m)); + EXPECT_EQ("0.0.0.0/0.0.0.0", STRNPRINTF(32, "%s", m)); +} + +TEST(ip4mask, cover) +{ + IP4Address a; + IP4Address b = IP4_BROADCAST; + IP4Address l = IP4_LOCALHOST; + IP4Address h({127, 255, 255, 255}); + IP4Address p24l({10, 0, 0, 0}); + IP4Address p24h({10, 255, 255, 255}); + IP4Address p20l({172, 16, 0, 0}); + IP4Address p20h({172, 31, 255, 255}); + IP4Address p16l({192, 168, 0, 0}); + IP4Address p16h({192, 168, 255, 255}); + IP4Mask m; + EXPECT_TRUE(m.covers(a)); + EXPECT_TRUE(m.covers(b)); + EXPECT_TRUE(m.covers(l)); + EXPECT_TRUE(m.covers(h)); + EXPECT_TRUE(m.covers(p24l)); + EXPECT_TRUE(m.covers(p24h)); + EXPECT_TRUE(m.covers(p20l)); + EXPECT_TRUE(m.covers(p20h)); + EXPECT_TRUE(m.covers(p16l)); + EXPECT_TRUE(m.covers(p16h)); + m = IP4Mask(l, a); + EXPECT_TRUE(m.covers(a)); + EXPECT_TRUE(m.covers(b)); + EXPECT_TRUE(m.covers(l)); + EXPECT_TRUE(m.covers(h)); + EXPECT_TRUE(m.covers(p24l)); + EXPECT_TRUE(m.covers(p24h)); + EXPECT_TRUE(m.covers(p20l)); + EXPECT_TRUE(m.covers(p20h)); + EXPECT_TRUE(m.covers(p16l)); + EXPECT_TRUE(m.covers(p16h)); + m = IP4Mask(l, b); + EXPECT_FALSE(m.covers(a)); + EXPECT_FALSE(m.covers(b)); + EXPECT_TRUE(m.covers(l)); + EXPECT_FALSE(m.covers(h)); + EXPECT_FALSE(m.covers(p24l)); + EXPECT_FALSE(m.covers(p24h)); + EXPECT_FALSE(m.covers(p20l)); + EXPECT_FALSE(m.covers(p20h)); + EXPECT_FALSE(m.covers(p16l)); + EXPECT_FALSE(m.covers(p16h)); + + // but the really useful ones are with partial masks + m = IP4Mask(IP4Address({10, 0, 0, 0}), IP4Address({255, 0, 0, 0})); + EXPECT_FALSE(m.covers(a)); + EXPECT_FALSE(m.covers(b)); + EXPECT_FALSE(m.covers(l)); + EXPECT_FALSE(m.covers(h)); + EXPECT_TRUE(m.covers(p24l)); + EXPECT_TRUE(m.covers(p24h)); + EXPECT_FALSE(m.covers(p20l)); + EXPECT_FALSE(m.covers(p20h)); + EXPECT_FALSE(m.covers(p16l)); + EXPECT_FALSE(m.covers(p16h)); + EXPECT_FALSE(m.covers(IP4Address({9, 255, 255, 255}))); + EXPECT_FALSE(m.covers(IP4Address({11, 0, 0, 0}))); + m = IP4Mask(IP4Address({127, 0, 0, 0}), IP4Address({255, 0, 0, 0})); + EXPECT_FALSE(m.covers(a)); + EXPECT_FALSE(m.covers(b)); + EXPECT_TRUE(m.covers(l)); + EXPECT_TRUE(m.covers(h)); + EXPECT_FALSE(m.covers(p24l)); + EXPECT_FALSE(m.covers(p24h)); + EXPECT_FALSE(m.covers(p20l)); + EXPECT_FALSE(m.covers(p20h)); + EXPECT_FALSE(m.covers(p16l)); + EXPECT_FALSE(m.covers(p16h)); + EXPECT_FALSE(m.covers(IP4Address({126, 255, 255, 255}))); + EXPECT_FALSE(m.covers(IP4Address({128, 0, 0, 0}))); + m = IP4Mask(IP4Address({172, 16, 0, 0}), IP4Address({255, 240, 0, 0})); + EXPECT_FALSE(m.covers(a)); + EXPECT_FALSE(m.covers(b)); + EXPECT_FALSE(m.covers(l)); + EXPECT_FALSE(m.covers(h)); + EXPECT_FALSE(m.covers(p24l)); + EXPECT_FALSE(m.covers(p24h)); + EXPECT_TRUE(m.covers(p20l)); + EXPECT_TRUE(m.covers(p20h)); + EXPECT_FALSE(m.covers(p16l)); + EXPECT_FALSE(m.covers(p16h)); + EXPECT_FALSE(m.covers(IP4Address({172, 15, 255, 255}))); + EXPECT_FALSE(m.covers(IP4Address({172, 32, 0, 0}))); + m = IP4Mask(IP4Address({192, 168, 0, 0}), IP4Address({255, 255, 0, 0})); + EXPECT_FALSE(m.covers(a)); + EXPECT_FALSE(m.covers(b)); + EXPECT_FALSE(m.covers(l)); + EXPECT_FALSE(m.covers(h)); + EXPECT_FALSE(m.covers(p24l)); + EXPECT_FALSE(m.covers(p24h)); + EXPECT_FALSE(m.covers(p20l)); + EXPECT_FALSE(m.covers(p20h)); + EXPECT_TRUE(m.covers(p16l)); + EXPECT_TRUE(m.covers(p16h)); + EXPECT_FALSE(m.covers(IP4Address({192, 167, 255, 255}))); + EXPECT_FALSE(m.covers(IP4Address({192, 169, 0, 0}))); + + // OTOH this is crazy + EXPECT_TRUE(extract("120.248.200.217/89.57.126.5", &m)); + EXPECT_TRUE(m.covers(IP4Address({120, 248, 200, 217}))); + EXPECT_TRUE(m.covers(IP4Address({88, 56, 72, 1}))); + EXPECT_FALSE(m.covers(IP4Address({88, 56, 72, 0}))); + EXPECT_FALSE(m.covers(IP4Address({88, 56, 72, 255}))); +} |