sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-16 19:40:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Using cPickle in BigArray (faster and potentially less memory used)

Browse Source
This commit is contained in:
Miroslav Stampar 2013-10-21 20:48:00 +02:00
parent 8dac47f7e5
commit 9f21406a4b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/core/bigarray.py
View File

@ -5,8 +5,12 @@ Copyright (c) 2006-2013 sqlmap developers (http://sqlmap.org/)
See the file 'doc/COPYING' for copying permission See the file 'doc/COPYING' for copying permission
""" """
try:
import cPickle as pickle
except:
import pickle
import os import os
import pickle
import tempfile import tempfile
from lib.core.settings import BIGARRAY_CHUNK_LENGTH from lib.core.settings import BIGARRAY_CHUNK_LENGTH
@ -62,7 +66,7 @@ class BigArray(list):
self.filenames.add(filename) self.filenames.add(filename)
os.close(handle) os.close(handle)
with open(filename, "w+b") as fp: with open(filename, "w+b") as fp:
pickle.dump(value, fp) pickle.dump(value, fp, pickle.HIGHEST_PROTOCOL)
return filename return filename
def _checkcache(self, index): def _checkcache(self, index):