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

Update for a last update

Browse Source
This commit is contained in:
Miroslav Stampar 2013-02-12 12:58:15 +01:00
parent 72984a578d
commit c67b39d14d
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lib/core/common.py
View File

@ -3355,15 +3355,21 @@ def resetCookieJar(cookieJar):
cookieJar.clear()
else:
try:
content = readCachedFileContent(conf.loadCookies)
lines = filter(None, (line.strip() for line in content.split("\n") if not line.startswith('#')))
handle, filename = tempfile.mkstemp()
os.close(handle)
with open(filename, "w+b") as f:
f.write("%s\n" % NETSCAPE_FORMAT_HEADER_COOKIES)
for line in lines:
f.write("\n%s" % "\t".join(line.split()))
cookieJar.load(filename)
if not cookieJar.filename:
content = readCachedFileContent(conf.loadCookies)
lines = filter(None, (line.strip() for line in content.split("\n") if not line.startswith('#')))
handle, filename = tempfile.mkstemp()
os.close(handle)
with open(filename, "w+b") as f:
f.write("%s\n" % NETSCAPE_FORMAT_HEADER_COOKIES)
for line in lines:
f.write("\n%s" % "\t".join(line.split()))
cookieJar.filename = filename
cookieJar.load(cookieJar.filename)
if not cookieJar._cookies:
errMsg = "no valid cookies found. Please "
errMsg += "make sure that cookies aren't expired"
raise SqlmapGenericException(errMsg)
except cookielib.LoadError, msg:
errMsg = "there was a problem loading "
errMsg += "cookies file ('%s')" % msg