mirrors/DeDRM_tools
mirrors/DeDRM_tools
1
0
Fork 0
mirror of https://github.com/noDRM/DeDRM_tools.git synced 2024-11-16 19:06:09 +06:00
Code Issues Projects Releases Wiki Activity

CMBDTC 1.1

Browse Source
This commit is contained in:
CMBDTC 2009-12-30 12:02:29 +00:00 committed by Apprentice Alf
parent 7027ac9b3e
commit 3a83053af9
1 changed files with 63 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

103
Topaz_Tools/lib/CMBDTC.py
View File

@ -2,7 +2,7 @@
""" """
Comprehensive Mazama Book DRM with Topaz Cryptography V1.0 Comprehensive Mazama Book DRM with Topaz Cryptography V1.1
-----BEGIN PUBLIC KEY----- -----BEGIN PUBLIC KEY-----
MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDdBHJ4CNc6DNFCw4MRCw4SWAK6 MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDdBHJ4CNc6DNFCw4MRCw4SWAK6
@ -142,7 +142,7 @@ def MD5(message):
return ctx.digest() return ctx.digest()
# #
# Returns the SHA1 digest of "message" # Returns the MD5 digest of "message"
# #
def SHA1(message): def SHA1(message):
@ -229,7 +229,7 @@ def findNameForHash(hash):
if hash == encodeHash(name, charMap2): if hash == encodeHash(name, charMap2):
result = name result = name
break break
return result return name
# #
# Print all the records from the kindle.info file (option -i) # Print all the records from the kindle.info file (option -i)
@ -297,7 +297,7 @@ def bookReadString():
# #
def bookReadHeaderRecordData(): def bookReadHeaderRecordData():
nbValues = ord(bookFile.read(1)) nbValues = bookReadEncodedNumber()
values = [] values = []
for i in range (0,nbValues): for i in range (0,nbValues):
values.append([bookReadEncodedNumber(),bookReadEncodedNumber(),bookReadEncodedNumber()]) values.append([bookReadEncodedNumber(),bookReadEncodedNumber(),bookReadEncodedNumber()])
@ -327,7 +327,7 @@ def parseTopazHeader():
if magic != 'TPZ0': if magic != 'TPZ0':
raise CMBDTCFatal("Parse Error : Invalid Header, not a Topaz file") raise CMBDTCFatal("Parse Error : Invalid Header, not a Topaz file")
nbRecords = ord(bookFile.read(1)) nbRecords = bookReadEncodedNumber()
bookHeaderRecords = {} bookHeaderRecords = {}
for i in range (0,nbRecords): for i in range (0,nbRecords):
@ -591,10 +591,11 @@ def generateDevicePID(table,dsn,nbRoll):
def usage(): def usage():
print("\nUsage:") print("\nUsage:")
print("\nCMBDTC.py [options] bookFileName\n") print("\nCMBDTC.py [options] bookFileName\n")
print("-r prints a record indicated in the form name:index (e.g \"img:0\")") print("-p Adds a PID to the list of PIDs that are tried to decrypt the book key (can be used several times)")
print("-r Prints or writes to disk a record indicated in the form name:index (e.g \"img:0\")")
print("-o Output file name to write records") print("-o Output file name to write records")
print("-v verbose (can be used several times)") print("-v Verbose (can be used several times)")
print("-i print kindle.info database") print("-i Print kindle.info database")
# #
# Main # Main
@ -612,9 +613,12 @@ def main(argv=sys.argv):
recordName = "" recordName = ""
recordIndex = 0 recordIndex = 0
outputFile = "" outputFile = ""
PIDs = []
kindleDatabase = None
try: try:
opts, args = getopt.getopt(sys.argv[1:], "vir:o:") opts, args = getopt.getopt(sys.argv[1:], "vir:o:p:")
except getopt.GetoptError, err: except getopt.GetoptError, err:
# print help information and exit: # print help information and exit:
print str(err) # will print something like "option -a not recognized" print str(err) # will print something like "option -a not recognized"
@ -634,43 +638,52 @@ def main(argv=sys.argv):
outputFile = a outputFile = a
if o =="-r": if o =="-r":
recordName,recordIndex = a.split(':') recordName,recordIndex = a.split(':')
if o =="-p":
PIDs.append(a)
# #
# Read the encrypted database # Read the encrypted database
# #
kindleDatabase = parseKindleInfo() try:
kindleDatabase = parseKindleInfo()
except Exception as message:
if verbose>0:
print(message)
if printInfo: if kindleDatabase != None :
printKindleInfo() if printInfo:
printKindleInfo()
# #
# Compute the DSN # Compute the DSN
# #
# Get the Mazama Random number # Get the Mazama Random number
MazamaRandomNumber = getKindleInfoValueForKey("MazamaRandomNumber") MazamaRandomNumber = getKindleInfoValueForKey("MazamaRandomNumber")
# Get the HDD serial # Get the HDD serial
encodedSystemVolumeSerialNumber = encodeHash(str(GetVolumeSerialNumber(GetSystemDirectory().split('\\')[0] + '\\')),charMap1) encodedSystemVolumeSerialNumber = encodeHash(str(GetVolumeSerialNumber(GetSystemDirectory().split('\\')[0] + '\\')),charMap1)
# Get the current user name # Get the current user name
encodedUsername = encodeHash(GetUserName(),charMap1) encodedUsername = encodeHash(GetUserName(),charMap1)
# concat, hash and encode # concat, hash and encode
DSN = encode(SHA1(MazamaRandomNumber+encodedSystemVolumeSerialNumber+encodedUsername),charMap1) DSN = encode(SHA1(MazamaRandomNumber+encodedSystemVolumeSerialNumber+encodedUsername),charMap1)
if verbose >1:
print("DSN: " + DSN) if verbose >1:
print("DSN: " + DSN)
# #
# Compute the device PID # Compute the device PID
# #
table = generatePidEncryptionTable()
devicePID = generateDevicePID(table,DSN,4)
PIDs.append(devicePID)
table = generatePidEncryptionTable() if verbose > 0:
devicePID = generateDevicePID(table,DSN,4) print("Device PID: " + devicePID)
if verbose > 0:
print("Device PID: " + devicePID)
# #
# Open book and parse metadata # Open book and parse metadata
@ -687,36 +700,46 @@ def main(argv=sys.argv):
# #
# Get the account token # Get the account token
kindleAccountToken = getKindleInfoValueForKey("kindle.account.tokens")
if verbose >1: if kindleDatabase != None:
print("Account Token: " + kindleAccountToken) kindleAccountToken = getKindleInfoValueForKey("kindle.account.tokens")
if verbose >1:
print("Account Token: " + kindleAccountToken)
keysRecord = bookMetadata["keys"] keysRecord = bookMetadata["keys"]
keysRecordRecord = bookMetadata[keysRecord] keysRecordRecord = bookMetadata[keysRecord]
pidHash = SHA1(DSN+kindleAccountToken+keysRecord+keysRecordRecord) pidHash = SHA1(DSN+kindleAccountToken+keysRecord+keysRecordRecord)
PID = encodePID(pidHash) bookPID = encodePID(pidHash)
PIDs.append(bookPID)
if verbose > 0: if verbose > 0:
print ("Book PID: " + PID ) print ("Book PID: " + bookPID )
# #
# Decrypt book key # Decrypt book key
# #
dkey = getBookPayloadRecord('dkey', 0) dkey = getBookPayloadRecord('dkey', 0)
bookKey = decryptDkeyRecords(dkey,PID)[0]
if verbose > 0: bookKeys = []
print("Book key: " + bookKey.encode('hex')) for PID in PIDs :
bookKeys+=decryptDkeyRecords(dkey,PID)
if recordName != "" :
extractBookPayloadRecord(recordName,int(recordIndex),outputFile) if len(bookKeys) == 0 :
if verbose > 0 :
print ("Book key could not be found. Maybe this book is not registered with this device.")
else :
bookKey = bookKeys[0]
if verbose > 0:
print("Book key: " + bookKey.encode('hex'))
if recordName != "" :
extractBookPayloadRecord(recordName,int(recordIndex),outputFile)
if outputFile != "" and verbose>0 : if outputFile != "" and verbose>0 :
print("Wrote record to file: "+outputFile) print("Wrote record to file: "+outputFile)
return 0 return 0