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