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

minor patch

Browse Source
This commit is contained in:
Miroslav Stampar 2011-11-22 11:04:43 +00:00
parent 9697e80013
commit ac041399f0
2 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/core/settings.py
View File

@ -417,4 +417,4 @@ DEFAULT_GET_POST_DELIMITER = '&'
DEFAULT_COOKIE_DELIMITER = ';'
# Skip unforced HashDB flush requests below the threshold number of cached items
HASHDB_FLUSH_THRESHOLD = 10
HASHDB_FLUSH_THRESHOLD = 32

28
lib/utils/hashdb.py
View File

@ -90,20 +90,22 @@ class HashDB(object):
self._write_cache.clear()
self._cache_lock.release()
self.beginTransaction()
for hash_, value in items:
while True:
try:
try:
self.beginTransaction()
for hash_, value in items:
while True:
try:
self.cursor.execute("INSERT INTO storage VALUES (?, ?)", (hash_, value,))
except sqlite3.IntegrityError:
self.cursor.execute("UPDATE storage SET value=? WHERE id=?", (value, hash_,))
except sqlite3.OperationalError, ex:
if not 'locked' in ex.message:
raise
else:
break
self.endTransaction()
try:
self.cursor.execute("INSERT INTO storage VALUES (?, ?)", (hash_, value,))
except sqlite3.IntegrityError:
self.cursor.execute("UPDATE storage SET value=? WHERE id=?", (value, hash_,))
except sqlite3.OperationalError, ex:
if not 'locked' in ex.message:
raise
else:
break
finally:
self.endTransaction()
def beginTransaction(self):
self.cursor.execute('BEGIN TRANSACTION')