from setup import *
import asm

code_len = 12 * 16 * 8 + 4
data_len = 8 * 16 * 8

code_buffer = p.malloc(code_len)
data_buffer = p.malloc(data_len)

template = asm.ARMAsm("""
    mov x2, x0
    mrs x2, s3_0_c0_c0_0
    str x2, [x1], #8
    ret
""", code_buffer)

mov, mrs, st, ret = struct.unpack("4I", template.data)

data = []

BAD = 0xacce5515abad1dea

for op1 in range(1 << 3):
    for CRn in (0b1011, 0b1111):
        mrs0 = mrs | (op1 << 16) | (CRn << 12)
        insns = []
        for CRm in range(1 << 4):
            for op2 in range(1 << 3):
                insns.extend((mov, mrs0 | (CRm << 8) | (op2 << 5), st))
        insns.append(ret)
        iface.writemem(code_buffer, struct.pack("<385I", *insns))
        p.dc_cvau(code_buffer, code_len)
        p.ic_ivau(code_buffer, code_len)

        p.set_exc_guard(p.GUARD_SILENT | p.GUARD_SKIP)
        p.call(code_buffer, BAD, data_buffer)
        cnt = p.get_exc_count()

        data = iface.readmem(data_buffer, data_len)
        d = struct.unpack("<128Q", data)
        i = 0
        for CRm in range(1 << 4):
            for op2 in range(1 << 3):
                v = d[i]
                if v != BAD:
                    print("s3_%d_c%d_c%d_%d (3, %d, %d, %d, %d) = 0x%x" % (
                        op1, CRn, CRm, op2, op1, CRn, CRm, op2, v))
                    try:
                        u.msr((3, op1, CRn, CRm, op2), v, silent=True)
                    except:
                        print(" - READONLY")
                    try:
                        u.mrs((3, op1, CRn, CRm, op2), silent=True, call=p.el1_call)
                    except:
                        print(" - ### EL2 only ###")
                    try:
                        u.mrs((3, op1, CRn, CRm, op2), silent=True, call=p.el0_call)
                    except:
                        pass
                    else:
                        print(" - *** EL0 accessible ***")
                i += 1

p.set_exc_guard(p.GUARD_OFF)