1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
"""Render the graph view headless at a chosen size and print it.
~/ida-venv/bin/python experiments/graph_shot.py [func] [cols] [rows] [zoom]
The pane a person runs this in is usually too small to judge the layout, and the
pilot lays out synchronously at whatever size you ask for -- so this is the way
to actually look at the thing.
"""
import asyncio
import os
import shutil
import sys
REPO = os.path.expanduser("~/dev/ida-tui-maybe")
sys.path.insert(0, REPO)
os.chdir(REPO)
func = sys.argv[1] if len(sys.argv) > 1 else "sub_2297"
cols = int(sys.argv[2]) if len(sys.argv) > 2 else 170
rows = int(sys.argv[3]) if len(sys.argv) > 3 else 55
zoom = int(sys.argv[4]) if len(sys.argv) > 4 else 0
src, tmp = f"{REPO}/targets/echo", "/tmp/echo_shot"
shutil.copy(src, tmp)
for e in ".i64 .id0 .id1 .id2 .nam .til".split():
try:
os.remove(tmp + e)
except OSError:
pass
from idatui.app import IdaTui, GraphView # noqa: E402
from idatui._sync import wait_for # noqa: E402
from idatui.rpc import screen_text # noqa: E402
async def main() -> None:
app = IdaTui(tmp, keepalive=False)
async with app.run_test(size=(cols, rows)) as pilot:
await wait_for(lambda: app.program is not None and app._cur is not None,
pilot.pause, 120)
ea = app.program.resolve(func)
fn = app.program.function_of(ea)
app._open_function(fn.addr, fn.name)
await wait_for(lambda: app._cur is not None and app._cur.ea == fn.addr,
pilot.pause, 60)
await pilot.pause(0.2)
await pilot.press("space")
gv = app.query_one(GraphView)
ok = await wait_for(lambda: app._active == "graph" and gv.lay is not None,
pilot.pause, 90)
if not ok:
print("graph never opened:", app.query_one("#status").render())
return
for _ in range(zoom):
await pilot.press("z")
await pilot.pause(0.2)
await pilot.pause(0.4)
print(screen_text(app)["text"])
print()
print("status:", app.query_one("#status").render())
print("stats :", gv.lay.stats, "canvas",
f"{gv.lay.width}x{gv.lay.height}", "zoom", gv.ZOOMS[gv._zoom])
app._save_on_exit = False
asyncio.run(main())
|
