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
67
68
69
70
71
72
73
74
75
76
77
78
79
|
"""Measure cold installation versus warm calls for typed remote modules.
Historical note: this file used to benchmark ``_PACK_EPILOGUE``. Application
scripts are no longer strings and packing is gone; the relevant design cost is
now the one-time content-addressed module installation versus steady-state calls.
Usage::
PYTHONPATH=. python experiments/bench_pack_trace.py [FILE]
"""
from __future__ import annotations
import argparse
import os
import statistics
import time
from idatui import remote_ops
from idatui.codemode_client import CodeModeClient
def timed(function, reps: int = 1) -> tuple[object, float]:
samples = []
result = None
for _ in range(reps):
started = time.perf_counter()
result = function()
samples.append((time.perf_counter() - started) * 1000.0)
return result, statistics.median(samples)
def main() -> int:
parser = argparse.ArgumentParser()
parser.add_argument("target", nargs="?", default="targets/bash")
parser.add_argument("--reps", type=int, default=25)
parser.add_argument("--rows", type=int, default=200)
args = parser.parse_args()
client = CodeModeClient(os.path.abspath(args.target)).connect()
index, operations_cold = timed(
lambda: client.call(remote_ops.list_funcs, queries=[{"offset": 0, "count": 40}])
)
funcs = (index.get("result") or [{}])[0].get("data") or []
biggest = max(funcs, key=lambda function: function.get("size") or 0, default=None)
if not biggest:
print("VERDICT: FAIL - no functions")
return 1
addr = biggest["addr"]
if isinstance(addr, int):
addr = hex(addr)
_, operations_warm = timed(
lambda: client.call(
remote_ops.list_funcs, queries=[{"offset": 0, "count": 40}]
),
args.reps,
)
page = lambda: client.call( # noqa: E731
remote_ops.heads, addr=addr, count=args.rows, annotate=True
)
payload, tools_cold = timed(page)
_, tools_warm = timed(page, args.reps)
print(f"target {os.path.basename(args.target)} backend={client.backend}")
print(f"function {addr}")
print(f"listing rows {len(payload.get('heads', []))}")
print(
f"operations.py cold {operations_cold:8.3f}ms warm {operations_warm:8.3f}ms"
)
print(f"remote_tools.py cold {tools_cold:8.3f}ms warm {tools_warm:8.3f}ms")
print(f"tools install overhead {tools_cold / max(tools_warm, 0.001):.2f}x one time")
client.close()
return 0
if __name__ == "__main__":
raise SystemExit(main())
|
