|  | /* | 
|  | * Copyright (C) 2013 The Android Open Source Project | 
|  | * | 
|  | * Licensed under the Apache License, Version 2.0 (the "License"); | 
|  | * you may not use this file except in compliance with the License. | 
|  | * You may obtain a copy of the License at | 
|  | * | 
|  | *      http://www.apache.org/licenses/LICENSE-2.0 | 
|  | * | 
|  | * Unless required by applicable law or agreed to in writing, software | 
|  | * distributed under the License is distributed on an "AS IS" BASIS, | 
|  | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | * See the License for the specific language governing permissions and | 
|  | * limitations under the License. | 
|  | */ | 
|  |  | 
|  | #include <netdb.h> | 
|  |  | 
|  | #include <gtest/gtest.h> | 
|  |  | 
|  | #include <arpa/inet.h> | 
|  | #include <string.h> | 
|  | #include <sys/types.h> | 
|  | #include <sys/socket.h> | 
|  | #include <netinet/in.h> | 
|  |  | 
|  | // https://code.google.com/p/android/issues/detail?id=13228 | 
|  | TEST(netdb, freeaddrinfo_NULL) { | 
|  | freeaddrinfo(nullptr); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_NULL_host) { | 
|  | // It's okay for the host argument to be NULL, as long as service isn't. | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo(nullptr, "smtp", nullptr, &ai)); | 
|  | // (sockaddr_in::sin_port and sockaddr_in6::sin6_port overlap.) | 
|  | ASSERT_EQ(25U, ntohs(reinterpret_cast<sockaddr_in*>(ai->ai_addr)->sin_port)); | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_NULL_service) { | 
|  | // It's okay for the service argument to be NULL, as long as host isn't. | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo("localhost", nullptr, nullptr, &ai)); | 
|  | ASSERT_TRUE(ai != nullptr); | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_NULL_hints) { | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo("localhost", "9999", nullptr, &ai)); | 
|  |  | 
|  | bool saw_tcp = false; | 
|  | bool saw_udp = false; | 
|  | for (addrinfo* p = ai; p != nullptr; p = p->ai_next) { | 
|  | ASSERT_TRUE(p->ai_family == AF_INET || p->ai_family == AF_INET6); | 
|  | if (p->ai_socktype == SOCK_STREAM) { | 
|  | ASSERT_EQ(IPPROTO_TCP, p->ai_protocol); | 
|  | saw_tcp = true; | 
|  | } else if (p->ai_socktype == SOCK_DGRAM) { | 
|  | ASSERT_EQ(IPPROTO_UDP, p->ai_protocol); | 
|  | saw_udp = true; | 
|  | } | 
|  | } | 
|  | ASSERT_TRUE(saw_tcp); | 
|  | ASSERT_TRUE(saw_udp); | 
|  |  | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_service_lookup) { | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo("localhost", "smtp", nullptr, &ai)); | 
|  | ASSERT_EQ(SOCK_STREAM, ai->ai_socktype); | 
|  | ASSERT_EQ(IPPROTO_TCP, ai->ai_protocol); | 
|  | ASSERT_EQ(25, ntohs(reinterpret_cast<sockaddr_in*>(ai->ai_addr)->sin_port)); | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_hints) { | 
|  | addrinfo hints; | 
|  | memset(&hints, 0, sizeof(hints)); | 
|  | hints.ai_family = AF_INET; | 
|  | hints.ai_socktype = SOCK_STREAM; | 
|  | hints.ai_protocol = IPPROTO_TCP; | 
|  |  | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo( "localhost", "9999", &hints, &ai)); | 
|  | ASSERT_TRUE(ai != nullptr); | 
|  | // In glibc, getaddrinfo() converts ::1 to 127.0.0.1 for localhost, | 
|  | // so one or two addrinfo may be returned. | 
|  | addrinfo* tai = ai; | 
|  | while (tai != nullptr) { | 
|  | ASSERT_EQ(AF_INET, tai->ai_family); | 
|  | ASSERT_EQ(SOCK_STREAM, tai->ai_socktype); | 
|  | ASSERT_EQ(IPPROTO_TCP, tai->ai_protocol); | 
|  | tai = tai->ai_next; | 
|  | } | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getaddrinfo_ip6_localhost) { | 
|  | addrinfo* ai = nullptr; | 
|  | ASSERT_EQ(0, getaddrinfo("ip6-localhost", nullptr, nullptr, &ai)); | 
|  | ASSERT_TRUE(ai != nullptr); | 
|  | ASSERT_GE(ai->ai_addrlen, static_cast<socklen_t>(sizeof(sockaddr_in6))); | 
|  | ASSERT_TRUE(ai->ai_addr != nullptr); | 
|  | sockaddr_in6 *addr = reinterpret_cast<sockaddr_in6*>(ai->ai_addr); | 
|  | ASSERT_EQ(addr->sin6_family, AF_INET6); | 
|  | ASSERT_EQ(0, memcmp(&addr->sin6_addr, &in6addr_loopback, sizeof(in6_addr))); | 
|  | freeaddrinfo(ai); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getnameinfo_salen) { | 
|  | sockaddr_storage ss; | 
|  | memset(&ss, 0, sizeof(ss)); | 
|  | sockaddr* sa = reinterpret_cast<sockaddr*>(&ss); | 
|  | char tmp[16]; | 
|  |  | 
|  | ss.ss_family = AF_INET; | 
|  | socklen_t too_much = sizeof(ss); | 
|  | socklen_t just_right = sizeof(sockaddr_in); | 
|  | socklen_t too_little = sizeof(sockaddr_in) - 1; | 
|  |  | 
|  | ASSERT_EQ(0, getnameinfo(sa, too_much, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  | ASSERT_STREQ("0.0.0.0", tmp); | 
|  | ASSERT_EQ(0, getnameinfo(sa, just_right, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  | ASSERT_STREQ("0.0.0.0", tmp); | 
|  | ASSERT_EQ(EAI_FAMILY, getnameinfo(sa, too_little, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  |  | 
|  | ss.ss_family = AF_INET6; | 
|  | just_right = sizeof(sockaddr_in6); | 
|  | too_little = sizeof(sockaddr_in6) - 1; | 
|  | too_much = just_right + 1; | 
|  |  | 
|  | ASSERT_EQ(0, getnameinfo(sa, too_much, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  | ASSERT_STREQ("::", tmp); | 
|  | ASSERT_EQ(0, getnameinfo(sa, just_right, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  | ASSERT_STREQ("::", tmp); | 
|  | ASSERT_EQ(EAI_FAMILY, getnameinfo(sa, too_little, tmp, sizeof(tmp), nullptr, 0, NI_NUMERICHOST)); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getnameinfo_localhost) { | 
|  | sockaddr_in addr; | 
|  | char host[NI_MAXHOST]; | 
|  | memset(&addr, 0, sizeof(sockaddr_in)); | 
|  | addr.sin_family = AF_INET; | 
|  | addr.sin_addr.s_addr = htonl(0x7f000001); | 
|  | ASSERT_EQ(0, getnameinfo(reinterpret_cast<sockaddr*>(&addr), sizeof(addr), | 
|  | host, sizeof(host), nullptr, 0, 0)); | 
|  | ASSERT_STREQ(host, "localhost"); | 
|  | } | 
|  |  | 
|  | static void VerifyLocalhostName(const char* name) { | 
|  | // Test possible localhost name and aliases, which depend on /etc/hosts or /system/etc/hosts. | 
|  | ASSERT_TRUE(strcmp(name, "localhost") == 0 || | 
|  | strcmp(name, "ip6-localhost") == 0 || | 
|  | strcmp(name, "ip6-loopback") == 0) << name; | 
|  | } | 
|  |  | 
|  | TEST(netdb, getnameinfo_ip6_localhost) { | 
|  | sockaddr_in6 addr; | 
|  | char host[NI_MAXHOST]; | 
|  | memset(&addr, 0, sizeof(sockaddr_in6)); | 
|  | addr.sin6_family = AF_INET6; | 
|  | addr.sin6_addr = in6addr_loopback; | 
|  | ASSERT_EQ(0, getnameinfo(reinterpret_cast<sockaddr*>(&addr), sizeof(addr), | 
|  | host, sizeof(host), nullptr, 0, 0)); | 
|  | VerifyLocalhostName(host); | 
|  | } | 
|  |  | 
|  | static void VerifyLocalhost(hostent *hent) { | 
|  | ASSERT_TRUE(hent != nullptr); | 
|  | VerifyLocalhostName(hent->h_name); | 
|  | for (size_t i = 0; hent->h_aliases[i] != nullptr; ++i) { | 
|  | VerifyLocalhostName(hent->h_aliases[i]); | 
|  | } | 
|  | ASSERT_EQ(hent->h_addrtype, AF_INET); | 
|  | ASSERT_EQ(hent->h_addr[0], 127); | 
|  | ASSERT_EQ(hent->h_addr[1], 0); | 
|  | ASSERT_EQ(hent->h_addr[2], 0); | 
|  | ASSERT_EQ(hent->h_addr[3], 1); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname) { | 
|  | hostent* hp = gethostbyname("localhost"); | 
|  | VerifyLocalhost(hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname2) { | 
|  | hostent* hp = gethostbyname2("localhost", AF_INET); | 
|  | VerifyLocalhost(hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname_r) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[512]; | 
|  | int err; | 
|  | int result = gethostbyname_r("localhost", &hent, buf, sizeof(buf), &hp, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp); | 
|  |  | 
|  | // Change hp->h_addr to test reentrancy. | 
|  | hp->h_addr[0] = 0; | 
|  |  | 
|  | hostent hent2; | 
|  | hostent *hp2; | 
|  | char buf2[512]; | 
|  | result = gethostbyname_r("localhost", &hent2, buf2, sizeof(buf2), &hp2, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp2); | 
|  |  | 
|  | ASSERT_EQ(0, hp->h_addr[0]); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname2_r) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[512]; | 
|  | int err; | 
|  | int result = gethostbyname2_r("localhost", AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp); | 
|  |  | 
|  | // Change hp->h_addr to test reentrancy. | 
|  | hp->h_addr[0] = 0; | 
|  |  | 
|  | hostent hent2; | 
|  | hostent *hp2; | 
|  | char buf2[512]; | 
|  | result = gethostbyname2_r("localhost", AF_INET, &hent2, buf2, sizeof(buf2), &hp2, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp2); | 
|  |  | 
|  | ASSERT_EQ(0, hp->h_addr[0]); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyaddr) { | 
|  | in_addr addr = { htonl(0x7f000001) }; | 
|  | hostent *hp = gethostbyaddr(&addr, sizeof(addr), AF_INET); | 
|  | VerifyLocalhost(hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyaddr_r) { | 
|  | in_addr addr = { htonl(0x7f000001) }; | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[512]; | 
|  | int err; | 
|  | int result = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp); | 
|  |  | 
|  | // Change hp->h_addr to test reentrancy. | 
|  | hp->h_addr[0] = 0; | 
|  |  | 
|  | hostent hent2; | 
|  | hostent *hp2; | 
|  | char buf2[512]; | 
|  | result = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &hent2, buf2, sizeof(buf2), &hp2, &err); | 
|  | ASSERT_EQ(0, result); | 
|  | VerifyLocalhost(hp2); | 
|  |  | 
|  | ASSERT_EQ(0, hp->h_addr[0]); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname_r_ERANGE) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[4]; // Use too small buffer. | 
|  | int err; | 
|  | int result = gethostbyname_r("localhost", &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(NETDB_INTERNAL, err); | 
|  | EXPECT_EQ(ERANGE, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname2_r_ERANGE) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[4]; // Use too small buffer. | 
|  | int err; | 
|  | int result = gethostbyname2_r("localhost", AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(NETDB_INTERNAL, err); | 
|  | EXPECT_EQ(ERANGE, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyaddr_r_ERANGE) { | 
|  | in_addr addr = { htonl(0x7f000001) }; | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[4]; // Use too small buffer. | 
|  | int err; | 
|  | int result = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(NETDB_INTERNAL, err); | 
|  | EXPECT_EQ(ERANGE, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname_r_HOST_NOT_FOUND) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[BUFSIZ]; | 
|  | int err; | 
|  | int result = gethostbyname_r("does.not.exist.google.com", &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(HOST_NOT_FOUND, err); | 
|  | EXPECT_EQ(0, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyname2_r_HOST_NOT_FOUND) { | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[BUFSIZ]; | 
|  | int err; | 
|  | int result = gethostbyname2_r("does.not.exist.google.com", AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(HOST_NOT_FOUND, err); | 
|  | EXPECT_EQ(0, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, gethostbyaddr_r_HOST_NOT_FOUND) { | 
|  | in_addr addr = { htonl(0xffffffff) }; | 
|  | hostent hent; | 
|  | hostent *hp; | 
|  | char buf[BUFSIZ]; | 
|  | int err; | 
|  | int result = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &hent, buf, sizeof(buf), &hp, &err); | 
|  | EXPECT_EQ(HOST_NOT_FOUND, err); | 
|  | EXPECT_EQ(0, result); | 
|  | EXPECT_EQ(nullptr, hp); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getservbyname) { | 
|  | // smtp is TCP-only, so we know we'll get 25/tcp back. | 
|  | servent* s = getservbyname("smtp", nullptr); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("smtp", s->s_name); | 
|  | ASSERT_EQ(25, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("tcp", s->s_proto); | 
|  |  | 
|  | // We get the same result by explicitly asking for tcp. | 
|  | s = getservbyname("smtp", "tcp"); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("smtp", s->s_name); | 
|  | ASSERT_EQ(25, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("tcp", s->s_proto); | 
|  |  | 
|  | // And we get a failure if we explicitly ask for udp. | 
|  | s = getservbyname("smtp", "udp"); | 
|  | ASSERT_TRUE(s == nullptr); | 
|  |  | 
|  | // But there are actually udp services. | 
|  | s = getservbyname("echo", "udp"); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("echo", s->s_name); | 
|  | ASSERT_EQ(7, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("udp", s->s_proto); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getservbyport) { | 
|  | // smtp is TCP-only, so we know we'll get 25/tcp back. | 
|  | servent* s = getservbyport(htons(25), nullptr); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("smtp", s->s_name); | 
|  | ASSERT_EQ(25, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("tcp", s->s_proto); | 
|  |  | 
|  | // We get the same result by explicitly asking for tcp. | 
|  | s = getservbyport(htons(25), "tcp"); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("smtp", s->s_name); | 
|  | ASSERT_EQ(25, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("tcp", s->s_proto); | 
|  |  | 
|  | // And we get a failure if we explicitly ask for udp. | 
|  | s = getservbyport(htons(25), "udp"); | 
|  | ASSERT_TRUE(s == nullptr); | 
|  |  | 
|  | // But there are actually udp services. | 
|  | s = getservbyport(htons(7), "udp"); | 
|  | ASSERT_TRUE(s != nullptr); | 
|  | ASSERT_STREQ("echo", s->s_name); | 
|  | ASSERT_EQ(7, ntohs(s->s_port)); | 
|  | ASSERT_STREQ("udp", s->s_proto); | 
|  | } | 
|  |  | 
|  | TEST(netdb, endnetent_getnetent_setnetent) { | 
|  | setnetent(0); | 
|  | setnetent(1); | 
|  | endnetent(); | 
|  | while (getnetent() != nullptr) { | 
|  | } | 
|  | } | 
|  |  | 
|  | TEST(netdb, getnetbyaddr) { | 
|  | getnetbyaddr(0, 0); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getnetbyname) { | 
|  | getnetbyname("x"); | 
|  | } | 
|  |  | 
|  | TEST(netdb, endprotoent_getprotoent_setprotoent) { | 
|  | setprotoent(0); | 
|  | setprotoent(1); | 
|  | endprotoent(); | 
|  | while (getprotoent() != nullptr) { | 
|  | } | 
|  | } | 
|  |  | 
|  | TEST(netdb, getprotobyname) { | 
|  | getprotobyname("tcp"); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getprotobynumber) { | 
|  | getprotobynumber(6); | 
|  | } | 
|  |  | 
|  | TEST(netdb, endservent_getservent_setservent) { | 
|  | setservent(0); | 
|  | setservent(1); | 
|  | endservent(); | 
|  | size_t service_count = 0; | 
|  | while (getservent() != nullptr) { | 
|  | ++service_count; | 
|  | } | 
|  | ASSERT_GT(service_count, 0U); | 
|  | } | 
|  |  | 
|  | TEST(netdb, getservbyname_getservent_conflicts) { | 
|  | // Calling getservbyname shouldn't affect getservent's iteration order. | 
|  | endservent(); | 
|  | while (getservent() != nullptr) { | 
|  | ASSERT_TRUE(getservbyname("smtp", "tcp") != nullptr); | 
|  | } | 
|  | } | 
|  |  | 
|  | TEST(netdb, getservbyport_getservent_conflicts) { | 
|  | // Calling getservbyport shouldn't affect getservent's iteration order. | 
|  | endservent(); | 
|  | while (getservent() != nullptr) { | 
|  | ASSERT_TRUE(getservbyport(htons(25), "tcp") != nullptr); | 
|  | } | 
|  | } | 
|  |  | 
|  | TEST(netdb, endservent_resets) { | 
|  | endservent(); | 
|  | std::string first_service(getservent()->s_name); | 
|  | endservent(); | 
|  | ASSERT_EQ(first_service, std::string(getservent()->s_name)); | 
|  | } | 
|  |  | 
|  | TEST(netdb, setservent_resets) { | 
|  | endservent(); | 
|  | std::string first_service(getservent()->s_name); | 
|  | setservent(0); | 
|  | ASSERT_EQ(first_service, std::string(getservent()->s_name)); | 
|  | } | 
|  |  | 
|  | TEST(netdb, endhostent_gethostent_sethostent) { | 
|  | sethostent(0); | 
|  | sethostent(1); | 
|  | endhostent(); | 
|  | size_t host_count = 0; | 
|  | while (gethostent() != nullptr) { | 
|  | ++host_count; | 
|  | } | 
|  | ASSERT_GT(host_count, 0U); | 
|  | } | 
|  |  | 
|  | TEST(netdb, endhostent_resets) { | 
|  | endhostent(); | 
|  | std::string first_host(gethostent()->h_name); | 
|  | endhostent(); | 
|  | ASSERT_EQ(first_host, std::string(gethostent()->h_name)); | 
|  | } | 
|  |  | 
|  | TEST(netdb, sethostent_resets) { | 
|  | endhostent(); | 
|  | std::string first_host(gethostent()->h_name); | 
|  | sethostent(0); | 
|  | ASSERT_EQ(first_host, std::string(gethostent()->h_name)); | 
|  | } |