Merge pull request #89 from UserJHansen/patch-1

Fix for incorrect key size with 0 length keySize var
nolze · May 24, 2024 · f9c9e1a · f9c9e1a
2 parents 7336859 f144612
commit f9c9e1a
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 60 deletions.
diff --git a/msoffcrypto/format/common.py b/msoffcrypto/format/common.py
@@ -1,3  1,4 @@
 import io
 import logging
 from struct import unpack
 
@@ -49,3  50,25 @@ def _parse_encryptionverifier(blob, algorithm):
         "encryptedVerifierHash": encryptedVerifierHash,
     }
     return verifier
 
 
 def _parse_header_RC4CryptoAPI(encryptionHeader):
     flags = encryptionHeader.read(4)
     (headerSize,) = unpack("<I", encryptionHeader.read(4))
     logger.debug(headerSize)
     blob = io.BytesIO(encryptionHeader.read(headerSize))
     header = _parse_encryptionheader(blob)
     logger.debug(header)
     # NOTE: https://learn.microsoft.com/en-us/openspecs/office_file_formats/ms-offcrypto/36cfb17f-9b15-4a9b-911a-f401f60b3991
     keySize = 0x00000028 if header["keySize"] == 0 else header["keySize"]
 
     blob = io.BytesIO(encryptionHeader.read())
     verifier = _parse_encryptionverifier(blob, "RC4")  # TODO: Fix (cf. ooxml.py)
     logger.debug(verifier)
     info = {
         "salt": verifier["salt"],
         "keySize": keySize,
         "encryptedVerifier": verifier["encryptedVerifier"],
         "encryptedVerifierHash": verifier["encryptedVerifierHash"],
     }
     return info
diff --git a/msoffcrypto/format/doc97.py b/msoffcrypto/format/doc97.py
@@ -9,7  9,7 @@
 
 from msoffcrypto import exceptions
 from msoffcrypto.format import base
-from msoffcrypto.format.common import _parse_encryptionheader, _parse_encryptionverifier
 from msoffcrypto.format.common import _parse_header_RC4CryptoAPI
 from msoffcrypto.method.rc4 import DocumentRC4
 from msoffcrypto.method.rc4_cryptoapi import DocumentRC4CryptoAPI
 
@@ -258,25  258,6 @@ def _parse_header_RC4(encryptionHeader):
     return info
 
 
-def _parse_header_RC4CryptoAPI(encryptionHeader):
-    flags = encryptionHeader.read(4)
-    (headerSize,) = unpack("<I", encryptionHeader.read(4))
-    logger.debug(headerSize)
-    blob = io.BytesIO(encryptionHeader.read(headerSize))
-    header = _parse_encryptionheader(blob)
-    logger.debug(header)
-    blob = io.BytesIO(encryptionHeader.read())
-    verifier = _parse_encryptionverifier(blob, "RC4")  # TODO: Fix (cf. ooxml.py)
-    logger.debug(verifier)
-    info = {
-        "salt": verifier["salt"],
-        "keySize": header["keySize"],
-        "encryptedVerifier": verifier["encryptedVerifier"],
-        "encryptedVerifierHash": verifier["encryptedVerifierHash"],
-    }
-    return info
-
-
 class Doc97File(base.BaseOfficeFile):
     """Return a MS-DOC file object.
 

diff --git a/msoffcrypto/format/ppt97.py b/msoffcrypto/format/ppt97.py
@@ -9,7  9,7 @@
 
 from msoffcrypto import exceptions
 from msoffcrypto.format import base
-from msoffcrypto.format.common import _parse_encryptionheader, _parse_encryptionverifier
 from msoffcrypto.format.common import _parse_header_RC4CryptoAPI
 from msoffcrypto.method.rc4_cryptoapi import DocumentRC4CryptoAPI
 
 logger = logging.getLogger(__name__)
@@ -502,25  502,6 @@ def construct_persistobjectdirectory(data):
     return persistobjectdirectory
 
 
-def _parse_header_RC4CryptoAPI(encryptionInfo):
-    flags = encryptionInfo.read(4)
-    (headerSize,) = unpack("<I", encryptionInfo.read(4))
-    logger.debug(headerSize)
-    blob = io.BytesIO(encryptionInfo.read(headerSize))
-    header = _parse_encryptionheader(blob)
-    logger.debug(header)
-    blob = io.BytesIO(encryptionInfo.read())
-    verifier = _parse_encryptionverifier(blob, "RC4")  # TODO: Fix (cf. ooxml.py)
-    logger.debug(verifier)
-    info = {
-        "salt": verifier["salt"],
-        "keySize": header["keySize"],
-        "encryptedVerifier": verifier["encryptedVerifier"],
-        "encryptedVerifierHash": verifier["encryptedVerifierHash"],
-    }
-    return info
-
-
 class Ppt97File(base.BaseOfficeFile):
     """Return a MS-PPT file object.
 

diff --git a/msoffcrypto/format/xls97.py b/msoffcrypto/format/xls97.py
@@ -6,7  6,7 @@
 
 from msoffcrypto import exceptions
 from msoffcrypto.format import base
-from msoffcrypto.format.common import _parse_encryptionheader, _parse_encryptionverifier
 from msoffcrypto.format.common import _parse_header_RC4CryptoAPI
 from msoffcrypto.method.rc4 import DocumentRC4
 from msoffcrypto.method.rc4_cryptoapi import DocumentRC4CryptoAPI
 from msoffcrypto.method.xor_obfuscation import DocumentXOR
@@ -386,25  386,6 @@ def _parse_header_RC4(encryptionInfo):
     return info
 
 
-def _parse_header_RC4CryptoAPI(encryptionInfo):
-    flags = encryptionInfo.read(4)
-    (headerSize,) = unpack("<I", encryptionInfo.read(4))
-    logger.debug(headerSize)
-    blob = io.BytesIO(encryptionInfo.read(headerSize))
-    header = _parse_encryptionheader(blob)
-    logger.debug(header)
-    blob = io.BytesIO(encryptionInfo.read())
-    verifier = _parse_encryptionverifier(blob, "RC4")  # TODO: Fix (cf. ooxml.py)
-    logger.debug(verifier)
-    info = {
-        "salt": verifier["salt"],
-        "keySize": header["keySize"],
-        "encryptedVerifier": verifier["encryptedVerifier"],
-        "encryptedVerifierHash": verifier["encryptedVerifierHash"],
-    }
-    return info
-
-
 class _BIFFStream:
     def __init__(self, data):
         self.data = data