tests/sys/kern/kern_descrip_test.c

   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
  49 static 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
  57 ATF_TC_WITHOUT_HEAD(dup2__simple);
  58 ATF_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
  71 ATF_TC(dup2__ebadf_when_2nd_arg_out_of_range);
  72 ATF_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
  77 ATF_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
  88 static void
  89 handler(int s __unused)
  90 {
  91         done++;
  92 }
  93
  94 static void
  95 openfiles2(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
 118 static void
 119 openfiles(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
 142 ATF_TC_WITH_CLEANUP(kern_maxfiles__increase);
 143 ATF_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
 151 ATF_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
 186 ATF_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
 204 ATF_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 }