1 files changed, 109 insertions, 0 deletions
diff --git a/tools/proxyclient/experiments/timer_test.py b/tools/proxyclient/experiments/timer_test.py
new file mode 100755
index 0000000..b3422c9
--- /dev/null
+++ b/tools/proxyclient/experiments/timer_test.py
@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+# SPDX-License-Identifier: MIT
+import sys, pathlib
+sys.path.append(str(pathlib.Path(__file__).resolve().parents[1]))
+
+from m1n1.setup import *
+
+HV_VTMR_CTL = (3, 5, 15, 1, 3)
+HV_VTMR_CTL_VMASK = (1 << 0)
+HV_VTMR_CTL_PMASK = (1 << 1)
+
+HV_VTMR_LIST = (3, 5, 15, 1, 2)
+
+TGE = (1<<27)
+
+u.msr(CNTHCTL_EL2, 3 << 10) # EL1PTEN | EL1PCTEN
+
+def run_test(ctl, tval):
+    u.inst(0xd5033fdf) # isb
+
+    u.msr(ctl, 0)
+    u.msr(tval, int(freq * 0.8))
+    u.msr(ctl, 1)
+
+    for i in range(6):
+        p.nop()
+        time.sleep(0.2)
+        #u.inst(0xd5033fdf, call=p.el1_call)
+        print("      . (ISR_EL1=%d) CTL=%x VTMR_LIST=%x" % (u.mrs(ISR_EL1), u.mrs(ctl), u.mrs(HV_VTMR_LIST)))
+
+    u.msr(ctl, 0)
+
+def test_hv_timers():
+    u.msr(DAIF, 0x3c0)
+    print("Testing HV timers...")
+    print("  TGE = 1")
+
+    u.msr(HCR_EL2, u.mrs(HCR_EL2) | TGE | (1 << 3) | (1 << 4))
+
+    print("    P:")
+    run_test(CNTP_CTL_EL0, CNTP_TVAL_EL0)
+    print("    V:")
+    run_test(CNTV_CTL_EL0, CNTV_TVAL_EL0)
+
+def test_guest_timers():
+    u.msr(DAIF, 0)
+    print("Testing guest timers...")
+
+    print("  TGE = 1, vGIC mode=0, timers unmasked")
+    u.msr(HCR_EL2, (u.mrs(HCR_EL2) | TGE) | (1 << 3) | (1 << 4))
+    u.msr(HACR_EL2, 0)
+    u.msr(HV_VTMR_CTL, 3)
+
+    print("    P:")
+    #run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    #run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    print("  TGE = 1, vGIC mode=0, timers masked")
+    u.msr(HV_VTMR_CTL, 0)
+
+    print("    P:")
+    run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    print("  TGE = 0, vGIC mode=0, timers unmasked")
+    u.msr(HCR_EL2, (u.mrs(HCR_EL2) & ~TGE) | (1 << 3) | (1 << 4))
+    u.msr(HACR_EL2, 0)
+    u.msr(HV_VTMR_CTL, 3)
+
+    print("    P:")
+    run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    print("  TGE = 0, vGIC mode=0, timers masked")
+    u.msr(HV_VTMR_CTL, 0)
+
+    print("    P:")
+    run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    print("  TGE = 0, vGIC mode=1, timers unmasked")
+    u.msr(HCR_EL2, (u.mrs(HCR_EL2) & ~TGE) | (1 << 3) | (1 << 4))
+    u.msr(HACR_EL2, 1<<20)
+    u.msr(HV_VTMR_CTL, 3)
+
+    print("    P:")
+    run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    print("  TGE = 0, vGIC mode=1, timers masked")
+    u.msr(HV_VTMR_CTL, 0)
+
+    print("    P:")
+    run_test(CNTP_CTL_EL02, CNTP_TVAL_EL02)
+    print("    V:")
+    run_test(CNTV_CTL_EL02, CNTV_TVAL_EL02)
+
+    return
+
+freq = u.mrs(CNTFRQ_EL0)
+print("Timer freq: %d" % freq)
+
+test_hv_timers()
+test_guest_timers()