1
/* env.c - Environment variables */
3
* GRUB -- GRand Unified Bootloader
4
* Copyright (C) 2003,2005,2006,2007,2008,2009 Free Software Foundation, Inc.
6
* GRUB is free software: you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation, either version 3 of the License, or
9
* (at your option) any later version.
11
* GRUB is distributed in the hope that it will be useful,
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
* GNU General Public License for more details.
16
* You should have received a copy of the GNU General Public License
17
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
21
#include <grub/env_private.h>
22
#include <grub/misc.h>
24
#include <grub/command.h>
25
#include <grub/normal.h>
26
#include <grub/i18n.h>
31
struct menu_pointer *prev;
34
struct menu_pointer initial_menu;
35
struct menu_pointer *current_menu = &initial_menu;
38
grub_env_unset_menu (void)
40
current_menu->menu = NULL;
44
grub_env_get_menu (void)
46
return current_menu->menu;
50
grub_env_set_menu (grub_menu_t nmenu)
52
current_menu->menu = nmenu;
56
grub_env_new_context (int export_all)
58
struct grub_env_context *context;
60
struct menu_pointer *menu;
62
context = grub_zalloc (sizeof (*context));
65
menu = grub_zalloc (sizeof (*menu));
69
context->prev = grub_current_context;
70
grub_current_context = context;
72
menu->prev = current_menu;
75
/* Copy exported variables. */
76
for (i = 0; i < HASHSZ; i++)
78
struct grub_env_var *var;
80
for (var = context->prev->vars[i]; var; var = var->next)
81
if (var->global || export_all)
83
if (grub_env_set (var->name, var->value) != GRUB_ERR_NONE)
85
grub_env_context_close ();
88
grub_env_export (var->name);
89
grub_register_variable_hook (var->name, var->read_hook, var->write_hook);
97
grub_env_context_open (void)
99
return grub_env_new_context (0);
102
int grub_extractor_level = 0;
105
grub_env_extractor_open (int source)
107
grub_extractor_level++;
108
return grub_env_new_context (source);
112
grub_env_context_close (void)
114
struct grub_env_context *context;
116
struct menu_pointer *menu;
118
if (! grub_current_context->prev)
119
return grub_error (GRUB_ERR_BAD_ARGUMENT,
120
"cannot close the initial context");
122
/* Free the variables associated with this context. */
123
for (i = 0; i < HASHSZ; i++)
125
struct grub_env_var *p, *q;
127
for (p = grub_current_context->vars[i]; p; p = q)
131
grub_free (p->value);
136
/* Restore the previous context. */
137
context = grub_current_context->prev;
138
grub_free (grub_current_context);
139
grub_current_context = context;
141
menu = current_menu->prev;
142
grub_free (current_menu);
145
return GRUB_ERR_NONE;
149
grub_env_extractor_close (int source)
151
grub_menu_t menu, menu2;
152
grub_menu_entry_t *last;
157
menu = grub_env_get_menu ();
158
grub_env_unset_menu ();
160
err = grub_env_context_close ();
164
menu2 = grub_env_get_menu ();
166
last = &menu2->entry_list;
168
last = &(*last)->next;
170
*last = menu->entry_list;
171
menu2->size += menu->size;
174
grub_extractor_level--;
179
grub_env_export (const char *name)
181
struct grub_env_var *var;
183
var = grub_env_find (name);
188
err = grub_env_set (name, "");
191
var = grub_env_find (name);
195
return GRUB_ERR_NONE;
198
static grub_command_t export_cmd;
201
grub_cmd_export (struct grub_command *cmd __attribute__ ((unused)),
202
int argc, char **args)
207
return grub_error (GRUB_ERR_BAD_ARGUMENT,
208
"no environment variable specified");
210
for (i = 0; i < argc; i++)
211
grub_env_export (args[i]);
217
grub_context_init (void)
219
grub_env_export ("root");
220
grub_env_export ("prefix");
222
export_cmd = grub_register_command ("export", grub_cmd_export,
223
N_("ENVVAR [ENVVAR] ..."),
224
N_("Export variables."));
228
grub_context_fini (void)
230
grub_unregister_command (export_cmd);