#!/usr/bin/env python3

import os
import sys
import pylink
import serial.tools.list_ports
from test_common import (
    DeviceUnderTestAeadUARTP,
    eprint,
    FileMutex,
    run_nist_lws_aead_test,
)


def get_serial():
    ports = serial.tools.list_ports.comports()
    devices = [
        p.device
        for p in ports
        if p.serial_number == 'FT2XA9MY'
    ]
    devices.sort()
    return serial.Serial(
        devices[0],
        baudrate=115200,
        timeout=5)


class F7(DeviceUnderTestAeadUARTP):
    RAM_SIZE = 0x50000

    def __init__(self, build_dir):
        DeviceUnderTestAeadUARTP.__init__(self)

        self.uart_device = get_serial()
        devname = os.path.basename(self.uart_device)
        self.lock = FileMutex('/var/lock/lwc-compare.%s.lock' % devname)
        self.build_dir = build_dir

        self.jlink = pylink.JLink()
        self.jlink.open(779340002)

        self.firmware_path = os.path.join(build_dir, 'build', 'f7.bin')
        self.ram_pattern_path = os.path.join(build_dir, 'ram_pattern.bin')

    def flash(self):
        jlink = self.jlink
        jlink.connect('STM32F746ZG')
        jlink.flash_file(self.firmware_path, 0x8000000)
        eprint("Firmware flashed.")
        jlink.flash_file(self.ram_pattern_path, 0x20000000)
        eprint("RAM flashed.")
        jlink.reset()
        jlink.restart()

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


def main(argv):
    if len(argv) != 3:
        print("Usage: test LWC_AEAD_KAT.txt build_dir")
        return 1

    run_nist_lws_aead_test(argv[1], argv[2], 0x0008, F7)
    return 0


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