首页 /安全 / 渗透测试

渗透测试报告自动生成工具

2021-12-06|

公众号：白帽子左一
领取配套练手靶场、安全全套课程及工具...

背景

在安服仔的日子里，发现其他人输出的渗透测试报告结果不规范，主要在报告质量、内容、字体、及修复方案中存在诸多问题，而且大部分安服仔需要对每次的项目结果进行统计整理，方便后续跟踪复测。

因此研发了Savior—渗透测试报告辅助生成系统，起这个名字也是为了拯救大多数渗透测试工程师，

告别繁琐的渗透测试报告编写过程及漏洞统计过程。

系统框架

前端：Ant Design Pro

后端：Django REST Framework

数据库：Mysql

主要功能

用户管理：主要是方便统计漏洞的发现者，后续可能大概也许会添加漏洞统计模块，根据提交数据、漏洞类型、时间等进行统计报表，当前用户管理模块仅允许通过Django后台进行修改，前端只负责展示，主要是我太懒了。

项目管理：根据项目的不同可上传项目的专属渗透测报告模板，并可以根据需要进行模板自定义模板（/Demo/demo.docx）；

模板自定义：不用修改源代码，仅需修改word即可进行模板自定义；

整改建议管理：此平台主要就是为了体现标准化输出，因此可通过内置漏洞描述及修复建议进行快速输出，并支持自定义修改（/Demo/常规WEB渗透测试漏洞描述及修复方法.docx）；

一键生成：通过提交报告模块，内联项目模板，快速生成渗透测试报告，真正达到了一键生成，并确保报告内数据准确、字体统一、格式标准；

自动邮件：在生成报告后可通过用户管理配置的自动邮件发送功能进行邮件通知，可自定义邮件模板，这样再报告给客户的时候就可以直接转发了（暖男功能）；

漏洞统计：每次渗透过后，需要挨个查找报告进行统计整理，现在只要提交报告后，后台会自动联动；

漏洞报表一键导出

漏洞跟踪：增加了漏洞状态字段，创建报告后，漏洞状态默认为新增，漏洞管理模块可进行复测，包括已整改、未整改两种状态；

安装指南

首先将代码clone到本地：

git clone https://github.com/Mustard404/Savior.git

Docker部署

我们推荐使用Docker进行部署，相对于源码部署更为简单和快速。

部署前请务必先安装Docker及docker-compose。

修改配置文件
首先复制根目录的.env.docker并重命名为.env，修改其中的Email Settings和initial Administrator配置。

这两个配置分别控制邮件提醒，以及初始管理帐号密码及邮箱。

同时需要注意以下两点：

1. 务必把邮箱修改为自己邮箱，不然可能会出现非预期错误！

2. 如果使用阿里云、腾讯云服务器，请使用smtp的ssl协议，两家云厂商默认封禁了25端口。

一键启动

docker-compose up -d

访问 http://127.0.0.1:8000即可看到页面。

修改启动端口

如果想修改启动端口，可以修改docker-compose.yaml文件中web容器的ports。

默认为8000:8000，比如要修改为8080端口可改为8080:8000。

源码部署

所需环境：

python 3.8+(测试过3.6也没问题，其他版本自行测试)

yarn 1.22+

Pandoc

Mysql 5.7

前端环境

cd app  
yarn && yarn start:no-mock

环境变量设置创建字符集为utf-8编码的数据库。

复制**.env.docker为.env**，并配置数据库、邮箱、管理员等信息。

务必把邮箱修改为自己邮箱，不然可能会出现非预期错误！

如果使用阿里云、腾讯云服务器，请使用smtp的ssl协议，两家云厂商默认封禁了25端口。

后端环境

#开启env环境
python3 -m venv env
source env/bin/activate
#安装依赖
python -m pip install -r requirements.txt -i http://pypi.doubanio.com/simple --trusted-host pypi.doubanio.com 
#同步数据库
python manage.py makemigrations api
python manage.py migrate
python manage.py init_admin
#启动后端
python manage.py runserver 0.0.0.0:8000

源码部署环境：

前台页面：http://127.0.0.1:8001

Django管理后台：
http://127.0.0.1:8000/api/admin/

使用手册

初始化说明

其中Savior平台包含两个后台页面。考虑到安全性，目前用户管理、项目管理托管于Django管理后台（主要是这两个模块不会写），其余功能均可通过前台页面实现。

前台页面：http://127.0.0.1:8000

Django管理后台：
http://127.0.0.1:8000/api/admin/

用户管理

访问Django管理后台：
http://127.0.0.1:8000/api/admin/,

请完善API>用户的Name、Avatar、Autosentmail三个字段，分别控制报告的作者、头像（图片Url）、生成报告后自动发送渗透测试报告到邮箱。

项目管理

访问Django管理后台：
http://127.0.0.1:8000/api/admin/

请通过API>Projects进行添加项目，可根据不通项目选择不通的渗透测试报告模板。

参数说明：Project logo（项目Logo）、Project center（项目名称）、Project description（项目描述）、Project template（渗透测试报告模板，目前标准模板可使用Demo/demo.docx，如需自定义模板，请参考模版自定义部分）

整改设置

访问http://127.0.0.1:8000可进入Savior平台，通过个人设置>整改设置>添加漏洞模板可进行设置漏洞类型、漏洞描述、修复建议从而达到标准化。

目前整理了一些通用的修复建议模板，请参考Demo/常规WEB渗透测试漏洞描述及修复方法.docx。

模板自定义

目前根据我经常使用的渗透测试报告模板生成了一个demo版本（请参考/Demo/demo.docx）。

当然您也可以根据自己的需求进行模板自定义，其中仅需在WORD模板中进行参数替换，目前Savior中具体参数如下：

{{report_no}} - 漏洞编号，通过时间戳自动生成，确保漏洞编号的唯一性

{{report_center}} - 测试项目，为项目管理中项目名称

{{report_systemname}} - 系统名称

{{report_start_time}} - 测试开始时间

{{report_end_time}} - 测试结束时间

{{report_author}} - 测试提交人，对应用户管理的Name参数

{{report_test_url}} - 测试Url

{{vuls|length}} - 漏洞个数

{{vuls|vul_statistics(3)}} - 高危漏洞个数

{{vuls|vul_statistics(2)}} - 中危漏洞个数

{{vuls|vul_statistics(1)}} - 低危漏洞个数

以下漏洞详情请利用{%tr for vuls in vuls %}{%tr endfor %}进行循环遍历。

如想列出所有漏洞URL,则使用参数{%tr for vuls in vuls %}{{item.vul_url}}{%tr endfor %}

{{item.vul_url}} - 漏洞Url

{{item.vul_recurrence}} - 漏洞复现

{{item.vul_level}} - 漏洞危险等级

{{item.vul_describe}} - 漏洞描述

{{item.vul_modify_repair}} - 修复建议

注：Savior平台渗透测试模板遵循Jinja2语法，更多内容请参考
https://jinja.palletsprojects.com/en/3.0.x/