From 9119672c5fa5d446c66fb9c6d6b1412f4897e8fd Mon Sep 17 00:00:00 2001 From: okuji Date: Fri, 12 Aug 2005 19:53:59 +0000 Subject: [PATCH] Forgot to add this file. --- normal/completion.c | 387 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 387 insertions(+) create mode 100644 normal/completion.c diff --git a/normal/completion.c b/normal/completion.c new file mode 100644 index 000000000..cde420fe7 --- /dev/null +++ b/normal/completion.c @@ -0,0 +1,387 @@ +/* completion.c - complete a command, a disk, a partition or a file */ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 1999,2000,2001,2002,2003,2004,2005 Free Software Foundation, Inc. + * + * This program 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 this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + */ + +#include +#include +#include +#include +#include +#include +#include + +/* The disk that is used for grub_partition_iterate. */ +static grub_device_t disk_dev; + +/* The current word. */ +static char *current_word; + +/* The matched string. */ +static char *match; + +/* The count of candidates. */ +static int num_found; + +/* The string to be appended. */ +static const char *suffix; + + + +static void +print_simple_completion (const char *str) +{ + grub_printf (" %s", str); +} + +static void +print_partition_completion (const char *str) +{ + grub_print_partinfo (disk_dev, (char *) str); + grub_errno = GRUB_ERR_NONE; +} + +/* Add a string to the list of possible completions. COMPLETION is the + string that should be added. EXTRA will be appended if COMPLETION + matches uniquely. The string WHAT contains a description of the + kind of data that is added. Use PRINT to show the completions + if there are multiple matches. */ +static int +add_completion (const char *completion, const char *extra, const char *what, + void (*print) (const char *)) +{ + if (grub_strncmp (current_word, completion, grub_strlen (current_word)) == 0) + { + num_found++; + + switch (num_found) + { + case 1: + match = grub_strdup (completion); + if (! match) + return 1; + suffix = extra; + break; + + case 2: + grub_printf ("\nPossible %s are:\n", what); + print (match); + + /* Fall through. */ + + default: + { + char *s = match; + const char *t = completion; + + print (completion); + + /* Detect the matched portion. */ + while (*s && *t && *s == *t) + { + s++; + t++; + } + + *s = '\0'; + } + break; + } + } + + return 0; +} + +static int +iterate_partition (const grub_partition_t p) +{ + return add_completion (grub_partition_get_name (p), ")", "partitions", + print_partition_completion); +} + +static int +iterate_dir (const char *filename, int dir) +{ + if (! dir) + { + if (add_completion (filename, " ", "files", print_simple_completion)) + return 1; + } + else + { + char fname[grub_strlen (filename) + 2]; + + grub_sprintf (fname, "%s/", filename); + if (add_completion (fname, "", "files", print_simple_completion)) + return 1; + } + + return 0; +} + +static int +iterate_dev (const char *devname) +{ + grub_device_t dev; + + /* Complete the partition part. */ + dev = grub_device_open (devname); + + if (dev) + { + if (dev->disk && dev->disk->has_partitions) + { + if (add_completion (devname, ",", "devices", + print_simple_completion)) + return 1; + } + else + { + if (add_completion (devname, ")", "devices", + print_simple_completion)) + return 1; + } + } + + grub_errno = GRUB_ERR_NONE; + return 0; +} + +static int +iterate_command (grub_command_t cmd) +{ + if (grub_command_find (cmd->name)) + { + if (cmd->flags & GRUB_COMMAND_FLAG_CMDLINE) + { + if (add_completion (cmd->name, " ", "commands", + print_simple_completion)) + return 1; + } + } + + return 0; +} + +/* Complete a device. */ +static int +complete_device (void) +{ + /* Check if this is a device or a partition. */ + char *p = grub_strchr (++current_word, ','); + + if (! p) + { + /* Complete the disk part. */ + if (grub_disk_dev_iterate (iterate_dev)) + return 1; + } + else + { + /* Complete the partition part. */ + *p = '\0'; + disk_dev = grub_device_open (current_word); + *p = ','; + grub_errno = GRUB_ERR_NONE; + + if (disk_dev) + { + if (disk_dev->disk && disk_dev->disk->has_partitions) + { + current_word = p + 1; + + if (grub_partition_iterate (disk_dev->disk, iterate_partition)) + { + grub_device_close (disk_dev); + return 1; + } + } + + grub_device_close (disk_dev); + } + else + return 1; + } + + return 0; +} + +/* Complete a file. */ +static int +complete_file (void) +{ + char *device; + char *dir; + char *last_dir; + grub_fs_t fs; + grub_device_t dev; + int ret = 0; + + device = grub_file_get_device_name (current_word); + if (grub_errno != GRUB_ERR_NONE) + return 1; + + dev = grub_device_open (device); + if (! dev) + { + ret = 1; + goto fail; + } + + fs = grub_fs_probe (dev); + if (! fs) + { + ret = 1; + goto fail; + } + + dir = grub_strchr (current_word, '/'); + last_dir = grub_strrchr (current_word, '/'); + if (dir) + { + char *dirfile; + + current_word = last_dir + 1; + + dir = grub_strdup (dir); + if (! dir) + { + ret = 1; + goto fail; + } + + /* Cut away the filename part. */ + dirfile = grub_strrchr (dir, '/'); + dirfile[1] = '\0'; + + /* Iterate the directory. */ + (fs->dir) (dev, dir, iterate_dir); + + grub_free (dir); + + if (grub_errno) + { + ret = 1; + goto fail; + } + } + else + { + current_word += grub_strlen (current_word); + match = grub_strdup ("/"); + if (! match) + { + ret = 1; + goto fail; + } + + suffix = ""; + num_found = 1; + } + + fail: + if (dev) + grub_device_close (dev); + grub_free (device); + return ret; +} + +/* Try to complete the string in BUF. Return the characters that + should be added to the string. This command outputs the possible + completions, in that case set RESTORE to 1 so the caller can + restore the prompt. */ +char * +grub_normal_do_completion (char *buf, int *restore) +{ + char *first_word; + + /* Initialize variables. */ + disk_dev = 0; + match = 0; + num_found = 0; + suffix = ""; + + *restore = 1; + + /* Find the first word. */ + for (first_word = buf; *first_word == ' '; first_word++) + ; + + /* Find the delimeter of the current word. */ + for (current_word = first_word + grub_strlen (first_word); + current_word > first_word; + current_word--) + if (*current_word == ' ' || *current_word == '=') + break; + + if (current_word == first_word) + { + /* Complete a command. */ + if (grub_iterate_commands (iterate_command)) + goto fail; + } + else + { + current_word++; + + if (*current_word == '(' && ! grub_strchr (current_word, ')')) + { + /* Complete a device. */ + if (complete_device ()) + goto fail; + } + else + { + /* Complete a file. */ + if (complete_file ()) + goto fail; + } + } + + /* If more than one match is found those matches will be printed and + the prompt should be restored. */ + if (num_found > 1) + *restore = 1; + else + *restore = 0; + + /* Return the part that matches. */ + if (match) + { + char *ret; + int current_len; + int match_len; + + current_len = grub_strlen (current_word); + match_len = grub_strlen (match); + ret = grub_malloc (match_len - current_len + grub_strlen (suffix) + 1); + grub_strcpy (ret, match + current_len); + if (num_found == 1) + grub_strcat (ret, suffix); + + grub_free (match); + + return ret; + } + + fail: + grub_free (match); + grub_errno = GRUB_ERR_NONE; + + return 0; +}