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 <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 }