river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改banner获取

Browse Source
This commit is contained in:
Jing Ling
2020-03-02 03:09:11 +08:00
parent 6180b7d2ce
commit 71e5c89689
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/request.py
+1 -4
View File
@@ -172,10 +172,7 @@ def request_callback(future, index, datas):
else: else:
datas[index]['valid'] = 1 datas[index]['valid'] = 1
headers = resp.headers headers = resp.headers
banner = str({'Server': headers.get('Server'), datas[index]['banner'] = utils.get_sample_banner(headers)
'Via': headers.get('Via'),
'X-Powered-By': headers.get('X-Powered-By')})
datas[index]['banner'] = banner[1:-1]
datas[index]['header'] = str(dict(headers))[1:-1] datas[index]['header'] = str(dict(headers))[1:-1]
if isinstance(text, str): if isinstance(text, str):
title = get_title(text).strip() title = get_title(text).strip()
oneforall/common/utils.py
+8
View File
@@ -374,3 +374,11 @@ def get_filtered_data(data):
if valid == 0 or valid == 1: if valid == 0 or valid == 1:
filtered_data.append(item) filtered_data.append(item)
return filtered_data return filtered_data
def get_sample_banner(headers):
server = headers.get('Server') or ''
via = headers.get('Via') or ''
power = headers.get('X-Powered-By') or ''
banner = server + via + power
return banner