From 1be9c6e9c5afe3e4b2fe680876e6091ef0309d18 Mon Sep 17 00:00:00 2001
From: Uwe Bonnes <bon@elektron.ikp.physik.tu-darmstadt.de>
Date: Sun, 25 Jun 2017 15:18:08 +0200
Subject: [PATCH] stm32_mem.py: Set address before writing to inhibit
 bootloader overwrite.

---
 scripts/stm32_mem.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/scripts/stm32_mem.py b/scripts/stm32_mem.py
index 9f2e228..780da32 100755
--- a/scripts/stm32_mem.py
+++ b/scripts/stm32_mem.py
@@ -39,6 +39,16 @@ def stm32_erase(dev, addr):
 		if status.bState == dfu.STATE_DFU_DOWNLOAD_IDLE:
 			break
 
+def stm32_set_address(dev, addr):
+	set_address_cmd = struct.pack("<BL", CMD_SETADDRESSPOINTER, addr)
+	dev.download(0, set_address_cmd)
+	while True:
+		status = dev.get_status()
+		if status.bState == dfu.STATE_DFU_DOWNLOAD_BUSY:
+			sleep(status.bwPollTimeout / 1000.0)
+		if status.bState == dfu.STATE_DFU_DOWNLOAD_IDLE:
+			break
+
 def stm32_write(dev, data):
 	dev.download(2, data)
 	while True:
@@ -120,6 +130,11 @@ if __name__ == "__main__":
                 except:
 			print "\nErase Timed out\n"
 			break
+		try:
+			stm32_set_address(dfudev, addr)
+		except:
+			print "\nSet Address Timed out\n"
+			break
 		stm32_write(dfudev, bin[:1024])
 
 		bin = bin[1024:]