xref: /minix3/tests/lib/libc/gen/t_closefrom.c (revision 11be35a165022172ed3cea20f2b5df0307540b0e)
1*11be35a1SLionel Sambuc /* $NetBSD: t_closefrom.c,v 1.4 2011/05/11 08:11:36 jruoho Exp $ */
2*11be35a1SLionel Sambuc 
3*11be35a1SLionel Sambuc /*-
4*11be35a1SLionel Sambuc  * Copyright (c) 2011 The NetBSD Foundation, Inc.
5*11be35a1SLionel Sambuc  * All rights reserved.
6*11be35a1SLionel Sambuc  *
7*11be35a1SLionel Sambuc  * This code is derived from software contributed to The NetBSD Foundation
8*11be35a1SLionel Sambuc  * by Jukka Ruohonen.
9*11be35a1SLionel Sambuc  *
10*11be35a1SLionel Sambuc  * Redistribution and use in source and binary forms, with or without
11*11be35a1SLionel Sambuc  * modification, are permitted provided that the following conditions
12*11be35a1SLionel Sambuc  * are met:
13*11be35a1SLionel Sambuc  * 1. Redistributions of source code must retain the above copyright
14*11be35a1SLionel Sambuc  *    notice, this list of conditions and the following disclaimer.
15*11be35a1SLionel Sambuc  * 2. Redistributions in binary form must reproduce the above copyright
16*11be35a1SLionel Sambuc  *    notice, this list of conditions and the following disclaimer in the
17*11be35a1SLionel Sambuc  *    documentation and/or other materials provided with the distribution.
18*11be35a1SLionel Sambuc  *
19*11be35a1SLionel Sambuc  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20*11be35a1SLionel Sambuc  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21*11be35a1SLionel Sambuc  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22*11be35a1SLionel Sambuc  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23*11be35a1SLionel Sambuc  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24*11be35a1SLionel Sambuc  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25*11be35a1SLionel Sambuc  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26*11be35a1SLionel Sambuc  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27*11be35a1SLionel Sambuc  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28*11be35a1SLionel Sambuc  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29*11be35a1SLionel Sambuc  * POSSIBILITY OF SUCH DAMAGE.
30*11be35a1SLionel Sambuc  */
31*11be35a1SLionel Sambuc #include <sys/cdefs.h>
32*11be35a1SLionel Sambuc __RCSID("$NetBSD: t_closefrom.c,v 1.4 2011/05/11 08:11:36 jruoho Exp $");
33*11be35a1SLionel Sambuc 
34*11be35a1SLionel Sambuc #include <sys/wait.h>
35*11be35a1SLionel Sambuc 
36*11be35a1SLionel Sambuc #include <atf-c.h>
37*11be35a1SLionel Sambuc #include <errno.h>
38*11be35a1SLionel Sambuc #include <fcntl.h>
39*11be35a1SLionel Sambuc #include <limits.h>
40*11be35a1SLionel Sambuc #include <unistd.h>
41*11be35a1SLionel Sambuc 
42*11be35a1SLionel Sambuc static const char  path[] = "closefrom";
43*11be35a1SLionel Sambuc 
44*11be35a1SLionel Sambuc ATF_TC_WITH_CLEANUP(closefrom_basic);
ATF_TC_HEAD(closefrom_basic,tc)45*11be35a1SLionel Sambuc ATF_TC_HEAD(closefrom_basic, tc)
46*11be35a1SLionel Sambuc {
47*11be35a1SLionel Sambuc 	atf_tc_set_md_var(tc, "descr", "A basic test of closefrom(3), #1");
48*11be35a1SLionel Sambuc }
49*11be35a1SLionel Sambuc 
ATF_TC_BODY(closefrom_basic,tc)50*11be35a1SLionel Sambuc ATF_TC_BODY(closefrom_basic, tc)
51*11be35a1SLionel Sambuc {
52*11be35a1SLionel Sambuc 	int fd, cur1, cur2;
53*11be35a1SLionel Sambuc 
54*11be35a1SLionel Sambuc 	(void)closefrom(STDERR_FILENO + 1);
55*11be35a1SLionel Sambuc 
56*11be35a1SLionel Sambuc 	fd = open(path, O_RDONLY | O_CREAT, 0400);
57*11be35a1SLionel Sambuc 	ATF_REQUIRE(fd >= 0);
58*11be35a1SLionel Sambuc 
59*11be35a1SLionel Sambuc 	cur1 = fcntl(0, F_MAXFD);
60*11be35a1SLionel Sambuc 
61*11be35a1SLionel Sambuc 	ATF_REQUIRE(cur1 == STDERR_FILENO + 1);
62*11be35a1SLionel Sambuc 	ATF_REQUIRE(closefrom(cur1) == 0);
63*11be35a1SLionel Sambuc 
64*11be35a1SLionel Sambuc 	cur2 = fcntl(0, F_MAXFD);
65*11be35a1SLionel Sambuc 
66*11be35a1SLionel Sambuc 	ATF_REQUIRE(cur1 - 1 == cur2);
67*11be35a1SLionel Sambuc 	ATF_REQUIRE(close(fd) == -1);
68*11be35a1SLionel Sambuc 	ATF_REQUIRE(unlink(path) == 0);
69*11be35a1SLionel Sambuc }
70*11be35a1SLionel Sambuc 
ATF_TC_CLEANUP(closefrom_basic,tc)71*11be35a1SLionel Sambuc ATF_TC_CLEANUP(closefrom_basic, tc)
72*11be35a1SLionel Sambuc {
73*11be35a1SLionel Sambuc 	(void)unlink(path);
74*11be35a1SLionel Sambuc }
75*11be35a1SLionel Sambuc 
76*11be35a1SLionel Sambuc ATF_TC_WITH_CLEANUP(closefrom_buffer);
ATF_TC_HEAD(closefrom_buffer,tc)77*11be35a1SLionel Sambuc ATF_TC_HEAD(closefrom_buffer, tc)
78*11be35a1SLionel Sambuc {
79*11be35a1SLionel Sambuc 	atf_tc_set_md_var(tc, "descr", "A basic test of closefrom(3), #2");
80*11be35a1SLionel Sambuc }
81*11be35a1SLionel Sambuc 
ATF_TC_BODY(closefrom_buffer,tc)82*11be35a1SLionel Sambuc ATF_TC_BODY(closefrom_buffer, tc)
83*11be35a1SLionel Sambuc {
84*11be35a1SLionel Sambuc 	int buf[16], cur, half;
85*11be35a1SLionel Sambuc 	size_t i;
86*11be35a1SLionel Sambuc 
87*11be35a1SLionel Sambuc 	/*
88*11be35a1SLionel Sambuc 	 * Open a buffer of descriptors, close the half of
89*11be35a1SLionel Sambuc 	 * these and verify that the result is consistent.
90*11be35a1SLionel Sambuc 	 */
91*11be35a1SLionel Sambuc 	ATF_REQUIRE(closefrom(STDERR_FILENO + 1) == 0);
92*11be35a1SLionel Sambuc 
93*11be35a1SLionel Sambuc 	cur = fcntl(0, F_MAXFD);
94*11be35a1SLionel Sambuc 	ATF_REQUIRE(cur == STDERR_FILENO);
95*11be35a1SLionel Sambuc 
96*11be35a1SLionel Sambuc 	for (i = 0; i < __arraycount(buf); i++) {
97*11be35a1SLionel Sambuc 		buf[i] = open(path, O_RDWR | O_CREAT, 0600);
98*11be35a1SLionel Sambuc 		ATF_REQUIRE(buf[i] >= 0);
99*11be35a1SLionel Sambuc 	}
100*11be35a1SLionel Sambuc 
101*11be35a1SLionel Sambuc 	cur = fcntl(0, F_MAXFD);
102*11be35a1SLionel Sambuc 	ATF_REQUIRE(cur == __arraycount(buf) + STDERR_FILENO);
103*11be35a1SLionel Sambuc 
104*11be35a1SLionel Sambuc 	half = STDERR_FILENO + __arraycount(buf) / 2;
105*11be35a1SLionel Sambuc 	ATF_REQUIRE(closefrom(half) == 0);
106*11be35a1SLionel Sambuc 
107*11be35a1SLionel Sambuc 	cur = fcntl(0, F_MAXFD);
108*11be35a1SLionel Sambuc 	ATF_REQUIRE(cur == half - 1);
109*11be35a1SLionel Sambuc 
110*11be35a1SLionel Sambuc 	for (i = 0; i < __arraycount(buf); i++)
111*11be35a1SLionel Sambuc 		(void)close(buf[i]);
112*11be35a1SLionel Sambuc }
113*11be35a1SLionel Sambuc 
ATF_TC_CLEANUP(closefrom_buffer,tc)114*11be35a1SLionel Sambuc ATF_TC_CLEANUP(closefrom_buffer, tc)
115*11be35a1SLionel Sambuc {
116*11be35a1SLionel Sambuc 	(void)unlink(path);
117*11be35a1SLionel Sambuc }
118*11be35a1SLionel Sambuc 
119*11be35a1SLionel Sambuc ATF_TC(closefrom_err);
ATF_TC_HEAD(closefrom_err,tc)120*11be35a1SLionel Sambuc ATF_TC_HEAD(closefrom_err, tc)
121*11be35a1SLionel Sambuc {
122*11be35a1SLionel Sambuc 	atf_tc_set_md_var(tc, "descr", "Test errors from closefrom(3)");
123*11be35a1SLionel Sambuc }
124*11be35a1SLionel Sambuc 
ATF_TC_BODY(closefrom_err,tc)125*11be35a1SLionel Sambuc ATF_TC_BODY(closefrom_err, tc)
126*11be35a1SLionel Sambuc {
127*11be35a1SLionel Sambuc 
128*11be35a1SLionel Sambuc 	errno = 0;
129*11be35a1SLionel Sambuc 	ATF_REQUIRE_ERRNO(EBADF, closefrom(-INT_MAX) == -1);
130*11be35a1SLionel Sambuc }
131*11be35a1SLionel Sambuc 
132*11be35a1SLionel Sambuc ATF_TC(closefrom_one);
ATF_TC_HEAD(closefrom_one,tc)133*11be35a1SLionel Sambuc ATF_TC_HEAD(closefrom_one, tc)
134*11be35a1SLionel Sambuc {
135*11be35a1SLionel Sambuc 	atf_tc_set_md_var(tc, "descr", "Test closefrom(1)");
136*11be35a1SLionel Sambuc }
137*11be35a1SLionel Sambuc 
ATF_TC_BODY(closefrom_one,tc)138*11be35a1SLionel Sambuc ATF_TC_BODY(closefrom_one, tc)
139*11be35a1SLionel Sambuc {
140*11be35a1SLionel Sambuc 	pid_t pid;
141*11be35a1SLionel Sambuc 	int sta;
142*11be35a1SLionel Sambuc 
143*11be35a1SLionel Sambuc 	pid = fork();
144*11be35a1SLionel Sambuc 	ATF_REQUIRE(pid >= 0);
145*11be35a1SLionel Sambuc 
146*11be35a1SLionel Sambuc 	if (pid == 0) {
147*11be35a1SLionel Sambuc 
148*11be35a1SLionel Sambuc 		if (closefrom(1) != 0)
149*11be35a1SLionel Sambuc 			_exit(10);
150*11be35a1SLionel Sambuc 
151*11be35a1SLionel Sambuc 		_exit(fcntl(0, F_MAXFD));
152*11be35a1SLionel Sambuc 	}
153*11be35a1SLionel Sambuc 
154*11be35a1SLionel Sambuc 
155*11be35a1SLionel Sambuc 	(void)wait(&sta);
156*11be35a1SLionel Sambuc 
157*11be35a1SLionel Sambuc 	/*
158*11be35a1SLionel Sambuc 	 * STDIN_FILENO sould still be open; WEXITSTATUS(1) == 0.
159*11be35a1SLionel Sambuc 	 */
160*11be35a1SLionel Sambuc 	if (WIFEXITED(sta) == 0 || WEXITSTATUS(sta) != 0)
161*11be35a1SLionel Sambuc 		atf_tc_fail("not all descriptors were closed");
162*11be35a1SLionel Sambuc }
163*11be35a1SLionel Sambuc 
ATF_TP_ADD_TCS(tp)164*11be35a1SLionel Sambuc ATF_TP_ADD_TCS(tp)
165*11be35a1SLionel Sambuc {
166*11be35a1SLionel Sambuc 
167*11be35a1SLionel Sambuc 	ATF_TP_ADD_TC(tp, closefrom_basic);
168*11be35a1SLionel Sambuc 	ATF_TP_ADD_TC(tp, closefrom_buffer);
169*11be35a1SLionel Sambuc 	ATF_TP_ADD_TC(tp, closefrom_err);
170*11be35a1SLionel Sambuc 	ATF_TP_ADD_TC(tp, closefrom_one);
171*11be35a1SLionel Sambuc 
172*11be35a1SLionel Sambuc 	return atf_no_error();
173*11be35a1SLionel Sambuc }
174