#!/usr/bin/env bash
set -euo pipefail

[[ $# -eq 3 ]] || {
  echo "usage: test-native-physical-probe.sh <candidate-image> <probe-stage> <receipt-output>" >&2
  exit 64
}

candidate=$(readlink -f "$1")
probe_stage=$2
receipt=$(readlink -m "$3")
[[ ${probe_stage} =~ ^[1-9]$ ]]
source_root=$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)
native_root=${source_root}/native/x86_64-bios
test_root=$(mktemp -d)
serial_log=${test_root}/serial.log
disk_image=${test_root}/physical-probe.img
block_image=${test_root}/virtio-block.img
trap 'rm -rf "${test_root}"' EXIT

truncate -s 2M "${disk_image}"
truncate -s 8M "${block_image}"
nasm -f bin "${native_root}/physical-test-mbr.asm" \
  -o "${test_root}/physical-test-mbr.bin"
dd if="${test_root}/physical-test-mbr.bin" of="${disk_image}" \
  bs=512 seek=0 conv=notrunc status=none
dd if="${candidate}" of="${disk_image}" \
  bs=512 seek=34 conv=notrunc status=none

set +e
timeout 20 qemu-system-x86_64 \
  -machine pc,accel=tcg \
  -m 64M \
  -drive "format=raw,file=${disk_image}" \
  -drive "if=none,id=ghblk,format=raw,file=${block_image}" \
  -device virtio-blk-pci,drive=ghblk,disable-modern=on \
  -netdev user,id=ghnet \
  -device virtio-net-pci,netdev=ghnet,disable-modern=on,mac=52:54:00:26:71:98 \
  -display none \
  -monitor none \
  -serial "file:${serial_log}" \
  -device isa-debug-exit,iobase=0xf4,iosize=0x04
qemu_status=$?
set -e

[[ ${qemu_status} -eq 33 ]]
grep -q '^GHOS_NATIVE_KERNEL_ENTERED=true' "${serial_log}"
grep -q '^GHOS_DISK_PROOF_WRITTEN=LBA63' "${serial_log}"
grep -q '^GHOS_DISK_PROOF_OBSERVED_AFTER_RESET=LBA63' "${serial_log}"
python3 - "${disk_image}" "${probe_stage}" <<'PY'
import pathlib
import sys

disk = pathlib.Path(sys.argv[1]).read_bytes()
stage = int(sys.argv[2])
proof = disk[63 * 512:64 * 512]
assert proof[0] == 0xA5
assert proof[1:].startswith(b"GHOS_NATIVE_LONG64_DISK_PROOF\x00")
assert proof[42] == 0x80 + stage
assert proof[43] == 0
assert proof[32:34] == bytes([1, 1])
assert int.from_bytes(proof[50:52], "little") > 0
assert int.from_bytes(proof[52:54], "little") > 0
if stage >= 5:
    assert proof[36:42] == bytes.fromhex("525400267198")
if stage >= 6:
    assert int.from_bytes(proof[44:46], "little") > 0
if stage >= 7:
    assert int.from_bytes(proof[46:48], "little") > 0
    assert proof[34] == 1
if stage >= 9:
    assert int.from_bytes(proof[48:50], "little") > 0
    assert proof[35] == 1
PY

observed_at=$(date --iso-8601=seconds)
image_sha=$(sha256sum "${candidate}" | awk '{print $1}')
cat >"${receipt}" <<EOF
schema: guanghu.ghal-physical-probe-test/v1
receipt_id: GH-OS-LAB-001-GHAL-PROBE-${probe_stage}-QEMU
status: VERIFIED
observed_at: ${observed_at}
probe_stage: ${probe_stage}
image_sha256: ${image_sha}
proof:
  flag: 0xa5
  completion_stage: 0x8${probe_stage}
  disk_lba: 63
  disk_receipt_observed_after_reset: true
acceptance:
  qemu_probe_return: true
  physical_server_probe: false
next_action: RUN_ONE_TIME_PHYSICAL_PROBE_STAGE_${probe_stage}
EOF
cat "${serial_log}" >>"${receipt}.serial.log"