1#include "config.h" 2 3#include "ntp_stdlib.h" 4#include "ntp_calendar.h" 5 6#include "unity.h" 7#include "sockaddrtest.h" 8 9 10void setUp(void); 11void test_IPv4AddressWithPort(void); 12void test_IPv6AddressWithPort(void); 13void test_IgnoreIPv6Fields(void); 14void test_ScopedIPv6AddressWithPort(void); 15void test_HashEqual(void); 16void test_HashNotEqual(void); 17 18 19void 20setUp(void) 21{ 22 init_lib(); 23} 24 25 26void 27test_IPv4AddressWithPort(void) 28{ 29 sockaddr_u input = CreateSockaddr4("192.0.2.10", 123); 30 31 TEST_ASSERT_EQUAL_STRING("192.0.2.10", socktoa(&input)); 32 TEST_ASSERT_EQUAL_STRING("192.0.2.10:123", sockporttoa(&input)); 33} 34 35 36void 37test_IPv6AddressWithPort(void) 38{ 39#ifdef ISC_PLATFORM_WANTIPV6 40 41 const struct in6_addr address = { 42 0x20, 0x01, 0x0d, 0xb8, 43 0x85, 0xa3, 0x08, 0xd3, 44 0x13, 0x19, 0x8a, 0x2e, 45 0x03, 0x70, 0x73, 0x34 46 }; 47 48 const char* expected = 49 "2001:db8:85a3:8d3:1319:8a2e:370:7334"; 50 const char* expected_port = 51 "[2001:db8:85a3:8d3:1319:8a2e:370:7334]:123"; 52 53 sockaddr_u input; 54 memset(&input, 0, sizeof(input)); 55 AF(&input) = AF_INET6; 56 SET_ADDR6N(&input, address); 57 SET_PORT(&input, 123); 58 59 TEST_ASSERT_EQUAL_STRING(expected, socktoa(&input)); 60 TEST_ASSERT_EQUAL_STRING(expected_port, sockporttoa(&input)); 61 62#else 63 64 TEST_IGNORE_MESSAGE("IPV6 disabled in build, skipping."); 65 66#endif /* ISC_PLATFORM_HAVEIPV6 */ 67} 68 69 70void 71test_ScopedIPv6AddressWithPort(void) 72{ 73#ifdef ISC_PLATFORM_HAVESCOPEID 74 75 const struct in6_addr address = { { { 76 0xfe, 0x80, 0x00, 0x00, 77 0x00, 0x00, 0x00, 0x00, 78 0x02, 0x12, 0x3f, 0xff, 79 0xfe, 0x29, 0xff, 0xfa 80 } } }; 81 82 const char* expected = 83 "fe80::212:3fff:fe29:fffa%5"; 84 const char* expected_port = 85 "[fe80::212:3fff:fe29:fffa%5]:123"; 86 87 sockaddr_u input; 88 memset(&input, 0, sizeof(input)); 89 AF(&input) = AF_INET6; 90 SET_ADDR6N(&input, address); 91 SET_PORT(&input, 123); 92 SCOPE_VAR(&input) = 5; 93 94 TEST_ASSERT_EQUAL_STRING(expected, socktoa(&input)); 95 TEST_ASSERT_EQUAL_STRING(expected_port, sockporttoa(&input)); 96#else 97 98 TEST_IGNORE_MESSAGE("Skipping because ISC_PLATFORM does not have Scope ID"); 99 100#endif 101} 102 103 104void 105test_HashEqual(void) 106{ 107 sockaddr_u input1 = CreateSockaddr4("192.00.2.2", 123); 108 sockaddr_u input2 = CreateSockaddr4("192.0.2.2", 123); 109 110 TEST_ASSERT_TRUE(IsEqual(input1, input2)); 111 TEST_ASSERT_EQUAL(sock_hash(&input1), sock_hash(&input2)); 112} 113 114 115void 116test_HashNotEqual(void) 117{ 118 /* These two addresses should not generate the same hash. */ 119 sockaddr_u input1 = CreateSockaddr4("192.0.2.1", 123); 120 sockaddr_u input2 = CreateSockaddr4("192.0.2.2", 123); 121 122 TEST_ASSERT_FALSE(IsEqual(input1, input2)); 123 TEST_ASSERT_FALSE(sock_hash(&input1) == sock_hash(&input2)); 124} 125 126 127void 128test_IgnoreIPv6Fields(void) 129{ 130#ifdef ISC_PLATFORM_WANTIPV6 131 132 const struct in6_addr address = { 133 0x20, 0x01, 0x0d, 0xb8, 134 0x85, 0xa3, 0x08, 0xd3, 135 0x13, 0x19, 0x8a, 0x2e, 136 0x03, 0x70, 0x73, 0x34 137 }; 138 139 sockaddr_u input1, input2; 140 141 input1.sa6.sin6_family = AF_INET6; 142 input1.sa6.sin6_addr = address; 143 input1.sa6.sin6_flowinfo = 30L; // This value differs from input2. 144 SET_PORT(&input1, NTP_PORT); 145 146 input2.sa6.sin6_family = AF_INET6; 147 input2.sa6.sin6_addr = address; 148 input2.sa6.sin6_flowinfo = 10L; // This value differs from input1. 149 SET_PORT(&input2, NTP_PORT); 150 151 TEST_ASSERT_EQUAL(sock_hash(&input1), sock_hash(&input2)); 152 153#else 154 155 TEST_IGNORE_MESSAGE("IPV6 disabled in build, skipping."); 156 157#endif /* ISC_PLATFORM_HAVEIPV6 */ 158} 159