sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-13 10:00:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix for an Issue #101

Browse Source
This commit is contained in:
Miroslav Stampar 2012-07-17 00:19:33 +02:00
parent c96e44b30c
commit 81d15e5051
2 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib/core/common.py
View File

@ -3050,12 +3050,15 @@ def getHostHeader(url):
Returns proper Host header value for a given target URL Returns proper Host header value for a given target URL
""" """
retVal = urlparse.urlparse(url).netloc retVal = url
if re.search("http(s)?://\[.+\]", url, re.I): if url:
retVal = extractRegexResult("http(s)?://\[(?P<result>.+)\]", url) retVal = urlparse.urlparse(url).netloc
elif any(retVal.endswith(':%d' % _) for _ in (80, 443)):
retVal = retVal.split(':')[0] if re.search("http(s)?://\[.+\]", url, re.I):
retVal = extractRegexResult("http(s)?://\[(?P<result>.+)\]", url)
elif any(retVal.endswith(':%d' % _) for _ in (80, 443)):
retVal = retVal.split(':')[0]
return retVal return retVal

11
lib/request/redirecthandler.py
View File

@ -75,11 +75,12 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
content = None content = None
redurl = self._get_header_redirect(headers) redurl = self._get_header_redirect(headers)
if not urlparse.urlsplit(redurl).netloc: if redurl:
redurl = urlparse.urljoin(req.get_full_url(), redurl) if not urlparse.urlsplit(redurl).netloc:
redurl = urlparse.urljoin(req.get_full_url(), redurl)
self._infinite_loop_check(req) self._infinite_loop_check(req)
self._ask_redirect_choice(code, redurl) self._ask_redirect_choice(code, redurl)
try: try:
content = fp.read() content = fp.read()
@ -88,7 +89,7 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
dbgMsg += "redirect response content (%s)" % msg dbgMsg += "redirect response content (%s)" % msg
logger.debug(dbgMsg) logger.debug(dbgMsg)
if kb.redirectChoice == REDIRECTION.YES: if redurl and kb.redirectChoice == REDIRECTION.YES:
req.headers[HTTPHEADER.HOST] = getHostHeader(redurl) req.headers[HTTPHEADER.HOST] = getHostHeader(redurl)
result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers) result = urllib2.HTTPRedirectHandler.http_error_302(self, req, fp, code, msg, headers)
else: else: