#!/bin/bash
# SPDX-License-Identifier: AGPL-3.0-or-later
# Copyright (c) 2025-2026 Cristian Cezar Moisés
# ZUPT audit test suite — double-validated security checks.
# Each property is checked via TWO independent paths.

set -Eeuo pipefail

repo_root=$(CDPATH='' cd -- "$(dirname -- "$0")/.." && pwd -P)
ZUPT_BIN=${ZUPT_BIN:-$repo_root/zupt}
if [[ ! -x $ZUPT_BIN ]]; then
    printf '  FAIL: %s not found; build ZUPT first\n' "$ZUPT_BIN" >&2
    exit 1
fi
version=$("$ZUPT_BIN" --version 2>&1)
if ! grep -Fq 'libvuptsdk=enabled' <<<"$version"; then
    echo '  SKIP: system libvuptsdk integration is disabled (build with WITH_SDK=1)'
    exit 0
fi

TMPDIR=$(mktemp -d)
trap 'rm -rf -- "$TMPDIR"' EXIT
cd "$TMPDIR"

PASS=0; FAIL=0
DCHK() {
    local name="$1" a="$2" b="$3"
    if [ "$a" = "$b" ] && [ "$a" = "1" ]; then
        echo "  ✓ $name (A=$a B=$b agree)"; PASS=$((PASS+1))
    else
        echo "  ✗ $name (A=$a B=$b disagree)"; FAIL=$((FAIL+1))
    fi
}

# Setup: SDK keys
"$ZUPT_BIN" keygen --sdk -o k.priv > /dev/null 2>&1
"$ZUPT_BIN" keygen --sdk -o other.priv > /dev/null 2>&1
"$ZUPT_BIN" keygen -o legacy.key > /dev/null 2>&1

echo "  [A. Authenticated archives]"

# A1. Wrong key rejected: SDK key vs SDK archive (path A) + Legacy key vs SDK archive (path B)
echo "data" > input.txt
"$ZUPT_BIN" c --pq-sdk k.priv.pub a.zupt input.txt > /dev/null 2>&1
mkdir -p ea
set +e
(cd ea && "$ZUPT_BIN" x --pq-sdk ../other.priv ../a.zupt > /dev/null 2>&1)
A_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f ea/input.txt ] && echo 1 || echo 0)
mkdir -p eb
set +e
(cd eb && "$ZUPT_BIN" x --pq legacy.key ../a.zupt > /dev/null 2>&1)
B_RC=$?
set -e
B=$([ "$B_RC" -ne 0 ] && [ ! -f eb/input.txt ] && echo 1 || echo 0)
DCHK "Wrong key rejected (SDK key + legacy key paths)" "$A" "$B"

# A2. Tamper at byte position N detected.
#
# F-02 (ZUPT 2.2.4): the previous version flipped a byte at len-50 for
# path B. SDK-PQ archive sizes vary by 1-2 bytes per run (ciphertext
# encoding), so len-50 occasionally landed inside the *index* region
# (between footer.index_offset and the trailing 32-byte footer), which
# is NOT covered by the per-block HMAC. Roughly 10% of runs would let
# the tampered file extract successfully and the suite would flake.
#
# Fix: tamper at absolute offsets known to be inside the encrypted
# body of any non-empty SDK-PQ archive. With "data\n" (5 bytes) as
# input, the archive is ~1769-1771 bytes and the body runs from
# offset ~80 to ~1610. Offsets 200 (early-body) and 500 (mid-body)
# are both deterministically authenticated.
#
# The unauthenticated index region is now tracked as F-02b (deferred
# to v2.2.5 format-v1.5).
cp a.zupt t1.zupt; cp a.zupt t2.zupt
python3 -c "
b = bytearray(open('t1.zupt','rb').read())
b[200] ^= 1
open('t1.zupt','wb').write(bytes(b))" 2>/dev/null
python3 -c "
b = bytearray(open('t2.zupt','rb').read())
b[500] ^= 1
open('t2.zupt','wb').write(bytes(b))" 2>/dev/null
mkdir -p t1e t2e
set +e
(cd t1e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../t1.zupt > /dev/null 2>&1)
A_RC=$?
(cd t2e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../t2.zupt > /dev/null 2>&1)
B_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f t1e/input.txt ] && echo 1 || echo 0)
B=$([ "$B_RC" -ne 0 ] && [ ! -f t2e/input.txt ] && echo 1 || echo 0)
DCHK "Tamper detected at body offset 200 and 500" "$A" "$B"

echo "  [B. Format security]"

# B1. Zero-byte file (path A) + 1-byte file (path B): both must roundtrip
: >empty.txt
echo -n "x" > one.txt
"$ZUPT_BIN" c --pq-sdk k.priv.pub e.zupt empty.txt > /dev/null 2>&1
"$ZUPT_BIN" c --pq-sdk k.priv.pub o.zupt one.txt > /dev/null 2>&1
mkdir -p eex && (cd eex && "$ZUPT_BIN" x --pq-sdk ../k.priv ../e.zupt > /dev/null 2>&1)
mkdir -p oex && (cd oex && "$ZUPT_BIN" x --pq-sdk ../k.priv ../o.zupt > /dev/null 2>&1)
A=$([ -f eex/empty.txt ] && [ ! -s eex/empty.txt ] && echo 1 || echo 0)
B=$([ -f oex/one.txt ] && [ "$(cat oex/one.txt)" = "x" ] && echo 1 || echo 0)
DCHK "Edge-size files (0/1 byte) roundtrip" "$A" "$B"

# B2. 1MB random file (path A) + structured-data 1MB (path B)
dd if=/dev/urandom of=big_a.bin bs=1M count=1 2>/dev/null
python3 -c "open('big_b.bin','wb').write(b'A'*1024*1024)"
"$ZUPT_BIN" c --pq-sdk k.priv.pub ba.zupt big_a.bin > /dev/null 2>&1
"$ZUPT_BIN" c --pq-sdk k.priv.pub bb.zupt big_b.bin > /dev/null 2>&1
mkdir -p ba_e && (cd ba_e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../ba.zupt > /dev/null 2>&1)
mkdir -p bb_e && (cd bb_e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../bb.zupt > /dev/null 2>&1)
A=$(diff -q big_a.bin ba_e/big_a.bin > /dev/null 2>&1 && echo 1 || echo 0)
B=$(diff -q big_b.bin bb_e/big_b.bin > /dev/null 2>&1 && echo 1 || echo 0)
DCHK "1MB roundtrip (random + structured)" "$A" "$B"

# B3. Truncated archive rejected (path A: cut last 50 bytes) (path B: cut at midpoint)
cp a.zupt tr1.zupt; cp a.zupt tr2.zupt
python3 - <<'PY'
from pathlib import Path

first = Path("tr1.zupt")
first.write_bytes(first.read_bytes()[:-50])
second = Path("tr2.zupt")
second.write_bytes(second.read_bytes()[:100])
PY
mkdir -p tr1e tr2e
set +e
(cd tr1e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../tr1.zupt > /dev/null 2>&1)
A_RC=$?
(cd tr2e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../tr2.zupt > /dev/null 2>&1)
B_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f tr1e/input.txt ] && echo 1 || echo 0)
B=$([ "$B_RC" -ne 0 ] && [ ! -f tr2e/input.txt ] && echo 1 || echo 0)
DCHK "Truncated archive rejected" "$A" "$B"

echo "  [C. Format compatibility]"

# C1. Mode confusion: SDK archive cannot be read with --pq (legacy)
mkdir -p mc1
set +e
(cd mc1 && "$ZUPT_BIN" x --pq ../legacy.key ../a.zupt > /dev/null 2>&1)
A_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f mc1/input.txt ] && echo 1 || echo 0)
# Also: legacy archive cannot be read with --pq-sdk
"$ZUPT_BIN" c --pq legacy.key leg.zupt input.txt > /dev/null 2>&1
mkdir -p mc2
set +e
(cd mc2 && "$ZUPT_BIN" x --pq-sdk ../k.priv ../leg.zupt > /dev/null 2>&1)
B_RC=$?
set -e
B=$([ "$B_RC" -ne 0 ] && [ ! -f mc2/input.txt ] && echo 1 || echo 0)
DCHK "Mode confusion prevented (SDK↔legacy)" "$A" "$B"

# C2. Legacy archive readable with legacy key (compat baseline)
mkdir -p lc && (cd lc && "$ZUPT_BIN" x --pq ../legacy.key ../leg.zupt > /dev/null 2>&1)
A=$(diff -q lc/input.txt input.txt > /dev/null 2>&1 && echo 1 || echo 0)
# B: SDK archive readable with SDK key (compat baseline)
mkdir -p sc && (cd sc && "$ZUPT_BIN" x --pq-sdk ../k.priv ../a.zupt > /dev/null 2>&1)
B=$(diff -q sc/input.txt input.txt > /dev/null 2>&1 && echo 1 || echo 0)
DCHK "Both SDK and legacy paths roundtrip independently" "$A" "$B"

echo "  [D. Robustness]"

# D1. Non-existent input handled
set +e
"$ZUPT_BIN" c --pq-sdk k.priv.pub nx.zupt /nonexistent_file_12345 > /dev/null 2>&1
A_RC=$?
"$ZUPT_BIN" c --pq-sdk k.priv.pub nx2.zupt /dev/nonexistent > /dev/null 2>&1
B_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f nx.zupt ] && echo 1 || echo 0)
B=$([ "$B_RC" -ne 0 ] && [ ! -f nx2.zupt ] && echo 1 || echo 0)
DCHK "Missing input file rejected cleanly" "$A" "$B"

# D2. Non-existent key handled
mkdir -p nk1
set +e
(cd nk1 && "$ZUPT_BIN" x --pq-sdk /nonexistent.key ../a.zupt > /dev/null 2>&1)
A_RC=$?
"$ZUPT_BIN" c --pq-sdk /nonexistent.pub bbnk.zupt input.txt > /dev/null 2>&1
B_RC=$?
set -e
A=$([ "$A_RC" -ne 0 ] && [ ! -f nk1/input.txt ] && echo 1 || echo 0)
B=$([ "$B_RC" -ne 0 ] && [ ! -s bbnk.zupt ] && echo 1 || echo 0)
DCHK "Missing key file rejected cleanly" "$A" "$B"

# D3. Multiple files in one archive
mkdir -p multi
echo "a" > multi/a.txt; echo "b" > multi/b.txt; echo "c" > multi/c.txt
"$ZUPT_BIN" c --pq-sdk k.priv.pub mm.zupt multi/a.txt multi/b.txt multi/c.txt > /dev/null 2>&1
mkdir -p mmex && (cd mmex && "$ZUPT_BIN" x --pq-sdk ../k.priv ../mm.zupt > /dev/null 2>&1)
A=$(diff -q mmex/multi/a.txt multi/a.txt > /dev/null 2>&1 && echo 1 || echo 0)
B=$([ -f mmex/multi/c.txt ] && [ "$(cat mmex/multi/c.txt 2>/dev/null)" = "c" ] && echo 1 || echo 0)
DCHK "Multiple files in archive roundtrip" "$A" "$B"

echo
echo "  ───────────────────────────────────────"
echo "  Audit results: $PASS passed, $FAIL failed"
echo "  ───────────────────────────────────────"
((FAIL == 0))