xref: /freebsd/tests/sys/kern/kern_descrip_test.c (revision af6a5351a1fdb1130f18be6c782c4d48916eb971)
1 /*-
2  * Copyright (c) 2014 EMC Corp.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  */
26 
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29 
30 #include <sys/param.h>
31 #include <errno.h>
32 #include <fcntl.h>
33 #include <signal.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <strings.h>
37 #include <sys/limits.h>
38 #include <sys/stat.h>
39 #include <sys/sysctl.h>
40 #include <unistd.h>
41 #include <atf-c.h>
42 
43 static volatile sig_atomic_t done;
44 
45 #define AFILE "afile"
46 #define EXPANDBY 1000
47 #define PARALLEL 4
48 #define RENDEZVOUS "rendezvous"
49 #define VALUE "value"
50 
51 ATF_TC_WITHOUT_HEAD(dup2__simple);
52 ATF_TC_BODY(dup2__simple, tc)
53 {
54 	int fd1, fd2;
55 	struct stat sb1, sb2;
56 
57 	ATF_REQUIRE((fd1 = open(AFILE, O_CREAT, 0644)) != -1);
58 	fd2 = 27;
59 	ATF_REQUIRE(dup2(fd1, fd2) != -1);
60 	ATF_REQUIRE(fstat(fd1, &sb1) != -1);
61 	ATF_REQUIRE(fstat(fd2, &sb2) != -1);
62 	ATF_REQUIRE(bcmp(&sb1, &sb2, sizeof(sb1)) == 0);
63 }
64 
65 ATF_TC(dup2__ebadf_when_2nd_arg_out_of_range);
66 ATF_TC_HEAD(dup2__ebadf_when_2nd_arg_out_of_range, tc)
67 {
68 	atf_tc_set_md_var(tc, "descr", "Regression test for r234131");
69 }
70 
71 ATF_TC_BODY(dup2__ebadf_when_2nd_arg_out_of_range, tc)
72 {
73 	int fd1, fd2, ret;
74 
75 	ATF_REQUIRE((fd1 = open(AFILE, O_CREAT, 0644)) != -1);
76 	fd2 = INT_MAX;
77 	ret = dup2(fd1, fd2);
78 	ATF_CHECK_EQ(-1, ret);
79 	ATF_CHECK_EQ(EBADF, errno);
80 }
81 
82 static void
83 handler(int s __unused)
84 {
85 	done++;
86 }
87 
88 static void
89 openfiles2(size_t n)
90 {
91 	size_t i;
92 	int r;
93 
94 	errno = 0;
95 	for (i = 0; i < n; i++)
96 		ATF_REQUIRE((r = open(AFILE, O_RDONLY)) != -1);
97 	kill(getppid(), SIGUSR1);
98 
99 	for (;;) {
100 		if (access(RENDEZVOUS, R_OK) != 0)
101 			break;
102 		usleep(1000);
103 	}
104 	_exit(0);
105 }
106 
107 static void
108 openfiles(size_t n)
109 {
110 	int i, fd;
111 
112 	signal(SIGUSR1, handler);
113 	ATF_REQUIRE((fd = open(AFILE, O_CREAT, 0644)) != -1);
114 	close(fd);
115 	ATF_REQUIRE((fd = open(RENDEZVOUS, O_CREAT, 0644)) != -1);
116 	close(fd);
117 	done = 0;
118 	for (i = 0; i < PARALLEL; i++)
119 		if (fork() == 0)
120 			openfiles2(n / PARALLEL);
121 	while (done != PARALLEL)
122 		usleep(1000);
123 	unlink(RENDEZVOUS);
124 	usleep(40000);
125 }
126 
127 ATF_TC_WITH_CLEANUP(kern_maxfiles__increase);
128 ATF_TC_HEAD(kern_maxfiles__increase, tc)
129 {
130 	atf_tc_set_md_var(tc, "require.user", "root");
131 	atf_tc_set_md_var(tc, "require.config", "allow_sysctl_side_effects");
132 	atf_tc_set_md_var(tc, "descr",
133 	    "Check kern.maxfiles expansion");
134 }
135 
136 ATF_TC_BODY(kern_maxfiles__increase, tc)
137 {
138 	size_t oldlen;
139 	int maxfiles, oldmaxfiles, current;
140 	char buf[80];
141 
142 	oldlen = sizeof(maxfiles);
143 	if (sysctlbyname("kern.maxfiles", &maxfiles, &oldlen, NULL, 0) == -1)
144 		atf_tc_fail("getsysctlbyname(%s): %s", "kern.maxfiles",
145 		    strerror(errno));
146 	if (sysctlbyname("kern.openfiles", &current, &oldlen, NULL, 0) == -1)
147 		atf_tc_fail("getsysctlbyname(%s): %s", "kern.openfiles",
148 		    strerror(errno));
149 
150 	oldmaxfiles = maxfiles;
151 
152 	/* Store old kern.maxfiles in a symlink for cleanup */
153 	snprintf(buf, sizeof(buf), "%d", oldmaxfiles);
154 	if (symlink(buf, VALUE) == 1)
155 		atf_tc_fail("symlink(%s, %s): %s", buf, VALUE,
156 		    strerror(errno));
157 
158 	maxfiles += EXPANDBY;
159 	if (sysctlbyname("kern.maxfiles", NULL, 0, &maxfiles, oldlen) == -1)
160 		atf_tc_fail("getsysctlbyname(%s): %s", "kern.maxfiles",
161 		    strerror(errno));
162 
163 	openfiles(oldmaxfiles - current + 1);
164 	(void)unlink(VALUE);
165 }
166 
167 ATF_TC_CLEANUP(kern_maxfiles__increase, tc)
168 {
169 	size_t oldlen;
170 	int n, oldmaxfiles;
171 	char buf[80];
172 
173 	if ((n = readlink(VALUE, buf, sizeof(buf))) > 0) {
174 		buf[MIN((size_t)n, sizeof(buf) - 1)] = '\0';
175 		if (sscanf(buf, "%d", &oldmaxfiles) == 1) {
176 			oldlen = sizeof(oldmaxfiles);
177 			(void) sysctlbyname("kern.maxfiles", NULL, 0,
178 			    &oldmaxfiles, oldlen);
179 		}
180 	}
181 }
182 
183 ATF_TP_ADD_TCS(tp)
184 {
185 
186 	ATF_TP_ADD_TC(tp, dup2__simple);
187 	ATF_TP_ADD_TC(tp, dup2__ebadf_when_2nd_arg_out_of_range);
188 	ATF_TP_ADD_TC(tp, kern_maxfiles__increase);
189 
190 	return (atf_no_error());
191 }
192