partition.c 6.37 KB
Newer Older
1 2
/*
 *  GRUB  --  GRand Unified Bootloader
3
 *  Copyright (C) 2004,2007  Free Software Foundation, Inc.
4
 *
5
 *  GRUB is free software: you can redistribute it and/or modify
6
 *  it under the terms of the GNU General Public License as published by
7
 *  the Free Software Foundation, either version 3 of the License, or
8 9
 *  (at your option) any later version.
 *
10
 *  GRUB is distributed in the hope that it will be useful,
11 12 13 14 15
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
16
 *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
17 18
 */

19
#include <grub/misc.h>
20
#include <grub/mm.h>
21 22
#include <grub/partition.h>
#include <grub/disk.h>
23
#include <grub/i18n.h>
24

25 26 27 28
#ifdef GRUB_UTIL
#include <grub/util/misc.h>
#endif

29
grub_partition_map_t grub_partition_map_list;
30

31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
/*
 * Checks that disk->partition contains part.  This function assumes that the
 * start of part is relative to the start of disk->partition.  Returns 1 if
 * disk->partition is null.
 */
static int
grub_partition_check_containment (const grub_disk_t disk,
				  const grub_partition_t part)
{
  if (disk->partition == NULL)
    return 1;

  if (part->start + part->len > disk->partition->len)
    {
      char *partname;

      partname = grub_partition_get_name (disk->partition);
      grub_dprintf ("partition", "sub-partition %s%d of (%s,%s) ends after parent.\n",
		    part->partmap->name, part->number + 1, disk->name, partname);
#ifdef GRUB_UTIL
51
      grub_util_warn (_("Discarding improperly nested partition (%s,%s,%s%d)"),
52 53 54 55 56 57 58 59 60 61
		      disk->name, partname, part->partmap->name, part->number + 1);
#endif
      grub_free (partname);

      return 0;
    }

  return 1;
}

62 63
/* Context for grub_partition_map_probe.  */
struct grub_partition_map_probe_ctx
64
{
65 66 67
  int partnum;
  grub_partition_t p;
};
68

69 70 71 72 73
/* Helper for grub_partition_map_probe.  */
static int
probe_iter (grub_disk_t dsk, const grub_partition_t partition, void *data)
{
  struct grub_partition_map_probe_ctx *ctx = data;
74

75 76
  if (ctx->partnum != partition->number)
    return 0;
77

78 79
  if (!(grub_partition_check_containment (dsk, partition)))
    return 0;
80

81 82 83
  ctx->p = (grub_partition_t) grub_malloc (sizeof (*ctx->p));
  if (! ctx->p)
    return 1;
84

85 86 87
  grub_memcpy (ctx->p, partition, sizeof (*ctx->p));
  return 1;
}
88

89 90 91 92 93 94 95 96 97 98
static grub_partition_t
grub_partition_map_probe (const grub_partition_map_t partmap,
			  grub_disk_t disk, int partnum)
{
  struct grub_partition_map_probe_ctx ctx = {
    .partnum = partnum,
    .p = 0
  };

  partmap->iterate (disk, probe_iter, &ctx);
99 100 101
  if (grub_errno)
    goto fail;

102
  return ctx.p;
103 104

 fail:
105
  grub_free (ctx.p);
106 107 108
  return 0;
}

109 110 111
grub_partition_t
grub_partition_probe (struct grub_disk *disk, const char *str)
{
112
  grub_partition_t part = 0;
113 114 115
  grub_partition_t curpart = 0;
  grub_partition_t tail;
  const char *ptr;
116

117 118 119 120
  part = tail = disk->partition;

  for (ptr = str; *ptr;)
    {
121 122 123 124 125 126 127 128 129
      grub_partition_map_t partmap;
      int num;
      const char *partname, *partname_end;

      partname = ptr;
      while (*ptr && grub_isalpha (*ptr))
	ptr++;
      partname_end = ptr; 
      num = grub_strtoul (ptr, (char **) &ptr, 0) - 1;
130 131 132

      curpart = 0;
      /* Use the first partition map type found.  */
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
      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;
      }
155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171

      if (! curpart)
	{
	  while (part)
	    {
	      curpart = part->parent;
	      grub_free (part);
	      part = curpart;
	    }
	  return 0;
	}
      curpart->parent = part;
      part = curpart;
      if (! ptr || *ptr != ',')
	break;
      ptr++;
    }
172

173 174 175
  return part;
}

176 177
/* Context for grub_partition_iterate.  */
struct grub_partition_iterate_ctx
178
{
179 180 181 182
  int ret;
  grub_partition_iterate_hook_t hook;
  void *hook_data;
};
183

184 185 186 187 188 189
/* Helper for grub_partition_iterate.  */
static int
part_iterate (grub_disk_t dsk, const grub_partition_t partition, void *data)
{
  struct grub_partition_iterate_ctx *ctx = data;
  struct grub_partition p = *partition;
190

191 192
  if (!(grub_partition_check_containment (dsk, partition)))
    return 0;
193

194 195 196 197 198 199
  p.parent = dsk->partition;
  dsk->partition = 0;
  if (ctx->hook (dsk, &p, ctx->hook_data))
    {
      ctx->ret = 1;
      return 1;
200
    }
201
  if (p.start != 0)
202
    {
203 204 205 206 207 208 209 210 211 212 213
      const struct grub_partition_map *partmap;
      dsk->partition = &p;
      FOR_PARTITION_MAPS(partmap)
      {
	grub_err_t err;
	err = partmap->iterate (dsk, part_iterate, ctx);
	if (err)
	  grub_errno = GRUB_ERR_NONE;
	if (ctx->ret)
	  break;
      }
214
    }
215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237
  dsk->partition = p.parent;
  return ctx->ret;
}

int
grub_partition_iterate (struct grub_disk *disk,
			grub_partition_iterate_hook_t hook, void *hook_data)
{
  struct grub_partition_iterate_ctx ctx = {
    .ret = 0,
    .hook = hook,
    .hook_data = hook_data
  };
  const struct grub_partition_map *partmap;

  FOR_PARTITION_MAPS(partmap)
  {
    grub_err_t err;
    err = partmap->iterate (disk, part_iterate, &ctx);
    if (err)
      grub_errno = GRUB_ERR_NONE;
    if (ctx.ret)
      break;
238
  }
239

240
  return ctx.ret;
241 242 243 244 245
}

char *
grub_partition_get_name (const grub_partition_t partition)
{
246
  char *out = 0, *ptr;
247
  grub_size_t needlen = 0;
248
  grub_partition_t part;
249 250
  if (!partition)
    return grub_strdup ("");
251 252 253
  for (part = partition; part; part = part->parent)
    /* Even on 64-bit machines this buffer is enough to hold
       longest number.  */
254 255
    needlen += grub_strlen (part->partmap->name) + 1 + 27;
  out = grub_malloc (needlen + 1);
256 257 258
  if (!out)
    return NULL;

259 260
  ptr = out + needlen;
  *ptr = 0;
261 262
  for (part = partition; part; part = part->parent)
    {
263 264 265 266 267 268 269 270 271 272
      char buf[27];
      grub_size_t len;
      grub_snprintf (buf, sizeof (buf), "%d", part->number + 1);
      len = grub_strlen (buf);
      ptr -= len;
      grub_memcpy (ptr, buf, len);
      len = grub_strlen (part->partmap->name);
      ptr -= len;
      grub_memcpy (ptr, part->partmap->name, len);
      *--ptr = ',';
273
    }
274
  grub_memmove (out, ptr + 1, out + needlen - ptr);
275
  return out;
276
}