Changes in kernel/generic/src/proc/task.c [f35749e:5a5269d] in mainline

File:

• kernel/generic/src/proc/task.c (modified) (20 diffs)

kernel/generic/src/proc/task.c

@@ -1 +1 @@
 /*
- * Copyright (c) 2025 Jiri Svoboda
  * Copyright (c) 2010 Jakub Jermar
+ * Copyright (c) 2018 Jiri Svoboda
  * All rights reserved.
  *
@@ -104 +104 @@
  *
  */
-void task_done(task_t *cur_task)
+void task_done(void)
 {
         size_t tasks_left;
@@ -112 +112 @@
                 task_t *task_0 = ipc_box_0->task;
                 ipc_box_0 = NULL;
-
                 /*
                  * The first task is held by kinit(), we need to release it or
@@ -130 +129 @@
                 task = task_first();
                 while (task != NULL) {
-                        if (task != cur_task) {
+                        if (task != TASK) {
                                 tasks_left++;
 #ifdef CONFIG_DEBUG
@@ -159 +158 @@
                 return rc;
 
+        atomic_store(&task->refcount, 0);
         atomic_store(&task->lifecount, 0);
 
@@ -201 +201 @@
         if (!task)
                 return NULL;
-
-        refcount_init(&task->refcount);
 
         task_create_arch(task);
@@ -226 +224 @@
 
         task->answerbox.active = true;
-
-        task->debug_sections = NULL;
 
 #ifdef CONFIG_UDEBUG
@@ -270 +266 @@
  *
  */
-static void task_destroy(task_t *task)
+void task_destroy(task_t *task)
 {
         /*
@@ -301 +297 @@
 void task_hold(task_t *task)
 {
-        refcount_up(&task->refcount);
+        atomic_inc(&task->refcount);
 }
 
@@ -313 +309 @@
 void task_release(task_t *task)
 {
-        if (refcount_down(&task->refcount))
+        if ((atomic_predec(&task->refcount)) == 0)
                 task_destroy(task);
 }
@@ -389 +385 @@
         irq_spinlock_lock(&tasks_lock, true);
         irq_spinlock_lock(&TASK->lock, false);
+        irq_spinlock_lock(&threads_lock, false);
 
         /* Set task name */
         str_cpy(TASK->name, TASK_NAME_BUFLEN, namebuf);
 
+        irq_spinlock_unlock(&threads_lock, false);
         irq_spinlock_unlock(&TASK->lock, false);
         irq_spinlock_unlock(&tasks_lock, true);
@@ -418 +416 @@
 /** Find task structure corresponding to task ID.
  *
+ * The tasks_lock must be already held by the caller of this function and
+ * interrupts must be disabled.
+ *
  * @param id Task ID.
  *
- * @return Task reference or NULL if there is no such task ID.
+ * @return Task structure address or NULL if there is no such task ID.
  *
  */
 task_t *task_find_by_id(task_id_t id)
 {
-        task_t *task = NULL;
-
-        irq_spinlock_lock(&tasks_lock, true);
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&tasks_lock));
 
         odlink_t *odlink = odict_find_eq(&tasks, &id, NULL);
-        if (odlink != NULL) {
-                task = odict_get_instance(odlink, task_t, ltasks);
-
-                /*
-                 * The directory of tasks can't hold a reference, since that would
-                 * prevent task from ever being destroyed. That means we have to
-                 * check for the case where the task is already being destroyed, but
-                 * not yet removed from the directory.
-                 */
-                if (!refcount_try_up(&task->refcount))
-                        task = NULL;
-        }
-
-        irq_spinlock_unlock(&tasks_lock, true);
-
-        return task;
+        if (odlink != NULL)
+                return odict_get_instance(odlink, task_t, ltasks);
+
+        return NULL;
 }
 
@@ -518 +506 @@
         /* Current values of threads */
         list_foreach(task->threads, th_link, thread_t, thread) {
+                irq_spinlock_lock(&thread->lock, false);
+
                 /* Process only counted threads */
                 if (!thread->uncounted) {
@@ -525 +515 @@
                         }
 
-                        uret += atomic_time_read(&thread->ucycles);
-                        kret += atomic_time_read(&thread->kcycles);
+                        uret += thread->ucycles;
+                        kret += thread->kcycles;
                 }
+
+                irq_spinlock_unlock(&thread->lock, false);
         }
 
@@ -536 +528 @@
 static void task_kill_internal(task_t *task)
 {
-        irq_spinlock_lock(&task->lock, true);
+        irq_spinlock_lock(&task->lock, false);
+        irq_spinlock_lock(&threads_lock, false);
 
         /*
@@ -543 +536 @@
 
         list_foreach(task->threads, th_link, thread_t, thread) {
-                thread_interrupt(thread);
+                bool sleeping = false;
+
+                irq_spinlock_lock(&thread->lock, false);
+
+                thread->interrupted = true;
+                if (thread->state == Sleeping)
+                        sleeping = true;
+
+                irq_spinlock_unlock(&thread->lock, false);
+
+                if (sleeping)
+                        waitq_interrupt_sleep(thread);
         }
 
-        irq_spinlock_unlock(&task->lock, true);
+        irq_spinlock_unlock(&threads_lock, false);
+        irq_spinlock_unlock(&task->lock, false);
 }
 
@@ -564 +569 @@
                 return EPERM;
 
+        irq_spinlock_lock(&tasks_lock, true);
+
         task_t *task = task_find_by_id(id);
-        if (!task)
+        if (!task) {
+                irq_spinlock_unlock(&tasks_lock, true);
                 return ENOENT;
+        }
 
         task_kill_internal(task);
-        task_release(task);
+        irq_spinlock_unlock(&tasks_lock, true);
+
         return EOK;
 }
@@ -599 +609 @@
         }
 
+        irq_spinlock_lock(&tasks_lock, true);
         task_kill_internal(TASK);
+        irq_spinlock_unlock(&tasks_lock, true);
+
         thread_exit();
 }
@@ -628 +641 @@
         if (additional)
                 printf("%-8" PRIu64 " %9zu", task->taskid,
-                    atomic_load(&task->lifecount));
+                    atomic_load(&task->refcount));
         else
                 printf("%-8" PRIu64 " %-14s %-5" PRIu32 " %10p %10p"
@@ -640 +653 @@
                 printf("%-8" PRIu64 " %9" PRIu64 "%c %9" PRIu64 "%c "
                     "%9zu\n", task->taskid, ucycles, usuffix, kcycles,
-                    ksuffix, atomic_load(&task->lifecount));
+                    ksuffix, atomic_load(&task->refcount));
         else
                 printf("%-8" PRIu64 " %-14s %-5" PRIu32 " %18p %18p\n",