From bc3ae6b7688aca881a4b1e61e01f86c9e316d600 Mon Sep 17 00:00:00 2001
From: Federico Cerutti <federico@ceres-c.it>
Date: Wed, 14 Oct 2020 19:47:38 +0200
Subject: [PATCH] Elf sections check before flash

Due to the addition of new codecs and applications, code size grew to
overwrite the part of FRAM currently used to store card slots (settings).
This simple (although hacky) makefile command verifies if such an
issue will occur before flashing the firmware.
---
 Firmware/Chameleon-Mini/Makefile | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/Firmware/Chameleon-Mini/Makefile b/Firmware/Chameleon-Mini/Makefile
index f67fa3b..0bdbcbf 100644
--- a/Firmware/Chameleon-Mini/Makefile
+++ b/Firmware/Chameleon-Mini/Makefile
@@ -134,6 +134,7 @@ AVRDUDE_FLAGS = -p $(AVRDUDE_MCU) -P $(AVRDUDE_PORT) -c $(AVRDUDE_PROGRAMMER)
 AVRDUDE_WRITE_APP_LATEST = -U application:w:Latest/Chameleon-RevG.hex
 AVRDUDE_WRITE_EEPROM_LATEST = -U eeprom:w:Latest/Chameleon-RevG.eep
 
+.PHONY: program program-latest dfu-flip dfu-prog check_size style
 
 # Default target
 all:
@@ -165,27 +166,46 @@ spmhelper: $(TARGET).elf
 	$(CROSS)-objcopy -O ihex -j .spmhelper $(SPM_HELPER_OBJCOPY) $(TARGET).elf $(TARGET).hex
 
 # Program the device using avrdude
-program: $(TARGET).hex $(TARGET).eep
+program: $(TARGET).hex $(TARGET).eep check_size
 	avrdude $(AVRDUDE_FLAGS) $(AVRDUDE_WRITE_APP) $(AVRDUDE_WRITE_EEPROM)
 
 # Program the device using avrdude with the latest official firmware
-program-latest:
+program-latest: check_size
 	avrdude $(AVRDUDE_FLAGS) $(AVRDUDE_WRITE_APP_LATEST) $(AVRDUDE_WRITE_EEPROM_LATEST)
 
 # Program the device using batchisp and the DFU bootloader
 # Note that the device has to be in bootloader mode already
-dfu-flip: $(TARGET).hex $(TARGET).eep
+dfu-flip: $(TARGET).hex $(TARGET).eep check_size
 	cp $(TARGET).eep EEPROM.hex
 	batchisp -hardware usb -device $(MCU) -operation erase f memory FLASH loadbuffer $(TARGET).hex program verify memory EEPROM loadbuffer EEPROM.hex program verify start reset 0
 	rm EEPROM.hex
 
 # Program the device using dfu-programmer
-dfu-prog: $(TARGET).hex $(TARGET).eep
+dfu-prog: $(TARGET).hex $(TARGET).eep check_size
 	dfu-programmer $(MCU) erase
 	dfu-programmer $(MCU) flash-eeprom $(TARGET).eep
 	dfu-programmer $(MCU) flash $(TARGET).hex
 	dfu-programmer $(MCU) reset
 
+check_size:
+ifeq ($(OS),Windows_NT)
+# TODO Windows version of this check
+	@PROGMEM_SIZE = 0
+else
+	@{ \
+	set -e; \
+	if [ ! -f $(TARGET).elf ]; then \
+		exit 0; \
+	fi; \
+	PROGMEM_SIZE=$$(avr-size $(TARGET).elf | grep -oP "\d+" | sed -n 4p); \
+	MAX_PRGMEM_SIZE=$$(printf "%d\n" $(FLASH_DATA_ADDR)); \
+	if [ $$PROGMEM_SIZE -gt $$MAX_PRGMEM_SIZE ]; then \
+		echo "make: *** $(TARGET).elf Application Section size $$PROGMEM_SIZE excedes maximum allowed $$MAX_PRGMEM_SIZE. Please disable some features in Makefile"; \
+		exit 1; \
+	fi; \
+	}
+endif
+
 style:
 	# Make sure astyle is installed
 	@which astyle >/dev/null || ( echo "Please install 'astyle' package first" ; exit 1 )