mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2026-08-04 14:55:40 +00:00
PyPy patch for testing module
This commit is contained in:
parent
08eeb0fd88
commit
aeaa776ca9
4 changed files with 13 additions and 3 deletions
|
|
@ -33,6 +33,7 @@ class HashDB(object):
|
|||
self.filepath = filepath
|
||||
self._write_cache = {}
|
||||
self._cache_lock = threading.Lock()
|
||||
self._connections = []
|
||||
|
||||
def _get_cursor(self):
|
||||
threadData = getCurrentThreadData()
|
||||
|
|
@ -40,6 +41,7 @@ class HashDB(object):
|
|||
if threadData.hashDBCursor is None:
|
||||
try:
|
||||
connection = sqlite3.connect(self.filepath, timeout=3, isolation_level=None)
|
||||
self._connections.append(connection)
|
||||
threadData.hashDBCursor = connection.cursor()
|
||||
threadData.hashDBCursor.execute("CREATE TABLE IF NOT EXISTS storage (id INTEGER PRIMARY KEY, value TEXT)")
|
||||
connection.commit()
|
||||
|
|
@ -66,6 +68,14 @@ class HashDB(object):
|
|||
except:
|
||||
pass
|
||||
|
||||
def closeAll(self):
|
||||
for connection in self._connections:
|
||||
try:
|
||||
connection.commit()
|
||||
connection.close()
|
||||
except:
|
||||
pass
|
||||
|
||||
@staticmethod
|
||||
def hashKey(key):
|
||||
key = getBytes(key if isinstance(key, six.text_type) else repr(key), errors="xmlcharrefreplace")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue