Changes in uspace/app/trace/trace.c [fafb8e5:01900b6] in mainline

File:

• uspace/app/trace/trace.c (modified) (13 diffs)

uspace/app/trace/trace.c

@@ -47 +47 @@
 #include <mem.h>
 #include <str.h>
-#include <loader/loader.h>
 #include <io/console.h>
 #include <io/keycode.h>
@@ -86 +85 @@
 void thread_trace_start(uintptr_t thread_hash);
 
+static char *cmd_path;
+static char **cmd_args;
+
 static task_id_t task_id;
-static loader_t *task_ldr;
+static task_wait_t task_w;
 static bool task_wait_for;
 
@@ -93 +95 @@
 display_mask_t display_mask;
 
-static errno_t program_run_fibril(void *arg);
 static errno_t cev_fibril(void *arg);
-
-static void program_run(void)
-{
-        fid_t fid;
-
-        fid = fibril_create(program_run_fibril, NULL);
-        if (fid == 0) {
-                printf("Error creating fibril\n");
-                exit(1);
-        }
-
-        fibril_add_ready(fid);
-}
 
 static void cev_fibril_start(void)
@@ -122 +110 @@
 }
 
-static errno_t program_run_fibril(void *arg)
-{
-        errno_t rc;
-
-        /*
-         * This must be done in background as it will block until
-         * we let the task reply to this call.
-         */
-        rc = loader_run(task_ldr);
+static errno_t program_run(void)
+{
+        errno_t rc;
+
+        rc = task_spawnv_debug(&task_id, &task_w, cmd_path,
+            (const char *const *)cmd_args, &sess);
+
+        if (rc == ENOTSUP) {
+                printf("You do not have userspace debugging support "
+                    "compiled in the kernel.\n");
+                printf("Compile kernel with 'Support for userspace debuggers' "
+                    "(CONFIG_UDEBUG) enabled.\n");
+        }
+
         if (rc != EOK) {
-                printf("Error running program\n");
-                exit(1);
-        }
-
-        task_ldr = NULL;
-
-        printf("program_run_fibril exiting\n");
-        return 0;
+                printf("Error running program (%s)\n", str_error_name(rc));
+                return rc;
+        }
+
+        return EOK;
 }
 
 static errno_t connect_task(task_id_t task_id)
 {
-        async_sess_t *ksess = async_connect_kbox(task_id);
-
-        if (!ksess) {
-                if (errno == ENOTSUP) {
-                        printf("You do not have userspace debugging support "
-                            "compiled in the kernel.\n");
-                        printf("Compile kernel with 'Support for userspace debuggers' "
-                            "(CONFIG_UDEBUG) enabled.\n");
-                        return errno;
-                }
-
-                printf("Error connecting\n");
-                printf("ipc_connect_task(%" PRIu64 ") -> %s ", task_id, str_error_name(errno));
-                return errno;
-        }
-
-        errno_t rc = udebug_begin(ksess);
-        if (rc != EOK) {
-                printf("udebug_begin() -> %s\n", str_error_name(rc));
-                return rc;
-        }
-
-        rc = udebug_set_evmask(ksess, UDEBUG_EM_ALL);
+        errno_t rc;
+        bool debug_started = false;
+        bool wait_set_up = false;
+
+        if (sess == NULL) {
+                sess = async_connect_kbox(task_id, &rc);
+                if (sess == NULL) {
+                        printf("Error connecting to task %" PRIu64 ".\n",
+                            task_id);
+                        goto error;
+                }
+
+                rc = udebug_begin(sess);
+                if (rc != EOK) {
+                        printf("Error starting debug session.\n");
+                        goto error;
+                }
+
+                debug_started = true;
+
+                rc = task_setup_wait(task_id, &task_w);
+                if (rc != EOK) {
+                        printf("Error setting up wait for task termination.\n");
+                        goto error;
+                }
+
+                wait_set_up = true;
+        }
+
+        rc = udebug_set_evmask(sess, UDEBUG_EM_ALL);
         if (rc != EOK) {
                 printf("udebug_set_evmask(0x%x) -> %s\n ", UDEBUG_EM_ALL, str_error_name(rc));
@@ -172 +169 @@
         }
 
-        sess = ksess;
-        return 0;
+        return EOK;
+error:
+        if (wait_set_up)
+                task_cancel_wait(&task_w);
+        if (debug_started)
+                udebug_end(sess);
+        if (sess != NULL)
+                async_hangup(sess);
+        return rc;
 }
 
@@ -198 +202 @@
         printf("\ntotal of %zu threads\n", tb_needed / sizeof(uintptr_t));
 
-        return 0;
+        return EOK;
 }
 
@@ -488 +492 @@
 
         printf("Finished tracing thread [%d].\n", thread_id);
-        return 0;
+        return EOK;
 }
 
@@ -502 +506 @@
         }
         fibril_add_ready(fid);
-}
-
-static loader_t *preload_task(const char *path, char **argv,
-    task_id_t *task_id)
-{
-        loader_t *ldr;
-        errno_t rc;
-
-        /* Spawn a program loader */
-        ldr = loader_connect();
-        if (ldr == NULL)
-                return NULL;
-
-        /* Get task ID. */
-        rc = loader_get_task_id(ldr, task_id);
-        if (rc != EOK)
-                goto error;
-
-        /* Send program. */
-        rc = loader_set_program_path(ldr, path);
-        if (rc != EOK)
-                goto error;
-
-        /* Send arguments */
-        rc = loader_set_args(ldr, (const char **) argv);
-        if (rc != EOK)
-                goto error;
-
-        /* Send default files */
-        int fd_root;
-        int fd_stdin;
-        int fd_stdout;
-        int fd_stderr;
-
-        fd_root = vfs_root();
-        if (fd_root >= 0) {
-                rc = loader_add_inbox(ldr, "root", fd_root);
-                vfs_put(fd_root);
-                if (rc != EOK)
-                        goto error;
-        }
-
-        if ((stdin != NULL) && (vfs_fhandle(stdin, &fd_stdin) == EOK)) {
-                rc = loader_add_inbox(ldr, "stdin", fd_stdin);
-                if (rc != EOK)
-                        goto error;
-        }
-
-        if ((stdout != NULL) && (vfs_fhandle(stdout, &fd_stdout) == EOK)) {
-                rc = loader_add_inbox(ldr, "stdout", fd_stdout);
-                if (rc != EOK)
-                        goto error;
-        }
-
-        if ((stderr != NULL) && (vfs_fhandle(stderr, &fd_stderr) == EOK)) {
-                rc = loader_add_inbox(ldr, "stderr", fd_stderr);
-                if (rc != EOK)
-                        goto error;
-        }
-
-        /* Load the program. */
-        rc = loader_load_program(ldr);
-        if (rc != EOK)
-                goto error;
-
-        /* Success */
-        return ldr;
-
-        /* Error exit */
-error:
-        loader_abort(ldr);
-        return NULL;
 }
 
@@ -807 +739 @@
                                 ++argv;
                                 task_id = strtol(*argv, &err_p, 10);
-                                task_ldr = NULL;
                                 task_wait_for = false;
                                 if (*err_p) {
@@ -848 +779 @@
                 printf("'%s'\n", *cp++);
 
-        task_ldr = preload_task(*argv, argv, &task_id);
+        cmd_path = *argv;
+        cmd_args = argv;
         task_wait_for = true;
 
@@ -869 +801 @@
 
         main_init();
+
+        if (cmd_path != NULL)
+                program_run();
 
         rc = connect_task(task_id);
@@ -878 +813 @@
         printf("Connected to task %" PRIu64 ".\n", task_id);
 
-        if (task_ldr != NULL)
-                program_run();
-
         cev_fibril_start();
         trace_task(task_id);
@@ -887 +819 @@
                 printf("Waiting for task to exit.\n");
 
-                rc = task_wait_task_id(task_id, &texit, &retval);
+                rc = task_wait(&task_w, &texit, &retval);
                 if (rc != EOK) {
                         printf("Failed waiting for task.\n");