2N/A

2N/A#include <grub/env.h>

2N/A#include <grub/env_private.h>

2N/A#include <grub/misc.h>

2N/A#include <grub/mm.h>

2N/A

2N/Astatic struct grub_env_context initial_context;

2N/A

2N/Astruct grub_env_context *grub_current_context = &initial_context;

2N/A

2N/Astatic unsigned int

2N/Agrub_env_hashval (const char *s)

2N/A{

2N/A unsigned int i = 0;

2N/A

2N/A /* XXX: This can be done much more efficiently. */

2N/A while (*s)

2N/A i += 5 * *(s++);

2N/A

2N/A return i % HASHSZ;

2N/A}

2N/A

2N/Astruct grub_env_var *

2N/Agrub_env_find (const char *name)

2N/A{

2N/A struct grub_env_var *var;

2N/A int idx = grub_env_hashval (name);

2N/A

2N/A /* Look for the variable in the current context. */

2N/A for (var = grub_current_context->vars[idx]; var; var = var->next)

2N/A if (grub_strcmp (var->name, name) == 0)

2N/A return var;

2N/A

2N/A return 0;

2N/A}

2N/A

2N/Astatic void

2N/Agrub_env_insert (struct grub_env_context *context,

2N/A struct grub_env_var *var)

2N/A{

2N/A int idx = grub_env_hashval (var->name);

2N/A

2N/A /* Insert the variable into the hashtable. */

2N/A var->prevp = &context->vars[idx];

2N/A var->next = context->vars[idx];

2N/A if (var->next)

2N/A var->next->prevp = &(var->next);

2N/A context->vars[idx] = var;

2N/A}

2N/A

2N/Astatic void

2N/Agrub_env_remove (struct grub_env_var *var)

2N/A{

2N/A /* Remove the entry from the variable table. */

2N/A *var->prevp = var->next;

2N/A if (var->next)

2N/A var->next->prevp = var->prevp;

2N/A}

2N/A

2N/Agrub_env_set (const char *name, const char *val)

2N/A{

2N/A struct grub_env_var *var;

2N/A

2N/A /* If the variable does already exist, just update the variable. */

2N/A var = grub_env_find (name);

2N/A if (var)

2N/A {

2N/A char *old = var->value;

2N/A

2N/A if (var->write_hook)

2N/A var->value = var->write_hook (var, val);

2N/A else

2N/A var->value = grub_strdup (val);

2N/A

2N/A if (! var->value)

2N/A {

2N/A var->value = old;

2N/A return grub_errno;

2N/A }

2N/A

2N/A grub_free (old);

2N/A return GRUB_ERR_NONE;

2N/A }

2N/A

2N/A /* The variable does not exist, so create a new one. */

2N/A var = grub_zalloc (sizeof (*var));

2N/A if (! var)

2N/A return grub_errno;

2N/A

2N/A /* This is not necessary. But leave this for readability. */

2N/A var->global = 0;

2N/A

2N/A var->name = grub_strdup (name);

2N/A if (! var->name)

2N/A goto fail;

2N/A

2N/A var->value = grub_strdup (val);

2N/A if (! var->value)

2N/A goto fail;

2N/A

2N/A grub_env_insert (grub_current_context, var);

2N/A

2N/A return GRUB_ERR_NONE;

2N/A

2N/A fail:

2N/A grub_free (var->name);

2N/A grub_free (var->value);

2N/A grub_free (var);

2N/A

2N/A return grub_errno;

2N/A}

2N/A

2N/Aconst char *

2N/Agrub_env_get (const char *name)

2N/A{

2N/A struct grub_env_var *var;

2N/A

2N/A var = grub_env_find (name);

2N/A if (! var)

2N/A return 0;

2N/A

2N/A if (var->read_hook)

2N/A return var->read_hook (var, var->value);

2N/A

2N/A return var->value;

2N/A}

2N/A

2N/Agrub_env_unset (const char *name)

2N/A{

2N/A struct grub_env_var *var;

2N/A

2N/A var = grub_env_find (name);

2N/A if (! var)

2N/A return;

2N/A

2N/A if (var->read_hook || var->write_hook)

2N/A {

2N/A grub_env_set (name, "");

2N/A return;

2N/A }

2N/A

2N/A grub_env_remove (var);

2N/A

2N/A grub_free (var->name);

2N/A grub_free (var->value);

2N/A grub_free (var);

2N/A}

2N/A

2N/Agrub_env_iterate (int (*func) (struct grub_env_var *var))

2N/A{

2N/A struct grub_env_sorted_var *sorted_list = 0;

2N/A struct grub_env_sorted_var *sorted_var;

2N/A int i;

2N/A

2N/A /* Add variables associated with this context into a sorted list. */

2N/A for (i = 0; i < HASHSZ; i++)

2N/A {

2N/A struct grub_env_var *var;

2N/A

2N/A for (var = grub_current_context->vars[i]; var; var = var->next)

2N/A {

2N/A struct grub_env_sorted_var *p, **q;

2N/A

2N/A sorted_var = grub_malloc (sizeof (*sorted_var));

2N/A if (! sorted_var)

2N/A goto fail;

2N/A

2N/A sorted_var->var = var;

2N/A

2N/A for (q = &sorted_list, p = *q; p; q = &((*q)->next), p = *q)

2N/A {

2N/A if (grub_strcmp (p->var->name, var->name) > 0)

2N/A break;

2N/A }

2N/A

2N/A sorted_var->next = *q;

2N/A *q = sorted_var;

2N/A }

2N/A }

2N/A

2N/A /* Iterate FUNC on the sorted list. */

2N/A for (sorted_var = sorted_list; sorted_var; sorted_var = sorted_var->next)

2N/A if (func (sorted_var->var))

2N/A break;

2N/A

2N/A fail:

2N/A

2N/A /* Free the sorted list. */

2N/A for (sorted_var = sorted_list; sorted_var; )

2N/A {

2N/A struct grub_env_sorted_var *tmp = sorted_var->next;

2N/A

2N/A grub_free (sorted_var);

2N/A sorted_var = tmp;

2N/A }

2N/A}

2N/A

2N/Agrub_register_variable_hook (const char *name,

2N/A grub_env_read_hook_t read_hook,

2N/A grub_env_write_hook_t write_hook)

2N/A{

2N/A struct grub_env_var *var = grub_env_find (name);

2N/A

2N/A if (! var)

2N/A {

2N/A if (grub_env_set (name, "") != GRUB_ERR_NONE)

2N/A return grub_errno;

2N/A

2N/A var = grub_env_find (name);

2N/A /* XXX Insert an assertion? */

2N/A }

2N/A

2N/A var->read_hook = read_hook;

2N/A var->write_hook = write_hook;

2N/A

2N/A return GRUB_ERR_NONE;

2N/A}

2N/A

2N/Agrub_env_export (const char *name)

2N/A{

2N/A struct grub_env_var *var;

2N/A

2N/A var = grub_env_find (name);

2N/A if (! var)

2N/A {

2N/A grub_err_t err;

2N/A

2N/A err = grub_env_set (name, "");

2N/A if (err)

2N/A return err;

2N/A var = grub_env_find (name);

2N/A }

2N/A var->global = 1;

2N/A

2N/A return GRUB_ERR_NONE;

2N/A}