fix(badapple): --no-load must not wait for ICMP; survive a slow readback

Two faults the 60 fps work exposed, both in the delivery tooling:

* `demo.sh --no-load` waited for the terminal to answer ping.  A loaded payload
  owns the LANCE and stops the OS's stack with it, so that gate can never pass
  once the payload is running -- the recovery path documented for "already
  jumped" deadlocked for its full 180 s.  The gate now runs only when this
  invocation is going to load the payload.

* `rdebug_load.py` used a 5 s socket timeout and no retry, so a readback peek
  that timed out on a freshly booted OS failed a load whose pokes had already
  been acknowledged.  Timeout is 15 s (--timeout) and the readback retries once
  before it is believed.

Tests: the retry path is pinned with a fake remote that times out on the first
peek (one timeout, then the successful pass over every chunk) and one that never
answers (it must raise rather than loop).  9 tests in harness/badapple pass.
This commit is contained in:
arti committed 2026-10-31 22:20:58 +02:00
1 parent c166c6fdec
commit d211ff0770
3 files changed
+80 -13

No files matched your search

+17 -9
View File
@@ -62,15 +62,23 @@ if [ "$SEND" = 1 ]; then
fi
# --- readiness: an ICMP reply from the terminal's own stack -----------------
start=$(date +%s)
while ! ping -c1 -W1 "$HOST" > /dev/null 2>&1; do
if [ $(( $(date +%s) - start )) -ge "$WAIT" ]; then
echo "demo.sh: $HOST did not answer within ${WAIT}s" >&2
echo " is the terminal booted and on the tap? (harness/mame/run.sh net)" >&2
exit 1
fi
done
echo "demo.sh: $HOST answers ($(( $(date +%s) - start ))s)"
# Only when this run is going to load the payload. A loaded payload owns the
# LANCE (and stops the OS's stack with it), so it never answers ICMP: waiting for
# a ping after the load would deadlock -- which is exactly what --no-load used to
# do. --no-load therefore skips the gate and trusts the operator.
if [ "$LOAD" = 1 ]; then
start=$(date +%s)
while ! ping -c1 -W1 "$HOST" > /dev/null 2>&1; do
if [ $(( $(date +%s) - start )) -ge "$WAIT" ]; then
echo "demo.sh: $HOST did not answer within ${WAIT}s" >&2
echo " is the terminal booted and on the tap? (harness/mame/run.sh net)" >&2
exit 1
fi
done
echo "demo.sh: $HOST answers ($(( $(date +%s) - start ))s)"
else
echo "demo.sh: --no-load: not waiting for ICMP (a loaded payload owns the LANCE)"
fi
# --- the payload, over the firmware's own remote-debug service --------------
if [ "$LOAD" = 1 ]; then
+29 -4
View File
@@ -46,6 +46,11 @@ CHUNK = 512
#: only when the environment file is absent, so a hand-built image still loads.
DEFAULT_LOAD = 0x018C0000
#: Socket timeout. Generous on purpose: a freshly booted OS answers rdebug
#: slowly (measured: a readback peek timed out at 5 s while the machine was still
#: settling), and a slow peek must not fail a load that already succeeded.
DEFAULT_TIMEOUT = 15.0
def read_env(path) -> dict:
"""Parse ``vidstream.env`` (``KEY=VALUE`` lines) into ints where possible."""
@@ -85,14 +90,31 @@ def plan(image: bytes, load: int, size: int = CHUNK):
for n, part in enumerate(chunks(image, size))]
def readback(rd, transfers, retries=1):
"""Peek every chunk back, retrying a stalled read once.
A timeout here is the machine being busy, not a bad image -- the poke has
already been acknowledged -- so it is retried before it is believed.
"""
back = b""
for attempt in range(retries + 1):
try:
return b"".join(rd.peek(addr, len(part)) for addr, part in transfers)
except (OSError, TimeoutError):
if attempt == retries:
raise
return back
def load_and_jump(image_path, entry, load, do_jump=True, chunk=CHUNK,
host="192.168.99.2", port=771, verbose=False):
host="192.168.99.2", port=771, verbose=False,
timeout=DEFAULT_TIMEOUT):
"""Poke ``image_path`` at ``load``, verify it, then jump to ``entry``."""
import exploit # dwt/exploit.py: the protocol the firmware speaks
data = open(image_path, "rb").read()
transfers = plan(data, load, chunk)
rd = exploit.RDebug(host, port, 5.0, verbose)
rd = exploit.RDebug(host, port, timeout, verbose)
rd.connect()
try:
for addr, part in transfers:
@@ -100,7 +122,7 @@ def load_and_jump(image_path, entry, load, do_jump=True, chunk=CHUNK,
print("poked %d bytes in %d chunk(s) at %#x"
% (len(data), len(transfers), load))
back = b"".join(rd.peek(addr, len(part)) for addr, part in transfers)
back = readback(rd, transfers)
if back != data:
bad = next(i for i in range(min(len(back), len(data)))
if back[i] != data[i])
@@ -140,6 +162,8 @@ def main(argv=None) -> int:
ap.add_argument("--chunk", type=int, default=CHUNK,
help="poke size in bytes (max %d; default %%(default)s)"
% CHUNK)
ap.add_argument("--timeout", type=float, default=DEFAULT_TIMEOUT,
help="socket timeout in seconds (default %(default)s)")
ap.add_argument("--no-jump", action="store_true",
help="load and verify, do not start it")
ap.add_argument("--dry-run", action="store_true",
@@ -182,7 +206,8 @@ def main(argv=None) -> int:
try:
return load_and_jump(args.image, entry, load,
do_jump=not args.no_jump, chunk=args.chunk,
host=args.host, port=args.port)
host=args.host, port=args.port,
timeout=args.timeout)
except Exception as exc: # noqa: BLE001 - one line, not a traceback
print("error: %s: %s" % (type(exc).__name__, exc), file=sys.stderr)
return 1
@@ -54,6 +54,40 @@ def test_missing_env_file_says_so_instead_of_guessing(tmp_path, capsys):
assert "no build environment" in err and "build-stream.sh" in err
# --- readback ------------------------------------------------------------
class _Flaky:
"""A remote end whose first peek times out, as a busy OS does."""
def __init__(self, transfers, fail_times=1):
self.mem = dict(transfers)
self.fail = fail_times
self.calls = 0
def peek(self, addr, length):
self.calls += 1
if self.fail:
self.fail -= 1
raise TimeoutError("timed out")
return self.mem[addr][:length]
def test_readback_retries_a_stalled_peek_instead_of_failing_the_load():
transfers = rdebug_load.plan(b"0123456789", 0x018C0000, 4)
rd = _Flaky(transfers, fail_times=1)
assert rdebug_load.readback(rd, transfers) == b"0123456789"
# The retry re-reads the whole image: one timed-out chunk plus the three
# chunks of the successful pass.
assert rd.calls == len(transfers) + 1
def test_readback_gives_up_after_the_retry():
transfers = rdebug_load.plan(b"0123", 0x018C0000, 4)
rd = _Flaky(transfers, fail_times=9)
with pytest.raises(TimeoutError):
rdebug_load.readback(rd, transfers)
# --- the clip format ------------------------------------------------------
def test_packed_rows_put_pixel_zero_on_bit_zero_of_byte_zero():