PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 20:51:31 +00:00
Code Issues Projects Releases Wiki Activity

added support for proper unicode session(s) storage/retrieval

Browse Source
This commit is contained in:
Miroslav Stampar
2010-05-24 11:00:49 +00:00
parent f34e6badfd
commit e9be60e1ac
5 changed files with 24 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/core/target.py
View File

@@ -22,6 +22,7 @@ with sqlmap; if not, write to the Free Software Foundation, Inc., 51
Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
"""
import codecs
import os
import time
@@ -121,11 +122,11 @@ def __setOutputResume():
if os.path.exists(conf.sessionFile):
if not conf.flushSession:
readSessionFP = open(conf.sessionFile, "r")
readSessionFP = codecs.open(conf.sessionFile, "r", "utf-8")
__url_cache = set()
__expression_cache = {}
for line in readSessionFP.xreadlines():
for line in readSessionFP.readlines(): #xreadlines doesn't return unicode strings when codec.open used
if line.count("][") == 4:
line = line.split("][")
@@ -170,7 +171,7 @@ def __setOutputResume():
raise sqlmapFilePathException, errMsg
try:
conf.sessionFP = open(conf.sessionFile, "a")
conf.sessionFP = codecs.open(conf.sessionFile, "a", "utf-8")
dataToSessionFile("\n[%s]\n" % time.strftime("%X %x"))
except IOError:
errMsg = "unable to write on the session file specified"