From e57792c95e77a90c16a69223678dc1cba2df0df4 Mon Sep 17 00:00:00 2001 From: Uwe Bonnes Date: Thu, 19 Mar 2020 19:43:54 +0100 Subject: [PATCH] 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. --- scripts/stm32_mem.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/scripts/stm32_mem.py b/scripts/stm32_mem.py index 1c8495a..ae99402 100755 --- a/scripts/stm32_mem.py +++ b/scripts/stm32_mem.py @@ -20,6 +20,7 @@ from time import sleep import struct +import os from sys import stdout, argv import argparse @@ -195,6 +196,10 @@ if __name__ == "__main__": exit(0) dfudev.make_idle() file = open(args.progfile, "rb") + if (os.path.getsize(args.progfile) > 0x1f800): + print("File too large") + exit(0) + bin = file.read() product = dfudev.handle.getString(dfudev.dev.iProduct, 64)