13cedbec3SEnji Cooper /*-
23cedbec3SEnji Cooper * Copyright (c) 2009 Hudson River Trading LLC
33cedbec3SEnji Cooper * Written by: John H. Baldwin <jhb@FreeBSD.org>
43cedbec3SEnji Cooper * All rights reserved.
53cedbec3SEnji Cooper *
63cedbec3SEnji Cooper * Redistribution and use in source and binary forms, with or without
73cedbec3SEnji Cooper * modification, are permitted provided that the following conditions
83cedbec3SEnji Cooper * are met:
93cedbec3SEnji Cooper * 1. Redistributions of source code must retain the above copyright
103cedbec3SEnji Cooper * notice, this list of conditions and the following disclaimer.
113cedbec3SEnji Cooper * 2. Redistributions in binary form must reproduce the above copyright
123cedbec3SEnji Cooper * notice, this list of conditions and the following disclaimer in the
133cedbec3SEnji Cooper * documentation and/or other materials provided with the distribution.
143cedbec3SEnji Cooper *
153cedbec3SEnji Cooper * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
163cedbec3SEnji Cooper * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
173cedbec3SEnji Cooper * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
183cedbec3SEnji Cooper * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
193cedbec3SEnji Cooper * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
203cedbec3SEnji Cooper * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
213cedbec3SEnji Cooper * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
223cedbec3SEnji Cooper * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
233cedbec3SEnji Cooper * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
243cedbec3SEnji Cooper * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
253cedbec3SEnji Cooper * SUCH DAMAGE.
263cedbec3SEnji Cooper */
273cedbec3SEnji Cooper
283cedbec3SEnji Cooper #include <sys/cdefs.h>
293cedbec3SEnji Cooper /*
303cedbec3SEnji Cooper * Regression tests for the closefrom(2) system call.
313cedbec3SEnji Cooper */
323cedbec3SEnji Cooper
333cedbec3SEnji Cooper #include <sys/param.h>
343cedbec3SEnji Cooper #include <sys/mman.h>
353cedbec3SEnji Cooper #include <sys/user.h>
363cedbec3SEnji Cooper #include <sys/wait.h>
373cedbec3SEnji Cooper #include <errno.h>
383cedbec3SEnji Cooper #include <fcntl.h>
393cedbec3SEnji Cooper #include <libutil.h>
403cedbec3SEnji Cooper #include <paths.h>
413cedbec3SEnji Cooper #include <stdarg.h>
423cedbec3SEnji Cooper #include <stdio.h>
433cedbec3SEnji Cooper #include <stdlib.h>
443cedbec3SEnji Cooper #include <string.h>
453cedbec3SEnji Cooper #include <unistd.h>
463cedbec3SEnji Cooper
473cedbec3SEnji Cooper struct shared_info {
483cedbec3SEnji Cooper int failed;
493cedbec3SEnji Cooper char tag[64];
503cedbec3SEnji Cooper char message[0];
513cedbec3SEnji Cooper };
523cedbec3SEnji Cooper
533cedbec3SEnji Cooper static int test = 1;
543cedbec3SEnji Cooper
553cedbec3SEnji Cooper static void
ok(const char * descr)563cedbec3SEnji Cooper ok(const char *descr)
573cedbec3SEnji Cooper {
583cedbec3SEnji Cooper
593cedbec3SEnji Cooper printf("ok %d - %s\n", test, descr);
603cedbec3SEnji Cooper test++;
613cedbec3SEnji Cooper }
623cedbec3SEnji Cooper
633cedbec3SEnji Cooper static void
fail(const char * descr,const char * fmt,...)643cedbec3SEnji Cooper fail(const char *descr, const char *fmt, ...)
653cedbec3SEnji Cooper {
663cedbec3SEnji Cooper va_list ap;
673cedbec3SEnji Cooper
683cedbec3SEnji Cooper printf("not ok %d - %s", test, descr);
693cedbec3SEnji Cooper test++;
703cedbec3SEnji Cooper if (fmt) {
713cedbec3SEnji Cooper va_start(ap, fmt);
723cedbec3SEnji Cooper printf(" # ");
733cedbec3SEnji Cooper vprintf(fmt, ap);
743cedbec3SEnji Cooper va_end(ap);
753cedbec3SEnji Cooper }
763cedbec3SEnji Cooper printf("\n");
773cedbec3SEnji Cooper exit(1);
783cedbec3SEnji Cooper }
793cedbec3SEnji Cooper
803cedbec3SEnji Cooper #define fail_err(descr) fail((descr), "%s", strerror(errno))
813cedbec3SEnji Cooper
823cedbec3SEnji Cooper static void
cok(struct shared_info * info,const char * descr)833cedbec3SEnji Cooper cok(struct shared_info *info, const char *descr)
843cedbec3SEnji Cooper {
853cedbec3SEnji Cooper
863cedbec3SEnji Cooper info->failed = 0;
873cedbec3SEnji Cooper strlcpy(info->tag, descr, sizeof(info->tag));
883cedbec3SEnji Cooper exit(0);
893cedbec3SEnji Cooper }
903cedbec3SEnji Cooper
913cedbec3SEnji Cooper static void
cfail(struct shared_info * info,const char * descr,const char * fmt,...)923cedbec3SEnji Cooper cfail(struct shared_info *info, const char *descr, const char *fmt, ...)
933cedbec3SEnji Cooper {
943cedbec3SEnji Cooper va_list ap;
953cedbec3SEnji Cooper
963cedbec3SEnji Cooper info->failed = 1;
973cedbec3SEnji Cooper strlcpy(info->tag, descr, sizeof(info->tag));
983cedbec3SEnji Cooper if (fmt) {
993cedbec3SEnji Cooper va_start(ap, fmt);
1003cedbec3SEnji Cooper vsprintf(info->message, fmt, ap);
1013cedbec3SEnji Cooper va_end(ap);
1023cedbec3SEnji Cooper }
1033cedbec3SEnji Cooper exit(0);
1043cedbec3SEnji Cooper }
1053cedbec3SEnji Cooper
1063cedbec3SEnji Cooper #define cfail_err(info, descr) cfail((info), (descr), "%s", strerror(errno))
1073cedbec3SEnji Cooper
1083cedbec3SEnji Cooper /*
1093cedbec3SEnji Cooper * Use kinfo_getfile() to fetch the list of file descriptors and figure out
1103cedbec3SEnji Cooper * the highest open file descriptor.
1113cedbec3SEnji Cooper */
1123cedbec3SEnji Cooper static int
highest_fd(void)1133cedbec3SEnji Cooper highest_fd(void)
1143cedbec3SEnji Cooper {
1153cedbec3SEnji Cooper struct kinfo_file *kif;
1163cedbec3SEnji Cooper int cnt, i, highest;
1173cedbec3SEnji Cooper
1183cedbec3SEnji Cooper kif = kinfo_getfile(getpid(), &cnt);
1193cedbec3SEnji Cooper if (kif == NULL)
1203cedbec3SEnji Cooper fail_err("kinfo_getfile");
1213cedbec3SEnji Cooper highest = INT_MIN;
1223cedbec3SEnji Cooper for (i = 0; i < cnt; i++)
1233cedbec3SEnji Cooper if (kif[i].kf_fd > highest)
1243cedbec3SEnji Cooper highest = kif[i].kf_fd;
1253cedbec3SEnji Cooper free(kif);
1263cedbec3SEnji Cooper return (highest);
1273cedbec3SEnji Cooper }
1283cedbec3SEnji Cooper
1293cedbec3SEnji Cooper static int
devnull(void)1303cedbec3SEnji Cooper devnull(void)
1313cedbec3SEnji Cooper {
1323cedbec3SEnji Cooper int fd;
1333cedbec3SEnji Cooper
1343cedbec3SEnji Cooper fd = open(_PATH_DEVNULL, O_RDONLY);
1353cedbec3SEnji Cooper if (fd < 0)
1363cedbec3SEnji Cooper fail_err("open(\" "_PATH_DEVNULL" \")");
1373cedbec3SEnji Cooper return (fd);
1383cedbec3SEnji Cooper }
1393cedbec3SEnji Cooper
1403cedbec3SEnji Cooper int
main(void)1413cedbec3SEnji Cooper main(void)
1423cedbec3SEnji Cooper {
1433cedbec3SEnji Cooper struct shared_info *info;
1443cedbec3SEnji Cooper pid_t pid;
145f3f3e3c4SMateusz Guzik int fd, flags, i, start;
1463cedbec3SEnji Cooper
1473a01dcc9SMark Johnston printf("1..21\n");
1483cedbec3SEnji Cooper
149664812f7SYen-Cheng Chang /* We'd better start up with fd's 0, 1, and 2 open. */
1503cedbec3SEnji Cooper start = devnull();
1513cedbec3SEnji Cooper if (start == -1)
1523cedbec3SEnji Cooper fail("open", "bad descriptor %d", start);
1533cedbec3SEnji Cooper ok("open");
1543cedbec3SEnji Cooper
1553cedbec3SEnji Cooper /* Make sure highest_fd() works. */
1563cedbec3SEnji Cooper fd = highest_fd();
1573cedbec3SEnji Cooper if (start != fd)
1583cedbec3SEnji Cooper fail("highest_fd", "bad descriptor %d != %d", start, fd);
1593cedbec3SEnji Cooper ok("highest_fd");
1603cedbec3SEnji Cooper
1613cedbec3SEnji Cooper /* Try to use closefrom() for just closing fd 3. */
1623cedbec3SEnji Cooper closefrom(start + 1);
1633cedbec3SEnji Cooper fd = highest_fd();
1643cedbec3SEnji Cooper if (fd != start)
1653cedbec3SEnji Cooper fail("closefrom", "highest fd %d", fd);
1663cedbec3SEnji Cooper ok("closefrom");
1673cedbec3SEnji Cooper
1683cedbec3SEnji Cooper /* Eat up 16 descriptors. */
1693cedbec3SEnji Cooper for (i = 0; i < 16; i++)
1703cedbec3SEnji Cooper (void)devnull();
1713cedbec3SEnji Cooper fd = highest_fd();
1723cedbec3SEnji Cooper if (fd != start + 16)
1733cedbec3SEnji Cooper fail("open 16", "highest fd %d", fd);
1743cedbec3SEnji Cooper ok("open 16");
1753cedbec3SEnji Cooper
1763cedbec3SEnji Cooper /* Close half of them. */
1773cedbec3SEnji Cooper closefrom(11);
1783cedbec3SEnji Cooper fd = highest_fd();
1793cedbec3SEnji Cooper if (fd != 10)
1803cedbec3SEnji Cooper fail("closefrom", "highest fd %d", fd);
1813cedbec3SEnji Cooper ok("closefrom");
1823cedbec3SEnji Cooper
1833cedbec3SEnji Cooper /* Explicitly close descriptors 6 and 8 to create holes. */
1843cedbec3SEnji Cooper if (close(6) < 0 || close(8) < 0)
1853cedbec3SEnji Cooper fail_err("close2 ");
1863cedbec3SEnji Cooper ok("close 2");
1873cedbec3SEnji Cooper
1883cedbec3SEnji Cooper /* Verify that close on 6 and 8 fails with EBADF. */
1893cedbec3SEnji Cooper if (close(6) == 0)
1903cedbec3SEnji Cooper fail("close(6)", "did not fail");
1913cedbec3SEnji Cooper if (errno != EBADF)
1923cedbec3SEnji Cooper fail_err("close(6)");
1933cedbec3SEnji Cooper ok("close(6)");
1943cedbec3SEnji Cooper if (close(8) == 0)
1953cedbec3SEnji Cooper fail("close(8)", "did not fail");
1963cedbec3SEnji Cooper if (errno != EBADF)
1973cedbec3SEnji Cooper fail_err("close(8)");
1983cedbec3SEnji Cooper ok("close(8)");
1993cedbec3SEnji Cooper
2003cedbec3SEnji Cooper /* Close from 4 on. */
2013cedbec3SEnji Cooper closefrom(4);
2023cedbec3SEnji Cooper fd = highest_fd();
2033cedbec3SEnji Cooper if (fd != 3)
2043cedbec3SEnji Cooper fail("closefrom", "highest fd %d", fd);
2053cedbec3SEnji Cooper ok("closefrom");
2063cedbec3SEnji Cooper
2073cedbec3SEnji Cooper /* Allocate a small SHM region for IPC with our child. */
2083cedbec3SEnji Cooper info = mmap(NULL, getpagesize(), PROT_READ | PROT_WRITE, MAP_ANON |
2093cedbec3SEnji Cooper MAP_SHARED, -1, 0);
2103cedbec3SEnji Cooper if (info == MAP_FAILED)
2113cedbec3SEnji Cooper fail_err("mmap");
2123cedbec3SEnji Cooper ok("mmap");
2133cedbec3SEnji Cooper
2143cedbec3SEnji Cooper /* Fork a child process to test closefrom(0). */
2153cedbec3SEnji Cooper pid = fork();
2163cedbec3SEnji Cooper if (pid < 0)
2173cedbec3SEnji Cooper fail_err("fork");
2183cedbec3SEnji Cooper if (pid == 0) {
2193cedbec3SEnji Cooper /* Child. */
2203cedbec3SEnji Cooper closefrom(0);
2213cedbec3SEnji Cooper fd = highest_fd();
2223cedbec3SEnji Cooper if (fd >= 0)
2233cedbec3SEnji Cooper cfail(info, "closefrom(0)", "highest fd %d", fd);
2243cedbec3SEnji Cooper cok(info, "closefrom(0)");
2253cedbec3SEnji Cooper }
2263cedbec3SEnji Cooper if (wait(NULL) < 0)
2273cedbec3SEnji Cooper fail_err("wait");
2283cedbec3SEnji Cooper if (info->failed)
2293cedbec3SEnji Cooper fail(info->tag, "%s", info->message);
2303cedbec3SEnji Cooper ok(info->tag);
2313cedbec3SEnji Cooper
2323cedbec3SEnji Cooper /* Fork a child process to test closefrom(-1). */
2333cedbec3SEnji Cooper pid = fork();
2343cedbec3SEnji Cooper if (pid < 0)
2353cedbec3SEnji Cooper fail_err("fork");
2363cedbec3SEnji Cooper if (pid == 0) {
2373cedbec3SEnji Cooper /* Child. */
2383cedbec3SEnji Cooper closefrom(-1);
2393cedbec3SEnji Cooper fd = highest_fd();
2403cedbec3SEnji Cooper if (fd >= 0)
2413cedbec3SEnji Cooper cfail(info, "closefrom(-1)", "highest fd %d", fd);
2423cedbec3SEnji Cooper cok(info, "closefrom(-1)");
2433cedbec3SEnji Cooper }
2443cedbec3SEnji Cooper if (wait(NULL) < 0)
2453cedbec3SEnji Cooper fail_err("wait");
2463cedbec3SEnji Cooper if (info->failed)
2473cedbec3SEnji Cooper fail(info->tag, "%s", info->message);
2483cedbec3SEnji Cooper ok(info->tag);
2493cedbec3SEnji Cooper
2503cedbec3SEnji Cooper /* Dup stdout to 6. */
2513cedbec3SEnji Cooper if (dup2(1, 6) < 0)
2523cedbec3SEnji Cooper fail_err("dup2");
2533cedbec3SEnji Cooper fd = highest_fd();
2543cedbec3SEnji Cooper if (fd != 6)
2553cedbec3SEnji Cooper fail("dup2", "highest fd %d", fd);
2563cedbec3SEnji Cooper ok("dup2");
2573cedbec3SEnji Cooper
2583cedbec3SEnji Cooper /* Do a closefrom() starting in a hole. */
2593cedbec3SEnji Cooper closefrom(4);
2603cedbec3SEnji Cooper fd = highest_fd();
2613cedbec3SEnji Cooper if (fd != 3)
2623cedbec3SEnji Cooper fail("closefrom", "highest fd %d", fd);
2633cedbec3SEnji Cooper ok("closefrom");
2643cedbec3SEnji Cooper
2653cedbec3SEnji Cooper /* Do a closefrom() beyond our highest open fd. */
2663cedbec3SEnji Cooper closefrom(32);
2673cedbec3SEnji Cooper fd = highest_fd();
2683cedbec3SEnji Cooper if (fd != 3)
2693cedbec3SEnji Cooper fail("closefrom", "highest fd %d", fd);
2703cedbec3SEnji Cooper ok("closefrom");
2713cedbec3SEnji Cooper
272472ced39SKyle Evans /* Chew up another 8 fd */
273472ced39SKyle Evans for (i = 0; i < 8; i++)
274472ced39SKyle Evans (void)devnull();
275472ced39SKyle Evans fd = highest_fd();
276472ced39SKyle Evans start = fd - 7;
277472ced39SKyle Evans
278472ced39SKyle Evans /* close_range() a hole in the middle */
279472ced39SKyle Evans close_range(start + 3, start + 5, 0);
280472ced39SKyle Evans for (i = start + 3; i < start + 6; ++i) {
281472ced39SKyle Evans if (close(i) == 0 || errno != EBADF) {
282472ced39SKyle Evans --i;
283472ced39SKyle Evans break;
284472ced39SKyle Evans }
285472ced39SKyle Evans }
286472ced39SKyle Evans if (i != start + 6)
287472ced39SKyle Evans fail("close_range", "failed to close at %d in %d - %d", i + 1,
288472ced39SKyle Evans start + 3, start + 6);
289472ced39SKyle Evans ok("close_range");
290472ced39SKyle Evans
291472ced39SKyle Evans /* close_range from the middle of the hole */
292472ced39SKyle Evans close_range(start + 4, start + 6, 0);
293472ced39SKyle Evans if ((i = highest_fd()) != fd)
294472ced39SKyle Evans fail("close_range", "highest fd %d", i);
295472ced39SKyle Evans ok("close_range");
296472ced39SKyle Evans
297472ced39SKyle Evans /* close_range to the end; effectively closefrom(2) */
298472ced39SKyle Evans close_range(start + 3, ~0L, 0);
299472ced39SKyle Evans if ((i = highest_fd()) != start + 2)
300472ced39SKyle Evans fail("close_range", "highest fd %d", i);
301472ced39SKyle Evans ok("close_range");
302472ced39SKyle Evans
303472ced39SKyle Evans /* Now close the rest */
304472ced39SKyle Evans close_range(start, start + 4, 0);
305472ced39SKyle Evans fd = highest_fd();
306472ced39SKyle Evans if (fd != 3)
307472ced39SKyle Evans fail("close_range", "highest fd %d", fd);
308472ced39SKyle Evans ok("close_range");
309472ced39SKyle Evans
310605c4cdaSKyle Evans /* Fork a child process to test closefrom(0) twice. */
311605c4cdaSKyle Evans pid = fork();
312605c4cdaSKyle Evans if (pid < 0)
313605c4cdaSKyle Evans fail_err("fork");
314605c4cdaSKyle Evans if (pid == 0) {
315605c4cdaSKyle Evans /* Child. */
316605c4cdaSKyle Evans closefrom(0);
317605c4cdaSKyle Evans closefrom(0);
318605c4cdaSKyle Evans cok(info, "closefrom(0)");
319605c4cdaSKyle Evans }
320605c4cdaSKyle Evans if (wait(NULL) < 0)
321605c4cdaSKyle Evans fail_err("wait");
322605c4cdaSKyle Evans if (info->failed)
323605c4cdaSKyle Evans fail(info->tag, "%s", info->message);
324605c4cdaSKyle Evans ok(info->tag);
325605c4cdaSKyle Evans
326f3f3e3c4SMateusz Guzik /* test CLOSE_RANGE_CLOEXEC */
327f3f3e3c4SMateusz Guzik for (i = 0; i < 8; i++)
328f3f3e3c4SMateusz Guzik (void)devnull();
329f3f3e3c4SMateusz Guzik fd = highest_fd();
330f3f3e3c4SMateusz Guzik start = fd - 8;
331f3f3e3c4SMateusz Guzik if (close_range(start + 1, start + 4, CLOSE_RANGE_CLOEXEC) < 0)
332f3f3e3c4SMateusz Guzik fail_err("close_range(..., CLOSE_RANGE_CLOEXEC)");
333f3f3e3c4SMateusz Guzik flags = fcntl(start, F_GETFD);
334f3f3e3c4SMateusz Guzik if (flags < 0)
335f3f3e3c4SMateusz Guzik fail_err("fcntl(.., F_GETFD)");
336f3f3e3c4SMateusz Guzik if ((flags & FD_CLOEXEC) != 0)
337f3f3e3c4SMateusz Guzik fail("close_range", "CLOSE_RANGE_CLOEXEC set close-on-exec "
338f3f3e3c4SMateusz Guzik "when it should not have on fd %d", start);
339f3f3e3c4SMateusz Guzik for (i = start + 1; i <= start + 4; i++) {
340f3f3e3c4SMateusz Guzik flags = fcntl(i, F_GETFD);
341f3f3e3c4SMateusz Guzik if (flags < 0)
342f3f3e3c4SMateusz Guzik fail_err("fcntl(.., F_GETFD)");
343f3f3e3c4SMateusz Guzik if ((flags & FD_CLOEXEC) == 0)
344f3f3e3c4SMateusz Guzik fail("close_range", "CLOSE_RANGE_CLOEXEC did not set "
345f3f3e3c4SMateusz Guzik "close-on-exec on fd %d", i);
346f3f3e3c4SMateusz Guzik }
347f3f3e3c4SMateusz Guzik for (; i < start + 8; i++) {
348f3f3e3c4SMateusz Guzik flags = fcntl(i, F_GETFD);
349f3f3e3c4SMateusz Guzik if (flags < 0)
350f3f3e3c4SMateusz Guzik fail_err("fcntl(.., F_GETFD)");
351f3f3e3c4SMateusz Guzik if ((flags & FD_CLOEXEC) != 0)
352f3f3e3c4SMateusz Guzik fail("close_range", "CLOSE_RANGE_CLOEXEC set close-on-exec "
353f3f3e3c4SMateusz Guzik "when it should not have on fd %d", i);
354f3f3e3c4SMateusz Guzik }
355f3f3e3c4SMateusz Guzik if (close_range(start, start + 8, 0) < 0)
356f3f3e3c4SMateusz Guzik fail_err("close_range");
357f3f3e3c4SMateusz Guzik ok("close_range(..., CLOSE_RANGE_CLOEXEC)");
358f3f3e3c4SMateusz Guzik
3593cedbec3SEnji Cooper return (0);
3603cedbec3SEnji Cooper }
361