You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.6 KiB
96 lines
2.6 KiB
8 years ago
|
#!/usr/bin/env python
|
||
|
|
from __future__ import print_function
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
import StringIO
|
||
|
|
|
||
|
|
def flash_release(path=None, st_serial=None):
|
||
|
|
from panda import Panda, PandaDFU, ESPROM, CesantaFlasher
|
||
|
|
from zipfile import ZipFile
|
||
|
|
|
||
|
|
def status(x):
|
||
|
|
print("\033[1;32;40m"+x+"\033[00m")
|
||
|
|
|
||
|
|
if st_serial == None:
|
||
|
|
# look for Panda
|
||
|
|
panda_list = Panda.list()
|
||
|
|
if len(panda_list) == 0:
|
||
|
|
raise Exception("panda not found, make sure it's connected and your user can access it")
|
||
|
|
elif len(panda_list) > 1:
|
||
|
|
raise Exception("Please only connect one panda")
|
||
|
|
st_serial = panda_list[0]
|
||
|
|
print("Using panda with serial %s" % st_serial)
|
||
|
|
|
||
|
|
if path == None:
|
||
|
|
print("Fetching latest firmware from github.com/commaai/panda-artifacts")
|
||
|
|
r = requests.get("https://raw.githubusercontent.com/commaai/panda-artifacts/master/latest.json")
|
||
|
|
url = json.loads(r.text)['url']
|
||
|
|
r = requests.get(url)
|
||
|
|
print("Fetching firmware from %s" % url)
|
||
|
|
path = StringIO.StringIO(r.content)
|
||
|
|
|
||
|
|
zf = ZipFile(path)
|
||
|
|
zf.printdir()
|
||
|
|
|
||
|
|
version = zf.read("version")
|
||
|
|
status("0. Preparing to flash "+version)
|
||
|
|
|
||
|
|
code_bootstub = zf.read("bootstub.panda.bin")
|
||
|
|
code_panda = zf.read("panda.bin")
|
||
|
|
|
||
|
|
code_boot_15 = zf.read("boot_v1.5.bin")
|
||
|
|
code_boot_15 = code_boot_15[0:2] + "\x00\x30" + code_boot_15[4:]
|
||
|
|
|
||
|
|
code_user1 = zf.read("user1.bin")
|
||
|
|
code_user2 = zf.read("user2.bin")
|
||
|
|
|
||
|
|
# enter DFU mode
|
||
|
|
status("1. Entering DFU mode")
|
||
|
|
panda = Panda(st_serial)
|
||
|
|
panda.enter_bootloader()
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# program bootstub
|
||
|
|
status("2. Programming bootstub")
|
||
|
|
dfu = PandaDFU(PandaDFU.st_serial_to_dfu_serial(st_serial))
|
||
|
|
dfu.program_bootstub(code_bootstub)
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# flash main code
|
||
|
|
status("3. Flashing main code")
|
||
|
|
panda = Panda(st_serial)
|
||
|
|
panda.flash(code=code_panda)
|
||
|
|
panda.close()
|
||
|
|
|
||
|
|
# flashing ESP
|
||
|
|
status("4. Flashing ESP (slow!)")
|
||
|
|
align = lambda x, sz=0x1000: x+"\xFF"*((sz-len(x)) % sz)
|
||
|
|
esp = ESPROM(st_serial)
|
||
|
|
esp.connect()
|
||
|
|
flasher = CesantaFlasher(esp, 230400)
|
||
|
|
flasher.flash_write(0x0, align(code_boot_15), True)
|
||
|
|
flasher.flash_write(0x1000, align(code_user1), True)
|
||
|
|
flasher.flash_write(0x81000, align(code_user2), True)
|
||
|
|
flasher.flash_write(0x3FE000, "\xFF"*0x1000)
|
||
|
|
flasher.boot_fw()
|
||
|
|
del flasher
|
||
|
|
del esp
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
# check for connection
|
||
|
|
status("5. Verifying version")
|
||
|
|
panda = Panda(st_serial)
|
||
|
|
my_version = panda.get_version()
|
||
|
|
print("dongle id: %s" % panda.get_serial()[0])
|
||
|
|
print(my_version, "should be", version)
|
||
|
|
assert(str(version) == str(my_version))
|
||
|
|
|
||
|
|
# done!
|
||
|
|
status("6. Success!")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
flash_release(*sys.argv[1:])
|
||
|
|
|