Add Python script to parse ML-KEM encapDecap KAT files

Signed-off-by: Anjan Roy <hello@itzmeanjan.in>

kats/scripts/parse_ml_kem_encaps_acvp_kat.py

@@ -0,0 +1,44 @@
+#!/usr/bin/python
+
+import json
+import sys
+import typing
+
+ML_KEM_512_ACVP_KAT_FILE_NAME="ml_kem_512_encaps.acvp.kat"
+ML_KEM_768_ACVP_KAT_FILE_NAME="ml_kem_768_encaps.acvp.kat"
+ML_KEM_1024_ACVP_KAT_FILE_NAME="ml_kem_1024_encaps.acvp.kat"
+
+def extract_and_write_ml_kem_encaps_kats(test_group: dict[str, typing.Any], write_to_file: str):
+    assert test_group["testType"] == "AFT"
+
+    with open(write_to_file, "wt") as fd:
+        for test in test_group["tests"]:
+            fd.write(f'pk = {test["ek"]}\n')
+            fd.write(f'sk = {test["dk"]}\n')
+            fd.write(f'm = {test["m"]}\n')
+            fd.write(f'ct = {test["c"]}\n')
+            fd.write(f'ss = {test["k"]}\n')
+
+            fd.write('\n')
+        
+        fd.flush()
+
+
+def main():
+    json_as_str = ''
+    for line in sys.stdin:
+        json_as_str += line
+    
+    acvp_kats = json.loads(json_as_str)
+
+    ml_kem_512_param_set = acvp_kats["testGroups"][0]
+    ml_kem_768_param_set = acvp_kats["testGroups"][1]
+    ml_kem_1024_param_set = acvp_kats["testGroups"][2]
+
+    extract_and_write_ml_kem_encaps_kats(ml_kem_512_param_set, ML_KEM_512_ACVP_KAT_FILE_NAME)
+    extract_and_write_ml_kem_encaps_kats(ml_kem_768_param_set, ML_KEM_768_ACVP_KAT_FILE_NAME)
+    extract_and_write_ml_kem_encaps_kats(ml_kem_1024_param_set, ML_KEM_1024_ACVP_KAT_FILE_NAME)
+
+
+if __name__=='__main__':
+    main()

kats/scripts/sync_acvp_kats.sh

@@ -15,6 +15,7 @@ else
 fi
 
 cat ./ACVP-Server/gen-val/json-files/ML-KEM-keyGen-FIPS203/internalProjection.json | python parse_ml_kem_keygen_acvp_kat.py
+cat ./ACVP-Server/gen-val/json-files/ML-KEM-encapDecap-FIPS203/internalProjection.json | python parse_ml_kem_encaps_acvp_kat.py
 mv *.acvp.kat ..
 
 echo "> Generated all NIST ACVP KATs."