Changeset e1b6742 in mainline for uspace/app/tasks/tasks.c

Timestamp:

2010-04-18T12:17:11Z (15 years ago)

Author:

Martin Decky <martin@…>

Branches:

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

Children:

e535eeb

Parents:

bbda5ab

Message:

export threads to user space
the "tasks" command can now print all threads or threads belonging to a task

File:

• uspace/app/tasks/tasks.c (modified) (10 diffs)

uspace/app/tasks/tasks.c

@@ -1 +1 @@
 /*
  * Copyright (c) 2010 Stanislav Kozina
+ * Copyright (c) 2010 Martin Decky
  * All rights reserved.
  *
@@ -43 +44 @@
 #include <malloc.h>
 #include <inttypes.h>
+#include <bool.h>
 #include <arg_parse.h>
 #include "func.h"
@@ -53 +55 @@
 #define PRINT_LOAD1(x)  ((x) >> 11)
 #define PRINT_LOAD2(x)  (((x) & 0x7ff) / 2)
-
-/** Thread states */
-//static const char *thread_states[] = {
-//      "Invalid",
-//      "Running",
-//      "Sleeping",
-//      "Ready",
-//      "Entering",
-//      "Exiting",
-//      "Lingering"
-//};
 
 static void list_tasks(void)
@@ -90 +81 @@
                         
                         printf("%8" PRIu64 "%8u %8" PRIu64"%c %12"
-                            PRIu64"%c %12" PRIu64"%c %s\n", ids[i], stats_task->threads,
+                            PRIu64 "%c %12" PRIu64 "%c %s\n", ids[i], stats_task->threads,
                             virtmem, vmsuffix, ucycles, usuffix, kcycles, ksuffix,
                             stats_task->name);
@@ -102 +93 @@
 }
 
-static void list_threads(task_id_t task_id)
-{
-        /* TODO:
-        size_t thread_count = THREAD_COUNT;
-        thread_info_t *threads = malloc(thread_count * sizeof(thread_info_t));
-        size_t result = get_task_threads(threads, sizeof(thread_info_t) * thread_count);
-
-        while (result > thread_count) {
-                thread_count *= 2;
-                threads = realloc(threads, thread_count * sizeof(thread_info_t));
-                result = get_task_threads(threads, sizeof(thread_info_t) * thread_count);
-        }
-
-        if (result == 0) {
-                printf("No task with given pid!\n");
-                exit(1);
-        }
-
-        size_t i;
+static void list_threads(task_id_t task_id, bool all)
+{
+        size_t count;
+        thread_id_t *ids =
+            (thread_id_t *) stats_get_threads(&count);
+        
+        if (ids == NULL) {
+                fprintf(stderr, "%s: Unable to get threads\n", NAME);
+                return;
+        }
+        
         printf("    ID    State  CPU   Prio    [k]uCycles    [k]kcycles   Cycle fault\n");
-        for (i = 0; i < result; ++i) {
-                if (threads[i].taskid != taskid) {
-                        continue;
-                }
-                uint64_t ucycles, kcycles;
-                char usuffix, ksuffix;
-                order(threads[i].ucycles, &ucycles, &usuffix);
-                order(threads[i].kcycles, &kcycles, &ksuffix);
-                printf("%6llu %-8s %4u %6d %12llu%c %12llu%c\n", threads[i].tid,
-                        thread_states[threads[i].state], threads[i].cpu,
-                        threads[i].priority, ucycles, usuffix,
-                        kcycles, ksuffix);
-        }
-
-        free(threads); */
+        size_t i;
+        for (i = 0; i < count; i++) {
+                stats_thread_t *stats_thread = stats_get_thread(ids[i]);
+                if (stats_thread != NULL) {
+                        if ((all) || (stats_thread->task_id == task_id)) {
+                                uint64_t ucycles, kcycles;
+                                char usuffix, ksuffix;
+                                
+                                order(stats_thread->ucycles, &ucycles, &usuffix);
+                                order(stats_thread->kcycles, &kcycles, &ksuffix);
+                                
+                                if (stats_thread->on_cpu) {
+                                        printf("%8" PRIu64 " %-8s %4u %6d %12"
+                                            PRIu64"%c %12" PRIu64"%c\n", ids[i],
+                                            thread_get_state(stats_thread->state),
+                                            stats_thread->cpu, stats_thread->priority,
+                                            ucycles, usuffix, kcycles, ksuffix);
+                                } else {
+                                        printf("%8" PRIu64 " %-8s ---- %6d %12"
+                                            PRIu64"%c %12" PRIu64"%c\n", ids[i],
+                                            thread_get_state(stats_thread->state),
+                                            stats_thread->priority,
+                                            ucycles, usuffix, kcycles, ksuffix);
+                                }
+                        }
+                        
+                        free(stats_thread);
+                } else if (all)
+                        printf("%8" PRIu64 "\n", ids[i]);
+        }
+        
+        free(ids);
 }
 
@@ -190 +190 @@
 {
         printf(
-            "Usage: tasks [-t task_id] [-l] [-c]\n" \
+            "Usage: tasks [-t task_id] [-a] [-l] [-c]\n" \
             "\n" \
             "Options:\n" \
@@ -197 +197 @@
             "\t\tList threads of the given task\n" \
             "\n" \
+            "\t-a\n" \
+            "\t--all\n" \
+            "\t\tList all threads\n" \
+            "\n" \
             "\t-l\n" \
             "\t--load\n" \
@@ -217 +221 @@
         bool toggle_tasks = true;
         bool toggle_threads = false;
+        bool toggle_all = false;
         bool toggle_load = false;
         bool toggle_cpus = false;
         
-        task_id_t task_id;
+        task_id_t task_id = 0;
         
         int i;
@@ -230 +235 @@
                         usage();
                         return 0;
+                }
+                
+                /* All threads */
+                if ((off = arg_parse_short_long(argv[i], "-a", "--all")) != -1) {
+                        toggle_tasks = false;
+                        toggle_threads = true;
+                        toggle_all = true;
+                        continue;
                 }
                 
@@ -268 +281 @@
         
         if (toggle_threads)
-                list_threads(task_id);
+                list_threads(task_id, toggle_all);
         
         if (toggle_load)