#Target: POST请求/api/v1/targets
try:
    #data包含目标URL和类型，auth_headers包含API_KEY
       response = requests.post(targets_api, auth_headers, data, False)
result = response.json()
target_id = result.get('target_id')
       return target_id
except Exception:
return None
#Scan: POST请求/api/v1/scans
try:
    response = requests.post(scan_api, data, auth_headers, False)
    status_code = 200
except Exception:
status_code = 404
return status_code

@csrf_exempt
def vuln_scan(request):
    #通过POST请求获取用户输入的URL和扫描类型
    url = request.POST.get('ip')
    scan_type = request.POST.get('scan_type')
    t = Target(API_URL, API_KEY)
    #将目标URL添加到扫描队列中
    target_id = t.add(url)
    #如果target_id不为None,则开始扫描
    if target_id is not None:
        s = Scan(API_URL, API_KEY)
        status_code = s.add(target_id, scan_type)
        if status_code == 200:
            return success()
    return error()

function get_scan_info(ip , scan_type) {
#使用POST请求发送用户输入
        $.post('/vuln_scan', {
            ip: ip ,
            scan_type: scan_type
        }, function (data) {
            if (data.code !== 200) {
                ......
            } else {
                ...... 
            }
           ......});
    }
var domain = $('input[name=scan_url]').val();
#使用循环判断用户选择的扫描类型
for(var i=0; i<document.getElementsByName("scan_type").length; i++) {
    if (document.getElementsByName("scan_type")[i].checked) {
      var scan_type=document.getElementsByName("scan_type")[i].value;
    }
}
if(domain){
    get_scan_info(domain,scan_type)
}else{
    ......
}

#获取所有扫描目标
response=requests.get(scan_api, self.auth_headers, False) 
scan_response=response.json().get('scans')
for scan in scan_response:
   scan['request_url'] = request_url
   scan_list.append(scan)
return scan_list
#搜索状态为“open“,对应target_id的扫描目标
vuln_search_api=f'{vuln_api}?q=status:{status};target_id:{target_id}'
try:
#使用get方式请求
    response = requests.get(vuln_search_api, auth_headers, False)
#返回搜索结果目标的所有漏洞信息
    return response.text
except Exception:
    return None

path('vulnscan', views.vulnscan, name="vulnscan"),
for target_id in target_ids:
      #使用正则匹配获取第二个参数：taget_id
urlpatterns.append(url(r'^vuln_result/(?P<target_id>.*)$', views.vuln_result, name='vuln_result/'+target_id))

@login_required
def vuln_result(request, target_id):
    d = Vuln(API_URL, API_KEY)
    data = []
    vuln_details = json.loads(d.search(None,None, "open", target_id=str(target_id)))
id = 1
for target in vuln_details['vulnerabilities']:
    item={
        'id': id,
        'severity': target['severity'],
        'target': target['affects_url'],
        'vuln_id':target['vuln_id'],
        'vuln_name': target['vt_name'],
        'time': re.sub(r'T|\..*$', " ", target['last_seen'])
    }
    id += 1
    data.append(item)
return render(request,&#39;vuln-reslut.html&#39;,{&#39;data&#39;: data})</code></pre></div></div><p>在这个子功能中，前端的数据展示使用的是 Bootstrap Table。这个模板有很多实用的功能，比如表格的搜索功能、分页展示功能等等，增加了用户体验。表格的数据在 HTML 中使用双花括号来接收，在 <strong>views.py</strong> 函数中返回的到相应的 HTML 页面时，将 data 字典一起返回。这样的返回方式可以将使用字典中的 key 值获取对应的 values 值。还可以是使用 if-else、for 等语句来分类展示数据。核心代码如下：</p><div class="rno-markdown-code"><div class="rno-markdown-code-toolbar"><div class="rno-markdown-code-toolbar-info"><div class="rno-markdown-code-toolbar-item is-type"><span class="is-m-hidden">代码语言：</span>javascript</div></div><div class="rno-markdown-code-toolbar-opt"><div class="rno-markdown-code-toolbar-copy"><i class="icon-copy"></i><span class="is-m-hidden">复制</span></div></div></div><div class="developer-code-block"><pre class="prism-token token line-numbers language-javascript"><code class="language-javascript" style="margin-left:0">{% for item in data %}

……………
这个只展示了扫描目标列，其他列类似
{{ item.target }}

……………

{% endfor %}

#获取指定漏洞的相关信息
def get(self, vuln_id):
    vuln_get_api = f'{self.vuln_api}/{vuln_id}'
    try:
        #使用GET请求将vuln_id传给API，结果以json格式返回
        response = requests.get(vuln_get_api, auth_headers, False)
        return response.json()
    except Exception:
        return None

for vuln_id in vuln_ids:
    urlpatterns.append(url(r'^vuln_detail/(?P<vuln_id>.*)$', views.vuln_detail, name='vuln_detail/' + vuln_id))

@login_required
def vuln_detail(request,vuln_id):
    d = Vuln(API_URL,API_KEY)
    data = d.get(vuln_id)
    print(data)
    parameter_list = BeautifulSoup(data['details'], features="html.parser").findAll('span')
    request_list = BeautifulSoup(data['details'], features="html.parser").findAll('li')
    data_dict = {
        'affects_url': data['affects_url'],
        'last_seen': re.sub(r'T|\..*$', " ", data['last_seen']),
        'vt_name': data['vt_name'],
        'details': data['details'].replace("  ",'').replace('</p>',''),
        'request': data['request'],
        'recommendation': data['recommendation'].replace('<br/>','\n')
    }
    try:
        data_dict['parameter_name'] = parameter_list[0].contents[0]
        data_dict['parameter_data'] = parameter_list[1].contents[0]
    except:
        pass
    num = 1
    try:
        Str = ''
        for i in range(len(request_list)):
            Str += str(request_list[i].contents[0])+str(request_list[i].contents[1]).replace('<strong>', '').replace('</strong>', '')+'\n'
            num += 1
    except:
        pass
    data_dict['Tests_performed'] = Str
    data_dict['num'] = num
    data_dict['details'] = data_dict['details'].replace('class="bb-dark"','style="color: #ff0000"')
    return render(request, "vuln-detail.html", {'data': data_dict})

#使用POST请求提交用户的输入
function get_Middleware_scan(ip , CVE_id) {
    $.post('/Middleware_scan', {
    ip: ip , #目标URL
    CVE_id: CVE_id  #选择的CVE编号
}, function (data) {
#处理返回结果
      ………
      ………}); 
}

$.ajax({
#使用POST请求发送目标URL和CVE编号，设置超时为1秒
    type: "POST",
    url: '/start_Middleware_scan',
    timeout: 10000,
    data: {
    ip: ip,
    CVE_id: CVE_id
    }
});

class Middleware_vuln(models.Model):
#类名为数据库表名，变量名为字段名，字段属性定义如下
    id = models.AutoField(primary_key=True)
    url = models.CharField(max_length=100, null=True)
    status = models.CharField(max_length=20, null=True)
    result = models.CharField(max_length=100, null=True)
    CVE_id = models.CharField(max_length=100, null=True)
    time = models.CharField(max_length=100, null=True, unique=True)

def insert_Middleware_data(url, CVE_id, Time, result=None, status="runing"):
    try:
        Middleware_vuln.objects.create(url=url, status=status, result=result, CVE_id=CVE_id, time=Time)
        print("insert success")
        return True
    except:
        print("data insert error")
        return False
def update_Middleware_data(url, CVE_id, Time, result):

try:

Middleware_vuln.objects.filter(url=url, status='runing', CVE_id=CVE_id, time=Time).update(status="completed", result=result)

print("update success")

except:

print("data updata error")

Time = 0.0

@csrf_exempt

@login_required

def Middleware_scan(request):

#使用POST请求获取用户输入，并将其插入数据库中。

#Time作为全局变量插入到数据库中，作为查询目标信息的条件。

global Time

try:

url= request.POST.get('ip')

CVE_id = request.POST.get('CVE_id').replace('-',"_")

Time = time.time()  # time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(t))时间戳转日期格式

if insert_Middleware_data(url, CVE_id, Time):

return success()

except:

return error()

@csrf_exempt

@login_required

def start_Middleware_scan(request):

try:

url = request.POST.get('ip')

ip, port = urlparse(url).netloc.split(':')

CVE_id = request.POST.get('CVE_id').replace('-', "_")

time.sleep(5) #等待数据插入成功后在查询出来扫描

msg = Middleware_vuln.objects.filter(url=url, status='runing', CVE_id=CVE_id, time=Time)

print(msg)

#扫描条目可能不止一条，需要使用循环来扫描

for target in msg:

result = POC_Check(target.url, target.CVE_id)

#将扫描结果和状态更新

update_Middleware_data(target.url, target.CVE_id, Time, result)

return success()

except:

return error()

def socket_scan(self, hosts):
    '''端口扫描核心代码'''
    global PROBE
    socket.setdefaulttimeout(1)
    ip, port = hosts.split(':')
    try:
        if len(self.port) < 25:
            # 创建套接字
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            # TCP/IP三次握手建立连接
            result = sock.connect_ex((ip, int(port)))   
# 调用socket.connect_ex((ip, port))，端口开放返回0，否则返回错误代码
# 实现和nmap的全连接扫描类似的功能。
            if result == 0:                     # 成功建立TCP链接
                self.port.append(port)          # 结果集中增加端口
                for i in PROBE:                 # 通过HTTP1.1刺探
                    sock.sendall(i.encode())    # 发送完整的TCP数据包
                    response = sock.recv(256)   # 接受最大256byte
                    sock.close()
                    if response:
                        break
                if response:
                    for pattern in SIGNS:
                        pattern = pattern.split(b'|')
                        if re.search(pattern[-1],response, re.IGNORECASE): 
# 正则匹配banner信息与字典中的服务
                            proto = '{}:{}'.format(pattern[1].decode(), port)
                            self.out.append(proto)  # 添加至输出结果
                            break
        else:
            self.num = 1
    except (socket.timeout, ConnectionResetError): # 异常处理
        pass
    except:
        pass

THREADNUM = 64  # 线程数
def run(self, ip):  #多线程扫描
    hosts = []
    global PORTS, THREADNUM
    for i in PORTS:
        hosts.append('{}:{}'.format(ip, i))
    try:
        with concurrent.futures.ThreadPoolExecutor(
                max_workers=THREADNUM) as executor:
            executor.map(self.socket_scan, hosts)
    except EOFError:
        pass

class PortList(models.Model):
    '''端口查询表'''
    num=models.BigIntegerField(verbose_name='端口号')
    service=models.TextField(max_length=100,verbose_name='服务')
    protocol=models.CharField(max_length=20,verbose_name='
协议',blank=True,default='未知')
    status=models.CharField(max_length=10,verbose_name='
状态',blank=True,default='未知')
    class Meta:  
    # 后台表头设置
        verbose_name=verbose_name_plural='端口列表'

@admin.register(PortList)
class PortListAdmin(ImportExportModelAdmin):
    # 设置哪些字段显示在后台
    list_display = ('num', 'service', 'protocol', 'status',)
    # 设置num字段进入编辑界面 
    list_display_links = ('num',)  
    search_fields = ('num', 'service',)
    # 过滤器，按字段进行筛选
    list_filter = ('protocol','status')
    # 设置num为默认排序字段
    ordering = ('num', )
    list_per_page = 15  #设置每页显示数据条数

class Item(models.Model):
    '''导航条目'''
    title = models.CharField(max_length=50,verbose_name='名称')
    desc = models.TextField(max_length=100,verbose_name='描述')
    url = models.URLField(verbose_name='网址',blank=True)
    img = models.URLField(default='https://jwt1399.top/favicon.png',verbose_name='logo')
    img_width = models.IntegerField(default=45, verbose_name='图片宽度')
    category = models.ForeignKey(Category, blank=True, null=True, verbose_name='分类', on_delete=models.CASCADE)
    class Meta:
        verbose_name=verbose_name_plural='导航条目'
    #后台条目图片预览
    def img_admin(self):
        return format_html( '<img src="{}" width="50px" height="50px" style="border-radius: 50%;" />',self.img,)
    img_admin.short_description = 'logo预览'
    def __str__(self):
        return self.title

class Category(models.Model):
    """条目分类"""
    name = models.CharField(max_length=20, verbose_name='名称')
    sort = models.IntegerField(default=1, verbose_name='显示顺序')
    add_menu = models.BooleanField(default=True, verbose_name='添加到导航栏')
    icon = models.CharField(max_length=30, default='fas fa-home',verbose_name='图标')
    class Meta:
        verbose_name_plural=verbose_name = '分类'
#统计分类对应条目数,并放入后台
def get_items(self):
    return len(self.item_set.all())
get_items.short_description = '条目数'  # 设置后台显示表头
#后台图标预览
def icon_data(self):#引入Font Awesome Free 5.11.1
    return format_html('<h1><i class="{}"></i></h1>',self.icon) #转化为<i class="{self.icon}"></i>
icon_data.short_description = '图标预览'
def __str__(self):
    return self.name</code></pre></div></div><h3 id="37q50" name="%E6%96%87%E6%A1%A3%E9%A1%B5">文档页</h3><p>将 Docsify 直接嵌入了 Django 中构造了文档页。</p><figure class=""><div class="rno-markdown-img-url" style="text-align:center"><div class="rno-markdown-img-url-inner" style="width:100%"><div style="width:100%"><img src="https://cdn.static.attains.cn/app/developer-bbs/upload/1723181217708115217.png" /></div><div class="figure-desc">文档页</div></div></div></figure><h3 id="anhmf" name="%E5%85%BC%E5%AE%B9%E6%80%A7">兼容性</h3><h4 id="69fcp" name="Phone%E7%AB%AF">Phone端</h4><figure class=""><div class="rno-markdown-img-url" style="text-align:center"><div class="rno-markdown-img-url-inner" style="width:100%"><div style="width:100%"><img src="https://cdn.static.attains.cn/app/developer-bbs/upload/1723181218180281538.png" /></div></div></div></figure><h4 id="avcl2" name="Pad%E7%AB%AF">Pad端</h4><figure class=""><div class="rno-markdown-img-url" style="text-align:center"><div class="rno-markdown-img-url-inner" style="width:100%"><div style="width:100%"><img src="https://cdn.static.attains.cn/app/developer-bbs/upload/1723181218661002317.png" /></div></div></div></figure><h3 id="e9imu" name="%E7%89%88%E6%9C%AC%E5%8F%98%E6%9B%B4%E8%AE%B0%E5%BD%95">版本变更记录</h3><h4 id="3lagd" name="v2.7%EF%BC%882021-04-18%EF%BC%89">v2.7（2021-04-18）</h4><ul class="ul-level-0"><li>新增<code>域名探测</code>功能；</li><li>新增<code>中间件漏洞</code>扫描；</li><li>修复<code>忘记密码</code>功能；</li><li>优化AWVS未启动报错信息；</li><li>优化<code>用户登录</code>逻辑；</li><li>优化<code>漏扫详情</code>页UI；</li><li>优化导航栏布局；</li><li>优化若干小细节；</li></ul><h4 id="8pcsv" name="v2.6%EF%BC%882021-03-31%EF%BC%89">v2.6（2021-03-31）</h4><ul class="ul-level-0"><li>新增漏洞<code>扫描详情</code>功能；</li><li>新增首页 <code>仪表盘</code>；</li><li>安全导航页导航栏<code>移动端优化</code>；</li><li>安全导航页目录栏<code>缩放优化</code>；</li><li>注册&amp;登录界面优化；</li><li>文档页<code>导航栏优化</code>；</li><li>新增 <code>UI</code> 夜间模式；</li><li>修复若干<code>UI</code> 显示Bug；</li></ul><h4 id="aaflg" name="v2.5%EF%BC%882021-03-02%EF%BC%89">v2.5（2021-03-02）</h4><ul class="ul-level-0"><li>新增了<code>漏洞扫描</code>功能；</li><li>端口扫描页新增<code>常见端口查询表</code>；</li><li>信息泄露页新增<code>常见信息泄露列表</code>；</li><li>指纹识别页新增<code>数据分析</code>图表；</li><li>漏洞扫描页<code>界面优化</code>；</li></ul><h4 id="fq99m" name="v2.4%EF%BC%882021-02-22%EF%BC%89">v2.4（2021-02-22）</h4><ul class="ul-level-0"><li>新增了<code>目录识别</code>功能；</li><li>重写<code>欢迎页</code>；</li><li>安全导航页<code>移动端界面</code>适配；</li><li>安全导航页<code>UI优化</code>；</li><li>目录识别页<code>界面优化</code>；</li><li>指纹识别页新增<code>常见指纹</code>显示与搜索；</li><li>引入<code>Boostrap Table</code>实现分页；</li><li>淘汰 <code>LayUI</code> 改用 <code>Layer</code> 进行弹窗；</li><li>文档页增加<code>导航栏</code>；</li></ul><h4 id="36s8s" name="v2.3%EF%BC%882021-02-08%EF%BC%89">v2.3（2021-02-08）</h4><ul class="ul-level-0"><li>全新的<code>页面布局</code>；</li><li>UI适配<code>移动端</code>；</li><li>优化<code>导航页布局</code>；</li><li>优化一系列<code>UI</code>显示问题；<ul class="ul-level-1"><li>优化了<code>手机端页脚</code>显示</li><li>优化了<code>平板端导航条</code>显示</li><li>页面底部增加<code>回到顶部</code>按钮</li><li>按钮<code>触发跳转</code>页面相对位置</li><li><code>回车键</code>触发查询按钮</li><li>优化<code>导航页页脚</code>显示</li></ul></li></ul><h4 id="5sded" name="v2.2-%EF%BC%882021-02-03%EF%BC%89">v2.2 （2021-02-03）</h4><ul class="ul-level-0"><li>新增了<code>信息泄露探测</code>功能；</li><li>新增了<code>旁站探测</code>功能；</li><li>新增了导航页<code>数据分析</code>功能；</li><li>新增了基于<code>Docsify</code>的<code>文档页</code>；</li><li>重构了<code>静态文件static</code>文件结构</li><li>优化了项目<code>文件结构</code>；</li><li>美化了<code>注册</code>页面；</li><li>引入了<code>动态粒子</code>背景效果；</li><li>修复了一些 <code>UI</code> 显示问题；</li></ul><h4 id="6tu9n" name="v2.1-%EF%BC%882021-01-13%EF%BC%89">v2.1 （2021-01-13）</h4><ul class="ul-level-0"><li>新增了<code>指纹识别</code>功能；</li><li>新增了<code>登录和注册功能</code>功能；</li><li>新增了<code>欢迎页</code>；</li><li><del>新增了<code>文档页</code>；</del></li><li>修复了一些 <code>UI</code> 显示问题；</li></ul><h4 id="653dg" name="v2.0%EF%BC%882021-01-04%EF%BC%89">v2.0（2021-01-04）</h4><ul class="ul-level-0"><li>新增了<code>端口扫描</code>功能；</li><li>新增了<code>安全导航</code>功能；</li><li>连入了 <code>SQLite</code> 数据库，后续考虑改为<code>MySQL</code>；</li><li>修复了一些 <code>UI</code> 显示问题；</li><li>修复了后台头部小图标无法显示问题；</li><li>新增了后台数据导入导出功能；</li></ul><h4 id="6tbe9" name="v1.0%EF%BC%882020-12-20%EF%BC%89">v1.0（2020-12-20）</h4><ul class="ul-level-0"><li>基于<code>Tabler</code>框架构造了前端页面；</li><li>采用基于<code>Python</code>的 Django 框架编写后端；</li><li>引入了<code>SimpleUi</code>美化 Django 后台框架；</li><li>引入了<code>Font-Awsome 5.15.1</code>图标；</li></ul><h3 id="enor" name="%E9%A1%B9%E7%9B%AE%E9%83%A8%E7%BD%B2">项目部署</h3><h4 id="2ric0" name="%E6%9C%AC%E5%9C%B0%E9%83%A8%E7%BD%B2">本地部署</h4><p>1.使用 Pycharm 打开本项目，在 Pycharm 的 <code>setting</code>—&gt;<code>Project Interpreter</code>中 add 一个新的虚拟环境。</p><p>2.在该虚拟环下打开 Pycharm 自带 Terminal 运行 <code>pip install -r requirements.txt</code> 下载项目所需的第三方包。</p><p>3.现在项目已经可以正常运行了，但是想要使用漏洞扫描功能，需要安装AWVS，并在项目的<code>setting.py</code> 中配置 AWVS的 <code>API URL</code>和<code>API KEY</code></p><p>4.忘记密码功能相关配置在项目的<code>setting.py</code> 中修改</p><div class="rno-markdown-code"><div class="rno-markdown-code-toolbar"><div class="rno-markdown-code-toolbar-info"><div class="rno-markdown-code-toolbar-item is-type"><span class="is-m-hidden">代码语言：</span>javascript</div></div><div class="rno-markdown-code-toolbar-opt"><div class="rno-markdown-code-toolbar-copy"><i class="icon-copy"></i><span class="is-m-hidden">复制</span></div></div></div><div class="developer-code-block"><pre class="prism-token token line-numbers language-javascript"><code class="language-javascript" style="margin-left:0">EMAIL_HOST = &#39;smtp.163.com&#39;

EMAIL_PORT = 25               # 发件箱的smtp服务器端口

EMAIL_HOST_USER = 'xxx'       # 你的邮箱账号

EMAIL_HOST_PASSWORD ="xxx"    # 邮箱授权码

EMAIL_USE_TLS = True          # 这里必须是 True，否则发送不成功

EMAIL_FROM = 'xxx'            # 你的邮箱账号

DEFAULT_FROM_EMAIL = 'xxx'    # 你的邮箱账号