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 <sys/limits.h>
32#include <sys/resource.h>
33#include <sys/stat.h>
34#include <sys/sysctl.h>
35#include <sys/time.h>
36#include <sys/wait.h>
37
38#include <errno.h>
39#include <fcntl.h>
40#include <signal.h>
41#include <stdio.h>
42#include <stdlib.h>
43#include <string.h>
44#include <strings.h>
45#include <unistd.h>
46
47#include <atf-c.h>
48
49static volatile sig_atomic_t done;
50
51#define AFILE "afile"
52#define EXPANDBY 1000
53#define PARALLEL 4
54#define RENDEZVOUS "rendezvous"
55#define VALUE "value"
56
57ATF_TC_WITHOUT_HEAD(dup2__simple);
58ATF_TC_BODY(dup2__simple, tc)
59{
60	int fd1, fd2;
61	struct stat sb1, sb2;
62
63	ATF_REQUIRE((fd1 = open(AFILE, O_CREAT, 0644)) != -1);
64	fd2 = 27;
65	ATF_REQUIRE(dup2(fd1, fd2) != -1);
66	ATF_REQUIRE(fstat(fd1, &sb1) != -1);
67	ATF_REQUIRE(fstat(fd2, &sb2) != -1);
68	ATF_REQUIRE(bcmp(&sb1, &sb2, sizeof(sb1)) == 0);
69}
70
71ATF_TC(dup2__ebadf_when_2nd_arg_out_of_range);
72ATF_TC_HEAD(dup2__ebadf_when_2nd_arg_out_of_range, tc)
73{
74	atf_tc_set_md_var(tc, "descr", "Regression test for r234131");
75}
76
77ATF_TC_BODY(dup2__ebadf_when_2nd_arg_out_of_range, tc)
78{
79	int fd1, fd2, ret;
80
81	ATF_REQUIRE((fd1 = open(AFILE, O_CREAT, 0644)) != -1);
82	fd2 = INT_MAX;
83	ret = dup2(fd1, fd2);
84	ATF_CHECK_EQ(-1, ret);
85	ATF_CHECK_EQ(EBADF, errno);
86}
87
88static void
89handler(int s __unused)
90{
91	done++;
92}
93
94static void
95openfiles2(size_t n)
96{
97	size_t i;
98	int r;
99
100	errno = 0;
101	for (i = 0; i < n; i++) {
102		r = open(AFILE, O_RDONLY);
103		if (r < 0) {
104			fprintf(stderr, "open: %s\n", strerror(errno));
105			_exit(1);
106		}
107	}
108	kill(getppid(), SIGUSR1);
109
110	for (;;) {
111		if (access(RENDEZVOUS, R_OK) != 0)
112			break;
113		usleep(1000);
114	}
115	_exit(0);
116}
117
118static void
119openfiles(size_t n)
120{
121	int i, fd;
122
123	signal(SIGUSR1, handler);
124	ATF_REQUIRE((fd = open(AFILE, O_CREAT, 0644)) != -1);
125	close(fd);
126	ATF_REQUIRE((fd = open(RENDEZVOUS, O_CREAT, 0644)) != -1);
127	close(fd);
128	done = 0;
129	for (i = 0; i < PARALLEL; i++)
130		if (fork() == 0)
131			openfiles2(n / PARALLEL);
132	while (done != PARALLEL) {
133		usleep(1000);
134		ATF_REQUIRE_EQ_MSG(0, waitpid(-1, NULL, WNOHANG),
135			"a child exited unexpectedly");
136	}
137	unlink(RENDEZVOUS);
138	for (i = 0; i < PARALLEL; i++)
139		ATF_CHECK_MSG(wait(NULL) > 0, "wait: %s", strerror(errno));
140}
141
142ATF_TC_WITH_CLEANUP(kern_maxfiles__increase);
143ATF_TC_HEAD(kern_maxfiles__increase, tc)
144{
145	atf_tc_set_md_var(tc, "require.user", "root");
146	atf_tc_set_md_var(tc, "require.config", "allow_sysctl_side_effects");
147	atf_tc_set_md_var(tc, "descr",
148	    "Check kern.maxfiles expansion");
149}
150
151ATF_TC_BODY(kern_maxfiles__increase, tc)
152{
153	size_t oldlen;
154	int maxfiles, oldmaxfiles, current;
155	char buf[80];
156	struct rlimit rl;
157
158	oldlen = sizeof(maxfiles);
159	if (sysctlbyname("kern.maxfiles", &maxfiles, &oldlen, NULL, 0) == -1)
160		atf_tc_fail("getsysctlbyname(%s): %s", "kern.maxfiles",
161		    strerror(errno));
162	if (sysctlbyname("kern.openfiles", &current, &oldlen, NULL, 0) == -1)
163		atf_tc_fail("getsysctlbyname(%s): %s", "kern.openfiles",
164		    strerror(errno));
165
166	oldmaxfiles = maxfiles;
167
168	/* Store old kern.maxfiles in a symlink for cleanup */
169	snprintf(buf, sizeof(buf), "%d", oldmaxfiles);
170	if (symlink(buf, VALUE) == 1)
171		atf_tc_fail("symlink(%s, %s): %s", buf, VALUE,
172		    strerror(errno));
173
174	maxfiles += EXPANDBY;
175	if (sysctlbyname("kern.maxfiles", NULL, 0, &maxfiles, oldlen) == -1)
176		atf_tc_fail("getsysctlbyname(%s): %s", "kern.maxfiles",
177		    strerror(errno));
178
179	rl.rlim_cur = rl.rlim_max = maxfiles;
180	ATF_REQUIRE_EQ_MSG(0, setrlimit(RLIMIT_NOFILE, &rl),
181		"setrlimit(RLIMIT_NOFILE, %d): %s", maxfiles, strerror(errno));
182
183	openfiles(oldmaxfiles - current + EXPANDBY / 2);
184}
185
186ATF_TC_CLEANUP(kern_maxfiles__increase, tc)
187{
188	size_t oldlen;
189	int n, oldmaxfiles;
190	char buf[80];
191
192	if ((n = readlink(VALUE, buf, sizeof(buf))) > 0) {
193		buf[MIN((size_t)n, sizeof(buf) - 1)] = '\0';
194		if (sscanf(buf, "%d", &oldmaxfiles) == 1) {
195			oldlen = sizeof(oldmaxfiles);
196			(void) sysctlbyname("kern.maxfiles", NULL, 0,
197			    &oldmaxfiles, oldlen);
198		}
199	}
200	(void)unlink(VALUE);
201	(void)unlink(AFILE);
202}
203
204ATF_TP_ADD_TCS(tp)
205{
206
207	ATF_TP_ADD_TC(tp, dup2__simple);
208	ATF_TP_ADD_TC(tp, dup2__ebadf_when_2nd_arg_out_of_range);
209	ATF_TP_ADD_TC(tp, kern_maxfiles__increase);
210
211	return (atf_no_error());
212}
213