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

#include <decaf/ed255.h>

/*
 * Erlang NIF functions
 */

/* decaf/ed255.h */

static ERL_NIF_TERM
libdecaf_ed25519_derive_public_key_1(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
	ErlNifBinary privkey;

	if (argc != 1 || !enif_inspect_binary(env, argv[0], &privkey)
			|| privkey.size != DECAF_EDDSA_25519_PRIVATE_BYTES) {
		return enif_make_badarg(env);
	}

	ERL_NIF_TERM out;
	uint8_t *pubkey = (uint8_t *)(enif_make_new_binary(env, DECAF_EDDSA_25519_PUBLIC_BYTES, &out));

	(void) decaf_ed25519_derive_public_key(pubkey, privkey.data);

	return out;
}

static ERL_NIF_TERM
libdecaf_ed25519_sign_5(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
	ErlNifBinary privkey;
	ErlNifBinary pubkey;
	ErlNifBinary message;
	unsigned int prehashed;
	ErlNifBinary context;

	if (argc != 5 || !enif_inspect_binary(env, argv[0], &privkey)
			|| privkey.size != DECAF_EDDSA_25519_PRIVATE_BYTES
			|| !enif_inspect_binary(env, argv[1], &pubkey)
			|| pubkey.size != DECAF_EDDSA_25519_PUBLIC_BYTES
			|| !enif_inspect_binary(env, argv[2], &message)
			|| !enif_get_uint(env, argv[3], &prehashed)
			|| (prehashed != 0 && prehashed != 1)) {
		return enif_make_badarg(env);
	}

	if (enif_compare(ATOM_no_context, argv[4]) == 0) {
		context.size = 0;
		context.data = (unsigned char *)(DECAF_ED25519_NO_CONTEXT);
	} else if (!enif_inspect_binary(env, argv[4], &context) || context.size > 255) {
		return enif_make_badarg(env);
	}

	ERL_NIF_TERM out;
	uint8_t *signature = (uint8_t *)(enif_make_new_binary(env, DECAF_EDDSA_25519_SIGNATURE_BYTES, &out));

	(void) decaf_ed25519_sign(signature, privkey.data, pubkey.data, message.data, message.size, prehashed, context.data, context.size);

	return out;
}

static ERL_NIF_TERM
libdecaf_ed25519_sign_prehash_4(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
	ErlNifBinary privkey;
	ErlNifBinary pubkey;
	ErlNifBinary message;
	ErlNifBinary context;

	if (argc != 4 || !enif_inspect_binary(env, argv[0], &privkey)
			|| privkey.size != DECAF_EDDSA_25519_PRIVATE_BYTES
			|| !enif_inspect_binary(env, argv[1], &pubkey)
			|| pubkey.size != DECAF_EDDSA_25519_PUBLIC_BYTES
			|| !enif_inspect_binary(env, argv[2], &message)) {
		return enif_make_badarg(env);
	}

	if (enif_compare(ATOM_no_context, argv[3]) == 0) {
		context.size = 0;
		context.data = (unsigned char *)(DECAF_ED25519_NO_CONTEXT);
	} else if (!enif_inspect_binary(env, argv[3], &context) || context.size > 255) {
		return enif_make_badarg(env);
	}

	decaf_ed25519_prehash_ctx_t hash;
	(void) decaf_ed25519_prehash_init(hash);
	(void) decaf_ed25519_prehash_update(hash, message.data, message.size);

	ERL_NIF_TERM out;
	uint8_t *signature = (uint8_t *)(enif_make_new_binary(env, DECAF_EDDSA_25519_SIGNATURE_BYTES, &out));

	(void) decaf_ed25519_sign_prehash(signature, privkey.data, pubkey.data, hash, context.data, context.size);

	(void) decaf_ed25519_prehash_destroy(hash);

	return out;
}

static ERL_NIF_TERM
libdecaf_ed25519_verify_5(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
	ErlNifBinary signature;
	ErlNifBinary pubkey;
	ErlNifBinary message;
	unsigned int prehashed;
	ErlNifBinary context;

	if (argc != 5 || !enif_inspect_binary(env, argv[0], &signature)
			|| signature.size != DECAF_EDDSA_25519_SIGNATURE_BYTES
			|| !enif_inspect_binary(env, argv[1], &pubkey)
			|| pubkey.size != DECAF_EDDSA_25519_PUBLIC_BYTES
			|| !enif_inspect_binary(env, argv[2], &message)
			|| !enif_get_uint(env, argv[3], &prehashed)
			|| (prehashed != 0 && prehashed != 1)) {
		return enif_make_badarg(env);
	}

	if (enif_compare(ATOM_no_context, argv[4]) == 0) {
		context.size = 0;
		context.data = (unsigned char *)(DECAF_ED25519_NO_CONTEXT);
	} else if (!enif_inspect_binary(env, argv[4], &context) || context.size > 255) {
		return enif_make_badarg(env);
	}

	if (decaf_ed25519_verify(signature.data, pubkey.data, message.data, message.size, prehashed, context.data, context.size) == DECAF_SUCCESS) {
		return ATOM_true;
	} else {
		return ATOM_false;
	}
}

static ERL_NIF_TERM
libdecaf_ed25519_verify_prehash_4(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[])
{
	ErlNifBinary signature;
	ErlNifBinary pubkey;
	ErlNifBinary message;
	ErlNifBinary context;

	if (argc != 4 || !enif_inspect_binary(env, argv[0], &signature)
			|| signature.size != DECAF_EDDSA_25519_SIGNATURE_BYTES
			|| !enif_inspect_binary(env, argv[1], &pubkey)
			|| pubkey.size != DECAF_EDDSA_25519_PUBLIC_BYTES
			|| !enif_inspect_binary(env, argv[2], &message)) {
		return enif_make_badarg(env);
	}

	if (enif_compare(ATOM_no_context, argv[3]) == 0) {
		context.size = 0;
		context.data = (unsigned char *)(DECAF_ED25519_NO_CONTEXT);
	} else if (!enif_inspect_binary(env, argv[3], &context) || context.size > 255) {
		return enif_make_badarg(env);
	}

	decaf_ed25519_prehash_ctx_t hash;
	(void) decaf_ed25519_prehash_init(hash);
	(void) decaf_ed25519_prehash_update(hash, message.data, message.size);

	if (decaf_ed25519_verify_prehash(signature.data, pubkey.data, hash, context.data, context.size) == DECAF_SUCCESS) {
		(void) decaf_ed25519_prehash_destroy(hash);
		return ATOM_true;
	} else {
		(void) decaf_ed25519_prehash_destroy(hash);
		return ATOM_false;
	}
}