import subprocess
import serial
from mupq import mupq
class M4Settings(mupq.PlatformSettings):
#: Specify folders to include
scheme_folders = [ # mupq.PlatformSettings.scheme_folders + [
('pqm4', 'crypto_kem', ''),
('pqm4', 'crypto_sign', ''),
('mupq', 'mupq/crypto_kem', ''),
('mupq', 'mupq/crypto_sign', ''),
('pqclean', 'mupq/pqclean/crypto_kem', "PQCLEAN"),
('pqclean', 'mupq/pqclean/crypto_sign', "PQCLEAN"),
]
#: List of dicts, in each dict specify (Scheme class) attributes of the
#: scheme with values, if all attributes match the scheme is skipped.
skip_list = (
{'scheme': 'falcon-1024-tree', 'implementation': 'opt-leaktime'},
{'scheme': 'falcon-1024-tree', 'implementation': 'opt-ct'},
{'scheme': 'frodokem640aes', 'implementation': 'clean'},
{'scheme': 'frodokem640aes', 'implementation': 'opt'},
{'scheme': 'frodokem976aes', 'implementation': 'clean'},
{'scheme': 'frodokem976aes', 'implementation': 'opt'},
{'scheme': 'frodokem1344aes', 'implementation': 'clean'},
{'scheme': 'frodokem1344aes', 'implementation': 'opt'},
{'scheme': 'frodokem640shake', 'implementation': 'clean'},
{'scheme': 'frodokem976shake', 'implementation': 'clean'},
{'scheme': 'frodokem976shake', 'implementation': 'opt'},
{'scheme': 'frodokem1344shake', 'implementation': 'clean'},
{'scheme': 'frodokem1344shake', 'implementation': 'opt'},
{'scheme': 'mqdss-48', 'implementation': 'clean'},
{'scheme': 'mqdss-64', 'implementation': 'clean'},
{'scheme': 'ledakemlt12', 'implementation': 'leaktime'},
{'scheme': 'ledakemlt32', 'implementation': 'leaktime'},
{'scheme': 'ledakemlt52', 'implementation': 'leaktime'},
{'scheme': 'luov-79-76-341-chacha', 'implementation': 'ref'},
{'scheme': 'luov-79-76-341-keccak', 'implementation': 'ref'},
{'scheme': 'luov-7-83-283-chacha', 'implementation': 'ref'},
{'scheme': 'luov-7-83-283-keccak', 'implementation': 'ref'},
{'scheme': 'luov-7-110-374-chacha', 'implementation': 'ref'},
{'scheme': 'luov-7-110-374-keccak', 'implementation': 'ref'},
{'scheme': 'rainbowIa-classic', 'implementation': 'clean'},
{'scheme': 'rainbowIa-cyclic', 'implementation': 'clean'},
{'scheme': 'rainbowIa-cyclic-compressed', 'implementation': 'clean'},
{'scheme': 'rainbowIIIc-classic', 'implementation': 'clean'},
{'scheme': 'rainbowIIIc-cyclic', 'implementation': 'clean'},
{'scheme': 'rainbowIIIc-cyclic-compressed', 'implementation': 'clean'},
{'scheme': 'rainbowVc-classic', 'implementation': 'clean'},
{'scheme': 'rainbowVc-cyclic', 'implementation': 'clean'},
{'scheme': 'rainbowVc-cyclic-compressed', 'implementation': 'clean'},
{'scheme': 'qtesla-p-I', 'implementation': 'clean'},
{'scheme': 'qtesla-p-III', 'implementation': 'clean'},
{'scheme': 'mceliece348864', 'implementation': 'clean'},
{'scheme': 'mceliece348864f', 'implementation': 'clean'},
{'scheme': 'mceliece460896', 'implementation': 'clean'},
{'scheme': 'mceliece460896f', 'implementation': 'clean'},
{'scheme': 'mceliece6688128', 'implementation': 'clean'},
{'scheme': 'mceliece6688128f', 'implementation': 'clean'},
{'scheme': 'mceliece6960119', 'implementation': 'clean'},
{'scheme': 'mceliece6960119f', 'implementation': 'clean'},
{'scheme': 'mceliece8192128', 'implementation': 'clean'},
{'scheme': 'mceliece8192128f', 'implementation': 'clean'},
{'scheme': 'mceliece348864', 'implementation': 'vec'},
{'scheme': 'mceliece348864f', 'implementation': 'vec'},
{'scheme': 'mceliece460896', 'implementation': 'vec'},
{'scheme': 'mceliece460896f', 'implementation': 'vec'},
{'scheme': 'mceliece6688128', 'implementation': 'vec'},
{'scheme': 'mceliece6688128f', 'implementation': 'vec'},
{'scheme': 'mceliece6960119', 'implementation': 'vec'},
{'scheme': 'mceliece6960119f', 'implementation': 'vec'},
{'scheme': 'mceliece8192128', 'implementation': 'vec'},
{'scheme': 'mceliece8192128f', 'implementation': 'vec'},
{'scheme': 'hqc-128-1-cca2', 'implementation': 'leaktime'},
{'scheme': 'hqc-192-1-cca2', 'implementation': 'leaktime'},
{'scheme': 'hqc-192-2-cca2', 'implementation': 'leaktime'},
{'scheme': 'hqc-256-1-cca2', 'implementation': 'leaktime'},
{'scheme': 'hqc-256-2-cca2', 'implementation': 'leaktime'},
{'scheme': 'hqc-256-3-cca2', 'implementation': 'leaktime'},
{'scheme': 'rqc256', 'implementation': 'ref'},
)
class M4(mupq.Platform):
def __enter__(self):
self._dev = serial.Serial("/dev/ttyUSB0", 115200, timeout=10)
return super().__enter__()
def __exit__(self,*args, **kwargs):
self._dev.close()
return super().__exit__(*args, **kwargs)
def device(self):
return self._dev
def flash(self, binary_path):
super().flash(binary_path)
subprocess.check_call(
["st-flash", "--reset", "write", binary_path, "0x8000000"],
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL
)