渗透测试系统penework设计及实现

整体重构以前写的penework，使用flask+vue2前后端分离的方式开发，提高使用效率

涉及技术

前端：

vue2 
vue-router
vuex
element-ui 
webpack3

后端

python
flask
masscan
nscan

前端设计

利用vue2 开发前端，element-ui做前端展示组件,vuex做数据存储交换

登录页面

项目展示

后端设计

后端采用mongodb

新建项目时，开启一个masscan扫描，在masscan扫描结束后，利用Nscan得到banner和title

后端api返回数据格式

{
    'code': 2000/3000/4000/5000/
    'message': xxx,
    'data':
}

其中

2000 =>   请求成功
4000 =>   已知错误
5000 =>   {
            5001: token过期
            5002: token非法
          }
6000 =>   未知错误

token 设计

token = jwt.encode({'time': login_time}, token_key, algorithm)

利用一个装饰器进行token检查

def auth_token(func):

    def wrapper(*args, **kwargs):
        token = ''
        if request.headers['token']:
            token  =  request.headers['token']
            decode_token = jwt.decode(token, Config.TOKEN_KEY, algorithm='HS256')
            if (int(time.time()) - decode_token['time']) > Config.EXPIRE_TIME
                return jsonify({'code':5001, 'message': 'Token Expire Time!'})
        return func(*args, **kwargs)

    return wrapper

需要利用任务队列来管理新建项目的创建，当有masscan时，只是录入项目的信息，并更新项目的状态为未开始。但没有项目正在扫描时，开启masscan扫描.为了缓解服务器的压力，每次只是运行固定个数的项目。

任务队列，利用celery，broker使用redis

队列

scanning
unscan
scaned

数据库设计

插件及相关扩展功能

会拥有的功能

masscan和nscan扫描，目前已经完成
漏洞库扫描，集合各种exp/shellcode，目前exp/shellcode收集中
子目录/子域名目前基本已经完成
爬虫彻底重构，还没有开始

开发进度

前端 40%左右

后端 30%左右

参考

vuex理解

鸣谢

特别感谢巡风，pocsuite等开源项目的激励和部分代码

Best Flask open-source libraries and packages

Penework