451 lines
21 KiB
Python
451 lines
21 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from __future__ import with_statement
|
||
|
|
__license__ = 'GPL v3'
|
||
|
|
__docformat__ = 'restructuredtext en'
|
||
|
|
|
||
|
|
|
||
|
|
# Released under the terms of the GNU General Public Licence, version 3
|
||
|
|
# <http://www.gnu.org/licenses/>
|
||
|
|
#
|
||
|
|
# Requires Calibre version 0.7.55 or higher.
|
||
|
|
#
|
||
|
|
# All credit given to i♥cabbages and The Dark Reverser for the original standalone scripts.
|
||
|
|
# We had the much easier job of converting them to a calibre plugin.
|
||
|
|
#
|
||
|
|
# This plugin is meant to decrypt eReader PDBs, Adobe Adept ePubs, Barnes & Noble ePubs,
|
||
|
|
# Adobe Adept PDFs, Amazon Kindle and Mobipocket files without having to
|
||
|
|
# install any dependencies... other than having calibre installed, of course.
|
||
|
|
#
|
||
|
|
# Configuration:
|
||
|
|
# Check out the plugin's configuration settings by clicking the "Customize plugin"
|
||
|
|
# button when you have the "DeDRM" plugin highlighted (under Preferences->
|
||
|
|
# Plugins->File type plugins). Once you have the configuration dialog open, you'll
|
||
|
|
# see a Help link on the top right-hand side.
|
||
|
|
#
|
||
|
|
# Revision history:
|
||
|
|
# 6.0.0 - Initial release
|
||
|
|
|
||
|
|
"""
|
||
|
|
Decrypt DRMed ebooks.
|
||
|
|
"""
|
||
|
|
|
||
|
|
PLUGIN_NAME = u"DeDRM"
|
||
|
|
PLUGIN_VERSION_TUPLE = (6, 0, 0)
|
||
|
|
PLUGIN_VERSION = u".".join([unicode(str(x)) for x in PLUGIN_VERSION_TUPLE])
|
||
|
|
# Include an html helpfile in the plugin's zipfile with the following name.
|
||
|
|
RESOURCE_NAME = PLUGIN_NAME + '_Help.htm'
|
||
|
|
|
||
|
|
import sys, os, re
|
||
|
|
import time
|
||
|
|
import zipfile
|
||
|
|
import traceback
|
||
|
|
from zipfile import ZipFile
|
||
|
|
|
||
|
|
class DeDRMError(Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
from calibre.customize import FileTypePlugin
|
||
|
|
from calibre.constants import iswindows, isosx
|
||
|
|
from calibre.gui2 import is_ok_to_use_qt
|
||
|
|
from calibre.utils.config import config_dir
|
||
|
|
|
||
|
|
|
||
|
|
# Wrap a stream so that output gets flushed immediately
|
||
|
|
# and also make sure that any unicode strings get safely
|
||
|
|
# 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)
|
||
|
|
|
||
|
|
class DeDRM(FileTypePlugin):
|
||
|
|
name = PLUGIN_NAME
|
||
|
|
description = u"Removes DRM from Amazon Kindle, Adobe Adept (including Kobo), Barnes & Noble, Mobipocket and eReader ebooks. Credit given to i♥cabbages and The Dark Reverser for the original stand-alone scripts."
|
||
|
|
supported_platforms = ['linux', 'osx', 'windows']
|
||
|
|
author = u"DiapDealer, Apprentice Alf, The Dark Reverser and i♥cabbages"
|
||
|
|
version = PLUGIN_VERSION_TUPLE
|
||
|
|
minimum_calibre_version = (0, 7, 55) # Compiled python libraries cannot be imported in earlier versions.
|
||
|
|
file_types = set(['epub','pdf','pdb','prc','mobi','azw','azw1','azw3','azw4','tpz'])
|
||
|
|
on_import = True
|
||
|
|
priority = 600
|
||
|
|
|
||
|
|
def initialize(self):
|
||
|
|
# convert old preferences, if necessary.
|
||
|
|
import calibre_plugins.dedrm.config
|
||
|
|
|
||
|
|
config.convertprefs()
|
||
|
|
|
||
|
|
"""
|
||
|
|
Dynamic modules can't be imported/loaded from a zipfile... so this routine
|
||
|
|
runs whenever the plugin gets initialized. This will extract the appropriate
|
||
|
|
library for the target OS and copy it to the 'alfcrypto' subdirectory of
|
||
|
|
calibre's configuration directory. That 'alfcrypto' directory is then
|
||
|
|
inserted into the syspath (as the very first entry) in the run function
|
||
|
|
so the CDLL stuff will work in the alfcrypto.py script.
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
if iswindows:
|
||
|
|
names = [u"alfcrypto.dll",u"alfcrypto64.dll"]
|
||
|
|
elif isosx:
|
||
|
|
names = [u"libalfcrypto.dylib"]
|
||
|
|
else:
|
||
|
|
names = [u"libalfcrypto32.so",u"libalfcrypto64.so"]
|
||
|
|
lib_dict = self.load_resources(names)
|
||
|
|
self.pluginsdir = os.path.join(config_dir,u"plugins")
|
||
|
|
if not os.path.exists(self.pluginsdir):
|
||
|
|
os.mkdir(self.pluginsdir)
|
||
|
|
self.maindir = os.path.join(self.pluginsdir,u"DeDRM")
|
||
|
|
if not os.path.exists(self.maindir):
|
||
|
|
os.mkdir(self.maindir)
|
||
|
|
self.helpdir = os.path.join(self.maindir,u"help")
|
||
|
|
if not os.path.exists(self.helpdir):
|
||
|
|
os.mkdir(self.helpdir)
|
||
|
|
self.alfdir = os.path.join(self.maindir,u"alfcrypto")
|
||
|
|
if not os.path.exists(self.alfdir):
|
||
|
|
os.mkdir(self.alfdir)
|
||
|
|
for entry, data in lib_dict.items():
|
||
|
|
file_path = os.path.join(self.alfdir, entry)
|
||
|
|
open(file_path,'wb').write(data)
|
||
|
|
except Exception, e:
|
||
|
|
traceback.print_exc()
|
||
|
|
raise
|
||
|
|
|
||
|
|
def ePubDecrypt(self,path_to_ebook):
|
||
|
|
# Create a TemporaryPersistent file to work with.
|
||
|
|
# Check original epub archive for zip errors.
|
||
|
|
import calibre_plugins.dedrm.zipfix
|
||
|
|
|
||
|
|
inf = self.temporary_file(u".epub")
|
||
|
|
try:
|
||
|
|
print u"{0} v{1}: Verifying zip archive integrity.".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
fr = zipfix.fixZip(path_to_ebook, inf.name)
|
||
|
|
fr.fix()
|
||
|
|
except Exception, e:
|
||
|
|
print u"{0} v{1}: Error \'{2}\' when checking zip archive.".format(PLUGIN_NAME, PLUGIN_VERSION, e.args[0])
|
||
|
|
raise Exception(e)
|
||
|
|
|
||
|
|
# import the decryption keys
|
||
|
|
import calibre_plugins.dedrm.config as config
|
||
|
|
|
||
|
|
# import the Barnes & Noble ePub handler
|
||
|
|
import calibre_plugins.dedrm.ignobleepub as ignobleepub
|
||
|
|
|
||
|
|
#check the book
|
||
|
|
if ignobleepub.ignobleBook(inf.name):
|
||
|
|
print u"{0} v{1}: “{2}” is a secure Barnes & Noble ePub.".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook))
|
||
|
|
|
||
|
|
# Attempt to decrypt epub with each encryption key (generated or provided).
|
||
|
|
for keyname, userkey in config.dedrmprefs['bandnkeys'].items():
|
||
|
|
keyname_masked = u"".join((u'X' if (x.isdigit()) else x) for x in keyname)
|
||
|
|
print u"{0} v{1}: Trying Encryption key {2:s}".format(PLUGIN_NAME, PLUGIN_VERSION, keyname_masked)
|
||
|
|
of = self.temporary_file(u".epub")
|
||
|
|
|
||
|
|
# Give the user key, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
result = ignobleepub.decryptBook(userkey, inf.name, of.name)
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
if result == 0:
|
||
|
|
# Decryption was successful.
|
||
|
|
# Return the modified PersistentTemporary file to calibre.
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
print u"{0} v{1}: Failed to decrypt with key {2:s} after {3:.1f} seconds.".format(PLUGIN_NAME, PLUGIN_VERSION,keyname_masked,time.time()-self.starttime)
|
||
|
|
|
||
|
|
print u"{0} v{1}: Ultimately failed to decrypt after {2:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
raise DeDRMError(u"{0} v{1}: Ultimately failed to decrypt “{2}” after {3:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook),time.time()-self.starttime))
|
||
|
|
|
||
|
|
# import the Adobe Adept ePub handler
|
||
|
|
import calibre_plugins.dedrm.ineptepub as ineptepub
|
||
|
|
|
||
|
|
if ineptepub.adeptBook(inf.name):
|
||
|
|
print u"{0} v{1}: {2} is a secure Adobe Adept ePub.".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook))
|
||
|
|
|
||
|
|
# Attempt to decrypt epub with each encryption key (generated or provided).
|
||
|
|
for keyname, userkeyhex in config.dedrmprefs['adeptkeys'].items():
|
||
|
|
userkey = userkeyhex.decode('hex')
|
||
|
|
print u"{0} v{1}: Trying Encryption key {2:s}".format(PLUGIN_NAME, PLUGIN_VERSION, keyname)
|
||
|
|
of = self.temporary_file(u".epub")
|
||
|
|
|
||
|
|
# Give the user key, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
try:
|
||
|
|
result = ineptepub.decryptBook(userkey, inf.name, of.name)
|
||
|
|
except:
|
||
|
|
result = 1
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
if result == 0:
|
||
|
|
# Decryption was successful.
|
||
|
|
# Return the modified PersistentTemporary file to calibre.
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
print u"{0} v{1}: Failed to decrypt with key {2:s} after {3:.1f} seconds.".format(PLUGIN_NAME, PLUGIN_VERSION,keyname,time.time()-self.starttime)
|
||
|
|
|
||
|
|
# perhaps we need to get a new default ADE key
|
||
|
|
if iswindows or isosx:
|
||
|
|
print u"{0} v{1}: Looking for new default Adobe Digital Editions Keys".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
|
||
|
|
# get the default Adobe keys
|
||
|
|
import calibre_plugins.dedrm.adobekey as adobe
|
||
|
|
|
||
|
|
try:
|
||
|
|
defaultkeys = adobe.adeptkeys()
|
||
|
|
except:
|
||
|
|
defaultkeys = []
|
||
|
|
|
||
|
|
newkeys = []
|
||
|
|
for keyvalue in defaultkeys:
|
||
|
|
if keyvalue.encode('hex') not in config.dedrmprefs['adeptkeys'].values():
|
||
|
|
newkeys.append(keyvalue)
|
||
|
|
|
||
|
|
if len(newkeys) > 0:
|
||
|
|
try:
|
||
|
|
for i,userkey in enumerate(newkeys):
|
||
|
|
print u"{0} v{1}: Trying a new default key.".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
of = self.temporary_file(u".epub")
|
||
|
|
|
||
|
|
# Give the user key, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
try:
|
||
|
|
result = ineptepub.decryptBook(userkey, inf.name, of.name)
|
||
|
|
except:
|
||
|
|
result = 1
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
if result == 0:
|
||
|
|
# Decryption was a success
|
||
|
|
# Store the new successful key in the defaults
|
||
|
|
print u"{0} v{1}: Saving a new default key.".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
try:
|
||
|
|
config.addnamedvaluetoprefs('adeptkeys','default_key',keyvalue.encode('hex'))
|
||
|
|
config.writeprefs()
|
||
|
|
except:
|
||
|
|
traceback.print_exc()
|
||
|
|
# Return the modified PersistentTemporary file to calibre.
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
print u"{0} v{1}: Failed to decrypt with new default key after {2:.1f} seconds.".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
except Exception, e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Something went wrong with decryption.
|
||
|
|
print u"{0} v{1}: Ultimately failed to decrypt after {2:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
raise DeDRMError(u"{0} v{1}: Ultimately failed to decrypt “{2}” after {3:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook),time.time()-self.starttime))
|
||
|
|
|
||
|
|
# Not a Barnes & Noble nor an Adobe Adept
|
||
|
|
# Import the fixed epub.
|
||
|
|
print u"{0} v{1}: “{2}” is neither an Adobe Adept nor a Barnes & Noble encrypted ePub.".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook))
|
||
|
|
return inf.name
|
||
|
|
|
||
|
|
def PDFDecrypt(self,path_to_ebook):
|
||
|
|
import calibre_plugins.dedrm.config as config
|
||
|
|
import calibre_plugins.dedrm.ineptpdf
|
||
|
|
|
||
|
|
# Attempt to decrypt epub with each encryption key (generated or provided).
|
||
|
|
print u"{0} v{1}: {2} is a PDF ebook.".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook))
|
||
|
|
for keyname, userkeyhex in config.dedrmprefs['adeptkeys'].items():
|
||
|
|
userkey = userkeyhex.decode('hex')
|
||
|
|
print u"{0} v{1}: Trying Encryption key {2:s}".format(PLUGIN_NAME, PLUGIN_VERSION, keyname)
|
||
|
|
of = self.temporary_file(u".pdf")
|
||
|
|
|
||
|
|
# Give the user key, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
try:
|
||
|
|
result = ineptpdf.decryptBook(userkey, path_to_ebook, of.name)
|
||
|
|
except:
|
||
|
|
result = 1
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
if result == 0:
|
||
|
|
# Decryption was successful.
|
||
|
|
# Return the modified PersistentTemporary file to calibre.
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
# perhaps we need to get a new default ADE key
|
||
|
|
if iswindows or isosx:
|
||
|
|
print u"{0} v{1}: Looking for new default Adobe Digital Editions Keys".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
|
||
|
|
# get the default Adobe keys
|
||
|
|
import calibre_plugins.dedrm.adobekey as adobe
|
||
|
|
|
||
|
|
try:
|
||
|
|
defaultkeys = adobe.adeptkeys()
|
||
|
|
except:
|
||
|
|
defaultkeys = []
|
||
|
|
|
||
|
|
newkeys = []
|
||
|
|
for keyvalue in defaultkeys:
|
||
|
|
if keyvalue.encode('hex') not in config.dedrmprefs['adeptkeys'].values():
|
||
|
|
newkeys.append(keyvalue)
|
||
|
|
|
||
|
|
if len(newkeys) > 0:
|
||
|
|
try:
|
||
|
|
for i,userkey in enumerate(newkeys):
|
||
|
|
print u"{0} v{1}: Trying a new default key.".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
of = self.temporary_file(u".pdf")
|
||
|
|
|
||
|
|
# Give the user key, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
try:
|
||
|
|
result = ineptepdf.decryptBook(userkey, inf.name, of.name)
|
||
|
|
except:
|
||
|
|
result = 1
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
if result == 0:
|
||
|
|
# Decryption was a success
|
||
|
|
# Store the new successful key in the defaults
|
||
|
|
print u"{0} v{1}: Saving a new default key.".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
try:
|
||
|
|
config.addnamedvaluetoprefs('adeptkeys','default_key',keyvalue.encode('hex'))
|
||
|
|
config.writeprefs()
|
||
|
|
except:
|
||
|
|
traceback.print_exc()
|
||
|
|
# Return the modified PersistentTemporary file to calibre.
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
print u"{0} v{1}: Failed to decrypt with new default key after {2:.1f} seconds.".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
except Exception, e:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Something went wrong with decryption.
|
||
|
|
print u"{0} v{1}: Ultimately failed to decrypt after {2:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
raise DeDRMError(u"{0} v{1}: Ultimately failed to decrypt “{2}” after {3:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook),time.time()-self.starttime))
|
||
|
|
|
||
|
|
def KindleMobiDecrypt(self,path_to_ebook):
|
||
|
|
|
||
|
|
# add the alfcrypto directory to sys.path so alfcrypto.py
|
||
|
|
# will be able to locate the custom lib(s) for CDLL import.
|
||
|
|
sys.path.insert(0, self.alfdir)
|
||
|
|
# Had to move this import here so the custom libs can be
|
||
|
|
# extracted to the appropriate places beforehand these routines
|
||
|
|
# look for them.
|
||
|
|
import calibre_plugins.dedrm.config as config
|
||
|
|
import calibre_plugins.dedrm.k4mobidedrm
|
||
|
|
|
||
|
|
pids = config.dedrmprefs['pids']
|
||
|
|
serials = config.dedrmprefs['serials']
|
||
|
|
kindleDatabases = config.dedrmprefs['kindlekeys'].items()
|
||
|
|
|
||
|
|
try:
|
||
|
|
book = k4mobidedrm.GetDecryptedBook(path_to_ebook,kindleDatabases,serials,pids,self.starttime)
|
||
|
|
except Exception, e:
|
||
|
|
decoded = False
|
||
|
|
# perhaps we need to get a new default Kindle for Mac/PC key
|
||
|
|
if iswindows or isosx:
|
||
|
|
print u"{0} v{1}: Failed to decrypt with error: {2}".format(PLUGIN_NAME, PLUGIN_VERSION,e.args[0])
|
||
|
|
print u"{0} v{1}: Looking for new default Kindle Key".format(PLUGIN_NAME, PLUGIN_VERSION)
|
||
|
|
import calibre_plugins.dedrm.kindlekey as amazon
|
||
|
|
|
||
|
|
try:
|
||
|
|
defaultkeys = amazon.kindlekeys()
|
||
|
|
except:
|
||
|
|
defaultkeys = []
|
||
|
|
newkeys = {}
|
||
|
|
for i,keyvalue in enumerate(defaultkeys):
|
||
|
|
keyname = u"default_key_{0:d}".format(i+1)
|
||
|
|
if keyvalue not in config.dedrmprefs['kindlekeys'].values():
|
||
|
|
newkeys[keyname] = keyvalue
|
||
|
|
if len(newkeys) > 0:
|
||
|
|
try:
|
||
|
|
book = k4mobidedrm.GetDecryptedBook(path_to_ebook,newkeys.items(),[],[],self.starttime)
|
||
|
|
decoded = True
|
||
|
|
# store the new successful keys in the defaults
|
||
|
|
for keyvalue in newkeys.values():
|
||
|
|
config.addnamedvaluetoprefs('kindlekeys','default_key',keyvalue)
|
||
|
|
config.writeprefs()
|
||
|
|
except Exception, e:
|
||
|
|
pass
|
||
|
|
if not decoded:
|
||
|
|
#if you reached here then no luck raise and exception
|
||
|
|
print u"{0} v{1}: Ultimately failed to decrypt after {2:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
traceback.print_exc()
|
||
|
|
raise DeDRMError(u"{0} v{1}: Ultimately failed to decrypt “{4}” after {3:.1f} seconds with error: {2}\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION, e.args[0],time.time()-self.starttime,os.path.basename(path_to_ebook)))
|
||
|
|
|
||
|
|
of = self.temporary_file(book.getBookExtension())
|
||
|
|
book.getFile(of.name)
|
||
|
|
of.close()
|
||
|
|
book.cleanup()
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
|
||
|
|
def eReaderDecrypt(self,path_to_ebook):
|
||
|
|
|
||
|
|
import calibre_plugins.dedrm.config as config
|
||
|
|
import calibre_plugins.dedrm.erdr2pml
|
||
|
|
|
||
|
|
# Attempt to decrypt epub with each encryption key (generated or provided).
|
||
|
|
for keyname, userkey in config.dedrmprefs['ereaderkeys'].items():
|
||
|
|
keyname_masked = u"".join((u'X' if (x.isdigit()) else x) for x in keyname)
|
||
|
|
print u"{0} v{1}: Trying Encryption key {2:s}".format(PLUGIN_NAME, PLUGIN_VERSION, keyname_masked)
|
||
|
|
of = self.temporary_file(u".pmlz")
|
||
|
|
|
||
|
|
# Give the userkey, ebook and TemporaryPersistent file to the decryption function.
|
||
|
|
result = erdr2pml.decryptBook(path_to_ebook, of.name, True, userkey.decode('hex'))
|
||
|
|
|
||
|
|
of.close()
|
||
|
|
|
||
|
|
# Decryption was successful return the modified PersistentTemporary
|
||
|
|
# file to Calibre's import process.
|
||
|
|
if result == 0:
|
||
|
|
return of.name
|
||
|
|
|
||
|
|
print u"{0} v{1}: Failed to decrypt with key {2:s} after {3:.1f} seconds.".format(PLUGIN_NAME, PLUGIN_VERSION,keyname_masked,time.time()-self.starttime)
|
||
|
|
|
||
|
|
print u"{0} v{1}: Ultimately failed to decrypt after {2:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
raise DeDRMError(u"{0} v{1}: Ultimately failed to decrypt “{2}” after {3:.1f} seconds.\nRead the FAQs at Alf's blog: http://apprenticealf.wordpress.com/".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook),time.time()-self.starttime))
|
||
|
|
|
||
|
|
|
||
|
|
def run(self, path_to_ebook):
|
||
|
|
|
||
|
|
# make sure any unicode output gets converted safely with 'replace'
|
||
|
|
sys.stdout=SafeUnbuffered(sys.stdout)
|
||
|
|
sys.stderr=SafeUnbuffered(sys.stderr)
|
||
|
|
|
||
|
|
print u"{0} v{1}: Trying to decrypt {2}.".format(PLUGIN_NAME, PLUGIN_VERSION, os.path.basename(path_to_ebook))
|
||
|
|
self.starttime = time.time()
|
||
|
|
|
||
|
|
booktype = os.path.splitext(path_to_ebook)[1].lower()[1:]
|
||
|
|
if booktype in ['prc','mobi','azw','azw1','azw3','azw4','tpz']:
|
||
|
|
# Kindle/Mobipocket
|
||
|
|
decrypted_ebook = self.KindleMobiDecrypt(path_to_ebook)
|
||
|
|
elif booktype == 'pdb':
|
||
|
|
# eReader
|
||
|
|
decrypted_ebook = self.eReaderDecrypt(path_to_ebook)
|
||
|
|
pass
|
||
|
|
elif booktype == 'pdf':
|
||
|
|
# Adobe Adept PDF (hopefully)
|
||
|
|
decrypted_ebook = self.PDFDecrypt(path_to_ebook)
|
||
|
|
pass
|
||
|
|
elif booktype == 'epub':
|
||
|
|
# Adobe Adept or B&N ePub
|
||
|
|
decrypted_ebook = self.ePubDecrypt(path_to_ebook)
|
||
|
|
else:
|
||
|
|
print u"Unknown booktype {0}. Passing back to calibre unchanged.".format(booktype)
|
||
|
|
return path_to_ebook
|
||
|
|
print u"{0} v{1}: Successfully decrypted book after {2:.1f} seconds".format(PLUGIN_NAME, PLUGIN_VERSION,time.time()-self.starttime)
|
||
|
|
return decrypted_ebook
|
||
|
|
|
||
|
|
def is_customizable(self):
|
||
|
|
# return true to allow customization via the Plugin->Preferences.
|
||
|
|
return True
|
||
|
|
|
||
|
|
def config_widget(self):
|
||
|
|
import calibre_plugins.dedrm.config as config
|
||
|
|
return config.ConfigWidget(self.plugin_path)
|
||
|
|
|
||
|
|
def save_settings(self, config_widget):
|
||
|
|
config_widget.save_settings()
|