Mirrors
/
DeDRM_tools
mirror of https://github.com/noDRM/DeDRM_tools.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
Kindle_Mobi_Tools/lib/mobidedrm.py
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,10 +73,10 @@ 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
return num return num
@ -136,7 +137,11 @@ 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])
extra_data_flags, = struct.unpack('>L', sect[0xF0:0xF4]) 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])
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:
@ -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:"
@ -178,4 +183,4 @@ else:
try: try:
file(outfile, 'wb').write(DrmStripper(data_file, pid).getResult()) file(outfile, 'wb').write(DrmStripper(data_file, pid).getResult())
except DrmException, e: except DrmException, e:
print "Error: %s" % e print "Error: %s" % e