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

better way for dealing with relative paths

Browse Source
This commit is contained in:
Miroslav Stampar 2011-05-24 05:26:51 +00:00
parent a536bf210f
commit ad25bcc2be
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/request/connect.py
View File

@ -104,6 +104,9 @@ class Connect:
retrying = kwargs.get('retrying', False) retrying = kwargs.get('retrying', False)
redirecting = kwargs.get('redirecting', False) redirecting = kwargs.get('redirecting', False)
if not urlparse.urlsplit(url).netloc:
url = urlparse.urljoin(conf.url, url)
# flag to know if we are dealing with the same target host # flag to know if we are dealing with the same target host
target = reduce(lambda x, y: x == y, map(lambda x: urlparse.urlparse(x).netloc.split(':')[0], [url, conf.url])) target = reduce(lambda x, y: x == y, map(lambda x: urlparse.urlparse(x).netloc.split(':')[0], [url, conf.url]))

2
lib/request/redirecthandler.py
View File

@ -54,7 +54,7 @@ class SmartRedirectHandler(urllib2.HTTPRedirectHandler):
result.redurl = headers.getheaders("uri")[0].split("?")[0] result.redurl = headers.getheaders("uri")[0].split("?")[0]
if hasattr(result, 'redurl'): if hasattr(result, 'redurl'):
if result.redurl.startswith('.') or result.redurl.startswith('/'): if not urlparse.urlsplit(result.redurl).netloc:
result.redurl = urlparse.urljoin(conf.url, result.redurl) result.redurl = urlparse.urljoin(conf.url, result.redurl)
if "set-cookie" in headers: if "set-cookie" in headers: