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

some more optimizations

Browse Source
This commit is contained in:
Miroslav Stampar
2011-11-22 10:54:29 +00:00
parent 267d67b024
commit 9697e80013
5 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
lib/utils/hashdb.py
View File

@@ -12,6 +12,7 @@ import sqlite3
import threading
from lib.core.data import conf
from lib.core.settings import HASHDB_FLUSH_THRESHOLD
from lib.core.settings import UNICODE_ENCODING
from lib.core.threads import getCurrentThreadData
from lib.core.threads import getCurrentThreadName
@@ -54,15 +55,17 @@ class HashDB(object):
retVal = None
if key:
hash_ = HashDB.hashKey(key)
while True:
try:
for row in self.cursor.execute("SELECT value FROM storage WHERE id=?", (hash_,)):
retVal = row[0]
except sqlite3.OperationalError, ex:
if not 'locked' in ex.message:
raise
else:
break
retVal = self._write_cache.get(hash_, None)
if not retVal:
while True:
try:
for row in self.cursor.execute("SELECT value FROM storage WHERE id=?", (hash_,)):
retVal = row[0]
except sqlite3.OperationalError, ex:
if not 'locked' in ex.message:
raise
else:
break
return retVal
def write(self, key, value):
@@ -75,10 +78,13 @@ class HashDB(object):
if getCurrentThreadName() in ('0', 'MainThread'):
self.flush()
def flush(self):
def flush(self, forced=False):
if not self._write_cache:
return
if not forced and len(self._write_cache) < HASHDB_FLUSH_THRESHOLD:
return
self._cache_lock.acquire()
items = self._write_cache.items()
self._write_cache.clear()