Firmware2/buildroot/share/PlatformIO/scripts/stm32_bootloader.py

41 lines
1.2 KiB
Python
Raw Normal View History

2021-02-28 05:38:57 +01:00
#
# stm32_bootloader.py
#
import os,sys,shutil,marlin
2020-07-07 00:08:52 +02:00
Import("env")
from SCons.Script import DefaultEnvironment
board = DefaultEnvironment().BoardConfig()
#
# Copy the firmware.bin file to build.firmware, no encryption
#
2020-07-07 00:08:52 +02:00
def noencrypt(source, target, env):
firmware = os.path.join(target[0].dir.path, board.get("build.firmware"))
shutil.copy(target[0].path, firmware)
2020-07-07 00:08:52 +02:00
#
# For build.offset define LD_FLASH_OFFSET, used by ldscript.ld
#
2020-07-07 00:08:52 +02:00
if 'offset' in board.get("build").keys():
LD_FLASH_OFFSET = board.get("build.offset")
2020-07-07 00:08:52 +02:00
marlin.relocate_vtab(LD_FLASH_OFFSET)
# Get upload.maximum_ram_size (defined by /buildroot/share/PlatformIO/boards/VARIOUS.json)
maximum_ram_size = board.get("upload.maximum_ram_size")
2020-07-07 00:08:52 +02:00
for i, flag in enumerate(env["LINKFLAGS"]):
if "-Wl,--defsym=LD_FLASH_OFFSET" in flag:
env["LINKFLAGS"][i] = "-Wl,--defsym=LD_FLASH_OFFSET=" + LD_FLASH_OFFSET
if "-Wl,--defsym=LD_MAX_DATA_SIZE" in flag:
env["LINKFLAGS"][i] = "-Wl,--defsym=LD_MAX_DATA_SIZE=" + str(maximum_ram_size - 40)
2020-07-07 00:08:52 +02:00
#
# Only copy the file if there's no encrypt
#
2020-12-24 03:01:21 +01:00
board_keys = board.get("build").keys()
if 'firmware' in board_keys and not 'encrypt' in board_keys:
import marlin
marlin.add_post_action(noencrypt)