"""IDA Pro MCP Plugin Loader
This file serves as the entry point for IDA Pro's plugin system.
It loads the actual implementation from the ida_mcp package.
"""
import sys
import idaapi
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from . import ida_mcp
def unload_package(package_name: str):
"""Remove every module that belongs to the package from sys.modules."""
to_remove = [
mod_name
for mod_name in sys.modules
if mod_name == package_name or mod_name.startswith(package_name + ".")
]
for mod_name in to_remove:
del sys.modules[mod_name]
class MCP(idaapi.plugin_t):
flags = idaapi.PLUGIN_KEEP
comment = "MCP Plugin"
help = "MCP"
wanted_name = "MCP"
wanted_hotkey = "Ctrl-Alt-M"
# TODO: make these configurable
HOST = "127.0.0.1"
PORT = 13337
def init(self):
hotkey = MCP.wanted_hotkey.replace("-", "+")
if __import__("sys").platform == "darwin":
hotkey = hotkey.replace("Alt", "Option")
print(
f"[MCP] Plugin loaded, use Edit -> Plugins -> MCP ({hotkey}) to start the server"
)
self.mcp: "ida_mcp.rpc.McpServer | None" = None
return idaapi.PLUGIN_KEEP
def run(self, arg):
if self.mcp:
self.mcp.stop()
self.mcp = None
# HACK: ensure fresh load of ida_mcp package
unload_package("ida_mcp")
if TYPE_CHECKING:
from .ida_mcp import MCP_SERVER, IdaMcpHttpRequestHandler, init_caches
else:
from ida_mcp import MCP_SERVER, IdaMcpHttpRequestHandler, init_caches
try:
init_caches()
except Exception as e:
print(f"[MCP] Cache init failed: {e}")
try:
MCP_SERVER.serve(
self.HOST, self.PORT, request_handler=IdaMcpHttpRequestHandler
)
print(f" Config: http://{self.HOST}:{self.PORT}/config.html")
self.mcp = MCP_SERVER
except OSError as e:
if e.errno in (48, 98, 10048): # Address already in use
print(f"[MCP] Error: Port {self.PORT} is already in use")
else:
raise
def term(self):
if self.mcp:
self.mcp.stop()
def PLUGIN_ENTRY():
return MCP()
# IDA plugin flags
PLUGIN_FLAGS = idaapi.PLUGIN_HIDE | idaapi.PLUGIN_FIX
