diff options
| author | blasty <blasty@local> | 2026-07-24 01:37:48 +0200 |
|---|---|---|
| committer | blasty <blasty@local> | 2026-07-24 01:37:48 +0200 |
| commit | f925701e003f05160d46cc3efd8eacf3ea256aeb (patch) | |
| tree | 6ca179a93898d66d7fcb462ca192168c8f852a65 /idatui/launch.py | |
| parent | worker: idalib worker + WorkerClient (drop-in for IDAClient) — migration st... (diff) | |
| download | ida-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.py | 42 |
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 |
