sys64738
/
blackmagic
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

stm32_mem: Check file size before flashing(#614) 

stm32mem is only meant to flash BMP to a STM32F103x(8|B). No need to
decipher the DFU interface string.
This commit is contained in:
Uwe Bonnes 2020-03-19 19:43:54 +01:00 committed by UweBonnes
parent 2b76ec0ea3
commit e57792c95e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/stm32_mem.py
View File

@ -20,6 +20,7 @@
from time import sleep from time import sleep
import struct import struct
import os
from sys import stdout, argv from sys import stdout, argv
import argparse import argparse
@ -195,6 +196,10 @@ if __name__ == "__main__":
exit(0) exit(0)
dfudev.make_idle() dfudev.make_idle()
file = open(args.progfile, "rb") file = open(args.progfile, "rb")
if (os.path.getsize(args.progfile) > 0x1f800):
print("File too large")
exit(0)
bin = file.read() bin = file.read()
product = dfudev.handle.getString(dfudev.dev.iProduct, 64) product = dfudev.handle.getString(dfudev.dev.iProduct, 64)