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
80
81
82
83
84
85
86
87
|
#!/usr/bin/env python3
"""Smoke test for idatui.pane's supervisor auto-start machinery (no IDA needed).
python3 tests/pane_smoke.py
Uses IDATUI_SERVER_CMD to launch a dummy port-binder in a tmux pane instead of
the real ./spawn.sh, so we can exercise _ensure_server end-to-end (start, detect,
idempotency, remote guard) without a real ida-pro-mcp server. Must run in tmux.
"""
import os
import socket
import subprocess
import sys
import tempfile
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from idatui import pane # noqa: E402
PASS = FAIL = 0
def check(name, cond, detail=""):
global PASS, FAIL
if cond:
PASS += 1
print(f" ok {name}")
else:
FAIL += 1
print(f" FAIL {name} {detail}")
def _free_port() -> int:
s = socket.socket()
s.bind(("127.0.0.1", 0))
p = s.getsockname()[1]
s.close()
return p
def main() -> int:
if not os.environ.get("TMUX"):
print("error: must run inside tmux", file=sys.stderr)
return 2
# a dummy "supervisor": bind the port and idle, so _server_up sees it.
fake = tempfile.NamedTemporaryFile("w", suffix=".py", delete=False)
fake.write(
"import socket,sys,time\n"
"s=socket.socket(); s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)\n"
"s.bind(('127.0.0.1',int(sys.argv[1]))); s.listen(); time.sleep(300)\n")
fake.close()
port = _free_port()
os.environ["IDATUI_SERVER_CMD"] = f"{sys.executable} {fake.name} {port}"
server_pane = None
try:
check("port starts down", not pane._server_up("127.0.0.1", port))
srv = pane._ensure_server("127.0.0.1", port, timeout=15.0)
server_pane = srv.get("server_pane")
check("ensure_server starts the supervisor and it comes up",
srv.get("server_started") and srv.get("server_up") and server_pane,
str(srv))
check("port is now up", pane._server_up("127.0.0.1", port))
srv2 = pane._ensure_server("127.0.0.1", port, timeout=5.0)
check("ensure_server is idempotent when already up",
srv2.get("server_started") is False and srv2.get("server_up") is True,
str(srv2))
srv3 = pane._ensure_server("10.255.255.1", 9, timeout=2.0)
check("remote+down server is not auto-started",
srv3.get("server_started") is False and "local" in (srv3.get("error") or ""),
str(srv3))
finally:
if server_pane:
subprocess.run(["tmux", "kill-pane", "-t", server_pane],
capture_output=True)
os.unlink(fake.name)
os.environ.pop("IDATUI_SERVER_CMD", None)
print(f"\n{PASS} passed, {FAIL} failed")
return 1 if FAIL else 0
if __name__ == "__main__":
raise SystemExit(main())
|
