#!/bin/bash

set -eu

PYTHON_DIR=$(dirname $0)
ROOT_DIR=${PYTHON_DIR}/../../
SRC_DIR=${ROOT_DIR}/src/add_python
RESULTS_DIR=${ROOT_DIR}/results
CRYPTO_AEAD=${RESULTS_DIR}/crypto_aead_py

mkdir -p ${CRYPTO_AEAD}/lilliputae{i,ii}{128,192,256}v1

declare -A NAMES=([1]=lilliputaei [2]=lilliputaeii)


generate ()
{
    local mode=$1
    local keylen=$2

    echo generating for ${mode} ${keylen}

    cat <<EOF > ${RESULTS_DIR}/parameters.py
MODE = ${mode}
KEYBYTES = $((keylen/8))
EOF

    PYTHONPATH=${RESULTS_DIR}:${SRC_DIR} ${PYTHON_DIR}/genkat_aead.py

    dest=${CRYPTO_AEAD}/${NAMES[${mode}]}${keylen}v1
    mv LWC_AEAD_KAT_${keylen}_120.txt ${dest}
}


for mode in 1 2
do
    for keylen in 128 192 256
    do
        generate ${mode} ${keylen}
    done
done