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