arch: custom cpu_idle and cpu_atomic harmonization

custom arch_cpu_idle and arch_cpu_atomic_idle implementation was done
differently on different architectures. riscv implemented those as weak
symbols, xtensa used a kconfig and all other architectures did not
really care, but this was a global kconfig that should apply to all
architectures.

Signed-off-by: Anas Nashif <[email protected]>

Author: nashif

arch/Kconfig

@@ -1084,3 +1084,10 @@ config ARCH_CPU_IDLE_CUSTOM
 	help
 	  This options allows applications to override the default arch idle implementation with
 	  a custom one.
+
+config ARCH_CPU_ATOMIC_IDLE_CUSTOM
+	bool "Custom arch_cpu_atomic_idle implementation"
+	default n
+	help
+	  This options allows applications to override the default arch idle implementation with
+	  a custom one.

arch/arc/core/cpu_idle.S

@@ -26,6 +26,7 @@ SECTION_VAR(BSS, z_arc_cpu_sleep_mode)
 	.align 4
 	.word 0
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 /*
  * @brief Put the CPU in low-power mode
  *
@@ -48,14 +49,17 @@ SECTION_FUNC(TEXT, arch_cpu_idle)
 	sleep r1
 	j_s [blink]
 	nop
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 /*
  * @brief Put the CPU in low-power mode, entered with IRQs locked
  *
  * This function exits with interrupts restored to <key>.
  *
  * void arch_cpu_atomic_idle(unsigned int key)
  */
+
 SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 
 #ifdef CONFIG_TRACING
@@ -70,3 +74,4 @@ SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 	sleep r1
 	j_s.d [blink]
 	seti r0
+#endif

arch/arm/core/cortex_a_r/cpu_idle.S

@@ -49,6 +49,7 @@ _skip_\@:
 #endif /* CONFIG_ARM_ON_ENTER_CPU_IDLE_HOOK */
 .endm
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 SECTION_FUNC(TEXT, arch_cpu_idle)
 #ifdef CONFIG_TRACING
 	push	{r0, lr}
@@ -68,6 +69,9 @@ SECTION_FUNC(TEXT, arch_cpu_idle)
 
 	bx	lr
 
+#endif
+
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 #ifdef CONFIG_TRACING
 	push	{r0, lr}
@@ -93,3 +97,4 @@ SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 _irq_disabled:
 
 	bx	lr
+#endif

arch/arm/core/cortex_m/cpu_idle.c

@@ -53,6 +53,7 @@ void z_arm_cpu_idle_init(void)
 } while (false)
 #endif
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 void arch_cpu_idle(void)
 {
 #if defined(CONFIG_TRACING)
@@ -96,7 +97,9 @@ void arch_cpu_idle(void)
 	__enable_irq();
 	__ISB();
 }
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 void arch_cpu_atomic_idle(unsigned int key)
 {
 #if defined(CONFIG_TRACING)
@@ -135,3 +138,4 @@ void arch_cpu_atomic_idle(unsigned int key)
 	__enable_irq();
 #endif
 }
+#endif

arch/arm64/core/cpu_idle.S

@@ -13,7 +13,7 @@
 #include <zephyr/arch/cpu.h>
 
 _ASM_FILE_PROLOGUE
-
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 GTEXT(arch_cpu_idle)
 SECTION_FUNC(TEXT, arch_cpu_idle)
 #ifdef CONFIG_TRACING
@@ -25,7 +25,9 @@ SECTION_FUNC(TEXT, arch_cpu_idle)
 	wfi
 	msr	daifclr, #(DAIFCLR_IRQ_BIT)
 	ret
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 GTEXT(arch_cpu_atomic_idle)
 SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 #ifdef CONFIG_TRACING
@@ -41,3 +43,5 @@ SECTION_FUNC(TEXT, arch_cpu_atomic_idle)
 	msr	daifclr, #(DAIFCLR_IRQ_BIT)
 _irq_disabled:
 	ret
+
+#endif

arch/mips/core/cpu_idle.c

@@ -19,12 +19,16 @@ static ALWAYS_INLINE void mips_idle(unsigned int key)
 	__asm__ volatile("wait");
 }
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 void arch_cpu_idle(void)
 {
 	mips_idle(1);
 }
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 void arch_cpu_atomic_idle(unsigned int key)
 {
 	mips_idle(key);
 }
+#endif

arch/nios2/core/cpu_idle.c

@@ -7,18 +7,22 @@
 #include <zephyr/kernel.h>
 #include <zephyr/kernel_structs.h>
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 void arch_cpu_idle(void)
 {
 	/* Do nothing but unconditionally unlock interrupts and return to the
 	 * caller. This CPU does not have any kind of power saving instruction.
 	 */
 	irq_unlock(NIOS2_STATUS_PIE_MSK);
 }
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 void arch_cpu_atomic_idle(unsigned int key)
 {
 	/* Do nothing but restore IRQ state. This CPU does not have any
 	 * kind of power saving instruction.
 	 */
 	irq_unlock(key);
 }
+#endif

arch/riscv/core/cpu_idle.c

@@ -7,16 +7,20 @@
 #include <zephyr/irq.h>
 #include <zephyr/tracing/tracing.h>
 
-void __weak arch_cpu_idle(void)
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
+void arch_cpu_idle(void)
 {
 	sys_trace_idle();
 	__asm__ volatile("wfi");
 	irq_unlock(MSTATUS_IEN);
 }
+#endif
 
-void __weak arch_cpu_atomic_idle(unsigned int key)
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
+void arch_cpu_atomic_idle(unsigned int key)
 {
 	sys_trace_idle();
 	__asm__ volatile("wfi");
 	irq_unlock(key);
 }
+#endif

arch/x86/core/cpuhalt.c

@@ -7,6 +7,7 @@
 #include <zephyr/tracing/tracing.h>
 #include <zephyr/arch/cpu.h>
 
+#ifndef CONFIG_ARCH_CPU_IDLE_CUSTOM
 __pinned_func
 void arch_cpu_idle(void)
 {
@@ -15,7 +16,9 @@ void arch_cpu_idle(void)
 	    "sti\n\t"
 	    "hlt\n\t");
 }
+#endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 __pinned_func
 void arch_cpu_atomic_idle(unsigned int key)
 {
@@ -42,3 +45,4 @@ void arch_cpu_atomic_idle(unsigned int key)
 		__asm__ volatile("cli");
 	}
 }
+#endif

arch/xtensa/core/cpu_idle.c

@@ -14,10 +14,12 @@ void arch_cpu_idle(void)
 }
 #endif
 
+#ifndef CONFIG_ARCH_CPU_ATOMIC_IDLE_CUSTOM
 void arch_cpu_atomic_idle(unsigned int key)
 {
 	sys_trace_idle();
 	__asm__ volatile ("waiti 0\n\t"
 			  "wsr.ps %0\n\t"
 			  "rsync" :: "a"(key));
 }
+#endif