206 lines
8 KiB
Shell
Executable file
206 lines
8 KiB
Shell
Executable file
#!/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))
|