Commit 7ed7497
committed
update safedog.py
update some feature1 parent af60f11 commit 7ed7497
1 file changed
Lines changed: 3 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -7,6 +7,7 @@ | |||
| 7 | 7 | ||
| 8 | 8 | import re | |
| 9 | 9 | ||
| 10 | + from lib.core.enums import HTTP_HEADER | ||
| 10 | 11 | from lib.core.settings import WAF_ATTACK_VECTORS | |
| 11 | 12 | ||
| 12 | 13 | __product__ = "Safedog Web Application Firewall (Safedog)" | |
@@ -17,6 +18,8 @@ def detect(get_page): | |||
| 17 | 18 | for vector in WAF_ATTACK_VECTORS: | |
| 18 | 19 | page, headers, code = get_page(get=vector) | |
| 19 | 20 | retval = re.search(r"WAF/2.0", headers.get("X-Powered-By", ""), re.I) is not None | |
| 21 | + retval |= re.search(r"Safedog", headers.get(HTTP_HEADER.SERVER, ""), re.I) is not None | ||
| 22 | + retval |= re.search(r"safedog", headers.get("Set-Cookie", ""), re.I) is not None | ||
| 20 | 23 | if retval: | |
| 21 | 24 | break | |
| 22 | 25 | ||
0 commit comments