def test_lzma_decompress(self):
default_buffer = "AAAAAAAA" * 90
compressed_buffer = efi_compressor.LzmaCompress(
default_buffer, len(default_buffer))
decompressed_buffer = efi_compressor.LzmaDecompress(
compressed_buffer, len(compressed_buffer))
self.assertTrue(decompressed_buffer is not None)
self.assertEqual(len(decompressed_buffer), len(default_buffer))
self.assertEqual(str(decompressed_buffer), str(default_buffer))
def dump_module(self, parent):
if self.compression == COMP_TYPE_LZMA:
dump_data("%s.module.lzma" %
os.path.join(parent, self.name), self.data)
try:
data = efi_compressor.LzmaDecompress(self.data, len(self.data))
dump_data("%s.module" % os.path.join(parent, self.name), data)
except Exception as e:
print("Cannot extract (%s), %s" % (self.name, str(e)))
return
elif self.compression == COMP_TYPE_NOT_COMPRESSED:
dump_data("%s.module" % os.path.join(parent, self.name), self.data)
def dump(self, parent=""):
if self.compression == COMP_TYPE_HUFFMAN:
pass
else:
dump_data("%s.module.lzma" % os.path.join(parent, self.name), self.data)
try:
data = efi_compressor.LzmaDecompress(self.data, len(self.data))
dump_data("%s.module" % os.path.join(parent, self.name), data)
except Exception, e:
print "Cannot extract GUID (%s), %s" % (sguid(self.guid), str(e))
return
pass
