Add tests for all functions of each algorithm against AEGIS test vectors (JSON). Incremental updates are tested with randomized splits to catch a variety of errors.

2025-11-06 19:56:12 +00:00
parent c16e785203
commit df6b1b2832
13 changed files with 979 additions and 0 deletions
@@ -0,0 +1,89 @@
+#! /usr/bin/env python3
+
+import json
+import re
+
+
+def tvdump(topic, tvs):
+    with open(filename(topic), "w") as f:
+        f.write(json.dumps(tvs, indent=2))
+
+    print(json.dumps(tvs, indent=2))
+
+
+def filename(topic):
+    return re.sub(r"[^a-z0-9]+", "-", topic.lower()) + ".json"
+
+
+header = True
+in_tv = False
+tv = {}
+tvs = []
+must_fail = False
+with open("../draft-irtf-cfrg-aegis-aead.md") as f:
+    for line in f:
+        line = line.strip()
+        if line == "":
+            continue
+        if line.startswith("# Test Vectors"):
+            header = False
+            continue
+        if header:
+            continue
+
+        if line.startswith("## "):
+            if len(tvs) > 0:
+                tvdump(topic, tvs)
+            topic = line[3:]
+            tv_name = topic
+            tvs = []
+            continue
+
+        if line.startswith("### "):
+            tv_name = line[4:]
+            tv = {"test": tv_name}
+            in_tv = False
+            continue
+
+        if line == "~~~ test-vectors":
+            in_tv = True
+            tv = {"name": tv_name}
+            if must_fail:
+                tv["error"] = "verification failed"
+                must_fail = False
+            continue
+
+        if line == "~~~":
+            tvs.append(tv)
+            in_tv = False
+            current_key = None
+            continue
+
+        if line.find("verification failed") != -1:
+            must_fail = True
+            continue
+
+        if line == "After initialization:":
+            tv_name = tv_name + " (after initialization)"
+
+        if not in_tv:
+            continue
+
+        parts = line.split(":")
+        if len(parts) == 2:
+            key = parts[0].strip()
+            value = parts[1].strip()
+            if key == "After Update":
+                continue
+            if key in tv:
+                key = key + "_2"
+            tv[key] = value
+            current_key = key
+            continue
+
+        if not current_key:
+            continue
+
+        tv[key] += line.strip()
+
+tvdump(topic, tvs)