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 *
makemap(int v,int f)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);
ATF_TC_HEAD(minherit_copy,tc)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 
ATF_TC_BODY(minherit_copy,tc)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);
ATF_TC_HEAD(minherit_share,tc)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 
ATF_TC_BODY(minherit_share,tc)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
segv(int n)117 segv(int n) {
118           _exit(n);
119 }
120 
121 ATF_TC(minherit_none);
ATF_TC_HEAD(minherit_none,tc)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 
ATF_TC_BODY(minherit_none,tc)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);
ATF_TC_HEAD(minherit_zero,tc)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 
ATF_TC_BODY(minherit_zero,tc)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);
ATF_TC_HEAD(minherit_bad,tc)177 ATF_TC_HEAD(minherit_bad, tc)
178 {
179           atf_tc_set_md_var(tc, "descr",
180               "Test for bad minherit(2)");
181 }
182 
ATF_TC_BODY(minherit_bad,tc)183 ATF_TC_BODY(minherit_bad, tc)
184 {
185           (void)makemap(0, 666);
186 }
187 
ATF_TP_ADD_TCS(tp)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 }
201