update safedog.py

getcode2git · getcode2git · commit 7ed7497fdaf8 · 2015-12-19T15:39:14.000+08:00
update some feature
diff --git a/waf/safedog.py b/waf/safedog.py
@@ -7,6 +7,7 @@
 
 import re
 
+from lib.core.enums import HTTP_HEADER
 from lib.core.settings import WAF_ATTACK_VECTORS
 
 __product__ = "Safedog Web Application Firewall (Safedog)"
@@ -17,6 +18,8 @@ def detect(get_page):
     for vector in WAF_ATTACK_VECTORS:
         page, headers, code = get_page(get=vector)
         retval = re.search(r"WAF/2.0", headers.get("X-Powered-By", ""), re.I) is not None
+        retval |= re.search(r"Safedog", headers.get(HTTP_HEADER.SERVER, ""), re.I) is not None
+        retval |= re.search(r"safedog", headers.get("Set-Cookie", ""), re.I) is not None
         if retval:
             break
 
