Subversion Repositories HelenOS

/* Copyright (c) 2008, Tim Post <tinkertim@gmail.com>

 * All rights reserved.

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

 *

 * Neither the name of the original program's authors nor the names of its

 * contributors may be used to endorse or promote products derived from this

 * software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT OWNER OR CONTRIBUTORS 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.

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <io/stream.h>

#include <console.h>

#include "config.h"

#include "util.h"

#include "scli.h"

#include "input.h"

#include "errors.h"

#include "exec.h"

extern volatile unsigned int cli_interactive;

/* Not exposed in input.h */

static void cli_restricted(char *);

static void read_line(char *, int);

/* More than a macro than anything */

static void cli_restricted(char *cmd)

{

    printf("%s is not available in %s mode\n", cmd,

        cli_interactive ? "interactive" : "non-interactive");

    return;

}

/* Tokenizes input from console, sees if the first word is a built-in, if so

 * invokes the built-in entry point (a[0]) passing all arguments in a[] to

 * the handler */

int tok_input(cliuser_t *usr)

{

    char *cmd[WORD_MAX];

    int n = 0, i = 0;

    int rc = 0;

    char *tmp;

    if (NULL == usr->line)

        return CL_EFAIL;

    tmp = cli_strdup(usr->line);

    /* Break up what the user typed, space delimited */

    /* TODO: Protect things in quotes / ticks, expand wildcards */

    cmd[n] = cli_strtok(tmp, " ");

    while (cmd[n] && n < WORD_MAX) {

        cmd[++n] = cli_strtok(NULL, " ");

    }

    /* We have rubbish */

    if (NULL == cmd[0]) {

        rc = CL_ENOENT;

        goto finit;

    }

    /* Its a builtin command */

    if ((i = (is_builtin(cmd[0]))) > -1) {

        /* Its not available in this mode, see what try_exec() thinks */

        if (builtin_is_restricted(i)) {

                rc = try_exec(cmd[0], cmd);

                if (rc)

                    /* No external matching it could be found, tell the

                     * user that the command does exist, but is not

                     * available in this mode. */

                    cli_restricted(cmd[0]);

                goto finit;

        }

        /* Its a builtin, its available, run it */

        rc = run_builtin(i, cmd, usr);

        goto finit;

    /* We repeat the same dance for modules */

    } else if ((i = (is_module(cmd[0]))) > -1) {

        if (module_is_restricted(i)) {

            rc = try_exec(cmd[0], cmd);

            if (rc)

                cli_restricted(cmd[0]);

            goto finit;

        }

        rc = run_module(i, cmd);

        goto finit;

    } else {

        /* Its not a module or builtin, restricted or otherwise.

         * See what try_exec() thinks of it and just pass its return

         * value back to the caller */

        rc = try_exec(cmd[0], cmd);

        goto finit;

    }

finit:

    if (NULL != usr->line) {

        free(usr->line);

        usr->line = (char *) NULL;

    }

    if (NULL != tmp)

        free(tmp);

    return rc;

}

/* Borrowed from Jiri Svoboda's 'cli' uspace app */

static void read_line(char *buffer, int n)

{

    char c;

    int chars;

    chars = 0;

    while (chars < n - 1) {

        c = getchar();

        if (c < 0)

            return;

        if (c == '\n')

            break;

        if (c == '\b') {

            if (chars > 0) {

                putchar('\b');

                --chars;

            }

            continue;

        }

        putchar(c);

        buffer[chars++] = c;

    }

    putchar('\n');

    buffer[chars] = '\0';

}

/* TODO:

 * Implement something like editline() / readline(), if even

 * just for command history and making arrows work. */

void get_input(cliuser_t *usr)

{

    char line[INPUT_MAX];

    size_t len = 0;

    console_set_style(STYLE_EMPHASIS);

    printf("%s", usr->prompt);

    console_set_style(STYLE_NORMAL);

    read_line(line, INPUT_MAX);

    len = strlen(line);

    /* Make sure we don't have rubbish or a C/R happy user */

    if (len == 0 || line[0] == '\n')

        return;

    usr->line = cli_strdup(line);

    return;

}