#!/bin/bash
set -e

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

git clone https://github.com/pq-crystals/kyber.git;
cd kyber/ref;
make all;

# Create vectors for each security level and mode
for tvec in test_vectors*[^.c];
  do
  sub_str=${tvec/est_/};
  ./$tvec > ${sub_str/tor/};
done;

echo 'Moving Files...'

# Move test vectors and sha256sums into the PQC-Kyber KAT folder
mv tvecs* ../..
cd ../..

echo 'Calculating SHA256 digests...'

# SHA256SUMS
for tvec in tvecs{5,7,1}*;
do
  shasum -a 256 $tvec >> SHA256SUMS;
done;

# Confirm SHA256SUMS match rust repo KAT's
# Please submit a github issue if upstream tests vectors have changed
echo "Checking SHA256SUMS of test vectors built match those used in the test suite..."
diff -s SHA256SUMS_ORIG SHA256SUMS