partition.c revision 2
8326d453818c9fb78ac1670cf49d26fa4da15003venki/*
8326d453818c9fb78ac1670cf49d26fa4da15003venki * GRUB -- GRand Unified Bootloader
8326d453818c9fb78ac1670cf49d26fa4da15003venki * Copyright (C) 2004,2007 Free Software Foundation, Inc.
8326d453818c9fb78ac1670cf49d26fa4da15003venki *
8326d453818c9fb78ac1670cf49d26fa4da15003venki * GRUB is free software: you can redistribute it and/or modify
8326d453818c9fb78ac1670cf49d26fa4da15003venki * it under the terms of the GNU General Public License as published by
8326d453818c9fb78ac1670cf49d26fa4da15003venki * the Free Software Foundation, either version 3 of the License, or
8326d453818c9fb78ac1670cf49d26fa4da15003venki * (at your option) any later version.
8326d453818c9fb78ac1670cf49d26fa4da15003venki *
8326d453818c9fb78ac1670cf49d26fa4da15003venki * GRUB is distributed in the hope that it will be useful,
8326d453818c9fb78ac1670cf49d26fa4da15003venki * but WITHOUT ANY WARRANTY; without even the implied warranty of
8326d453818c9fb78ac1670cf49d26fa4da15003venki * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
8326d453818c9fb78ac1670cf49d26fa4da15003venki * GNU General Public License for more details.
8326d453818c9fb78ac1670cf49d26fa4da15003venki *
8326d453818c9fb78ac1670cf49d26fa4da15003venki * You should have received a copy of the GNU General Public License
8326d453818c9fb78ac1670cf49d26fa4da15003venki * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
8326d453818c9fb78ac1670cf49d26fa4da15003venki */
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/misc.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/mm.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/partition.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/disk.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/i18n.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki#ifdef GRUB_UTIL
8326d453818c9fb78ac1670cf49d26fa4da15003venki#include <grub/util/misc.h>
8326d453818c9fb78ac1670cf49d26fa4da15003venki#endif
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venkigrub_partition_map_t grub_partition_map_list;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki/*
8326d453818c9fb78ac1670cf49d26fa4da15003venki * Checks that disk->partition contains part. This function assumes that the
8326d453818c9fb78ac1670cf49d26fa4da15003venki * start of part is relative to the start of disk->partition. Returns 1 if
8326d453818c9fb78ac1670cf49d26fa4da15003venki * disk->partition is null.
8326d453818c9fb78ac1670cf49d26fa4da15003venki */
8326d453818c9fb78ac1670cf49d26fa4da15003venkistatic int
8326d453818c9fb78ac1670cf49d26fa4da15003venkigrub_partition_check_containment (const grub_disk_t disk,
8326d453818c9fb78ac1670cf49d26fa4da15003venki const grub_partition_t part)
8326d453818c9fb78ac1670cf49d26fa4da15003venki{
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (disk->partition == NULL)
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 1;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (part->start + part->len > disk->partition->len)
8326d453818c9fb78ac1670cf49d26fa4da15003venki {
8326d453818c9fb78ac1670cf49d26fa4da15003venki char *partname;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki partname = grub_partition_get_name (disk->partition);
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_dprintf ("partition", "sub-partition %s%d of (%s,%s) ends after parent.\n",
8326d453818c9fb78ac1670cf49d26fa4da15003venki part->partmap->name, part->number + 1, disk->name, partname);
8326d453818c9fb78ac1670cf49d26fa4da15003venki#ifdef GRUB_UTIL
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_util_warn (_("Discarding improperly nested partition (%s,%s,%s%d)"),
8326d453818c9fb78ac1670cf49d26fa4da15003venki disk->name, partname, part->partmap->name, part->number + 1);
8326d453818c9fb78ac1670cf49d26fa4da15003venki#endif
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_free (partname);
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki }
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 1;
8326d453818c9fb78ac1670cf49d26fa4da15003venki}
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venkistatic grub_partition_t
8326d453818c9fb78ac1670cf49d26fa4da15003venkigrub_partition_map_probe (const grub_partition_map_t partmap,
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_disk_t disk, int partnum)
8326d453818c9fb78ac1670cf49d26fa4da15003venki{
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_partition_t p = 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki auto int find_func (grub_disk_t d, const grub_partition_t partition);
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki int find_func (grub_disk_t dsk,
8326d453818c9fb78ac1670cf49d26fa4da15003venki const grub_partition_t partition)
8326d453818c9fb78ac1670cf49d26fa4da15003venki {
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (partnum != partition->number)
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (!(grub_partition_check_containment (dsk, partition)))
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki p = (grub_partition_t) grub_malloc (sizeof (*p));
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (! p)
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 1;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_memcpy (p, partition, sizeof (*p));
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 1;
8326d453818c9fb78ac1670cf49d26fa4da15003venki }
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki partmap->iterate (disk, find_func);
8326d453818c9fb78ac1670cf49d26fa4da15003venki if (grub_errno)
8326d453818c9fb78ac1670cf49d26fa4da15003venki goto fail;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki return p;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki fail:
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_free (p);
8326d453818c9fb78ac1670cf49d26fa4da15003venki return 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki}
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venkigrub_partition_t
8326d453818c9fb78ac1670cf49d26fa4da15003venkigrub_partition_probe (struct grub_disk *disk, const char *str)
8326d453818c9fb78ac1670cf49d26fa4da15003venki{
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_partition_t part = 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_partition_t curpart = 0;
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_partition_t tail;
8326d453818c9fb78ac1670cf49d26fa4da15003venki const char *ptr;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki part = tail = disk->partition;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki for (ptr = str; *ptr;)
8326d453818c9fb78ac1670cf49d26fa4da15003venki {
8326d453818c9fb78ac1670cf49d26fa4da15003venki grub_partition_map_t partmap;
8326d453818c9fb78ac1670cf49d26fa4da15003venki int num;
8326d453818c9fb78ac1670cf49d26fa4da15003venki const char *partname, *partname_end;
8326d453818c9fb78ac1670cf49d26fa4da15003venki
8326d453818c9fb78ac1670cf49d26fa4da15003venki partname = ptr;
8326d453818c9fb78ac1670cf49d26fa4da15003venki while (*ptr && grub_isalpha (*ptr))
8326d453818c9fb78ac1670cf49d26fa4da15003venki ptr++;
8326d453818c9fb78ac1670cf49d26fa4da15003venki partname_end = ptr;
num = grub_strtoul (ptr, (char **) &ptr, 0) - 1;
curpart = 0;
/* Use the first partition map type found. */
FOR_PARTITION_MAPS(partmap)
{
if (partname_end != partname &&
(grub_strncmp (partmap->name, partname, partname_end - partname)
!= 0 || partmap->name[partname_end - partname] != 0))
continue;
disk->partition = part;
curpart = grub_partition_map_probe (partmap, disk, num);
disk->partition = tail;
if (curpart)
break;
if (grub_errno == GRUB_ERR_BAD_PART_TABLE)
{
/* Continue to next partition map type. */
grub_errno = GRUB_ERR_NONE;
continue;
}
break;
}
if (! curpart)
{
while (part)
{
curpart = part->parent;
grub_free (part);
part = curpart;
}
return 0;
}
curpart->parent = part;
part = curpart;
if (! ptr || *ptr != ',')
break;
ptr++;
}
return part;
}
int
grub_partition_iterate (struct grub_disk *disk,
int (*hook) (grub_disk_t disk,
const grub_partition_t partition))
{
int ret = 0;
auto int part_iterate (grub_disk_t dsk, const grub_partition_t p);
int part_iterate (grub_disk_t dsk,
const grub_partition_t partition)
{
struct grub_partition p = *partition;
if (!(grub_partition_check_containment (dsk, partition)))
return 0;
p.parent = dsk->partition;
dsk->partition = 0;
if (hook (dsk, &p))
{
ret = 1;
return 1;
}
if (p.start != 0)
{
const struct grub_partition_map *partmap;
dsk->partition = &p;
FOR_PARTITION_MAPS(partmap)
{
grub_err_t err;
err = partmap->iterate (dsk, part_iterate);
if (err)
grub_errno = GRUB_ERR_NONE;
if (ret)
break;
}
}
dsk->partition = p.parent;
return ret;
}
{
const struct grub_partition_map *partmap;
FOR_PARTITION_MAPS(partmap)
{
grub_err_t err;
err = partmap->iterate (disk, part_iterate);
if (err)
grub_errno = GRUB_ERR_NONE;
if (ret)
break;
}
}
return ret;
}
char *
grub_partition_get_name (const grub_partition_t partition)
{
char *out = 0;
int curlen = 0;
grub_partition_t part;
for (part = partition; part; part = part->parent)
{
/* Even on 64-bit machines this buffer is enough to hold
longest number. */
char buf[grub_strlen (part->partmap->name) + 25];
int strl;
grub_snprintf (buf, sizeof (buf), "%s%d", part->partmap->name,
part->number + 1);
strl = grub_strlen (buf);
if (curlen)
{
out = grub_realloc (out, curlen + strl + 2);
grub_memcpy (out + strl + 1, out, curlen);
out[curlen + 1 + strl] = 0;
grub_memcpy (out, buf, strl);
out[strl] = ',';
curlen = curlen + 1 + strl;
}
else
{
curlen = strl;
out = grub_strdup (buf);
}
}
return out;
}