137 lines
6.3 KiB
Shell
Executable file
137 lines
6.3 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 for zupt 2.2+
|
|
# Each property is checked via TWO independent paths.
|
|
|
|
ZUPT_BIN="$(realpath ./zupt)"
|
|
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 && (cd ea && "$ZUPT_BIN" x --pq-sdk ../other.priv ../a.zupt > /dev/null 2>&1)
|
|
A=$([ ! -f ea/input.txt ] && echo 1 || echo 0)
|
|
mkdir -p eb && (cd eb && "$ZUPT_BIN" x --pq legacy.key ../a.zupt > /dev/null 2>&1)
|
|
B=$([ ! -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 (path A: pos 200) (path B: pos at end)
|
|
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[len(b)-50] ^= 1
|
|
open('t2.zupt','wb').write(bytes(b))" 2>/dev/null
|
|
mkdir -p t1e && (cd t1e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../t1.zupt > /dev/null 2>&1)
|
|
mkdir -p t2e && (cd t2e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../t2.zupt > /dev/null 2>&1)
|
|
A=$([ ! -f t1e/input.txt ] && echo 1 || echo 0)
|
|
B=$([ ! -f t2e/input.txt ] && echo 1 || echo 0)
|
|
DCHK "Tamper detected at any byte position" "$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
|
|
truncate -s -50 tr1.zupt
|
|
truncate -s 100 tr2.zupt
|
|
mkdir -p tr1e tr2e
|
|
(cd tr1e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../tr1.zupt > /dev/null 2>&1)
|
|
(cd tr2e && "$ZUPT_BIN" x --pq-sdk ../k.priv ../tr2.zupt > /dev/null 2>&1)
|
|
A=$([ ! -f tr1e/input.txt ] && echo 1 || echo 0)
|
|
B=$([ ! -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 && (cd mc1 && "$ZUPT_BIN" x --pq ../legacy.key ../a.zupt > /dev/null 2>&1)
|
|
A=$([ ! -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 && (cd mc2 && "$ZUPT_BIN" x --pq-sdk ../k.priv ../leg.zupt > /dev/null 2>&1)
|
|
B=$([ ! -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
|
|
"$ZUPT_BIN" c --pq-sdk k.priv.pub nx.zupt /nonexistent_file_12345 > /dev/null 2>&1
|
|
A=$([ ! -f nx.zupt ] && echo 1 || echo 0)
|
|
"$ZUPT_BIN" c --pq-sdk k.priv.pub nx2.zupt /dev/nonexistent > /dev/null 2>&1
|
|
B=$([ ! -f nx2.zupt ] && echo 1 || echo 0)
|
|
DCHK "Missing input file rejected cleanly" "$A" "$B"
|
|
|
|
# D2. Non-existent key handled
|
|
mkdir -p nk1 && (cd nk1 && "$ZUPT_BIN" x --pq-sdk /nonexistent.key ../a.zupt > /dev/null 2>&1)
|
|
A=$([ ! -f nk1/input.txt ] && echo 1 || echo 0)
|
|
"$ZUPT_BIN" c --pq-sdk /nonexistent.pub bbnk.zupt input.txt > /dev/null 2>&1
|
|
B=$([ ! -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 -eq 0 ]
|