#!/bin/sh
# watch_ctx.sh - guest-side watcher: (re)start cdt, snapshot the ioctl stream
# (fd paths), attach gdb to the libcuda wait as soon as cuCtxCreate is reached
# and POKE the per-entry trackers (p+0x9410+0x10/0x18 = 1) to test whether that
# gate releases the wait. Runs under sudo; results land in /tmp/ctxdumps/.
mkdir -p /tmp/ctxdumps
rm -f /tmp/ctxdumps/*
nvidia-modprobe -c 0
nvidia-modprobe -u
modprobe nvidia-uvm
rm -f /tmp/cdt.out
setsid /tmp/cdt > /tmp/cdt.out 2>&1 &
i=0
while [ $i -lt 300 ]; do
    grep -q 'cuCtxCreate' /tmp/cdt.out 2>/dev/null && break
    sleep 0.3
    i=$((i+1))
done
P=$(pidof cdt | tr ' ' '\n' | head -1)
ls -l /proc/$P/fd/ > /tmp/ctxdumps/fds.txt 2>&1
if command -v strace >/dev/null 2>&1; then
    timeout 8 strace -f -y -p "$P" -e trace=ioctl -o /tmp/ctxdumps/ioctl.txt 2>/dev/null
fi
n=0
while [ $n -lt 80 ]; do
    P=$(pidof cdt | tr ' ' '\n' | head -1)
    [ -n "$P" ] || break
    gdb -p "$P" -batch -x /tmp/dumpctx.py > "/tmp/ctxdumps/d$n.txt" 2>&1
    if grep -q DONE_SPIN "/tmp/ctxdumps/d$n.txt"; then
        cp "/tmp/ctxdumps/d$n.txt" /tmp/ctxdumps/FOUND.txt
        break
    fi
    n=$((n+1))
    sleep 0.5
done
P=$(pidof cdt | tr ' ' '\n' | head -1)
if [ -f /tmp/ctxdumps/FOUND.txt ] && [ -n "$P" ]; then
    CTX_POKE=1 gdb -p "$P" -batch -x /tmp/dumpctx.py > /tmp/ctxdumps/poke_run.txt 2>&1
    k=0
    while [ $k -lt 12 ]; do
        echo "--- t=$((k*3))s $(tail -1 /tmp/cdt.out)" >> /tmp/ctxdumps/result.txt
        k=$((k+1))
        sleep 3
    done
    VAS=$(grep -oE 'type=1 p=0x[0-9a-f]+' /tmp/ctxdumps/FOUND.txt | sed 's/.*p=0x//' | while read -r p; do printf '0x%x ' $((0x$p + 0x9410)); done)
    echo "PID=$P TRACKER_VAS $VAS" > /tmp/ctxdumps/probe.txt
    python3 /tmp/pmap.py "$P" $VAS >> /tmp/ctxdumps/probe.txt 2>&1
fi
echo WATCH_DONE >> /tmp/ctxdumps/WATCH.log