2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2005,2007,2009,2010 Free Software Foundation, Inc.
2N/A *
2N/A * GRUB is free software: you can redistribute it and/or modify
2N/A * it under the terms of the GNU General Public License as published by
2N/A * the Free Software Foundation, either version 3 of the License, or
2N/A * (at your option) any later version.
2N/A *
2N/A * GRUB is distributed in the hope that it will be useful,
2N/A * but WITHOUT ANY WARRANTY; without even the implied warranty of
2N/A * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2N/A * GNU General Public License for more details.
2N/A *
2N/A * You should have received a copy of the GNU General Public License
2N/A * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
2N/A */
2N/A
2N/A#include <grub/misc.h>
2N/A#include <grub/script_sh.h>
2N/A#include <grub/parser.h>
2N/A#include <grub/mm.h>
2N/A#include <grub/charset.h>
2N/A
2N/Agrub_script_function_t grub_script_function_list;
2N/A
2N/Agrub_script_function_t
2N/Agrub_script_function_create (struct grub_script_arg *functionname_arg,
2N/A struct grub_script *cmd)
2N/A{
2N/A grub_script_function_t func;
2N/A grub_script_function_t *p;
2N/A
2N/A func = (grub_script_function_t) grub_malloc (sizeof (*func));
2N/A if (! func)
2N/A return 0;
2N/A
2N/A func->name = grub_strdup (functionname_arg->str);
2N/A if (! func->name)
2N/A {
2N/A grub_free (func);
2N/A return 0;
2N/A }
2N/A
2N/A func->func = cmd;
2N/A
2N/A /* Keep the list sorted for simplicity. */
2N/A p = &grub_script_function_list;
2N/A while (*p)
2N/A {
2N/A if (grub_strcmp ((*p)->name, func->name) >= 0)
2N/A break;
2N/A
2N/A p = &((*p)->next);
2N/A }
2N/A
2N/A /* If the function already exists, overwrite the old function. */
2N/A if (*p && grub_strcmp ((*p)->name, func->name) == 0)
2N/A {
2N/A grub_script_function_t q;
2N/A
2N/A q = *p;
2N/A grub_script_free (q->func);
2N/A q->func = cmd;
2N/A grub_free (func);
2N/A func = q;
2N/A }
2N/A else
2N/A {
2N/A func->next = *p;
2N/A *p = func;
2N/A }
2N/A
2N/A return func;
2N/A}
2N/A
2N/Avoid
2N/Agrub_script_function_remove (const char *name)
2N/A{
2N/A grub_script_function_t *p, q;
2N/A
2N/A for (p = &grub_script_function_list, q = *p; q; p = &(q->next), q = q->next)
2N/A if (grub_strcmp (name, q->name) == 0)
2N/A {
2N/A *p = q->next;
2N/A grub_free (q->name);
2N/A grub_script_free (q->func);
2N/A grub_free (q);
2N/A break;
2N/A }
2N/A}
2N/A
2N/Agrub_script_function_t
2N/Agrub_script_function_find (char *functionname)
2N/A{
2N/A grub_script_function_t func;
2N/A
2N/A for (func = grub_script_function_list; func; func = func->next)
2N/A if (grub_strcmp (functionname, func->name) == 0)
2N/A break;
2N/A
2N/A if (! func)
2N/A {
2N/A char tmp[21];
2N/A grub_strncpy (tmp, functionname, 20);
2N/A tmp[20] = 0;
2N/A /* Avoid truncating inside UTF-8 character. */
2N/A tmp[grub_getend (tmp, tmp + grub_strlen (tmp))] = 0;
2N/A grub_error (GRUB_ERR_UNKNOWN_COMMAND, "unknown command `%s'", tmp);
2N/A }
2N/A
2N/A return func;
2N/A}