12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- import shutil
- from pathlib import Path
- from SCons.Script import DefaultEnvironment
- env = DefaultEnvironment()
- def copytree(src, dst, symlinks=False, ignore=None):
- for item in src.iterdir():
- if item.is_dir():
- shutil.copytree(item, dst / item.name, symlinks, ignore)
- else:
- shutil.copy2(item, dst / item.name)
- def replace_define(field, value):
- envdefs = env['CPPDEFINES'].copy()
- for define in envdefs:
- if define[0] == field:
- env['CPPDEFINES'].remove(define)
- env['CPPDEFINES'].append((field, value))
- def relocate_firmware(address):
- replace_define("VECT_TAB_ADDR", address)
- def relocate_vtab(address):
- replace_define("VECT_TAB_OFFSET", address)
- def custom_ld_script(ldname):
- apath = str(Path("buildroot/share/PlatformIO/ldscripts", ldname).resolve())
- for i, flag in enumerate(env["LINKFLAGS"]):
- if "-Wl,-T" in flag:
- env["LINKFLAGS"][i] = "-Wl,-T" + apath
- elif flag == "-T":
- env["LINKFLAGS"][i + 1] = apath
- def encrypt_mks(source, target, env, new_name):
- import sys
- key = [0xA3, 0xBD, 0xAD, 0x0D, 0x41, 0x11, 0xBB, 0x8D, 0xDC, 0x80, 0x2D, 0xD0, 0xD2, 0xC4, 0x9B, 0x1E, 0x26, 0xEB, 0xE3, 0x33, 0x4A, 0x15, 0xE4, 0x0A, 0xB3, 0xB1, 0x3C, 0x93, 0xBB, 0xAF, 0xF7, 0x3E]
-
- mf = env["MARLIN_FEATURES"]
- if "FIRMWARE_BIN" in mf: new_name = mf["FIRMWARE_BIN"]
- fwpath = Path(target[0].path)
- fwfile = fwpath.open("rb")
- enfile = Path(target[0].dir.path, new_name).open("wb")
- length = fwpath.stat().st_size
- position = 0
- try:
- while position < length:
- byte = fwfile.read(1)
- if 320 <= position < 31040:
- byte = chr(ord(byte) ^ key[position & 31])
- if sys.version_info[0] > 2:
- byte = bytes(byte, 'latin1')
- enfile.write(byte)
- position += 1
- finally:
- fwfile.close()
- enfile.close()
- fwpath.unlink()
- def add_post_action(action):
- env.AddPostAction(str(Path("$BUILD_DIR", "${PROGNAME}.bin")), action)
|