1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#!/usr/bin/env python3
# guest-side helper: dump present pages of the target's /dev/nvidia* mappings
# as "va,pa" lines (root in the guest), starting at or above va-min and
# stopping after max-pairs lines so the guest agent's output stays small.
# Usage: sudo python3 guest_pmap_all.py <pid> [va-min-hex] [max-pairs]
import os
import sys
pid = int(sys.argv[1])
va_min = int(sys.argv[2], 16) if len(sys.argv) > 2 else 0
maxp = int(sys.argv[3]) if len(sys.argv) > 3 else 300
regions = []
with open('/proc/%d/maps' % pid) as f:
for line in f:
parts = line.split()
a, b = parts[0].split('-')
lo, hi = int(a, 16), int(b, 16)
path = parts[5] if len(parts) >= 6 else ''
if 'nvidia' in path:
regions.append((max(lo, va_min), hi))
fd = os.open('/proc/%d/pagemap' % pid, os.O_RDONLY)
n = 0
try:
for lo, hi in regions:
for va in range(lo, hi, 0x1000):
os.lseek(fd, (va // 4096) * 8, 0)
e = int.from_bytes(os.read(fd, 8), 'little')
if not ((e >> 63) & 1):
continue
pfn = e & ((1 << 55) - 1)
print('0x%x,0x%x' % (va, (pfn << 12)))
n += 1
if n >= maxp:
raise SystemExit
finally:
os.close(fd)
|
