2 * Copyright (c) 2009 Hudson River Trading LLC
3 * Written by: John H. Baldwin <jhb@FreeBSD.org>
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
32 * Regression tests for the closefrom(2) system call.
35 #include <sys/param.h>
61 printf("ok %d - %s\n", test, descr);
66 fail(const char *descr, const char *fmt, ...)
70 printf("not ok %d - %s", test, descr);
82 #define fail_err(descr) fail((descr), "%s", strerror(errno))
85 cok(struct shared_info *info, const char *descr)
89 strlcpy(info->tag, descr, sizeof(info->tag));
94 cfail(struct shared_info *info, const char *descr, const char *fmt, ...)
99 strlcpy(info->tag, descr, sizeof(info->tag));
102 vsprintf(info->message, fmt, ap);
108 #define cfail_err(info, descr) cfail((info), (descr), "%s", strerror(errno))
111 * Use kinfo_getfile() to fetch the list of file descriptors and figure out
112 * the highest open file descriptor.
117 struct kinfo_file *kif;
120 kif = kinfo_getfile(getpid(), &cnt);
122 fail_err("kinfo_getfile");
124 for (i = 0; i < cnt; i++)
125 if (kif[i].kf_fd > highest)
126 highest = kif[i].kf_fd;
136 fd = open(_PATH_DEVNULL, O_RDONLY);
138 fail_err("open(\" "_PATH_DEVNULL" \")");
145 struct shared_info *info;
147 int fd, flags, i, start;
151 /* We better start up with fd's 0, 1, and 2 open. */
154 fail("open", "bad descriptor %d", start);
157 /* Make sure highest_fd() works. */
160 fail("highest_fd", "bad descriptor %d != %d", start, fd);
163 /* Try to use closefrom() for just closing fd 3. */
164 closefrom(start + 1);
167 fail("closefrom", "highest fd %d", fd);
170 /* Eat up 16 descriptors. */
171 for (i = 0; i < 16; i++)
174 if (fd != start + 16)
175 fail("open 16", "highest fd %d", fd);
178 /* Close half of them. */
182 fail("closefrom", "highest fd %d", fd);
185 /* Explicitly close descriptors 6 and 8 to create holes. */
186 if (close(6) < 0 || close(8) < 0)
190 /* Verify that close on 6 and 8 fails with EBADF. */
192 fail("close(6)", "did not fail");
194 fail_err("close(6)");
197 fail("close(8)", "did not fail");
199 fail_err("close(8)");
202 /* Close from 4 on. */
206 fail("closefrom", "highest fd %d", fd);
209 /* Allocate a small SHM region for IPC with our child. */
210 info = mmap(NULL, getpagesize(), PROT_READ | PROT_WRITE, MAP_ANON |
212 if (info == MAP_FAILED)
216 /* Fork a child process to test closefrom(0). */
225 cfail(info, "closefrom(0)", "highest fd %d", fd);
226 cok(info, "closefrom(0)");
231 fail(info->tag, "%s", info->message);
234 /* Fork a child process to test closefrom(-1). */
243 cfail(info, "closefrom(-1)", "highest fd %d", fd);
244 cok(info, "closefrom(-1)");
249 fail(info->tag, "%s", info->message);
252 /* Dup stdout to 6. */
257 fail("dup2", "highest fd %d", fd);
260 /* Do a closefrom() starting in a hole. */
264 fail("closefrom", "highest fd %d", fd);
267 /* Do a closefrom() beyond our highest open fd. */
271 fail("closefrom", "highest fd %d", fd);
274 /* Chew up another 8 fd */
275 for (i = 0; i < 8; i++)
280 /* close_range() a hole in the middle */
281 close_range(start + 3, start + 5, 0);
282 for (i = start + 3; i < start + 6; ++i) {
283 if (close(i) == 0 || errno != EBADF) {
289 fail("close_range", "failed to close at %d in %d - %d", i + 1,
290 start + 3, start + 6);
293 /* close_range from the middle of the hole */
294 close_range(start + 4, start + 6, 0);
295 if ((i = highest_fd()) != fd)
296 fail("close_range", "highest fd %d", i);
299 /* close_range to the end; effectively closefrom(2) */
300 close_range(start + 3, ~0L, 0);
301 if ((i = highest_fd()) != start + 2)
302 fail("close_range", "highest fd %d", i);
305 /* Now close the rest */
306 close_range(start, start + 4, 0);
309 fail("close_range", "highest fd %d", fd);
312 /* Fork a child process to test closefrom(0) twice. */
320 cok(info, "closefrom(0)");
325 fail(info->tag, "%s", info->message);
328 /* test CLOSE_RANGE_CLOEXEC */
329 for (i = 0; i < 8; i++)
333 if (close_range(start + 1, start + 4, CLOSE_RANGE_CLOEXEC) < 0)
334 fail_err("close_range(..., CLOSE_RANGE_CLOEXEC)");
335 flags = fcntl(start, F_GETFD);
337 fail_err("fcntl(.., F_GETFD)");
338 if ((flags & FD_CLOEXEC) != 0)
339 fail("close_range", "CLOSE_RANGE_CLOEXEC set close-on-exec "
340 "when it should not have on fd %d", start);
341 for (i = start + 1; i <= start + 4; i++) {
342 flags = fcntl(i, F_GETFD);
344 fail_err("fcntl(.., F_GETFD)");
345 if ((flags & FD_CLOEXEC) == 0)
346 fail("close_range", "CLOSE_RANGE_CLOEXEC did not set "
347 "close-on-exec on fd %d", i);
349 for (; i < start + 8; i++) {
350 flags = fcntl(i, F_GETFD);
352 fail_err("fcntl(.., F_GETFD)");
353 if ((flags & FD_CLOEXEC) != 0)
354 fail("close_range", "CLOSE_RANGE_CLOEXEC set close-on-exec "
355 "when it should not have on fd %d", i);
357 if (close_range(start, start + 8, 0) < 0)
358 fail_err("close_range");
359 ok("close_range(..., CLOSE_RANGE_CLOEXEC)");