BinjaLattice

BinjaLattice 是一个用于 Binary Ninja 的安全通信协议，它支持与外部模型上下文协议 (MCP) 服务器和工具进行交互。它提供了一种结构化的方式来从 Binary Ninja 获取信息，并且能够通过带有 REST API 的 HTTP 修改活动的 Binary Ninja 数据库。

演示

功能特性

• 安全认证: 基于令牌的认证系统
• 加密通信: 可选的 SSL/TLS 加密
• 二进制分析上下文: 导出伪代码、反汇编、变量名、二进制信息等。
• 二进制修改: 更新函数名、添加注释、重命名变量
• 令牌管理: 自动过期和更新认证令牌

安装

1. 将 lattice_server_plugin.py 复制到你的 Binary Ninja 插件目录：

   • Linux: ~/.binaryninja/plugins/
   • macOS: ~/Library/Application Support/Binary Ninja/plugins/
   • Windows: %APPDATA%\Binary Ninja\plugins\

2. 创建一个虚拟环境 pip -m venv venv-test (或者你喜欢的依赖管理工具)

3. 激活你的虚拟环境并安装所需的 Python 依赖：

   • 使用以下命令安装：pip install -r requirements.txt (或者你喜欢的方法)

使用

在 Binary Ninja 中启动服务器

1. 打开 Binary Ninja 并加载一个二进制文件
2. 前往 Plugins > Start Lattice Protocol Server
3. 服务器将启动并在日志控制台中显示 API 密钥
4. 在你的 MCP 配置中将 API 密钥设置为 BNJLAT 环境变量

来自 Cursor 的 MCP 配置示例 (mcp.json)：

{
    "mcpServers": {
      "binja-lattice-mcp": {
        "command": "/path/to/venv/bin/python",
        "args": ["/path/to/mcp_server.py"],
        "env": {
            "BNJLAT": "your_api_key_here"
        }
      }
    }
}

可用的 MCP 工具

以下工具可以通过 MCP 服务器使用：

• get_all_function_names: 获取二进制文件中所有函数名称的列表
• get_binary_info: 获取有关正在分析的二进制文件的信息
• update_function_name: 更新函数的名称
• add_comment_to_address: 向特定地址添加注释
• add_comment_to_function: 向函数添加注释
• get_function_disassembly: 获取函数的反汇编代码
• get_function_pseudocode: 获取函数的伪代码
• get_function_variables: 获取函数的变量和参数
• get_cross_references_to_function: 获取对函数的交叉引用

客户端库用法

Lattice 客户端库提供了一个 Python 接口，用于与 BinjaLattice 服务器进行交互：

from lib.lattice import Lattice

# 初始化客户端
client = Lattice(host='localhost', port=9000, use_ssl=False)

# 使用 API 密钥进行身份验证
client.authenticate("username", "API_KEY")

# 示例：获取二进制文件信息
binary_info = client.get_binary_info()

# 示例：更新函数名称
client.update_function_name("old_name", "new_name")

# 示例：向函数添加注释
client.add_comment_to_function("function_name", "This function handles authentication")

命令行界面

该项目包括 lattice_client.py，它提供了一个交互式命令行界面，用于测试和调试 BinjaLattice 服务器：

python lattice_client.py --host localhost --port 9000 [--ssl] --username user --password YOUR_API_KEY

命令行选项

• --host: 服务器主机 (默认: localhost)
• --port: 服务器端口 (默认: 9000)
• --ssl: 启用 SSL/TLS 加密
• --interactive, -i: 以交互模式运行
• --username: 用于身份验证的用户名
• --password: 用于身份验证的密码/API 密钥
• --token: 身份验证令牌 (如果你有之前身份验证的令牌)

交互模式

交互模式提供了一个菜单驱动的界面，具有以下选项：

1. 获取二进制文件信息
2. 按地址获取函数上下文
3. 按名称获取函数上下文
4. 更新函数名称
5. 更新变量名称
6. 向函数添加注释
7. 向地址添加注释
8. 重新连接到服务器
9. 获取所有函数名称
10. 获取函数反汇编代码
11. 获取函数伪代码
12. 获取函数变量
13. 获取对函数的交叉引用
14. 退出

使用交互模式的示例：

python lattice_client.py -i --ssl --username user --password YOUR_API_KEY

非交互式命令

你还可以使用客户端执行单个命令：

# 获取二进制文件信息
python lattice_client.py --username user --password YOUR_API_KEY --get-binary-info

# 获取函数反汇编代码
python lattice_client.py --username user --password YOUR_API_KEY --get-function-disassembly "main"

# 向函数添加注释
python lattice_client.py --username user --password YOUR_API_KEY --add-comment-to-function "main" "Entry point of the program"

安全注意事项

• API 密钥在服务器启动时随机生成，并在 Binary Ninja 日志中显示
• 令牌默认在 8 小时后过期
• SSL/TLS 需要用户提供证书和密钥（默认禁用）
• 所有请求都需要通过 API 密钥或令牌进行身份验证
• 服务器默认在本地端口 9000 上运行

开发

• 主要服务器实现在 plugin/lattice_server_plugin.py 中
• MCP 服务器实现在 mcp_server.py 中
• 客户端库在 lib/lattice.py 中

添加新功能

要添加新功能：

1. 在 lattice_server_plugin.py 中的 LatticeRequestHandler 类中添加新的端点处理程序
2. 在 lib/lattice.py 中的 Lattice 类中添加相应的客户端方法
3. 在 mcp_server.py 中添加新的 MCP 工具

许可证

MIT 许可证