#!/usr/bin/env python3
# SPDX-License-Identifier: BSD-3-Clause
# Copyright (c) 2026 Open-IOV contributors
# 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 = 4096
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))