再次修改banner获取

shmilylty · Mar 1, 2020 · 6f3cdb6 · 6f3cdb6
1 parent 71e5c89
commit 6f3cdb6
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/oneforall/common/utils.py b/oneforall/common/utils.py
@@ -377,8 +377,15 @@ def get_filtered_data(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
+    temp_list = []
+    server = headers.get('Server')
+    if server:
+        temp_list.append(server)
+    via = headers.get('Via')
+    if via:
+        temp_list.append(via)
+    power = headers.get('X-Powered-By')
+    if power:
+        temp_list.append(power)
+    banner = ','.join(temp_list)
     return banner