t_pipe2.c revision 276478 
1/* $NetBSD: t_pipe2.c,v 1.8 2012/05/16 13:54:28 jruoho 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.8 2012/05/16 13:54:28 jruoho 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(fcntl(3, F_CLOSEM) != -1);
60#endif
61
62	ATF_REQUIRE(pipe2(fd, flags) == 0);
63
64	ATF_REQUIRE(fd[0] == 3);
65	ATF_REQUIRE(fd[1] == 4);
66
67	if (flags & O_CLOEXEC) {
68		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) != 0);
69		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) != 0);
70	} else {
71		ATF_REQUIRE((fcntl(fd[0], F_GETFD) & FD_CLOEXEC) == 0);
72		ATF_REQUIRE((fcntl(fd[1], F_GETFD) & FD_CLOEXEC) == 0);
73	}
74
75	if (flags & O_NONBLOCK) {
76		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) != 0);
77		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) != 0);
78	} else {
79		ATF_REQUIRE((fcntl(fd[0], F_GETFL) & O_NONBLOCK) == 0);
80		ATF_REQUIRE((fcntl(fd[1], F_GETFL) & O_NONBLOCK) == 0);
81	}
82
83#ifndef __FreeBSD__
84	if (flags & O_NOSIGPIPE) {
85		ATF_REQUIRE(fcntl(fd[0], F_GETNOSIGPIPE) != 0);
86		ATF_REQUIRE(fcntl(fd[1], F_GETNOSIGPIPE) != 0);
87	} else {
88		ATF_REQUIRE(fcntl(fd[0], F_GETNOSIGPIPE) == 0);
89		ATF_REQUIRE(fcntl(fd[1], F_GETNOSIGPIPE) == 0);
90	}
91#endif
92
93	ATF_REQUIRE(close(fd[0]) != -1);
94	ATF_REQUIRE(close(fd[1]) != -1);
95}
96
97ATF_TC(pipe2_basic);
98ATF_TC_HEAD(pipe2_basic, tc)
99{
100	atf_tc_set_md_var(tc, "descr", "A basic test of pipe2(2)");
101}
102
103ATF_TC_BODY(pipe2_basic, tc)
104{
105	run(0);
106}
107
108ATF_TC(pipe2_consume);
109ATF_TC_HEAD(pipe2_consume, tc)
110{
111	atf_tc_set_md_var(tc, "descr", "Test that consuming file descriptors "
112	    "with pipe2(2) does not crash the system (PR kern/46457)");
113}
114
115ATF_TC_BODY(pipe2_consume, tc)
116{
117	struct rlimit rl;
118	int err, filedes[2];
119#ifdef __FreeBSD__
120	int old;
121
122	closefrom(4);
123#else
124	err = fcntl(4, F_CLOSEM);
125	ATF_REQUIRE(err == 0);
126#endif
127
128	err = getrlimit(RLIMIT_NOFILE, &rl);
129	ATF_REQUIRE(err == 0);
130	/*
131	 * The heart of this test is to run against the number of open
132	 * file descriptor limit in the middle of a pipe2() call - i.e.
133	 * before the call only a single descriptor may be openend.
134	 */
135#ifdef __FreeBSD__
136	old = rl.rlim_cur;
137#endif
138	rl.rlim_cur = 4;
139	err = setrlimit(RLIMIT_NOFILE, &rl);
140	ATF_REQUIRE(err == 0);
141
142	err = pipe2(filedes, O_CLOEXEC);
143	ATF_REQUIRE(err == -1);
144#ifdef __FreeBSD__
145	rl.rlim_cur = old;
146	err = setrlimit(RLIMIT_NOFILE, &rl);
147#endif
148}
149
150ATF_TC(pipe2_nonblock);
151ATF_TC_HEAD(pipe2_nonblock, tc)
152{
153	atf_tc_set_md_var(tc, "descr", "A non-blocking test of pipe2(2)");
154}
155
156ATF_TC_BODY(pipe2_nonblock, tc)
157{
158	run(O_NONBLOCK);
159}
160
161ATF_TC(pipe2_cloexec);
162ATF_TC_HEAD(pipe2_cloexec, tc)
163{
164	atf_tc_set_md_var(tc, "descr", "A close-on-exec test of pipe2(2)");
165}
166
167ATF_TC_BODY(pipe2_cloexec, tc)
168{
169	run(O_CLOEXEC);
170}
171
172#ifdef __NetBSD__
173ATF_TC(pipe2_nosigpipe);
174ATF_TC_HEAD(pipe2_nosigpipe, tc)
175{
176	atf_tc_set_md_var(tc, "descr", "A no sigpipe test of pipe2(2)");
177}
178
179ATF_TC_BODY(pipe2_nosigpipe, tc)
180{
181	run(O_NOSIGPIPE);
182}
183#endif
184
185ATF_TC(pipe2_einval);
186ATF_TC_HEAD(pipe2_einval, tc)
187{
188	atf_tc_set_md_var(tc, "descr", "A error check of pipe2(2)");
189}
190
191ATF_TC_BODY(pipe2_einval, tc)
192{
193	int fd[2];
194	ATF_REQUIRE_ERRNO(EINVAL, pipe2(fd, O_ASYNC) == -1);
195}
196
197ATF_TP_ADD_TCS(tp)
198{
199
200	ATF_TP_ADD_TC(tp, pipe2_basic);
201	ATF_TP_ADD_TC(tp, pipe2_consume);
202	ATF_TP_ADD_TC(tp, pipe2_nonblock);
203	ATF_TP_ADD_TC(tp, pipe2_cloexec);
204#ifdef __NetBSD__
205	ATF_TP_ADD_TC(tp, pipe2_nosigpipe);
206#endif
207	ATF_TP_ADD_TC(tp, pipe2_einval);
208
209	return atf_no_error();
210}
211