aboutsummaryrefslogtreecommitdiffstats
path: root/idatui/launch.py
diff options
context:
space:
mode:
authorblasty <blasty@local>2026-07-24 01:37:48 +0200
committerblasty <blasty@local>2026-07-24 01:37:48 +0200
commitf925701e003f05160d46cc3efd8eacf3ea256aeb (patch)
tree6ca179a93898d66d7fcb462ca192168c8f852a65 /idatui/launch.py
parentworker: idalib worker + WorkerClient (drop-in for IDAClient) — migration st... (diff)
downloadida-tui-f925701e003f05160d46cc3efd8eacf3ea256aeb.tar.gz
ida-tui-f925701e003f05160d46cc3efd8eacf3ea256aeb.tar.xz
ida-tui-f925701e003f05160d46cc3efd8eacf3ea256aeb.zip
app: --backend {mcp,worker} — run the TUI on our idalib worker (migration step 3)
Wires WorkerClient into the app as a selectable backend, so you can launch: ida-tui --backend worker /path/to/binary # or IDATUI_BACKEND=worker * IdaTui gains a `backend` param. _connect is refactored into _open_mcp_client() (the existing ida-pro-mcp path, unchanged) and _open_worker_client() (spawns a private idalib worker via WorkerClient that opens+analyzes THIS binary and streams progress into the loading overlay). The common tail (health, keepalive, Program, load_functions) is shared, so domain.py and every view are untouched. * _reconnect branches the same way: a dropped worker (segfault → closed socket) respawns a fresh WorkerClient — the connection-loss recovery already built works verbatim for the worker. * launch.py adds --backend (env IDATUI_BACKEND); the worker path skips all the supervisor/server plumbing (it owns a private worker) and just sweeps stale locks + requires a binary. keepalive is a no-op for the worker (it never idles out). mcp remains the default — nothing changes unless you opt in. Verified without idalib: all four files parse; --backend is in --help; IdaTui constructs for both backends with both openers present; WorkerClient covers the full client surface. The idalib E2E (experiments/worker_smoke.py, and actually launching --backend worker) still can't run in this sandbox — it now reaps every idalib spawn before a byte is written — but the mcp default is untouched and the worker path reuses proven pieces (the unix protocol benched at ~50us/call; the worker dispatches the same tool functions the HTTP path does). To validate on a real box: ida-tui --backend worker targets/echo (or run experiments/worker_smoke.py for the headless read-path check).
Diffstat (limited to 'idatui/launch.py')
-rw-r--r--idatui/launch.py42
1 files changed, 28 insertions, 14 deletions
diff --git a/idatui/launch.py b/idatui/launch.py
index 2ecf8b5..15352d5 100644
--- a/idatui/launch.py
+++ b/idatui/launch.py
@@ -189,6 +189,10 @@ def main(argv: list[str] | None = None) -> int:
help="listen for RPC on this unix socket (puppeteer the TUI)")
p.add_argument("--no-server", action="store_true",
help="do not auto-start the supervisor if it's down")
+ p.add_argument("--backend", choices=("mcp", "worker"),
+ default=os.environ.get("IDATUI_BACKEND", "mcp"),
+ help="'mcp' (ida-pro-mcp supervisor, default) or 'worker' "
+ "(our own in-process idalib worker over a unix socket)")
args = p.parse_args(argv)
binary = None
@@ -202,21 +206,30 @@ def main(argv: list[str] | None = None) -> int:
f"{os.path.dirname(binary)}")
return 2
- # Everything slow — starting the supervisor and opening/analyzing the binary
- # — is deferred to the TUI so its chrome + "loading…" overlay are on screen
- # the whole time, instead of a blank terminal before the TUI even starts.
- # Here we only do the instant checks.
- server_down = not _server_up(*_server_addr(args.url))
- if args.no_server and server_down:
- _log("supervisor is down and --no-server was given")
- return 1
- if binary is not None and server_down:
- # About to start a fresh supervisor (nothing holds the DB): clear any
- # stale unpacked lock files a crashed worker left, so the supervisor's
- # seeded initial open doesn't crash-loop on a wedged DB.
- swept = _sweep_locks(binary)
+ if args.backend == "worker":
+ # The worker path owns no shared supervisor: the TUI spawns a private
+ # idalib worker that opens THIS binary. No server/lock plumbing here.
+ if binary is None:
+ _log("--backend worker needs a binary path")
+ return 2
+ swept = _sweep_locks(binary) # a crashed worker can leave the DB wedged
if swept:
_log(f"cleared {swept} stale lock file(s) from a crashed worker")
+ else:
+ # Everything slow — starting the supervisor and opening/analyzing the
+ # binary — is deferred to the TUI so its chrome + "loading…" overlay are
+ # on screen the whole time. Here we only do the instant checks.
+ server_down = not _server_up(*_server_addr(args.url))
+ if args.no_server and server_down:
+ _log("supervisor is down and --no-server was given")
+ return 1
+ if binary is not None and server_down:
+ # About to start a fresh supervisor (nothing holds the DB): clear any
+ # stale unpacked lock files a crashed worker left, so the seeded
+ # initial open doesn't crash-loop on a wedged DB.
+ swept = _sweep_locks(binary)
+ if swept:
+ _log(f"cleared {swept} stale lock file(s) from a crashed worker")
# Hand off to the TUI (imported late so --help works without textual). It
# ensures the server (with on-screen progress), then opens/adopts the binary
@@ -229,7 +242,8 @@ def main(argv: list[str] | None = None) -> int:
rpc_path = os.path.abspath(os.path.expanduser(args.rpc)) if args.rpc else None
IdaTui(url=args.url, db=args.db,
open_path=(binary if args.db is None else None), ttl=args.ttl,
- ensure_server=not args.no_server,
+ ensure_server=(args.backend == "mcp" and not args.no_server),
+ backend=args.backend,
keepalive=not args.no_keepalive, rpc_path=rpc_path).run()
return 0