BTCPrivate
/
blue-loader-python
mirror of https://github.com/BTCPrivate/blue-loader-python.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/LedgerHQ/blue-loader-python

This commit is contained in:
BTChip 2017-01-22 18:05:16 +01:00
parent 3738a758e3 e73a8a6389
commit 53a76b055e
3 changed files with 73 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ledgerblue/endorsementSetup.py
View File

@ -21,12 +21,15 @@ from .comm import getDongle
import binascii
import argparse
def auto_int(x):
return int(x, 0)
parser = argparse.ArgumentParser()
parser.add_argument("--key", help="Reference of the endorsement key to setup (1 or 2)", type=auto_int)
parser.add_argument("--certificate", help="Certificate to store if finalizing the endorsement (hex encoded)")
parser.add_argument(
"--key", help="Reference of the endorsement key to setup (1 or 2)", type=auto_int)
parser.add_argument(
"--certificate", help="Certificate to store if finalizing the endorsement (hex encoded)")
parser.add_argument("--apdu", help="Display APDU log", action='store_true')
args = parser.parse_args()
@ -40,10 +43,10 @@ dongle = getDongle(args.apdu)
if args.certificate == None:
apdu = bytearray([0xe0, 0xC0, args.key, 0x00, 0x00])
response = dongle.exchange(apdu)
print "Public key " + str(response[0:65]).encode('hex')
print "Certificate " + str(response[65:]).encode('hex')
print("Public key " + str(response[0:65]).encode('hex'))
print("Certificate " + str(response[65:]).encode('hex'))
else:
certificate = bytearray.fromhex(args.certificate)
apdu = bytearray([0xe0, 0xC2, 0x00, 0x00, len(certificate)]) + certificate
dongle.exchange(apdu)
print "Endorsement setup finalized"
print("Endorsement setup finalized")

2
ledgerblue/hashApp.py
View File

@ -44,4 +44,4 @@ for a in parser.getAreas():
m.update(a.data)
dataToSign = m.digest()
print dataToSign.encode('hex')
print(dataToSign.encode('hex'))

17
ledgerblue/runScript.py
View File

@ -25,13 +25,15 @@ import argparse
import sys
import fileinput
def auto_int(x):
return int(x, 0)
parser = argparse.ArgumentParser()
parser.add_argument("--fileName", help="Set the file name to load")
parser.add_argument("--apdu", help="Display APDU log", action='store_true')
parser.add_argument("--scp", help="open secure channel to exchange apdu", action='store_true')
parser.add_argument(
"--scp", help="open secure channel to exchange apdu", action='store_true')
parser.add_argument("--targetId", help="Set the chip target ID", type=auto_int)
parser.add_argument("--rootPrivateKey", help="Set the root private key")
@ -47,6 +49,7 @@ else:
class SCP:
def __init__(self, dongle, targetId, rootPrivateKey):
self.key = getDeployedSecretV2(dongle, rootPrivateKey, targetId)
self.iv = b'\x00' * 16
@ -67,7 +70,7 @@ class SCP:
decryptedData = cipher.decrypt(data)
l = len(decryptedData) - 1
while (decryptedData[l] != chr(0x80)):
l-=1
l -= 1
decryptedData = decryptedData[0:l]
self.iv = data[len(data) - 16:]
return decryptedData
@ -76,7 +79,8 @@ dongle = getDongle(args.apdu)
if args.scp:
if args.rootPrivateKey is None:
privateKey = PrivateKey()
publicKey = binascii.hexlify(privateKey.pubkey.serialize(compressed=False))
publicKey = binascii.hexlify(
privateKey.pubkey.serialize(compressed=False))
print("Generated random root public key : %s" % publicKey)
args.rootPrivateKey = privateKey.serialize()
scp = SCP(dongle, args.targetId, bytearray.fromhex(args.rootPrivateKey))
@ -87,10 +91,11 @@ for data in file:
continue
if args.scp:
data = bytearray(data)
if data[4] > 0 and len(data)>5:
apduData = data[5 : 5 + data[4]]
if data[4] > 0 and len(data) > 5:
apduData = data[5: 5 + data[4]]
apduData = scp.encryptAES(str(apduData))
result = dongle.exchange(data[0:4] + bytearray([len(apduData)]) + bytearray(apduData))
result = dongle.exchange(
data[0:4] + bytearray([len(apduData)]) + bytearray(apduData))
else:
result = dongle.exchange(data[0:5])
result = scp.decryptAES(str(result))