fix: use clean VM reset and resilient agent listener

This commit is contained in:
2026-08-16 16:36:24 +09:00
parent a637e591e1
commit 5d9cd90299
4 changed files with 63 additions and 33 deletions
+44 -20
View File
@@ -3,6 +3,7 @@ from __future__ import annotations
import argparse import argparse
import json import json
import shutil
import subprocess import subprocess
import sys import sys
import time import time
@@ -37,15 +38,41 @@ def rpc(payload: dict[str, object], start_daemon: bool = False) -> object:
return response["result"] return response["result"]
def wait_ready(timeout: int) -> bool:
"""Wait quietly; do not turn an expected boot gap into error-log spam."""
deadline = time.monotonic() + timeout
while time.monotonic() < deadline:
try:
status = rpc({"op": "status"})
if status["agent_connected"] and str(rpc({"op": "ping"})["response"]).startswith("OK 504F4E47"):
return True
except RuntimeError:
pass
time.sleep(.5)
return False
def follow_logs() -> int:
log_path = ROOT / ".qemu" / "ferro-vm.log"
lnav = shutil.which("lnav.exe") or shutil.which("lnav")
if lnav:
return subprocess.run([lnav, str(log_path)]).returncode
print("lnav was not found; following the log with PowerShell.", file=sys.stderr)
return subprocess.run([
"powershell", "-NoProfile", "-Command",
f"Get-Content -LiteralPath '{log_path}' -Wait",
]).returncode
def main() -> int: def main() -> int:
parser = argparse.ArgumentParser(description="Windows-only QEMU/FreeDOS automation") parser = argparse.ArgumentParser(description="Windows-only QEMU/FreeDOS automation")
commands = parser.add_subparsers(dest="op", required=True) commands = parser.add_subparsers(dest="op", required=True)
for name in ("start", "stop", "status", "ping", "screenshot", "ocr"): for name in ("start", "stop", "status", "ping", "screenshot", "ocr", "logs"):
commands.add_parser(name) commands.add_parser(name)
wait = commands.add_parser("wait-ready")
wait.add_argument("--timeout", type=int, default=45)
reset = commands.add_parser("reset") reset = commands.add_parser("reset")
reset.add_argument("--timeout", type=int, default=45) reset.add_argument("--timeout", type=int, default=45)
soft_reset = commands.add_parser("soft-reset")
soft_reset.add_argument("--timeout", type=int, default=45)
execute = commands.add_parser("exec") execute = commands.add_parser("exec")
execute.add_argument("command") execute.add_argument("command")
put = commands.add_parser("put") put = commands.add_parser("put")
@@ -57,6 +84,13 @@ def main() -> int:
args = parser.parse_args() args = parser.parse_args()
try: try:
if args.op == "logs":
return follow_logs()
if args.op == "wait-ready":
if wait_ready(args.timeout):
print(json.dumps({"agent": "PONG"}))
return 0
raise RuntimeError("TCPAGENT did not become ready")
if args.op == "ocr": if args.op == "ocr":
result = rpc({"op": "screenshot"}) result = rpc({"op": "screenshot"})
import logging import logging
@@ -65,25 +99,15 @@ def main() -> int:
recognized = RapidOCR()(result["path"]) recognized = RapidOCR()(result["path"])
print("\n".join(recognized.txts or ())) print("\n".join(recognized.txts or ()))
return 0 return 0
if args.op in ("reset", "soft-reset"): if args.op == "reset":
if args.op == "reset": rpc({"op": "stop"}, start_daemon=True)
rpc({"op": "stop"}, start_daemon=True) time.sleep(.5)
time.sleep(.5) rpc({"op": "start"}, start_daemon=True)
rpc({"op": "start"}, start_daemon=True)
else:
rpc({"op": "soft-reset"})
# FreeDOS displays its default boot menu before FDAUTO.BAT starts
# TCPAGENT. This is input, not a readiness delay.
time.sleep(2) time.sleep(2)
rpc({"op": "monitor", "command": "sendkey ret"}) rpc({"op": "monitor", "command": "sendkey ret"})
deadline = time.monotonic() + args.timeout if wait_ready(args.timeout):
while time.monotonic() < deadline: print(json.dumps({"reset": "complete", "agent": "PONG"}))
try: return 0
if str(rpc({"op": "ping"})["response"]).startswith("OK 504F4E47"):
print(json.dumps({"reset": args.op, "agent": "PONG"}))
return 0
except RuntimeError:
time.sleep(.5)
raise RuntimeError("TCPAGENT did not become ready") raise RuntimeError("TCPAGENT did not become ready")
payload: dict[str, object] = {"op": args.op} payload: dict[str, object] = {"op": args.op}
if args.op == "exec": payload["command"] = args.command if args.op == "exec": payload["command"] = args.command
+9 -6
View File
@@ -62,10 +62,12 @@ class Host:
log_event(logging.WARNING, "agent rejected", peer=str(peer), reason="already connected") log_event(logging.WARNING, "agent rejected", peer=str(peer), reason="already connected")
continue continue
self.agent = sock self.agent = sock
self.agent_ready.set()
try: try:
banner = self._read_line(sock).decode("ascii", "replace") banner = self._read_line(sock).decode("ascii", "replace")
log_event(logging.INFO, "agent connected", peer=f"{peer[0]}:{peer[1]}", banner=banner) if banner != "TCPAGENT READY":
raise ConnectionError(f"invalid TCPAGENT banner: {banner!r}")
self.agent_ready.set()
log_event(logging.INFO, "agent connected", peer=f"{peer[0]}:{peer[1]}")
# Request() owns protocol reads. Between requests, peek only # Request() owns protocol reads. Between requests, peek only
# for EOF so TCPAGENT can reconnect without being rejected. # for EOF so TCPAGENT can reconnect without being rejected.
while self.agent is sock: while self.agent is sock:
@@ -78,6 +80,11 @@ class Host:
self.agent_lock.release() self.agent_lock.release()
else: else:
time.sleep(.05) time.sleep(.05)
except (ConnectionError, OSError) as exc:
# Reset can close a connection during its banner. This is a
# per-connection event, never a reason to kill the listener.
log_event(logging.WARNING, "agent handshake/connection failed",
peer=f"{peer[0]}:{peer[1]}", error=str(exc))
finally: finally:
with self.agent_lock: with self.agent_lock:
if self.agent is sock: if self.agent is sock:
@@ -232,10 +239,6 @@ class Host:
return {"agent_connected": self.agent_ready.is_set(), "qemu_running": running, "log": str(LOG_PATH)} return {"agent_connected": self.agent_ready.is_set(), "qemu_running": running, "log": str(LOG_PATH)}
if op == "start": return self.start() if op == "start": return self.start()
if op == "stop": return self.stop() if op == "stop": return self.stop()
if op == "soft-reset":
self.monitor("system_reset")
log_event(logging.INFO, "qemu soft reset requested")
return {"reset": "requested"}
if op == "ping": return self.ping() if op == "ping": return self.ping()
if op == "exec": return self.exec(str(request["command"])) if op == "exec": return self.exec(str(request["command"]))
if op == "put": return self.put(str(request["source"]), str(request["destination"])) if op == "put": return self.put(str(request["source"]), str(request["destination"]))
+9 -5
View File
@@ -20,14 +20,14 @@ Windows named pipe `\\.\pipe\ferrolang-vm`; there is no controller or observer
TCP port. Monitor the append-only structured log in another terminal: TCP port. Monitor the append-only structured log in another terminal:
```powershell ```powershell
lnav .qemu/ferro-vm.log uv run ferro-vm logs
``` ```
Commands: Commands:
```powershell ```powershell
uv run ferro-vm reset # clean QEMU quit and restart uv run ferro-vm reset # clean QEMU quit and restart
uv run ferro-vm soft-reset # QEMU system_reset uv run ferro-vm wait-ready --timeout 45
uv run ferro-vm ping uv run ferro-vm ping
uv run ferro-vm exec 'dir C:\FEC' uv run ferro-vm exec 'dir C:\FEC'
uv run ferro-vm put fec/src/check.c 'C:\FEC\SRC\CHECK.C' uv run ferro-vm put fec/src/check.c 'C:\FEC\SRC\CHECK.C'
@@ -37,9 +37,13 @@ uv run ferro-vm ocr
uv run ferro-vm stop uv run ferro-vm stop
``` ```
Both reset modes wait for FreeDOS to boot, submit the default boot-menu Enter, `reset` cleanly quits and restarts QEMU, waits for FreeDOS to boot, submits
and require TCPAGENT `PING`/`PONG` before succeeding. The daemon logs command the default boot-menu Enter, and requires TCPAGENT `PING`/`PONG`. QEMU
metadata, DOS output, exit status, transfers, and agent lifecycle events as UTF-8 lines. It deliberately never logs raw binary `system_reset` is intentionally unsupported because repeated soft resets leave
the FreeDOS NE2000 packet driver stuck during initialization. `logs` starts
`lnav` when installed and otherwise falls back to PowerShell `Get-Content
-Wait`. The daemon logs command metadata, DOS output, exit status, transfers,
and agent lifecycle events as UTF-8 lines. It deliberately never logs raw binary
payloads or protocol hex. payloads or protocol hex.
## Standalone OCR ## Standalone OCR
+1 -2
View File
@@ -176,10 +176,9 @@ int main(void) {
int used,rc; uint16_t key; int used,rc; uint16_t key;
if(Utils::parseEnv()!=0)return 2; if(Utils::parseEnv()!=0)return 2;
if(Utils::initStack(1,TCP_SOCKET_RING_SIZE,ctrl_break,ctrl_c))return 3; if(Utils::initStack(1,TCP_SOCKET_RING_SIZE,ctrl_break,ctrl_c))return 3;
cprintf("[TCPAGENT] Connecting to 10.0.2.2:%u\r\n",SERVER_PORT);
while(!stop_requested) { while(!stop_requested) {
if(connect_host()!=0){unsigned long spins=0;while(spins++<60000UL&&!stop_requested)drive();continue;} if(connect_host()!=0){unsigned long spins=0;while(spins++<60000UL&&!stop_requested)drive();continue;}
cprintf("[TCPAGENT] Connected. Alt-X stops.\r\n"); write_text("TCPAGENT READY\r\n"); used=0; write_text("TCPAGENT READY\r\n"); used=0;
while(!stop_requested&&!socketp->isRemoteClosed()) { while(!stop_requested&&!socketp->isRemoteClosed()) {
drive(); rc=socketp->recv((uint8_t *)data,CHUNK_SIZE); drive(); rc=socketp->recv((uint8_t *)data,CHUNK_SIZE);
if(rc<0)break; if(rc<0)break;