2N/A/*
2N/A * GRUB -- GRand Unified Bootloader
2N/A * Copyright (C) 2004,2007 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/mm.h>
2N/A#include <grub/partition.h>
2N/A#include <grub/disk.h>
2N/A#include <grub/i18n.h>
2N/A
2N/A#ifdef GRUB_UTIL
2N/A#include <grub/util/misc.h>
2N/A#endif
2N/A
2N/Agrub_partition_map_t grub_partition_map_list;
2N/A
2N/A/*
2N/A * Checks that disk->partition contains part. This function assumes that the
2N/A * start of part is relative to the start of disk->partition. Returns 1 if
2N/A * disk->partition is null.
2N/A */
2N/Astatic int
2N/Agrub_partition_check_containment (const grub_disk_t disk,
2N/A const grub_partition_t part)
2N/A{
2N/A if (disk->partition == NULL)
2N/A return 1;
2N/A
2N/A if (part->start + part->len > disk->partition->len)
2N/A {
2N/A char *partname;
2N/A
2N/A partname = grub_partition_get_name (disk->partition);
2N/A grub_dprintf ("partition", "sub-partition %s%d of (%s,%s) ends after parent.\n",
2N/A part->partmap->name, part->number + 1, disk->name, partname);
2N/A#ifdef GRUB_UTIL
2N/A grub_util_warn (_("Discarding improperly nested partition (%s,%s,%s%d)"),
2N/A disk->name, partname, part->partmap->name, part->number + 1);
2N/A#endif
2N/A grub_free (partname);
2N/A
2N/A return 0;
2N/A }
2N/A
2N/A return 1;
2N/A}
2N/A
2N/Astatic grub_partition_t
2N/Agrub_partition_map_probe (const grub_partition_map_t partmap,
2N/A grub_disk_t disk, int partnum)
2N/A{
2N/A grub_partition_t p = 0;
2N/A
2N/A auto int find_func (grub_disk_t d, const grub_partition_t partition);
2N/A
2N/A int find_func (grub_disk_t dsk,
2N/A const grub_partition_t partition)
2N/A {
2N/A if (partnum != partition->number)
2N/A return 0;
2N/A
2N/A if (!(grub_partition_check_containment (dsk, partition)))
2N/A return 0;
2N/A
2N/A p = (grub_partition_t) grub_malloc (sizeof (*p));
2N/A if (! p)
2N/A return 1;
2N/A
2N/A grub_memcpy (p, partition, sizeof (*p));
2N/A return 1;
2N/A }
2N/A
2N/A partmap->iterate (disk, find_func);
2N/A if (grub_errno)
2N/A goto fail;
2N/A
2N/A return p;
2N/A
2N/A fail:
2N/A grub_free (p);
2N/A return 0;
2N/A}
2N/A
2N/Agrub_partition_t
2N/Agrub_partition_probe (struct grub_disk *disk, const char *str)
2N/A{
2N/A grub_partition_t part = 0;
2N/A grub_partition_t curpart = 0;
2N/A grub_partition_t tail;
2N/A const char *ptr;
2N/A
2N/A part = tail = disk->partition;
2N/A
2N/A for (ptr = str; *ptr;)
2N/A {
2N/A grub_partition_map_t partmap;
2N/A int num;
2N/A const char *partname, *partname_end;
2N/A
2N/A partname = ptr;
2N/A while (*ptr && grub_isalpha (*ptr))
2N/A ptr++;
2N/A partname_end = ptr;
2N/A num = grub_strtoul (ptr, (char **) &ptr, 0) - 1;
2N/A
2N/A curpart = 0;
2N/A /* Use the first partition map type found. */
2N/A FOR_PARTITION_MAPS(partmap)
2N/A {
2N/A if (partname_end != partname &&
2N/A (grub_strncmp (partmap->name, partname, partname_end - partname)
2N/A != 0 || partmap->name[partname_end - partname] != 0))
2N/A continue;
2N/A
2N/A disk->partition = part;
2N/A curpart = grub_partition_map_probe (partmap, disk, num);
2N/A disk->partition = tail;
2N/A if (curpart)
2N/A break;
2N/A
2N/A if (grub_errno == GRUB_ERR_BAD_PART_TABLE)
2N/A {
2N/A /* Continue to next partition map type. */
2N/A grub_errno = GRUB_ERR_NONE;
2N/A continue;
2N/A }
2N/A
2N/A break;
2N/A }
2N/A
2N/A if (! curpart)
2N/A {
2N/A while (part)
2N/A {
2N/A curpart = part->parent;
2N/A grub_free (part);
2N/A part = curpart;
2N/A }
2N/A return 0;
2N/A }
2N/A curpart->parent = part;
2N/A part = curpart;
2N/A if (! ptr || *ptr != ',')
2N/A break;
2N/A ptr++;
2N/A }
2N/A
2N/A return part;
2N/A}
2N/A
2N/Aint
2N/Agrub_partition_iterate (struct grub_disk *disk,
2N/A int (*hook) (grub_disk_t disk,
2N/A const grub_partition_t partition))
2N/A{
2N/A int ret = 0;
2N/A
2N/A auto int part_iterate (grub_disk_t dsk, const grub_partition_t p);
2N/A
2N/A int part_iterate (grub_disk_t dsk,
2N/A const grub_partition_t partition)
2N/A {
2N/A struct grub_partition p = *partition;
2N/A
2N/A if (!(grub_partition_check_containment (dsk, partition)))
2N/A return 0;
2N/A
2N/A p.parent = dsk->partition;
2N/A dsk->partition = 0;
2N/A if (hook (dsk, &p))
2N/A {
2N/A ret = 1;
2N/A return 1;
2N/A }
2N/A if (p.start != 0)
2N/A {
2N/A const struct grub_partition_map *partmap;
2N/A dsk->partition = &p;
2N/A FOR_PARTITION_MAPS(partmap)
2N/A {
2N/A grub_err_t err;
2N/A err = partmap->iterate (dsk, part_iterate);
2N/A if (err)
2N/A grub_errno = GRUB_ERR_NONE;
2N/A if (ret)
2N/A break;
2N/A }
2N/A }
2N/A dsk->partition = p.parent;
2N/A return ret;
2N/A }
2N/A
2N/A {
2N/A const struct grub_partition_map *partmap;
2N/A FOR_PARTITION_MAPS(partmap)
2N/A {
2N/A grub_err_t err;
2N/A err = partmap->iterate (disk, part_iterate);
2N/A if (err)
2N/A grub_errno = GRUB_ERR_NONE;
2N/A if (ret)
2N/A break;
2N/A }
2N/A }
2N/A
2N/A return ret;
2N/A}
2N/A
2N/Achar *
2N/Agrub_partition_get_name (const grub_partition_t partition)
2N/A{
2N/A char *out = 0;
2N/A int curlen = 0;
2N/A grub_partition_t part;
2N/A for (part = partition; part; part = part->parent)
2N/A {
2N/A /* Even on 64-bit machines this buffer is enough to hold
2N/A longest number. */
2N/A char buf[grub_strlen (part->partmap->name) + 25];
2N/A int strl;
2N/A grub_snprintf (buf, sizeof (buf), "%s%d", part->partmap->name,
2N/A part->number + 1);
2N/A strl = grub_strlen (buf);
2N/A if (curlen)
2N/A {
2N/A out = grub_realloc (out, curlen + strl + 2);
2N/A grub_memcpy (out + strl + 1, out, curlen);
2N/A out[curlen + 1 + strl] = 0;
2N/A grub_memcpy (out, buf, strl);
2N/A out[strl] = ',';
2N/A curlen = curlen + 1 + strl;
2N/A }
2N/A else
2N/A {
2N/A curlen = strl;
2N/A out = grub_strdup (buf);
2N/A }
2N/A }
2N/A return out;
2N/A}