t_pipe2.c revision 314817 
1/* $NetBSD: t_pipe2.c,v 1.9 2017/01/13 21:19:45 christos 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 Christos Zoulas.
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 * 3. All advertising materials mentioning features or use of this software
19 *    must display the following acknowledgement:
20 *        This product includes software developed by the NetBSD
21 *        Foundation, Inc. and its contributors.
22 * 4. Neither the name of The NetBSD Foundation nor the names of its
23 *    contributors may be used to endorse or promote products derived
24 *    from this software without specific prior written permission.
25 *
26 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
27 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
28 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
30 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
31 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
32 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
33 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
34 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
35 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36 * POSSIBILITY OF SUCH DAMAGE.
37 */
38#include <sys/cdefs.h>
39__RCSID("$NetBSD: t_pipe2.c,v 1.9 2017/01/13 21:19:45 christos Exp $");
40
41#include <atf-c.h>
42#include <fcntl.h>
43#include <unistd.h>
44#include <stdlib.h>
45#include <errno.h>
46#include <sys/resource.h>
47
48static void
49run(int flags)
50{
51	int fd[2], i;
52
53	while ((i = open("/", O_RDONLY)) < 3)
54		ATF_REQUIRE(i != -1);
55
56#ifdef __FreeBSD__
57	closefrom(3);
58#else
59	ATF_REQUIRE_MSG(closefrom(3) != -1, "closefrom failed: %s",
60	    strerror(errno));
61#endif
62
63	ATF_REQUIRE(pipe2(fd, flags) == 0);
64
65	ATF_REQUIRE(fd[0] == 3);
66	ATF_REQUIRE(fd[1] == 4);
67
68	if (flags & O_CLOEXEC) {
69		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) != 0);
70		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) != 0);
71	} else {
72		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) == 0);
73		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) == 0);
74	}
75
76	if (flags & O_NONBLOCK) {
77		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) != 0);
78		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) != 0);
79	} else {
80		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) == 0);
81		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) == 0);
82	}
83
84#ifndef __FreeBSD__
85	if (flags & O_NOSIGPIPE) {
86		ATF_REQUIRE(fcntl(fd[0], F_GETNOSIGPIPE) != 0);
87		ATF_REQUIRE(fcntl(fd[1], F_GETNOSIGPIPE) != 0);
88	} else {
89		ATF_REQUIRE(fcntl(fd[0], F_GETNOSIGPIPE) == 0);
90		ATF_REQUIRE(fcntl(fd[1], F_GETNOSIGPIPE) == 0);
91	}
92#endif
93
94	ATF_REQUIRE(close(fd[0]) != -1);
95	ATF_REQUIRE(close(fd[1]) != -1);
96}
97
98ATF_TC(pipe2_basic);
99ATF_TC_HEAD(pipe2_basic, tc)
100{
101	atf_tc_set_md_var(tc, "descr", "A basic test of pipe2(2)");
102}
103
104ATF_TC_BODY(pipe2_basic, tc)
105{
106	run(0);
107}
108
109ATF_TC(pipe2_consume);
110ATF_TC_HEAD(pipe2_consume, tc)
111{
112	atf_tc_set_md_var(tc, "descr", "Test that consuming file descriptors "
113	    "with pipe2(2) does not crash the system (PR kern/46457)");
114}
115
116ATF_TC_BODY(pipe2_consume, tc)
117{
118	struct rlimit rl;
119	int err, filedes[2];
120	int old;
121
122#ifdef __FreeBSD__
123	closefrom(4);
124#else
125	ATF_REQUIRE_MSG(closefrom(4) != -1, "closefrom failed: %s",
126	    strerror(errno));
127#endif
128
129	err = getrlimit(RLIMIT_NOFILE, &rl);
130	ATF_REQUIRE(err == 0);
131	/*
132	 * The heart of this test is to run against the number of open
133	 * file descriptor limit in the middle of a pipe2() call - i.e.
134	 * before the call only a single descriptor may be openend.
135	 */
136	old = rl.rlim_cur;
137	rl.rlim_cur = 4;
138	err = setrlimit(RLIMIT_NOFILE, &rl);
139	ATF_REQUIRE(err == 0);
140
141	err = pipe2(filedes, O_CLOEXEC);
142	ATF_REQUIRE(err == -1);
143	rl.rlim_cur = old;
144	err = setrlimit(RLIMIT_NOFILE, &rl);
145}
146
147ATF_TC(pipe2_nonblock);
148ATF_TC_HEAD(pipe2_nonblock, tc)
149{
150	atf_tc_set_md_var(tc, "descr", "A non-blocking test of pipe2(2)");
151}
152
153ATF_TC_BODY(pipe2_nonblock, tc)
154{
155	run(O_NONBLOCK);
156}
157
158ATF_TC(pipe2_cloexec);
159ATF_TC_HEAD(pipe2_cloexec, tc)
160{
161	atf_tc_set_md_var(tc, "descr", "A close-on-exec test of pipe2(2)");
162}
163
164ATF_TC_BODY(pipe2_cloexec, tc)
165{
166	run(O_CLOEXEC);
167}
168
169#ifdef __NetBSD__
170ATF_TC(pipe2_nosigpipe);
171ATF_TC_HEAD(pipe2_nosigpipe, tc)
172{
173	atf_tc_set_md_var(tc, "descr", "A no sigpipe test of pipe2(2)");
174}
175
176ATF_TC_BODY(pipe2_nosigpipe, tc)
177{
178	run(O_NOSIGPIPE);
179}
180#endif
181
182ATF_TC(pipe2_einval);
183ATF_TC_HEAD(pipe2_einval, tc)
184{
185	atf_tc_set_md_var(tc, "descr", "A error check of pipe2(2)");
186}
187
188ATF_TC_BODY(pipe2_einval, tc)
189{
190	int fd[2];
191	ATF_REQUIRE_ERRNO(EINVAL, pipe2(fd, O_ASYNC) == -1);
192}
193
194ATF_TP_ADD_TCS(tp)
195{
196
197	ATF_TP_ADD_TC(tp, pipe2_basic);
198	ATF_TP_ADD_TC(tp, pipe2_consume);
199	ATF_TP_ADD_TC(tp, pipe2_nonblock);
200	ATF_TP_ADD_TC(tp, pipe2_cloexec);
201#ifdef __NetBSD__
202	ATF_TP_ADD_TC(tp, pipe2_nosigpipe);
203#endif
204	ATF_TP_ADD_TC(tp, pipe2_einval);
205
206	return atf_no_error();
207}
208