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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#!/usr/bin/env python3
# guest_pmap_win.py - dump present pages of a process's VA range through the
# guest's own pagemap (root in the guest). Batched reads (512 entries per
# pread) so a 256 MB window takes about a second, not minutes.
#
# Usage: sudo python3 guest_pmap_win.py <pid> <lo-hex> <hi-hex> [max-pairs] [pa-max-hex]
# Output: one "0x<va>,0x<pa>,<npages>" line per contiguous run (ascending VA),
# capped at max-pairs pages. PAs of 0 are skipped; pages at/above pa-max are
# skipped unless pa-max is 0.
import os
import sys
pid = int(sys.argv[1])
lo = int(sys.argv[2], 16) & ~0xFFF
hi = int(sys.argv[3], 16)
maxp = int(sys.argv[4]) if len(sys.argv) > 4 else 20000
pa_max = int(sys.argv[5], 16) if len(sys.argv) > 5 else 0
BATCH = 512
out = []
run_va = run_pa = None
run_n = 0
pairs = 0
def flush():
if run_n:
out.append('0x%x,0x%x,%d' % (run_va, run_pa, run_n))
fd = os.open('/proc/%d/pagemap' % pid, os.O_RDONLY)
try:
va = lo
while va < hi:
n = min(BATCH, (hi - va) // 0x1000)
if n <= 0:
break
os.lseek(fd, (va // 4096) * 8, 0)
data = os.read(fd, n * 8)
got = len(data) // 8
for i in range(got):
e = int.from_bytes(data[i * 8:i * 8 + 8], 'little')
page_va = va + i * 0x1000
if not ((e >> 63) & 1):
flush()
run_va = run_pa = None
run_n = 0
continue
pfn = e & ((1 << 55) - 1)
pa = pfn << 12
if not pa or (pa_max and pa >= pa_max):
flush()
run_va = run_pa = None
run_n = 0
continue
if (run_va is not None and page_va == run_va + run_n * 0x1000
and pa == run_pa + run_n * 0x1000):
run_n += 1
else:
flush()
run_va, run_pa, run_n = page_va, pa, 1
pairs += 1
if pairs >= maxp:
flush()
print('\n'.join(out))
sys.exit(0)
va += got * 0x1000
finally:
os.close(fd)
flush()
print('\n'.join(out))
|
