2N/A/* parttool.c - common dispatcher and parser for partition operations */
2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2009 Free Software Foundation, Inc.
2N/A *
2N/A * This program 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 2 of the License, or
2N/A * (at your option) any later version.
2N/A *
2N/A * This program 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 this program; if not, write to the Free Software
2N/A * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
2N/A */
2N/A
2N/A#include <grub/types.h>
2N/A#include <grub/misc.h>
2N/A#include <grub/mm.h>
2N/A#include <grub/err.h>
2N/A#include <grub/dl.h>
2N/A#include <grub/normal.h>
2N/A#include <grub/device.h>
2N/A#include <grub/disk.h>
2N/A#include <grub/partition.h>
2N/A#include <grub/parttool.h>
2N/A#include <grub/command.h>
2N/A#include <grub/i18n.h>
2N/A
2N/AGRUB_MOD_LICENSE ("GPLv2+");
2N/A
2N/Astatic struct grub_parttool *parts = 0;
2N/Astatic int curhandle = 0;
2N/Astatic grub_dl_t mymod;
2N/Astatic char helpmsg[] =
2N/A N_("Perform COMMANDS on partition.\n"
2N/A "Use \"parttool PARTITION help\" for the list "
2N/A "of available commands.");
2N/A
2N/Aint
2N/Agrub_parttool_register(const char *part_name,
2N/A const grub_parttool_function_t func,
2N/A const struct grub_parttool_argdesc *args)
2N/A{
2N/A struct grub_parttool *cur;
2N/A int nargs = 0;
2N/A
2N/A if (! parts)
2N/A grub_dl_ref (mymod);
2N/A
2N/A cur = (struct grub_parttool *) grub_malloc (sizeof (struct grub_parttool));
2N/A cur->next = parts;
2N/A cur->name = grub_strdup (part_name);
2N/A cur->handle = curhandle++;
2N/A for (nargs = 0; args[nargs].name != 0; nargs++);
2N/A cur->nargs = nargs;
2N/A cur->args = (struct grub_parttool_argdesc *)
2N/A grub_malloc ((nargs + 1) * sizeof (struct grub_parttool_argdesc));
2N/A grub_memcpy (cur->args, args,
2N/A (nargs + 1) * sizeof (struct grub_parttool_argdesc));
2N/A
2N/A cur->func = func;
2N/A parts = cur;
2N/A return cur->handle;
2N/A}
2N/A
2N/Avoid
2N/Agrub_parttool_unregister (int handle)
2N/A{
2N/A struct grub_parttool *prev = 0, *cur, *t;
2N/A for (cur = parts; cur; )
2N/A if (cur->handle == handle)
2N/A {
2N/A grub_free (cur->args);
2N/A grub_free (cur->name);
2N/A if (prev)
2N/A prev->next = cur->next;
2N/A else
2N/A parts = cur->next;
2N/A t = cur;
2N/A cur = cur->next;
2N/A grub_free (t);
2N/A }
2N/A else
2N/A {
2N/A prev = cur;
2N/A cur = cur->next;
2N/A }
2N/A if (! parts)
2N/A grub_dl_unref (mymod);
2N/A}
2N/A
2N/Astatic grub_err_t
2N/Agrub_cmd_parttool (grub_command_t cmd __attribute__ ((unused)),
2N/A int argc, char **args)
2N/A{
2N/A grub_device_t dev;
2N/A struct grub_parttool *cur, *ptool;
2N/A int *parsed;
2N/A int i, j;
2N/A grub_err_t err = GRUB_ERR_NONE;
2N/A
2N/A auto grub_err_t show_help (void);
2N/A grub_err_t show_help (void)
2N/A {
2N/A int found = 0;
2N/A for (cur = parts; cur; cur = cur->next)
2N/A if (grub_strcmp (dev->disk->partition->partmap->name, cur->name) == 0)
2N/A {
2N/A struct grub_parttool_argdesc *curarg;
2N/A found = 1;
2N/A for (curarg = cur->args; curarg->name; curarg++)
2N/A {
2N/A int spacing = 20;
2N/A
2N/A spacing -= grub_strlen (curarg->name);
2N/A grub_printf ("%s", curarg->name);
2N/A
2N/A switch (curarg->type)
2N/A {
2N/A case GRUB_PARTTOOL_ARG_BOOL:
2N/A grub_printf ("+/-");
2N/A spacing -= 3;
2N/A break;
2N/A
2N/A case GRUB_PARTTOOL_ARG_VAL:
2N/A grub_xputs (_("=VAL"));
2N/A spacing -= 4;
2N/A break;
2N/A
2N/A case GRUB_PARTTOOL_ARG_END:
2N/A break;
2N/A }
2N/A while (spacing-- > 0)
2N/A grub_printf (" ");
2N/A grub_puts_ (curarg->desc);
2N/A }
2N/A }
2N/A if (! found)
2N/A grub_printf_ (N_("Sorry no parttool is available for %s\n"),
2N/A dev->disk->partition->partmap->name);
2N/A return GRUB_ERR_NONE;
2N/A }
2N/A
2N/A if (argc < 1)
2N/A {
2N/A grub_puts_ (helpmsg);
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "too few arguments");
2N/A }
2N/A
2N/A if (args[0][0] == '(' && args[0][grub_strlen (args[0]) - 1] == ')')
2N/A {
2N/A args[0][grub_strlen (args[0]) - 1] = 0;
2N/A dev = grub_device_open (args[0] + 1);
2N/A args[0][grub_strlen (args[0]) - 1] = ')';
2N/A }
2N/A else
2N/A dev = grub_device_open (args[0]);
2N/A
2N/A if (! dev)
2N/A return grub_errno;
2N/A
2N/A if (! dev->disk)
2N/A {
2N/A grub_device_close (dev);
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "not a disk");
2N/A }
2N/A
2N/A if (! dev->disk->partition)
2N/A {
2N/A grub_device_close (dev);
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "not a partition");
2N/A }
2N/A
2N/A /* Load modules. */
2N/A if (! grub_no_autoload)
2N/A {
2N/A const char *prefix;
2N/A prefix = grub_env_get ("prefix");
2N/A if (prefix)
2N/A {
2N/A char *filename;
2N/A
2N/A filename = grub_xasprintf ("%s/parttool.lst", prefix);
2N/A if (filename)
2N/A {
2N/A grub_file_t file;
2N/A
2N/A file = grub_file_open (filename);
2N/A if (file)
2N/A {
2N/A char *buf = 0;
2N/A for (;; grub_free(buf))
2N/A {
2N/A char *p, *name;
2N/A
2N/A buf = grub_file_getline (file);
2N/A
2N/A if (! buf)
2N/A break;
2N/A
2N/A name = buf;
2N/A
2N/A if (! grub_isgraph (name[0]))
2N/A continue;
2N/A
2N/A p = grub_strchr (name, ':');
2N/A if (! p)
2N/A continue;
2N/A
2N/A *p = '\0';
2N/A while (*++p == ' ')
2N/A ;
2N/A
2N/A if (! grub_isgraph (*p))
2N/A continue;
2N/A
2N/A if (grub_strcmp (name, dev->disk->partition->partmap->name)
2N/A != 0)
2N/A continue;
2N/A
2N/A grub_dl_load (p);
2N/A }
2N/A
2N/A grub_file_close (file);
2N/A }
2N/A
2N/A grub_free (filename);
2N/A }
2N/A }
2N/A /* Ignore errors. */
2N/A grub_errno = GRUB_ERR_NONE;
2N/A }
2N/A
2N/A if (argc == 1)
2N/A return show_help ();
2N/A
2N/A for (i = 1; i < argc; i++)
2N/A if (grub_strcmp (args[i], "help") == 0)
2N/A return show_help ();
2N/A
2N/A parsed = (int *) grub_zalloc (argc * sizeof (int));
2N/A
2N/A for (i = 1; i < argc; i++)
2N/A if (! parsed[i])
2N/A {
2N/A struct grub_parttool_argdesc *curarg;
2N/A struct grub_parttool_args *pargs;
2N/A for (cur = parts; cur; cur = cur->next)
2N/A if (grub_strcmp (dev->disk->partition->partmap->name, cur->name) == 0)
2N/A {
2N/A for (curarg = cur->args; curarg->name; curarg++)
2N/A if (grub_strncmp (curarg->name, args[i],
2N/A grub_strlen (curarg->name)) == 0
2N/A && ((curarg->type == GRUB_PARTTOOL_ARG_BOOL
2N/A && (args[i][grub_strlen (curarg->name)] == '+'
2N/A || args[i][grub_strlen (curarg->name)] == '-'
2N/A || args[i][grub_strlen (curarg->name)] == 0))
2N/A || (curarg->type == GRUB_PARTTOOL_ARG_VAL
2N/A && args[i][grub_strlen (curarg->name)] == '=')))
2N/A
2N/A break;
2N/A if (curarg->name)
2N/A break;
2N/A }
2N/A if (! cur)
2N/A return grub_error (GRUB_ERR_BAD_ARGUMENT, "unrecognised argument %s",
2N/A args[i]);
2N/A ptool = cur;
2N/A pargs = (struct grub_parttool_args *)
2N/A grub_zalloc (ptool->nargs * sizeof (struct grub_parttool_args));
2N/A for (j = i; j < argc; j++)
2N/A if (! parsed[j])
2N/A {
2N/A for (curarg = ptool->args; curarg->name; curarg++)
2N/A if (grub_strncmp (curarg->name, args[j],
2N/A grub_strlen (curarg->name)) == 0
2N/A && ((curarg->type == GRUB_PARTTOOL_ARG_BOOL
2N/A && (args[j][grub_strlen (curarg->name)] == '+'
2N/A || args[j][grub_strlen (curarg->name)] == '-'
2N/A || args[j][grub_strlen (curarg->name)] == 0))
2N/A || (curarg->type == GRUB_PARTTOOL_ARG_VAL
2N/A && args[j][grub_strlen (curarg->name)] == '=')))
2N/A {
2N/A parsed[j] = 1;
2N/A pargs[curarg - ptool->args].set = 1;
2N/A switch (curarg->type)
2N/A {
2N/A case GRUB_PARTTOOL_ARG_BOOL:
2N/A pargs[curarg - ptool->args].bool
2N/A = (args[j][grub_strlen (curarg->name)] != '-');
2N/A break;
2N/A
2N/A case GRUB_PARTTOOL_ARG_VAL:
2N/A pargs[curarg - ptool->args].str
2N/A = (args[j] + grub_strlen (curarg->name) + 1);
2N/A break;
2N/A
2N/A case GRUB_PARTTOOL_ARG_END:
2N/A break;
2N/A }
2N/A }
2N/A }
2N/A
2N/A err = ptool->func (dev, pargs);
2N/A grub_free (pargs);
2N/A if (err)
2N/A break;
2N/A }
2N/A
2N/A grub_free (parsed);
2N/A grub_device_close (dev);
2N/A return err;
2N/A}
2N/A
2N/Astatic grub_command_t cmd;
2N/A
2N/AGRUB_MOD_INIT(parttool)
2N/A{
2N/A mymod = mod;
2N/A cmd = grub_register_command ("parttool", grub_cmd_parttool,
2N/A N_("PARTITION COMMANDS"),
2N/A helpmsg);
2N/A}
2N/A
2N/AGRUB_MOD_FINI(parttool)
2N/A{
2N/A grub_unregister_command (cmd);
2N/A}