2019-11-10 03:34:12 +00:00
|
|
|
import os
|
2019-07-14 23:05:24 +00:00
|
|
|
Import("env")
|
|
|
|
|
|
|
|
# Relocate firmware from 0x08000000 to 0x08010000
|
|
|
|
for define in env['CPPDEFINES']:
|
|
|
|
if define[0] == "VECT_TAB_ADDR":
|
|
|
|
env['CPPDEFINES'].remove(define)
|
|
|
|
env['CPPDEFINES'].append(("VECT_TAB_ADDR", "0x08010000"))
|
2019-11-10 03:34:12 +00:00
|
|
|
|
|
|
|
custom_ld_script = os.path.abspath("buildroot/share/PlatformIO/ldscripts/STM32F103VE_longer.ld")
|
|
|
|
for i, flag in enumerate(env["LINKFLAGS"]):
|
|
|
|
if "-Wl,-T" in flag:
|
|
|
|
env["LINKFLAGS"][i] = "-Wl,-T" + custom_ld_script
|
|
|
|
elif flag == "-T":
|
|
|
|
env["LINKFLAGS"][i + 1] = custom_ld_script
|
|
|
|
|
2019-07-14 23:05:24 +00:00
|
|
|
|
2019-08-13 00:50:34 +00:00
|
|
|
# Rename ${PROGNAME}.bin and save it as 'project.bin' (No encryption on the Longer3D)
|
2019-07-14 23:05:24 +00:00
|
|
|
def encrypt(source, target, env):
|
|
|
|
import os
|
|
|
|
|
|
|
|
firmware = open(target[0].path, "rb")
|
|
|
|
marlin_alfa = open(target[0].dir.path +'/project.bin', "wb")
|
|
|
|
length = os.path.getsize(target[0].path)
|
|
|
|
position = 0
|
|
|
|
try:
|
|
|
|
while position < length:
|
|
|
|
byte = firmware.read(1)
|
|
|
|
marlin_alfa.write(byte)
|
|
|
|
position += 1
|
|
|
|
finally:
|
|
|
|
firmware.close()
|
|
|
|
marlin_alfa.close()
|
2019-08-13 00:50:34 +00:00
|
|
|
|
|
|
|
env.AddPostAction("$BUILD_DIR/${PROGNAME}.bin", encrypt);
|