mirror of
https://github.com/mikecarper/MeshCore.git
synced 2026-10-18 07:20:00 +00:00
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
"""Make the nRF52 core's loop task use the configured stack even under LTO.
|
|
|
|
GNU --wrap only redirects undefined references. The pinned core and FreeRTOS
|
|
share an LTO graph, so its xTaskCreate call can bind directly to the real
|
|
function and silently retain the framework's 4 KiB stack. Explicitly route
|
|
only the core loop-task call through MeshCore's existing stack policy in a
|
|
private build copy. Never edit the shared framework package.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
OLD_LOOP_CREATE = ' xTaskCreate(loop_task, "loop", LOOP_STACK_SZ, NULL, TASK_PRIO_LOW, &_loopHandle);'
|
|
LOOP_CREATE = OLD_LOOP_CREATE.replace("xTaskCreate(", "__wrap_xTaskCreate(", 1)
|
|
DECLARATION = """extern "C" BaseType_t __wrap_xTaskCreate(
|
|
TaskFunction_t task_code, const char* const task_name,
|
|
const configSTACK_DEPTH_TYPE stack_depth, void* const parameters,
|
|
UBaseType_t priority, TaskHandle_t* const created_task);
|
|
|
|
"""
|
|
LOOP_STACK = "#define LOOP_STACK_SZ (256*4)"
|
|
|
|
|
|
def patched_source(source):
|
|
source = source.replace("\r\n", "\n")
|
|
if (source.count(LOOP_CREATE) == 1 and source.count(DECLARATION) == 1
|
|
and OLD_LOOP_CREATE not in source and source.count(LOOP_STACK) == 1):
|
|
return source
|
|
if (source.count(OLD_LOOP_CREATE) != 1 or source.count(LOOP_STACK) != 1
|
|
or LOOP_CREATE in source or DECLARATION in source):
|
|
raise RuntimeError(
|
|
"nRF52 loop stack fix: unrecognized core task creation; review the "
|
|
"framework update before building (shared SDK not modified)"
|
|
)
|
|
return source.replace(LOOP_STACK, DECLARATION + LOOP_STACK).replace(
|
|
OLD_LOOP_CREATE, LOOP_CREATE)
|
|
|
|
|
|
def replace_core(build_env, node):
|
|
source = Path(node.srcnode().get_abspath())
|
|
patched = patched_source(source.read_text(encoding="utf-8"))
|
|
destination = Path(build_env.subst("$BUILD_DIR")) / "patched-nrf52-core" / "main.cpp"
|
|
destination.parent.mkdir(parents=True, exist_ok=True)
|
|
if not destination.exists() or destination.read_text(encoding="utf-8") != patched:
|
|
destination.write_text(patched, encoding="utf-8")
|
|
print("nRF52 loop stack: explicit LTO-safe core task policy enabled")
|
|
return build_env.File(str(destination))
|
|
|
|
|
|
def install(build_env):
|
|
build_env.AddBuildMiddleware(replace_core, "*cores*nRF5*main.cpp")
|
|
|
|
|
|
if "Import" in globals():
|
|
Import("env")
|
|
install(env)
|