修改banner获取

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
+1 -4
View File
@@ -172,10 +172,7 @@ def request_callback(future, index, datas):
else:
datas[index]['valid'] = 1
headers = resp.headers
banner = str({'Server': headers.get('Server'),
'Via': headers.get('Via'),
'X-Powered-By': headers.get('X-Powered-By')})
datas[index]['banner'] = banner[1:-1]
datas[index]['banner'] = utils.get_sample_banner(headers)
datas[index]['header'] = str(dict(headers))[1:-1]
if isinstance(text, str):
title = get_title(text).strip()
+8
View File
@@ -374,3 +374,11 @@ def get_filtered_data(data):
if valid == 0 or valid == 1:
filtered_data.append(item)
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