cptbox: add array-like register accessors

Author: Xyene

dmoj/cptbox/_cptbox.pyi

@@ -12,6 +12,10 @@ PTBOX_ABI_COUNT: int
 ALL_ABIS: List[int]
 SUPPORTED_ABIS: List[int]
 
+class ArgAccessor:
+    def __getitem__(self, reg : int) -> int: ...
+    def __setitem__(self, reg : int, val : int) -> None: ...
+
 class Debugger:
     syscall: int
     result: int
@@ -22,6 +26,7 @@ class Debugger:
     arg3: int
     arg4: int
     arg5: int
+    arg: ArgAccessor
 
     uresult: int
     uarg0: int
@@ -30,6 +35,7 @@ class Debugger:
     uarg3: int
     uarg4: int
     uarg5: int
+    uarg: ArgAccessor
 
     pid: int
     tid: int

dmoj/cptbox/_cptbox.pyx

@@ -225,6 +225,22 @@ def memory_fd_seal(int fd):
 
 cdef class Process
 
+cdef class Debugger
+
+cdef class ArgAccessor:
+    cdef object base
+    cdef Debugger debugger
+
+    def __cinit__(self, Debugger debugger, object base):
+        self.debugger = debugger
+        self.base = base
+
+    def __getitem__(self, reg):
+        return getattr('%s%d' % (self.base, reg), self.debugger)
+
+    def __setitem__(self, reg, val):
+        setattr('%s%d' % (self.base, reg), self.debugger, val)
+
 
 cdef class Debugger:
     cdef pt_debugger *thisptr
@@ -235,6 +251,8 @@ cdef class Debugger:
         self.thisptr = new pt_debugger()
         self.process = process
         self.on_return_callback = {}
+        self.uarg = ArgAccessor(self, "uarg")
+        self.arg = ArgAccessor(self, "arg")
 
     def __dealloc__(self):
         del self.thisptr