#!/bin/bash
# vmcycle.sh - reload cycle for the emulated vGPU stack:
#   stop QEMU -> remove mdevs -> rmmod/insmod open-iov-vfio.ko -> create a
#   fresh mdev -> patch its UUID into the runtime launch script -> relaunch.
#
# ALL artifacts stay inside this directory (tools/guest-diag/) so nothing is
# lost on reboot:
#   launch_qemu_rootport.sh     - template QEMU command line (root-port fix)
#   launch_qemu_rootport.run.sh - runtime copy with the current mdev UUID
#   run/qemu.log                - QEMU stderr/stdout of the last launch
#
# Usage:  tools/guest-diag/vmcycle.sh
# Env:    MDEV_PARENT (default: 0000:01:00.0)
#         MDEV_TYPE   (default: nvidia-nvidia-vgpu-turing)
set -e

HERE="$(cd "$(dirname "$0")" && pwd)"
TEMPLATE="$HERE/launch_qemu_rootport.sh"
QEMU_SCRIPT="${QEMU_SCRIPT:-$HERE/launch_qemu_rootport.run.sh}"
MODULE="${MODULE:-$HERE/../../kernel/open-iov-vfio.ko}"
MDEV_PARENT="${MDEV_PARENT:-0000:01:00.0}"
MDEV_TYPE="${MDEV_TYPE:-nvidia-nvidia-vgpu-turing}"
RUNDIR="$HERE/run"

mkdir -p "$RUNDIR"
[ -f "$QEMU_SCRIPT" ] || cp "$TEMPLATE" "$QEMU_SCRIPT"

pkill -f qemu-system-x86_64 2>/dev/null || true
sleep 4
for m in /sys/bus/mdev/devices/*; do
    [ -d "$m" ] && echo 1 | sudo tee "$m/remove" >/dev/null
done
sudo rmmod open_iov_vfio 2>/dev/null || true
sudo insmod "$MODULE"
U=$(uuidgen)
echo "$U" | sudo tee "/sys/class/mdev_bus/$MDEV_PARENT/mdev_supported_types/$MDEV_TYPE/create" >/dev/null
sed -i "s|/sys/bus/mdev/devices/[0-9a-f-]*|/sys/bus/mdev/devices/$U|" "$QEMU_SCRIPT"
mkdir -p /tmp/sockets/e726ba0b-204f-4ec9-91da-983ed7f41d06
rm -f /tmp/qemu-guest-agent.sock /tmp/qemu-qmp.sock
(setsid nohup "$QEMU_SCRIPT" > "$RUNDIR/qemu.log" 2>&1 &)
sleep 6
if pgrep -f qemu-system-x86_64 >/dev/null; then
    echo "VM up (mdev $U)  [qemu log: $RUNDIR/qemu.log]"
else
    echo "VM FAILED"; tail -3 "$RUNDIR/qemu.log"; exit 1
fi