Forgot to add this file.
This commit is contained in:
okuji
parent
8de3495c7b
commit
9119672c5f
1 changed files with 387 additions and 0 deletions
387
normal/completion.c
Normal file
387
normal/completion.c
Normal file
|
|
@ -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 <grub/normal.h>
|
||||
#include <grub/misc.h>
|
||||
#include <grub/err.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/partition.h>
|
||||
#include <grub/disk.h>
|
||||
#include <grub/file.h>
|
||||
|
||||
/* 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;
|
||||
}
|
||||
Loading…
Reference in a new issue