PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-06 12:41:30 +00:00
Code Issues Projects Releases Wiki Activity

Patch for an Issue #979

Browse Source
This commit is contained in:
Miroslav Stampar
2014-12-01 00:29:25 +01:00
parent 56b6bf72f4
commit f71a65a9a0
3 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/core/bigarray.py
View File

@@ -15,6 +15,7 @@ import tempfile
from lib.core.exception import SqlmapSystemException
from lib.core.settings import BIGARRAY_CHUNK_LENGTH
from lib.core.settings import BIGARRAY_TEMP_PREFIX
class Cache(object):
"""
@@ -35,14 +36,12 @@ class BigArray(list):
self.chunks = [[]]
self.cache = None
self.filenames = set()
self.protected = False
self._os_remove = os.remove
def append(self, value):
self.chunks[-1].append(value)
if len(self.chunks[-1]) >= BIGARRAY_CHUNK_LENGTH:
filename = self._dump(self.chunks[-1])
del(self.chunks[-1][:])
self.chunks[-1] = filename
self.chunks.append([])
@@ -65,7 +64,7 @@ class BigArray(list):
def _dump(self, value):
try:
handle, filename = tempfile.mkstemp(prefix="sqlmapba-")
handle, filename = tempfile.mkstemp(prefix=BIGARRAY_TEMP_PREFIX)
self.filenames.add(filename)
os.close(handle)
with open(filename, "w+b") as fp:
@@ -85,12 +84,11 @@ class BigArray(list):
self.cache = Cache(index, pickle.load(fp), False)
def __getstate__(self):
self.protected = True
return self.chunks, self.filenames, self.protected
return self.chunks, self.filenames
def __setstate__(self, state):
self.__init__()
self.chunks, self.filenames, self.protected = state
self.chunks, self.filenames = state
def __getslice__(self, i, j):
retval = BigArray()
@@ -132,11 +130,3 @@ class BigArray(list):
def __len__(self):
return len(self.chunks[-1]) if len(self.chunks) == 1 else (len(self.chunks) - 1) * BIGARRAY_CHUNK_LENGTH + len(self.chunks[-1])
def __del__(self):
if not self.protected:
for filename in self.filenames:
try:
self._os_remove(filename)
except:
pass