blob: a5bc098ad3ca85cfc8fa194b31d4f6c04b079d71 (
plain) (
blame)
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
|
#!/usr/bin/env bash
# Control the ida-pro-mcp server for stress testing. Assumes ~/ida-venv + spawn.sh.
set -u
REPO="$(cd "$(dirname "$0")/.." && pwd)"
PORT=8745
LOG=/tmp/ida-stress-spawn.log
wait_port() { # wait_port <up|down> <secs>
local want="$1" secs="${2:-40}" i
for ((i=0; i<secs*2; i++)); do
if ss -ltn 2>/dev/null | grep -q ":$PORT "; then
[ "$want" = up ] && return 0
else
[ "$want" = down ] && return 0
fi
sleep 0.5
done
return 1
}
wait_ready() { # wait until hexrays_ready via the client
local secs="${1:-60}" i
for ((i=0; i<secs; i++)); do
if python3 -c "
import sys; sys.path.insert(0,'$REPO')
from idatui.client import IDAClient
try:
c=IDAClient(); c.connect()
ss=[s for s in c.list_sessions() if s.session_id]
if ss:
c.set_db(ss[0].session_id)
if c.health().get('hexrays_ready'): print(ss[0].session_id); sys.exit(0)
except Exception: pass
sys.exit(1)
" 2>/dev/null; then return 0; fi
sleep 1
done
return 1
}
case "${1:-}" in
start)
cd "$REPO"
source ~/ida-venv/bin/activate 2>/dev/null
nohup ./spawn.sh >"$LOG" 2>&1 &
wait_port up 40 || { echo "PORT_TIMEOUT"; tail -5 "$LOG"; exit 1; }
wait_ready 90 || { echo "READY_TIMEOUT"; tail -5 "$LOG"; exit 1; }
;;
stop)
# Kill the whole tree: uv wrapper, supervisor, worker.
pkill -9 -f 'idalib_server' 2>/dev/null
pkill -9 -f 'idalib-mcp' 2>/dev/null
pkill -9 -f 'uv run idalib' 2>/dev/null
wait_port down 20 || { echo "STOP_TIMEOUT"; exit 1; }
;;
kill9)
# Hard kill only the worker (simulate a crash of the analysis process).
pkill -9 -f 'idalib_server' 2>/dev/null
;;
ready)
wait_ready "${2:-60}"
;;
*)
echo "usage: $0 {start|stop|kill9|ready [secs]}"; exit 2;;
esac
|
