Merge ad33aea18d into 3373d93874

DeDRM_plugin/erdr2pml.py

@@ -255,7 +255,7 @@ class EreaderProcessor(object):
             encrypted_key = r[172:172+8]
             encrypted_key_sha = r[56:56+20]
         self.content_key = des.decrypt(encrypted_key)
-        if hashlib.sha1(self.content_key).digest() != encrypted_key_sha:
+        if sha1(self.content_key).digest() != encrypted_key_sha:
             raise ValueError('Incorrect Name and/or Credit Card')
 
     def getNumImages(self):

DeDRM_plugin/ion.py

@@ -1,19 +1,25 @@
 #!/usr/bin/env python3
-"""ion.py: Decrypt Kindle KFX files.
+# -*- coding: utf-8 -*-
 
-Revision history:
-  Pascal implementation by lulzkabulz.
-  BinaryIon.pas + DrmIon.pas + IonSymbols.pas
-  1.0   - Python translation by apprenticenaomi.
-  1.1   - DeDRM integration by anon.
-  1.2   - Added pylzma import fallback
-  1.3   - Fixed lzma support for calibre 4.6+
-  2.0   - VoucherEnvelope v2/v3 support by apprenticesakuya.
-  3.0   - Added Python 3 compatibility for calibre 5.0
+# ion.py
+# Copyright © 2013-2020 Apprentice Harper et al.
+
+__license__ = 'GPL v3'
+__version__ = '3.0'
+
+# Revision history:
+#  Pascal implementation by lulzkabulz.
+#  BinaryIon.pas + DrmIon.pas + IonSymbols.pas
+#  1.0   - Python translation by apprenticenaomi.
+#  1.1   - DeDRM integration by anon.
+#  1.2   - Added pylzma import fallback
+#  1.3   - Fixed lzma support for calibre 4.6+
+#  2.0   - VoucherEnvelope v2/v3 support by apprenticesakuya.
+#  3.0   - Added Python 3 compatibility for calibre 5.0
 
-Copyright © 2013-2020 Apprentice Harper et al.
 """
-from __future__ import annotations
+Decrypt Kindle KFX files.
+"""
 
 import collections
 import hashlib
@@ -24,9 +30,6 @@ import struct
 
 from io import BytesIO
 
-__license__ = 'GPL v3'
-__version__ = '3.0'
-
 #@@CALIBRE_COMPAT_CODE@@
 
 
@@ -1093,11 +1096,11 @@ def process_V9888(st):
     ws.sbox(d0x6a0bf4d0,d0x6a0dab50,[1,2])
     ws.sbox(d0x6a0ba4d0,d0x6a0dab50,[1,2])
     ws.shuffle(repl)
-    ws.shuffle(repl)
+    ws.shuffle(repl)    
     ws.shuffle(repl)
     ws.sbox(d0x6a0bf4d0,d0x6a0dab50,[1,2])
     ws.sbox(d0x6a0ba4d0,d0x6a0dab50,[1,2])
-    ws.exlookup(d0x6a0be4d0)
+    ws.exlookup(d0x6a0be4d0)    
     dat=ws.mask(st[sto:sto+16])
     out+=dat
     sto+=16
@@ -1121,7 +1124,7 @@ def process_V4648(st):
     ws.sbox(d0x6a0c51a8,d0x6a0dab50,[1,3])
     ws.shuffle(repl)
     ws.shuffle(repl)
-    ws.exlookup(d0x6a0c91a8)
+    ws.exlookup(d0x6a0c91a8)    
     dat=ws.mask(st[sto:sto+16])
     out+=dat
     sto+=16
@@ -1145,7 +1148,7 @@ def process_V5683(st):
     ws.shuffle(repl)
     ws.sbox(d0x6a0cfe80,d0x6a0dab50,[])
     ws.shuffle(repl)
-    ws.exlookup(d0x6a0d3e80)
+    ws.exlookup(d0x6a0d3e80)    
     dat=ws.mask(st[sto:sto+16])
     out+=dat
     sto+=16
@@ -1160,12 +1163,12 @@ def process_V5683(st):
 #     if a<0: a=256+a
 #     ax.append(ha[(a>>4)]+ha[a%16])
 #   return "".join(ax)
-#
+# 
 # def memhex(adr,sz):
 #   emu=EmulatorHelper(currentProgram)
 #   arr=emu.readMemory(getAddress(adr),sz)
 #   return a2hex(arr)
-#
+# 
 
 
 
@@ -1251,7 +1254,7 @@ def scramble3(st,magic):
             iVar5 = iVar5 - magic
             index -= 1
             if uVar4<=-1: break
-      else:
+      else: 
         if (offset < magic):
           iVar3 = 0
         else :
@@ -1271,9 +1274,9 @@ def scramble3(st,magic):
             index=index-1
             iVar5 = iVar5 + magic;
             cntr += 1;
-            if iVar3>=divs: break
+            if iVar3>=divs: break 
       offset = offset + 1
-      if offset >= ((magic - 1) + divs) :break
+      if offset >= ((magic - 1) + divs) :break 
   return ret
 
 #not sure if the third variant is used anywhere, but it is in Kindle, so I tried to add it
@@ -1339,14 +1342,14 @@ class DrmIonVoucher(object):
                 _assert(False, "Unknown lock parameter: %s" % param)
 
 
-        # i know that version maps to scramble pretty much 1 to 1, but there was precendent where they changed it, so...
+        # i know that version maps to scramble pretty much 1 to 1, but there was precendent where they changed it, so... 
         sharedsecrets = [obfuscate(shared, self.version),obfuscate2(shared, self.version),obfuscate3(shared, self.version),
                          process_V9708(shared), process_V1031(shared), process_V2069(shared), process_V9041(shared),
                          process_V3646(shared), process_V6052(shared), process_V9479(shared), process_V9888(shared),
                          process_V4648(shared), process_V5683(shared)]
-
+        
         decrypted=False
-        lastexception: Exception | None = None
+        ex=None
         for sharedsecret in sharedsecrets:
             key = hmac.new(sharedsecret, b"PIDv3", digestmod=hashlib.sha256).digest()
             aes = AES.new(key[:32], AES.MODE_CBC, self.cipheriv[:16])
@@ -1358,15 +1361,14 @@ class DrmIonVoucher(object):
 
                 _assert(self.drmkey.hasnext() and self.drmkey.next() == TID_LIST and self.drmkey.gettypename() == "com.amazon.drm.KeySet@1.0",
                     "Expected KeySet, got %s" % self.drmkey.gettypename())
-                decrypted=True
+                decrypted=True 
 
                 print("Decryption succeeded")
                 break
             except Exception as ex:
-                lastexception = ex
                 print("Decryption failed, trying next fallback ")
         if not decrypted:
-            raise lastexception
+            raise ex
 
         self.drmkey.stepin()
         while self.drmkey.hasnext():

Obok_plugin/action.py

@@ -374,11 +374,7 @@ class InterfacePluginAction(InterfaceAction):
         result['success'] = False
         result['fileobj'] = None
 
-        try:
-            zin = zipfile.ZipFile(book.filename, 'r')
-        except FileNotFoundError:
-            print (_('{0} - File "{1}" not found. Make sure the eBook has been properly downloaded in the Kobo app.').format(PLUGIN_NAME, book.filename))
-            return result
+        zin = zipfile.ZipFile(book.filename, 'r')
         #print ('Kobo library filename: {0}'.format(book.filename))
         for userkey in self.userkeys:
             print (_('Trying key: '), codecs.encode(userkey, 'hex'))

Obok_plugin/obok/obok.py

@@ -463,15 +463,9 @@ class KoboLibrary(object):
             for m in matches:
                 # print "m:{0}".format(m[0])
                 macaddrs.append(m[0].upper())
-        elif sys.platform.startswith('linux'):
-            for interface in os.listdir('/sys/class/net'):
-                with open('/sys/class/net/' + interface + '/address', 'r') as f:
-                    mac = f.read().strip().upper()
-                # some interfaces, like Tailscale's VPN interface, do not have a MAC address
-                if mac != '':
-                    macaddrs.append(mac)
         else:
-            # final fallback
+            # probably linux
+
             # let's try ip
             c = re.compile('\s(' + '[0-9a-f]{2}:' * 5 + '[0-9a-f]{2})(\s|$)', re.IGNORECASE)
             for line in os.popen('ip -br link'):