Update for WAF scripts

This commit is contained in:
stamparm
2013-02-26 15:30:11 +01:00
parent 6c38afab35
commit e5835dc74f
10 changed files with 73 additions and 6 deletions

View File

@@ -5,6 +5,9 @@ Copyright (c) 2006-2013 sqlmap developers (http://sqlmap.org/)
See the file 'doc/COPYING' for copying permission
"""
import re
from lib.core.enums import HTTPHEADER
from lib.core.settings import WAF_ATTACK_VECTORS
__product__ = "ModSecurity: Open Source Web Application Firewall (Trustwave)"
@@ -14,8 +17,9 @@ def detect(get_page):
for vector in WAF_ATTACK_VECTORS:
page, headers, code = get_page(get=vector)
if code == 501:
retval = True
retval = code == 501
retval |= re.search(r"Mod_Security|NOYB", headers.get(HTTPHEADER.SERVER, ""), re.I) is not None
if retval:
break
return retval