jacob/DeDRM_tools
1
0
Fork 0
mirror of https://github.com/noDRM/DeDRM_tools.git synced 2025-10-23 23:07:47 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

test file type correctly

Browse source
This commit is contained in:
Apprentice Harper 2020-10-16 13:22:19 +01:00
parent 7262264b95
commit dc27c36761
1 changed files with 13 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
DeDRM_plugin/k4mobidedrm.py
View file

@ -87,11 +87,11 @@ if inCalibre:
from calibre_plugins.dedrm import androidkindlekey from calibre_plugins.dedrm import androidkindlekey
from calibre_plugins.dedrm import kfxdedrm from calibre_plugins.dedrm import kfxdedrm
else: else:
from . import mobidedrm import mobidedrm
from . import topazextract import topazextract
from . import kgenpids import kgenpids
from . import androidkindlekey import androidkindlekey
from . import kfxdedrm import kfxdedrm
# Wrap a stream so that output gets flushed immediately # Wrap a stream so that output gets flushed immediately
# and also make sure that any unicode strings get # and also make sure that any unicode strings get
@ -204,14 +204,14 @@ def GetDecryptedBook(infile, kDatabases, androidFiles, serials, pids, starttime
mobi = True mobi = True
magic8 = open(infile,'rb').read(8) magic8 = open(infile,'rb').read(8)
if magic8 == '\xeaDRMION\xee': if magic8 == b'\xeaDRMION\xee':
raise DrmException("The .kfx DRMION file cannot be decrypted by itself. A .kfx-zip archive containing a DRM voucher is required.") raise DrmException("The .kfx DRMION file cannot be decrypted by itself. A .kfx-zip archive containing a DRM voucher is required.")
magic3 = magic8[:3] magic3 = magic8[:3]
if magic3 == 'TPZ': if magic3 == b'TPZ':
mobi = False mobi = False
if magic8[:4] == 'PK\x03\x04': if magic8[:4] == b'PK\x03\x04':
mb = kfxdedrm.KFXZipBook(infile) mb = kfxdedrm.KFXZipBook(infile)
elif mobi: elif mobi:
mb = mobidedrm.MobiBook(infile) mb = mobidedrm.MobiBook(infile)
@ -309,10 +309,10 @@ def usage(progname):
def cli_main(): def cli_main():
argv=unicode_argv() argv=unicode_argv()
progname = os.path.basename(argv[0]) progname = os.path.basename(argv[0])
print("K4MobiDeDrm v{0}.\nCopyright © 2008-2017 Apprentice Harper et al.".format(__version__)) print("K4MobiDeDrm v{0}.\nCopyright © 2008-2020 Apprentice Harper et al.".format(__version__))
try: try:
opts, args = getopt.getopt(argv[1:], "k:p:s:a:") opts, args = getopt.getopt(argv[1:], "k:p:s:a:h")
except getopt.GetoptError as err: except getopt.GetoptError as err:
print("Error in options or arguments: {0}".format(err.args[0])) print("Error in options or arguments: {0}".format(err.args[0]))
usage(progname) usage(progname)
@ -329,6 +329,9 @@ def cli_main():
pids = [] pids = []
for o, a in opts: for o, a in opts:
if o == "-h":
usage(progname)
sys.exit(0)
if o == "-k": if o == "-k":
if a == None : if a == None :
raise DrmException("Invalid parameter for -k") raise DrmException("Invalid parameter for -k")
@ -346,9 +349,6 @@ def cli_main():
raise DrmException("Invalid parameter for -a") raise DrmException("Invalid parameter for -a")
androidFiles.append(a) androidFiles.append(a)
# try with built in Kindle Info files if not on Linux
k4 = not sys.platform.startswith('linux')
return decryptBook(infile, outdir, kDatabaseFiles, androidFiles, serials, pids) return decryptBook(infile, outdir, kDatabaseFiles, androidFiles, serials, pids)