aboutsummaryrefslogtreecommitdiffstats
path: root/experiments/worker_smoke.py
diff options
context:
space:
mode:
authoruser <user@clank>2026-08-07 15:14:30 +0200
committeruser <user@clank>2026-08-07 15:14:30 +0200
commit72fce7da1a1fd6527e389ffeb0f951157523589a (patch)
tree3f08a83f0d99f3d3fc7069b7be00d235bab82817 /experiments/worker_smoke.py
parentStop tracking 157MB of core dumps, and ignore them (diff)
parentdocs: upstream findings for the ida-codemode maintainers (diff)
downloadida-tui-72fce7da1a1fd6527e389ffeb0f951157523589a.tar.gz
ida-tui-72fce7da1a1fd6527e389ffeb0f951157523589a.tar.xz
ida-tui-72fce7da1a1fd6527e389ffeb0f951157523589a.zip
Merge the IDA Code Mode port
Replaces the private idalib worker (idatui/worker.py + worker_client.py, with server/patch_server.py injecting tools into ida-pro-mcp) with an ordinary client of ida_codemode.client.DatabaseHandle. A database open by an IDA GUI is reused; otherwise Code Mode starts or shares a managed idalib worker. The TUI no longer owns an IDA process, and closing it releases only its lease. Based on Duncan Ogilvie's port, rebased onto ~150 commits of local work it predated. The rebase itself was mechanical; landing it was not. Nine defects had to be fixed before the feature set was whole again, none of which the patch's own tests could catch: - DatabaseHandle.open() takes image_base, not loading_address: every connect() would have raised TypeError on the first call - five operations our tree had grown were simply missing (flowchart, so the graph view was dead; op_format/pc_nums/pc_num_format, so 'o'/'O' were; survey_binary) - set_comments wrote only the disassembly comment, so comments never appeared in pseudocode - xref_query returned rows in raw IDA order, and 'follow the call' silently followed the fall-through instead - rename accepted one edit per category, so bulk symbol import was dead - decompile ran decomp_map's full per-column ctree sweep to fill in a per-line address anchor - heads shipped without operand extents or the digest protocol - the package became unimportable without ida_codemode installed, which killed the offline test suites Verified against the pre-codemode tag rather than against assumptions: the full suite is 788 passed / 0 failed, and the pilot's 301 checks match the old backend exactly. Performance is within 2x on the listing hot path and faster on decompile, disasm and connect, after fixing two runtime costs that are documented for upstream in docs/CODEMODE_UPSTREAM.md. Test runtime came down from ~9m20s to 115s along the way -- not by removing checks, but by removing four kinds of waiting-on-a-guess that were also hiding real failures.
Diffstat (limited to 'experiments/worker_smoke.py')
-rw-r--r--experiments/worker_smoke.py100
1 files changed, 49 insertions, 51 deletions
diff --git a/experiments/worker_smoke.py b/experiments/worker_smoke.py
index b215a57..9b55138 100644
--- a/experiments/worker_smoke.py
+++ b/experiments/worker_smoke.py
@@ -1,58 +1,56 @@
-"""Runnable read-path smoke: drives the REAL domain.Program through WorkerClient
-(our idalib worker over a unix socket). Run when idalib can spawn:
- ~/ida-venv/bin/python experiments/worker_smoke.py
-"""
-import os, sys, shutil, time
-REPO=os.path.expanduser("~/dev/ida-tui-maybe"); sys.path.insert(0, REPO); os.chdir(REPO)
-# fresh copy so the worker's idalib doesn't fight any running server
-src=f"{REPO}/targets/echo"; tmp="/tmp/echo_worker"
-shutil.copy(src, tmp)
-for e in ".i64 .id0 .id1 .id2 .nam .til".split():
- try: os.remove(tmp+e)
- except OSError: pass
-
-from idatui.worker_client import WorkerClient
-from idatui.domain import Program
-
-print("spawning worker + opening echo…", flush=True)
-t=time.time()
-cl=WorkerClient(tmp)
-cl.connect(progress=lambda m: None)
-print(f" worker ready in {time.time()-t:.2f}s session={cl.resolve_db()}", flush=True)
-prog=Program(cl)
+"""Exercise the real domain.Program through an IDA Code Mode lease.
-# --- drive the REAL domain layer through the worker (read path) ---
-main=prog.resolve("main")
-print("resolve('main') =", hex(main), flush=True)
-
-idx=prog.functions(); idx.load_all()
-print("functions() ->", len(idx), "funcs", flush=True)
-
-fn=prog.function_of(main)
-print("function_of(main) ->", fn.name, hex(fn.addr), "size", fn.size, flush=True)
+A matching registered GUI is reused; otherwise Code Mode starts a managed
+idalib worker. Usage: ``uv run python experiments/worker_smoke.py FILE``.
+"""
+from __future__ import annotations
-b=prog.read_bytes(main, 16)
-print("read_bytes(main,16) ->", b.hex(), flush=True)
+import os
+import sys
+import time
-lm=prog.listing(main)
-for _ in range(3): lm.load_next_page()
-rows=[lm.get(i) for i in range(min(6,len(lm)))]
-print("listing() first rows:", flush=True)
-for h in rows:
- if h: print(" ", hex(h.ea), h.kind, repr(h.text[:44]), flush=True)
+from idatui.codemode_client import CodeModeClient
+from idatui.domain import Program
-d=prog.decompile(main)
-print("decompile(main) -> failed?", d.failed, "lines:", len((d.code or '').splitlines()), flush=True)
-regs=prog.file_regions()
-print("file_regions ->", len(regs), "segments", flush=True)
+def main() -> int:
+ target = os.path.abspath(sys.argv[1] if len(sys.argv) > 1 else "experiments/fibonacci.elf")
+ print(f"attaching Code Mode to {target}…", flush=True)
+ started = time.time()
+ client = CodeModeClient(target)
+ client.connect(progress=lambda message: print(f" {message}", flush=True))
+ print(
+ f" ready in {time.time() - started:.2f}s; backend={client.backend}; "
+ f"session={client.resolve_db()}",
+ flush=True,
+ )
+ program = Program(client)
+ try:
+ index = program.functions()
+ index.load_all()
+ print(f"functions() -> {len(index)}", flush=True)
+ first = index.get(0)
+ if first is None:
+ print("VERDICT: FAIL — no functions", flush=True)
+ return 1
+ fn = program.function_of(first.addr)
+ data = program.read_bytes(first.addr, 16)
+ decompilation = program.decompile(first.addr)
+ print(f"function_of() -> {fn}", flush=True)
+ print(f"read_bytes() -> {data.hex()}", flush=True)
+ print(
+ f"decompile() -> failed={decompilation.failed}; "
+ f"lines={len((decompilation.code or '').splitlines())}",
+ flush=True,
+ )
+ print(f"file_regions() -> {len(program.file_regions())}", flush=True)
+ ok = fn is not None and bool(data) and bool(program.file_regions())
+ print(f"VERDICT: {'OK' if ok else 'FAIL'}", flush=True)
+ return 0 if ok else 1
+ finally:
+ program.close()
+ client.close()
-# xrefs to a called function
-callee=next((f.addr for f in idx.all_loaded() if f.name.startswith("sub_")), None)
-if callee:
- xr=prog.xrefs_to(callee)
- print("xrefs_to(", hex(callee), ") ->", len(xr), "refs", flush=True)
-ok = (fn.name=="main" and len(idx)>100 and b and not d.failed and len(regs)>0)
-print("VERDICT:", "OK — domain.Program runs unchanged on the worker" if ok else "FAIL", flush=True)
-cl.close()
+if __name__ == "__main__":
+ raise SystemExit(main())