More attempted fixes from Paul

This commit is contained in:
Paul Durrant 2008-07-26 07:30:20 +01:00 committed by Apprentice Alf
parent 24f7ca219f
commit 87630e2bd4
1 changed files with 12 additions and 7 deletions

View File

@ -5,6 +5,7 @@
# 0.01 - Initial version # 0.01 - Initial version
# 0.02 - Huffdic compressed books were not properly decrypted # 0.02 - Huffdic compressed books were not properly decrypted
# 0.03 - fix 0.02 to work with all Mobipocket eBooks # 0.03 - fix 0.02 to work with all Mobipocket eBooks
# 0.04 - Wasn't checking MOBI header length
import sys,struct,binascii import sys,struct,binascii
@ -72,7 +73,7 @@ def getSizeOfTrailingDataEntries(ptr, size, flags):
return result return result
num = 0 num = 0
flags >>= 1 flags >>= 1
# while flags: while flags:
if flags & 1: if flags & 1:
num += getSizeOfTrailingDataEntry(ptr, size - num) num += getSizeOfTrailingDataEntry(ptr, size - num)
flags >>= 1 flags >>= 1
@ -136,8 +137,12 @@ class DrmStripper:
sect = self.loadSection(0) sect = self.loadSection(0)
records, = struct.unpack('>H', sect[0x8:0x8+2]) records, = struct.unpack('>H', sect[0x8:0x8+2])
mobi_length, = struct.unpack('>L',sect[0x14:0x18])
extra_data_flags = 0
if mobi_length >= 0xE4:
extra_data_flags, = struct.unpack('>L', sect[0xF0:0xF4]) extra_data_flags, = struct.unpack('>L', sect[0xF0:0xF4])
crypto_type, = struct.unpack('>H', sect[0xC:0xC+2]) crypto_type, = struct.unpack('>H', sect[0xC:0xC+2])
if crypto_type != 2: if crypto_type != 2:
raise DrmException("invalid encryption type: %d" % crypto_type) raise DrmException("invalid encryption type: %d" % crypto_type)
@ -165,7 +170,7 @@ class DrmStripper:
def getResult(self): def getResult(self):
return self.data_file return self.data_file
print "MobiDeDrm v0.03. Copyright (c) 2008 The Dark Reverser" print "MobiDeDrm v0.04. Copyright (c) 2008 The Dark Reverser"
if len(sys.argv)<4: if len(sys.argv)<4:
print "Removes protection from Mobipocket books" print "Removes protection from Mobipocket books"
print "Usage:" print "Usage:"