PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-01-02 04:49:03 +00:00
Code Issues Projects Releases Wiki Activity

Fixes #3514

Browse Source
This commit is contained in:
Miroslav Stampar
2019-03-04 16:36:19 +01:00
parent b1ef5d520c
commit 59d97968a7
5 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
extra/wafdetectify/wafdetectify.py
View File

@@ -73,7 +73,7 @@ def main():
print(colorize("%s #v%s\n by: %s\n" % (NAME, VERSION, AUTHOR)))
if len(sys.argv) < 2:
exit(colorize("[x] usage: python %s <hostname>" % os.path.split(__file__)[-1]))
sys.exit(colorize("[x] usage: python %s <hostname>" % os.path.split(__file__)[-1]))
cookie_jar = cookielib.CookieJar()
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookie_jar))
@@ -96,11 +96,11 @@ def main():
del sys.modules[filename[:-3]]
module = __import__(filename[:-3].encode(sys.getfilesystemencoding() or "utf8"))
except ImportError as ex:
exit(colorize("[x] cannot import WAF script '%s' (%s)" % (filename[:-3], ex)))
sys.exit(colorize("[x] cannot import WAF script '%s' (%s)" % (filename[:-3], ex)))
_ = dict(inspect.getmembers(module))
if "detect" not in _:
exit(colorize("[x] missing function 'detect(get_page)' in WAF script '%s'" % found))
sys.exit(colorize("[x] missing function 'detect(get_page)' in WAF script '%s'" % found))
else:
WAF_FUNCTIONS.append((_["detect"], _.get("__product__", filename[:-3])))
@@ -113,7 +113,7 @@ def main():
socket.getaddrinfo(hostname, None)
except socket.gaierror:
print(colorize("[x] host '%s' does not exist" % hostname))
exit(1)
sys.exit(1)
found = False
for function, product in WAF_FUNCTIONS:
@@ -121,14 +121,14 @@ def main():
continue
if function(get_page):
exit(colorize("[!] WAF/IPS identified as '%s'" % product))
sys.exit(colorize("[!] WAF/IPS identified as '%s'" % product))
if not found:
print(colorize("[o] nothing found"))
print()
exit(int(not found))
sys.exit(int(not found))
if __name__ == "__main__":
main()