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

speedup of initial session file handling

Browse Source
This commit is contained in:
Miroslav Stampar 2010-05-11 13:36:30 +00:00
parent 6752e66164
commit 1e5ecbaa97
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lib/core/target.py
View File

@ -122,9 +122,10 @@ def __setOutputResume():
if os.path.exists(conf.sessionFile): if os.path.exists(conf.sessionFile):
if not conf.flushSession: if not conf.flushSession:
readSessionFP = open(conf.sessionFile, "r") readSessionFP = open(conf.sessionFile, "r")
lines = readSessionFP.readlines() __url_cache = set()
__expression_cache = {}
for line in lines: for line in readSessionFP.xreadlines():
if line.count("][") == 4: if line.count("][") == 4:
line = line.split("][") line = line.split("][")
@ -145,14 +146,17 @@ def __setOutputResume():
if url not in ( conf.url, conf.hostname ): if url not in ( conf.url, conf.hostname ):
continue continue
if url not in kb.resumedQueries.keys(): if url not in __url_cache:
kb.resumedQueries[url] = {} kb.resumedQueries[url] = {}
kb.resumedQueries[url][expression] = value kb.resumedQueries[url][expression] = value
__url_cache.add(url)
__expression_cache[url] = set()
resumeConfKb(expression, url, value) resumeConfKb(expression, url, value)
if expression not in kb.resumedQueries[url].keys(): if expression not in __expression_cache[url]:
kb.resumedQueries[url][expression] = value kb.resumedQueries[url][expression] = value
__expression_cache[url].add(value)
elif len(value) >= len(kb.resumedQueries[url][expression]): elif len(value) >= len(kb.resumedQueries[url][expression]):
kb.resumedQueries[url][expression] = value kb.resumedQueries[url][expression] = value