More attempted fixes from Paul
This commit is contained in:
parent
24f7ca219f
commit
87630e2bd4
|
@ -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:"
|
||||||
|
|
Loading…
Reference in New Issue