#!/usr/bin/env python3
"""Was the 'wall' an address boundary, or exhaustion caused by the sweep itself?

The sweep's failures began at PA 0x3ef00000, which was ALSO probe #1006. The
failing value was (PA>>16)+0x101 -- but since PA = n*0x100000, that is exactly
16*n + 257. Address-derived and count-derived are indistinguishable in that data.

A fresh process probes 0x3ff00000 and 0x40000000 fine, which already argues
against the address hypothesis. This settles it: probe ONE fixed, known-good
address repeatedly and see whether it degrades at the same count.

LOAD_TA allocates a session on every successful load and this probe never
unloads, so session exhaustion is the obvious candidate.
"""
import sys, time
sys.path.insert(0, "/root/psp/work")
from tinygrad import Device
import tinygrad.runtime.autogen.am.am as am

def main():
    adev = Device["AMD"].iface.dev_impl
    cmd_pa = adev.mm.palloc(0x10000, align=0x1000, zero=True, boot=False)
    mcc = adev.paddr2mc(cmd_pa)
    FIXED = 0x20000000                      # one address, known good, never changes
    def probe(pa):
        mc = adev.paddr2mc(pa)
        c = am.struct_psp_gfx_cmd_resp(cmd_id=am.GFX_CMD_ID_LOAD_TA)
        c.cmd.cmd_load_ta.app_phy_addr_lo = mc & 0xFFFFFFFF
        c.cmd.cmd_load_ta.app_phy_addr_hi = mc >> 32
        c.cmd.cmd_load_ta.app_len = 0x1000
        c.cmd.cmd_load_ta.cmd_buf_phy_addr_lo = mcc & 0xFFFFFFFF
        c.cmd.cmd_load_ta.cmd_buf_phy_addr_hi = mcc >> 32
        c.cmd.cmd_load_ta.cmd_buf_len = 0x10000
        t = time.time()
        try: st = adev.psp._ring_submit(c).resp.status
        except Exception as e:
            try: st = int(str(e).strip().split()[-1])
            except Exception: st = -1
        return st, time.time() - t
    print("probing ONE fixed address %#x repeatedly" % FIXED)
    first_bad = None
    for n in range(1, 1400):
        st, dt = probe(FIXED)
        if st != 0x34 or dt > 0.5:
            if first_bad is None:
                first_bad = n
                print("   FIRST DEVIATION at probe #%d: status=%#x (%.2fs)" % (n, st, dt))
                print("   -> 16*n+257 = %#x   (PA>>16)+0x101 = %#x"
                      % (16 * n + 257, (FIXED >> 16) + 0x101))
            if n > (first_bad + 3): break
        if n % 250 == 0: print("   ... probe #%d status=%#x %.3fs" % (n, st, dt))
    if first_bad is None:
        print("   no deviation in 1399 probes at a fixed address")
    print()
    print("=> %s" % ("COUNT-dependent: the 'wall' was exhaustion, not an address boundary"
                     if first_bad else "not count-dependent at this address"))
    return 0

sys.exit(main())