Changeset 6e49dab in mainline

Timestamp:

2024-01-14T18:24:05Z (12 months ago)

Author:

Jiří Zárevúcky <zarevucky.jiri@…>

Branches:

master

Children:

8996582

Parents:

23f36a3

git-author:

Jiří Zárevúcky <zarevucky.jiri@…> (2023-03-11 19:05:47)

git-committer:

Jiří Zárevúcky <zarevucky.jiri@…> (2024-01-14 18:24:05)

Message:

Extract post-thread cleanup into a separate function

File:

• kernel/generic/src/proc/scheduler.c (modified) (3 diffs)

kernel/generic/src/proc/scheduler.c

@@ -99 +99 @@
 }
 
-/** Take actions after THREAD had run.
- *
- * Perform actions that need to be
- * taken after the running thread
- * had been preempted by the scheduler.
- *
- * THREAD->lock is locked on entry
- *
- */
-static void after_thread_ran(void)
-{
-        after_thread_ran_arch();
-}
-
 #ifdef CONFIG_FPU_LAZY
 void scheduler_fpu_lazy_request(void)
@@ -374 +360 @@
 
         scheduler_locked(ipl);
+}
+
+static void cleanup_after_thread(thread_t *thread, state_t out_state)
+{
+        assert(CURRENT->mutex_locks == 0);
+        assert(interrupts_disabled());
+
+        int expected;
+
+        switch (out_state) {
+        case Running:
+                thread_ready(thread);
+                break;
+
+        case Exiting:
+                waitq_close(&thread->join_wq);
+
+                /*
+                 * Release the reference CPU has for the thread.
+                 * If there are no other references (e.g. threads calling join),
+                 * the thread structure is deallocated.
+                 */
+                thread_put(thread);
+                break;
+
+        case Sleeping:
+                expected = SLEEP_INITIAL;
+
+                /* Only set SLEEP_ASLEEP in sleep pad if it's still in initial state */
+                if (!atomic_compare_exchange_strong_explicit(&thread->sleep_state,
+                    &expected, SLEEP_ASLEEP,
+                    memory_order_acq_rel, memory_order_acquire)) {
+
+                        assert(expected == SLEEP_WOKE);
+                        /* The thread has already been woken up, requeue immediately. */
+                        thread_ready(thread);
+                }
+                break;
+
+        default:
+                /*
+                 * Entering state is unexpected.
+                 */
+                panic("tid%" PRIu64 ": unexpected state %s.",
+                    thread->tid, thread_states[thread->state]);
+                break;
+        }
 }
 
@@ -460 +493 @@
 
         if (THREAD) {
-                /* Must be run after the switch to scheduler stack */
-                after_thread_ran();
-
-                int expected;
-
-                switch (THREAD->state) {
-                case Running:
-                        irq_spinlock_unlock(&THREAD->lock, false);
-                        thread_ready(THREAD);
-                        break;
-
-                case Exiting:
-                        irq_spinlock_unlock(&THREAD->lock, false);
-                        waitq_close(&THREAD->join_wq);
-
-                        /*
-                         * Release the reference CPU has for the thread.
-                         * If there are no other references (e.g. threads calling join),
-                         * the thread structure is deallocated.
-                         */
-                        thread_put(THREAD);
-                        break;
-
-                case Sleeping:
-                        expected = SLEEP_INITIAL;
-
-                        /* Only set SLEEP_ASLEEP in sleep pad if it's still in initial state */
-                        if (atomic_compare_exchange_strong_explicit(&THREAD->sleep_state,
-                            &expected, SLEEP_ASLEEP,
-                            memory_order_acq_rel, memory_order_acquire)) {
-
-                                /* Prefer the thread after it's woken up. */
-                                THREAD->priority = -1;
-                                irq_spinlock_unlock(&THREAD->lock, false);
-                        } else {
-                                assert(expected == SLEEP_WOKE);
-                                /* The thread has already been woken up, requeue immediately. */
-                                irq_spinlock_unlock(&THREAD->lock, false);
-                                thread_ready(THREAD);
-                        }
-
-                        break;
-
-                default:
-                        /*
-                         * Entering state is unexpected.
-                         */
-                        panic("tid%" PRIu64 ": unexpected state %s.",
-                            THREAD->tid, thread_states[THREAD->state]);
-                        break;
+                after_thread_ran_arch();
+
+                state_t state = THREAD->state;
+
+                if (state == Sleeping) {
+                        /* Prefer the thread after it's woken up. */
+                        THREAD->priority = -1;
                 }
+
+                irq_spinlock_unlock(&THREAD->lock, false);
+
+                cleanup_after_thread(THREAD, state);
 
                 THREAD = NULL;