summaryrefslogtreecommitdiff
path: root/ports/risc-v32/gnu/src/tx_thread_schedule.S
diff options
context:
space:
mode:
Diffstat (limited to 'ports/risc-v32/gnu/src/tx_thread_schedule.S')
-rw-r--r--ports/risc-v32/gnu/src/tx_thread_schedule.S318
1 files changed, 318 insertions, 0 deletions
diff --git a/ports/risc-v32/gnu/src/tx_thread_schedule.S b/ports/risc-v32/gnu/src/tx_thread_schedule.S
new file mode 100644
index 00000000..1c235a2d
--- /dev/null
+++ b/ports/risc-v32/gnu/src/tx_thread_schedule.S
@@ -0,0 +1,318 @@
+/***************************************************************************
+ * Copyright (c) 2025 10xEngineers
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the MIT License which is available at
+ * https://opensource.org/licenses/MIT.
+ *
+ * SPDX-License-Identifier: MIT
+ **************************************************************************/
+
+
+/**************************************************************************/
+/**************************************************************************/
+/** */
+/** ThreadX Component */
+/** */
+/** Thread */
+/** */
+/**************************************************************************/
+/**************************************************************************/
+
+
+ .section .text
+/**************************************************************************/
+/* */
+/* FUNCTION RELEASE */
+/* */
+/* _tx_thread_schedule RISC-V32/GNU */
+/* 6.4.x */
+/* AUTHOR */
+/* */
+/* Akif Ejaz, 10xEngineers */
+/* */
+/* DESCRIPTION */
+/* */
+/* This function waits for a thread control block pointer to appear in */
+/* the _tx_thread_execute_ptr variable. Once a thread pointer appears */
+/* in the variable, the corresponding thread is resumed. */
+/* */
+/* INPUT */
+/* */
+/* None */
+/* */
+/* OUTPUT */
+/* */
+/* None */
+/* */
+/* CALLS */
+/* */
+/* None */
+/* */
+/* CALLED BY */
+/* */
+/* _tx_initialize_kernel_enter ThreadX entry function */
+/* _tx_thread_system_return Return to system from thread */
+/* _tx_thread_context_restore Restore thread's context */
+/* */
+/**************************************************************************/
+/* VOID _tx_thread_schedule(VOID)
+{ */
+ .global _tx_thread_schedule
+_tx_thread_schedule:
+
+ /* Enable interrupts. */
+
+ csrsi mstatus, 0x08 // Enable interrupts (MIE bit 3)
+
+ /* Wait for a thread to execute. */
+ /* do
+ { */
+_tx_thread_schedule_loop:
+
+ la t0, _tx_thread_execute_ptr // Pickup address of execute ptr
+ lw t1, 0(t0) // Pickup execute pointer
+ bnez t1, _tx_thread_ready_to_run // If non-NULL, a thread is ready to run
+
+#ifndef TX_NO_WFI
+ wfi // Wait for interrupt
+#endif
+ j _tx_thread_schedule_loop // Check again
+
+ /* }
+ while (_tx_thread_execute_ptr == TX_NULL); */
+
+_tx_thread_ready_to_run:
+
+ /* At this point, t1 contains the pointer to the thread to execute.
+ Lockout interrupts. */
+
+ csrci mstatus, 0x08 // Disable interrupts (MIE bit 3)
+
+ /* Check _tx_thread_execute_ptr again, in case an interrupt occurred
+ between the check and the disable. */
+
+ lw t1, 0(t0) // Pickup execute pointer
+ beqz t1, _tx_thread_schedule_loop // If NULL, go back to wait loop
+
+ /* Yes! We have a thread to execute. */
+ /* _tx_thread_current_ptr = _tx_thread_execute_ptr; */
+
+ la t0, _tx_thread_current_ptr // Pickup address of current thread
+ sw t1, 0(t0) // Setup current thread pointer
+
+ /* Increment the run count for this thread. */
+ /* _tx_thread_current_ptr -> tx_thread_run_count++; */
+
+ lw t2, 4(t1) // Pickup run count
+ addi t2, t2, 1 // Increment run count
+ sw t2, 4(t1) // Store run count
+
+ /* Setup time-slice values. */
+ /* _tx_timer_time_slice = _tx_thread_current_ptr -> tx_thread_time_slice; */
+
+ lw t2, 24(t1) // Pickup thread time-slice
+ la t3, _tx_timer_time_slice // Pickup address of time-slice
+ sw t2, 0(t3) // Setup time-slice
+
+ /* Call the thread execution enter function if enabled. */
+#ifdef TX_ENABLE_EXECUTION_CHANGE_NOTIFY
+
+ call _tx_execution_thread_enter // Call the thread execution enter function
+#endif
+
+ /* Switch to the thread's stack. */
+ /* sp = _tx_thread_current_ptr -> tx_thread_stack_ptr; */
+
+ lw sp, 8(t1) // Switch to thread stack
+
+ /* Determine the type of stack frame. */
+ /* if (*sp)
+ { */
+
+ lw t0, 0(sp) // Pickup stack type
+ beqz t0, _tx_thread_solicited_return // If 0, solicited return
+
+ /* Recover floating point registers. */
+#if defined(__riscv_float_abi_single)
+ flw f0, 31*4(sp) // Recover ft0
+ flw f1, 32*4(sp) // Recover ft1
+ flw f2, 33*4(sp) // Recover ft2
+ flw f3, 34*4(sp) // Recover ft3
+ flw f4, 35*4(sp) // Recover ft4
+ flw f5, 36*4(sp) // Recover ft5
+ flw f6, 37*4(sp) // Recover ft6
+ flw f7, 38*4(sp) // Recover ft7
+ flw f8, 39*4(sp) // Recover fs0
+ flw f9, 40*4(sp) // Recover fs1
+ flw f10, 41*4(sp) // Recover fa0
+ flw f11, 42*4(sp) // Recover fa1
+ flw f12, 43*4(sp) // Recover fa2
+ flw f13, 44*4(sp) // Recover fa3
+ flw f14, 45*4(sp) // Recover fa4
+ flw f15, 46*4(sp) // Recover fa5
+ flw f16, 47*4(sp) // Recover fa6
+ flw f17, 48*4(sp) // Recover fa7
+ flw f18, 49*4(sp) // Recover fs2
+ flw f19, 50*4(sp) // Recover fs3
+ flw f20, 51*4(sp) // Recover fs4
+ flw f21, 52*4(sp) // Recover fs5
+ flw f22, 53*4(sp) // Recover fs6
+ flw f23, 54*4(sp) // Recover fs7
+ flw f24, 55*4(sp) // Recover fs8
+ flw f25, 56*4(sp) // Recover fs9
+ flw f26, 57*4(sp) // Recover fs10
+ flw f27, 58*4(sp) // Recover fs11
+ flw f28, 59*4(sp) // Recover ft8
+ flw f29, 60*4(sp) // Recover ft9
+ flw f30, 61*4(sp) // Recover ft10
+ flw f31, 62*4(sp) // Recover ft11
+ lw t0, 63*4(sp) // Recover fcsr
+ csrw fcsr, t0 // Restore fcsr
+#elif defined(__riscv_float_abi_double)
+ fld f0, 31*4(sp) // Recover ft0
+ fld f1, 32*4(sp) // Recover ft1
+ fld f2, 33*4(sp) // Recover ft2
+ fld f3, 34*4(sp) // Recover ft3
+ fld f4, 35*4(sp) // Recover ft4
+ fld f5, 36*4(sp) // Recover ft5
+ fld f6, 37*4(sp) // Recover ft6
+ fld f7, 38*4(sp) // Recover ft7
+ fld f8, 39*4(sp) // Recover fs0
+ fld f9, 40*4(sp) // Recover fs1
+ fld f10, 41*4(sp) // Recover fa0
+ fld f11, 42*4(sp) // Recover fa1
+ fld f12, 43*4(sp) // Recover fa2
+ fld f13, 44*4(sp) // Recover fa3
+ fld f14, 45*4(sp) // Recover fa4
+ fld f15, 46*4(sp) // Recover fa5
+ fld f16, 47*4(sp) // Recover fa6
+ fld f17, 48*4(sp) // Recover fa7
+ fld f18, 49*4(sp) // Recover fs2
+ fld f19, 50*4(sp) // Recover fs3
+ fld f20, 51*4(sp) // Recover fs4
+ fld f21, 52*4(sp) // Recover fs5
+ fld f22, 53*4(sp) // Recover fs6
+ fld f23, 54*4(sp) // Recover fs7
+ fld f24, 55*4(sp) // Recover fs8
+ fld f25, 56*4(sp) // Recover fs9
+ fld f26, 57*4(sp) // Recover fs10
+ fld f27, 58*4(sp) // Recover fs11
+ fld f28, 59*4(sp) // Recover ft8
+ fld f29, 60*4(sp) // Recover ft9
+ fld f30, 61*4(sp) // Recover ft10
+ fld f31, 62*4(sp) // Recover ft11
+ lw t0, 63*4(sp) // Recover fcsr
+ csrw fcsr, t0 // Restore fcsr
+#endif
+
+ /* Recover standard registers. */
+
+ lw t0, 30*4(sp) // Recover mepc
+ csrw mepc, t0 // Setup mepc
+
+ li t0, 0x1880 // Prepare mstatus: MPP=Machine(0x1800) | MPIE(0x80)
+#if defined(__riscv_float_abi_single) || defined(__riscv_float_abi_double)
+ li t1, 0x2000 // Set FS bits for FP state
+ or t0, t0, t1
+#endif
+ csrw mstatus, t0 // Set mstatus
+
+ lw ra, 28*4(sp) // Recover return address
+ lw t0, 19*4(sp) // Recover t0
+ lw t1, 18*4(sp) // Recover t1
+ lw t2, 17*4(sp) // Recover t2
+ lw s0, 12*4(sp) // Recover s0
+ lw x9, 11*4(sp) // Recover s1
+ lw a0, 27*4(sp) // Recover a0
+ lw a1, 26*4(sp) // Recover a1
+ lw a2, 25*4(sp) // Recover a2
+ lw a3, 24*4(sp) // Recover a3
+ lw a4, 23*4(sp) // Recover a4
+ lw a5, 22*4(sp) // Recover a5
+ lw a6, 21*4(sp) // Recover a6
+ lw a7, 20*4(sp) // Recover a7
+ lw t3, 16*4(sp) // Recover t3
+ lw t4, 15*4(sp) // Recover t4
+ lw t5, 14*4(sp) // Recover t5
+ lw t6, 13*4(sp) // Recover t6
+ lw x18, 10*4(sp) // Recover s2
+ lw x19, 9*4(sp) // Recover s3
+ lw x20, 8*4(sp) // Recover s4
+ lw x21, 7*4(sp) // Recover s5
+ lw x22, 6*4(sp) // Recover s6
+ lw x23, 5*4(sp) // Recover s7
+ lw x24, 4*4(sp) // Recover s8
+ lw x25, 3*4(sp) // Recover s9
+ lw x26, 2*4(sp) // Recover s10
+ lw x27, 1*4(sp) // Recover s11
+
+#if defined(__riscv_float_abi_single) || defined(__riscv_float_abi_double)
+ addi sp, sp, 65*4 // Recover stack frame - with floating point enabled
+#else
+ addi sp, sp, 32*4 // Recover stack frame - without floating point enabled
+#endif
+ mret // Return to thread
+
+_tx_thread_solicited_return:
+
+ /* Recover floating point registers. */
+#if defined(__riscv_float_abi_single)
+ flw f8, 15*4(sp) // Recover fs0
+ flw f9, 16*4(sp) // Recover fs1
+ flw f18, 17*4(sp) // Recover fs2
+ flw f19, 18*4(sp) // Recover fs3
+ flw f20, 19*4(sp) // Recover fs4
+ flw f21, 20*4(sp) // Recover fs5
+ flw f22, 21*4(sp) // Recover fs6
+ flw f23, 22*4(sp) // Recover fs7
+ flw f24, 23*4(sp) // Recover fs8
+ flw f25, 24*4(sp) // Recover fs9
+ flw f26, 25*4(sp) // Recover fs10
+ flw f27, 26*4(sp) // Recover fs11
+ lw t0, 27*4(sp) // Recover fcsr
+ csrw fcsr, t0 // Restore fcsr
+#elif defined(__riscv_float_abi_double)
+ fld f8, 15*4(sp) // Recover fs0
+ fld f9, 16*4(sp) // Recover fs1
+ fld f18, 17*4(sp) // Recover fs2
+ fld f19, 18*4(sp) // Recover fs3
+ fld f20, 19*4(sp) // Recover fs4
+ fld f21, 20*4(sp) // Recover fs5
+ fld f22, 21*4(sp) // Recover fs6
+ fld f23, 22*4(sp) // Recover fs7
+ fld f24, 23*4(sp) // Recover fs8
+ fld f25, 24*4(sp) // Recover fs9
+ fld f26, 25*4(sp) // Recover fs10
+ fld f27, 26*4(sp) // Recover fs11
+ lw t0, 27*4(sp) // Recover fcsr
+ csrw fcsr, t0 // Restore fcsr
+#endif
+
+ /* Recover standard registers. */
+
+ lw t0, 14*4(sp) // Recover mstatus
+ csrw mstatus, t0 // Restore mstatus
+
+ lw ra, 13*4(sp) // Recover return address
+ lw s0, 12*4(sp) // Recover s0
+ lw s1, 11*4(sp) // Recover s1
+ lw x18, 10*4(sp) // Recover s2
+ lw x19, 9*4(sp) // Recover s3
+ lw x20, 8*4(sp) // Recover s4
+ lw x21, 7*4(sp) // Recover s5
+ lw x22, 6*4(sp) // Recover s6
+ lw x23, 5*4(sp) // Recover s7
+ lw x24, 4*4(sp) // Recover s8
+ lw x25, 3*4(sp) // Recover s9
+ lw x26, 2*4(sp) // Recover s10
+ lw x27, 1*4(sp) // Recover s11
+
+#if defined(__riscv_float_abi_single) || defined(__riscv_float_abi_double)
+ addi sp, sp, 29*4 // Recover stack frame - with floating point enabled
+#else
+ addi sp, sp, 16*4 // Recover stack frame - without floating point enabled
+#endif
+ ret // Return to thread
+
+/* } */