ida-code 0.2.1__tar.gz

ida_code-0.2.1/.gitignore

@@ -0,0 +1,7 @@
+__pycache__/
+*.pyc
+*.pyo
+.venv/
+dist/
+*.egg-info/
+.mcp.json

ida_code-0.2.1/CHANGELOG.md

@@ -0,0 +1,84 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+Format based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
+
+## [Unreleased]
+
+## [0.2.1] - 2026-05-05
+
+### Added
+
+- **PyPI publish workflow** — `.github/workflows/publish-pypi.yml` builds with `uv build` and publishes to PyPI via Trusted Publishing on every push to `main` whose `pyproject.toml` version differs from the latest released version. No-op when versions match.
+- **Credits section** in README thanking [@p41l](https://github.com/p41l) for ideas and cross-LLM testing.
+
+### Changed
+
+- **fastmcp v3** — Bumped dependency from `fastmcp>=2.0,<3` to `fastmcp>=3.0,<4`. No code changes required: the `FastMCP` constructor, `@mcp.tool` / `@mcp.resource` / `@mcp.prompt` decorators, `fastmcp.exceptions.ToolError`, `fastmcp.server.auth.DebugTokenVerifier`, and `mcp.run(transport="streamable-http"|"sse", host=..., port=...)` are all still supported in v3.
+- **README polish** — Compressed install to a single primary path (`uv tool install ida-code`, with `pip` fallback), added a per-OS `IDA_INSTALL_DIR` table, moved the tool inventory above transport/env sections, and pushed source-install + fastmcp note to the bottom.
+
+## [0.2.0] - 2026-04-30
+
+### Added
+
+- **PyPI release prep** — Full PEP 639 metadata in `pyproject.toml` (license expression, classifiers, urls, authors, keywords, readme). Hatchling sdist allowlist excludes dev-only files (`.claude/`, `.mcp.json`, `CLAUDE.md`, `TODO.md`, `uv.lock`). README rewritten as a slim user-facing reference (~130 lines, table-of-tools instead of 30 per-tool sections).
+- **Friendlier `idapro` import error** — When `IDA_INSTALL_DIR` doesn't point at a valid IDA Pro install, the server now exits with a clear message naming the directory it tried instead of an opaque `ModuleNotFoundError`.
+- **`.mcp.json.example`** — Committed configuration template with placeholder paths. The real `.mcp.json` is now gitignored to avoid leaking local paths.
+- **Word-boundary matching** — `search_docs` and `search_examples` now use word-boundary-aware matching. Searching for "set" matches `set_name` and `ida_name.set_name` but not "reset", "offset", or "unset". Boundaries are start-of-string, underscore, dot, and whitespace.
+- **Field-weighted scoring in `search_docs`** — Title matches now score 4x higher than body matches, and Python API name matches score 5x higher. Previously all matches scored equally. Includes all-terms-match bonus (1.5x multiplier).
+- **Cross-linking docs → examples** — `search_docs` now includes a `related_examples` key with up to 2 matching example scripts. Controlled via `include_examples` parameter (default `True`).
+- **Server instructions** — Added `instructions` parameter to FastMCP server with workflow guidance (open → list → decompile → annotate → iterate). Visible to LLMs in the MCP `initialize` response.
+- **Literal types for enum parameters** — `comment_type`, `category`, and `level` parameters now use `Literal` types, exposing valid values in the JSON schema `enum` field instead of only in docstrings.
+- **`rename_function` tool** — Rename a function by name or address. Returns old and new names.
+- **`retype_function` tool** — Change a function's type signature with a C type string.
+- **`get_xrefs_to` tool** — Get cross-references to an address (who calls/references this?). Returns typed xref list with human-readable type names.
+- **`get_xrefs_from` tool** — Get cross-references from an address (what does this call/reference?).
+- **`get_strings` tool** — List strings in the database with min-length and filter options.
+- **`get_imports` tool** — List all imported functions grouped by module.
+- **`get_exports` tool** — List all exported functions/symbols with ordinals.
+- **Database file guard** — Before every IDA API call, `session.require_open()` now checks that the `.i64`/`.idb` database file still exists on disk. If the file has been moved or deleted, the server resets internal state and returns a clean `ToolError` instead of letting idalib segfault and crash the process. All tools that require an open database use this centralized check.
+- **`open_database` `overwrite` flag** — Delete existing `.i64`/`.idb` database files before opening, forcing a fresh analysis from the original binary.
+- **`close_database` tool** — Explicitly close the current database and free resources. Clears the executor namespace.
+- **`execute_file` tool** — Run IDAPython script files directly by path. Optional `args` parameter for inline follow-up code in the same namespace.
+- **Coding guidelines resources** — Three MCP Resources (`guidelines://standalone_script`, `guidelines://plugin`, `guidelines://idapython_script`) providing architecture templates and best practices for writing standalone idalib scripts, IDA plugins, and classic IDAPython scripts.
+- **Execution timeout** — `execute` and `execute_file` now enforce a wall-clock timeout (default 30s, configurable via `timeout` parameter, 0 = unlimited). Prevents infinite loops from hanging the server.
+- **Process-killing exception guard** — `SystemExit` and `KeyboardInterrupt` raised by user code are now intercepted and returned as error text instead of killing the server process.
+- **`decompile` tool** — Decompile a function by name or address. Resolves names via `ida_name`, accepts hex/decimal addresses, returns pseudocode with a header comment. Requires Hex-Rays.
+- **Structured logging** — All modules now use Python `logging`. Output goes to stderr (won't interfere with stdio MCP transport). Controlled via `LOG_LEVEL` env var (default `WARNING`).
+- **Unit tests** — 35 tests covering executor (output capture, timeout, exception handling, namespace persistence, truncation) and doc_search (HTML stripping, scoring, excerpt extraction). Run with `uv run pytest`.
+- **`open_database` timeout** — New `timeout` parameter (default 0 = unlimited) limits auto-analysis wait time. When the timeout expires, the database stays open with partial analysis and a warning is appended. Progress (function count) is logged during analysis.
+- **`get_database_info` tool** — Read-only tool returning current database summary (processor, segments, entry points, function count) without opening or closing anything.
+- **`list_functions` tool** — Paginated function listing with address, size, and name. Supports `offset`/`limit` pagination and case-insensitive name `filter`.
+- **REPL-like expression output** — `execute` now returns the `repr()` of the last expression if it's a bare expression (not an assignment or statement), just like the interactive Python prompt. No need to wrap everything in `print()`.
+- **Database snapshots** — Four new tools for checkpointing and rolling back database state: `list_snapshots`, `create_snapshot`, `restore_snapshot`, `remove_snapshot`. Built on `ida_loader.snapshot_t` and `ida_kernwin.take_database_snapshot` / `restore_database_snapshot`.
+- **`search_examples` tool** — Search 125 official IDAPython example scripts. Indexes metadata from `index.md` (title, description, keywords, APIs used, level, category) and AST-parses each `.py` file for imports, definitions, and `ida_*` API call patterns. Weighted scoring ranks API matches highest. Supports `category` and `level` filters.
+- **Structure management tools** — Five new tools for managing IDA structs and unions: `list_structures` (paginated listing with filter), `get_structure` (detailed info with C definition), `create_structure` (from C definition string via `idc.parse_decls`), `edit_structure` (replace existing definition), `delete_structure` (remove from type library).
+- **Variable management tools** — Two new tools for inspecting and modifying variables: `get_variable` (read local or global variable info) and `set_variable` (rename and/or retype). Local variables use Hex-Rays decompiler APIs (`ida_hexrays`); globals use `ida_name` and `ida_typeinf`.
+- **Comment management tools** — Three new tools for managing comments: `get_comment` (read one or all comment types at an address), `set_comment` (write a comment), `delete_comment` (remove a comment). Supports all five IDA comment types: regular, repeatable, function, anterior, and posterior.
+- **Undo/redo tools** — Three new tools for undoing and redoing database changes: `get_undo_status` (check availability and next action labels), `perform_undo` (undo one or more steps), `perform_redo` (redo one or more steps). Built on `ida_undo`. Multi-step undo/redo in a single call with partial success support.
+- **MCP prompts** — Two new MCP prompts for guided workflows: `reverse_engineer` (comprehensive five-phase binary analysis workflow covering reconnaissance, triage, deep analysis, annotation, and iteration) and `create_script` (coding guidelines for standalone scripts, plugins, or IDAPython scripts plus IDAPython best practices for error handling, performance, naming conventions, and common pitfalls).
+
+### Changed
+
+- **`remove_snapshot` → `delete_snapshot`** — Renamed for consistency with `delete_structure` and `delete_comment`.
+- **`filter` → `name_filter`** — Renamed in `list_functions` and `list_structures` to avoid shadowing the Python builtin and clarify semantics.
+- **`id` → `snapshot_id`** — Renamed in `restore_snapshot` and `delete_snapshot` to avoid shadowing the Python builtin and be self-documenting.
+- **`function` → `scope`** — Renamed in `get_variable` and `set_variable` to clarify it's the containing scope, not the target variable.
+- **Tool descriptions** — All 30 tools that require an open database now say "Requires an open database" in the first line. Added `Returns:` lines listing dict keys. Enriched `execute` description with exhaustive pre-imported module list. Improved `search_docs` and `search_examples` descriptions to clarify when and why to use them.
+- **`reverse_engineer` prompt** — Updated to reference new dedicated tools (`get_strings`, `get_imports`, `get_exports`, `get_xrefs_to`, `get_xrefs_from`, `rename_function`, `retype_function`) instead of raw `execute` boilerplate. Recommends dedicated tools over `execute` where available.
+- **`execute` behavior** — Last-expression values are now auto-printed. `None` results are suppressed. Explicit `print()` calls still work as before.
+- **Auto-import idalib on startup** — `idapro` is now loaded automatically from `IDA_INSTALL_DIR/idalib/python/` and `IDADIR` is set via `os.environ.setdefault`. No manual `pip install` or `py-activate-idalib.py` needed.
+
+## [0.1.0] - 2026-02-10
+
+Initial release.
+
+### Added
+
+- **`open_database` tool** — Open binaries and IDA databases via idalib with auto-analysis support. Returns summary info: processor type, bitness, segments, entry points, function count.
+- **`execute` tool** — Run arbitrary IDAPython code with persistent namespace across calls. Pre-imports common `ida_*` modules. Captures stdout/stderr with 50K character truncation.
+- **`search_docs` tool** — Keyword search over IDA's HTML documentation (2628 indexed pages) and IDAPython API source files (`ida_*.py` signatures and docstrings).
+- Implicit database lifecycle management (close-on-open, atexit cleanup).
+- stdio MCP transport via fastmcp.
+- Claude Code `.mcp.json` configuration template.

ida_code-0.2.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Dil4rd
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ida_code-0.2.1/PKG-INFO

@@ -0,0 +1,167 @@
+Metadata-Version: 2.4
+Name: ida-code
+Version: 0.2.1
+Summary: MCP server for AI-assisted IDAPython scripting via idalib
+Project-URL: Homepage, https://github.com/Dil4rd/ida-code
+Project-URL: Repository, https://github.com/Dil4rd/ida-code
+Project-URL: Issues, https://github.com/Dil4rd/ida-code/issues
+Project-URL: Changelog, https://github.com/Dil4rd/ida-code/blob/main/CHANGELOG.md
+Author: Dil4rd
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ida,ida-pro,idalib,idapython,mcp,reverse-engineering
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Information Technology
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Security
+Classifier: Topic :: Software Development :: Disassemblers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.12
+Requires-Dist: fastmcp<4,>=3.0
+Requires-Dist: lief>=0.15
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# ida-code
+
+MCP server that lets AI coding agents interact with IDA Pro. Open binaries, decompile, run IDAPython, search the API docs — all through tool calls.
+
+Built on [idalib](https://docs.hex-rays.com/developer-guide/idalib) for headless in-process operation and [fastmcp](https://github.com/jlowin/fastmcp) for the MCP transport.
+
+> **Requires** a licensed IDA Pro 9.2+ with idalib support. `ida-code` does not install or replace IDA Pro — it loads `idapro` from your existing install at startup.
+
+## Install
+
+```bash
+uv tool install ida-code
+```
+
+This puts the `ida-code` CLI on your `PATH` (via `uv`'s tool dir) so MCP clients can launch it directly. Don't have uv? `pip install ida-code` works too.
+
+Then point `IDA_INSTALL_DIR` at your IDA Pro install (the directory that contains `idalib/python/`):
+
+| OS | Typical path |
+|---|---|
+| Linux | `/opt/ida-pro-9.3` |
+| macOS | `/Applications/IDA Professional 9.3.app/Contents/MacOS` |
+| Windows | `C:\Program Files\IDA Professional 9.3` |
+
+## Use with Claude Code
+
+Add `ida-code` to your project's `.mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "ida-code": {
+      "command": "ida-code",
+      "env": {
+        "IDA_INSTALL_DIR": "/opt/ida-pro-9.3"
+      }
+    }
+  }
+}
+```
+
+Restart Claude Code; the server is picked up automatically. You can confirm it's wired up by asking Claude to open a binary — it should call `open_database` and report architecture, entry point, and load address.
+
+For other MCP clients, run the server directly:
+
+```bash
+IDA_INSTALL_DIR=/opt/ida-pro-9.3 ida-code   # stdio transport
+```
+
+## Tools (35)
+
+Full parameter docs live in each tool's docstring — surfaced automatically to MCP clients via `tools/list`.
+
+| Domain | Tools |
+|---|---|
+| Database | `open_database`, `close_database`, `get_database_info`, `list_architectures` |
+| Code execution | `execute`, `execute_file` |
+| Navigation | `list_functions`, `decompile`, `get_disassembly`, `get_xrefs_to`, `get_xrefs_from` |
+| Annotation | `rename_function`, `retype_function`, `get_comment`, `set_comment`, `delete_comment`, `get_variable`, `set_variable` |
+| Structures | `list_structures`, `get_structure`, `create_structure`, `edit_structure`, `delete_structure` |
+| Snapshots | `list_snapshots`, `create_snapshot`, `restore_snapshot`, `delete_snapshot` |
+| Undo/redo | `get_undo_status`, `perform_undo`, `perform_redo` |
+| Inventory | `get_strings`, `get_imports`, `get_exports` |
+| Search | `search_docs`, `search_examples` |
+
+## Resources & prompts
+
+| Type | URI / name | Purpose |
+|---|---|---|
+| Resource | `guidelines://standalone_script` | Boilerplate for standalone idalib scripts |
+| Resource | `guidelines://plugin` | Boilerplate for IDA plugins (`plugin_t`) |
+| Resource | `guidelines://idapython_script` | Boilerplate for IDAPython scripts run inside IDA GUI |
+| Prompt | `reverse_engineer` | Five-phase RE workflow (recon, triage, analysis, annotation, iteration) |
+| Prompt | `create_script` | Coding guidelines for a chosen target script type |
+
+## Transport modes
+
+```bash
+ida-code                          # stdio (default)
+ida-code --http                   # streamable-http on 127.0.0.1:8080
+ida-code --http 0.0.0.0:9090      # custom host:port
+ida-code --sse                    # SSE on 127.0.0.1:8080
+ida-code --sse :9090              # SSE on 127.0.0.1:9090
+```
+
+HTTP/SSE require bearer token auth. Set `MCP_AUTH_TOKEN` or let the server generate one (printed to stderr on startup).
+
+## Environment variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `IDA_INSTALL_DIR` | `/opt/ida-pro-9.3` | IDA Pro installation directory (must contain `idalib/python/`) |
+| `LOG_LEVEL` | `WARNING` | Logging verbosity (`DEBUG`, `INFO`, `WARNING`, `ERROR`) |
+| `MCP_AUTH_TOKEN` | (auto-generated) | Bearer token for HTTP/SSE transports |
+
+Doc and example paths are derived from `IDA_INSTALL_DIR` (`docs/`, `python/`, `python/examples/`).
+
+## Install from source
+
+```bash
+git clone https://github.com/Dil4rd/ida-code
+cd ida-code
+uv sync
+uv run ida-code
+```
+
+When wiring a source checkout into `.mcp.json`, use `uv` as the command:
+
+```json
+{
+  "mcpServers": {
+    "ida-code": {
+      "command": "uv",
+      "args": ["run", "--directory", "/path/to/ida-code", "ida-code"],
+      "env": { "IDA_INSTALL_DIR": "/opt/ida-pro-9.3" }
+    }
+  }
+}
+```
+
+> **Note:** the `fastmcp` dependency is the [community fastmcp](https://github.com/jlowin/fastmcp) package, not the official `mcp` SDK. Don't install `mcp` by mistake.
+
+## Development
+
+```bash
+uv sync --extra dev
+uv run pytest
+```
+
+The test suite covers the executor, doc/example search, comments, snapshots, structures, undo, variables, and Mach-O parsing. Tests that need idalib are skipped if it's not available.
+
+## Credits
+
+Thanks to [@p41l](https://github.com/p41l) for ideas and for testing the tool across different LLMs.
+
+## License
+
+MIT — see `LICENSE`.

ida_code-0.2.1/README.md

@@ -0,0 +1,138 @@
+# ida-code
+
+MCP server that lets AI coding agents interact with IDA Pro. Open binaries, decompile, run IDAPython, search the API docs — all through tool calls.
+
+Built on [idalib](https://docs.hex-rays.com/developer-guide/idalib) for headless in-process operation and [fastmcp](https://github.com/jlowin/fastmcp) for the MCP transport.
+
+> **Requires** a licensed IDA Pro 9.2+ with idalib support. `ida-code` does not install or replace IDA Pro — it loads `idapro` from your existing install at startup.
+
+## Install
+
+```bash
+uv tool install ida-code
+```
+
+This puts the `ida-code` CLI on your `PATH` (via `uv`'s tool dir) so MCP clients can launch it directly. Don't have uv? `pip install ida-code` works too.
+
+Then point `IDA_INSTALL_DIR` at your IDA Pro install (the directory that contains `idalib/python/`):
+
+| OS | Typical path |
+|---|---|
+| Linux | `/opt/ida-pro-9.3` |
+| macOS | `/Applications/IDA Professional 9.3.app/Contents/MacOS` |
+| Windows | `C:\Program Files\IDA Professional 9.3` |
+
+## Use with Claude Code
+
+Add `ida-code` to your project's `.mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "ida-code": {
+      "command": "ida-code",
+      "env": {
+        "IDA_INSTALL_DIR": "/opt/ida-pro-9.3"
+      }
+    }
+  }
+}
+```
+
+Restart Claude Code; the server is picked up automatically. You can confirm it's wired up by asking Claude to open a binary — it should call `open_database` and report architecture, entry point, and load address.
+
+For other MCP clients, run the server directly:
+
+```bash
+IDA_INSTALL_DIR=/opt/ida-pro-9.3 ida-code   # stdio transport
+```
+
+## Tools (35)
+
+Full parameter docs live in each tool's docstring — surfaced automatically to MCP clients via `tools/list`.
+
+| Domain | Tools |
+|---|---|
+| Database | `open_database`, `close_database`, `get_database_info`, `list_architectures` |
+| Code execution | `execute`, `execute_file` |
+| Navigation | `list_functions`, `decompile`, `get_disassembly`, `get_xrefs_to`, `get_xrefs_from` |
+| Annotation | `rename_function`, `retype_function`, `get_comment`, `set_comment`, `delete_comment`, `get_variable`, `set_variable` |
+| Structures | `list_structures`, `get_structure`, `create_structure`, `edit_structure`, `delete_structure` |
+| Snapshots | `list_snapshots`, `create_snapshot`, `restore_snapshot`, `delete_snapshot` |
+| Undo/redo | `get_undo_status`, `perform_undo`, `perform_redo` |
+| Inventory | `get_strings`, `get_imports`, `get_exports` |
+| Search | `search_docs`, `search_examples` |
+
+## Resources & prompts
+
+| Type | URI / name | Purpose |
+|---|---|---|
+| Resource | `guidelines://standalone_script` | Boilerplate for standalone idalib scripts |
+| Resource | `guidelines://plugin` | Boilerplate for IDA plugins (`plugin_t`) |
+| Resource | `guidelines://idapython_script` | Boilerplate for IDAPython scripts run inside IDA GUI |
+| Prompt | `reverse_engineer` | Five-phase RE workflow (recon, triage, analysis, annotation, iteration) |
+| Prompt | `create_script` | Coding guidelines for a chosen target script type |
+
+## Transport modes
+
+```bash
+ida-code                          # stdio (default)
+ida-code --http                   # streamable-http on 127.0.0.1:8080
+ida-code --http 0.0.0.0:9090      # custom host:port
+ida-code --sse                    # SSE on 127.0.0.1:8080
+ida-code --sse :9090              # SSE on 127.0.0.1:9090
+```
+
+HTTP/SSE require bearer token auth. Set `MCP_AUTH_TOKEN` or let the server generate one (printed to stderr on startup).
+
+## Environment variables
+
+| Variable | Default | Description |
+|---|---|---|
+| `IDA_INSTALL_DIR` | `/opt/ida-pro-9.3` | IDA Pro installation directory (must contain `idalib/python/`) |
+| `LOG_LEVEL` | `WARNING` | Logging verbosity (`DEBUG`, `INFO`, `WARNING`, `ERROR`) |
+| `MCP_AUTH_TOKEN` | (auto-generated) | Bearer token for HTTP/SSE transports |
+
+Doc and example paths are derived from `IDA_INSTALL_DIR` (`docs/`, `python/`, `python/examples/`).
+
+## Install from source
+
+```bash
+git clone https://github.com/Dil4rd/ida-code
+cd ida-code
+uv sync
+uv run ida-code
+```
+
+When wiring a source checkout into `.mcp.json`, use `uv` as the command:
+
+```json
+{
+  "mcpServers": {
+    "ida-code": {
+      "command": "uv",
+      "args": ["run", "--directory", "/path/to/ida-code", "ida-code"],
+      "env": { "IDA_INSTALL_DIR": "/opt/ida-pro-9.3" }
+    }
+  }
+}
+```
+
+> **Note:** the `fastmcp` dependency is the [community fastmcp](https://github.com/jlowin/fastmcp) package, not the official `mcp` SDK. Don't install `mcp` by mistake.
+
+## Development
+
+```bash
+uv sync --extra dev
+uv run pytest
+```
+
+The test suite covers the executor, doc/example search, comments, snapshots, structures, undo, variables, and Mach-O parsing. Tests that need idalib are skipped if it's not available.
+
+## Credits
+
+Thanks to [@p41l](https://github.com/p41l) for ideas and for testing the tool across different LLMs.
+
+## License
+
+MIT — see `LICENSE`.

ida_code-0.2.1/pyproject.toml

@@ -0,0 +1,60 @@
+[project]
+name = "ida-code"
+version = "0.2.1"
+description = "MCP server for AI-assisted IDAPython scripting via idalib"
+readme = "README.md"
+requires-python = ">=3.12"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [{name = "Dil4rd"}]
+keywords = ["ida-pro", "ida", "idapython", "reverse-engineering", "mcp", "idalib"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Information Technology",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Security",
+    "Topic :: Software Development :: Disassemblers",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = ["fastmcp>=3.0,<4", "lief>=0.15"]
+
+[project.urls]
+Homepage = "https://github.com/Dil4rd/ida-code"
+Repository = "https://github.com/Dil4rd/ida-code"
+Issues = "https://github.com/Dil4rd/ida-code/issues"
+Changelog = "https://github.com/Dil4rd/ida-code/blob/main/CHANGELOG.md"
+
+[project.scripts]
+ida-code = "ida_code.server:main"
+
+[project.optional-dependencies]
+dev = ["pytest>=8.0"]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+filterwarnings = [
+    "ignore:builtin type SwigPyPacked has no __module__ attribute:DeprecationWarning",
+    "ignore:builtin type SwigPyObject has no __module__ attribute:DeprecationWarning",
+    "ignore:builtin type swigvarlink has no __module__ attribute:DeprecationWarning",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/ida_code"]
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "src/",
+    "tests/",
+    "README.md",
+    "LICENSE",
+    "CHANGELOG.md",
+    "pyproject.toml",
+]

ida_code-0.2.1/src/ida_code/__init__.py

@@ -0,0 +1,2 @@
+# Import session first to ensure idapro is loaded before any other ida_* imports.
+from ida_code import session  # noqa: F401

ida_code-0.2.1/src/ida_code/_search_utils.py

@@ -0,0 +1,33 @@
+"""Shared search utilities for doc_search and example_search.
+
+Provides word-boundary-aware matching: "set" matches "set_name" and
+"ida_name.set_name" but not "reset" or "offset".
+
+Boundaries are: start-of-string, underscore, dot, whitespace.
+"""
+
+import re
+from functools import lru_cache
+
+
+@lru_cache(maxsize=128)
+def _boundary_pattern(term: str) -> re.Pattern:
+    """Regex that matches term at an underscore/dot/whitespace boundary."""
+    escaped = re.escape(term)
+    return re.compile(rf"(?:^|[_.\s]){escaped}", re.IGNORECASE)
+
+
+def term_matches(term: str, text: str) -> bool:
+    """Check if term appears in text at a word boundary.
+
+    Boundaries are: start-of-string, underscore, dot, whitespace.
+    Fast path: rejects via substring check before running regex.
+
+    Dotted terms (e.g. "ida_funcs.get_func") use plain substring
+    matching since the dot is already specific enough.
+    """
+    if term not in text.lower():
+        return False
+    if "." in term:
+        return True  # dotted terms: substring match is sufficient
+    return _boundary_pattern(term).search(text) is not None