2004-03-14 Marco Gerards <metgerards@student.han.nl>
* commands/boot.c: New file. * commands/cat.c: Likewise. * commands/cmp.c: Likewise. * commands/ls.c: Likewise. * commands/terminal.c: Likewise. * normal/command.c: Include <pupa/env.h> and <pupa/dl.h>. (pupa_register_command): Changed interface to match the new argument parser. (pupa_command_execute): Changed (almost rewritten) so it uses pupa_split_command. Added support for setting variables using the syntax `foo=bar'. (rescue_command): Changed to work with the new argument parser. (terminal_command): Moved from here to commands/terminal.c. (set_command): New function. (unset_command): New function. (insmod_command): New function. (rmmod_command): New function. (lsmod_command): New function. (pupa_command_init): Don't initialize the command terminal anymore. Initialize the commands set, unset, insmod, rmmod and lsmod. * conf/i386-pc.rmk (kernel_img_SOURCES): Add kern/env.c. (kernel_img_HEADERS): Add arg.h and env.h. (pupa_mkimage_LDFLAGS): Add kern/env.c. (pupa_emu_SOURCES): Add kern/env.c, commands/ls.c, commands/terminal.c commands/boot.c commands/cmp.c commands/cat.c, normal/arg.c. (pkgdata_MODULES): Add ls.mod, boot.mod, cmp.mod, cat.mod and terminal.mod. (normal_mod_SOURCES): Add normal/arg.c and normal/arg.c. (boot_mod_SOURCES): New variable. (terminal_mod_SOURCES): Likewise. (ls_mod_SOURCES): Likewise. (cmp_mod_SOURCES): Likewise. (cat_mod_SOURCES): Likewise. * normal/arg.c: New file. * kern/env.c: Likewise. * include/pupa/arg.h: Likewise. * include/pupa/env.h: Likewise. * font/manager.c (font_command): Changed to match argument parsing interface changes. (PUPA_MOD_INIT): Likewise. * hello/hello.c (pupa_cmd_hello): Likewise. (PUPA_MOD_INIT): Likewise. * include/pupa/disk.h: Include <pupa/device.h>. (pupa_print_partinfo): New prototype. * include/pupa/dl.h (pupa_dl_set_prefix): Prototype removed. (pupa_dl_get_prefix): Likewise. * include/pupa/misc.h: Include <pupa/err.h>. (pupa_isgraph): New prototype. (pupa_isdigit): Likewise. (pupa_split_cmdline): Likewise. * include/pupa/normal.h: Include <pupa/arg.h>. (pupa_command): Changed the prototype of the member `func' to match the argument parsing interface. Added member `options'. (pupa_register_command): Updated to match function. (pupa_arg_parse): New prototype. (pupa_hello_init) [PUPA_UTIL]: New prototype. (pupa_hello_fini) [PUPA_UTIL]: Likewise. (pupa_ls_init) [PUPA_UTIL]: Likewise. (pupa_ls_fini) [PUPA_UTIL]: Likewise. (pupa_cat_init) [PUPA_UTIL]: Likewise. (pupa_cat_fini) [PUPA_UTIL]: Likewise. (pupa_boot_init) [PUPA_UTIL]: Likewise. (pupa_boot_fini) [PUPA_UTIL]: Likewise. (pupa_cmp_init) [PUPA_UTIL]: Likewise. (pupa_cmp_fini) [PUPA_UTIL]: Likewise. (pupa_terminal_init) [PUPA_UTIL]: Likewise. (pupa_terminal_fini) [PUPA_UTIL]: Likewise. * kern/disk.c: Include <pupa/file.h>. (pupa_print_partinfo): New function. * kern/dl.c: Include <pupa/env.h>. (pupa_dl_dir): Variable removed. (pupa_dl_load): Use the environment variable `prefix' instead of the variable pupa_dl_dir. (pupa_dl_set_prefix): Function removed. (pupa_dl_get_prefix): Likewise. * kern/i386/pc/init.c: Include <pupa/env.h>. (pupa_machine_init): Use the environment variable `prefix' instead of using pupa_dl_set_prefix to set the prefix. * kern/main.c: Include <pupa/env.h>. (pupa_set_root_dev): Use the environment variable `prefix' instead of using pupa_dl_get_prefix to get the prefix. * kern/misc.c: Include <pupa/env.h>. (pupa_isdigit): New function. (pupa_isgraph): Likewise. (pupa_ftoa): Likewise. (pupa_vsprintf): Added support for printing values of the type `double'. Make it possible to format variable output when using formatting like `%1.2%f'. (pupa_split_cmdline): New function. * kern/rescue.c: Include <pupa/env.h>. (next_word): Removed function. (pupa_rescue_cmd_prefix): Likewise. (pupa_rescue_cmd_set): New function. (pupa_rescue_cmd_unset): New function. (pupa_enter_rescue_mode): Use the `pupa_split_cmdline' function to split the command line instead of splitting it here. Added support for setting variables using the syntax `foo=bar'. Don't initialize the prefix command anymore. Initialized the set and unset commands. * normal/cmdline.c: Include <pupa/env.h>. (pupa_tab_complete): Added prototypes for print_simple_completion, print_partition_completion, add_completion, iterate_commands, iterate_dev, iterate_part and iterate_dir. Moved code to print partition information from here to kern/disk.c. (pupa_cmdline_run): Don't check if the funtion exists anymore. * normal/main.c: Include <pupa/env.h>. (pupa_rescue_cmd_normal): Use the environment variable `prefix' instead of using pupa_dl_get_prefix to get the prefix. * term/i386/pc/vga.c: Include <pupa/arg.h>. (check_vga_mem): Cast pointers to `void *' to silence a gcc warning. (pupa_vga_putchar) [! DEBUG_VGA]: Removed for this case. (pupa_vga_setcolor): Declare unused variables with `__attribute__ ((unused))' to silence a gcc warning. (pupa_vga_setcolor): Likewise. (debug_command): Changed to match argument parsing interface changes. * util/pupa-emu.c: Include <pupa/env.h>. (options): Added 0's for unused fields to silence a gcc warning. (argp): Likewise. (main): Use the environment variable `prefix' instead of using pupa_dl_set_prefix to set the prefix. Initialize the commands ls, boot, cmp, cat and terminal. Finish the commands boot, cmp, cat and terminal. * util/i386/pc/getroot.c: Include <pupa/i386/pc/util/biosdisk.h>. * util/misc.c: Include <malloc.h>. (pupa_malloc): Rewritten so errors are correctly reported. (pupa_realloc): Likewise. (pupa_memalign): Likewise. (pupa_mm_init_region): Declare unused variables with `__attribute__ ((unused))' to silence a gcc warning. * normal/i386/setjmp.S: Remove tab at the end of the file to silence a gcc warning. * loader/i386/pc/linux.c (pupa_rescue_cmd_initrd): Declare unused variables with `__attribute__ ((unused))' to silence a gcc warning. * loader/i386/pc/multiboot.c (pupa_multiboot_unload): Make the local variable i unsigned to silence a gcc warning. * kern/term.c: Include <pupa/misc.h>. (pupa_more_lines): New variable. (pupa_more): Likewise. (pupa_putcode): When the pager is active pause at the end of every screen. (pupa_set_more): New function. * include/pupa/term.h (pupa_set_more): New prototype.
This commit is contained in:
marco_g
parent
3b1139cb9d
commit
db1771cfbe
35 changed files with 2607 additions and 285 deletions
354
normal/arg.c
Normal file
354
normal/arg.c
Normal file
|
|
@ -0,0 +1,354 @@
|
|||
/* arg.c - argument parser */
|
||||
/*
|
||||
* PUPA -- Preliminary Universal Programming Architecture for GRUB
|
||||
* Copyright (C) 2003, 2004 Free Software Foundation, Inc.
|
||||
*
|
||||
* PUPA is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with PUPA; if not, write to the Free Software
|
||||
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*/
|
||||
|
||||
#include "pupa/arg.h"
|
||||
#include "pupa/misc.h"
|
||||
#include "pupa/mm.h"
|
||||
#include "pupa/err.h"
|
||||
#include "pupa/normal.h"
|
||||
|
||||
/* Build in parser for default options. */
|
||||
static const struct pupa_arg_option help_options[] =
|
||||
{
|
||||
{"help", 'h', 0, "Display help", 0, ARG_TYPE_NONE},
|
||||
{"usage", 'u', 0, "Show how to use this command", 0, ARG_TYPE_NONE},
|
||||
{0, 0, 0, 0, 0, 0}
|
||||
};
|
||||
|
||||
static struct pupa_arg_option *
|
||||
find_short (const struct pupa_arg_option *options, char c)
|
||||
{
|
||||
struct pupa_arg_option *found = 0;
|
||||
auto struct pupa_arg_option *fnd_short (const struct pupa_arg_option *opt);
|
||||
|
||||
struct pupa_arg_option *fnd_short (const struct pupa_arg_option *opt)
|
||||
{
|
||||
while (opt->doc)
|
||||
{
|
||||
if (opt->shortarg == c)
|
||||
return (struct pupa_arg_option *) opt;
|
||||
opt++;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (options)
|
||||
found = fnd_short (options);
|
||||
if (! found)
|
||||
found = fnd_short (help_options);
|
||||
|
||||
return found;
|
||||
}
|
||||
|
||||
static char *
|
||||
find_long_option (char *s)
|
||||
{
|
||||
char *argpos = pupa_strchr (s, '=');
|
||||
|
||||
if (argpos)
|
||||
{
|
||||
*argpos = '\0';
|
||||
return ++argpos;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
static struct pupa_arg_option *
|
||||
find_long (const struct pupa_arg_option *options, char *s)
|
||||
{
|
||||
struct pupa_arg_option *found = 0;
|
||||
auto struct pupa_arg_option *fnd_long (const struct pupa_arg_option *opt);
|
||||
|
||||
struct pupa_arg_option *fnd_long (const struct pupa_arg_option *opt)
|
||||
{
|
||||
while (opt->doc)
|
||||
{
|
||||
if (opt->longarg && !pupa_strcmp (opt->longarg, s))
|
||||
return (struct pupa_arg_option *) opt;
|
||||
opt++;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (options)
|
||||
found = fnd_long (options);
|
||||
if (!found)
|
||||
found = fnd_long (help_options);
|
||||
|
||||
return found;
|
||||
}
|
||||
|
||||
static void
|
||||
show_usage (pupa_command_t cmd)
|
||||
{
|
||||
pupa_printf ("Usage: %s\n", cmd->summary);
|
||||
}
|
||||
|
||||
static void
|
||||
show_help (pupa_command_t cmd)
|
||||
{
|
||||
static void showargs (const struct pupa_arg_option *opt)
|
||||
{
|
||||
for (; opt->doc; opt++)
|
||||
{
|
||||
if (opt->shortarg && pupa_isgraph (opt->shortarg))
|
||||
pupa_printf ("-%c%c ", opt->shortarg, opt->longarg ? ',':' ');
|
||||
else
|
||||
pupa_printf (" ");
|
||||
if (opt->longarg)
|
||||
{
|
||||
pupa_printf ("--%s", opt->longarg);
|
||||
if (opt->arg)
|
||||
pupa_printf ("=%s", opt->arg);
|
||||
}
|
||||
else
|
||||
pupa_printf ("\t");
|
||||
|
||||
pupa_printf ("\t\t%s\n", opt->doc);
|
||||
}
|
||||
}
|
||||
|
||||
show_usage (cmd);
|
||||
pupa_printf ("%s\n\n", cmd->description);
|
||||
if (cmd->options)
|
||||
showargs (cmd->options);
|
||||
showargs (help_options);
|
||||
pupa_printf ("\nReport bugs to <%s>.\n", PACKAGE_BUGREPORT);
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
parse_option (pupa_command_t cmd, int key, char *arg, struct pupa_arg_list *usr)
|
||||
{
|
||||
switch (key)
|
||||
{
|
||||
case 'h':
|
||||
show_help (cmd);
|
||||
return -1;
|
||||
|
||||
case 'u':
|
||||
show_usage (cmd);
|
||||
return -1;
|
||||
|
||||
default:
|
||||
{
|
||||
int found = -1;
|
||||
int i = 0;
|
||||
const struct pupa_arg_option *opt = cmd->options;
|
||||
|
||||
while (opt->doc)
|
||||
{
|
||||
if (opt->shortarg && key == opt->shortarg)
|
||||
{
|
||||
found = i;
|
||||
break;
|
||||
}
|
||||
opt++;
|
||||
i++;
|
||||
}
|
||||
|
||||
if (found == -1)
|
||||
return -1;
|
||||
|
||||
usr[found].set = 1;
|
||||
usr[found].arg = arg;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
pupa_arg_parse (pupa_command_t cmd, int argc, char **argv,
|
||||
struct pupa_arg_list *usr, char ***args, int *argnum)
|
||||
{
|
||||
int curarg;
|
||||
char *longarg = 0;
|
||||
int complete = 0;
|
||||
char **argl = 0;
|
||||
int num = 0;
|
||||
auto pupa_err_t add_arg (char *s);
|
||||
|
||||
pupa_err_t add_arg (char *s)
|
||||
{
|
||||
argl = pupa_realloc (argl, (++num) * sizeof (char *));
|
||||
if (! args)
|
||||
return pupa_errno;
|
||||
argl[num - 1] = s;
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
for (curarg = 0; curarg < argc; curarg++)
|
||||
{
|
||||
char *arg = argv[curarg];
|
||||
struct pupa_arg_option *opt;
|
||||
char *option = 0;
|
||||
|
||||
/* No option is used. */
|
||||
if (arg[0] != '-' || pupa_strlen (arg) == 1)
|
||||
{
|
||||
if (add_arg (arg) != 0)
|
||||
goto fail;
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
/* One or more short options. */
|
||||
if (arg[1] != '-')
|
||||
{
|
||||
char *curshort = arg + 1;
|
||||
|
||||
while (1)
|
||||
{
|
||||
opt = find_short (cmd->options, *curshort);
|
||||
if (!opt)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT,
|
||||
"Unknown argument `-%c'\n", *curshort);
|
||||
goto fail;
|
||||
}
|
||||
|
||||
curshort++;
|
||||
|
||||
/* Parse all arguments here except the last one because
|
||||
it can have an argument value. */
|
||||
if (*curshort)
|
||||
{
|
||||
if (parse_option (cmd, opt->shortarg, 0, usr) || pupa_errno)
|
||||
goto fail;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (opt->type != ARG_TYPE_NONE)
|
||||
{
|
||||
if (curarg + 1 < argc)
|
||||
{
|
||||
char *nextarg = argv[curarg + 1];
|
||||
if (!(opt->flags & PUPA_ARG_OPTION_OPTIONAL)
|
||||
|| (pupa_strlen (nextarg) < 2 || nextarg[0] != '-'))
|
||||
option = argv[++curarg];
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
else /* The argument starts with "--". */
|
||||
{
|
||||
/* If the argument "--" is used just pass the other
|
||||
arguments. */
|
||||
if (pupa_strlen (arg) == 2)
|
||||
{
|
||||
for (curarg++; curarg < argc; curarg++)
|
||||
if (add_arg (arg) != 0)
|
||||
goto fail;
|
||||
break;
|
||||
}
|
||||
|
||||
longarg = (char *) pupa_strdup (arg);
|
||||
if (! longarg)
|
||||
goto fail;
|
||||
|
||||
option = find_long_option (longarg);
|
||||
arg = longarg;
|
||||
|
||||
opt = find_long (cmd->options, arg + 2);
|
||||
if (!opt)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT, "Unknown argument `%s'\n", arg);
|
||||
goto fail;
|
||||
}
|
||||
}
|
||||
|
||||
if (! (opt->type == ARG_TYPE_NONE
|
||||
|| (!option && (opt->flags & PUPA_ARG_OPTION_OPTIONAL))))
|
||||
{
|
||||
if (!option)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT,
|
||||
"Missing mandatory option for `%s'\n", opt->longarg);
|
||||
goto fail;
|
||||
}
|
||||
|
||||
switch (opt->type)
|
||||
{
|
||||
case ARG_TYPE_NONE:
|
||||
/* This will never happen. */
|
||||
break;
|
||||
|
||||
case ARG_TYPE_STRING:
|
||||
/* No need to do anything. */
|
||||
break;
|
||||
|
||||
case ARG_TYPE_INT:
|
||||
{
|
||||
char *tail;
|
||||
|
||||
pupa_strtoul (option, &tail, 0);
|
||||
if (tail == 0 || tail == option || *tail != '\0' || pupa_errno)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT,
|
||||
"The argument `%s' requires an integer.",
|
||||
arg);
|
||||
|
||||
goto fail;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
case ARG_TYPE_DEVICE:
|
||||
case ARG_TYPE_DIR:
|
||||
case ARG_TYPE_FILE:
|
||||
case ARG_TYPE_PATHNAME:
|
||||
/* XXX: Not implemented. */
|
||||
break;
|
||||
}
|
||||
if (parse_option (cmd, opt->shortarg, option, usr) || pupa_errno)
|
||||
goto fail;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (option)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT,
|
||||
"A value was assigned to the argument `%s' while it "
|
||||
"doesn't require an argument\n", arg);
|
||||
goto fail;
|
||||
}
|
||||
|
||||
if (parse_option (cmd, opt->shortarg, 0, usr) || pupa_errno)
|
||||
goto fail;
|
||||
}
|
||||
pupa_free (longarg);
|
||||
longarg = 0;
|
||||
}
|
||||
|
||||
complete = 1;
|
||||
|
||||
*args = argl;
|
||||
*argnum = num;
|
||||
|
||||
fail:
|
||||
pupa_free (longarg);
|
||||
|
||||
return complete;
|
||||
}
|
||||
|
|
@ -26,6 +26,7 @@
|
|||
#include <pupa/machine/partition.h>
|
||||
#include <pupa/disk.h>
|
||||
#include <pupa/file.h>
|
||||
#include <pupa/env.h>
|
||||
|
||||
static char *kill_buf;
|
||||
|
||||
|
|
@ -156,6 +157,18 @@ pupa_tab_complete (char *buf, int *restore)
|
|||
/* String that is added when matched. */
|
||||
char *matchstr;
|
||||
|
||||
auto void print_simple_completion (char *comp);
|
||||
auto void print_partition_completion (char *comp);
|
||||
auto int NESTED_FUNC_ATTR add_completion (const char *comp, const char *match,
|
||||
const char *what,
|
||||
void (*print_completion) (char *));
|
||||
auto int iterate_commands (pupa_command_t cmd);
|
||||
auto int iterate_dev (const char *devname);
|
||||
auto int iterate_part (const pupa_partition_t p);
|
||||
auto int iterate_dir (const char *filename, int dir);
|
||||
|
||||
|
||||
|
||||
void print_simple_completion (char *comp)
|
||||
{
|
||||
pupa_printf (" %s", comp);
|
||||
|
|
@ -163,39 +176,8 @@ pupa_tab_complete (char *buf, int *restore)
|
|||
|
||||
void print_partition_completion (char *comp)
|
||||
{
|
||||
pupa_fs_t fs = 0;
|
||||
pupa_device_t part;
|
||||
char devname[20];
|
||||
|
||||
pupa_sprintf (devname, "%s,%s", partdev->disk->name, comp);
|
||||
part = pupa_device_open (devname);
|
||||
if (!part)
|
||||
pupa_printf ("\n\tPartition num:%s, Filesystem cannot be accessed",
|
||||
comp);
|
||||
else
|
||||
{
|
||||
char *label;
|
||||
|
||||
fs = pupa_fs_probe (part);
|
||||
/* Ignore all errors. */
|
||||
pupa_errno = 0;
|
||||
|
||||
pupa_printf ("\n\tPartition num:%s, Filesystem type %s",
|
||||
comp, fs ? fs->name : "Unknown");
|
||||
|
||||
if (fs)
|
||||
{
|
||||
(fs->label) (part, &label);
|
||||
if (pupa_errno == PUPA_ERR_NONE)
|
||||
{
|
||||
if (label && pupa_strlen (label))
|
||||
pupa_printf (", Label: %s", label);
|
||||
pupa_free (label);
|
||||
}
|
||||
pupa_errno = PUPA_ERR_NONE;
|
||||
}
|
||||
pupa_device_close (part);
|
||||
}
|
||||
pupa_print_partinfo (partdev, comp);
|
||||
pupa_errno = 0;
|
||||
}
|
||||
|
||||
/* Add a string to the list of possible completions. COMP is the
|
||||
|
|
@ -206,9 +188,9 @@ pupa_tab_complete (char *buf, int *restore)
|
|||
multiple matches. XXX: Because of a bug in gcc it is required to
|
||||
use __regparm__ in some cases. */
|
||||
|
||||
int NESTED_FUNC_ATTR
|
||||
add_completion (const char *comp, const char *match, const char *what,
|
||||
void (*print_completion) (char *))
|
||||
int NESTED_FUNC_ATTR add_completion (const char *comp, const char *match,
|
||||
const char *what,
|
||||
void (*print_completion) (char *))
|
||||
{
|
||||
/* Bug in strncmp then len ==0. */
|
||||
if (!len || pupa_strncmp (pos, comp, len) == 0)
|
||||
|
|
@ -460,7 +442,6 @@ pupa_cmdline_run (int nested)
|
|||
while (1)
|
||||
{
|
||||
static char cmdline[PUPA_MAX_CMDLINE];
|
||||
pupa_command_t cmd;
|
||||
|
||||
pupa_print_error ();
|
||||
pupa_errno = PUPA_ERR_NONE;
|
||||
|
|
@ -473,17 +454,6 @@ pupa_cmdline_run (int nested)
|
|||
if (! *cmdline)
|
||||
continue;
|
||||
|
||||
cmd = pupa_command_find (cmdline);
|
||||
if (! cmd)
|
||||
continue;
|
||||
|
||||
if (! (cmd->flags & PUPA_COMMAND_FLAG_CMDLINE))
|
||||
{
|
||||
pupa_error (PUPA_ERR_UNKNOWN_COMMAND,
|
||||
"invalid command `%s'", cmd->name);
|
||||
continue;
|
||||
}
|
||||
|
||||
pupa_command_execute (cmdline);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
268
normal/command.c
268
normal/command.c
|
|
@ -22,15 +22,19 @@
|
|||
#include <pupa/mm.h>
|
||||
#include <pupa/err.h>
|
||||
#include <pupa/term.h>
|
||||
#include <pupa/env.h>
|
||||
#include <pupa/dl.h>
|
||||
|
||||
static pupa_command_t pupa_command_list;
|
||||
|
||||
void
|
||||
pupa_register_command (const char *name,
|
||||
int (*func) (int argc, char *argv[]),
|
||||
pupa_err_t (*func) (struct pupa_arg_list *state,
|
||||
int argc, char **args),
|
||||
unsigned flags,
|
||||
const char *summary,
|
||||
const char *description)
|
||||
const char *description,
|
||||
const struct pupa_arg_option *options)
|
||||
{
|
||||
pupa_command_t cmd, *p;
|
||||
|
||||
|
|
@ -43,6 +47,7 @@ pupa_register_command (const char *name,
|
|||
cmd->flags = flags;
|
||||
cmd->summary = summary;
|
||||
cmd->description = description;
|
||||
cmd->options = options;
|
||||
|
||||
/* Keep the list sorted for simplicity. */
|
||||
p = &pupa_command_list;
|
||||
|
|
@ -107,66 +112,69 @@ pupa_iterate_commands (int (*iterate) (pupa_command_t))
|
|||
int
|
||||
pupa_command_execute (char *cmdline)
|
||||
{
|
||||
pupa_command_t cmd;
|
||||
char *p;
|
||||
char **args;
|
||||
int num = 0;
|
||||
int i;
|
||||
int ret;
|
||||
auto pupa_err_t cmdline_get (char **s);
|
||||
pupa_err_t cmdline_get (char **s)
|
||||
{
|
||||
*s = pupa_malloc (PUPA_MAX_CMDLINE);
|
||||
*s[0] = '\0';
|
||||
return pupa_cmdline_get (">", *s, PUPA_MAX_CMDLINE, 0, 1);
|
||||
}
|
||||
|
||||
cmd = pupa_command_find (cmdline);
|
||||
pupa_command_t cmd;
|
||||
pupa_err_t ret = 0;
|
||||
char *pager;
|
||||
int num;
|
||||
char **args;
|
||||
struct pupa_arg_list *state;
|
||||
struct pupa_arg_option *parser;
|
||||
int maxargs = 0;
|
||||
char **arglist;
|
||||
int numargs;
|
||||
|
||||
if (pupa_split_cmdline (cmdline, cmdline_get, &num, &args))
|
||||
return 0;
|
||||
|
||||
/* In case of an assignment set the environment accordingly instead
|
||||
of calling a function. */
|
||||
if (num == 0 && pupa_strchr (args[0], '='))
|
||||
{
|
||||
char *val = pupa_strchr (args[0], '=');
|
||||
val[0] = 0;
|
||||
pupa_env_set (args[0], val + 1);
|
||||
val[0] = '=';
|
||||
return 0;
|
||||
}
|
||||
|
||||
cmd = pupa_command_find (args[0]);
|
||||
if (! cmd)
|
||||
return -1;
|
||||
|
||||
/* Count arguments. */
|
||||
p = cmdline;
|
||||
while (1)
|
||||
{
|
||||
while (*p && *p != ' ')
|
||||
p++;
|
||||
/* Enable the pager if the environment pager is set to 1. */
|
||||
pager = pupa_env_get ("pager");
|
||||
if (pager && (! pupa_strcmp (pager, "1")))
|
||||
pupa_set_more (1);
|
||||
|
||||
parser = (struct pupa_arg_option *) cmd->options;
|
||||
while (parser && (parser++)->doc)
|
||||
maxargs++;
|
||||
|
||||
if (! *p)
|
||||
break;
|
||||
|
||||
while (*p == ' ')
|
||||
p++;
|
||||
|
||||
num++;
|
||||
}
|
||||
|
||||
args = (char **) pupa_malloc (sizeof (char *) * (num + 1));
|
||||
if (! args)
|
||||
return -1;
|
||||
|
||||
/* Fill arguments. */
|
||||
for (i = 0, p = pupa_strchr (cmdline, ' '); i < num && p; i++)
|
||||
{
|
||||
if (! p)
|
||||
break;
|
||||
|
||||
while (*p == ' ')
|
||||
p++;
|
||||
|
||||
args[i] = p;
|
||||
|
||||
while (*p && *p != ' ')
|
||||
p++;
|
||||
|
||||
*p++ = '\0';
|
||||
}
|
||||
|
||||
/* Terminate the array with NULL. */
|
||||
args[i] = 0;
|
||||
|
||||
ret = (cmd->func) (num, args);
|
||||
state = pupa_malloc (sizeof (struct pupa_arg_list) * maxargs);
|
||||
pupa_memset (state, 0, sizeof (struct pupa_arg_list) * maxargs);
|
||||
if (pupa_arg_parse (cmd, num, &args[1], state, &arglist, &numargs))
|
||||
ret = (cmd->func) (state, numargs, arglist);
|
||||
pupa_free (state);
|
||||
|
||||
if (pager && (! pupa_strcmp (pager, "1")))
|
||||
pupa_set_more (0);
|
||||
|
||||
pupa_free (args);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int
|
||||
rescue_command (int argc __attribute__ ((unused)),
|
||||
char *argv[] __attribute__ ((unused)))
|
||||
static pupa_err_t
|
||||
rescue_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc __attribute__ ((unused)),
|
||||
char **args __attribute__ ((unused)))
|
||||
{
|
||||
pupa_longjmp (pupa_exit_env, 0);
|
||||
|
||||
|
|
@ -174,61 +182,145 @@ rescue_command (int argc __attribute__ ((unused)),
|
|||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
terminal_command (int argc, char *argv[])
|
||||
|
||||
static pupa_err_t
|
||||
set_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc, char **args)
|
||||
{
|
||||
pupa_term_t term = 0;
|
||||
|
||||
auto int print_terminal (pupa_term_t);
|
||||
auto int find_terminal (pupa_term_t);
|
||||
|
||||
int print_terminal (pupa_term_t t)
|
||||
char *var;
|
||||
char *val;
|
||||
|
||||
auto int print_env (struct pupa_env_var *env);
|
||||
int print_env (struct pupa_env_var *env)
|
||||
{
|
||||
pupa_printf (" %s", t->name);
|
||||
pupa_printf ("%s=%s\n", env->name, env->value);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int find_terminal (pupa_term_t t)
|
||||
|
||||
if (! argc)
|
||||
{
|
||||
if (pupa_strcmp (t->name, argv[0]) == 0)
|
||||
{
|
||||
term = t;
|
||||
return 1;
|
||||
}
|
||||
|
||||
pupa_env_iterate (print_env);
|
||||
return 0;
|
||||
}
|
||||
|
||||
var = args[0];
|
||||
val = pupa_strchr (var, '=');
|
||||
if (! val)
|
||||
{
|
||||
pupa_error (PUPA_ERR_BAD_ARGUMENT, "not an assignment");
|
||||
return pupa_errno;
|
||||
}
|
||||
|
||||
val[0] = 0;
|
||||
pupa_env_set (var, val + 1);
|
||||
val[0] = '=';
|
||||
return 0;
|
||||
}
|
||||
|
||||
static pupa_err_t
|
||||
unset_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc, char **args)
|
||||
{
|
||||
if (argc < 1)
|
||||
return pupa_error (PUPA_ERR_BAD_ARGUMENT,
|
||||
"no environment variable specified");
|
||||
|
||||
pupa_env_unset (args[0]);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static pupa_err_t
|
||||
insmod_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc, char **args)
|
||||
{
|
||||
char *p;
|
||||
pupa_dl_t mod;
|
||||
|
||||
if (argc == 0)
|
||||
{
|
||||
pupa_printf ("Available terminal(s):");
|
||||
pupa_term_iterate (print_terminal);
|
||||
pupa_putchar ('\n');
|
||||
|
||||
pupa_printf ("Current terminal: %s\n", pupa_term_get_current ()->name);
|
||||
}
|
||||
return pupa_error (PUPA_ERR_BAD_ARGUMENT, "no module specified");
|
||||
|
||||
p = pupa_strchr (args[0], '/');
|
||||
if (! p)
|
||||
mod = pupa_dl_load (args[0]);
|
||||
else
|
||||
{
|
||||
pupa_term_iterate (find_terminal);
|
||||
if (! term)
|
||||
return pupa_error (PUPA_ERR_BAD_ARGUMENT, "no such terminal");
|
||||
mod = pupa_dl_load_file (args[0]);
|
||||
|
||||
pupa_term_set_current (term);
|
||||
if (mod)
|
||||
pupa_dl_ref (mod);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static pupa_err_t
|
||||
rmmod_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc, char **args)
|
||||
{
|
||||
pupa_dl_t mod;
|
||||
|
||||
if (argc == 0)
|
||||
return pupa_error (PUPA_ERR_BAD_ARGUMENT, "no module specified");
|
||||
|
||||
mod = pupa_dl_get (args[0]);
|
||||
if (! mod)
|
||||
return pupa_error (PUPA_ERR_BAD_ARGUMENT, "no such module");
|
||||
|
||||
if (! pupa_dl_unref (mod))
|
||||
pupa_dl_unload (mod);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static pupa_err_t
|
||||
lsmod_command (struct pupa_arg_list *state __attribute__ ((unused)),
|
||||
int argc __attribute__ ((unused)),
|
||||
char **args __attribute__ ((unused)))
|
||||
{
|
||||
auto int print_module (pupa_dl_t mod);
|
||||
|
||||
int print_module (pupa_dl_t mod)
|
||||
{
|
||||
pupa_dl_dep_t dep;
|
||||
|
||||
pupa_printf ("%s\t%d\t\t", mod->name, mod->ref_count);
|
||||
for (dep = mod->dep; dep; dep = dep->next)
|
||||
{
|
||||
if (dep != mod->dep)
|
||||
pupa_putchar (',');
|
||||
|
||||
pupa_printf ("%s", dep->mod->name);
|
||||
}
|
||||
pupa_putchar ('\n');
|
||||
pupa_refresh ();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
return PUPA_ERR_NONE;
|
||||
pupa_printf ("Name\tRef Count\tDependencies\n");
|
||||
pupa_dl_iterate (print_module);
|
||||
return 0;
|
||||
}
|
||||
|
||||
void
|
||||
pupa_command_init (void)
|
||||
{
|
||||
/* This is a special command, because this never be called actually. */
|
||||
pupa_register_command ("title", 0, PUPA_COMMAND_FLAG_TITLE, 0, 0);
|
||||
pupa_register_command ("title", 0, PUPA_COMMAND_FLAG_TITLE, 0, 0, 0);
|
||||
|
||||
pupa_register_command ("rescue", rescue_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"rescue",
|
||||
"Enter into the rescue mode.");
|
||||
pupa_register_command ("terminal", terminal_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"terminal [TERM...]",
|
||||
"Select a terminal.");
|
||||
"rescue", "Enter into the rescue mode.", 0);
|
||||
|
||||
pupa_register_command ("set", set_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"unset ENVVAR", "Set an environment variable.", 0);
|
||||
|
||||
pupa_register_command ("unset", unset_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"set [ENVVAR=VALUE]", "Remove an environment variable.", 0);
|
||||
|
||||
pupa_register_command ("insmod", insmod_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"insmod MODULE|FILE", "Insert a module.", 0);
|
||||
|
||||
pupa_register_command ("rmmod", rmmod_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"rmmod MODULE", "Remove a module.", 0);
|
||||
|
||||
pupa_register_command ("lsmod", lsmod_command, PUPA_COMMAND_FLAG_BOTH,
|
||||
"lsmod", "Show loaded modules.", 0);
|
||||
}
|
||||
|
|
|
|||
|
|
@ -26,6 +26,7 @@
|
|||
#include <pupa/file.h>
|
||||
#include <pupa/mm.h>
|
||||
#include <pupa/term.h>
|
||||
#include <pupa/env.h>
|
||||
|
||||
pupa_jmp_buf pupa_exit_env;
|
||||
|
||||
|
|
@ -315,7 +316,7 @@ pupa_rescue_cmd_normal (int argc, char *argv[])
|
|||
char *config;
|
||||
const char *prefix;
|
||||
|
||||
prefix = pupa_dl_get_prefix ();
|
||||
prefix = pupa_env_get ("prefix");
|
||||
if (prefix)
|
||||
{
|
||||
config = pupa_malloc (pupa_strlen (prefix) + sizeof ("/pupa.cfg"));
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue