kern_descrip_test.c revision 301805 
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: stable/10/tests/sys/kern/kern_descrip_test.c 301805 2016-06-10 18:07:35Z ngie $");
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
43static 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
51ATF_TC_WITHOUT_HEAD(dup2__simple);
52ATF_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
65ATF_TC(dup2__ebadf_when_2nd_arg_out_of_range);
66ATF_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
71ATF_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
82static void
83handler(int s __unused)
84{
85	done++;
86}
87
88static void
89openfiles2(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
107static void
108openfiles(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
127ATF_TC_WITH_CLEANUP(kern_maxfiles__increase);
128ATF_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
136ATF_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
167ATF_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
183ATF_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