Irtoolshed MCP Server

🚀 安全響應工具箱 MCP 服務器

安全響應工具箱 MCP 服務器為安全專業人士量身打造，提供了全面的模型上下文協議（MCP）服務。它集成了豐富的事件響應和網絡分析工具，支持諸如 Claude 之類的 AI 代理開展各類網絡查詢與分析工作，極大地助力了安全調查的高效進行。

🚀 快速開始

安全響應工具箱 MCP 服務器提供了通過模型上下文協議訪問的一系列網絡和安全工具。它旨在作為通用目的服務，供網絡安全事件響應人員使用，使其能夠執行基本查找操作，包括：

✓ ASN 查找 ✓ DNS 記錄查找（A, AAAA, MX 等） ✓ WHOIS 記錄檢索 ✓ IP 地理位置服務

✨ 主要特性

1. ASN 查找

• 功能：查詢給定 IP 的自治系統編號 (ASN) 和相關詳細信息。
• 輸入：有效的 IPv4 或 IPv6 地址。
• 輸出：包含 ASN、組織名稱、註冊商和地理位置的 JSON 響應。

2. DNS 記錄查找

• 功能：查找指定域名的 DNS 記錄（A, AAAA, MX 等）。
• 輸入：正確的域名。
• 輸出：包含所有相關 DNS 記錄類型的詳細信息，如 A、AAAA、MX 等。

3. WHOIS 記錄檢索

• 功能：檢索指定域名的 WHOIS 註冊信息。
• 輸入：正確的域名。
• 輸出：包含註冊商、registrant、創建日期、到期日期等詳細信息的 JSON 響應。

4. IP 地理位置服務

• 功能：根據 IP 地址獲取地理位置信息。
• 輸入：有效的 IPv4 或 IPv6 地址。
• 輸出：包含國家、城市、郵政編碼、經緯度和時區等詳細信息的 JSON 響應。

🔧 技術細節

錯誤處理

每個工具遵循一致的錯誤處理模式：

通用錯誤響應格式：

{
    "狀態": "錯誤",
    "錯誤": "詳細錯誤消息",
    "查詢": "原始查詢值"
}

工具特定錯誤示例：

• ASN 查找：

{
    "ip_addr": "<查詢的IP>",
    "status": "error",
    "error": "無效的 IP 地址格式"
}

• DNS 查找：

{
    "domain": "<查詢的域名>",
    "record_type": "<請求的類型>",
    "status": "error",
    "error": "DNS 解決失敗"
}

• WHOIS 查找：

{
    "domain": "<查詢的域名>",
    "status": "error",
    "error": "WHOIS 服務器不可用"
}

• 地理位置：

{
    "ip_addr": "<查詢的IP>",
    "status": "error",
    "error": "MaxMind 數據庫未找到或許可證密鑰無效"
}

項目結構

該項目遵循標準的 Python 包結構：

irtoolshed_mcp_server/     # 主包目錄
├── __init__.py           # 包初始化
├── asnlookup.py         # ASN 查找功能
├── dnslookup.py         # DNS 查找功能
├── geolookup.py         # 地理位置功能
├── mcp_server.py        # 主 MCP 服務器實現
└── whoislookup.py       # WHOIS 查找功能

tests/                    # 測試目錄
├── test_asnlookup.py    # ASN 查找測試
├── test_dnslookup.py    # DNS 查找測試
├── test_geolookup.py    # 地理位置測試
└── test_whoislookup.py  # WHOIS 查找測試

📦 安裝指南

設置開發環境

1. 克隆此倉庫：

git clone <repository-url>
cd ir-toolshed-mcp-server

2. 創建虛擬環境並安裝依賴項：

uv venv
source .venv/bin/activate  # 在 Windows 上使用: .venv\Scripts\activate
uv pip install -e ".[dev]"

運行測試

要運行測試套件：

uv run pytest

這將：

• 運行 tests/ 目錄中的所有測試
• 顯示測試覆蓋率信息
• 顯示任何失敗的詳細輸出

⚠️ 重要提示

某些測試需要額外配置：

• 地理位置測試需要 MaxMind GeoLite2 數據庫和許可證密鑰
• WHOIS 測試可能會失敗，如果 WHOIS 服務不可用

代碼質量

該項目使用多種工具來維護代碼質量：

• 使用 Black 格式化代碼：

uv run black .

• 使用 isort 排序導入：

uv run isort .

• 使用 mypy 運行類型檢查：

uv run mypy .

• 使用 ruff 運行 linting：

uv run ruff .

📚 詳細文檔

未來計劃

已完成： ✓ ASN 查找 ✓ DNS 記錄查找（A, AAAA, MX 等）