1 files changed, 37 insertions, 0 deletions
diff --git a/tools/guest-diag/guest_pmap.py b/tools/guest-diag/guest_pmap.py
new file mode 100644
index 0000000..96b7ddc
--- /dev/null
+++ b/tools/guest-diag/guest_pmap.py
@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+# guest-side helper: translate GPU/user VAs of a process through its pagemap.
+# Usage: sudo python3 guest_pmap.py <pid> <hex-va> [hex-va ...]
+import os
+import sys
+
+pid = int(sys.argv[1])
+vas = [int(x, 16) for x in sys.argv[2:]]
+
+maps = []
+try:
+    with open('/proc/%d/maps' % pid) as f:
+        for line in f:
+            parts = line.split()
+            a, b = parts[0].split('-')
+            maps.append((int(a, 16), int(b, 16), line.strip()))
+except OSError as e:
+    print('MAPS_ERROR: %s' % e)
+
+def in_maps(va):
+    for a, b, line in maps:
+        if a <= va < b:
+            return line
+    return None
+
+fd = os.open('/proc/%d/pagemap' % pid, os.O_RDONLY)
+try:
+    for va in vas:
+        os.lseek(fd, (va // 4096) * 8, 0)
+        e = int.from_bytes(os.read(fd, 8), 'little')
+        present = (e >> 63) & 1
+        pfn = e & ((1 << 55) - 1)
+        pa = (pfn << 12) | (va & 0xfff)
+        line = in_maps(va)
+        print('VA=0x%x present=%d PA=0x%x map=%s' % (va, present, pa, line if line else '-'))
+finally:
+    os.close(fd)