#!/usr/bin/env python # # stm32_mem.py: STM32 memory access using USB DFU class # Copyright (C) 2011 Black Sphere Technologies # Written by Gareth McMullin # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . from time import sleep import struct from sys import stdout, argv import argparse import usb import dfu CMD_GETCOMMANDS = 0x00 CMD_SETADDRESSPOINTER = 0x21 CMD_ERASE = 0x41 def stm32_erase(dev, addr): erase_cmd = struct.pack("" 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() if not devs: print "No devices found!" exit(-1) for dev in devs: dfudev = dfu.dfu_device(*dev) man = dfudev.handle.getString(dfudev.dev.iManufacturer, 30) product = dfudev.handle.getString(dfudev.dev.iProduct, 64) serial_no = dfudev.handle.getString(dfudev.dev.iSerialNumber, 30) if args.serial_target: if man == "Black Sphere Technologies" and serial_no == args.serial_target: break if man == "STMicroelectronics" and serial_no == args.serial_target: break else: if man == "Black Sphere Technologies": break if man == "STMicroelectronics": break print "Device %s: ID %04x:%04x %s - %s\n\tSerial %s" % ( dfudev.dev.filename, dfudev.dev.idVendor, dfudev.dev.idProduct, man, product, serial_no) if args.serial_target and serial_no != args.serial_target: print "Serial number doesn't match!\n" exit(-2) try: state = dfudev.get_state() except: print "Failed to read device state! Assuming APP_IDLE" state = dfu.STATE_APP_IDLE if state == dfu.STATE_APP_IDLE: dfudev.detach() print "Run again to upgrade firmware." exit(0) dfudev.make_idle() bin = open(args.progfile, "rb").read() if "F4" in product: addr = 0x8004000 else: addr = 0x8002000 while bin: print ("Programming memory at 0x%08X\r" % addr), stdout.flush() try: stm32_erase(dfudev, addr) except: print "\nErase Timed out\n" break stm32_write(dfudev, bin[:1024]) bin = bin[1024:] addr += 1024 stm32_manifest(dfudev) print "\nAll operations complete!\n"