sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-06-28 17:03:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

few minor just in case updates

Browse Source
This commit is contained in:
Miroslav Stampar 2012-02-27 11:15:53 +00:00
parent 85125018a1
commit 3909658fc2
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/core/dump.py
View File

@ -79,6 +79,7 @@ class Dump:
self._outputBP = StringIO.StringIO() self._outputBP = StringIO.StringIO()
def getOutputFile(self): def getOutputFile(self):
self.flush()
return self._outputFile return self._outputFile
def string(self, header, data, sort=True): def string(self, header, data, sort=True):

5
lib/utils/hashdb.py
View File

@ -10,6 +10,7 @@ See the file 'doc/COPYING' for copying permission
import hashlib import hashlib
import sqlite3 import sqlite3
import threading import threading
import time
from lib.core.common import serializeObject from lib.core.common import serializeObject
from lib.core.common import unserializeObject from lib.core.common import unserializeObject
@ -101,8 +102,10 @@ class HashDB(object):
except sqlite3.IntegrityError: except sqlite3.IntegrityError:
self.cursor.execute("UPDATE storage SET value=? WHERE id=?", (value, hash_,)) self.cursor.execute("UPDATE storage SET value=? WHERE id=?", (value, hash_,))
except sqlite3.OperationalError, ex: except sqlite3.OperationalError, ex:
if not 'locked' in ex.message: if not any(_ in ex.message for _ in ('locked', 'I/O')):
raise raise
else:
time.sleep(1)
else: else:
break break
finally: finally: