使用说明
MCP映射器和开发树桩生成器
mcp_map.py 是一种攻击性安全工具，旨在枚举、映射和交互 模型上下文协议（MCP） 服务器。
它连接到MCP端点（通常通过SSE），枚举整个攻击面（工具、资源、模板），主动查询可用数据，并生成人类可读的审计图和即用型Python客户端存根，用于快速利用或测试。
🚀 特性
-
深度枚举：
-
资源： 列出所有公开的数据端点和 积极阅读 它们（带有安全截断）用于查找敏感数据或配置。
-
资源模板： 标识数据访问的动态URI模式。
-
工具： 转储工具名称、描述和参数的完整JSON模式（对于查找RCE或Logic缺陷至关重要）。
-
-
主动侦察：
-
自动尝试读取已发现的资源。
-
优雅地处理二进制数据和大型文本文件，以避免控制台泛滥。
-
-
双输出格式：
-
Markdown(
.md): 专门为 黑曜石 或Notion pentest报告（语法干净，代码块清晰，没有HTML混乱）。 -
JSON(
.json): 将管道的原始数据转储到其他自动化工具中。
-
-
漏洞生成(
--generate-stub):-
自动创建独立
client.py根据目标量身定制的脚本。 -
用发现的模式和虚拟值预填充函数调用，允许 即时互动/利用 无需手动编码。
-
📋 先决条件
-
Python 3.10+
-
fastmcp（该工具依赖于此库实现稳定的SSE传输）
Bash
pip install fastmcp
🛠️ 用法
基本映射（显示到控制台）
连接到目标并将Markdown摘要打印到stdout。
Bash
python mcp_map.py http://localhost:8000/sse
审计报告生成
将完整地图保存到黑曜石友好的Markdown文件中。
Bash
python mcp_map.py http://target-ip:53060/mcp -o audit_report.md
全面侦察和利用树桩生成
执行映射、保存映射并生成Python脚本(exploit.py)预先配置以攻击目标上发现的特定工具。
Bash
python mcp_map.py http://target-ip:53060/mcp -o map.md -g exploit.py
用于自动化的JSON输出
Bash
python mcp_map.py http://target-ip:53060/mcp -f json -o raw_data.json
📄 输出示例
1.Markdown地图(-o map.md)
针对直接复制到pentest报告进行了优化。
标记语言
# MCP Server Map: http://10.10.10.5:8000/sse
## 📂 Resources (Queried)
- **App Config**
- URI: `config://main`
- Desc: _Main application settings_
- **Content**:
```json
{
"debug": true,
"db_host": "127.0.0.1"
}
```
## 🛠️ Tools
- **execute_command**
- Desc: _Executes a shell command_
- Inputs: `command`, `timeout`
- Schema:
```json
{
"properties": {
"command": { "type": "string" },
"timeout": { "type": "integer" }
}
}
```
2.客户失误(-g exploit.py)
生成的Python代码已准备好运行。只需取消注释要测试的行即可。
python
import asyncio
from fastmcp import Client
TARGET_URL = 'http://10.10.10.5:8000/sse'
client = Client(TARGET_URL)
async def main():
async with client:
print(f'[*] Connected to {TARGET_URL}')
# --- Tools ---
print('\n[+] Calling Tool: execute_command')
# Description: Executes a shell command
# Arguments Schema: {"command": {"type": "string"}, "timeout": {"type": "integer"}}
# UNCOMMENT TO EXECUTE:
# result = await client.call_tool(name='execute_command', arguments={'command': 'TEST_VALUE', 'timeout': 0})
# print(result)
if __name__ == '__main__':
asyncio.run(main())
⚠️ 法律免责声明
此工具仅用于法律安全评估和授权渗透测试。在扫描或与任何MCP服务器交互之前，请确保您拥有明确的权限。
