mirror of
https://github.com/noDRM/DeDRM_tools.git
synced 2024-08-25 02:26:10 +06:00
320 lines
10 KiB
Python
320 lines
10 KiB
Python
# standlone set of Mac OSX specific routines needed for K4DeDRM
|
|
|
|
from __future__ import with_statement
|
|
|
|
import sys
|
|
import os
|
|
|
|
#Exception Handling
|
|
class K4MDrmException(Exception):
|
|
pass
|
|
|
|
import signal
|
|
import threading
|
|
import subprocess
|
|
from subprocess import Popen, PIPE, STDOUT
|
|
|
|
# **heavily** chopped up and modfied version of asyncproc.py
|
|
# to make it actually work on Windows as well as Mac/Linux
|
|
# For the original see:
|
|
# "http://www.lysator.liu.se/~bellman/download/"
|
|
# author is "Thomas Bellman <bellman@lysator.liu.se>"
|
|
# available under GPL version 3 or Later
|
|
|
|
# create an asynchronous subprocess whose output can be collected in
|
|
# a non-blocking manner
|
|
|
|
# What a mess! Have to use threads just to get non-blocking io
|
|
# in a cross-platform manner
|
|
|
|
# luckily all thread use is hidden within this class
|
|
|
|
class Process(object):
|
|
def __init__(self, *params, **kwparams):
|
|
if len(params) <= 3:
|
|
kwparams.setdefault('stdin', subprocess.PIPE)
|
|
if len(params) <= 4:
|
|
kwparams.setdefault('stdout', subprocess.PIPE)
|
|
if len(params) <= 5:
|
|
kwparams.setdefault('stderr', subprocess.PIPE)
|
|
self.__pending_input = []
|
|
self.__collected_outdata = []
|
|
self.__collected_errdata = []
|
|
self.__exitstatus = None
|
|
self.__lock = threading.Lock()
|
|
self.__inputsem = threading.Semaphore(0)
|
|
self.__quit = False
|
|
|
|
self.__process = subprocess.Popen(*params, **kwparams)
|
|
|
|
if self.__process.stdin:
|
|
self.__stdin_thread = threading.Thread(
|
|
name="stdin-thread",
|
|
target=self.__feeder, args=(self.__pending_input,
|
|
self.__process.stdin))
|
|
self.__stdin_thread.setDaemon(True)
|
|
self.__stdin_thread.start()
|
|
|
|
if self.__process.stdout:
|
|
self.__stdout_thread = threading.Thread(
|
|
name="stdout-thread",
|
|
target=self.__reader, args=(self.__collected_outdata,
|
|
self.__process.stdout))
|
|
self.__stdout_thread.setDaemon(True)
|
|
self.__stdout_thread.start()
|
|
|
|
if self.__process.stderr:
|
|
self.__stderr_thread = threading.Thread(
|
|
name="stderr-thread",
|
|
target=self.__reader, args=(self.__collected_errdata,
|
|
self.__process.stderr))
|
|
self.__stderr_thread.setDaemon(True)
|
|
self.__stderr_thread.start()
|
|
|
|
def pid(self):
|
|
return self.__process.pid
|
|
|
|
def kill(self, signal):
|
|
self.__process.send_signal(signal)
|
|
|
|
# check on subprocess (pass in 'nowait') to act like poll
|
|
def wait(self, flag):
|
|
if flag.lower() == 'nowait':
|
|
rc = self.__process.poll()
|
|
else:
|
|
rc = self.__process.wait()
|
|
if rc != None:
|
|
if self.__process.stdin:
|
|
self.closeinput()
|
|
if self.__process.stdout:
|
|
self.__stdout_thread.join()
|
|
if self.__process.stderr:
|
|
self.__stderr_thread.join()
|
|
return self.__process.returncode
|
|
|
|
def terminate(self):
|
|
if self.__process.stdin:
|
|
self.closeinput()
|
|
self.__process.terminate()
|
|
|
|
# thread gets data from subprocess stdout
|
|
def __reader(self, collector, source):
|
|
while True:
|
|
data = os.read(source.fileno(), 65536)
|
|
self.__lock.acquire()
|
|
collector.append(data)
|
|
self.__lock.release()
|
|
if data == "":
|
|
source.close()
|
|
break
|
|
return
|
|
|
|
# thread feeds data to subprocess stdin
|
|
def __feeder(self, pending, drain):
|
|
while True:
|
|
self.__inputsem.acquire()
|
|
self.__lock.acquire()
|
|
if not pending and self.__quit:
|
|
drain.close()
|
|
self.__lock.release()
|
|
break
|
|
data = pending.pop(0)
|
|
self.__lock.release()
|
|
drain.write(data)
|
|
|
|
# non-blocking read of data from subprocess stdout
|
|
def read(self):
|
|
self.__lock.acquire()
|
|
outdata = "".join(self.__collected_outdata)
|
|
del self.__collected_outdata[:]
|
|
self.__lock.release()
|
|
return outdata
|
|
|
|
# non-blocking read of data from subprocess stderr
|
|
def readerr(self):
|
|
self.__lock.acquire()
|
|
errdata = "".join(self.__collected_errdata)
|
|
del self.__collected_errdata[:]
|
|
self.__lock.release()
|
|
return errdata
|
|
|
|
# non-blocking write to stdin of subprocess
|
|
def write(self, data):
|
|
if self.__process.stdin is None:
|
|
raise ValueError("Writing to process with stdin not a pipe")
|
|
self.__lock.acquire()
|
|
self.__pending_input.append(data)
|
|
self.__inputsem.release()
|
|
self.__lock.release()
|
|
|
|
# close stdinput of subprocess
|
|
def closeinput(self):
|
|
self.__lock.acquire()
|
|
self.__quit = True
|
|
self.__inputsem.release()
|
|
self.__lock.release()
|
|
|
|
|
|
# interface to needed routines in openssl's libcrypto
|
|
def _load_crypto_libcrypto():
|
|
from ctypes import CDLL, byref, POINTER, c_void_p, c_char_p, c_int, c_long, \
|
|
Structure, c_ulong, create_string_buffer, addressof, string_at, cast
|
|
from ctypes.util import find_library
|
|
|
|
libcrypto = find_library('crypto')
|
|
if libcrypto is None:
|
|
raise K4MDrmException('libcrypto not found')
|
|
libcrypto = CDLL(libcrypto)
|
|
|
|
AES_MAXNR = 14
|
|
c_char_pp = POINTER(c_char_p)
|
|
c_int_p = POINTER(c_int)
|
|
|
|
class AES_KEY(Structure):
|
|
_fields_ = [('rd_key', c_long * (4 * (AES_MAXNR + 1))), ('rounds', c_int)]
|
|
AES_KEY_p = POINTER(AES_KEY)
|
|
|
|
def F(restype, name, argtypes):
|
|
func = getattr(libcrypto, name)
|
|
func.restype = restype
|
|
func.argtypes = argtypes
|
|
return func
|
|
|
|
AES_cbc_encrypt = F(None, 'AES_cbc_encrypt',[c_char_p, c_char_p, c_ulong, AES_KEY_p, c_char_p,c_int])
|
|
|
|
AES_set_decrypt_key = F(c_int, 'AES_set_decrypt_key',[c_char_p, c_int, AES_KEY_p])
|
|
|
|
PKCS5_PBKDF2_HMAC_SHA1 = F(c_int, 'PKCS5_PBKDF2_HMAC_SHA1',
|
|
[c_char_p, c_ulong, c_char_p, c_ulong, c_ulong, c_ulong, c_char_p])
|
|
|
|
class LibCrypto(object):
|
|
def __init__(self):
|
|
self._blocksize = 0
|
|
self._keyctx = None
|
|
self.iv = 0
|
|
|
|
def set_decrypt_key(self, userkey, iv):
|
|
self._blocksize = len(userkey)
|
|
if (self._blocksize != 16) and (self._blocksize != 24) and (self._blocksize != 32) :
|
|
raise K4MDrmException('AES improper key used')
|
|
return
|
|
keyctx = self._keyctx = AES_KEY()
|
|
self.iv = iv
|
|
rv = AES_set_decrypt_key(userkey, len(userkey) * 8, keyctx)
|
|
if rv < 0:
|
|
raise K4MDrmException('Failed to initialize AES key')
|
|
|
|
def decrypt(self, data):
|
|
out = create_string_buffer(len(data))
|
|
rv = AES_cbc_encrypt(data, out, len(data), self._keyctx, self.iv, 0)
|
|
if rv == 0:
|
|
raise K4MDrmException('AES decryption failed')
|
|
return out.raw
|
|
|
|
def keyivgen(self, passwd):
|
|
salt = '16743'
|
|
saltlen = 5
|
|
passlen = len(passwd)
|
|
iter = 0x3e8
|
|
keylen = 80
|
|
out = create_string_buffer(keylen)
|
|
rv = PKCS5_PBKDF2_HMAC_SHA1(passwd, passlen, salt, saltlen, iter, keylen, out)
|
|
return out.raw
|
|
return LibCrypto
|
|
|
|
def _load_crypto():
|
|
LibCrypto = None
|
|
try:
|
|
LibCrypto = _load_crypto_libcrypto()
|
|
except (ImportError, K4MDrmException):
|
|
pass
|
|
return LibCrypto
|
|
|
|
LibCrypto = _load_crypto()
|
|
|
|
#
|
|
# Utility Routines
|
|
#
|
|
|
|
# uses a sub process to get the Hard Drive Serial Number using ioreg
|
|
# returns with the first found serial number in that class
|
|
def GetVolumeSerialNumber():
|
|
cmdline = '/usr/sbin/ioreg -r -c AppleAHCIDiskDriver'
|
|
cmdline = cmdline.encode(sys.getfilesystemencoding())
|
|
p = Process(cmdline, shell=True, bufsize=1, stdin=None, stdout=PIPE, stderr=PIPE, close_fds=False)
|
|
poll = p.wait('wait')
|
|
results = p.read()
|
|
reslst = results.split('\n')
|
|
sernum = '9999999999'
|
|
cnt = len(reslst)
|
|
for j in xrange(cnt):
|
|
resline = reslst[j]
|
|
pp = resline.find('"Serial Number" = "')
|
|
if pp >= 0:
|
|
sernum = resline[pp+19:]
|
|
sernum = sernum[:-1]
|
|
sernum = sernum.lstrip()
|
|
break
|
|
return sernum
|
|
|
|
# uses unix env to get username instead of using sysctlbyname
|
|
def GetUserName():
|
|
username = os.getenv('USER')
|
|
return username
|
|
|
|
# Various character maps used to decrypt books. Probably supposed to act as obfuscation
|
|
charMap1 = "n5Pr6St7Uv8Wx9YzAb0Cd1Ef2Gh3Jk4M"
|
|
charMap2 = "ZB0bYyc1xDdW2wEV3Ff7KkPpL8UuGA4gz-Tme9Nn_tHh5SvXCsIiR6rJjQaqlOoM"
|
|
charMap3 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
|
charMap4 = "ABCDEFGHIJKLMNPQRSTUVWXYZ123456789"
|
|
|
|
def encode(data, map):
|
|
result = ""
|
|
for char in data:
|
|
value = ord(char)
|
|
Q = (value ^ 0x80) // len(map)
|
|
R = value % len(map)
|
|
result += map[Q]
|
|
result += map[R]
|
|
return result
|
|
|
|
import hashlib
|
|
|
|
def SHA256(message):
|
|
ctx = hashlib.sha256()
|
|
ctx.update(message)
|
|
return ctx.digest()
|
|
|
|
# implements an Pseudo Mac Version of Windows built-in Crypto routine
|
|
def CryptUnprotectData(encryptedData):
|
|
sp = GetVolumeSerialNumber() + '!@#' + GetUserName()
|
|
passwdData = encode(SHA256(sp),charMap1)
|
|
crp = LibCrypto()
|
|
key_iv = crp.keyivgen(passwdData)
|
|
key = key_iv[0:32]
|
|
iv = key_iv[32:48]
|
|
crp.set_decrypt_key(key,iv)
|
|
cleartext = crp.decrypt(encryptedData)
|
|
return cleartext
|
|
|
|
# Locate and open the .kindle-info file
|
|
def openKindleInfo():
|
|
home = os.getenv('HOME')
|
|
cmdline = 'find "' + home + '/Library/Application Support" -name ".kindle-info"'
|
|
cmdline = cmdline.encode(sys.getfilesystemencoding())
|
|
p1 = Process(cmdline, shell=True, bufsize=1, stdin=None, stdout=PIPE, stderr=PIPE, close_fds=False)
|
|
poll = p1.wait('wait')
|
|
results = p1.read()
|
|
reslst = results.split('\n')
|
|
kinfopath = 'NONE'
|
|
cnt = len(reslst)
|
|
for j in xrange(cnt):
|
|
resline = reslst[j]
|
|
pp = resline.find('.kindle-info')
|
|
if pp >= 0:
|
|
kinfopath = resline
|
|
break
|
|
if not os.path.exists(kinfopath):
|
|
raise K4MDrmException('Error: .kindle-info file can not be found')
|
|
return open(kinfopath,'r')
|