CERBERUS/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2026-08-04 14:55:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor improvement of BigArray's iter

Browse source
This commit is contained in:
Miroslav Stampar 2026-01-27 11:03:40 +01:00
parent 657eb464f8
commit f31ea1e2f5
3 changed files with 26 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

28
lib/core/bigarray.py
View file

@ -305,11 +305,29 @@ class BigArray(list):
return "%s%s" % ("..." if len(self.chunks) > 1 else "", self.chunks[-1].__repr__())
def __iter__(self):
for i in xrange(len(self)):
try:
yield self[i]
except IndexError:
break
with self._lock:
chunks = list(self.chunks)
cache_index = self.cache.index if isinstance(self.cache, Cache) else None
cache_data = self.cache.data if isinstance(self.cache, Cache) else None
for idx, chunk in enumerate(chunks):
if isinstance(chunk, list):
for item in chunk:
yield item
else:
try:
if cache_index == idx and cache_data is not None:
data = cache_data
else:
with open(chunk, "rb") as f:
data = pickle.loads(zlib.decompress(f.read()))
except Exception as ex:
errMsg = "exception occurred while retrieving data "
errMsg += "from a temporary file ('%s')" % ex
raise SqlmapSystemException(errMsg)
for item in data:
yield item
def __len__(self):
return len(self.chunks[-1]) if len(self.chunks) == 1 else (len(self.chunks) - 1) * self.chunk_length + len(self.chunks[-1])