sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2024-11-25 11:03:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Major bug fix following the last commit

Browse Source
This commit is contained in:
Bernardo Damele 2008-10-27 23:56:02 +00:00
parent eb6e6f4d03
commit 9895338630
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lib/request/connect.py
View File

@ -77,12 +77,14 @@ class Connect:
params = urlencode(params).replace("%%", "%") params = urlencode(params).replace("%%", "%")
url = "%s?%s" % (url, params) url = "%s?%s" % (url, params)
requestMsg += "?%s" % params requestMsg += "?%s" % params
elif multipart: elif multipart:
multipartOpener = urllib2.build_opener(multipartpost.MultipartPostHandler) multipartOpener = urllib2.build_opener(multipartpost.MultipartPostHandler)
conn = multipartOpener.open(url, multipart) conn = multipartOpener.open(url, multipart)
page = conn.read() page = conn.read()
return page return page
elif conf.method in ( "GET", "POST" ):
else:
if conf.parameters.has_key("GET") and not get: if conf.parameters.has_key("GET") and not get:
get = conf.parameters["GET"] get = conf.parameters["GET"]
@ -90,11 +92,12 @@ class Connect:
get = urlencode(get).replace("%%", "%") get = urlencode(get).replace("%%", "%")
url = "%s?%s" % (url, get) url = "%s?%s" % (url, get)
requestMsg += "?%s" % get requestMsg += "?%s" % get
elif conf.method == "POST":
if conf.parameters.has_key("POST") and not post:
post = conf.parameters["POST"]
post = urlencode(post).replace("%%", "%") if conf.method == "POST":
if conf.parameters.has_key("POST") and not post:
post = conf.parameters["POST"]
post = urlencode(post).replace("%%", "%")
requestMsg += " HTTP/1.1" requestMsg += " HTTP/1.1"
@ -107,7 +110,7 @@ class Connect:
req = urllib2.Request(url, post, headers) req = urllib2.Request(url, post, headers)
conn = urllib2.urlopen(req) conn = urllib2.urlopen(req)
if "Accept-Encoding" not in req.headers: if not req.has_header("Accept-Encoding"):
requestHeaders += "\nAccept-Encoding: identity" requestHeaders += "\nAccept-Encoding: identity"
requestHeaders = "\n".join(["%s: %s" % (header, value) for header, value in req.header_items()]) requestHeaders = "\n".join(["%s: %s" % (header, value) for header, value in req.header_items()])
@ -121,10 +124,10 @@ class Connect:
cookieStr += "%s; " % cookie[8:index] cookieStr += "%s; " % cookie[8:index]
if "Cookie" not in req.headers and cookieStr: if not req.has_header("Cookie") and cookieStr:
requestHeaders += "\n%s" % cookieStr[:-2] requestHeaders += "\n%s" % cookieStr[:-2]
if "Connection" not in req.headers: if not req.has_header("Connection"):
requestHeaders += "\nConnection: close" requestHeaders += "\nConnection: close"
requestMsg += "\n%s" % requestHeaders requestMsg += "\n%s" % requestHeaders