Red-team-mcp

Red-team-mcp

供 AI Agent 在红队演练中使用的 MCP（Model Context Protocol）服务。

概览

此服务为 AI Agent 提供红队演练场景中所需的资源和工具。它实现了 Model Context Protocol，使 AI Agent 能够检索信息、访问工具并执行安全测试与评估所需的操作。

架构

graph TD
    Agent[AI Agent Client] <--> |MCP Protocol| Server[FastMCP Server]

    subgraph "MCP Server Components"
        Server --> PortScanner[Port Scanner Module]
        Server --> VulnScanner[Vulnerability Scanner Module]
        Server --> SSHTools[SSH Tools Module]
        Server --> Metasploit[Metasploit Integration]
        Server --> DomainDiscovery[Domain Discovery Module]
        Server <--> Database[(MongoDB Database)]
    end

    subgraph "External Tools"
        PortScanner --> Masscan[masscan]
        VulnScanner --> Nuclei[nuclei]
        SSHTools --> SSH[SSH Client]
        Metasploit --> MSFRPC[Metasploit RPC]
        DomainDiscovery --> Subfinder[subfinder]
    end

    Database --> Findings[Findings]

组件

1. 端口扫描器

• 集成 masscan，实现高速端口扫描
• 允许 AI Agent 发现目标网络上的开放端口和服务
• 可配置扫描参数（IP 范围、端口范围、扫描速率）
• 解析结果并整理为适合 Agent 使用的格式

2. 漏洞扫描器

• 集成 Nuclei 进行漏洞检测
• 支持自定义模板和按严重程度筛选
• 将结果存储在 MongoDB 中，以便快速搜索

3. SSH 工具

• 使用凭据进行 SSH 命令执行
• 支持使用用户名/密码列表进行 SSH 暴力破解
• 将结果与主机及端口信息一同存储

4. Metasploit 集成

• 漏洞利用数据库：预先填充了所有 Metasploit 漏洞利用的 MongoDB 集合
• 快速搜索：按平台、CVE、等级、作者或关键词搜索漏洞利用
• 执行漏洞利用：通过配置有效载荷，针对目标执行漏洞利用
• 实时查询：从 Metasploit RPC 服务实时列出漏洞利用

实施计划

1. 搭建基本的 MCP 服务结构
2. 集成 masscan，并验证参数
3. 创建结果解析和存储机制
4. 添加其他组件（即将推出）
5. 实现日志记录和监控
6. 构建权限和边界系统

待办事项

• 添加 nuclei 扫描器
• 添加 SSH 扫描器
• 添加 Metasploit 集成
• 实现完全异步的 FastMCP 服务
• 添加测试
• 编写逐步入侵一台机器的提示词。（参见 agents/simple_agent.py）

设置

安装第三方工具

设置需要以下工具

• mongodb
• nuclei

漏洞利用数据库设置

若要启用快速漏洞利用搜索，请填充 Metasploit 漏洞利用数据库：

1. 先决条件：

   • 已安装 Metasploit Framework
   • MongoDB 正在 localhost:27017 上运行
   • Metasploit RPC 服务正在运行：msfrpcd -P msf -a 127.0.0.1
   • Python 依赖项：pip install pymetasploit3

2. 填充数据库：

   python scripts/setup_exploits_db.py
   

3. 测试设置：

   python scripts/test_exploits_db.py
   

这会创建一个可搜索的数据库，其中包含所有 Metasploit 漏洞利用及其元数据，包括：

• CVE 参考信息
• 平台信息
• 漏洞利用等级和作者
• 兼容的有效载荷
• 必需选项

使用方法

漏洞利用搜索示例

# Search Windows SMB exploits
search_exploits_fast(platform='windows', search_term='smb')

# Find EternalBlue exploit
search_exploits_fast(cve='CVE-2017-0144')

# Get high-quality exploits
search_exploits_fast(rank='excellent')

# Search by author
search_exploits_fast(author='hdm')

开发

有关如何通过新工具和资源扩展服务的详情将另行补充。