9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * This file and its contents are supplied under the terms of the
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Common Development and Distribution License ("CDDL"), version 1.0.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * You may only use this file in accordance with the terms of version
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * 1.0 of the CDDL.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi *
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * A full copy of the text of the CDDL should have accompanied this
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * source. A copy of the CDDL is also available via the Internet at
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * http://www.illumos.org/license/CDDL.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Copyright (c) 2015, Joyent, Inc.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi/*
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * Verify that using MC_INHERIT_ZERO works just fine when applied to an entire
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi * region that is mlocked.
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi */
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <sys/types.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <unistd.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <assert.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <sys/mman.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <string.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <errno.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <stdlib.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi#include <wait.h>
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchiint
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchimain(void)
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi{
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi void *buf;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi pid_t child;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi int ret, i;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi siginfo_t info;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi uint8_t *ubuf;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi size_t mapsz = sysconf(_SC_PAGESIZE) * 2;
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE,
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi MAP_PRIVATE | MAP_ANON, -1, 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(buf != MAP_FAILED);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi memset(buf, 'a', mapsz);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi ret = mlock(buf, mapsz);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ret == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ret == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi child = fork();
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi if (child == 0) {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi for (i = 0, ubuf = buf; i < mapsz; i++)
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ubuf[i] == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi exit(0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi }
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(child != -1);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi do {
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi ret = waitid(P_PID, child, &info, WEXITED);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi } while (ret == -1 && errno == EINTR);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ret == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(info.si_pid == child);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(info.si_status == 0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi for (i = 0, ubuf = buf; i < mapsz; i++)
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi assert(ubuf[i] == 'a');
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi return (0);
9d12795f87b63c2e39e87bff369182edd34677d3Robert Mustacchi}