Skip to content

Latest commit

 

History

History
58 lines (44 loc) · 1.67 KB

README.md

File metadata and controls

58 lines (44 loc) · 1.67 KB

vulhub-chatgpt-assistant

基于ChatGPT的漏洞靶机环境自动化部署助手

介绍

本项目基于ChatGPT,通过聊天机器人自动创建漏洞靶机环境,自动关闭漏洞环境,自动获取漏洞环境的poc文档,自动build基础应用镜像并启动容器。

项目可以独立运行,也可以作为ChatGPT的插件运行。(插件方式需要有chatgpt开发者权限,没测试过)

本来计划写成一个chatgpt插件的,结果chatgpt插件需要申请开发者权限,开发者又不支持中国区,只能用gpt的function calling api实现了一个独立的项目。

功能

  • 通过聊天机器人自动创建漏洞环境
  • 通过聊天机器人自动关闭漏洞环境
  • 通过聊天机器人自动获取漏洞环境的验证文档
  • 通过聊天机器人自动build基础应用镜像,并启动容器

功能截图

web_screenshot

安装

1. 安装依赖

pip install -r requirements.txt

注:需要提前安装好docker和docker-compose,和openai的api key

2. 下载vulhub仓库

cd src/vulhub
git clone git@github.com:vulhub/vulhub.git

3. 配置参数

# 配置文件地址: src/config.py
# 主要是配置openai的api key

4. 启动服务

cd src
uvicorn main:app --reload

5. 访问服务

http://127.0.0.1:8000/

TODO

  • 优化消息排版,支持markdown
  • 返回消息打字机效果
  • 支持漏洞验证文档内容返回

感谢

感谢 wolf 大佬给的灵感,感谢 vulhub 项目提供的漏洞环境。