/*
 * Copyright (c) 2017 Linaro Limited.
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <device.h>
#include <random.h>
#include <drivers/rand32.h>
#include <init.h>

#include "fsl_trng.h"

static int random_mcux_trng_get_entropy(struct device *dev, u8_t *buffer,
					u16_t length)
{
	status_t status;

	ARG_UNUSED(dev);

	status = TRNG_GetRandomData(TRNG0, buffer, length);
	__ASSERT_NO_MSG(!status);

	return 0;
}

static const struct random_driver_api random_mcux_trng_api_funcs = {
	.get_entropy = random_mcux_trng_get_entropy
};

static int random_mcux_trng_init(struct device *);

DEVICE_AND_API_INIT(random_mcux_trng, CONFIG_RANDOM_NAME,
		    random_mcux_trng_init, NULL, NULL,
		    PRE_KERNEL_2, CONFIG_KERNEL_INIT_PRIORITY_DEVICE,
		    &random_mcux_trng_api_funcs);

static int random_mcux_trng_init(struct device *dev)
{
	trng_config_t conf;
	status_t status;

	ARG_UNUSED(dev);

	status = TRNG_GetDefaultConfig(&conf);
	__ASSERT_NO_MSG(!status);

	status = TRNG_Init(TRNG0, &conf);
	__ASSERT_NO_MSG(!status);

	return 0;
}

u32_t sys_rand32_get(void)
{
	u32_t output;
	int rc;

	rc = random_mcux_trng_get_entropy(DEVICE_GET(random_mcux_trng),
					  (u8_t *) &output, sizeof(output));
	__ASSERT_NO_MSG(!rc);

	return output;
}