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
47static long page;
48
49static void *
50makemap(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
62ATF_TC(minherit_copy);
63ATF_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
69ATF_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
88ATF_TC(minherit_share);
89ATF_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
95ATF_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
116static void
117segv(int n) {
118	_exit(n);
119}
120
121ATF_TC(minherit_none);
122ATF_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
128ATF_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
148ATF_TC(minherit_zero);
149ATF_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
155ATF_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
176ATF_TC(minherit_bad);
177ATF_TC_HEAD(minherit_bad, tc)
178{
179	atf_tc_set_md_var(tc, "descr",
180	    "Test for bad minherit(2)");
181}
182
183ATF_TC_BODY(minherit_bad, tc)
184{
185	(void)makemap(0, 666);
186}
187
188ATF_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