1272343Sngie/* $NetBSD: t_getprotoent.c,v 1.2 2012/04/04 10:03:53 joerg Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2011 The NetBSD Foundation, Inc. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * This code is derived from software contributed to The NetBSD Foundation 8272343Sngie * by Jukka Ruohonen. 9272343Sngie * 10272343Sngie * Redistribution and use in source and binary forms, with or without 11272343Sngie * modification, are permitted provided that the following conditions 12272343Sngie * are met: 13272343Sngie * 1. Redistributions of source code must retain the above copyright 14272343Sngie * notice, this list of conditions and the following disclaimer. 15272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 16272343Sngie * notice, this list of conditions and the following disclaimer in the 17272343Sngie * documentation and/or other materials provided with the distribution. 18272343Sngie * 19272343Sngie * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20272343Sngie * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21272343Sngie * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22272343Sngie * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23272343Sngie * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24272343Sngie * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25272343Sngie * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26272343Sngie * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27272343Sngie * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28272343Sngie * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29272343Sngie * POSSIBILITY OF SUCH DAMAGE. 30272343Sngie */ 31272343Sngie#include <sys/cdefs.h> 32272343Sngie__RCSID("$NetBSD: t_getprotoent.c,v 1.2 2012/04/04 10:03:53 joerg Exp $"); 33272343Sngie 34272343Sngie#include <atf-c.h> 35272343Sngie#include <netdb.h> 36272343Sngie#include <limits.h> 37272343Sngie#include <stdio.h> 38272343Sngie#include <string.h> 39272343Sngie 40272343Sngiestatic const struct { 41272343Sngie const char *name; 42272343Sngie int number; 43272343Sngie} protos[] = { 44272343Sngie 45272343Sngie { "icmp", 1 }, { "tcp", 6 }, { "udp", 17 }, { "gre", 47 }, 46272343Sngie { "esp", 50 }, { "ah", 51 }, { "sctp", 132}, { "ipv6-icmp", 58 } 47272343Sngie}; 48272343Sngie 49272343SngieATF_TC(endprotoent_rewind); 50272343SngieATF_TC_HEAD(endprotoent_rewind, tc) 51272343Sngie{ 52272343Sngie atf_tc_set_md_var(tc, "descr", "Check that endprotoent(3) rewinds"); 53272343Sngie} 54272343Sngie 55272343SngieATF_TC_BODY(endprotoent_rewind, tc) 56272343Sngie{ 57272343Sngie struct protoent *p; 58272343Sngie int i = 0; 59272343Sngie 60272343Sngie setprotoent(0); 61272343Sngie 62272343Sngie while ((p = getprotoent()) != NULL && i <= 10) { 63272343Sngie ATF_REQUIRE(p->p_proto == i); 64272343Sngie i++; 65272343Sngie } 66272343Sngie 67272343Sngie i = 0; 68272343Sngie endprotoent(); 69272343Sngie 70272343Sngie while ((p = getprotoent()) != NULL && i <= 10) { 71272343Sngie ATF_REQUIRE(p->p_proto == i); 72272343Sngie i++; 73272343Sngie } 74272343Sngie 75272343Sngie endprotoent(); 76272343Sngie} 77272343Sngie 78272343SngieATF_TC(getprotobyname_basic); 79272343SngieATF_TC_HEAD(getprotobyname_basic, tc) 80272343Sngie{ 81272343Sngie atf_tc_set_md_var(tc, "descr", "A naive test of getprotobyname(3)"); 82272343Sngie} 83272343Sngie 84272343SngieATF_TC_BODY(getprotobyname_basic, tc) 85272343Sngie{ 86272343Sngie struct protoent *p; 87272343Sngie size_t i; 88272343Sngie 89272343Sngie for (i = 0; i < __arraycount(protos); i++) { 90272343Sngie 91272343Sngie p = getprotobyname(protos[i].name); 92272343Sngie 93272343Sngie ATF_REQUIRE(p != NULL); 94272343Sngie ATF_REQUIRE(p->p_proto == protos[i].number); 95272343Sngie ATF_REQUIRE(strcmp(p->p_name, protos[i].name) == 0); 96272343Sngie } 97272343Sngie 98272343Sngie endprotoent(); 99272343Sngie} 100272343Sngie 101272343SngieATF_TC(getprotobyname_err); 102272343SngieATF_TC_HEAD(getprotobyname_err, tc) 103272343Sngie{ 104272343Sngie atf_tc_set_md_var(tc, "descr", "Test EOF from getprotobyname(3)"); 105272343Sngie} 106272343Sngie 107272343SngieATF_TC_BODY(getprotobyname_err, tc) 108272343Sngie{ 109272343Sngie static const char * name[] = 110272343Sngie { "xxx", "yyy", "xyz", ".as.d}9x.._?!!#\xa4,\xa8^//&%%,", 111272343Sngie "0", "", "tCp", "uDp", "t c p", "tcp ", " tcp" }; 112272343Sngie 113272343Sngie size_t i; 114272343Sngie 115272343Sngie for (i = 0; i < __arraycount(name); i++) 116272343Sngie ATF_REQUIRE(getprotobyname(name[i]) == NULL); 117272343Sngie 118272343Sngie endprotoent(); 119272343Sngie} 120272343Sngie 121272343SngieATF_TC(getprotobynumber_basic); 122272343SngieATF_TC_HEAD(getprotobynumber_basic, tc) 123272343Sngie{ 124272343Sngie atf_tc_set_md_var(tc, "descr", "A naive test of getprotobynumber(3)"); 125272343Sngie} 126272343Sngie 127272343SngieATF_TC_BODY(getprotobynumber_basic, tc) 128272343Sngie{ 129272343Sngie struct protoent *p; 130272343Sngie size_t i; 131272343Sngie 132272343Sngie /* 133272343Sngie * No ATF_CHECK() due static storage. 134272343Sngie */ 135272343Sngie for (i = 0; i < __arraycount(protos); i++) { 136272343Sngie 137272343Sngie p = getprotobynumber(protos[i].number); 138272343Sngie 139272343Sngie ATF_REQUIRE(p != NULL); 140272343Sngie ATF_REQUIRE(p->p_proto == protos[i].number); 141272343Sngie ATF_REQUIRE(strcmp(p->p_name, protos[i].name) == 0); 142272343Sngie } 143272343Sngie 144272343Sngie endprotoent(); 145272343Sngie} 146272343Sngie 147272343SngieATF_TC(getprotobynumber_err); 148272343SngieATF_TC_HEAD(getprotobynumber_err, tc) 149272343Sngie{ 150272343Sngie atf_tc_set_md_var(tc, "descr", "Test EOF from getprotobynumber(3)"); 151272343Sngie} 152272343Sngie 153272343SngieATF_TC_BODY(getprotobynumber_err, tc) 154272343Sngie{ 155272343Sngie static const int number[] = { -1, -99999, INT_MAX, 1000000000 }; 156272343Sngie size_t i; 157272343Sngie 158272343Sngie for (i = 0; i < __arraycount(number); i++) 159272343Sngie ATF_REQUIRE(getprotobynumber(number[i]) == NULL); 160272343Sngie 161272343Sngie endprotoent(); 162272343Sngie} 163272343Sngie 164272343SngieATF_TC(getprotoent_next); 165272343SngieATF_TC_HEAD(getprotoent_next, tc) 166272343Sngie{ 167272343Sngie atf_tc_set_md_var(tc, "descr", "getprotoent(3) returns next line?"); 168272343Sngie} 169272343Sngie 170272343SngieATF_TC_BODY(getprotoent_next, tc) 171272343Sngie{ 172272343Sngie struct protoent *p; 173272343Sngie int i = 0; 174272343Sngie 175272343Sngie /* 176272343Sngie * The range [0, 60] is already reserved by IANA. 177272343Sngie */ 178272343Sngie while ((p = getprotoent()) != NULL && i <= 60) { 179272343Sngie ATF_CHECK(p->p_proto == i); 180272343Sngie i++; 181272343Sngie } 182272343Sngie 183272343Sngie endprotoent(); 184272343Sngie} 185272343Sngie 186272343SngieATF_TC(setprotoent_rewind); 187272343SngieATF_TC_HEAD(setprotoent_rewind, tc) 188272343Sngie{ 189272343Sngie atf_tc_set_md_var(tc, "descr", "Check that setprotoent(3) rewinds"); 190272343Sngie} 191272343Sngie 192272343SngieATF_TC_BODY(setprotoent_rewind, tc) 193272343Sngie{ 194272343Sngie struct protoent *p; 195272343Sngie 196272343Sngie setprotoent(0); 197272343Sngie 198272343Sngie p = getprotoent(); 199272343Sngie ATF_REQUIRE(p->p_proto == 0); 200272343Sngie 201272343Sngie p = getprotoent(); 202272343Sngie ATF_REQUIRE(p->p_proto == 1); 203272343Sngie 204272343Sngie p = getprotoent(); 205272343Sngie ATF_REQUIRE(p->p_proto == 2); 206272343Sngie 207272343Sngie setprotoent(0); 208272343Sngie 209272343Sngie p = getprotoent(); 210272343Sngie ATF_REQUIRE(p->p_proto == 0); 211272343Sngie 212272343Sngie p = getprotoent(); 213272343Sngie ATF_REQUIRE(p->p_proto == 1); 214272343Sngie 215272343Sngie p = getprotoent(); 216272343Sngie ATF_REQUIRE(p->p_proto == 2); 217272343Sngie 218272343Sngie endprotoent(); 219272343Sngie} 220272343Sngie 221272343SngieATF_TP_ADD_TCS(tp) 222272343Sngie{ 223272343Sngie 224272343Sngie ATF_TP_ADD_TC(tp, getprotobyname_basic); 225272343Sngie ATF_TP_ADD_TC(tp, getprotobyname_err); 226272343Sngie ATF_TP_ADD_TC(tp, getprotobynumber_basic); 227272343Sngie ATF_TP_ADD_TC(tp, getprotobynumber_err); 228272343Sngie ATF_TP_ADD_TC(tp, endprotoent_rewind); 229272343Sngie ATF_TP_ADD_TC(tp, getprotoent_next); 230272343Sngie ATF_TP_ADD_TC(tp, setprotoent_rewind); 231272343Sngie 232272343Sngie return atf_no_error(); 233272343Sngie} 234