Changeset aab5e46 in mainline for kernel/generic/src

Timestamp:

2018-11-03T23:32:39Z (7 years ago)

Author:

Jiri Svoboda <jiri@…>

Branches:

lfn, master, serial, ticket/834-toolchain-update, topic/msim-upgrade, topic/simplify-dev-export

Children:

790f3a3

Parents:

ef1eab7

Message:

Thread and task iterator functions.

Location:

kernel/generic/src

Files:

• proc/task.c (modified) (5 diffs)
• proc/thread.c (modified) (4 diffs)
• sysinfo/stats.c (modified) (6 diffs)

kernel/generic/src/proc/task.c

@@ -107 +107 @@
 {
         size_t tasks_left;
-        odlink_t *odlink;
         task_t *task;
 
@@ -128 +127 @@
                 tasks_left = 0;
 
-                odlink = odict_first(&tasks);
-                while (odlink != NULL) {
-                        task = odict_get_instance(odlink, task_t, ltasks);
-
+                task = task_first();
+                while (task != NULL) {
                         if (task != TASK) {
                                 tasks_left++;
@@ -140 +137 @@
                         }
 
-                        odlink = odict_next(odlink, &tasks);
+                        task = task_next(task);
                 }
 
@@ -452 +449 @@
 }
 
+/** Get count of tasks.
+ *
+ * @return Number of tasks in the system
+ */
+size_t task_count(void)
+{
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&tasks_lock));
+
+        return odict_count(&tasks);
+}
+
+/** Get first task (task with lowest ID).
+ *
+ * @return Pointer to first task or @c NULL if there are none.
+ */
+task_t *task_first(void)
+{
+        odlink_t *odlink;
+
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&tasks_lock));
+
+        odlink = odict_first(&tasks);
+        if (odlink == NULL)
+                return NULL;
+
+        return odict_get_instance(odlink, task_t, ltasks);
+}
+
+/** Get next task (with higher task ID).
+ *
+ * @param cur Current task
+ * @return Pointer to next task or @c NULL if there are no more tasks.
+ */
+task_t *task_next(task_t *cur)
+{
+        odlink_t *odlink;
+
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&tasks_lock));
+
+        odlink = odict_next(&cur->ltasks, &tasks);
+        if (odlink == NULL)
+                return NULL;
+
+        return odict_get_instance(odlink, task_t, ltasks);
+}
+
 /** Get accounting data of given task.
  *
@@ -658 +704 @@
 #endif
 
-        odlink_t *odlink;
         task_t *task;
 
-        odlink = odict_first(&tasks);
-        while (odlink != NULL) {
-                task = odict_get_instance(odlink, task_t, ltasks);
+        task = task_first();
+        while (task != NULL) {
                 task_print(task, additional);
-                odlink = odict_next(odlink, &tasks);
+                task = task_next(task);
         }
 

kernel/generic/src/proc/thread.c

@@ -772 +772 @@
 void thread_print_list(bool additional)
 {
-        odlink_t *odlink;
         thread_t *thread;
 
@@ -796 +795 @@
 #endif
 
-        odlink = odict_first(&threads);
-        while (odlink != NULL) {
-                thread = odict_get_instance(odlink, thread_t, lthreads);
+        thread = thread_first();
+        while (thread != NULL) {
                 thread_print(thread, additional);
-                odlink = odict_next(odlink, &threads);
+                thread = thread_next(thread);
         }
 
@@ -860 +858 @@
 thread_t *thread_find_by_id(thread_id_t thread_id)
 {
-        odlink_t *odlink;
         thread_t *thread;
 
@@ -866 +863 @@
         assert(irq_spinlock_locked(&threads_lock));
 
-        odlink = odict_first(&threads);
-        while (odlink != NULL) {
-                thread = odict_get_instance(odlink, thread_t, lthreads);
+        thread = thread_first();
+        while (thread != NULL) {
                 if (thread->tid == thread_id)
                         return thread;
 
-                odlink = odict_next(odlink, &threads);
+                thread = thread_next(thread);
         }
 
         return NULL;
+}
+
+/** Get count of threads.
+ *
+ * @return Number of threads in the system
+ */
+size_t thread_count(void)
+{
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&threads_lock));
+
+        return odict_count(&threads);
+}
+
+/** Get first thread.
+ *
+ * @return Pointer to first thread or @c NULL if there are none.
+ */
+thread_t *thread_first(void)
+{
+        odlink_t *odlink;
+
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&threads_lock));
+
+        odlink = odict_first(&threads);
+        if (odlink == NULL)
+                return NULL;
+
+        return odict_get_instance(odlink, thread_t, lthreads);
+}
+
+/** Get next thread.
+ *
+ * @param cur Current thread
+ * @return Pointer to next thread or @c NULL if there are no more threads.
+ */
+thread_t *thread_next(thread_t *cur)
+{
+        odlink_t *odlink;
+
+        assert(interrupts_disabled());
+        assert(irq_spinlock_locked(&threads_lock));
+
+        odlink = odict_next(&cur->lthreads, &threads);
+        if (odlink == NULL)
+                return NULL;
+
+        return odict_get_instance(odlink, thread_t, lthreads);
 }
 

kernel/generic/src/sysinfo/stats.c

@@ -246 +246 @@
 
         /* Count the tasks */
-        size_t count = odict_count(&tasks);
+        size_t count = task_count();
 
         if (count == 0) {
@@ -271 +271 @@
         /* Gather the statistics for each task */
         size_t i = 0;
-        odlink_t *odlink = odict_first(&tasks);
-        while (odlink != NULL) {
-                task_t *task = odict_get_instance(odlink, task_t, ltasks);
-
+        task_t *task = task_first();
+        while (task != NULL) {
                 /* Interrupts are already disabled */
                 irq_spinlock_lock(&(task->lock), false);
@@ -283 +281 @@
 
                 irq_spinlock_unlock(&(task->lock), false);
-                odlink = odict_next(odlink, &tasks);
+                task = task_next(task);
         }
 
@@ -336 +334 @@
 
         /* Count the threads */
-        size_t count = odict_count(&threads);
+        size_t count = thread_count();
 
         if (count == 0) {
@@ -362 +360 @@
         size_t i = 0;
 
-        odlink_t *odlink = odict_first(&threads);
-        while (odlink != NULL) {
-                thread_t *thread = odict_get_instance(odlink, thread_t,
-                    lthreads);
-
+        thread_t *thread = thread_first();
+        while (thread != NULL) {
                 /* Interrupts are already disabled */
                 irq_spinlock_lock(&thread->lock, false);
@@ -376 +371 @@
                 irq_spinlock_unlock(&thread->lock, false);
 
-                odlink = odict_next(odlink, &threads);
+                thread = thread_next(thread);
         }