使用Django编写简易测试报告生成器

2018-04-07 401741人围观 ,发现 24 个不明物体 工具

*本文原创作者:海带头,本文属FreeBuf原创奖励计划,未经许可禁止转载

概述

安全测试人员在工作的时候,不单单需要发现问题,还需要花费精力去撰写安全测试报告来记录结果。为了方便,我利用django框架写了个简易的安全测试报告撰写系统。

image.png

撰写报告的时候,需要写明漏洞名称、漏洞风险等很多项目内容。对于某一类型漏洞而言,其实有很多项在大多数情况是固定不变的(比如漏洞名称、修复建议)。我们可以在编写漏洞的时候直接调用漏洞模板,再添加和修改需要记录的数据,最终一键导出我们设计好的模板。

配置和安装环境

Django-1.10.8

python2.7

Wndows/Mac

安装

python –m pip install –r requirements.txt

说明

Mac下安装mysql以及MySQL-python

brew install mysql

pip install MySQL-python

如果第二条安装出现”Failed building wheel for mysql-python”错误,则采用以下方法

LDFLAGS=-L/usr/local/opt/openssl/lib pip install mysql-python

启动服务

建库

本地mysql需要新建配置文件中的数据库settings.py文件:

DATABASES ={
 'default':{ 'ENGINE':'django.db.backends.mysql', 'HOST':'127.0.0.1', 'PORT':'3306', 'NAME':'bugwrite', # 数据库名 'USER':'root', 'PASSWORD':'root', 'OPTIONS':{ 'init_command':"SET sql_mode='STRICT_TRANS_TABLES'" }, }
}

建表

cd myweb01

python manage.py makemigrations

python manage.py migrate

启服务

python manage.py runserver

功能描述

启动本地服务后,访问http://localhost:8000/blog/work可以在本地新建一个项目,如下图所示:

image.png

点开项目详情,可以在其中创建漏洞,并且可以引用不同的漏洞模板:

image.png

点击新建XSS漏洞后,XSS漏洞名称、风险分析、风险等级、修复建议会自动填充,只需要填写记录测试过程即可:

image.png

回到之前创建的项目,可以导出报告:

image.pngimage.png

实现了漏洞管理

扩展和说明

漏洞模板创建

models.py文件中创建了BugModel类作为漏洞模板,并添加到admin中,在django的管理后台可以添加模板,访问http://localhost:8000/admin ,编辑BugModel即可:

image.png

然后在views.py中编辑form表单SelectTestFrom,添加模板选择项:

image.png

Word模板扩展

漏洞详情的使用ckeditor获取用户输入,提交的数据类型为html格式,我这边自己写了一个解析html数据的类Html2Word,处理了p/ul/ol/strong/em/img便签,为防止数据丢失,建议编辑器只用加粗、斜体、图片、序标的格式:html2word.py

image.png

下面是将漏洞字段填充到word中,如果有需要丰富生成的word模板,可以修改下面的内容:Object2word.py

image.png

附件

源码已经上传至github:https://github.com/Gin-Cy/bugeditor.git

*本文原创作者:海带头,本文属FreeBuf原创奖励计划,未经许可禁止转载

更多精彩
相关推荐
发表评论

已有 24 条评论

取消
Loading...

特别推荐

推荐关注

活动预告

填写个人信息

姓名
电话
邮箱
公司
行业
职位
css.php