test_ecdsa_dbg.py

#!/usr/bin/env python3
import sys
import random as rn
import os

import test_common as tc
import models.p256 as p256

if __name__ == "__main__":

    defines_set = tc.get_main_defines()
    if "DEBUG_OPS" not in defines_set:
        tc.print_test_skipped("Debug ops are disabled.")
        sys.exit(0)

    args = tc.parser.parse_args()

    ops_cfg = tc.get_ops_config()
    test_name = "ecdsa_sign_dbg"
    run_name = test_name

    tc.print_run_name(test_name)

    if args.testvec != "":
        print(f"Reading test vector from {args.testvec}")
        data_dir, rng_list = tc.parse_testvec(args.testvec, tc.rng_luts[test_name])
        z = bytes.fromhex(data_dir["z"])
        sch = bytes.fromhex(data_dir["sch"])
        scn = bytes.fromhex(data_dir["scn"])
        d = data_dir["d"]
        w = bytes.fromhex(data_dir["w"])
        Ax = data_dir["Ax"]
        Ay = data_dir["Ay"]
    else:
        seed = tc.set_seed(args)
        rn.seed(seed)
        print("seed:", seed)
        # Generate test vector
        d, w, Ax, Ay = p256.key_gen(int.to_bytes(rn.randint(0, 2**256-1), 32, 'big'))
        sch = int.to_bytes(rn.randint(0, 2**256-1), 32, 'big')
        scn = int.to_bytes(rn.randint(0, 2**32-1), 4, 'little')
        z = int.to_bytes(rn.randint(0, 2**256-1), 32, 'big')

        rng_list = [rn.randint(0, 2**256-1) for i in range(16)]

    test_dir = tc.make_test_dir(test_name)
    cmd_file = tc.get_cmd_file(test_dir)

    tc.set_rng(test_dir, rng_list)

    r_ref, s_ref = p256.sign(d, w, sch, scn, z)

    signature_ref = r_ref.to_bytes(32, 'big') + s_ref.to_bytes(32, 'big')

    # Write Keys and inputs
    slot = rn.randint(0, 7)

    wint = int.from_bytes(w, 'big')
    tc.write_int256(cmd_file, d, 0x0040)
    tc.write_int256(cmd_file, wint, 0x0060)
    tc.write_int256(cmd_file, Ax, 0x0160)
    tc.write_int256(cmd_file, Ay, 0x0180)

    insrc = 0x0
    outsrc = 0x1

    tc.write_bytes(cmd_file, z, (insrc<<12) + 0x10)
    tc.write_bytes(cmd_file, sch, 0x00A0)
    tc.write_bytes(cmd_file, scn, 0x00C0)

    input_word = (slot << 8) + tc.find_in_list("ecdsa_sign", ops_cfg)["id"]

    tc.write_int32(cmd_file, input_word, (insrc<<12))

    # Run Op
    ctx = tc.run_op(
        cmd_file, "ecdsa_sign_dbg", insrc, outsrc, 0, ops_cfg, test_dir,
        run_name=run_name
    )

    SPECT_OP_STATUS, SPECT_OP_DATA_OUT_SIZE = tc.get_res_word(test_dir, run_name)

    if (SPECT_OP_STATUS):
        print("SPECT_OP_STATUS:", hex(SPECT_OP_STATUS))
        tc.print_failed()
        sys.exit(1)

    if (SPECT_OP_DATA_OUT_SIZE != 80):
        print("SPECT_OP_DATA_OUT_SIZE:", SPECT_OP_DATA_OUT_SIZE)
        tc.print_failed()
        sys.exit(1)

    # Read result
    l3_result = tc.read_output(test_dir, run_name, outsrc<<12, 1)
    l3_result &= 0xFF

    signature = tc.read_output(test_dir, run_name, (outsrc<<12) + 0x10, 16, string=True)

    if not(signature_ref == signature):
        tc.print_failed()
        sys.exit(1)

    tc.print_passed()

    if "TS_SPECT_FW_TEST_DONT_DUMP" in os.environ.keys():
        os.system(f"rm -r {test_dir}")

    sys.exit(0)