GzdAgent

简介

GzdAgent 是一个面向 CTF（Capture The Flag）安全竞赛场景的智能辅助分析平台。用户通过自然语言与 AI Agent 交互，Agent 自动调用工具、检索知识库、参考历史 Write-up 解题报告，为参赛者提供解题思路和攻击路径建议。

核心能力：AI 对话 + MCP 工具调用 + 知识库检索 + CTF WP 经验匹配。

功能特性

Agent 对话

• 多轮对话，SSE 流式输出，Markdown 渲染
• 单 Agent ReAct 模式 + 多 Agent 编排（Deep / Plan-Execute / Supervisor）
• 支持 OpenAI 兼容接口和 Claude 原生 API
• 推理模式（thinking/effort）可配置

CTF 解题辅助

• 题目类型自动识别（Web / PWN / Reverse / Crypto / Misc）
• 调用 search_ctf_wp 工具检索历史 WP 知识库
• 基于关键词匹配 + 类别过滤，返回相似题目的解题思路和 Payload 参考
• WP 知识库按类别目录组织，Markdown 格式，支持 CLI 管理

工具系统

• 55+ 内置 MCP 工具：漏洞管理、资产扫描、WebShell 管理、C2 框架、知识库检索、项目事实图等
• 支持外部 MCP 服务器接入，扩展工具能力
• 工具执行监控：超时控制、取消、结果截断、并发限制

知识库

• RAG 向量检索：文档上传、嵌入、多查询检索、重排序
• WP 倒排索引：CJK + 英文分词，关键词精确匹配
• 检索历史审计

项目管理

• 项目创建、事实黑板（Fact Graph）
• 攻击链可视化（节点 + 边 DAG）
• 漏洞记录与资产关联

企业功能

• RBAC 权限控制（用户 → 角色 → 权限，三级资源范围）
• 操作审计日志
• 可视化工作流编排
• 批量任务调度
• 8 平台聊天机器人集成（微信、企业微信、钉钉、飞书、Telegram、Slack、Discord、QQ）

快速开始

环境要求

• Go 1.25+
• Linux 64-bit（推荐）或 macOS
• 可访问的 LLM API（OpenAI 兼容或 Claude）

编译

git clone https://github.com/Toyar-mo/GzdAgent.git
cd GzdAgent
go build -o gzd-agent ./cmd/server/

配置

cp config.example.yaml config.yaml

编辑 config.yaml，至少配置模型 API：

ai:
  default_channel: "deepseek"
  channels:
    deepseek:
      provider: "openai"
      base_url: "https://api.deepseek.com/v1"
      api_key: "${DEEPSEEK_API_KEY}"
      model: "deepseek-chat"

启动

./run.sh

浏览器访问 http://localhost:8080，默认管理员账号 admin，初始密码在启动日志中打印。

CTF 竞赛专用配置启动

针对 CTF 比赛场景，项目提供了专用的 config-ctf.yaml，启用更激进的竞速策略（WP 命中 → 直接用 Payload → 拿到 Flag → 结束，3 轮无结果立刻换策略）：

# 1. 编译
go build -o gzd-agent ./cmd/server/

# 2. 用 CTF 专用配置启动
./gzd-agent --config config-ctf.yaml

# 3. 在对话页面选择 "CTF 竞赛" 角色

架构概览

┌─────────────────────────────────────────┐
│              客户端层                     │
│  Web SPA (Vanilla JS)  │  机器人 (8平台)  │
└─────────────────────────────────────────┘
                    │ HTTP/SSE
┌─────────────────────────────────────────┐
│            HTTP 服务层 (Gin)              │
│   认证 + RBAC + 限流 + 审计               │
└─────────────────────────────────────────┘
                    │
┌─────────────────────────────────────────┐
│             应用服务层                     │
│  ┌─────────────────────────────────┐    │
│  │ Agent 引擎 (Eino ReAct)         │    │
│  │  单Agent │ Deep │ Plan │ Sup    │    │
│  ├─────────────────────────────────┤    │
│  │ MCP 工具系统 (55+ 内置工具)      │    │
│  ├─────────────────────────────────┤    │
│  │ 知识库 (RAG + WP 倒排索引)       │    │
│  └─────────────────────────────────┘    │
└─────────────────────────────────────────┘
                    │
┌─────────────────────────────────────────┐
│              数据层 (SQLite)              │
│   30+ 表: 对话/消息/工具执行/项目/审计...   │
└─────────────────────────────────────────┘

技术栈

项目结构

GzdAgent/
├── cmd/server/          # Web 服务入口
├── cmd/ctfcli/          # CTF WP CLI 工具
├── internal/
│   ├── agent/           # Agent 引擎
│   ├── app/             # 应用启动
│   ├── config/          # 配置管理
│   ├── ctfknowledge/    # CTF WP 知识库
│   ├── database/        # 数据库层
│   ├── handler/         # HTTP API
│   ├── knowledge/       # RAG 知识库
│   ├── mcp/             # MCP 协议
│   ├── multiagent/      # 多 Agent 编排
│   ├── openai/          # LLM 客户端
│   ├── project/         # 项目事实图
│   ├── robot/           # 机器人集成
│   ├── security/        # 安全工具/鉴权
│   ├── workflow/        # 工作流引擎
│   └── ...
├── web/                 # 前端 SPA
├── wp/                  # CTF WP 知识库
├── skills/              # Agent Skills
├── agents/              # 多 Agent 定义
├── roles/               # 角色配置
└── config.example.yaml  # 配置模板

配置

主要配置项：

API Key 支持环境变量：api_key: "${DEEPSEEK_API_KEY}"

添加 CTF WP

在 wp/ 目录下按类别创建 Markdown 文件：

wp/
├── web/sql_injection/basic_union_sqli.md
├── pwn/rop/ret2libc.md
└── reverse/crackme.md

WP 格式参考 wp/README.md。重启服务后自动索引。

CLI 管理：

go run ./cmd/ctfcli/ index
go run ./cmd/ctfcli/ search "sql注入"
go run ./cmd/ctfcli/ status

安全说明

• 仅可对自有系统或已获得明确授权的目标使用
• 在共享或生产环境启用高风险工具前，请审阅安全配置
• API Key 通过环境变量注入，不写入配置文件
• 默认生成随机管理员密码，启动时打印

贡献者

贡献者列表持续更新中，欢迎提交 Pull Request 加入！

License

Apache License 2.0