/*
 * 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) 2018, Joyent, Inc.
 */

/*
 * getrandom system call implementation
 */

#include <sys/types.h>
#include <sys/errno.h>
#include <sys/systm.h>
#include <sys/random.h>
#include <sys/ddi.h>
#include <sys/sunddi.h>
#include <sys/sysmacros.h>

#include <sys/random.h>

/*
 * Impose a maximum upper bound on the number of bytes that we'll read in one
 * go, ala a read of /dev/random. For /dev/urandom, we clamp it based on our
 * return value, because the system call returns an int, we can't handle more
 * than INT_MAX.
 */
#define	MAXRANDBYTES	1024
#define	MAXURANDBYTES	INT_MAX

ssize_t
getrandom(void *bufp, size_t buflen, unsigned int flags)
{
	int out = 0;
	uint8_t rbytes[128];
	uint8_t *buf = bufp;

	if (flags & ~(GRND_NONBLOCK | GRND_RANDOM))
		return (set_errno(EINVAL));

	if ((flags & GRND_RANDOM) && buflen > MAXRANDBYTES) {
		buflen = MAXRANDBYTES;
	} else if (buflen > MAXURANDBYTES) {
		buflen = MAXURANDBYTES;
	}

	while (out < buflen) {
		int err;
		size_t len = MIN(sizeof (rbytes), buflen - out);

		if (flags & GRND_RANDOM) {
			if (flags & GRND_NONBLOCK)
				err = random_get_bytes(rbytes, len);
			else
				err = random_get_blocking_bytes(rbytes, len);
		} else {
			err = random_get_pseudo_bytes(rbytes, len);
		}

		if (err == 0) {
			if (ddi_copyout(rbytes, buf + out, len, 0) != 0)
				return (set_errno(EFAULT));
			out += len;
		} else if (err == EAGAIN && out > 0) {
			break;
		} else {
			return (set_errno(err));
		}
	}

	return (out);
}