Home | History | Annotate | Line # | Download | only in net
      1 /* $NetBSD: t_getprotoent.c,v 1.2 2012/04/04 10:03:53 joerg Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2011 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Jukka Ruohonen.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 #include <sys/cdefs.h>
     32 __RCSID("$NetBSD: t_getprotoent.c,v 1.2 2012/04/04 10:03:53 joerg Exp $");
     33 
     34 #include <atf-c.h>
     35 #include <netdb.h>
     36 #include <limits.h>
     37 #include <stdio.h>
     38 #include <string.h>
     39 
     40 static const struct {
     41 	const char *name;
     42 	int	    number;
     43 } protos[] = {
     44 
     45 	{ "icmp", 1 },	{ "tcp", 6 },	{ "udp", 17 },	{ "gre", 47 },
     46 	{ "esp", 50 },	{ "ah", 51 },	{ "sctp", 132},	{ "ipv6-icmp", 58 }
     47 };
     48 
     49 ATF_TC(endprotoent_rewind);
     50 ATF_TC_HEAD(endprotoent_rewind, tc)
     51 {
     52 	atf_tc_set_md_var(tc, "descr", "Check that endprotoent(3) rewinds");
     53 }
     54 
     55 ATF_TC_BODY(endprotoent_rewind, tc)
     56 {
     57 	struct protoent *p;
     58 	int i = 0;
     59 
     60 	setprotoent(0);
     61 
     62 	while ((p = getprotoent()) != NULL && i <= 10) {
     63 		ATF_REQUIRE(p->p_proto == i);
     64 		i++;
     65 	}
     66 
     67 	i = 0;
     68 	endprotoent();
     69 
     70 	while ((p = getprotoent()) != NULL && i <= 10) {
     71 		ATF_REQUIRE(p->p_proto == i);
     72 		i++;
     73 	}
     74 
     75 	endprotoent();
     76 }
     77 
     78 ATF_TC(getprotobyname_basic);
     79 ATF_TC_HEAD(getprotobyname_basic, tc)
     80 {
     81 	atf_tc_set_md_var(tc, "descr", "A naive test of getprotobyname(3)");
     82 }
     83 
     84 ATF_TC_BODY(getprotobyname_basic, tc)
     85 {
     86 	struct protoent *p;
     87 	size_t i;
     88 
     89 	for (i = 0; i < __arraycount(protos); i++) {
     90 
     91 		p = getprotobyname(protos[i].name);
     92 
     93 		ATF_REQUIRE(p != NULL);
     94 		ATF_REQUIRE(p->p_proto == protos[i].number);
     95 		ATF_REQUIRE(strcmp(p->p_name, protos[i].name) == 0);
     96 	}
     97 
     98 	endprotoent();
     99 }
    100 
    101 ATF_TC(getprotobyname_err);
    102 ATF_TC_HEAD(getprotobyname_err, tc)
    103 {
    104 	atf_tc_set_md_var(tc, "descr", "Test EOF from getprotobyname(3)");
    105 }
    106 
    107 ATF_TC_BODY(getprotobyname_err, tc)
    108 {
    109 	static const char * name[] =
    110 	    { "xxx", "yyy", "xyz", ".as.d}9x.._?!!#\xa4,\xa8^//&%%,",
    111 	      "0", "", "tCp", "uDp", "t c p", "tcp ", " tcp" };
    112 
    113 	size_t i;
    114 
    115 	for (i = 0; i < __arraycount(name); i++)
    116 		ATF_REQUIRE(getprotobyname(name[i]) == NULL);
    117 
    118 	endprotoent();
    119 }
    120 
    121 ATF_TC(getprotobynumber_basic);
    122 ATF_TC_HEAD(getprotobynumber_basic, tc)
    123 {
    124 	atf_tc_set_md_var(tc, "descr", "A naive test of getprotobynumber(3)");
    125 }
    126 
    127 ATF_TC_BODY(getprotobynumber_basic, tc)
    128 {
    129 	struct protoent *p;
    130 	size_t i;
    131 
    132 	/*
    133 	 * No ATF_CHECK() due static storage.
    134 	 */
    135 	for (i = 0; i < __arraycount(protos); i++) {
    136 
    137 		p = getprotobynumber(protos[i].number);
    138 
    139 		ATF_REQUIRE(p != NULL);
    140 		ATF_REQUIRE(p->p_proto == protos[i].number);
    141 		ATF_REQUIRE(strcmp(p->p_name, protos[i].name) == 0);
    142 	}
    143 
    144 	endprotoent();
    145 }
    146 
    147 ATF_TC(getprotobynumber_err);
    148 ATF_TC_HEAD(getprotobynumber_err, tc)
    149 {
    150 	atf_tc_set_md_var(tc, "descr", "Test EOF from getprotobynumber(3)");
    151 }
    152 
    153 ATF_TC_BODY(getprotobynumber_err, tc)
    154 {
    155 	static const int number[] = { -1, -99999, INT_MAX, 1000000000 };
    156 	size_t i;
    157 
    158 	for (i = 0; i < __arraycount(number); i++)
    159 		ATF_REQUIRE(getprotobynumber(number[i]) == NULL);
    160 
    161 	endprotoent();
    162 }
    163 
    164 ATF_TC(getprotoent_next);
    165 ATF_TC_HEAD(getprotoent_next, tc)
    166 {
    167 	atf_tc_set_md_var(tc, "descr", "getprotoent(3) returns next line?");
    168 }
    169 
    170 ATF_TC_BODY(getprotoent_next, tc)
    171 {
    172 	struct protoent *p;
    173 	int i = 0;
    174 
    175 	/*
    176 	 * The range [0, 60] is already reserved by IANA.
    177 	 */
    178 	while ((p = getprotoent()) != NULL && i <= 60) {
    179 		ATF_CHECK(p->p_proto == i);
    180 		i++;
    181 	}
    182 
    183 	endprotoent();
    184 }
    185 
    186 ATF_TC(setprotoent_rewind);
    187 ATF_TC_HEAD(setprotoent_rewind, tc)
    188 {
    189 	atf_tc_set_md_var(tc, "descr", "Check that setprotoent(3) rewinds");
    190 }
    191 
    192 ATF_TC_BODY(setprotoent_rewind, tc)
    193 {
    194 	struct protoent *p;
    195 
    196 	setprotoent(0);
    197 
    198 	p = getprotoent();
    199 	ATF_REQUIRE(p->p_proto == 0);
    200 
    201 	p = getprotoent();
    202 	ATF_REQUIRE(p->p_proto == 1);
    203 
    204 	p = getprotoent();
    205 	ATF_REQUIRE(p->p_proto == 2);
    206 
    207 	setprotoent(0);
    208 
    209 	p = getprotoent();
    210 	ATF_REQUIRE(p->p_proto == 0);
    211 
    212 	p = getprotoent();
    213 	ATF_REQUIRE(p->p_proto == 1);
    214 
    215 	p = getprotoent();
    216 	ATF_REQUIRE(p->p_proto == 2);
    217 
    218 	endprotoent();
    219 }
    220 
    221 ATF_TP_ADD_TCS(tp)
    222 {
    223 
    224 	ATF_TP_ADD_TC(tp, getprotobyname_basic);
    225 	ATF_TP_ADD_TC(tp, getprotobyname_err);
    226 	ATF_TP_ADD_TC(tp, getprotobynumber_basic);
    227 	ATF_TP_ADD_TC(tp, getprotobynumber_err);
    228 	ATF_TP_ADD_TC(tp, endprotoent_rewind);
    229 	ATF_TP_ADD_TC(tp, getprotoent_next);
    230 	ATF_TP_ADD_TC(tp, setprotoent_rewind);
    231 
    232 	return atf_no_error();
    233 }
    234