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;

}