tools v2.2

Kindle_Mobi_Tools/Kindle_4_Mac_Unswindle/lib/mobidedrm.py

@@ -37,10 +37,17 @@
 #         in utf8 file are encrypted. (Although neither kind gets compressed.)
 #         This knowledge leads to a simplification of the test for the 
 #         trailing data byte flags - version 5 and higher AND header size >= 0xE4. 
-#  0.15 - Now outputs 'hearbeat', and is also quicker for long files.
+#  0.15 - Now outputs 'heartbeat', and is also quicker for long files.
 #  0.16 - And reverts to 'done' not 'done.' at the end for unswindle compatibility.
+#  0.17 - added modifications to support its use as an imported python module
+#         both inside calibre and also in other places (ie K4DeDRM tools)
+#  0.17a- disabled the standalone plugin feature since a plugin can not import
+#         a plugin
+#  0.18 - It seems that multibyte entries aren't encrypted in a v7 file...
+#         Removed the disabled Calibre plug-in code
+#         Permit use of 8-digit PIDs
 
-__version__ = '0.16'
+__version__ = '0.18'
 
 import sys
 import struct
@@ -123,10 +130,11 @@ def getSizeOfTrailingDataEntries(ptr, size, flags):
         if testflags & 1:
             num += getSizeOfTrailingDataEntry(ptr, size - num)
         testflags >>= 1
-    # Multibyte data, if present, is included in the encryption, so
-    # we do not need to check the low bit.
-    # if flags & 1:
-    #    num += (ord(ptr[size - num - 1]) & 0x3) + 1
+    # Check the low bit to see if there's multibyte data present.
+    # if multibyte data is included in the encryped data, we'll
+    # have already cleared this flag.
+    if flags & 1:
+        num += (ord(ptr[size - num - 1]) & 0x3) + 1
     return num
 
 class DrmStripper:
@@ -177,9 +185,14 @@ class DrmStripper:
         return found_key
 
     def __init__(self, data_file, pid):
-        if checksumPid(pid[0:-2]) != pid:
-            raise DrmException("invalid PID checksum")
-        pid = pid[0:-2]
+        if len(pid)==10:
+            if checksumPid(pid[0:-2]) != pid:
+                raise DrmException("invalid PID checksum")
+            pid = pid[0:-2]
+        elif len(pid)==8:
+            print "PID without checksum given. With checksum PID is "+checksumPid(pid)
+        else:
+            raise DrmException("Invalid PID length")
 
         self.data_file = data_file
         header = data_file[0:72]
@@ -202,6 +215,10 @@ class DrmStripper:
         if (mobi_length >= 0xE4) and (mobi_version >= 5):
             extra_data_flags, = struct.unpack('>H', sect[0xF2:0xF4])
             print "Extra Data Flags = %d" %extra_data_flags
+        if mobi_version < 7:
+            # multibyte utf8 data is included in the encryption for mobi_version 5 (& 6?)
+            # so clear that byte so that we leave it to be decrypted.
+            extra_data_flags &= 0xFFFE
 
         crypto_type, = struct.unpack('>H', sect[0xC:0xC+2])
         if crypto_type == 0:
@@ -248,63 +265,33 @@ class DrmStripper:
     def getResult(self):
         return self.data_file
 
-if not __name__ == "__main__":
-    from calibre.customize import FileTypePlugin
+def getUnencryptedBook(infile,pid):
+    sys.stdout=Unbuffered(sys.stdout)
+    data_file = file(infile, 'rb').read()
+    strippedFile = DrmStripper(data_file, pid)
+    return strippedFile.getResult()
 
-    class MobiDeDRM(FileTypePlugin):
-        name                = 'MobiDeDRM' # Name of the plugin
-        description         = 'Removes DRM from secure Mobi files'
-        supported_platforms = ['linux', 'osx', 'windows'] # Platforms this plugin will run on
-        author              = 'The Dark Reverser' # The author of this plugin
-        version             = (0, 1, 6)   # The version number of this plugin
-        file_types          = set(['prc','mobi','azw']) # The file types that this plugin will be applied to
-        on_import           = True # Run this plugin during the import
-
-        def run(self, path_to_ebook):
-            from calibre.gui2 import is_ok_to_use_qt
-            from PyQt4.Qt import QMessageBox
-            PID = self.site_customization
-            data_file = file(path_to_ebook, 'rb').read()
-            ar = PID.split(',')
-            for i in ar:
-                try:
-                    unlocked_file = DrmStripper(data_file, i).getResult()
-                except DrmException:
-                    # ignore the error
-                    pass
-                else:
-                    of = self.temporary_file('.mobi')
-                    of.write(unlocked_file)
-                    of.close()
-                    return of.name
-            if is_ok_to_use_qt():
-                d = QMessageBox(QMessageBox.Warning, "MobiDeDRM Plugin", "Couldn't decode: %s\n\nImporting encrypted version." % path_to_ebook)
-                d.show()
-                d.raise_()
-                d.exec_()
-            return path_to_ebook
-
-        def customization_help(self, gui=False):
-            return 'Enter PID (separate multiple PIDs with comma)'
-
-if __name__ == "__main__":
+def main(argv=sys.argv):
     sys.stdout=Unbuffered(sys.stdout)
     print ('MobiDeDrm v%(__version__)s. '
 	   'Copyright 2008-2010 The Dark Reverser.' % globals())
-    if len(sys.argv)<4:
+    if len(argv)<4:
         print "Removes protection from Mobipocket books"
         print "Usage:"
         print "    %s <infile> <outfile> <PID>" % sys.argv[0]
-        sys.exit(1)
+        return 1
     else:
-        infile = sys.argv[1]
-        outfile = sys.argv[2]
-        pid = sys.argv[3]
-        data_file = file(infile, 'rb').read()
+        infile = argv[1]
+        outfile = argv[2]
+        pid = argv[3]
         try:
-            strippedFile = DrmStripper(data_file, pid)
-            file(outfile, 'wb').write(strippedFile.getResult())
+            stripped_file = getUnencryptedBook(infile, pid)
+            file(outfile, 'wb').write(stripped_file)
         except DrmException, e:
             print "Error: %s" % e
-            sys.exit(1)
-    sys.exit(0)
+            return 1
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())