run-json.py

#! /usr/bin/python3

import json
from subprocess import Popen, PIPE
import sys

debug = False

def docmd(p, str, wait=True):
    p.stdin.write((str + "\r\n").encode('ascii'))

    while wait:
        line = p.stdout.readline()

        line = line.decode('ascii', 'ignore').rstrip('\n')

        if debug:
            print(line)

        if len(line) < 5:
            continue

        idx = line.find('<SET')
        if idx != -1:
            return line[idx + 5:]

        idx = line.find('>GET')
        if idx != -1:
            return line[idx + 5:]

process = Popen(['dotnet', 'run', '-c', 'Debug', '--', '-d', '-P', '-l', 'logfile.dat', '-x', 'blank', '-B'], stdin=PIPE, stdout=PIPE, stderr=PIPE, bufsize=0)

process.stdin.write('echo\r\n'.encode('ascii'))  # disable echo

test_file = sys.argv[1]

sub_cmd = test_file[-7] == '.' and test_file[-6].isdigit() and test_file[-5:] == '.json'

j = json.loads(open(test_file, 'rb').read())

jm = json.loads(open(sys.argv[2], 'rb').read())

for set in j:
    if not 'name' in set:
        continue

    print(set['name'])

    process.stdin.write(f'reset\r\n'.encode('ascii'))

    process.stdin.write(f'dolog {set["name"]}\r\n'.encode('ascii'))

    b = set['bytes']

    flags_mask = 65535

    byte_offset = 0

    while b[byte_offset] in (0x26, 0x2e, 0x36, 0x3e, 0xf2, 0xf3):
        byte_offset += 1

    first_byte = f'{b[byte_offset]:02X}'

    if sub_cmd:
        second_byte = b[byte_offset + 1]
        reg = f'{(second_byte >> 3) & 7}'

        if 'flags-mask' in jm[first_byte]['reg'][reg]:
            flags_mask = int(jm[first_byte]['reg'][reg]['flags-mask'])

    else:
        if first_byte in jm:
            if 'flags-mask' in jm[first_byte]:
                flags_mask = int(jm[first_byte]['flags-mask'])

    initial = set['initial']

    # set registers
    regs = initial['regs']
    for reg in regs:
        docmd(process, f'set reg {reg} {regs[reg]}')

    # fill ram
    for addr, value in initial['ram']:
        docmd(process, f'set ram {addr} {value}')

    docmd(process, 'S', False)  # step

    # verify
    final = set['final']

    ok = True

    # verify registers
    regs = final['regs']
    is_ = dict()
    for reg in regs:
        result = int(docmd(process, f'get reg {reg}').split()[1])
        is_[reg] = result

        compare = regs[reg]

        result &= flags_mask
        compare &= flags_mask

        if result != compare:
            print(f' *** {reg} failed ***')
            print(f': {result} (emulator) != {regs[reg]} (test set)')
            ok = False

    # verify mem
    for addr, value in final['ram']:
        result = int(docmd(process, f'get ram {addr}').split()[1])

        if result != value:
            print(f' *** {addr} failed ***')
            print(f': {result} (emulator) != {value} (test set)')
            ok = False

    if not ok:
        for reg in regs:
            print(f'{reg} was at start {initial["regs"][reg]}, should have become {final["regs"][reg]}, is: {is_[reg]}')

        print(f'Test set: {sys.argv[1]}')

        process.stdin.write(f'q\r\n'.encode('ascii'))

        sys.exit(1)

sys.exit(0)