#include "api.h"
#include "isap.h"
#include "crypto_aead.h"

int crypto_aead_encrypt(
	unsigned char *c, unsigned long long *clen,
	const unsigned char *m, unsigned long long mlen,
	const unsigned char *ad, unsigned long long adlen,
	const unsigned char *nsec,
	const unsigned char *npub,
	const unsigned char *k)
{
	(void)nsec;

	// Ciphertext length is mlen + tag length
	*clen = mlen + ISAP_TAG_SZ;

	// Encrypt plaintext
	if (mlen > 0)
	{
		isap_enc(k, npub, m, mlen, c);
	}

	// Generate tag
	unsigned char *tag = c + mlen;
	isap_mac(k, npub, ad, adlen, c, mlen, tag);
	return 0;
}

int crypto_aead_decrypt(
	unsigned char *m, unsigned long long *mlen,
	unsigned char *nsec,
	const unsigned char *c, unsigned long long clen,
	const unsigned char *ad, unsigned long long adlen,
	const unsigned char *npub,
	const unsigned char *k)
{
	(void)nsec;

	// Plaintext length is clen - tag length
	*mlen = clen - ISAP_TAG_SZ;

	// Generate tag
	unsigned char T[16];
	isap_mac(k, npub, ad, adlen, c, *mlen, T);

	// Compare tag
	unsigned char T_star[16];
	for (int i = 0; i < 16; i++)
    {
        T_star[i] = *(c + *mlen + i);
    }
	int eq_cnt = pvp(T, T_star);

	// Perform decryption if tag is correct
	if (eq_cnt == ISAP_TAG_SZ)
	{
		if (*mlen > 0)
		{
			isap_enc(k, npub, c, *mlen, m);
		}
		return 0;
	}
	else
	{
		return -1;
	}
}