Subversion Repositories HelenOS

/*

 * Copyright (C) 2005 Jakub Jermar

 * 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.

 */

/**

 * This file is meant to contain all wrapper functions for

 * all kconsole commands. The point is in separating

 * kconsole specific wrappers from kconsole-unaware functions

 * from other subsystems.

 */

#include <console/cmd.h>

#include <console/kconsole.h>

#include <print.h>

#include <panic.h>

#include <typedefs.h>

#include <arch/types.h>

#include <list.h>

#include <arch.h>

#include <func.h>

#include <macros.h>

#include <debug.h>

#include <symtab.h>

#include <mm/tlb.h>

#include <arch/mm/tlb.h>

/** Data and methods for 'help' command. */

static int cmd_help(cmd_arg_t *argv);

static cmd_info_t help_info = {

    .name = "help",

    .description = "List of supported commands.",

    .func = cmd_help,

    .argc = 0

};

static cmd_info_t exit_info = {

    .name = "exit",

    .description ="Exit kconsole",

    .argc = 0

};

/** Data and methods for 'description' command. */

static int cmd_desc(cmd_arg_t *argv);

static void desc_help(void);

static char desc_buf[MAX_CMDLINE+1];

static cmd_arg_t desc_argv = {

    .type = ARG_TYPE_STRING,

    .buffer = desc_buf,

    .len = sizeof(desc_buf)

};

static cmd_info_t desc_info = {

    .name = "describe",

    .description = "Describe specified command.",

    .help = desc_help,

    .func = cmd_desc,

    .argc = 1,

    .argv = &desc_argv

};

/** Data and methods for 'symaddr' command. */

static int cmd_symaddr(cmd_arg_t *argv);

static char symaddr_buf[MAX_CMDLINE+1];

static cmd_arg_t symaddr_argv = {

    .type = ARG_TYPE_STRING,

    .buffer = symaddr_buf,

    .len = sizeof(symaddr_buf)

};

static cmd_info_t symaddr_info = {

    .name = "symaddr",

    .description = "Return symbol address.",

    .func = cmd_symaddr,

    .argc = 1,

    .argv = &symaddr_argv

};

static char set_buf[MAX_CMDLINE+1];

static int cmd_set4(cmd_arg_t *argv);

static cmd_arg_t set4_argv[] = {

    {

        .type = ARG_TYPE_STRING,

        .buffer = set_buf,

        .len = sizeof(set_buf)

    },

    {

        .type = ARG_TYPE_INT

    }

};

static cmd_info_t set4_info = {

    .name = "set4",

    .description = "set <dest_addr> <value> - 4byte version",

    .func = cmd_set4,

    .argc = 2,

    .argv = set4_argv

};

/** Data and methods for 'call0' command. */

static char call0_buf[MAX_CMDLINE+1];

static char carg1_buf[MAX_CMDLINE+1];

static char carg2_buf[MAX_CMDLINE+1];

static char carg3_buf[MAX_CMDLINE+1];

static int cmd_call0(cmd_arg_t *argv);

static cmd_arg_t call0_argv = {

    .type = ARG_TYPE_STRING,

    .buffer = call0_buf,

    .len = sizeof(call0_buf)

};

static cmd_info_t call0_info = {

    .name = "call0",

    .description = "call0 <function> -> call function().",

    .func = cmd_call0,

    .argc = 1,

    .argv = &call0_argv

};

/** Data and methods for 'call1' command. */

static int cmd_call1(cmd_arg_t *argv);

static cmd_arg_t call1_argv[] = {

    {

        .type = ARG_TYPE_STRING,

        .buffer = call0_buf,

        .len = sizeof(call0_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg1_buf,

        .len = sizeof(carg1_buf)

    }

};

static cmd_info_t call1_info = {

    .name = "call1",

    .description = "call1 <function> <arg1> -> call function(arg1).",

    .func = cmd_call1,

    .argc = 2,

    .argv = call1_argv

};

/** Data and methods for 'call2' command. */

static int cmd_call2(cmd_arg_t *argv);

static cmd_arg_t call2_argv[] = {

    {

        .type = ARG_TYPE_STRING,

        .buffer = call0_buf,

        .len = sizeof(call0_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg1_buf,

        .len = sizeof(carg1_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg2_buf,

        .len = sizeof(carg2_buf)

    }

};

static cmd_info_t call2_info = {

    .name = "call2",

    .description = "call2 <function> <arg1> <arg2> -> call function(arg1,arg2).",

    .func = cmd_call2,

    .argc = 3,

    .argv = call2_argv

};

/** Data and methods for 'call3' command. */

static int cmd_call3(cmd_arg_t *argv);

static cmd_arg_t call3_argv[] = {

    {

        .type = ARG_TYPE_STRING,

        .buffer = call0_buf,

        .len = sizeof(call0_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg1_buf,

        .len = sizeof(carg1_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg2_buf,

        .len = sizeof(carg2_buf)

    },

    {

        .type = ARG_TYPE_VAR,

        .buffer = carg3_buf,

        .len = sizeof(carg3_buf)

    }

};

static cmd_info_t call3_info = {

    .name = "call3",

    .description = "call3 <function> <arg1> <arg2> <arg3> -> call function(arg1,arg2,arg3).",

    .func = cmd_call3,

    .argc = 4,

    .argv = call3_argv

};

/** Data and methods for 'halt' command. */

static int cmd_halt(cmd_arg_t *argv);

static cmd_info_t halt_info = {

    .name = "halt",

    .description = "Halt the kernel.",

    .func = cmd_halt,

    .argc = 0

};

/** Data and methods for 'ptlb' command. */

static int cmd_ptlb(cmd_arg_t *argv);

cmd_info_t ptlb_info = {

    .name = "ptlb",

    .description = "Print TLB of current processor.",

    .help = NULL,

    .func = cmd_ptlb,

    .argc = 0,

    .argv = NULL

};

/** Initialize command info structure.

 *

 * @param cmd Command info structure.

 *

 */

void cmd_initialize(cmd_info_t *cmd)

{

    spinlock_initialize(&cmd->lock, "cmd");

    link_initialize(&cmd->link);

}

/** Initialize and register commands. */

void cmd_init(void)

{

    cmd_initialize(&help_info);

    if (!cmd_register(&help_info))

        panic("could not register command %s\n", help_info.name);

    cmd_initialize(&desc_info);

    if (!cmd_register(&desc_info))

        panic("could not register command %s\n", desc_info.name);

    cmd_initialize(&exit_info);

    if (!cmd_register(&exit_info))

        panic("could not register command %s\n", exit_info.name);

    cmd_initialize(&symaddr_info);

    if (!cmd_register(&symaddr_info))

        panic("could not register command %s\n", symaddr_info.name);

    cmd_initialize(&call0_info);

    if (!cmd_register(&call0_info))

        panic("could not register command %s\n", call0_info.name);

    cmd_initialize(&call1_info);

    if (!cmd_register(&call1_info))

        panic("could not register command %s\n", call1_info.name);

    cmd_initialize(&call2_info);

    if (!cmd_register(&call2_info))

        panic("could not register command %s\n", call2_info.name);

    cmd_initialize(&call3_info);

    if (!cmd_register(&call3_info))

        panic("could not register command %s\n", call3_info.name);

    cmd_initialize(&set4_info);

    if (!cmd_register(&set4_info))

        panic("could not register command %s\n", set4_info.name);

    cmd_initialize(&halt_info);

    if (!cmd_register(&halt_info))

        panic("could not register command %s\n", halt_info.name);

    cmd_initialize(&ptlb_info);

    if (!cmd_register(&ptlb_info))

        panic("could not register command %s\n", ptlb_info.name);

}

/** List supported commands.

 *

 * @param argv Argument vector.

 *

 * @return 0 on failure, 1 on success.

 */

int cmd_help(cmd_arg_t *argv)

{

    link_t *cur;

    ipl_t ipl;

    spinlock_lock(&cmd_lock);

    for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {

        cmd_info_t *hlp;

        hlp = list_get_instance(cur, cmd_info_t, link);

        spinlock_lock(&hlp->lock);

        printf("%s - %s\n", hlp->name, hlp->description);

        spinlock_unlock(&hlp->lock);

    }

    spinlock_unlock(&cmd_lock);

    return 1;

}

/** Describe specified command.

 *

 * @param argv Argument vector.

 *

 * @return 0 on failure, 1 on success.

 */

int cmd_desc(cmd_arg_t *argv)

{

    link_t *cur;

    ipl_t ipl;

    spinlock_lock(&cmd_lock);

    for (cur = cmd_head.next; cur != &cmd_head; cur = cur->next) {

        cmd_info_t *hlp;

        hlp = list_get_instance(cur, cmd_info_t, link);

        spinlock_lock(&hlp->lock);

        if (strncmp(hlp->name, (const char *) argv->buffer, strlen(hlp->name)) == 0) {

            printf("%s - %s\n", hlp->name, hlp->description);

            if (hlp->help)

                hlp->help();

            spinlock_unlock(&hlp->lock);

            break;

        }

        spinlock_unlock(&hlp->lock);

    }

    spinlock_unlock(&cmd_lock);

    return 1;

}

/** Search symbol table */

int cmd_symaddr(cmd_arg_t *argv)

{

    __address symaddr;

    char *symbol;

    symtab_print_search(argv->buffer);

    return 1;

}

/** Call function with zero parameters */

int cmd_call0(cmd_arg_t *argv)

{

    __address symaddr;

    char *symbol;

    __native (*f)(void);

    symaddr = get_symbol_addr(argv->buffer);

    if (!symaddr)

        printf("Symbol %s not found.\n", argv->buffer);

    else if (symaddr == (__address) -1) {

        symtab_print_search(argv->buffer);

        printf("Duplicate symbol, be more specific.\n");

    } else {

        symbol = get_symtab_entry(symaddr);

        printf("Calling f(): 0x%p: %s\n", symaddr, symbol);

        f =  (__native (*)(void)) symaddr;

        printf("Result: 0x%p\n", f());

    }

    return 1;

}

/** Call function with one parameter */

int cmd_call1(cmd_arg_t *argv)

{

    __address symaddr;

    char *symbol;

    __native (*f)(__native);

    __native arg1 = argv[1].intval;

    symaddr = get_symbol_addr(argv->buffer);

    if (!symaddr)

        printf("Symbol %s not found.\n", argv->buffer);

    else if (symaddr == (__address) -1) {

        symtab_print_search(argv->buffer);

        printf("Duplicate symbol, be more specific.\n");

    } else {

        symbol = get_symtab_entry(symaddr);

        printf("Calling f(0x%x): 0x%p: %s\n", arg1, symaddr, symbol);

        f =  (__native (*)(__native)) symaddr;

        printf("Result: 0x%p\n", f(arg1));

    }

    return 1;

}

/** Call function with two parameters */

int cmd_call2(cmd_arg_t *argv)

{

    __address symaddr;

    char *symbol;

    __native (*f)(__native,__native);

    __native arg1 = argv[1].intval;

    __native arg2 = argv[2].intval;

    symaddr = get_symbol_addr(argv->buffer);

    if (!symaddr)

        printf("Symbol %s not found.\n", argv->buffer);

    else if (symaddr == (__address) -1) {

        symtab_print_search(argv->buffer);

        printf("Duplicate symbol, be more specific.\n");

    } else {

        symbol = get_symtab_entry(symaddr);

        printf("Calling f(0x%x,0x%x): 0x%p: %s\n",

               arg1, arg2, symaddr, symbol);

        f =  (__native (*)(__native,__native)) symaddr;

        printf("Result: 0x%p\n", f(arg1, arg2));

    }

    return 1;

}

/** Call function with three parameters */

int cmd_call3(cmd_arg_t *argv)

{

    __address symaddr;

    char *symbol;

    __native (*f)(__native,__native,__native);

    __native arg1 = argv[1].intval;

    __native arg2 = argv[2].intval;

    __native arg3 = argv[3].intval;

    symaddr = get_symbol_addr(argv->buffer);

    if (!symaddr)

        printf("Symbol %s not found.\n", argv->buffer);

    else if (symaddr == (__address) -1) {

        symtab_print_search(argv->buffer);

        printf("Duplicate symbol, be more specific.\n");

    } else {

        symbol = get_symtab_entry(symaddr);

        printf("Calling f(0x%x,0x%x, 0x%x): 0x%p: %s\n",

               arg1, arg2, arg3, symaddr, symbol);

        f =  (__native (*)(__native,__native,__native)) symaddr;

        printf("Result: 0x%p\n", f(arg1, arg2, arg3));

    }

    return 1;

}

/** Print detailed description of 'describe' command. */

void desc_help(void)

{

    printf("Syntax: describe command_name\n");

}

/** Halt the kernel.

 *

 * @param argv Argument vector (ignored).

 *

 * @return 0 on failure, 1 on success (never returns).

 */

int cmd_halt(cmd_arg_t *argv)

{

    halt();

    return 1;

}

/** Command for printing TLB contents.

 *

 * @param argv Not used.

 *

 * @return Always returns 1.

 */

int cmd_ptlb(cmd_arg_t *argv)

{

    tlb_print();

    return 1;

}

/** Write 4 byte value to address */

int cmd_set4(cmd_arg_t *argv)

{

    char *symbol;

    __u32 *addr ;

    __u32 arg1 = argv[1].intval;

    bool pointer = false;

    if (((char *)argv->buffer)[0] == '*') {

        addr = (__u32 *) get_symbol_addr(argv->buffer+1);

        pointer = true;

    } else if (((char *)argv->buffer)[0] >= '0' &&

           ((char *)argv->buffer)[0] <= '9')

        addr = (__u32 *)atoi((char *)argv->buffer);

    else

        addr = (__u32 *)get_symbol_addr(argv->buffer);

    if (!addr)

        printf("Symbol %s not found.\n", argv->buffer);

    else if (addr == (__u32 *) -1) {

        symtab_print_search(argv->buffer);

        printf("Duplicate symbol, be more specific.\n");

    } else {

        if (pointer)

            addr = (__u32 *)(*(__native *)addr);

        printf("Writing 0x%x -> 0x%p\n", arg1, addr);

        *addr = arg1;

    }

    return 1;

}