Red-team-mcp

Red-team-mcp

AI エージェントがレッドチーム演習で使用するための MCP（Model Context Protocol）サーバーです。

概要

このサーバーは、レッドチームのシナリオで AI エージェントが利用できるリソースとツールを提供します。Model Context Protocol を実装し、セキュリティテストと評価に必要な情報の取得、ツールへのアクセス、アクションの実行を AI エージェントに許可します。

アーキテクチャ

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 エージェントが対象ネットワーク上の開いているポートとサービスを検出できます
• スキャンパラメーター（IP 範囲、ポート範囲、スキャン速度）を設定できます
• エージェントが利用できるよう、結果を解析してフォーマットします

2. 脆弱性スキャナー

• 脆弱性検出に Nuclei と連携します
• カスタムテンプレートと重大度フィルタリングに対応しています
• 高速検索のため、結果を MongoDB に保存します

3. SSH ツール

• 認証情報を使用した SSH コマンドの実行
• ユーザー名とパスワードのリストを使用した SSH ブルートフォース機能
• ホストおよびポート情報とともに結果を保存します

4. Metasploit 連携

• エクスプロイトデータベース：すべての Metasploit エクスプロイトをあらかじめ格納した MongoDB コレクション
• 高速検索：プラットフォーム、CVE、ランク、作成者、キーワードでエクスプロイトを検索します
• エクスプロイトの実行：ペイロード設定を指定して、対象に対してエクスプロイトを実行します
• ライブクエリ：Metasploit RPC サーバーからリアルタイムでエクスプロイトを一覧表示します

実装計画

1. 基本的な MCP サーバー構造を設定します
2. パラメーター検証を伴う masscan 連携を実装します
3. 結果の解析と保存の仕組みを作成します
4. 追加コンポーネントを追加します（今後対応予定）
5. ロギングとモニタリングを実装します
6. 権限と境界のシステムを構築します

TODO

• 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')

開発

サーバーに新しいツールやリソースを追加する方法の詳細は、今後追記されます。