tests/lib/libc/sys/t_minherit.c

   1 /* $NetBSD: t_minherit.c,v 1.1 2014/07/18 12:34:52 christos Exp $ */
   2
   3 /*-
   4  * Copyright (c) 2014 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Christos Zoulas
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 __RCSID("$NetBSD: t_minherit.c,v 1.1 2014/07/18 12:34:52 christos Exp $");
  33
  34 #include <sys/param.h>
  35 #include <sys/mman.h>
  36 #include <sys/sysctl.h>
  37 #include <sys/wait.h>
  38
  39 #include <errno.h>
  40 #include <fcntl.h>
  41 #include <stdlib.h>
  42 #include <string.h>
  43 #include <unistd.h>
  44
  45 #include <atf-c.h>
  46
  47 static long page;
  48
  49 static void *
  50 makemap(int v, int f) {
  51         void *map = mmap(NULL, page, PROT_READ|PROT_WRITE,
  52             MAP_SHARED|MAP_ANON, -1, 0);
  53         ATF_REQUIRE(map != MAP_FAILED);
  54         memset(map, v, page);
  55         if (f != 666)
  56                 ATF_REQUIRE(minherit(map, page, f) == 0);
  57         else
  58                 ATF_REQUIRE(minherit(map, page, f) == -1);
  59         return map;
  60 }
  61
  62 ATF_TC(minherit_copy);
  63 ATF_TC_HEAD(minherit_copy, tc)
  64 {
  65         atf_tc_set_md_var(tc, "descr",
  66             "Test for MAP_INHERIT_COPY from minherit(2)");
  67 }
  68
  69 ATF_TC_BODY(minherit_copy, tc)
  70 {
  71         void *map1 = makemap(1, MAP_INHERIT_COPY);
  72         void *map2 = makemap(1, MAP_INHERIT_COPY);
  73         switch (fork()) {
  74         default:
  75                 ATF_REQUIRE(wait(NULL) != -1);
  76                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
  77                 break;
  78         case -1:
  79                 ATF_REQUIRE(0);
  80                 break;
  81         case 0:
  82                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
  83                 memset(map1, 0, page);
  84                 exit(0);
  85         }
  86 }
  87
  88 ATF_TC(minherit_share);
  89 ATF_TC_HEAD(minherit_share, tc)
  90 {
  91         atf_tc_set_md_var(tc, "descr",
  92             "Test for MAP_INHERIT_SHARE from minherit(2)");
  93 }
  94
  95 ATF_TC_BODY(minherit_share, tc)
  96 {
  97         void *map1 = makemap(1, MAP_INHERIT_SHARE);
  98         void *map2 = makemap(1, MAP_INHERIT_SHARE);
  99
 100         switch (fork()) {
 101         default:
 102                 ATF_REQUIRE(wait(NULL) != -1);
 103                 memset(map2, 0, page);
 104                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
 105                 break;
 106         case -1:
 107                 ATF_REQUIRE(0);
 108                 break;
 109         case 0:
 110                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
 111                 memset(map1, 0, page);
 112                 exit(0);
 113         }
 114 }
 115
 116 static void
 117 segv(int n) {
 118         _exit(n);
 119 }
 120
 121 ATF_TC(minherit_none);
 122 ATF_TC_HEAD(minherit_none, tc)
 123 {
 124         atf_tc_set_md_var(tc, "descr",
 125             "Test for MAP_INHERIT_NONE from minherit(2)");
 126 }
 127
 128 ATF_TC_BODY(minherit_none, tc)
 129 {
 130         void *map1 = makemap(0, MAP_INHERIT_NONE);
 131         int status;
 132
 133         switch (fork()) {
 134         default:
 135                 ATF_REQUIRE(wait(&status) != -1);
 136                 ATF_REQUIRE(WEXITSTATUS(status) == SIGSEGV);
 137                 break;
 138         case -1:
 139                 ATF_REQUIRE(0);
 140                 break;
 141         case 0:
 142                 ATF_REQUIRE(signal(SIGSEGV, segv) != SIG_ERR);
 143                 memset(map1, 0, page);
 144                 exit(0);
 145         }
 146 }
 147
 148 ATF_TC(minherit_zero);
 149 ATF_TC_HEAD(minherit_zero, tc)
 150 {
 151         atf_tc_set_md_var(tc, "descr",
 152             "Test for MAP_INHERIT_ZERO from minherit(2)");
 153 }
 154
 155 ATF_TC_BODY(minherit_zero, tc)
 156 {
 157         void *map1 = makemap(1, MAP_INHERIT_ZERO);
 158         void *map2 = makemap(0, MAP_INHERIT_SHARE);
 159
 160         switch (fork()) {
 161         default:
 162                 ATF_REQUIRE(wait(NULL) != -1);
 163                 memset(map2, 1, page);
 164                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
 165                 break;
 166         case -1:
 167                 ATF_REQUIRE(0);
 168                 break;
 169         case 0:
 170                 ATF_REQUIRE(memcmp(map1, map2, page) == 0);
 171                 memset(map1, 2, page);
 172                 exit(0);
 173         }
 174 }
 175
 176 ATF_TC(minherit_bad);
 177 ATF_TC_HEAD(minherit_bad, tc)
 178 {
 179         atf_tc_set_md_var(tc, "descr",
 180             "Test for bad minherit(2)");
 181 }
 182
 183 ATF_TC_BODY(minherit_bad, tc)
 184 {
 185         (void)makemap(0, 666);
 186 }
 187
 188 ATF_TP_ADD_TCS(tp)
 189 {
 190         page = sysconf(_SC_PAGESIZE);
 191         ATF_REQUIRE(page >= 0);
 192
 193         ATF_TP_ADD_TC(tp, minherit_copy);
 194         ATF_TP_ADD_TC(tp, minherit_share);
 195         ATF_TP_ADD_TC(tp, minherit_none);
 196         ATF_TP_ADD_TC(tp, minherit_zero);
 197         ATF_TP_ADD_TC(tp, minherit_bad);
 198
 199         return atf_no_error();
 200 }