/**
 * API defined by the Cryptographic Engineering Research Group (CREG) from
 * George Mason University (GMU) in their call for protected software
 * implementations of NIST LWC finalists.
 */ 
#include "api.h"
#include <stdint.h>

typedef struct {
    uint32_t shares[NUM_SHARES_M];
} mask_m_uint32_t;

typedef struct {
    uint32_t shares[NUM_SHARES_C];
} mask_c_uint32_t;

typedef struct {
    uint32_t shares[NUM_SHARES_AD];
} mask_ad_uint32_t;

typedef struct {
    uint32_t shares[NUM_SHARES_NPUB];
} mask_npub_uint32_t;

typedef struct {
    uint32_t shares[NUM_SHARES_KEY];
} mask_key_uint32_t;


int crypto_aead_encrypt_shared(
    mask_c_uint32_t* cs, unsigned long long *clen,
    const mask_m_uint32_t *ms, unsigned long long mlen,
    const mask_ad_uint32_t *ads, unsigned long long adlen,
    const mask_npub_uint32_t *npubs,
    const mask_key_uint32_t *ks
);

int crypto_aead_decrypt_shared(
    mask_m_uint32_t* ms, unsigned long long *mlen,
    const mask_c_uint32_t *cs, unsigned long long clen,
    const mask_ad_uint32_t *ads, unsigned long long adlen,
    const mask_npub_uint32_t *npubs,
    const mask_key_uint32_t *ks
);

void generate_shares_encrypt(
    const unsigned char *m, mask_m_uint32_t *ms, const unsigned long long mlen,
    const unsigned char *ad, mask_ad_uint32_t *ads, const unsigned long long adlen,
    const unsigned char *npub, mask_npub_uint32_t *npubs,
    const unsigned char *k, mask_key_uint32_t *ks
);

void generate_shares_decrypt(
    //const unsigned char *c, mask_m_uint32_t *cs, const unsigned long long clen,
    const unsigned char *c, mask_c_uint32_t *cs, const unsigned long long clen,
    const unsigned char *ad, mask_ad_uint32_t *ads, const unsigned long long adlen,
    const unsigned char *npub, mask_npub_uint32_t *npubs,
    const unsigned char *k, mask_key_uint32_t *ks
);

void combine_shares_encrypt(
    const mask_c_uint32_t *cs, unsigned char *c, unsigned long long clen
);

void combine_shares_decrypt(
    const mask_m_uint32_t *ms, unsigned char *m, unsigned long long mlen
);