1 /* 2 * Copyright (C) 2014 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include <gtest/gtest.h> 18 19 #include <arpa/inet.h> 20 21 TEST(arpa_inet, inet_addr) { 22 ASSERT_EQ((htonl)(0x7f000001), inet_addr("127.0.0.1")); 23 } 24 25 TEST(arpa_inet, inet_aton) { 26 in_addr a; 27 28 // a.b.c.d 29 a.s_addr = 0; 30 ASSERT_EQ(1, inet_aton("127.1.2.3", &a)); 31 ASSERT_EQ((htonl)(0x7f010203), a.s_addr); 32 33 // a.b.c 34 a.s_addr = 0; 35 ASSERT_EQ(1, inet_aton("127.1.2", &a)); 36 ASSERT_EQ((htonl)(0x7f010002), a.s_addr); 37 38 // a.b 39 a.s_addr = 0; 40 ASSERT_EQ(1, inet_aton("127.1", &a)); 41 ASSERT_EQ((htonl)(0x7f000001), a.s_addr); 42 43 // a 44 a.s_addr = 0; 45 ASSERT_EQ(1, inet_aton("0x7f000001", &a)); 46 ASSERT_EQ((htonl)(0x7f000001), a.s_addr); 47 48 // Hex (0x) and mixed-case hex digits. 49 a.s_addr = 0; 50 ASSERT_EQ(1, inet_aton("0xFf.0.0.1", &a)); 51 ASSERT_EQ((htonl)(0xff000001), a.s_addr); 52 53 // Hex (0X) and mixed-case hex digits. 54 a.s_addr = 0; 55 ASSERT_EQ(1, inet_aton("0XfF.0.0.1", &a)); 56 ASSERT_EQ((htonl)(0xff000001), a.s_addr); 57 58 // Octal. 59 a.s_addr = 0; 60 ASSERT_EQ(1, inet_aton("0177.0.0.1", &a)); 61 ASSERT_EQ((htonl)(0x7f000001), a.s_addr); 62 63 a.s_addr = 0; 64 ASSERT_EQ(1, inet_aton("036", &a)); 65 ASSERT_EQ((htonl)(036U), a.s_addr); 66 } 67 68 TEST(arpa_inet, inet_aton_nullptr) { 69 ASSERT_EQ(0, inet_aton("", nullptr)); 70 ASSERT_EQ(1, inet_aton("127.0.0.1", nullptr)); 71 } 72 73 TEST(arpa_inet, inet_aton_invalid) { 74 ASSERT_EQ(0, inet_aton("", nullptr)); // Empty. 75 ASSERT_EQ(0, inet_aton("x", nullptr)); // Leading junk. 76 ASSERT_EQ(0, inet_aton("127.0.0.1x", nullptr)); // Trailing junk. 77 ASSERT_EQ(0, inet_aton("09.0.0.1", nullptr)); // Invalid octal. 78 ASSERT_EQ(0, inet_aton("0xg.0.0.1", nullptr)); // Invalid hex. 79 80 ASSERT_EQ(0, inet_aton("1.2.3.4.5", nullptr)); // Too many dots. 81 ASSERT_EQ(0, inet_aton("1.2.3.4.", nullptr)); // Trailing dot. 82 83 // Out of range a.b.c.d form. 84 ASSERT_EQ(0, inet_aton("999.0.0.1", nullptr)); 85 ASSERT_EQ(0, inet_aton("0.999.0.1", nullptr)); 86 ASSERT_EQ(0, inet_aton("0.0.999.1", nullptr)); 87 ASSERT_EQ(0, inet_aton("0.0.0.999", nullptr)); 88 89 // Out of range a.b.c form. 90 ASSERT_EQ(0, inet_aton("256.0.0", nullptr)); 91 ASSERT_EQ(0, inet_aton("0.256.0", nullptr)); 92 ASSERT_EQ(0, inet_aton("0.0.0x10000", nullptr)); 93 94 // Out of range a.b form. 95 ASSERT_EQ(0, inet_aton("256.0", nullptr)); 96 ASSERT_EQ(0, inet_aton("0.0x1000000", nullptr)); 97 98 // Out of range a form. 99 ASSERT_EQ(0, inet_aton("0x100000000", nullptr)); 100 101 // 64-bit overflow. 102 ASSERT_EQ(0, inet_aton("0x10000000000000000", nullptr)); 103 104 // Out of range octal. 105 ASSERT_EQ(0, inet_aton("0400.0.0.1", nullptr)); 106 } 107 108 TEST(arpa_inet, inet_lnaof) { 109 in_addr a = { htonl(0x12345678) }; 110 ASSERT_EQ(0x00345678U, inet_lnaof(a)); 111 } 112 113 TEST(arpa_inet, inet_makeaddr) { 114 in_addr a = inet_makeaddr(0x12U, 0x345678); 115 ASSERT_EQ((htonl)(0x12345678), a.s_addr); 116 } 117 118 TEST(arpa_inet, inet_netof) { 119 in_addr a = { htonl(0x12345678) }; 120 ASSERT_EQ(0x12U, inet_netof(a)); 121 } 122 123 TEST(arpa_inet, inet_network) { 124 ASSERT_EQ(0x7f000001U, inet_network("127.0.0.1")); 125 ASSERT_EQ(0x7fU, inet_network("0x7f")); 126 ASSERT_EQ(~0U, inet_network("")); 127 } 128 129 TEST(arpa_inet, inet_ntoa) { 130 in_addr a = { (htonl)(0x7f000001) }; 131 ASSERT_STREQ("127.0.0.1", inet_ntoa(a)); 132 } 133 134 TEST(arpa_inet, inet_pton__inet_ntop) { 135 sockaddr_storage ss; 136 ASSERT_EQ(1, inet_pton(AF_INET, "127.0.0.1", &ss)); 137 138 char s[INET_ADDRSTRLEN]; 139 ASSERT_STREQ("127.0.0.1", inet_ntop(AF_INET, &ss, s, INET_ADDRSTRLEN)); 140 } 141 142 TEST(arpa_inet, inet_ntop_overflow) { 143 // OpenBSD's inet_ntop had a bug where passing a 'size' larger than INET_ADDRSTRLEN 144 // for AF_INET or INET6_ADDRSTRLEN for AF_INET6 would cause inet_ntop to overflow an 145 // internal buffer. 146 147 sockaddr_storage ss4; 148 ASSERT_EQ(1, inet_pton(AF_INET, "127.0.0.1", &ss4)); 149 150 sockaddr_storage ss6; 151 ASSERT_EQ(1, inet_pton(AF_INET6, "::1", &ss6)); 152 153 char s4[INET_ADDRSTRLEN]; 154 char s6[INET6_ADDRSTRLEN]; 155 ASSERT_STREQ("127.0.0.1", inet_ntop(AF_INET, &ss4, s4, INET_ADDRSTRLEN)); 156 ASSERT_STREQ("127.0.0.1", inet_ntop(AF_INET, &ss4, s4, 2*INET_ADDRSTRLEN)); 157 ASSERT_STREQ("::1", inet_ntop(AF_INET6, &ss6, s6, INET_ADDRSTRLEN)); 158 ASSERT_STREQ("::1", inet_ntop(AF_INET6, &ss6, s6, INET6_ADDRSTRLEN)); 159 ASSERT_STREQ("::1", inet_ntop(AF_INET6, &ss6, s6, 2*INET6_ADDRSTRLEN)); 160 } 161
