diff --git a/include/sbi_utils/fdt/fdt_helper.h b/include/sbi_utils/fdt/fdt_helper.h index 67251f8f..d3b065a6 100644 --- a/include/sbi_utils/fdt/fdt_helper.h +++ b/include/sbi_utils/fdt/fdt_helper.h @@ -9,6 +9,18 @@ #ifndef __FDT_HELPER_H__ #define __FDT_HELPER_H__ +/** + * Fix up the CPU node in the device tree + * + * This routine updates the "status" property of a CPU node in the device tree + * to "disabled" if that hart is in disabled state in OpenSBI. + * + * It is recommended that platform codes call this helper in their final_init() + * + * @param fdt: device tree blob + */ +void fdt_cpu_fixup(void *fdt); + /** * Fix up the PLIC node in the device tree * diff --git a/lib/utils/fdt/fdt_helper.c b/lib/utils/fdt/fdt_helper.c index fcae968e..4ef7a5e4 100644 --- a/lib/utils/fdt/fdt_helper.c +++ b/lib/utils/fdt/fdt_helper.c @@ -12,6 +12,32 @@ #include #include +void fdt_cpu_fixup(void *fdt) +{ + struct sbi_scratch *scratch = sbi_scratch_thishart_ptr(); + const struct sbi_platform *plat = sbi_platform_ptr(scratch); + char cpu_node[32] = ""; + int cpu_offset; + int err; + u32 i; + + err = fdt_open_into(fdt, fdt, fdt_totalsize(fdt) + 32); + if (err < 0) + return; + + /* assume hart ids are continuous */ + for (i = 0; i < sbi_platform_hart_count(plat); i++) { + sbi_sprintf(cpu_node, "/cpus/cpu@%d", i); + cpu_offset = fdt_path_offset(fdt, cpu_node); + + if (sbi_platform_hart_disabled(plat, i)) + fdt_setprop_string(fdt, cpu_offset, "status", + "disabled"); + + memset(cpu_node, 0, sizeof(cpu_node)); + } +} + void fdt_plic_fixup(void *fdt, const char *compat) { u32 *cells;