tools v5.5

Plugins now include unaltered stand-alone scripts, so no longer need to keep separate copies.

DeDRM_Macintosh_Application/DeDRM.app/Contents/Resources/kindlepid.py

@@ -1,29 +1,80 @@
 #!/usr/bin/python
-# Mobipocket PID calculator v0.2 for Amazon Kindle.
+# -*- coding: utf-8 -*-
+
+# Mobipocket PID calculator v0.4 for Amazon Kindle.
 # Copyright (c) 2007, 2009 Igor Skochinsky <skochinsky@mail.ru>
 # History:
 #  0.1 Initial release
 #  0.2 Added support for generating PID for iPhone (thanks to mbp)
 #  0.3 changed to autoflush stdout, fixed return code usage
-class Unbuffered:
+#  0.3 updated for unicode
+
+import sys
+import binascii
+
+# Wrap a stream so that output gets flushed immediately
+# and also make sure that any unicode strings get
+# encoded using "replace" before writing them.
+class SafeUnbuffered:
     def __init__(self, stream):
         self.stream = stream
+        self.encoding = stream.encoding
+        if self.encoding == None:
+            self.encoding = "utf-8"
     def write(self, data):
+        if isinstance(data,unicode):
+            data = data.encode(self.encoding,"replace")
         self.stream.write(data)
         self.stream.flush()
     def __getattr__(self, attr):
         return getattr(self.stream, attr)
 
-import sys
-sys.stdout=Unbuffered(sys.stdout)
+iswindows = sys.platform.startswith('win')
+isosx = sys.platform.startswith('darwin')
 
-import binascii
+def unicode_argv():
+    if iswindows:
+        # Uses shell32.GetCommandLineArgvW to get sys.argv as a list of Unicode
+        # strings.
+
+        # Versions 2.x of Python don't support Unicode in sys.argv on
+        # Windows, with the underlying Windows API instead replacing multi-byte
+        # characters with '?'.
+
+
+        from ctypes import POINTER, byref, cdll, c_int, windll
+        from ctypes.wintypes import LPCWSTR, LPWSTR
+
+        GetCommandLineW = cdll.kernel32.GetCommandLineW
+        GetCommandLineW.argtypes = []
+        GetCommandLineW.restype = LPCWSTR
+
+        CommandLineToArgvW = windll.shell32.CommandLineToArgvW
+        CommandLineToArgvW.argtypes = [LPCWSTR, POINTER(c_int)]
+        CommandLineToArgvW.restype = POINTER(LPWSTR)
+
+        cmd = GetCommandLineW()
+        argc = c_int(0)
+        argv = CommandLineToArgvW(cmd, byref(argc))
+        if argc.value > 0:
+            # Remove Python executable and commands if present
+            start = argc.value - len(sys.argv)
+            return [argv[i] for i in
+                    xrange(start, argc.value)]
+        # if we don't have any arguments at all, just pass back script name
+        # this should never happen
+        return [u"mobidedrm.py"]
+    else:
+        argvencoding = sys.stdin.encoding
+        if argvencoding == None:
+            argvencoding = "utf-8"
+        return [arg if (type(arg) == unicode) else unicode(arg,argvencoding) for arg in sys.argv]
 
 if sys.hexversion >= 0x3000000:
-    print "This script is incompatible with Python 3.x. Please install Python 2.6.x from python.org"
+    print 'This script is incompatible with Python 3.x. Please install Python 2.7.x.'
     sys.exit(2)
 
-letters = "ABCDEFGHIJKLMNPQRSTUVWXYZ123456789"
+letters = 'ABCDEFGHIJKLMNPQRSTUVWXYZ123456789'
 
 def crc32(s):
     return (~binascii.crc32(s,-1))&0xFFFFFFFF
@@ -53,39 +104,39 @@ def pidFromSerial(s, l):
     for i in xrange(l):
         arr1[i] ^= crc_bytes[i&3]
 
-    pid = ""
+    pid = ''
     for i in xrange(l):
         b = arr1[i] & 0xff
         pid+=letters[(b >> 7) + ((b >> 5 & 3) ^ (b & 0x1f))]
 
     return pid
 
-def main(argv=sys.argv):
-    print "Mobipocket PID calculator for Amazon Kindle. Copyright (c) 2007, 2009 Igor Skochinsky"
+def cli_main(argv=unicode_argv()):
+    print u"Mobipocket PID calculator for Amazon Kindle. Copyright © 2007, 2009 Igor Skochinsky"
     if len(sys.argv)==2:
         serial = sys.argv[1]
     else:
-        print "Usage: kindlepid.py <Kindle Serial Number>/<iPhone/iPod Touch UDID>"
+        print u"Usage: kindlepid.py <Kindle Serial Number>/<iPhone/iPod Touch UDID>"
         return 1
     if len(serial)==16:
         if serial.startswith("B"):
-            print "Kindle serial number detected"
+            print u"Kindle serial number detected"
         else:
-            print "Warning: unrecognized serial number. Please recheck input."
+            print u"Warning: unrecognized serial number. Please recheck input."
             return 1
-        pid = pidFromSerial(serial,7)+"*"
-        print "Mobipocket PID for Kindle serial# "+serial+" is "+checksumPid(pid)
+        pid = pidFromSerial(serial.encode("utf-8"),7)+'*'
+        print u"Mobipocket PID for Kindle serial#{0} is {1} ".format(serial,checksumPid(pid))
         return 0
     elif len(serial)==40:
-        print "iPhone serial number (UDID) detected"
-        pid = pidFromSerial(serial,8)
-        print "Mobipocket PID for iPhone serial# "+serial+" is "+checksumPid(pid)
+        print u"iPhone serial number (UDID) detected"
+        pid = pidFromSerial(serial.encode("utf-8"),8)
+        print u"Mobipocket PID for iPhone serial#{0} is {1} ".format(serial,checksumPid(pid))
         return 0
-    else:
-        print "Warning: unrecognized serial number. Please recheck input."
-        return 1
-    return 0
+    print u"Warning: unrecognized serial number. Please recheck input."
+     return 1
 
 
 if __name__ == "__main__":
-    sys.exit(main())
+    sys.stdout=SafeUnbuffered(sys.stdout)
+    sys.stderr=SafeUnbuffered(sys.stderr)
+    sys.exit(cli_main())