概述
本篇文章主要讲一下BBScan 是如何生成html 报告的
Template
其实这块内容说起来也还挺简单的,主要的知识就是使用的是string 中的 Template,使用的时候需要从string 中进行导入
from string import Template
使用方法示例
from string import Template
value = """you name is ${name}"""
name = "join"
name = {"name": name}
temp = Template(value) //得到一个Template 对象
result = temp.substitute(name) //使用字典的格式,一一对应的对内容进行填充
print(result)
----------------------------------------------------------------
from string import Template
value = """you name is ${name}"""
temp = Template(value)
result = temp.substitute(name="Bob") //直接使用值进行填充
print(result)
Template的使用是通过预先的定义字符串或者某些格式内容,然后通过substitute()函数进行填充,来得到动态内容的方法。
扫描器中报告生成功能实现的分析
首先就是使用字符串生成了报告的大体格式以及样式,均是这种带有变量的字符串,后续进行填充
这两块就是把对应的字符串通过字典组装成一个完成的模板了,通过代码和传输的参数进行选择
生成报告函数内代码分析
is_markdown = args.md //是否选择了markdown 格式的报告模板
no_browser = args.no_browser
start_time = time.time()
a_template = template['markdown'] if is_markdown else template['html']// 判断报告模板,若是没有选择,则默认是html
t_general = Template(a_template['general']) //根据报告的各个部分,创建多个Template对象。
t_host = Template(a_template['host'])
t_list_item = Template(a_template['list_item'])
output_file_suffix = a_template['suffix']
report_name = '%s_%s%s' % (os.path.basename(_file).lower().replace('.txt', ''),
time.strftime('%Y%m%d_%H%M%S', time.localtime()),
output_file_suffix)
下面的代码其实也就没啥意思了,但是我感觉最有意思的部分应该是下面这一部分,这到不是说有多难,但是给从来没有写过扫描器的我们一个提示,可以通过变量(开关)这种方式来控制,扫描的进行和是否完成的标志。