#!/bin/bash
# SPDX-License-Identifier: BSD-3-Clause
# Copyright (c) 2026 Open-IOV contributors
# inject_maps.sh - while cdt runs, collect pushbuffer/ring VAs from the host
# dmesg (sema-chan entry logs) plus channel GPFIFO ring VAs, translate them
# through the guest's own pagemap, and inject the VA->PA pairs into the
# service's learned map with sema_map_add.
#
# Usage: tools/guest-diag/inject_maps.sh [iterations]
set -u
cd "$(dirname "$0")/../.."
ITERS="${1:-20}"
LOG=/tmp/inject_maps.log
: > "$LOG"

map_add() {   # map_add <va-hex> <pa-hex>
    echo "0x$1,0x$2" | sudo tee /sys/module/vgsp/parameters/sema_map_add >/dev/null 2>&1
}

inject_list() {  # inject_list <va...>
    local vas="$*" chunk out n=0
    [ -n "${vas// /}" ] || return 0
    # chunk to keep the guest agent output under its size cap
    chunk=""
    for va in $vas; do
        chunk="$chunk $va"
        if [ "$(echo "$chunk" | wc -w)" -ge 12 ]; then
            out=$(tools/guest-diag/pmap_va.sh cdt $chunk 2>/dev/null | grep '^VA=')
            echo "$out" | while read -r line; do
                local v p
                v=$(echo "$line" | sed -n 's/.*VA=0x\([0-9a-f]*\).*/\1/p')
                p=$(echo "$line" | sed -n 's/.*present=1.*PA=0x\([0-9a-f]*\).*/\1/p')
                [ -n "$v" ] && [ -n "$p" ] && { map_add "$v" "$p"; echo "add 0x$v -> 0x$p" >> "$LOG"; }
            done
            chunk=""
        fi
    done
    if [ -n "${chunk// /}" ]; then
        out=$(tools/guest-diag/pmap_va.sh cdt $chunk 2>/dev/null | grep '^VA=')
        echo "$out" | while read -r line; do
            local v p
            v=$(echo "$line" | sed -n 's/.*VA=0x\([0-9a-f]*\).*/\1/p')
            p=$(echo "$line" | sed -n 's/.*present=1.*PA=0x\([0-9a-f]*\).*/\1/p')
            [ -n "$v" ] && [ -n "$p" ] && { map_add "$v" "$p"; echo "add 0x$v -> 0x$p" >> "$LOG"; }
        done
    fi
}

i=0
while [ "$i" -lt "$ITERS" ]; do
    i=$((i+1))
    PID=$(timeout 15 ./scripts/vgpu_vm.py exec --timeout 10 'pgrep -x cdt | head -1' 2>/dev/null | tail -1)
    [ -n "$PID" ] || { echo "cdt gone at iter $i" >> "$LOG"; break; }

    RINGS=$(sudo dmesg | grep -aoE "gpfifo=0x[0-9a-f]+" | grep -aoE "0x[0-9a-f]+" | sort -u | tr '\n' ' ')
    RINGVAS=$(sudo dmesg | grep -aoE "entry\[[0-9]+\] raw=[0-9a-f]+ va=0x[0-9a-f]+ len=[0-9]+" | grep -aoE "va=0x[0-9a-f]+" | grep -aoE "0x[0-9a-f]+" | sort -u | tr '\n' ' ')

    echo "== iter $i: rings=[$RINGS] entries=[$RINGVAS]" >> "$LOG"
    inject_list $RINGS
    inject_list $RINGVAS

    DONE=$(sudo dmesg | grep -aoE "gpfifo done=[0-9]+ nomap=[0-9]+" | tail -1)
    TAIL=$(timeout 15 ./scripts/vgpu_vm.py exec --timeout 10 'tail -1 /tmp/cdt.out' 2>/dev/null | tail -1)
    echo "   $DONE | cdt: $TAIL" >> "$LOG"
    case "$TAIL" in
        *cuCtxCreate*FAILED*|*ALL*|*cuMemAlloc*) echo "cdt moved on: $TAIL" >> "$LOG"; break;;
    esac
    sleep 10
done
echo "INJECT_DONE" >> "$LOG"