1 files changed, 128 insertions, 0 deletions

diff --git a/nix/test_all.py b/nix/test_all.py
new file mode 100644
index 0000000..2017a2c
--- /dev/null
+++ b/nix/test_all.py
@@ -0,0 +1,128 @@
+#!/usr/bin/env python
+
+import argparse
+import json
+import time
+
+from pathlib import Path
+
+import subprocess as sp
+
+
+def base_options(library):
+    match library:
+        case "openssl" | "botan" | "boringssl" | "ippcp" | "libressl" | "gcrypt" | "nettle":
+            return ["-ps", "123412341234123412341234123412341234123412341234123412341234123412341234123412341234123412341234"]
+        case "cryptopp" | "mbedtls":
+            return ["-ps", "12345678"]
+        case _:
+            return []
+
+def default_options(library):
+    match library:
+        case "botan" | "cryptopp":
+            return ["-gt", "ECDH"]
+        case _:
+            return []
+
+def test_vectors_options(library):
+    return default_options(library)
+
+def performance_options(library):
+    return default_options(library)
+
+def signature_options(library):
+    match library:
+        case "nettle" | "gcrypt" | "boringssl" | "openssl" | "tomcrypt" | "libressl" | "ippcp" | "mbedtls":
+            return ["-st", "NONEwithECDSA"]
+        case _:
+            return []
+
+def miscellaneous_options(library):
+    return default_options(library)
+
+def twist_options(library):
+    return default_options(library)
+
+def invalid_options(library):
+    return default_options(library)
+
+def degenerate_options(library):
+    return default_options(library)
+
+def cofactor_options(library):
+    return default_options(library)
+
+def composite_options(library):
+    return default_options(library)
+
+def edge_cases_options(library):
+    return default_options(library)
+
+def wrong_options(library):
+    return default_options(library)
+
+def test_library(library, test_suite, version):
+    opts = base_options(library)
+    opts.extend(globals()[f"{test_suite.replace('-', '_')}_options"](library))
+    command = ["nix", "run", f"?submodules=1#{library}.{version}", "--", "test", f"-oyml:results/{library}_{test_suite}_{version}.yml", *opts, test_suite, library]
+    print(command)
+    process = sp.run(command)
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-l", "--library")
+    parser.add_argument("-s", "--suite")
+    args = parser.parse_args()
+    library = args.library
+    suite = args.suite
+
+    libraries = [
+         "botan", 
+         "cryptopp", 
+         "openssl", 
+         "boringssl", 
+         "gcrypt", 
+         "mbedtls", 
+         "ippcp", 
+         "nettle", 
+         "libressl", 
+    ]
+
+    suites = [
+        "default",
+        "test-vectors",
+        "performance",
+        "signature",
+        "miscellaneous",
+        "invalid",
+        "twist",
+        "degenerate",
+        "edge-cases",
+        "cofactor",
+        "composite",
+        "wrong"
+    ]
+
+
+    if library is None:
+        libraries2test = libraries
+    else:
+        libraries2test = [library]
+
+    if suite is None:
+        suites2test = suites
+    else:
+        suites2test = [suite]
+
+    for library in libraries2test:
+        with open(f"./nix/{library}_pkg_versions.json", "r") as f:
+            versions = list(json.load(f).keys())
+        for version in versions:
+            for suite in suites2test:
+                test_library(library, suite, version)
+
+
+if __name__ == '__main__':
+    main()