Merge pull request #54 from yaroslav-o/master

Use bytearray for greatly improved file processing performance
onekey-sec · Jan 17, 2022 · 38d3706 · 38d3706
2 parents 175f7fa + e2f63dc
commit 38d3706
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ubireader/ubifs/list.py b/ubireader/ubifs/list.py
@@ -148,7 +148,7 @@ def file_leng(ubifs, inode):
 
 def _process_reg_file(ubifs, inode, path):
     try:
-        buf = b''
+        buf = bytearray()
         if 'data' in inode:
             compr_type = 0
             sorted_data = sorted(inode['data'], key=lambda x: x.key['khash'])
@@ -177,4 +177,4 @@ def _process_reg_file(ubifs, inode, path):
     if inode['ino'].size > len(buf):
         buf += b'\x00' * (inode['ino'].size - len(buf))
 
-    return buf
+    return bytes(buf)
diff --git a/ubireader/ubifs/output.py b/ubireader/ubifs/output.py
@@ -165,7 +165,7 @@ def _write_reg_file(path, data):
 
 def _process_reg_file(ubifs, inode, path):
     try:
-        buf = b''
+        buf = bytearray()
         if 'data' in inode:
             compr_type = 0
             sorted_data = sorted(inode['data'], key=lambda x: x.key['khash'])
@@ -194,4 +194,4 @@ def _process_reg_file(ubifs, inode, path):
     if inode['ino'].size > len(buf):
         buf += b'\x00' * (inode['ino'].size - len(buf))
 
-    return buf
+    return bytes(buf)