2N/A/* argv.c - methods for constructing argument vector */
2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 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/mm.h>
2N/A#include <grub/misc.h>
2N/A#include <grub/script_sh.h>
2N/A
2N/A/* Return nearest power of two that is >= v. */
2N/Astatic unsigned
2N/Around_up_exp (unsigned v)
2N/A{
2N/A COMPILE_TIME_ASSERT (sizeof (v) == 4);
2N/A
2N/A v--;
2N/A v |= v >> 1;
2N/A v |= v >> 2;
2N/A v |= v >> 4;
2N/A v |= v >> 8;
2N/A v |= v >> 16;
2N/A
2N/A v++;
2N/A v += (v == 0);
2N/A
2N/A return v;
2N/A}
2N/A
2N/Avoid
2N/Agrub_script_argv_free (struct grub_script_argv *argv)
2N/A{
2N/A unsigned i;
2N/A
2N/A if (argv->args)
2N/A {
2N/A for (i = 0; i < argv->argc; i++)
2N/A grub_free (argv->args[i]);
2N/A
2N/A grub_free (argv->args);
2N/A }
2N/A
2N/A argv->argc = 0;
2N/A argv->args = 0;
2N/A argv->script = 0;
2N/A}
2N/A
2N/A/* Make argv from argc, args pair. */
2N/Aint
2N/Agrub_script_argv_make (struct grub_script_argv *argv, int argc, char **args)
2N/A{
2N/A int i;
2N/A struct grub_script_argv r = { 0, 0, 0 };
2N/A
2N/A for (i = 0; i < argc; i++)
2N/A if (grub_script_argv_next (&r)
2N/A || grub_script_argv_append (&r, args[i], grub_strlen (args[i])))
2N/A {
2N/A grub_script_argv_free (&r);
2N/A return 1;
2N/A }
2N/A *argv = r;
2N/A return 0;
2N/A}
2N/A
2N/A/* Prepare for next argc. */
2N/Aint
2N/Agrub_script_argv_next (struct grub_script_argv *argv)
2N/A{
2N/A char **p = argv->args;
2N/A
2N/A if (argv->args && argv->argc && argv->args[argv->argc - 1] == 0)
2N/A return 0;
2N/A
2N/A p = grub_realloc (p, round_up_exp ((argv->argc + 2) * sizeof (char *)));
2N/A if (! p)
2N/A return 1;
2N/A
2N/A argv->argc++;
2N/A argv->args = p;
2N/A
2N/A if (argv->argc == 1)
2N/A argv->args[0] = 0;
2N/A argv->args[argv->argc] = 0;
2N/A return 0;
2N/A}
2N/A
2N/A/* Append `s' to the last argument. */
2N/Aint
2N/Agrub_script_argv_append (struct grub_script_argv *argv, const char *s,
2N/A grub_size_t slen)
2N/A{
2N/A grub_size_t a;
2N/A char *p = argv->args[argv->argc - 1];
2N/A
2N/A if (! s)
2N/A return 0;
2N/A
2N/A a = p ? grub_strlen (p) : 0;
2N/A
2N/A p = grub_realloc (p, round_up_exp ((a + slen + 1) * sizeof (char)));
2N/A if (! p)
2N/A return 1;
2N/A
2N/A grub_memcpy (p + a, s, slen);
2N/A p[a+slen] = 0;
2N/A argv->args[argv->argc - 1] = p;
2N/A
2N/A return 0;
2N/A}
2N/A
2N/A/* Split `s' and append words as multiple arguments. */
2N/Aint
2N/Agrub_script_argv_split_append (struct grub_script_argv *argv, const char *s)
2N/A{
2N/A const char *p;
2N/A int errors = 0;
2N/A
2N/A if (! s)
2N/A return 0;
2N/A
2N/A while (! errors && *s)
2N/A {
2N/A p = s;
2N/A while (*s && ! grub_isspace (*s))
2N/A s++;
2N/A
2N/A errors += grub_script_argv_append (argv, p, s - p);
2N/A
2N/A while (*s && grub_isspace (*s))
2N/A s++;
2N/A
2N/A if (*s)
2N/A errors += grub_script_argv_next (argv);
2N/A }
2N/A return errors;
2N/A}