Commit 680492cb by heffnercj

Updated common.BlockFile to include the total file size; deflate.py now uses…

Updated common.BlockFile to include the total file size; deflate.py now uses this during file extraction
parent 5b9a72eb
...@@ -120,13 +120,15 @@ class BlockFile(file): ...@@ -120,13 +120,15 @@ class BlockFile(file):
self.total_read = 0 self.total_read = 0
self.offset = offset self.offset = offset
try:
self.size = file_size(fname)
except:
self.size = 0
if length: if length:
self.length = length self.length = length
else: else:
try: self.length = self.size
self.length = file_size(fname)
except:
self.length = 0
file.__init__(self, fname, mode) file.__init__(self, fname, mode)
......
...@@ -56,7 +56,7 @@ class Plugin: ...@@ -56,7 +56,7 @@ class Plugin:
# Extract the file # Extract the file
if self.binwalk.extractor.enabled: if self.binwalk.extractor.enabled:
self.binwalk.extractor.extract(loc, self.DESCRIPTION, fp.name, (fp.length - loc)) self.binwalk.extractor.extract(loc, self.DESCRIPTION, fp.name, (fp.size - loc))
if (current_total + i) > self.binwalk.scan_length: if (current_total + i) > self.binwalk.scan_length:
break break
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment