diff --git a/oneforall/common/request.py b/oneforall/common/request.py index e7a1c14..809c849 100644 --- a/oneforall/common/request.py +++ b/oneforall/common/request.py @@ -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() diff --git a/oneforall/common/utils.py b/oneforall/common/utils.py index 9388548..db17131 100644 --- a/oneforall/common/utils.py +++ b/oneforall/common/utils.py @@ -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