summaryrefslogtreecommitdiffstats
path: root/debug/risc-v-gsp-rm-vm/examples/scripts/trace_rpc.py
blob: 12b912b357c6777898feeac12feeefa79464728f (plain)
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
"""Trace every uproc RPC handler with enter/leave hooks.

Run from the tool root:

    ./greeneye.sh run examples/scripts/trace_rpc.py

The script namespace provides ``s``/``session``, ``memory``, ``symbols``,
``recorder`` and ``ge`` (the greeneye package).
"""
NAMES = ["rpc_ping", "rpc_echo", "rpc_fill", "rpc_sum", "rpc_state", "rpc_fault"]
events = []

for name in NAMES:
    sym = symbols.get(name)
    if sym is None:
        print(f"symbol {name} not found")
        continue
    s.hook(
        sym.addr,
        on_enter=lambda hit: events.append(("enter", hit.label(), hit.arg(0))),
        on_leave=lambda hit: events.append(("leave", hit.label(), hit.retval)),
    )

s.run(duration=1.0)

print(f"{len(events)} hook events:")
for kind, label, value in events:
    print(f"  {kind:5} {label:14} {value:#018x}")
recorder.event("trace_complete", events=len(events))