jackcartersmith/blackmagic
1
0
Fork 0
Code Issues Pull Requests Releases Activity

stm32_mem.py: Run automatically after switching to DFU mode.

Browse Source
This commit is contained in:
Uwe Bonnes 2017-09-27 21:38:54 +02:00
parent 4f3f4cb898
commit c41dfaef9a
2 changed files with 30 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/dfu.py
View File

@ -84,6 +84,8 @@ class dfu_device(object):
self.index = self.iface.interfaceNumber self.index = self.iface.interfaceNumber
else: self.index = self.iface else: self.index = self.iface
def release(self):
self.handle.releaseInterface()
def detach(self, wTimeout=255): def detach(self, wTimeout=255):
self.handle.controlMsg(usb.ENDPOINT_OUT | usb.TYPE_CLASS | self.handle.controlMsg(usb.ENDPOINT_OUT | usb.TYPE_CLASS |
usb.RECIP_INTERFACE, DFU_DETACH, usb.RECIP_INTERFACE, DFU_DETACH,

46
scripts/stm32_mem.py
View File

@ -69,18 +69,7 @@ def stm32_manifest(dev):
sleep(status.bwPollTimeout / 1000.0) sleep(status.bwPollTimeout / 1000.0)
if status.bState == dfu.STATE_DFU_MANIFEST: if status.bState == dfu.STATE_DFU_MANIFEST:
break break
def stm32_scan(args):
if __name__ == "__main__":
print
print "USB Device Firmware Upgrade - Host Utility -- version 1.2"
print "Copyright (C) 2011 Black Sphere Technologies"
print "License GPLv3+: GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>"
print
parser = argparse.ArgumentParser()
parser.add_argument("progfile", help="Binary file to program")
parser.add_argument("-s", "--serial_target", help="Match Serial Number")
args = parser.parse_args()
devs = dfu.finddevs() devs = dfu.finddevs()
bmp = 0 bmp = 0
if not devs: if not devs:
@ -120,10 +109,24 @@ if __name__ == "__main__":
print "Device ID:\t %04x:%04x" % (dfudev.dev.idVendor, dfudev.dev.idProduct) print "Device ID:\t %04x:%04x" % (dfudev.dev.idVendor, dfudev.dev.idProduct)
print "Manufacturer:\t %s" % man print "Manufacturer:\t %s" % man
print "Product:\t %s" % product print "Product:\t %s" % product
print "Serial:\t\t %s\n" % serial_no print "Serial:\t\t %s" % serial_no
if args.serial_target and serial_no != args.serial_target: if args.serial_target and serial_no != args.serial_target:
print "Serial number doesn't match!\n" print "Serial number doesn't match!\n"
exit(-2) exit(-2)
return dfudev
if __name__ == "__main__":
print
print "USB Device Firmware Upgrade - Host Utility -- version 1.2"
print "Copyright (C) 2011 Black Sphere Technologies"
print "License GPLv3+: GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>"
print
parser = argparse.ArgumentParser()
parser.add_argument("progfile", help="Binary file to program")
parser.add_argument("-s", "--serial_target", help="Match Serial Number")
args = parser.parse_args()
dfudev = stm32_scan(args)
try: try:
state = dfudev.get_state() state = dfudev.get_state()
except: except:
@ -131,11 +134,18 @@ if __name__ == "__main__":
state = dfu.STATE_APP_IDLE state = dfu.STATE_APP_IDLE
if state == dfu.STATE_APP_IDLE: if state == dfu.STATE_APP_IDLE:
dfudev.detach() dfudev.detach()
print "Run again to upgrade firmware." dfudev.release()
exit(0) print "Invoking DFU Device"
timeout = 0
while True :
sleep(0.5)
timeout = timeout + 0.5
dfudev = stm32_scan(args)
if dfudev: break
if timeout > 5 :
print "Error: DFU device did not appear"
exit(-1)
dfudev.make_idle() dfudev.make_idle()
bin = open(args.progfile, "rb").read() bin = open(args.progfile, "rb").read()
product = dfudev.handle.getString(dfudev.dev.iProduct, 64) product = dfudev.handle.getString(dfudev.dev.iProduct, 64)
@ -152,7 +162,7 @@ if __name__ == "__main__":
# To support the STM DFU bootloader, the interface descriptor must # To support the STM DFU bootloader, the interface descriptor must
# get evaluated and erase called only once per sector! # get evaluated and erase called only once per sector!
stm32_erase(dfudev, addr) stm32_erase(dfudev, addr)
except: except:
print "\nErase Timed out\n" print "\nErase Timed out\n"
break break
try: try: