using byte string in kfxdedrm.py

pull/1380/head
xxyzz 4 years ago
parent e0c7d7d382
commit 66bab7bd7d
No known key found for this signature in database
GPG Key ID: F796163E6DCFEE9D

@ -35,7 +35,7 @@ class KFXZipBook:
for filename in zf.namelist():
with zf.open(filename) as fh:
data = fh.read(8)
if data != '\xeaDRMION\xee':
if data != b'\xeaDRMION\xee':
continue
data += fh.read()
if self.voucher is None:
@ -53,11 +53,11 @@ class KFXZipBook:
for info in zf.infolist():
with zf.open(info.filename) as fh:
data = fh.read(4)
if data != '\xe0\x01\x00\xea':
if data != b'\xe0\x01\x00\xea':
continue
data += fh.read()
if 'ProtectedData' in data:
if b'ProtectedData' in data:
break # found DRM voucher
else:
raise Exception("The .kfx-zip archive contains an encrypted DRMION file without a DRM voucher")

Loading…
Cancel
Save