1 /*
2 * This file and its contents are supplied under the terms of the
3 * Common Development and Distribution License ("CDDL"), version 1.0.
4 * You may only use this file in accordance with the terms of version
5 * 1.0 of the CDDL.
6 *
7 * A full copy of the text of the CDDL should have accompanied this
8 * source. A copy of the CDDL is also available via the Internet at
9 * http://www.illumos.org/license/CDDL.
10 */
11
12 /*
13 * Copyright (c) 2015, Joyent, Inc.
14 */
15
16 /*
17 * Verify that using MC_INHERIT_ZERO works just fine when applied to an entire
18 * region across multiple children.
19 */
20
21 #include <sys/types.h>
22 #include <unistd.h>
23 #include <assert.h>
24 #include <sys/mman.h>
25 #include <string.h>
26 #include <errno.h>
27 #include <stdlib.h>
28 #include <wait.h>
29
30 static int nchild = 5;
31
32 int
main(void)33 main(void)
34 {
35 void *buf;
36 pid_t child;
37 int ret, i;
38 siginfo_t info;
39 uint8_t *ubuf;
40 size_t mapsz = sysconf(_SC_PAGESIZE) * 2;
41
42 buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE,
43 MAP_PRIVATE | MAP_ANON, -1, 0);
44 assert(buf != MAP_FAILED);
45
46 ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
47 assert(ret == 0);
48
49 again:
50 memset(buf, 'a' + nchild, mapsz);
51 child = fork();
52 if (child == 0) {
53 nchild--;
54 for (i = 0, ubuf = buf; i < mapsz; i++)
55 assert(ubuf[i] == 0);
56 if (nchild != 0)
57 goto again;
58 exit(0);
59 }
60 assert(child != -1);
61
62 do {
63 ret = waitid(P_PID, child, &info, WEXITED);
64 } while (ret == -1 && errno == EINTR);
65 assert(ret == 0);
66 assert(info.si_pid == child);
67 assert(info.si_status == 0);
68
69 for (i = 0, ubuf = buf; i < mapsz; i++)
70 assert(ubuf[i] == 'a' + nchild);
71
72 return (0);
73 }
74