test 1.88 KB
Newer Older
Enrico Pozzobon committed
1 2 3 4
#!/usr/bin/env python3

import os
import sys
5
import pylink
6
import serial.tools.list_ports
7 8
from test_common import (
    DeviceUnderTestAeadUARTP,
lwc-tester committed
9
    eprint,
10
    FileMutex,
lwc-tester committed
11
    run_nist_lws_aead_test,
12
)
Enrico Pozzobon committed
13 14


15 16 17 18 19
def get_serial():
    ports = serial.tools.list_ports.comports()
    devices = [
        p.device
        for p in ports
lwc-tester committed
20
        if p.serial_number == '00000000'
21 22
    ]
    devices.sort()
23
    return devices[0]
24 25


26 27
class F7(DeviceUnderTestAeadUARTP):
    RAM_SIZE = 0x50000
lwc-tester committed
28

29 30 31 32 33
    def __init__(self, build_dir):
        DeviceUnderTestAeadUARTP.__init__(self)

        self.uart_device = get_serial()
        devname = os.path.basename(self.uart_device)
34 35 36 37
        self.ser = serial.Serial(
            self.uart_device,
            baudrate=115200,
            timeout=5)
38 39
        self.lock = FileMutex('/var/lock/lwc-compare.%s.lock' % devname)
        self.build_dir = build_dir
lwc-tester committed
40
        self.template_path = os.path.dirname(sys.argv[0])
41

42 43
        self.jlink = pylink.JLink()
        self.jlink.open(779340002)
44

lwc-tester committed
45 46 47 48
        self.firmware_path = os.path.join(
            build_dir, 'f7.bin')
        self.ram_pattern_path = os.path.join(
            self.template_path, 'ram_pattern.bin')
49

50 51 52
    def flash(self):
        jlink = self.jlink
        jlink.connect('STM32F746ZG')
lwc-tester committed
53 54 55 56
        jlink.flash_file(self.firmware_path, 0x8000000)
        eprint("Firmware flashed.")
        jlink.flash_file(self.ram_pattern_path, 0x20000000)
        eprint("RAM flashed.")
57 58
        jlink.reset()
        jlink.restart()
59

60 61 62
    def dump_ram(self):
        jlink = self.jlink
        return bytes(jlink.memory_read8(0x20000000, F7.RAM_SIZE))
63

Enrico Pozzobon committed
64

65
def main(argv):
lwc-tester committed
66 67
    if len(argv) != 2:
        print("Usage: test build_dir")
68
        return 1
Enrico Pozzobon committed
69

lwc-tester committed
70 71 72 73
    build_dir = argv[1]
    kat_path = os.path.join(build_dir, 'LWC_AEAD_KAT.txt')

    run_nist_lws_aead_test(kat_path, build_dir, 0x0008, F7)
lwc-tester committed
74
    return 0
Enrico Pozzobon committed
75 76 77 78


if __name__ == "__main__":
    sys.exit(main(sys.argv))