1 files changed, 41 insertions, 0 deletions
diff --git a/test/python/generate-vectors.sh b/test/python/generate-vectors.sh
new file mode 100755
index 0000000..d37f15f
--- /dev/null
+++ b/test/python/generate-vectors.sh
@@ -0,0 +1,41 @@
+#!/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