#ifndef __THREE_BEARS_MAMABEAR_H__
#define __THREE_BEARS_MAMABEAR_H__

#include <stddef.h> /* for size_t */
#include <stdint.h>

#define MAMABEAR_KEYGEN_SEED_BYTES 40
#define MAMABEAR_PRIVATE_KEY_BYTES MAMABEAR_KEYGEN_SEED_BYTES
#define MAMABEAR_SHARED_SECRET_BYTES 32
#define MAMABEAR_ENC_SEED_AND_IV_BYTES 32
#define MAMABEAR_PUBLIC_KEY_BYTES 1194
#define MAMABEAR_CAPSULE_BYTES 1307

/**
 * Expand a secret seed to a public/private keypair.
 *
 * @param[out] pk The public key.
 * @param[in] sk The private key, which must be uniformly random.
 */
void MamaBear_get_pubkey (
    uint8_t pk[MAMABEAR_PUBLIC_KEY_BYTES],
    const uint8_t sk[MAMABEAR_PRIVATE_KEY_BYTES]
);

/**
 * Create a shared secret using a random seed and another party's public key.
 *
 * Input and output parameters may not alias.
 *
 * @param[out] shared_secret The shared secret key.
 * @param[out] capsule A ciphertext to send to the other party.
 * @param[in] pk The other party's public key.
 * @param[in] seed A random seed.
 */
void MamaBear_encapsulate (
    uint8_t shared_secret[MAMABEAR_SHARED_SECRET_BYTES],
    uint8_t capsule[MAMABEAR_CAPSULE_BYTES],
    const uint8_t pk[MAMABEAR_PUBLIC_KEY_BYTES],
    const uint8_t seed[MAMABEAR_ENC_SEED_AND_IV_BYTES]
);

/**
 * Extract the shared secret from a capsule using the private key.
 * Has a negligible but nonzero probability of failure.
 *
 * Input and output parameters may not alias.
 *
 * @param[out] shared_secret The shared secret.
 * @param[in] capsule The capsule produced by encapsulate_cca2.
 * @param[in] sk The private key.
 */
void MamaBear_decapsulate (
    uint8_t shared_secret[MAMABEAR_SHARED_SECRET_BYTES],
    const uint8_t capsule[MAMABEAR_CAPSULE_BYTES],
    const uint8_t sk[MAMABEAR_PRIVATE_KEY_BYTES]
);

void PQCLEAN_MAMABEAR_CLEAN_secure_bzero (void *s, size_t size);

#endif