Minor bug fix

bdamele · bdamele · commit 36bc410333d5 · 2010-10-18T09:50:23.000Z
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -169,11 +169,11 @@ def getPage(**kwargs):
 
             conn = urllib2.urlopen(req)
 
-            if req.has_header("Authorization"):
-                kb.authHeader = req.headers["Authorization"]
+            if not kb.authHeader and req.has_header("Authorization"):
+                kb.authHeader = req.get_header("Authorization")
 
-            if req.has_header("Proxy-authorization"):
-                kb.proxyAuthHeader = req.headers["Proxy-authorization"]
+            if not kb.proxyAuthHeader and req.has_header("Proxy-authorization"):
+                kb.proxyAuthHeader = req.get_header("Proxy-authorization")
 
             if hasattr(conn, "redurl") and hasattr(conn, "redcode") and not conf.redirectHandled:
                 msg  = "sqlmap got a %d redirect to " % conn.redcode
