m1n1.trace.agx: Log structure UAT permissions as meta

Signed-off-by: Asahi Lina <lina@asahilina.net>

proxyclient/m1n1/constructutils.py

@@ -347,7 +347,12 @@ class ConstructClassBase(Reloadable, metaclass=ReloadableConstructMeta):
         self._apply(obj)
 
         if self._addr > 0x10000:
-            g_struct_trace.add((self._addr, f"{cls.name} (end: {self._addr + size:#x})"))
+            desc = f"{cls.name} (end: {self._addr + size:#x})"
+            if getattr(stream, "meta_fn", None):
+                meta = stream.meta_fn(self._addr, None)
+                if meta is not None:
+                    desc += " " + meta
+            g_struct_trace.add((self._addr, desc))
             g_struct_addrmap[self._addr] = f"{cls.name}"
         return self
 

proxyclient/m1n1/fw/agx/channels.py

@@ -483,7 +483,7 @@ class Channel(Reloadable):
         assert index < count
         addr = self.rb_base[ring] + index * size
         stream = self.uat.iostream(0, addr)
-        stream.meta_fn = meta_fn
+        stream.meta_fn = lambda a, b: meta_fn(0, a, b)
         return msgcls.parse_stream(stream)
 
     def clear_message(self, ring, index):

proxyclient/m1n1/trace/agx.py

@@ -586,7 +586,11 @@ class AGXTracer(ASCTracer):
         self.vmcnt += 1
         #self.mon.poll()
 
-    def meta_gpuvm(self, iova, size):
+    def meta_gpuvm(self, ctx, iova, size=None):
+        if size is None:
+            pte = self.uat.ioperm(ctx, iova)
+            return f"PTE: {pte.describe()}"
+
         meta = ""
         iova &= 0xfffffffffff
         for off in range(size):
@@ -605,7 +609,7 @@ class AGXTracer(ASCTracer):
 
     def get_stream(self, context, off):
         stream = self.uat.iostream(context, off)
-        stream.meta_fn = self.meta_gpuvm
+        stream.meta_fn = lambda a, b: self.meta_gpuvm(context, a, b)
         return stream
 
     def start(self):