2008-07-02 Bean <bean123ch@gmail.com>
* conf/common.rmk (bin_UTILITIES): Add grub-editenv. (grub_editenv_SOURCES): New variable. (pkglib_MODULES): Add loadenv.mod. (loadenv_mod_SOURCES): New variable. (loadenv_mod_CFLAGS): Likewise. (loadenv_mod_LDFLAGS): Likewise. * include/grub/envblk.h: New file. * util/envblk.c: New file. * util/grub-editenv.c: New file. * commands/loadenv.c: New file.
This commit is contained in:
bean
parent
d89b76343e
commit
2270f77bea
7 changed files with 973 additions and 124 deletions
156
util/envblk.c
Normal file
156
util/envblk.c
Normal file
|
|
@ -0,0 +1,156 @@
|
|||
/* envblk.c - Common function for environment block. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2008 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
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB 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 GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <config.h>
|
||||
#include <grub/types.h>
|
||||
#include <grub/envblk.h>
|
||||
#include <grub/misc.h>
|
||||
|
||||
grub_envblk_t
|
||||
grub_envblk_find (char *buf)
|
||||
{
|
||||
grub_uint32_t *pd;
|
||||
int len;
|
||||
|
||||
pd = (grub_uint32_t *) buf;
|
||||
|
||||
for (len = GRUB_ENVBLK_MAXLEN - 6; len > 0; len -= 4, pd++)
|
||||
if (*pd == GRUB_ENVBLK_SIGNATURE)
|
||||
{
|
||||
grub_envblk_t p;
|
||||
|
||||
p = (grub_envblk_t) pd;
|
||||
if (p->length <= len)
|
||||
return p;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
grub_envblk_insert (grub_envblk_t envblk, char *name, char *value)
|
||||
{
|
||||
char *p, *pend;
|
||||
char *found = 0;
|
||||
int nl;
|
||||
|
||||
nl = grub_strlen (name);
|
||||
p = envblk->data;
|
||||
pend = p + envblk->length;
|
||||
|
||||
while (*p)
|
||||
{
|
||||
if ((! found) && (! grub_memcmp (name, p, nl)) && (p[nl] == '='))
|
||||
found = p + nl + 1;
|
||||
|
||||
p += grub_strlen (p) + 1;
|
||||
if (p >= pend)
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (found)
|
||||
{
|
||||
int len1, len2;
|
||||
|
||||
len1 = grub_strlen (found);
|
||||
len2 = grub_strlen (value);
|
||||
if ((p - envblk->data) + 1 - len1 + len2 > envblk->length)
|
||||
return 1;
|
||||
|
||||
grub_memcpy (found + len2 + 1, found + len1 + 1, (p - found) - len1);
|
||||
grub_strcpy (found, value);
|
||||
}
|
||||
else
|
||||
{
|
||||
int len2 = grub_strlen (value);
|
||||
|
||||
if ((p - envblk->data) + nl + 1 + len2 + 2 > envblk->length)
|
||||
return 1;
|
||||
|
||||
grub_strcpy (p, name);
|
||||
p[nl] = '=';
|
||||
grub_strcpy (p + nl + 1, value);
|
||||
p[nl + 1 + len2 + 1] = 0;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void
|
||||
grub_envblk_delete (grub_envblk_t envblk, char *name)
|
||||
{
|
||||
char *p, *pend;
|
||||
char *found = 0;
|
||||
int nl;
|
||||
|
||||
nl = grub_strlen (name);
|
||||
p = envblk->data;
|
||||
pend = p + envblk->length;
|
||||
|
||||
while (*p)
|
||||
{
|
||||
if ((! found) && (! grub_memcmp (name, p, nl)) && (p[nl] == '='))
|
||||
found = p;
|
||||
|
||||
p += grub_strlen (p) + 1;
|
||||
if (p >= pend)
|
||||
return;
|
||||
}
|
||||
|
||||
if (found)
|
||||
{
|
||||
int len;
|
||||
|
||||
len = grub_strlen (found);
|
||||
grub_memcpy (found, found + len + 1, (p - found) - len);
|
||||
}
|
||||
}
|
||||
|
||||
void
|
||||
grub_envblk_iterate (grub_envblk_t envblk,
|
||||
int hook (char *name, char *value))
|
||||
{
|
||||
char *p, *pend;
|
||||
|
||||
p = envblk->data;
|
||||
pend = p + envblk->length;
|
||||
|
||||
while (*p)
|
||||
{
|
||||
char *v;
|
||||
int r;
|
||||
|
||||
v = grub_strchr (p, '=');
|
||||
if (v)
|
||||
{
|
||||
*v = 0;
|
||||
r = hook (p, v + 1);
|
||||
*v = '=';
|
||||
}
|
||||
else
|
||||
r = hook (p, "");
|
||||
|
||||
if (r)
|
||||
break;
|
||||
|
||||
p += grub_strlen (p) + 1;
|
||||
if (p >= pend)
|
||||
break;
|
||||
}
|
||||
}
|
||||
258
util/grub-editenv.c
Normal file
258
util/grub-editenv.c
Normal file
|
|
@ -0,0 +1,258 @@
|
|||
/* grub-editenv.c - tool to edit environment block. */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2008 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
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB 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 GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <config.h>
|
||||
#include <grub/types.h>
|
||||
#include <grub/util/misc.h>
|
||||
|
||||
#include <grub/envblk.h>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <getopt.h>
|
||||
|
||||
void
|
||||
grub_putchar (int c)
|
||||
{
|
||||
putchar (c);
|
||||
}
|
||||
|
||||
void
|
||||
grub_refresh (void)
|
||||
{
|
||||
fflush (stdout);
|
||||
}
|
||||
|
||||
void *
|
||||
grub_term_get_current (void)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
grub_getkey (void)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
char *
|
||||
grub_env_get (const char *name __attribute__ ((unused)))
|
||||
{
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static struct option options[] = {
|
||||
{"help", no_argument, 0, 'h'},
|
||||
{"version", no_argument, 0, 'V'},
|
||||
{"verbose", no_argument, 0, 'v'},
|
||||
{0, 0, 0, 0}
|
||||
};
|
||||
|
||||
char buffer[GRUB_ENVBLK_MAXLEN];
|
||||
grub_envblk_t envblk;
|
||||
|
||||
static void
|
||||
usage (int status)
|
||||
{
|
||||
if (status)
|
||||
fprintf (stderr, "Try ``grub-editenv --help'' for more information.\n");
|
||||
else
|
||||
printf ("\
|
||||
Usage: grub-editenv [OPTIONS] FILENAME COMMAND\n\
|
||||
\n\
|
||||
Tool to edit environment block.\n\
|
||||
\nCommands:\n\
|
||||
create create a blank environment block file\n\
|
||||
info show information about the environment block\n\
|
||||
list list the current variables\n\
|
||||
set [name=value] ... change/delete variables\n\
|
||||
clear delete all variables\n\
|
||||
\nOptions:\n\
|
||||
-h, --help display this message and exit\n\
|
||||
-V, --version print version information and exit\n\
|
||||
-v, --verbose print verbose messages\n\
|
||||
\n\
|
||||
Report bugs to <%s>.\n", PACKAGE_BUGREPORT);
|
||||
|
||||
exit (status);
|
||||
}
|
||||
|
||||
int
|
||||
create_envblk_file (char *name)
|
||||
{
|
||||
FILE *f;
|
||||
grub_envblk_t p;
|
||||
|
||||
f = fopen (name, "wb");
|
||||
if (! f)
|
||||
return 1;
|
||||
|
||||
/* Just in case OS don't save 0s. */
|
||||
memset (buffer, -1, sizeof (buffer));
|
||||
|
||||
p = (grub_envblk_t) &buffer[0];
|
||||
p->signature = GRUB_ENVBLK_SIGNATURE;
|
||||
p->length = sizeof (buffer) - sizeof (struct grub_envblk);
|
||||
p->data[0] = p->data[1] = 0;
|
||||
|
||||
fwrite (buffer, sizeof (buffer), 1, f);
|
||||
|
||||
fclose (f);
|
||||
return 0;
|
||||
}
|
||||
|
||||
FILE *
|
||||
open_envblk_file (char *name)
|
||||
{
|
||||
FILE *f;
|
||||
|
||||
f = fopen (name, "r+b");
|
||||
if (! f)
|
||||
grub_util_error ("Can\'t open file %s", name);
|
||||
|
||||
if (fread (buffer, 1, sizeof (buffer), f) != sizeof (buffer))
|
||||
grub_util_error ("The envblk file is too short");
|
||||
|
||||
envblk = grub_envblk_find (buffer);
|
||||
if (! envblk)
|
||||
grub_util_error ("Can\'t find environment block");
|
||||
|
||||
return f;
|
||||
}
|
||||
|
||||
static void
|
||||
cmd_info (void)
|
||||
{
|
||||
printf ("Envblk offset: %d\n", envblk->data - buffer);
|
||||
printf ("Envblk length: %d\n", envblk->length);
|
||||
}
|
||||
|
||||
static void
|
||||
cmd_list (void)
|
||||
{
|
||||
auto int hook (char *name, char *value);
|
||||
int hook (char *name, char *value)
|
||||
{
|
||||
printf ("%s=%s\n", name, value);
|
||||
return 0;
|
||||
}
|
||||
|
||||
grub_envblk_iterate (envblk, hook);
|
||||
}
|
||||
|
||||
static void
|
||||
cmd_set (int argc, char *argv[])
|
||||
{
|
||||
while (argc)
|
||||
{
|
||||
char *p;
|
||||
|
||||
p = strchr (argv[0], '=');
|
||||
if (! p)
|
||||
grub_util_error ("Invalid parameter");
|
||||
|
||||
*(p++) = 0;
|
||||
|
||||
if (*p)
|
||||
{
|
||||
if (grub_envblk_insert (envblk, argv[0], p))
|
||||
grub_util_error ("Environment block too small");
|
||||
}
|
||||
else
|
||||
grub_envblk_delete (envblk, argv[0]);
|
||||
|
||||
argc--;
|
||||
argv++;
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
cmd_clear (void)
|
||||
{
|
||||
envblk->data[0] = envblk->data[1] = 0;
|
||||
}
|
||||
|
||||
int
|
||||
main (int argc, char *argv[])
|
||||
{
|
||||
FILE *f;
|
||||
|
||||
progname = "grub-editenv";
|
||||
|
||||
/* Check for options. */
|
||||
while (1)
|
||||
{
|
||||
int c = getopt_long (argc, argv, "hVv", options, 0);
|
||||
|
||||
if (c == -1)
|
||||
break;
|
||||
else
|
||||
switch (c)
|
||||
{
|
||||
case 'h':
|
||||
usage (0);
|
||||
break;
|
||||
|
||||
case 'V':
|
||||
printf ("%s (%s) %s\n", progname, PACKAGE_NAME, PACKAGE_VERSION);
|
||||
return 0;
|
||||
|
||||
case 'v':
|
||||
verbosity++;
|
||||
break;
|
||||
|
||||
default:
|
||||
usage (1);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/* Obtain PATH. */
|
||||
if (optind + 1 >= argc)
|
||||
{
|
||||
fprintf (stderr, "Not enough parameter.\n");
|
||||
usage (1);
|
||||
}
|
||||
|
||||
if (! strcmp (argv[optind + 1], "create"))
|
||||
return create_envblk_file (argv[optind]);
|
||||
|
||||
f = open_envblk_file (argv[optind]);
|
||||
|
||||
optind++;
|
||||
if (! strcmp (argv[optind], "info"))
|
||||
cmd_info ();
|
||||
else if (! strcmp (argv[optind], "list"))
|
||||
cmd_list ();
|
||||
else
|
||||
{
|
||||
if (! strcmp (argv[optind], "set"))
|
||||
cmd_set (argc - optind - 1, argv + optind + 1);
|
||||
else if (! strcmp (argv[optind], "clear"))
|
||||
cmd_clear ();
|
||||
|
||||
fseek (f, 0, SEEK_SET);
|
||||
fwrite (buffer, sizeof (buffer), 1, f);
|
||||
}
|
||||
fclose (f);
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue