Skip to main content
Glama

根据用户输入的漏洞信息生成符合pocsuite3标准的POC代码。

项目结构

pocgenius/ ├── __init__.py ├── server.py ├── requirements.txt └── README.md

server.py

requirements.txt

mcp>=1.0.0 requests>=2.28.0 pocsuite3>=2.0.0

README.md

PocGenius - POC生成专家

POCGeneratorMCP 是一个基于 Model Context Protocol (MCP) 的服务,专门用于生成符合 pocsuite3 标准的 Proof of Concept (POC) 代码。它能够根据用户提供的漏洞信息自动生成安全、可靠且符合最佳实践的漏洞验证代码。

功能特性

🛡️ 安全性优先

  • 无害性设计: 所有生成的Payload都经过安全过滤,避免对目标系统造成破坏

  • 安全审查: 自动检测和移除可能有害的指令和操作

  • 环境恢复: 确保测试后能够将环境恢复到初始状态

🎯 智能生成

  • 上下文感知: 根据漏洞类型自动调整POC结构和检测逻辑

  • 随机化策略: 关键变量自动随机化,避免重复性问题和误报

  • 通用性设计: 生成的POC适用于多种目标环境

📊 专业标准

  • pocsuite3兼容: 完全符合pocsuite3框架标准

  • 模块化设计: 代码结构清晰,易于理解和扩展

  • 严谨验证: 多重特征校验,降低误报和漏报率

安装和使用

前置要求

  • Python 3.8+

  • pocsuite3 框架

安装步骤

  1. 克隆项目:

git clone <repository-url> cd pocgenius
  1. 安装依赖:

pip install -r requirements.txt
  1. 配置MCP客户端(以Claude为例): 在Claude的MCP配置中添加:

{ "mcpServers": { "pocgenius": { "command": "python", "args": ["/path/to/pocgenius/server.py"] } } }

使用方法

基本用法

通过MCP客户端调用PocGenius服务:

# 示例漏洞信息 vuln_info = { "title": "Example Vulnerability", "severity": "high", "description": "这是一个示例漏洞描述", "type": "rce", "request": { "method": "POST", "url": "/api/v1/test", "headers": { "Content-Type": "application/json" }, "data": "{\"command\": \"echo test\"}" }, "response": { "status_code": 200, "patterns": ["test", "success"] } } # 生成POC poc_code = generate_poc(json.dumps(vuln_info))

输入格式

漏洞信息需要以JSON格式提供,包含以下字段:

字段

类型

必填

描述

title

string

漏洞标题

severity

string

漏洞等级 (low/medium/high/critical)

description

string

漏洞详细描述

type

string

漏洞类型 (rce/sqli/xss/等)

request

object

请求信息

response

object

响应信息

生成示例

PocGenius 生成的POC代码包含:

  1. 安全验证逻辑: 多重特征匹配确保准确性

  2. 随机化参数: 避免固定值导致的重复性问题

  3. 错误处理: 完善的异常处理机制

  4. 结果解析: 清晰的输出格式

安全特性

  • 🔒 自动过滤危险命令 (rm, mkfs, dd等)

  • 🛡️ 防止文件覆盖和系统修改

  • 🔄 测试环境自动恢复

  • 📝 完整的日志记录

支持的类型

  • RCE (远程代码执行)

  • SQL注入

  • XSS (跨站脚本)

  • 文件包含

  • 文件上传

  • 目录遍历

  • 以及其他常见Web漏洞

贡献指南

欢迎提交Issue和Pull Request来改进PocGenius!

许可证

MIT License

免责声明

本工具仅用于安全研究和授权测试,请勿用于非法用途。使用者需对自身行为负责。


PocGenius - 让POC生成更智能、更安全! 🚀

-
security - not tested
A
license - permissive license
-
quality - not tested

Latest Blog Posts

MCP directory API

We provide all the information about MCP servers via our MCP API.

curl -X GET 'https://glama.ai/api/mcp/v1/servers/rockmelodies/POCGeneratorMCP'

If you have feedback or need assistance with the MCP directory API, please join our Discord server