Merge "Migrate system/extra getaddrinfo test, and fix a bug in getservbyname(3)."

This commit is contained in:
Elliott Hughes 2014-10-23 17:57:05 +00:00 committed by Gerrit Code Review
commit 6c5694b6c8
2 changed files with 76 additions and 21 deletions

View File

@ -25,29 +25,19 @@
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE. * SUCH DAMAGE.
*/ */
#include <sys/cdefs.h>
#include <sys/types.h>
#include <netdb.h> #include <netdb.h>
#include "servent.h" #include "servent.h"
struct servent * struct servent* getservbyname(const char* name, const char* proto) {
getservbyname(const char *name, const char *proto) res_static rs = __res_get_static();
{ rs->servent_ptr = NULL;
res_static rs = __res_get_static(); struct servent* s;
while ((s = getservent_r(rs)) != NULL) {
if (rs == NULL || proto == NULL || name == NULL) { if (strcmp(s->s_name, name) == 0 && (proto == NULL || strcmp(s->s_proto, proto) == 0)) {
errno = EINVAL; return s;
return NULL;
} }
}
rs->servent_ptr = NULL; return NULL;
while (1) {
struct servent* s = getservent_r(rs);
if (s == NULL)
break;
if ( !strcmp( s->s_name, name ) && !strcmp( s->s_proto, proto ) )
return s;
}
return NULL;
} }

View File

@ -24,6 +24,47 @@
TEST(netdb, getaddrinfo_NULL_hints) { TEST(netdb, getaddrinfo_NULL_hints) {
addrinfo* ai = NULL; addrinfo* ai = NULL;
ASSERT_EQ(0, getaddrinfo("localhost", "9999", NULL, &ai)); ASSERT_EQ(0, getaddrinfo("localhost", "9999", NULL, &ai));
bool saw_tcp = false;
bool saw_udp = false;
for (addrinfo* p = ai; p != NULL; 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 = NULL;
ASSERT_EQ(0, getaddrinfo("localhost", "smtp", NULL, &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_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
hints.ai_protocol = IPPROTO_TCP;
addrinfo* ai = NULL;
ASSERT_EQ(0, getaddrinfo( "localhost", "9999", &hints, &ai));
ASSERT_EQ(AF_INET, ai->ai_family);
ASSERT_EQ(SOCK_STREAM, ai->ai_socktype);
ASSERT_EQ(IPPROTO_TCP, ai->ai_protocol);
ASSERT_TRUE(ai->ai_next == NULL);
freeaddrinfo(ai); freeaddrinfo(ai);
} }
@ -65,3 +106,27 @@ TEST(netdb, gethostbyname) {
ASSERT_EQ(hent->h_addr[2], 0); ASSERT_EQ(hent->h_addr[2], 0);
ASSERT_EQ(hent->h_addr[3], 1); ASSERT_EQ(hent->h_addr[3], 1);
} }
TEST(netdb, getservbyname) {
// smtp is TCP-only, so we know we'll get 25/tcp back.
servent* s = getservbyname("smtp", NULL);
ASSERT_TRUE(s != NULL);
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 != NULL);
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 == NULL);
// But there are actually udp services.
s = getservbyname("echo", "udp");
ASSERT_TRUE(s != NULL);
ASSERT_EQ(7, ntohs(s->s_port));
ASSERT_STREQ("udp", s->s_proto);
}