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

fix for Bug #202

Browse Source
This commit is contained in:
Miroslav Stampar 2010-10-18 08:54:08 +00:00
parent 683184cc8f
commit aaebb4336e
3 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/core/option.py
View File

@ -1031,6 +1031,7 @@ def __setKnowledgeBaseAttributes():
kb.absFilePaths = set() kb.absFilePaths = set()
kb.assumeEmpty = False kb.assumeEmpty = False
kb.authHeader = None
kb.bannerFp = advancedDict() kb.bannerFp = advancedDict()
kb.cache = advancedDict() kb.cache = advancedDict()

1
lib/core/target.py
View File

@ -286,6 +286,7 @@ def initTargetEnv():
conf.parameters = {} conf.parameters = {}
conf.sessionFile = None conf.sessionFile = None
kb.authHeader = None
kb.dbms = None kb.dbms = None
kb.dbmsDetected = False kb.dbmsDetected = False
kb.dbmsVersion = [ "Unknown" ] kb.dbmsVersion = [ "Unknown" ]

10
lib/request/connect.py
View File

@ -121,6 +121,9 @@ class Connect:
headers["Referer"] = "%s://%s" % (conf.scheme, conf.hostname) headers["Referer"] = "%s://%s" % (conf.scheme, conf.hostname)
if kb.authHeader:
headers["Authorization"] = kb.authHeader
if auxHeaders: if auxHeaders:
for key, item in auxHeaders.items(): for key, item in auxHeaders.items():
headers[key] = item headers[key] = item
@ -140,10 +143,10 @@ class Connect:
for _, cookie in enumerate(conf.cj): for _, cookie in enumerate(conf.cj):
if not cookieStr: if not cookieStr:
cookieStr = "Cookie: " cookieStr = "Cookie: "
cookie = getUnicode(cookie) cookie = getUnicode(cookie)
index = cookie.index(" for ") index = cookie.index(" for ")
cookieStr += "%s; " % cookie[8:index] cookieStr += "%s; " % cookie[8:index]
if not req.has_header("Cookie") and cookieStr: if not req.has_header("Cookie") and cookieStr:
@ -163,6 +166,9 @@ class Connect:
conn = urllib2.urlopen(req) conn = urllib2.urlopen(req)
if req.has_header("Authorization"):
kb.authHeader = req.headers["Authorization"]
if hasattr(conn, "redurl") and hasattr(conn, "redcode") and not conf.redirectHandled: if hasattr(conn, "redurl") and hasattr(conn, "redcode") and not conf.redirectHandled:
msg = "sqlmap got a %d redirect to " % conn.redcode msg = "sqlmap got a %d redirect to " % conn.redcode
msg += "%s - What target address do you " % conn.redurl msg += "%s - What target address do you " % conn.redurl