/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright (c) 2015, Joyent, Inc.
 */

/*
 * Verify that using MC_INHERIT_ZERO doesn't work on mappings that aren't
 * anonymous private mappings.
 */

#include <sys/types.h>
#include <unistd.h>
#include <assert.h>
#include <sys/mman.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include <wait.h>
#include <sys/stat.h>
#include <fcntl.h>

int
main(void)
{
	void *buf;
	int ret, fd;
	char *template = "/tmp/inz_inval.XXXXXX";
	char *tmpfile;
	size_t mapsz = sysconf(_SC_PAGESIZE) * 2;
	caddr_t bad = (caddr_t)(uintptr_t)23;

	buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE,
	    MAP_PRIVATE | MAP_ANON, -1, 0);
	assert(buf != MAP_FAILED);

	/* Bad arguments to memcntl */
	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, bad, 0, 0);
	assert(ret == -1);
	assert(errno == EINVAL);

	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, PROT_READ, 0);
	assert(ret == -1);
	assert(errno == EINVAL);

	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, bad, PROT_READ | PRIVATE, 0);
	assert(ret == -1);
	assert(errno == EINVAL);

	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 1);
	assert(ret == -1);
	assert(errno == EINVAL);

	ret = munmap(buf, mapsz);
	assert(ret == 0);

	/* Mapping non-existant region */
	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
	assert(ret == -1);
	assert(errno == ENOMEM);

	/* Map anon MAP_SHARED */
	buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE,
	    MAP_SHARED | MAP_ANON, -1, 0);
	assert(buf != MAP_FAILED);
	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
	assert(ret == -1);
	assert(errno == EINVAL);
	ret = munmap(buf, mapsz);
	assert(ret == 0);

	/* Grab a temp file and get it to be the right size */
	tmpfile = strdup(template);
	assert(tmpfile != NULL);
	fd = mkstemp(tmpfile);
	assert(fd >= 0);
	ret = ftruncate(fd, mapsz);
	assert(ret == 0);

	/* MAP_PRIVATE file */
	buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd, 0);
	assert(buf != MAP_FAILED);
	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
	assert(ret == -1);
	assert(errno == EINVAL);
	ret = munmap(buf, mapsz);
	assert(ret == 0);

	/* MAP_SHARED file */
	buf = mmap(NULL, mapsz, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
	assert(buf != MAP_FAILED);
	ret = memcntl(buf, mapsz, MC_INHERIT_ZERO, 0, 0, 0);
	assert(ret == -1);
	assert(errno == EINVAL);
	ret = munmap(buf, mapsz);
	assert(ret == 0);

	ret = close(fd);
	assert(ret == 0);
	(void) unlink(tmpfile);
	free(tmpfile);

	return (0);
}