mirrors/DeDRM_tools
mirrors/DeDRM_tools
1
0
Fork 0
mirror of https://github.com/noDRM/DeDRM_tools.git synced 2025-01-16 12:50:37 +06:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1380 from xxyzz/byte-string

Browse Source 
Fix byte string error for KFX
This commit is contained in:
Apprentice Harper 2020-11-28 16:19:17 +00:00 committed by GitHub
commit 981aadc497
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
DeDRM_plugin/ion.py
View File

@ -31,7 +31,7 @@ import struct
from io import BytesIO from io import BytesIO
from Crypto.Cipher import AES from Crypto.Cipher import AES
from Crypto.Util.py3compat import bchr, bord from Crypto.Util.py3compat import bchr
try: try:
# lzma library from calibre 4.6.0 or later # lzma library from calibre 4.6.0 or later
@ -89,7 +89,7 @@ LEN_IS_VAR_LEN = 0xE
LEN_IS_NULL = 0xF LEN_IS_NULL = 0xF
VERSION_MARKER = b"\x01\x00\xEA" VERSION_MARKER = [b"\x01", b"\x00", b"\xEA"]
# asserts must always raise exceptions for proper functioning # asserts must always raise exceptions for proper functioning
@ -347,7 +347,7 @@ class BinaryIonParser(object):
b = self.stream.read(1) b = self.stream.read(1)
if len(b) < 1: if len(b) < 1:
return -1 return -1
b = bord(b) b = ord(b)
result = b >> 4 result = b >> 4
ln = b & 0xF ln = b & 0xF
@ -372,13 +372,13 @@ class BinaryIonParser(object):
return result return result
def readvarint(self): def readvarint(self):
b = bord(self.read()) b = ord(self.read())
negative = ((b & 0x40) != 0) negative = ((b & 0x40) != 0)
result = (b & 0x3F) result = (b & 0x3F)
i = 0 i = 0
while (b & 0x80) == 0 and i < 4: while (b & 0x80) == 0 and i < 4:
b = bord(self.read()) b = ord(self.read())
result = (result << 7) | (b & 0x7F) result = (result << 7) | (b & 0x7F)
i += 1 i += 1
@ -389,12 +389,12 @@ class BinaryIonParser(object):
return result return result
def readvaruint(self): def readvaruint(self):
b = bord(self.read()) b = ord(self.read())
result = (b & 0x7F) result = (b & 0x7F)
i = 0 i = 0
while (b & 0x80) == 0 and i < 4: while (b & 0x80) == 0 and i < 4:
b = bord(self.read()) b = ord(self.read())
result = (result << 7) | (b & 0x7F) result = (result << 7) | (b & 0x7F)
i += 1 i += 1
@ -414,7 +414,7 @@ class BinaryIonParser(object):
_assert(self.localremaining <= 8, "Decimal overflow") _assert(self.localremaining <= 8, "Decimal overflow")
signed = False signed = False
b = [bord(x) for x in self.read(self.localremaining)] b = [ord(x) for x in self.read(self.localremaining)]
if (b[0] & 0x80) != 0: if (b[0] & 0x80) != 0:
b[0] = b[0] & 0x7F b[0] = b[0] & 0x7F
signed = True signed = True
@ -579,7 +579,7 @@ class BinaryIonParser(object):
_assert(self.valuelen <= 4, "int too long: %d" % self.valuelen) _assert(self.valuelen <= 4, "int too long: %d" % self.valuelen)
v = 0 v = 0
for i in range(self.valuelen - 1, -1, -1): for i in range(self.valuelen - 1, -1, -1):
v = (v | (bord(self.read()) << (i * 8))) v = (v | (ord(self.read()) << (i * 8)))
if self.valuetid == TID_NEGINT: if self.valuetid == TID_NEGINT:
self.value = -v self.value = -v
@ -649,7 +649,7 @@ class BinaryIonParser(object):
result = "" result = ""
for i in b: for i in b:
result += ("%02x " % bord(i)) result += ("%02x " % ord(i))
if len(result) > 0: if len(result) > 0:
result = result[:-1] result = result[:-1]
@ -748,7 +748,8 @@ def pkcs7pad(msg, blocklen):
def pkcs7unpad(msg, blocklen): def pkcs7unpad(msg, blocklen):
_assert(len(msg) % blocklen == 0) _assert(len(msg) % blocklen == 0)
paddinglen = bord(msg[-1]) paddinglen = msg[-1]
_assert(paddinglen > 0 and paddinglen <= blocklen, "Incorrect padding - Wrong key") _assert(paddinglen > 0 and paddinglen <= blocklen, "Incorrect padding - Wrong key")
_assert(msg[-paddinglen:] == bchr(paddinglen) * paddinglen, "Incorrect padding - Wrong key") _assert(msg[-paddinglen:] == bchr(paddinglen) * paddinglen, "Incorrect padding - Wrong key")
@ -806,7 +807,7 @@ class DrmIonVoucher(object):
secretkey = b"" secretkey = b""
def __init__(self, voucherenv, dsn, secret): def __init__(self, voucherenv, dsn, secret):
self.dsn,self.secret = dsn,secret self.dsn, self.secret = dsn, secret
self.lockparams = [] self.lockparams = []
@ -827,7 +828,7 @@ class DrmIonVoucher(object):
sharedsecret = obfuscate(shared.encode('ASCII'), self.version) sharedsecret = obfuscate(shared.encode('ASCII'), self.version)
key = hmac.new(sharedsecret, "PIDv3", digestmod=hashlib.sha256).digest() key = hmac.new(sharedsecret, b"PIDv3", digestmod=hashlib.sha256).digest()
aes = AES.new(key[:32], AES.MODE_CBC, self.cipheriv[:16]) aes = AES.new(key[:32], AES.MODE_CBC, self.cipheriv[:16])
b = aes.decrypt(self.ciphertext) b = aes.decrypt(self.ciphertext)
b = pkcs7unpad(b, 16) b = pkcs7unpad(b, 16)
@ -1024,7 +1025,7 @@ class DrmIon(object):
outpages.write(msg) outpages.write(msg)
return return
_assert(msg[0] == b"\x00", "LZMA UseFilter not supported") _assert(msg[0] == 0, "LZMA UseFilter not supported")
if calibre_lzma is not None: if calibre_lzma is not None:
with calibre_lzma.decompress(msg[1:], bufsize=0x1000000) as f: with calibre_lzma.decompress(msg[1:], bufsize=0x1000000) as f:

15
DeDRM_plugin/kfxdedrm.py
View File

@ -11,6 +11,7 @@ import shutil
import zipfile import zipfile
from io import BytesIO from io import BytesIO
from calibre_plugins.dedrm.ion import DrmIon, DrmIonVoucher
__license__ = 'GPL v3' __license__ = 'GPL v3'
@ -27,22 +28,18 @@ class KFXZipBook:
return (None, None) return (None, None)
def processBook(self, totalpids): def processBook(self, totalpids):
try:
import ion
except:
from calibre_plugins.dedrm import ion
with zipfile.ZipFile(self.infile, 'r') as zf: with zipfile.ZipFile(self.infile, 'r') as zf:
for filename in zf.namelist(): for filename in zf.namelist():
with zf.open(filename) as fh: with zf.open(filename) as fh:
data = fh.read(8) data = fh.read(8)
if data != '\xeaDRMION\xee': if data != b'\xeaDRMION\xee':
continue continue
data += fh.read() data += fh.read()
if self.voucher is None: if self.voucher is None:
self.decrypt_voucher(totalpids) self.decrypt_voucher(totalpids)
print("Decrypting KFX DRMION: {0}".format(filename)) print("Decrypting KFX DRMION: {0}".format(filename))
outfile = BytesIO() outfile = BytesIO()
ion.DrmIon(BytesIO(data[8:-8]), lambda name: self.voucher).parse(outfile) DrmIon(BytesIO(data[8:-8]), lambda name: self.voucher).parse(outfile)
self.decrypted[filename] = outfile.getvalue() self.decrypted[filename] = outfile.getvalue()
if not self.decrypted: if not self.decrypted:
@ -53,11 +50,11 @@ class KFXZipBook:
for info in zf.infolist(): for info in zf.infolist():
with zf.open(info.filename) as fh: with zf.open(info.filename) as fh:
data = fh.read(4) data = fh.read(4)
if data != '\xe0\x01\x00\xea': if data != b'\xe0\x01\x00\xea':
continue continue
data += fh.read() data += fh.read()
if 'ProtectedData' in data: if b'ProtectedData' in data:
break # found DRM voucher break # found DRM voucher
else: else:
raise Exception("The .kfx-zip archive contains an encrypted DRMION file without a DRM voucher") raise Exception("The .kfx-zip archive contains an encrypted DRMION file without a DRM voucher")
@ -72,7 +69,7 @@ class KFXZipBook:
continue continue
try: try:
voucher = ion.DrmIonVoucher(BytesIO(data), pid[:dsn_len], pid[dsn_len:]) voucher = DrmIonVoucher(BytesIO(data), pid[:dsn_len], pid[dsn_len:])
voucher.parse() voucher.parse()
voucher.decryptvoucher() voucher.decryptvoucher()
break break