2 files changed, 40 insertions, 0 deletions
diff --git a/nist/make-package.sh b/nist/make-package.sh
index 234532a..c3bbd32 100755
--- a/nist/make-package.sh
+++ b/nist/make-package.sh
@@ -116,3 +116,5 @@ done
 cp -r ${TMP_DIR}/crypto_aead .
 
 cleanup
+
+${NIST_DIR}/package-python.sh
diff --git a/nist/package-python.sh b/nist/package-python.sh
new file mode 100755
index 0000000..88c4ef1
--- /dev/null
+++ b/nist/package-python.sh
@@ -0,0 +1,38 @@
+#!/bin/bash
+
+set -eu
+
+NIST_DIR=$(dirname $0)
+ROOT=${NIST_DIR}/..
+ADD_PYTHON=${ROOT}/src/add_python
+PYTHON_RESOURCES=${ROOT}/test/python/
+
+
+add-variant ()
+{
+    local mode=$1
+    local keylen=$2
+
+    local -A names=([1]=lilliputaei [2]=lilliputaeii)
+
+    variant_dir=crypto_aead/${names[${mode}]}${keylen}v1
+
+    cp -r ${ADD_PYTHON} ${variant_dir}
+
+    cat <<EOF > ${variant_dir}/add_python/parameters.py
+MODE = ${mode}
+KEYBYTES = $((keylen/8))
+EOF
+
+    cp ${PYTHON_RESOURCES}/{crypto_aead.py,genkat_aead.py} \
+       ${variant_dir}/add_python
+}
+
+
+for mode in 1 2
+do
+    for keylen in 128 192 256
+    do
+        add-variant ${mode} ${keylen}
+    done
+done