update safedog.py

update some feature
This commit is contained in:
getcode2git 2015-12-19 15:39:14 +08:00
parent af60f11319
commit 7ed7497fda

View File

@ -7,6 +7,7 @@ See the file 'doc/COPYING' for copying permission
import re import re
from lib.core.enums import HTTP_HEADER
from lib.core.settings import WAF_ATTACK_VECTORS from lib.core.settings import WAF_ATTACK_VECTORS
__product__ = "Safedog Web Application Firewall (Safedog)" __product__ = "Safedog Web Application Firewall (Safedog)"
@ -17,6 +18,8 @@ def detect(get_page):
for vector in WAF_ATTACK_VECTORS: for vector in WAF_ATTACK_VECTORS:
page, headers, code = get_page(get=vector) 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"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: if retval:
break break