#include <limits.h>
#include <stdint.h>
#include <sys/types.h>
#include <stdlib.h>
#include <errno.h>
#include <linux/ncr.h>
#include "ncrypto.h"

int
ncr_key_generate_params_init(ncr_key_generate_params_t *params)
{
	ncr_key_generate_params_t rv;

	if (!params) {
		errno = EINVAL;
		return -1;
	}

	if (!(rv = calloc(1, sizeof(*rv)))) {
		errno = ENOMEM;
		return -1;
	}

	rv->algorithm = NCR_ALG_NONE;
	*params = rv;

	return 0;
}

int
ncr_key_generate_params_deinit(ncr_key_generate_params_t params)
{
	if (params)
		free(params);

	return 0;
}

int
ncr_key_generate_params_set_algorithm(ncr_key_generate_params_t params, ncr_algorithm_t algorithm)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}

	params->algorithm = algorithm;

	return 0;
}

int
ncr_key_generate_params_set_keyflags(ncr_key_generate_params_t params, unsigned int keyflags)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}

	params->keyflags = keyflags;

	return 0;
}

int
ncr_key_generate_params_set_secret_bits(ncr_key_generate_params_t params, unsigned int bits)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}
	params->params.secret.bits = bits;

	return 0;
}

int
ncr_key_generate_params_set_rsa_bits(ncr_key_generate_params_t params, unsigned int bits)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}
	params->params.rsa.bits = bits;

	return 0;
}

int
ncr_key_generate_params_set_rsa_e(ncr_key_generate_params_t params, void *e, size_t e_size)
{
	unsigned long value;
	const uint8_t *p;

	if (!params || !e) {
		errno = EINVAL;
		return -1;
	}
	value = 0;
	for (p = e; p < (const uint8_t *)e + e_size; p++) {
		if (value > (ULONG_MAX - *p) / 256) {
			errno = EOVERFLOW;
			return -1;
		}
		value = value * 256 + *p;
	}

	params->params.rsa.e = value;
	return 0;
}

int
ncr_key_generate_params_set_dsa_p_bits(ncr_key_generate_params_t params,
				       unsigned int p_bits)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}
	params->params.dsa.p_bits = p_bits;

	return 0;
}

int
ncr_key_generate_params_set_dsa_q_bits(ncr_key_generate_params_t params,
				       unsigned int q_bits)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}
	params->params.dsa.q_bits = q_bits;

	return 0;
}

int
ncr_key_generate_params_set_dh(ncr_key_generate_params_t params, void *p, size_t p_size, void *g, size_t g_size)
{
	if (!params) {
		errno = EINVAL;
		return -1;
	}

	params->params.dh.p = p;
	params->params.dh.p_size = p_size;
	params->params.dh.g = g;
	params->params.dh.g_size = g_size;

	return 0;
}