Subversion Repositories HelenOS

/*

 * Copyright (c) 2008 Jiri Svoboda

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * - Redistributions of source code must retain the above copyright

 *   notice, this list of conditions and the following disclaimer.

 * - Redistributions in binary form must reproduce the above copyright

 *   notice, this list of conditions and the following disclaimer in the

 *   documentation and/or other materials provided with the distribution.

 * - The name of the author may not be used to endorse or promote products

 *   derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

/** @addtogroup trace

 * @{

 */

/** @file

 */

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <ipc/ipc.h>

#include <fibril.h>

#include <errno.h>

#include <udebug.h>

#include <async.h>

#include <task.h>

#include <mem.h>

#include <string.h>

#include <bool.h>

#include <loader/loader.h>

#include <io/console.h>

#include <io/keycode.h>

#include <fibril_sync.h>

#include <libc.h>

// Temporary: service and method names

#include "proto.h"

#include <ipc/services.h>

#include "../../srv/vfs/vfs.h"

#include <ipc/console.h>

#include "syscalls.h"

#include "ipcp.h"

#include "errors.h"

#include "trace.h"

#define THBUF_SIZE 64

uintptr_t thread_hash_buf[THBUF_SIZE];

int n_threads;

int next_thread_id;

int phoneid;

bool abort_trace;

uintptr_t thash;

static bool paused;

static fibril_condvar_t state_cv;

static fibril_mutex_t state_lock;

static bool cev_valid;

static console_event_t cev;

void thread_trace_start(uintptr_t thread_hash);

static proto_t *proto_console;

static task_id_t task_id;

static loader_t *task_ldr;

static bool task_wait_for;

/** Combination of events/data to print. */

display_mask_t display_mask;

static int program_run_fibril(void *arg);

static int 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)

{

    fid_t fid;

    fid = fibril_create(cev_fibril, NULL);

    if (fid == 0) {

        printf("Error creating fibril\n");

        exit(1);

    }

    fibril_add_ready(fid);

}

static int program_run_fibril(void *arg)

{

    int 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);

    if (rc != 0) {

        printf("Error running program\n");

        exit(1);

    }

    free(task_ldr);

    task_ldr = NULL;

    printf("program_run_fibril exiting\n");

    return 0;

}

static int connect_task(task_id_t task_id)

{

    int rc;

    rc = ipc_connect_kbox(task_id);

    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");

        return rc;

    }

    if (rc < 0) {

        printf("Error connecting\n");

        printf("ipc_connect_task(%lld) -> %d ", task_id, rc);

        return rc;

    }

    phoneid = rc;

    rc = udebug_begin(phoneid);

    if (rc < 0) {

        printf("udebug_begin() -> %d\n", rc);

        return rc;

    }

    rc = udebug_set_evmask(phoneid, UDEBUG_EM_ALL);

    if (rc < 0) {

        printf("udebug_set_evmask(0x%x) -> %d\n ", UDEBUG_EM_ALL, rc);

        return rc;

    }

    return 0;

}

static int get_thread_list(void)

{

    int rc;

    size_t tb_copied;

    size_t tb_needed;

    int i;

    rc = udebug_thread_read(phoneid, thread_hash_buf,

        THBUF_SIZE*sizeof(unsigned), &tb_copied, &tb_needed);

    if (rc < 0) {

        printf("udebug_thread_read() -> %d\n", rc);

        return rc;

    }

    n_threads = tb_copied / sizeof(uintptr_t);

    printf("Threads:");

    for (i = 0; i < n_threads; i++) {

        printf(" [%d] (hash 0x%lx)", 1+i, thread_hash_buf[i]);

    }

    printf("\ntotal of %u threads\n", tb_needed / sizeof(uintptr_t));

    return 0;

}

void val_print(sysarg_t val, val_type_t v_type)

{

    switch (v_type) {

    case V_VOID:

        printf("<void>");

        break;

    case V_INTEGER:

        printf("%ld", val);

        break;

    case V_HASH:

    case V_PTR:

        printf("0x%08lx", val);

        break;

    case V_ERRNO:

        if (val >= -15 && val <= 0) {

            printf("%ld %s (%s)", val,

                err_desc[-val].name,

                err_desc[-val].desc);

        } else {

            printf("%ld", val);

        }

        break;

    case V_INT_ERRNO:

        if (val >= -15 && val < 0) {

            printf("%ld %s (%s)", val,

                err_desc[-val].name,

                err_desc[-val].desc);

        } else {

            printf("%ld", val);

        }

        break;

    case V_CHAR:

        if (val >= 0x20 && val < 0x7f) {

            printf("'%c'", val);

        } else {

            switch (val) {

            case '\a': printf("'\\a'"); break;

            case '\b': printf("'\\b'"); break;

            case '\n': printf("'\\n'"); break;

            case '\r': printf("'\\r'"); break;

            case '\t': printf("'\\t'"); break;

            case '\\': printf("'\\\\'"); break;

            default: printf("'\\x%02lX'", val); break;

            }

        }

        break;

    }

}

static void print_sc_retval(sysarg_t retval, val_type_t val_type)

{

    printf(" -> ");

    val_print(retval, val_type);

    putchar('\n');

}

static void print_sc_args(sysarg_t *sc_args, int n)

{

    int i;

    putchar('(');

    if (n > 0) printf("%ld", sc_args[0]);

    for (i = 1; i < n; i++) {

        printf(", %ld", sc_args[i]);

    }

    putchar(')');

}

static void sc_ipc_call_async_fast(sysarg_t *sc_args, sysarg_t sc_rc)

{

    ipc_call_t call;

    ipcarg_t phoneid;

    if (sc_rc == IPC_CALLRET_FATAL || sc_rc == IPC_CALLRET_TEMPORARY)

        return;

    phoneid = sc_args[0];

    IPC_SET_METHOD(call, sc_args[1]);

    IPC_SET_ARG1(call, sc_args[2]);

    IPC_SET_ARG2(call, sc_args[3]);

    IPC_SET_ARG3(call, sc_args[4]);

    IPC_SET_ARG4(call, sc_args[5]);

    IPC_SET_ARG5(call, 0);

    ipcp_call_out(phoneid, &call, sc_rc);

}

static void sc_ipc_call_async_slow(sysarg_t *sc_args, sysarg_t sc_rc)

{

    ipc_call_t call;

    int rc;

    if (sc_rc == IPC_CALLRET_FATAL || sc_rc == IPC_CALLRET_TEMPORARY)

        return;

    memset(&call, 0, sizeof(call));

    rc = udebug_mem_read(phoneid, &call.args, sc_args[1], sizeof(call.args));

    if (rc >= 0) {

        ipcp_call_out(sc_args[0], &call, sc_rc);

    }

}

static void sc_ipc_call_sync_fast(sysarg_t *sc_args)

{

    ipc_call_t question, reply;

    int rc;

    int phoneidx;

//  printf("sc_ipc_call_sync_fast()\n");

    phoneidx = sc_args[0];

    IPC_SET_METHOD(question, sc_args[1]);

    IPC_SET_ARG1(question, sc_args[2]);

    IPC_SET_ARG2(question, sc_args[3]);

    IPC_SET_ARG3(question, sc_args[4]);

    IPC_SET_ARG4(question, 0);

    IPC_SET_ARG5(question, 0);

//  printf("memset\n");

    memset(&reply, 0, sizeof(reply));

//  printf("udebug_mem_read(phone=%d, buffer_ptr=%u, src_addr=%d, n=%d\n",

//      phoneid, &reply.args, sc_args[5], sizeof(reply.args));

    rc = udebug_mem_read(phoneid, &reply.args, sc_args[5], sizeof(reply.args));

//  printf("dmr->%d\n", rc);

    if (rc < 0) return;

//  printf("call ipc_call_sync\n");

    ipcp_call_sync(phoneidx, &question, &reply);

}

static void sc_ipc_call_sync_slow(sysarg_t *sc_args)

{

    ipc_call_t question, reply;

    int rc;

    memset(&question, 0, sizeof(question));

    rc = udebug_mem_read(phoneid, &question.args, sc_args[1], sizeof(question.args));

    printf("dmr->%d\n", rc);

    if (rc < 0) return;

    memset(&reply, 0, sizeof(reply));

    rc = udebug_mem_read(phoneid, &reply.args, sc_args[2], sizeof(reply.args));

    printf("dmr->%d\n", rc);

    if (rc < 0) return;

    ipcp_call_sync(sc_args[0], &question, &reply);

}

static void sc_ipc_wait(sysarg_t *sc_args, int sc_rc)

{

    ipc_call_t call;

    int rc;

    if (sc_rc == 0) return;

    memset(&call, 0, sizeof(call));

    rc = udebug_mem_read(phoneid, &call, sc_args[0], sizeof(call));

//  printf("udebug_mem_read(phone %d, dest %d, app-mem src %d, size %d -> %d\n",

//      phoneid, (int)&call, sc_args[0], sizeof(call), rc);

    if (rc >= 0) {

        ipcp_call_in(&call, sc_rc);

    }

}

static void event_syscall_b(unsigned thread_id, uintptr_t thread_hash,

    unsigned sc_id, sysarg_t sc_rc)

{

    sysarg_t sc_args[6];

    int rc;

    /* Read syscall arguments */

    rc = udebug_args_read(phoneid, thread_hash, sc_args);

    async_serialize_start();

//  printf("[%d] ", thread_id);

    if (rc < 0) {

        printf("error\n");

        async_serialize_end();

        return;

    }

    if ((display_mask & DM_SYSCALL) != 0) {

        /* Print syscall name and arguments */

        printf("%s", syscall_desc[sc_id].name);

        print_sc_args(sc_args, syscall_desc[sc_id].n_args);

    }

    async_serialize_end();

}

static void event_syscall_e(unsigned thread_id, uintptr_t thread_hash,

    unsigned sc_id, sysarg_t sc_rc)

{

    sysarg_t sc_args[6];

    int rv_type;

    int rc;

    /* Read syscall arguments */

    rc = udebug_args_read(phoneid, thread_hash, sc_args);

    async_serialize_start();

//  printf("[%d] ", thread_id);

    if (rc < 0) {

        printf("error\n");

        async_serialize_end();

        return;

    }

    if ((display_mask & DM_SYSCALL) != 0) {

        /* Print syscall return value */

        rv_type = syscall_desc[sc_id].rv_type;

        print_sc_retval(sc_rc, rv_type);

    }

    switch (sc_id) {

    case SYS_IPC_CALL_ASYNC_FAST:

        sc_ipc_call_async_fast(sc_args, sc_rc);

        break;

    case SYS_IPC_CALL_ASYNC_SLOW:

        sc_ipc_call_async_slow(sc_args, sc_rc);

        break;

    case SYS_IPC_CALL_SYNC_FAST:

        sc_ipc_call_sync_fast(sc_args);

        break;

    case SYS_IPC_CALL_SYNC_SLOW:

        sc_ipc_call_sync_slow(sc_args);

        break;

    case SYS_IPC_WAIT:

        sc_ipc_wait(sc_args, sc_rc);

        break;

    default:

        break;

    }

    async_serialize_end();

}

static void event_thread_b(uintptr_t hash)

{

    async_serialize_start();

    printf("New thread, hash 0x%lx\n", hash);

    async_serialize_end();

    thread_trace_start(hash);

}

static int trace_loop(void *thread_hash_arg)

{

    int rc;

    unsigned ev_type;

    uintptr_t thread_hash;

    unsigned thread_id;

    sysarg_t val0, val1;

    thread_hash = (uintptr_t)thread_hash_arg;

    thread_id = next_thread_id++;

    printf("Start tracing thread [%d] (hash 0x%lx).\n", thread_id, thread_hash);

    while (!abort_trace) {

        fibril_mutex_lock(&state_lock);

        if (paused) {

            printf("Thread [%d] paused. Press R to resume.\n",

                thread_id);

            while (paused)

                fibril_condvar_wait(&state_cv, &state_lock);

            printf("Thread [%d] resumed.\n", thread_id);

        }

        fibril_mutex_unlock(&state_lock);

        /* Run thread until an event occurs */

        rc = udebug_go(phoneid, thread_hash,

            &ev_type, &val0, &val1);

//      printf("rc = %d, ev_type=%d\n", rc, ev_type);

        if (ev_type == UDEBUG_EVENT_FINISHED) {

            /* Done tracing this thread */

            break;

        }

        if (rc >= 0) {

            switch (ev_type) {

            case UDEBUG_EVENT_SYSCALL_B:

                event_syscall_b(thread_id, thread_hash, val0, (int)val1);

                break;

            case UDEBUG_EVENT_SYSCALL_E:

                event_syscall_e(thread_id, thread_hash, val0, (int)val1);

                break;

            case UDEBUG_EVENT_STOP:

                printf("Stop event\n");

                fibril_mutex_lock(&state_lock);

                paused = true;

                fibril_mutex_unlock(&state_lock);

                break;

            case UDEBUG_EVENT_THREAD_B:

                event_thread_b(val0);

                break;

            case UDEBUG_EVENT_THREAD_E:

                printf("Thread 0x%lx exited.\n", val0);

                fibril_mutex_lock(&state_lock);

                abort_trace = true;

                fibril_condvar_broadcast(&state_cv);

                fibril_mutex_unlock(&state_lock);

                break;

            default:

                printf("Unknown event type %d.\n", ev_type);

                break;

            }

        }

    }

    printf("Finished tracing thread [%d].\n", thread_id);

    return 0;

}

void thread_trace_start(uintptr_t thread_hash)

{

    fid_t fid;

    thash = thread_hash;

    fid = fibril_create(trace_loop, (void *)thread_hash);

    if (fid == 0) {

        printf("Warning: Failed creating fibril\n");

    }

    fibril_add_ready(fid);

}

static loader_t *preload_task(const char *path, char *const argv[],

    task_id_t *task_id)

{

    loader_t *ldr;

    int rc;

    /* Spawn a program loader */   

    ldr = loader_connect();

    if (ldr == NULL)

        return 0;

    /* Get task ID. */

    rc = loader_get_task_id(ldr, task_id);

    if (rc != EOK)

        goto error;

    /* Send program pathname */

    rc = loader_set_pathname(ldr, path);

    if (rc != EOK)

        goto error;

    /* Send arguments */

    rc = loader_set_args(ldr, argv);

    if (rc != EOK)

        goto error;

    /* Send default files */

    fdi_node_t *files[4];

    fdi_node_t stdin_node;

    fdi_node_t stdout_node;

    fdi_node_t stderr_node;

    if ((stdin != NULL) && (fnode(stdin, &stdin_node) == EOK))

        files[0] = &stdin_node;

    else

        files[0] = NULL;

    if ((stdout != NULL) && (fnode(stdout, &stdout_node) == EOK))

        files[1] = &stdout_node;

    else

        files[1] = NULL;

    if ((stderr != NULL) && (fnode(stderr, &stderr_node) == EOK))

        files[2] = &stderr_node;

    else

        files[2] = NULL;

    files[3] = NULL;

    rc = loader_set_files(ldr, files);

    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);

    free(ldr);

    return NULL;

}

static int cev_fibril(void *arg)

{

    (void) arg;

    while (true) {

        fibril_mutex_lock(&state_lock);

        while (cev_valid)

            fibril_condvar_wait(&state_cv, &state_lock);

        fibril_mutex_unlock(&state_lock);

        if (!console_get_event(fphone(stdin), &cev))

            return -1;

        fibril_mutex_lock(&state_lock);

        cev_valid = true;

        fibril_condvar_broadcast(&state_cv);

        fibril_mutex_unlock(&state_lock);      

    }

}

static void trace_task(task_id_t task_id)

{

    console_event_t ev;

    bool done;

    int i;

    int rc;

    ipcp_init();

    /*

     * User apps now typically have console on phone 3.

     * (Phones 1 and 2 are used by the loader).

     */

    ipcp_connection_set(3, 0, proto_console);

    rc = get_thread_list();

    if (rc < 0) {

        printf("Failed to get thread list (error %d)\n", rc);

        return;

    }

    abort_trace = false;

    for (i = 0; i < n_threads; i++) {

        thread_trace_start(thread_hash_buf[i]);

    }

    done = false;

    while (!done) {

        fibril_mutex_lock(&state_lock);

        while (!cev_valid && !abort_trace)

            fibril_condvar_wait(&state_cv, &state_lock);

        fibril_mutex_unlock(&state_lock);

        ev = cev;

        fibril_mutex_lock(&state_lock);

        cev_valid = false;

        fibril_condvar_broadcast(&state_cv);

        fibril_mutex_unlock(&state_lock);

        if (abort_trace)

            break;

        if (ev.type != KEY_PRESS)

            continue;

        switch (ev.key) {

        case KC_Q:

            done = true;

            break;

        case KC_P:

            printf("Pause...\n");

            rc = udebug_stop(phoneid, thash);

            if (rc != EOK)

                printf("Error: stop -> %d\n", rc);

            break;

        case KC_R:

            fibril_mutex_lock(&state_lock);

            paused = false;

            fibril_condvar_broadcast(&state_cv);

            fibril_mutex_unlock(&state_lock);

            printf("Resume...\n");

            break;

        }

    }

    printf("\nTerminate debugging session...\n");

    abort_trace = true;

    udebug_end(phoneid);

    ipc_hangup(phoneid);

    ipcp_cleanup();

    printf("Done\n");

    return;

}

static void main_init(void)

{

    proto_t *p;

    oper_t *o;

    val_type_t arg_def[OPER_MAX_ARGS] = {

        V_INTEGER,

        V_INTEGER,

        V_INTEGER,

        V_INTEGER,

        V_INTEGER      

    };

    val_type_t resp_def[OPER_MAX_ARGS] = {

        V_INTEGER,

        V_INTEGER,

        V_INTEGER,

        V_INTEGER,

        V_INTEGER

    };

    next_thread_id = 1;

    paused = false;

    cev_valid = false;

    fibril_mutex_initialize(&state_lock);

    fibril_condvar_initialize(&state_cv);

    proto_init();

    p = proto_new("vfs");

    o = oper_new("read", 1, arg_def, V_ERRNO, 1, resp_def);

    proto_add_oper(p, VFS_IN_READ, o);

    o = oper_new("write", 1, arg_def, V_ERRNO, 1, resp_def);

    proto_add_oper(p, VFS_IN_WRITE, o);

    o = oper_new("truncate", 5, arg_def, V_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_TRUNCATE, o);

    o = oper_new("mount", 2, arg_def, V_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_MOUNT, o);

/*  o = oper_new("unmount", 0, arg_def);

    proto_add_oper(p, VFS_IN_UNMOUNT, o);*/

    o = oper_new("open", 2, arg_def, V_INT_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_OPEN, o);

    o = oper_new("close", 1, arg_def, V_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_CLOSE, o);

    o = oper_new("seek", 3, arg_def, V_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_SEEK, o);

    o = oper_new("mkdir", 1, arg_def, V_ERRNO, 0, resp_def);

    proto_add_oper(p, VFS_IN_MKDIR, o);

    o = oper_new("unlink", 0, arg_def, V_ERRNO, 0, resp_def);