#!/bin/sh

#
# Copyright (c) 2018 Dell EMC Isilon
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#

# msetdomain(2) fuzz test.
# No problems seen.

. ../default.cfg
[ `id -u` -ne 0 ] && echo "Must be root!" && exit 1

nm /usr/lib/libc.a | grep -q __sys_msetdomain || exit 0
dir=/tmp
odir=`pwd`
cd $dir
sed '1,/^EOF/d' < $odir/$0 > $dir/msetdomain.c
mycc -o msetdomain -Wall -Wextra -O0 -g msetdomain.c || exit 1
rm -f msetdomain.c
cd $odir

$dir/msetdomain
s=$?
[ -f msetdomain.core -a $s -eq 0 ] &&
    { ls -l msetdomain.core; mv msetdomain.core /tmp; s=1; }
rm -rf $dir/msetdomain
exit $s

EOF
#include <sys/param.h>
#include <sys/mman.h>
#include <sys/resource.h>
#include <sys/cpuset.h>
#include <sys/domainset.h>

#include <err.h>
#include <stdlib.h>
#include <time.h>

/*
struct msetdomain_args {
        void            *addr;
        size_t          size;
        size_t          domainsetsize;
        domainset_t     *mask;
        int             policy;
        int             flags;
*/

static long
random_long(long mi, long ma)
{
        return (arc4random()  % (ma - mi + 1) + mi);
}

void
flip(void *ap, size_t len)
{
	unsigned char *cp;
	int byte;
	unsigned char bit, buf, mask, old;

	cp = (unsigned char *)ap;
	byte = random_long(0, len);
	bit = random_long(0,7);
	mask = ~(1 << bit);
	buf = cp[byte];
	old = cp[byte];
	buf = (buf & mask) | (~buf & ~mask);
	cp[byte] = buf;
}

int
main(void)
{
	size_t len;
	time_t start;
	void *share;
	domainset_t rootmask;
	int flags, policy;

	len = PAGE_SIZE;
	if ((share = mmap(NULL, len, PROT_READ | PROT_WRITE,
	    MAP_ANON | MAP_SHARED, -1, 0)) == MAP_FAILED)
		err(1, "mmap");

	start = time(NULL);
	while (time(NULL) - start < 60) {
		if (cpuset_getdomain(CPU_LEVEL_ROOT, CPU_WHICH_PID, -1,
		    sizeof(rootmask), &rootmask, &policy) != 0)
			err(EXIT_FAILURE, "getdomain");

		flags = 0;
		flip(&flags, sizeof(flags));
		msetdomain(share, len, sizeof(rootmask), &rootmask, policy,
		    flags);
	}

	start = time(NULL);
	while (time(NULL) - start < 60) {
		if (cpuset_getdomain(CPU_LEVEL_ROOT, CPU_WHICH_PID, -1,
		    sizeof(rootmask), &rootmask, &policy) != 0)
			err(EXIT_FAILURE, "getdomain");

		flip(&policy, sizeof(policy));
		msetdomain(share, len, sizeof(rootmask), &rootmask, policy,
		    flags);
	}

	start = time(NULL);
	while (time(NULL) - start < 60) {
		if (cpuset_getdomain(CPU_LEVEL_ROOT, CPU_WHICH_PID, -1,
		    sizeof(rootmask), &rootmask, &policy) != 0)
			err(EXIT_FAILURE, "getdomain");

		flip(&rootmask, sizeof(rootmask));
		msetdomain(share, len, sizeof(rootmask), &rootmask, policy,
		    flags);
	}

	return (0);
}