mirror of
https://github.com/henrydcase/pqc.git
synced 2024-11-29 19:01:22 +00:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import subprocess
|
|
import sys
|
|
|
|
"""
|
|
For a given SCHEME, this script verifies that all exported symbols are properly
|
|
namespaced, i.e., all start with "PQCLEAN_SCHEMENAME_"
|
|
"""
|
|
|
|
if len(sys.argv) != 2:
|
|
print("Provide a scheme name (e.g. crypto_kem/kyber768) as argv[1]")
|
|
exit(1)
|
|
|
|
SCHEME = sys.argv[1]
|
|
SCHEMEFULL = SCHEME.replace('/', '_') # e.g. crypto_kem_kyber768
|
|
SCHEMESHORT = SCHEME.split('/')[1].upper()
|
|
namespace = f"PQCLEAN_{SCHEMESHORT}_".replace('-', '')
|
|
|
|
# TODO can we do this using object files instead, to preserve file origin?
|
|
sharedlib = "bin/shared_{}_clean.so".format(SCHEMEFULL)
|
|
subprocess.run(["make", sharedlib, "SCHEME={}".format(SCHEME)])
|
|
p = subprocess.run(["nm", "-D", sharedlib], capture_output=True)
|
|
|
|
symbols = p.stdout.decode('utf-8').strip().split("\n")
|
|
non_namespaced = []
|
|
|
|
for symbolstr in symbols:
|
|
*_, symtype, symbol = symbolstr.split()
|
|
if symtype in 'TR':
|
|
if not symbol.startswith(namespace):
|
|
non_namespaced.append(symbol)
|
|
|
|
if non_namespaced:
|
|
print("! Not all symbols were properly namespaced.", file=sys.stderr)
|
|
print("! Missing namespace literal {}".format(namespace), file=sys.stderr)
|
|
for symbol in non_namespaced:
|
|
print("\t{}".format(symbol), file=sys.stderr)
|
|
sys.exit(1)
|