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