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

Signature and icon support

This commit is contained in:
BTChip 2016-07-28 09:45:29 +02:00
parent f51eff7d96
commit f7253c0a52
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ledgerblue/hexLoader.py
View File

@ -27,7 +27,7 @@ class HexLoader:
self.cla = cla self.cla = cla
self.secure = secure self.secure = secure
self.key = key self.key = key
self.iv = "\x00" * 16; self.iv = "\x00" * 16
self.relative = relative self.relative = relative
def crc16(self, data): def crc16(self, data):
@ -65,7 +65,7 @@ class HexLoader:
0xef1f, 0xff3e, 0xcf5d, 0xdf7c, 0xaf9b, 0xbfba, 0x8fd9, 0x9ff8, 0xef1f, 0xff3e, 0xcf5d, 0xdf7c, 0xaf9b, 0xbfba, 0x8fd9, 0x9ff8,
0x6e17, 0x7e36, 0x4e55, 0x5e74, 0x2e93, 0x3eb2, 0x0ed1, 0x1ef0 0x6e17, 0x7e36, 0x4e55, 0x5e74, 0x2e93, 0x3eb2, 0x0ed1, 0x1ef0
] ]
crc = 0xFFFF; crc = 0xFFFF
for i in range(0, len(data)): for i in range(0, len(data)):
b = data[i] & 0xff b = data[i] & 0xff
b = (b ^ ((crc >> 8) & 0xff)) & 0xff b = (b ^ ((crc >> 8) & 0xff)) & 0xff
@ -114,15 +114,21 @@ class HexLoader:
data = struct.pack('>I', targetId) data = struct.pack('>I', targetId)
self.exchange(self.cla, 0x04, 0x00, 0x00, data) self.exchange(self.cla, 0x04, 0x00, 0x00, data)
def boot(self, bootadr): def boot(self, bootadr, signature=None):
# Force jump into Thumb mode # Force jump into Thumb mode
bootadr |= 1 bootadr |= 1
data = '\x09' + struct.pack('>I', bootadr) data = '\x09' + struct.pack('>I', bootadr)
if (signature != None):
data += chr(len(signature)) + signature
data = self.encryptAES(data) data = self.encryptAES(data)
self.exchange(self.cla, 0x00, 0x00, 0x00, data) self.exchange(self.cla, 0x00, 0x00, 0x00, data)
def createApp(self, appflags, applength, appname): def createApp(self, appflags, applength, appname, icon=None, path=None):
data = '\x0B' + struct.pack('>I', applength) + struct.pack('>I', appflags) + chr(len(appname)) + appname data = '\x0B' + struct.pack('>I', applength) + struct.pack('>I', appflags) + chr(len(appname)) + appname
if (icon != None):
data += chr(len(icon)) + icon
if (path != None):
data += chr(len(path)) + path
data = self.encryptAES(data) data = self.encryptAES(data)
self.exchange(self.cla, 0x00, 0x00, 0x00, data) self.exchange(self.cla, 0x00, 0x00, 0x00, data)
@ -161,9 +167,9 @@ class HexLoader:
self.crcSegment(0, len(data), crc) self.crcSegment(0, len(data), crc)
return sha256.hexdigest() return sha256.hexdigest()
def run(self, hexAreas, bootaddr): def run(self, hexAreas, bootaddr, signature=None):
initialAddress = 0 initialAddress = 0
if (len(hexAreas) <> 0) and self.relative: if (len(hexAreas) <> 0) and self.relative:
initialAddress = hexAreas[0].getStart() initialAddress = hexAreas[0].getStart()
self.boot(bootaddr - initialAddress) self.boot(bootaddr - initialAddress, signature)