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

another minor improvement regarding logging of http traffic

Browse Source
This commit is contained in:
Miroslav Stampar 2010-11-17 12:16:48 +00:00
parent 3e569a1693
commit 3d25071d06
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
lib/request/connect.py
View File

@ -90,6 +90,7 @@ class Connect:
responseMsg = "HTTP response "
requestHeaders = ""
responseHeaders = ""
logHeaders = ""
try:
if silent:
@ -224,8 +225,23 @@ class Connect:
code = e.code
status = e.msg
responseMsg = "\n%s[#%d] (%d %s)\n" % (responseMsg, requestID, code, status)
logHTTPTraffic(requestMsg, responseMsg)
try:
page = e.read()
responseHeaders = e.info()
except socket.timeout:
warnMsg = "connection timed out while trying "
warnMsg += "to get error page information (%d)" % code
logger.warn(warnMsg)
return None, None
except:
pass
responseMsg = "\n%s[#%d] (%d %s):\n" % (responseMsg, requestID, code, status)
if responseHeaders:
logHeaders = "\n".join(["%s: %s" % (key.capitalize() if isinstance(key, basestring) else key, value) for (key, value) in responseHeaders.items()])
logHTTPTraffic(requestMsg, "%s%s\n\n%s" % (responseMsg, logHeaders, page))
if e.code == 401:
errMsg = "not authorized, try to provide right HTTP "
@ -235,15 +251,6 @@ class Connect:
errMsg = "page not found (%d)" % code
raise sqlmapConnectionException, errMsg
else:
try:
page = e.read()
responseHeaders = e.info()
except socket.timeout:
warnMsg = "connection timed out while trying "
warnMsg += "to get error page information (%d)" % code
logger.warn(warnMsg)
return None, None
debugMsg = "got HTTP error code: %d (%s)" % (code, status)
logger.debug(debugMsg)