// -*- mode: c; tab-width: 8; indent-tabs-mode: 1; st-rulers: [70] -*-
// vim: ts=8 sw=8 ft=c noet

#include "libsodium_api_crypto_generichash.h"

static void	LS_API_EXEC(crypto_generichash, bytes_min);
static void	LS_API_EXEC(crypto_generichash, bytes_max);
static void	LS_API_EXEC(crypto_generichash, bytes);
static void	LS_API_EXEC(crypto_generichash, keybytes_min);
static void	LS_API_EXEC(crypto_generichash, keybytes_max);
static void	LS_API_EXEC(crypto_generichash, keybytes);
static void	LS_API_EXEC(crypto_generichash, primitive);
static void	LS_API_EXEC(crypto_generichash, statebytes);
static int	LS_API_INIT(crypto_generichash, crypto_generichash);
static void	LS_API_EXEC(crypto_generichash, crypto_generichash);
static int	LS_API_INIT(crypto_generichash, init);
static void	LS_API_EXEC(crypto_generichash, init);
static int	LS_API_INIT(crypto_generichash, update);
static void	LS_API_EXEC(crypto_generichash, update);
static int	LS_API_INIT(crypto_generichash, final);
static void	LS_API_EXEC(crypto_generichash, final);

libsodium_function_t	libsodium_functions_crypto_generichash[] = {
	LS_API_R_ARG0(crypto_generichash, bytes_min),
	LS_API_R_ARG0(crypto_generichash, bytes_max),
	LS_API_R_ARG0(crypto_generichash, bytes),
	LS_API_R_ARG0(crypto_generichash, keybytes_min),
	LS_API_R_ARG0(crypto_generichash, keybytes_max),
	LS_API_R_ARG0(crypto_generichash, keybytes),
	LS_API_R_ARG0(crypto_generichash, primitive),
	LS_API_R_ARG0(crypto_generichash, statebytes),
	LS_API_R_ARGV(crypto_generichash, crypto_generichash, 3),
	LS_API_R_ARGV(crypto_generichash, init, 2),
	LS_API_R_ARGV(crypto_generichash, update, 2),
	LS_API_R_ARGV(crypto_generichash, final, 2),
	{NULL}
};

/* crypto_generichash_bytes_min/0 */

static void
LS_API_EXEC(crypto_generichash, bytes_min)
{
	size_t bytes_min;

	bytes_min = crypto_generichash_bytes_min();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(bytes_min),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_bytes_max/0 */

static void
LS_API_EXEC(crypto_generichash, bytes_max)
{
	size_t bytes_max;

	bytes_max = crypto_generichash_bytes_max();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(bytes_max),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_bytes/0 */

static void
LS_API_EXEC(crypto_generichash, bytes)
{
	size_t bytes;

	bytes = crypto_generichash_bytes();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(bytes),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_keybytes_min/0 */

static void
LS_API_EXEC(crypto_generichash, keybytes_min)
{
	size_t keybytes_min;

	keybytes_min = crypto_generichash_keybytes_min();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(keybytes_min),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_keybytes_max/0 */

static void
LS_API_EXEC(crypto_generichash, keybytes_max)
{
	size_t keybytes_max;

	keybytes_max = crypto_generichash_keybytes_max();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(keybytes_max),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_keybytes/0 */

static void
LS_API_EXEC(crypto_generichash, keybytes)
{
	size_t keybytes;

	keybytes = crypto_generichash_keybytes();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(keybytes),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_primitive/0 */

static void
LS_API_EXEC(crypto_generichash, primitive)
{
	const char *primitive;

	primitive = crypto_generichash_primitive();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_ATOM, driver_mk_atom((char *)(primitive)),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_statebytes/0 */

static void
LS_API_EXEC(crypto_generichash, statebytes)
{
	size_t statebytes;

	statebytes = crypto_generichash_statebytes();

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_UINT, (ErlDrvUInt)(statebytes),
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_crypto_generichash/3 */

typedef struct LS_API_F_ARGV(crypto_generichash, crypto_generichash) {
	size_t			outlen;
	const unsigned char	*in;
	unsigned long long	inlen;
	const unsigned char	*key;
	size_t			keylen;
} LS_API_F_ARGV_T(crypto_generichash, crypto_generichash);

static int
LS_API_INIT(crypto_generichash, crypto_generichash)
{
	LS_API_F_ARGV_T(crypto_generichash, crypto_generichash) *argv;
	int skip;
	int type;
	int type_length;
	size_t outlen;
	unsigned long long inlen;
	size_t keylen;
	ErlDrvSizeT x;
	void *p;

	if (ei_decode_ulong(buffer, index, (unsigned long *)&(outlen)) < 0) {
		return -1;
	}

	if (outlen > crypto_generichash_bytes_max() || outlen < crypto_generichash_bytes_min()) {
		return -1;
	}

	if (ei_get_type(buffer, index, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT) {
		return -1;
	}

	inlen = (unsigned long long)(type_length);

	skip = *index;

	if (ei_skip_term(buffer, &skip) < 0) {
		return -1;
	}

	if (ei_get_type(buffer, &skip, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT) {
		return -1;
	}

	keylen = (size_t)(type_length);

	if (keylen != 0 && (keylen > crypto_generichash_keybytes_max() || keylen < crypto_generichash_keybytes_min())) {
		return -1;
	}

	x = (ErlDrvSizeT)(inlen + keylen + (sizeof (LS_API_F_ARGV_T(crypto_generichash, crypto_generichash))));
	p = (void *)(driver_alloc(x));

	if (p == NULL) {
		return -1;
	}

	argv = (LS_API_F_ARGV_T(crypto_generichash, crypto_generichash) *)(p);
	argv->outlen = outlen;
	p += (sizeof (LS_API_F_ARGV_T(crypto_generichash, crypto_generichash)));
	argv->in = (const unsigned char *)(p);

	if (keylen == 0) {
		argv->key = NULL;
		argv->keylen = 0;
	} else {
		p += inlen;
		argv->key = (const unsigned char *)(p);
	}

	if (ei_decode_binary(buffer, index, (void *)(argv->in), (long *)&(argv->inlen)) < 0) {
		(void) driver_free(argv);
		return -1;
	}

	if (keylen > 0) {
		if (ei_decode_binary(buffer, index, (void *)(argv->key), (long *)&(argv->keylen)) < 0) {
			(void) driver_free(argv);
			return -1;
		}
	}

	request->argv = (void *)(argv);

	return 0;
}

static void
LS_API_EXEC(crypto_generichash, crypto_generichash)
{
	LS_API_F_ARGV_T(crypto_generichash, crypto_generichash) *argv;
	LS_API_READ_ARGV(crypto_generichash, crypto_generichash);
	unsigned char *out;

	out = (unsigned char *)(driver_alloc((ErlDrvSizeT)(argv->outlen)));

	if (out == NULL) {
		LS_FAIL_OOM(request->port->drv_port);
		return;
	}

	(void) crypto_generichash(out, argv->outlen, argv->in, argv->inlen, argv->key, argv->keylen);

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_BUF2BINARY, (ErlDrvTermData)(out), argv->outlen,
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);

	(void) driver_free(out);
}

/* crypto_generichash_init/2 */

typedef struct LS_API_F_ARGV(crypto_generichash, init) {
	const unsigned char	*key;
	size_t			keylen;
	size_t			outlen;
} LS_API_F_ARGV_T(crypto_generichash, init);

static int
LS_API_INIT(crypto_generichash, init)
{
	LS_API_F_ARGV_T(crypto_generichash, init) *argv;
	int skip;
	int type;
	int type_length;
	size_t keylen;
	size_t outlen;
	ErlDrvSizeT x;
	void *p;

	if (ei_get_type(buffer, index, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT) {
		return -1;
	}

	keylen = (size_t)(type_length);

	if (keylen != 0 && (keylen > crypto_generichash_keybytes_max() || keylen < crypto_generichash_keybytes_min())) {
		return -1;
	}

	skip = *index;

	if (ei_skip_term(buffer, &skip) < 0) {
		return -1;
	}

	if (ei_decode_ulong(buffer, &skip, (unsigned long *)&(outlen)) < 0) {
		return -1;
	}

	if (outlen > crypto_generichash_bytes_max() || outlen < crypto_generichash_bytes_min()) {
		return -1;
	}

	x = (ErlDrvSizeT)(keylen + (sizeof (LS_API_F_ARGV_T(crypto_generichash, init))));
	p = (void *)(driver_alloc(x));

	if (p == NULL) {
		return -1;
	}

	argv = (LS_API_F_ARGV_T(crypto_generichash, init) *)(p);

	if (keylen == 0) {
		argv->key = NULL;
		argv->keylen = 0;
	} else {
		p += (sizeof (LS_API_F_ARGV_T(crypto_generichash, init)));
		argv->key = (const unsigned char *)(p);
		if (ei_decode_binary(buffer, index, (void *)(argv->key), (long *)&(argv->keylen)) < 0) {
			(void) driver_free(argv);
			return -1;
		}
	}

	argv->outlen = outlen;

	request->argv = (void *)(argv);

	return 0;
}

static void
LS_API_EXEC(crypto_generichash, init)
{
	LS_API_F_ARGV_T(crypto_generichash, init) *argv;
	LS_API_READ_ARGV(crypto_generichash, init);
	size_t statebytes;
	crypto_generichash_state *state;

	statebytes = crypto_generichash_statebytes();

	state = (crypto_generichash_state *)(sodium_malloc(statebytes));

	if (state == NULL) {
		LS_FAIL_OOM(request->port->drv_port);
		return;
	}

	(void) crypto_generichash_init(state, argv->key, argv->keylen, argv->outlen);

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_BUF2BINARY, (ErlDrvTermData)(char *)(state), statebytes,
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);

	(void) sodium_free(state);
}

/* crypto_generichash_update/2 */

typedef struct LS_API_F_ARGV(crypto_generichash, update) {
	crypto_generichash_state	*state;
	const unsigned char		*in;
	unsigned long long		inlen;
} LS_API_F_ARGV_T(crypto_generichash, update);

static int
LS_API_INIT(crypto_generichash, update)
{
	LS_API_F_ARGV_T(crypto_generichash, update) *argv;
	int skip;
	int type;
	int type_length;
	size_t statebytes;
	ErlDrvSizeT x;
	void *p;

	statebytes = crypto_generichash_statebytes();

	if (ei_get_type(buffer, index, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT
			|| type_length != statebytes) {
		return -1;
	}

	skip = *index;

	if (ei_skip_term(buffer, &skip) < 0) {
		return -1;
	}

	if (ei_get_type(buffer, &skip, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT) {
		return -1;
	}

	x = (ErlDrvSizeT)(statebytes + type_length + (sizeof (LS_API_F_ARGV_T(crypto_generichash, update))));
	p = (void *)(driver_alloc(x));

	if (p == NULL) {
		return -1;
	}

	argv = (LS_API_F_ARGV_T(crypto_generichash, update) *)(p);
	p += (sizeof (LS_API_F_ARGV_T(crypto_generichash, update)));
	argv->state = (crypto_generichash_state *)(p);
	p += statebytes;
	argv->in = (const unsigned char *)(p);

	if (ei_decode_binary(buffer, index, (void *)(argv->state), NULL) < 0) {
		(void) driver_free(argv);
		return -1;
	}

	if (ei_decode_binary(buffer, index, (void *)(argv->in), (long *)&(argv->inlen)) < 0) {
		(void) driver_free(argv);
		return -1;
	}

	request->argv = (void *)(argv);

	return 0;
}

static void
LS_API_EXEC(crypto_generichash, update)
{
	LS_API_F_ARGV_T(crypto_generichash, update) *argv;
	LS_API_READ_ARGV(crypto_generichash, update);
	size_t statebytes;

	statebytes = crypto_generichash_statebytes();

	(void) crypto_generichash_update(argv->state, argv->in, argv->inlen);

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_BUF2BINARY, (ErlDrvTermData)(char *)(argv->state), statebytes,
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}

/* crypto_generichash_final/2 */

typedef struct LS_API_F_ARGV(crypto_generichash, final) {
	crypto_generichash_state	*state;
	size_t				outlen;
} LS_API_F_ARGV_T(crypto_generichash, final);

static int
LS_API_INIT(crypto_generichash, final)
{
	LS_API_F_ARGV_T(crypto_generichash, final) *argv;
	int skip;
	int type;
	int type_length;
	size_t statebytes;
	size_t outlen;
	ErlDrvSizeT x;
	void *p;

	statebytes = crypto_generichash_statebytes();

	if (ei_get_type(buffer, index, &type, &type_length) < 0
			|| type != ERL_BINARY_EXT
			|| type_length != statebytes) {
		return -1;
	}

	skip = *index;

	if (ei_skip_term(buffer, &skip) < 0) {
		return -1;
	}

	if (ei_decode_ulong(buffer, &skip, (unsigned long *)&(outlen)) < 0) {
		return -1;
	}

	if (outlen > crypto_generichash_bytes_max() || outlen < crypto_generichash_bytes_min()) {
		return -1;
	}

	x = (ErlDrvSizeT)(statebytes + (sizeof (LS_API_F_ARGV_T(crypto_generichash, final))));
	p = (void *)(driver_alloc(x));

	if (p == NULL) {
		return -1;
	}

	argv = (LS_API_F_ARGV_T(crypto_generichash, final) *)(p);
	p += (sizeof (LS_API_F_ARGV_T(crypto_generichash, final)));
	argv->state = (crypto_generichash_state *)(p);
	argv->outlen = outlen;

	if (ei_decode_binary(buffer, index, (void *)(argv->state), NULL) < 0) {
		(void) driver_free(argv);
		return -1;
	}

	request->argv = (void *)(argv);

	return 0;
}

static void
LS_API_EXEC(crypto_generichash, final)
{
	LS_API_F_ARGV_T(crypto_generichash, final) *argv;
	LS_API_READ_ARGV(crypto_generichash, final);
	unsigned char *out;

	out = (unsigned char *)(driver_alloc((ErlDrvSizeT)(argv->outlen)));

	if (out == NULL) {
		LS_FAIL_OOM(request->port->drv_port);
		return;
	}

	(void) crypto_generichash_final(argv->state, out, argv->outlen);

	ErlDrvTermData spec[] = {
		LS_RES_TAG(request),
		ERL_DRV_BUF2BINARY, (ErlDrvTermData)(out), argv->outlen,
		ERL_DRV_TUPLE, 2
	};

	LS_RESPOND(request, spec, __FILE__, __LINE__);
}