Google Threat Intelligence (GTI) MCP 服务器

一个生产级的 Model Context Protocol (MCP) 服务器，提供与 Google Threat Intelligence (GTI) 和 VirusTotal API v3 的全面集成。基于高性能异步 Python、FastAPI 和 MCP Streamable HTTP 传输标准（/mcp）构建，专为在 Google Cloud Run 上部署以及与 Gemini Enterprise 和 AI 安全代理无缝交互而设计。

🌟 主要特性

• Streamable HTTP 传输 (/mcp)：原生支持 MCP Streamable HTTP 协议规范，使用零重定向路由。

• 22+ 威胁情报工具：直接访问 Google Threat Intelligence 集合、威胁行为者、活动、恶意软件家族、报告、文件沙箱分析、IP/域名/URL 遥测以及 IoC 查询。

• 企业级安全：原生集成 Google Cloud Secret Manager（VT_APIKEY / VT_SECRET_NAME），确保无需在源代码中存储任何密钥或 API 密钥。

• Gemini Enterprise 与 Agent 就绪：IAM 保护端点（roles/run.invoker），支持 Google Cloud 身份验证。

• 自动化云部署：使用 Google Cloud Build 和 Cloud Run 的一键构建与部署脚本（deploy.sh）。

Related MCP server: OSINT MCP Server

🛠️ MCP 工具套件

🚀 快速开始

1. 前提条件

• Python 3.10+

• 已配置项目访问权限的 Google Cloud SDK (gcloud)

• 有效的 Google Threat Intelligence / VirusTotal API 密钥

2. 本地设置

克隆仓库并安装依赖：

git clone https://github.com/yayefa/GTI-MCP-Server.git
cd GTI-MCP-Server

python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt

3. 环境配置

复制示例环境文件：

cp .env.example .env

编辑 .env 以配置你的设置：

PROJECT_ID=your-gcp-project-id
REGION=us-central1
SERVICE_NAME=mcp-gti-mcp-server
VT_SECRET_NAME=VT_APIKEY
SECRET_PROJECT_ID=your-gcp-project-id
LOG_LEVEL=INFO

4. 本地运行

uvicorn server:app --host 0.0.0.0 --port 8080 --reload

☁️ 部署到 Google Cloud Run

1. 在 Google Secret Manager 中存储 API 密钥

echo -n "YOUR_GTI_VT_API_KEY" | gcloud secrets create "VT_APIKEY" \
    --data-file=- \
    --project="YOUR_PROJECT_ID" \
    --replication-policy="automatic"

2. 通过脚本部署

执行自动化部署脚本：

chmod +x deploy.sh
./deploy.sh

有关完整的部署详情和 IAM 配置，请参阅 DEPLOYMENT.md。

🧪 测试与验证

针对正在运行的实例或已部署的 Cloud Run 服务运行自动化测试客户端：

AUTH_TOKEN=$(gcloud auth print-identity-token) \
TARGET_URL="https://<YOUR-CLOUD-RUN-URL>" \
python3 test_client.py

或者使用 curl 直接查询 MCP 端点：

curl -X POST https://<YOUR-CLOUD-RUN-URL>/mcp \
  -H "Authorization: Bearer $(gcloud auth print-identity-token)" \
  -H "Content-Type: application/json" \
  -d '{
    "jsonrpc": "2.0",
    "id": 1,
    "method": "tools/call",
    "params": {
      "name": "get_ip_report",
      "arguments": {
        "ip_address": "8.8.8.8"
      }
    }
  }'

📄 许可证

本项目采用 Apache 2.0 许可证 - 详情请参阅 LICENSE 文件。