process xor recognizes 1-byte case

kaitai-io · Apr 1, 2018 · a9d9d9a · a9d9d9a
1 parent 47faa45
commit a9d9d9a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kaitaistruct.py b/kaitaistruct.py
@@ -348,6 +348,8 @@ def process_xor_one(data, key):
     def process_xor_many(data, key):
         if key == b'\x00' * len(key):
             return data
+        if len(key) == 1:
+            return KaitaiStream.process_xor_one(data, ord(key))
 
         if PY2:
             return bytes(bytearray(a ^ b for a, b in zip(bytearray(data), itertools.cycle(bytearray(key)))))