CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-03 22:31:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Patch for an Issue #193

Browse source
This commit is contained in:
Miroslav Stampar 2012-09-25 11:21:39 +02:00
parent c9e7e71ea2
commit fccdb824bb
2 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
lib/utils/hashdb.py
View file

@ -14,6 +14,8 @@ import time
from lib.core.common import getUnicode
from lib.core.common import serializeObject
from lib.core.common import unserializeObject
from lib.core.data import logger
from lib.core.settings import HASHDB_FLUSH_RETRIES
from lib.core.settings import HASHDB_FLUSH_THRESHOLD
from lib.core.settings import UNICODE_ENCODING
from lib.core.threads import getCurrentThreadData
@ -95,6 +97,7 @@ class HashDB(object):
try:
self.beginTransaction()
for hash_, value in _.items():
retries = 0
while True:
try:
try:
@ -102,9 +105,16 @@ class HashDB(object):
except sqlite3.IntegrityError:
self.cursor.execute("UPDATE storage SET value=? WHERE id=?", (value, hash_,))
except sqlite3.OperationalError, ex:
if not any(_ in ex.message for _ in ('locked', 'I/O')):
raise
if retries == 0:
warnMsg = "there has been a problem while writing to "
warnMsg += "the session file ('%s')" % ex.message
logger.warn(warnMsg)
if retries >= HASHDB_FLUSH_RETRIES:
return
else:
retries += 1
time.sleep(1)
else:
break