diff --git a/include/sbi/sbi_platform.h b/include/sbi/sbi_platform.h index 010328eb..72bb7e29 100644 --- a/include/sbi/sbi_platform.h +++ b/include/sbi/sbi_platform.h @@ -76,6 +76,11 @@ struct sbi_platform_operations { /** Platform final initialization */ int (*final_init)(bool cold_boot); + /** Platform early exit */ + void (*early_exit)(void); + /** Platform final exit */ + void (*final_exit)(void); + /** For platforms that do not implement misa, non-standard * methods are needed to determine cpu extension. */ @@ -301,6 +306,28 @@ static inline int sbi_platform_final_init(const struct sbi_platform *plat, return 0; } +/** + * Early exit for current HART + * + * @param plat pointer to struct sbi_platform + */ +static inline void sbi_platform_early_exit(const struct sbi_platform *plat) +{ + if (plat && sbi_platform_ops(plat)->early_exit) + sbi_platform_ops(plat)->early_exit(); +} + +/** + * Final exit for current HART + * + * @param plat pointer to struct sbi_platform + */ +static inline void sbi_platform_final_exit(const struct sbi_platform *plat) +{ + if (plat && sbi_platform_ops(plat)->final_exit) + sbi_platform_ops(plat)->final_exit(); +} + /** * Check CPU extension in MISA * diff --git a/include/sbi/sbi_system.h b/include/sbi/sbi_system.h index bf768d9a..c1ef3742 100644 --- a/include/sbi/sbi_system.h +++ b/include/sbi/sbi_system.h @@ -18,6 +18,10 @@ int sbi_system_early_init(struct sbi_scratch *scratch, bool cold_boot); int sbi_system_final_init(struct sbi_scratch *scratch, bool cold_boot); +void sbi_system_early_exit(struct sbi_scratch *scratch); + +void sbi_system_final_exit(struct sbi_scratch *scratch); + void __attribute__((noreturn)) sbi_system_reboot(struct sbi_scratch *scratch, u32 type); diff --git a/lib/sbi/sbi_init.c b/lib/sbi/sbi_init.c index b4c242f2..d5d35130 100644 --- a/lib/sbi/sbi_init.c +++ b/lib/sbi/sbi_init.c @@ -203,5 +203,9 @@ void __noreturn sbi_exit(struct sbi_scratch *scratch) sbi_hart_unmark_available(hartid); + sbi_platform_early_exit(plat); + + sbi_platform_final_exit(plat); + sbi_hart_hang(); } diff --git a/lib/sbi/sbi_system.c b/lib/sbi/sbi_system.c index cdc85d6f..9ea3c21d 100644 --- a/lib/sbi/sbi_system.c +++ b/lib/sbi/sbi_system.c @@ -23,6 +23,16 @@ int sbi_system_final_init(struct sbi_scratch *scratch, bool cold_boot) return sbi_platform_final_init(sbi_platform_ptr(scratch), cold_boot); } +void sbi_system_early_exit(struct sbi_scratch *scratch) +{ + sbi_platform_early_exit(sbi_platform_ptr(scratch)); +} + +void sbi_system_final_exit(struct sbi_scratch *scratch) +{ + sbi_platform_final_exit(sbi_platform_ptr(scratch)); +} + void __attribute__((noreturn)) sbi_system_reboot(struct sbi_scratch *scratch, u32 type)