MCP Server

🚀 scanmalware - mcp

這是一個輕量級的Python MCP服務器，它封裝了公共的ScanMalware.com API。

🚀 快速開始

本地運行（可流式傳輸的HTTP）

python -m venv .venv
source .venv/bin/activate
pip install -U pip
pip install .

export MCP_TRANSPORT=streamable-http
export MCP_HOST=127.0.0.1
export MCP_PORT=8000

scanmalware-mcp

使用Docker運行

docker build -t scanmalware-mcp .
docker run --rm -p 127.0.0.1:8000:8000 \
  -e MCP_TRANSPORT=streamable-http \
  -e MCP_HOST=0.0.0.0 \
  -e MCP_PORT=8000 \
  scanmalware-mcp

可選配置

• 可設置MCP_AUTH_TOKEN，以便HTTP傳輸需要Authorization: Bearer <MCP_AUTH_TOKEN>。
• 可選的認證環境變量（僅認證受限端點需要）：
  • SCANMALWARE_BEARER_TOKEN
• 其他環境變量：
  • SCANMALWARE_BASE_URL（默認值：https://scanmalware.com）
  • SCANMALWARE_ALLOW_HTTP（默認值：false）
  • SCANMALWARE_TIMEOUT_S（默認值：30）
  • SCANMALWARE_MAX_DOWNLOAD_BYTES（默認值：10485760）
  • SCANMALWARE_ALLOW_PRIVATE_TARGETS（默認值：false）
  • SCANMALWARE_CA_CERT（可選；SSL驗證的CA證書包路徑）
• MCP服務器安全環境變量：
  • MCP_AUTH_TOKEN（如果設置，HTTP傳輸需要Authorization: Bearer <token>）
  • MCP_RESOURCE_SERVER_URL / MCP_ISSUER_URL（可選；僅在設置MCP_AUTH_TOKEN時使用）

工具說明

• submit_scan 不會調用 /api/v1/csrf - token；沒有CSRF令牌工具。
• 一些上游端點被禁用，不包含在工具列表中（例如，get_improvements、find_screenshot_duplicates、get_ai_stats、search_js_fingerprinter2_code_hash、search_js_segments_by_tlsh）。
• 一些搜索工具至少需要一個過濾器，如果未提供則會引發驗證錯誤。

💻 使用示例

釣魚郵件分類（提交 → 等待 → 總結）

提交對https://example - login - update.com的掃描，等待完成，然後返回狀態、風險評分和主要指標。如果風險高，包括AI分析和截圖資源。

品牌濫用監控

搜索包含 "acme login" 的掃描結果（限制5條）。對於每個結果，列出掃描ID、狀態、風險評分和URL。突出顯示標記為高風險的結果。

TLS/證書檢查

對於掃描ID 1234...abcd，獲取TLS詳細信息和證書PEM下載。總結頒發者、主題、有效期和SANs；標記不匹配項。

📦 安裝指南

部署到DigitalOcean（Debian + Docker + Nginx）

部署包位於deploy/目錄，運行兩個容器：

• mcp（此服務器，在8000端口上使用可流式傳輸的HTTP）
• nginx（前端在80端口；將/mcp代理到MCP服務器）

前提條件

• doctl 已認證（doctl auth init）
• SSH密鑰已上傳到DigitalOcean（供doctl compute droplet create使用）

在德國（法蘭克福）創建一個小的Droplet

DROPLET_NAME=scanmalware-mcp-small
REGION=fra1
SIZE=s-1vcpu-2gb
IMAGE=debian-12-x64
SSH_KEYS=$(doctl compute ssh-key list --format ID --no-header | paste -sd, -)

doctl compute droplet create "$DROPLET_NAME" \
  --region "$REGION" \
  --size "$SIZE" \
  --image "$IMAGE" \
  --ssh-keys "$SSH_KEYS" \
  --tag-name scanmalware-mcp \
  --wait

防火牆（公共HTTP/HTTPS + SSH）

doctl compute firewall create \
  --name scanmalware-mcp-fw \
  --inbound-rules "protocol:tcp,ports:22,address:0.0.0.0/0,address:::0/0" \
  --inbound-rules "protocol:tcp,ports:80,address:0.0.0.0/0,address:::0/0" \
  --inbound-rules "protocol:tcp,ports:443,address:0.0.0.0/0,address:::0/0" \
  --outbound-rules "protocol:icmp,ports:0,address:0.0.0.0/0,address:::0/0" \
  --outbound-rules "protocol:tcp,ports:0,address:0.0.0.0/0,address:::0/0" \
  --outbound-rules "protocol:udp,ports:0,address:0.0.0.0/0,address:::0/0" \
  --droplet-ids <droplet-id>

在Droplet上安裝Docker + Compose

ssh -i /path/to/key root@<droplet-ip> \
  "apt-get update -y && apt-get install -y docker.io docker-compose"

上傳並運行

tar --exclude=.git --exclude=.venv --exclude=__pycache__ -czf /tmp/scanmalware-mcp.tar.gz -C . .
scp -i /path/to/key /tmp/scanmalware-mcp.tar.gz root@<droplet-ip>:/tmp/
ssh -i /path/to/key root@<droplet-ip> \
  "mkdir -p /opt/scanmalware-mcp && tar -xzf /tmp/scanmalware-mcp.tar.gz -C /opt/scanmalware-mcp"
ssh -i /path/to/key root@<droplet-ip> \
  "cd /opt/scanmalware-mcp && docker-compose -f deploy/docker-compose.yml up -d --build"

驗證

curl -I https://mcp.scanmalware.com/
curl -I https://mcp.scanmalware.com/mcp

/ 應該從Nginx返回200狀態碼。/mcp 在沒有MCP Accept頭的GET請求下返回406，這是預期的。

冒煙測試（MCP初始化 + 工具列表）

import json
import httpx

URL = "http://<droplet-ip>/mcp"
HEADERS = {
    "accept": "application/json, text/event-stream",
    "content-type": "application/json",
}

init_payload = {
    "jsonrpc": "2.0",
    "id": 1,
    "method": "initialize",
    "params": {
        "protocolVersion": "2025-06-18",
        "capabilities": {},
        "clientInfo": {"name": "mcp-smoke-test", "version": "0.1.0"},
    },
}

with httpx.Client(timeout=10) as client:
    init_resp = client.post(URL, headers=HEADERS, json=init_payload)
    init_resp.raise_for_status()
    session_id = init_resp.headers.get("mcp-session-id")

    def extract_sse_data(text: str) -> dict:
        for line in text.splitlines():
            if line.startswith("data: "):
                return json.loads(line[len("data: "):])
        raise ValueError("No SSE data line found")

    init_message = extract_sse_data(init_resp.text)
    protocol_version = init_message["result"]["protocolVersion"]

    # 發送初始化通知
    client.post(
        URL,
        headers={
            **HEADERS,
            "mcp-session-id": session_id,
            "mcp-protocol-version": protocol_version,
        },
        json={"jsonrpc": "2.0", "method": "notifications/initialized"},
    )

    tools_resp = client.post(
        URL,
        headers={
            **HEADERS,
            "mcp-session-id": session_id,
            "mcp-protocol-version": protocol_version,
        },
        json={"jsonrpc": "2.0", "id": 2, "method": "tools/list"},
    )
    tools_resp.raise_for_status()
    tools_message = extract_sse_data(tools_resp.text)
    tool_names = [tool["name"] for tool in tools_message["result"]["tools"]]

print("protocol_version:", protocol_version)
print("tool_count:", len(tool_names))
print("tools:", ", ".join(tool_names))

重新部署 / 新部署

1. 原地更新（同一Droplet）

tar --exclude=.git --exclude=.venv --exclude=__pycache__ -czf /tmp/scanmalware-mcp.tar.gz -C . .
scp -i /path/to/key /tmp/scanmalware-mcp.tar.gz root@<droplet-ip>:/tmp/
ssh -i /path/to/key root@<droplet-ip> \
  "bash /opt/scanmalware-mcp/deploy/redeploy.sh /tmp/scanmalware-mcp.tar.gz"

重新部署腳本在替換文件之前會停止容器，以避免掛載inode問題。如果腳本還不在Droplet上，先運行一次舊的tar + docker - compose命令來安裝它。

可選的一次性輔助腳本（從倉庫根目錄運行）

./deploy/push-redeploy.sh root@<droplet-ip> /path/to/key

2. 滾動部署（新Droplet）

• 創建一個新的Droplet（上述步驟）
• 部署相同的包
• 將DNS切換到新IP
• 準備好後銷燬舊的Droplet

doctl compute droplet delete <old-droplet-id> --force

📚 詳細文檔

有關部署、TLS、日誌記錄以及如何連接到DigitalOcean Droplet的詳細信息，請參閱 docs/OPERATIONS.md。