2005-07-20 Yoshinori K. Okuji <okuji@enbug.org>
Change the semantics of variable hooks. They now return strings instead of error values. * util/i386/pc/grub-setup.c: Include grub/env.h. (setup): Use grub_device_set_root instead of grub_env_set. * kern/rescue.c (grub_rescue_cmd_root): Use grub_env_set and grub_env_get instead of grub_device_set_root and grub_device_get_root, respectively. * kern/main.c (grub_env_write_root): New function. (grub_set_root_dev): Register grub_env_write_hook for "root". Use grub_env_set instead of grub_device_set_root. * kern/env.c (HASHSZ): Reduced to 13, because GRUB does not need many variables. (grub_env_set): Set ENV->VALUE to the result of ENV->WRITE_HOOK rather than calling ENV->WRITE_HOOK afterwards. (grub_env_get): Return the result of ENV->READ_HOOK rather than passing a pointer of a pointer. (grub_register_variable_hook): Change the types of "read_hook" and "write_hook" to grub_env_read_hook_t and grub_env_write_hook_t, respectively. Allocate the default empty string on the heap, because this string may be freed later. * kern/device.c: Include grub/env.h. (grub_device_set_root): Removed. (grub_device_get_root): Likewise. (grub_device_open): Use grub_env_get instead of grub_device_get_root. * include/grub/env.h (grub_env_read_hook_t): New type. (grub_env_write_hook_t): Likewise. (grub_env_var): Change the types of "read_hook" and "write_hook" to grub_env_read_hook_t and grub_env_write_hook_t, respectively. (grub_register_variable_hook): Likewise. * include/grub/device.h (grub_device_set_root): Removed. (grub_device_set_root): Likewise. * fs/fat.c (grub_fat_dir): Make a copy of PATH in DIRNAME, and make sure that DIRNAME terminates with '/', so that grub_fat_find_dir will fail if PATH is not a directory. * commands/ls.c (grub_ls_list_files): Remove the qualifier const from DIRNAME. Use the qualifier auto for print_files and print_files_long. If FS->DIR sets GRUB_ERRNO to GRUB_ERR_BAD_FILE_TYPE, try DIRNAME as a regular file. Put a newline only if there is no error. (grub_cmd_ls): Remove grub_ls_print_files, because this is not used.
This commit is contained in:
okuji
parent
896f0afdb7
commit
5f968e1e61
10 changed files with 173 additions and 72 deletions
|
|
@ -1,7 +1,7 @@
|
|||
/* device.c - device manager */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2002 Free Software Foundation, Inc.
|
||||
* Copyright (C) 2002,2005 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
|
|
@ -24,25 +24,7 @@
|
|||
#include <grub/fs.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/misc.h>
|
||||
|
||||
static char *grub_device_root;
|
||||
|
||||
grub_err_t
|
||||
grub_device_set_root (const char *name)
|
||||
{
|
||||
grub_free (grub_device_root);
|
||||
grub_device_root = grub_strdup (name);
|
||||
return grub_errno;
|
||||
}
|
||||
|
||||
const char *
|
||||
grub_device_get_root (void)
|
||||
{
|
||||
if (! grub_device_root)
|
||||
grub_error (GRUB_ERR_BAD_DEVICE, "no root device");
|
||||
|
||||
return grub_device_root;
|
||||
}
|
||||
#include <grub/env.h>
|
||||
|
||||
grub_device_t
|
||||
grub_device_open (const char *name)
|
||||
|
|
@ -52,13 +34,12 @@ grub_device_open (const char *name)
|
|||
|
||||
if (! name)
|
||||
{
|
||||
if (! grub_device_root)
|
||||
name = grub_env_get ("root");
|
||||
if (*name == '\0')
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_DEVICE, "no device is set");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
name = grub_device_root;
|
||||
}
|
||||
|
||||
dev = grub_malloc (sizeof (*dev));
|
||||
|
|
|
|||
40
kern/env.c
40
kern/env.c
|
|
@ -1,7 +1,7 @@
|
|||
/* env.c - Environment variables */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2003 Free Software Foundation, Inc.
|
||||
* Copyright (C) 2003,2005 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
|
|
@ -22,8 +22,8 @@
|
|||
#include <grub/misc.h>
|
||||
#include <grub/mm.h>
|
||||
|
||||
/* XXX: What would be a good size for the hashtable? */
|
||||
#define HASHSZ 123
|
||||
/* The size of the hash table. */
|
||||
#define HASHSZ 13
|
||||
|
||||
/* A hashtable for quick lookup of variables. */
|
||||
static struct grub_env_var *grub_env[HASHSZ];
|
||||
|
|
@ -68,16 +68,18 @@ grub_env_set (const char *var, const char *val)
|
|||
if (env)
|
||||
{
|
||||
char *old = env->value;
|
||||
env->value = grub_strdup (val);
|
||||
if (! env->name)
|
||||
|
||||
if (env->write_hook)
|
||||
env->value = env->write_hook (env, val);
|
||||
else
|
||||
env->value = grub_strdup (val);
|
||||
|
||||
if (! env->value)
|
||||
{
|
||||
env->value = old;
|
||||
return grub_errno;
|
||||
}
|
||||
|
||||
if (env->write_hook)
|
||||
(env->write_hook) (env);
|
||||
|
||||
grub_free (old);
|
||||
return 0;
|
||||
}
|
||||
|
|
@ -141,12 +143,7 @@ grub_env_get (const char *name)
|
|||
return 0;
|
||||
|
||||
if (env->read_hook)
|
||||
{
|
||||
char *val;
|
||||
env->read_hook (env, &val);
|
||||
|
||||
return val;
|
||||
}
|
||||
return env->read_hook (env, env->value);
|
||||
|
||||
return env->value;
|
||||
}
|
||||
|
|
@ -190,14 +187,21 @@ grub_env_iterate (int (* func) (struct grub_env_var *var))
|
|||
|
||||
grub_err_t
|
||||
grub_register_variable_hook (const char *var,
|
||||
grub_err_t (*read_hook) (struct grub_env_var *var, char **),
|
||||
grub_err_t (*write_hook) (struct grub_env_var *var))
|
||||
grub_env_read_hook_t read_hook,
|
||||
grub_env_write_hook_t write_hook)
|
||||
{
|
||||
struct grub_env_var *env = grub_env_find (var);
|
||||
|
||||
if (! env)
|
||||
if (grub_env_set (var, "") != GRUB_ERR_NONE)
|
||||
return grub_errno;
|
||||
{
|
||||
char *val = grub_strdup ("");
|
||||
|
||||
if (! val)
|
||||
return grub_errno;
|
||||
|
||||
if (grub_env_set (var, val) != GRUB_ERR_NONE)
|
||||
return grub_errno;
|
||||
}
|
||||
|
||||
env = grub_env_find (var);
|
||||
/* XXX Insert an assertion? */
|
||||
|
|
|
|||
18
kern/main.c
18
kern/main.c
|
|
@ -57,12 +57,28 @@ grub_load_modules (void)
|
|||
grub_mm_init_region ((void *) modinfo, modinfo->size);
|
||||
}
|
||||
|
||||
/* Write hook for the environment variables of root. Remove surrounding
|
||||
parentheses, if any. */
|
||||
static char *
|
||||
grub_env_write_root (struct grub_env_var *var, const char *val)
|
||||
{
|
||||
/* XXX Is it better to check the existence of the device? */
|
||||
grub_size_t len = grub_strlen (val);
|
||||
|
||||
if (val[0] == '(' && val[len - 1] == ')')
|
||||
return grub_strndup (val + 1, len - 2);
|
||||
|
||||
return grub_strdup (val);
|
||||
}
|
||||
|
||||
/* Set the root device according to the dl prefix. */
|
||||
static void
|
||||
grub_set_root_dev (void)
|
||||
{
|
||||
const char *prefix;
|
||||
|
||||
grub_register_variable_hook ("root", 0, grub_env_write_root);
|
||||
|
||||
prefix = grub_env_get ("prefix");
|
||||
|
||||
if (prefix)
|
||||
|
|
@ -72,7 +88,7 @@ grub_set_root_dev (void)
|
|||
dev = grub_file_get_device_name (prefix);
|
||||
if (dev)
|
||||
{
|
||||
grub_device_set_root (dev);
|
||||
grub_env_set ("root", dev);
|
||||
grub_free (dev);
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -336,8 +336,8 @@ grub_rescue_cmd_root (int argc, char *argv[])
|
|||
char *device_name = grub_file_get_device_name (argv[0]);
|
||||
if (! device_name)
|
||||
return;
|
||||
|
||||
grub_device_set_root (device_name);
|
||||
|
||||
grub_env_set ("root", device_name);
|
||||
grub_free (device_name);
|
||||
}
|
||||
|
||||
|
|
@ -350,7 +350,7 @@ grub_rescue_cmd_root (int argc, char *argv[])
|
|||
grub_errno = GRUB_ERR_NONE;
|
||||
|
||||
grub_printf ("(%s): Filesystem is %s.\n",
|
||||
grub_device_get_root (), fs ? fs->name : "unknown");
|
||||
grub_env_get ("root"), fs ? fs->name : "unknown");
|
||||
|
||||
grub_device_close (dev);
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue