gpt.c revision 1.2 1 1.2 martin /* $NetBSD: gpt.c,v 1.2 2019/06/23 11:47:08 martin Exp $ */
2 1.1 martin
3 1.1 martin /*
4 1.1 martin * Copyright 2018 The NetBSD Foundation, Inc.
5 1.1 martin * All rights reserved.
6 1.1 martin *
7 1.1 martin * Redistribution and use in source and binary forms, with or without
8 1.1 martin * modification, are permitted provided that the following conditions
9 1.1 martin * are met:
10 1.1 martin * 1. Redistributions of source code must retain the above copyright
11 1.1 martin * notice, this list of conditions and the following disclaimer.
12 1.1 martin * 2. Redistributions in binary form must reproduce the above copyright
13 1.1 martin * notice, this list of conditions and the following disclaimer in the
14 1.1 martin * documentation and/or other materials provided with the distribution.
15 1.1 martin *
16 1.1 martin * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
17 1.1 martin * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 1.1 martin * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 1.1 martin * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
20 1.1 martin * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 1.1 martin * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 1.1 martin * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 1.1 martin * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 1.1 martin * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 1.1 martin * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
26 1.1 martin * THE POSSIBILITY OF SUCH DAMAGE.
27 1.1 martin *
28 1.1 martin */
29 1.1 martin
30 1.1 martin #include "defs.h"
31 1.1 martin #include "mbr.h"
32 1.1 martin #include "md.h"
33 1.1 martin #include "gpt_uuid.h"
34 1.1 martin #include <assert.h>
35 1.1 martin #include <paths.h>
36 1.1 martin #include <sys/param.h>
37 1.1 martin #include <sys/ioctl.h>
38 1.1 martin #include <util.h>
39 1.1 martin
40 1.1 martin bool gpt_parts_check(void); /* check for needed binaries */
41 1.1 martin
42 1.1 martin
43 1.1 martin /*************** GPT ************************************************/
44 1.1 martin /* a GPT based disk_partitions interface */
45 1.1 martin
46 1.1 martin #define GUID_STR_LEN 40
47 1.1 martin #define GPT_PTYPE_MAX 32 /* should be > gpt type -l | wc -l */
48 1.1 martin #define GPT_DEV_LEN 16 /* dkNN */
49 1.1 martin
50 1.1 martin #define GPT_PARTS_PER_SEC 4 /* a 512 byte sector hols 4 entries */
51 1.1 martin #define GPT_DEFAULT_MAX_PARTS 128
52 1.1 martin
53 1.1 martin /* a usable label will be short, so we can get away with an arbitrary limit */
54 1.1 martin #define GPT_LABEL_LEN 96
55 1.1 martin
56 1.1 martin #define GPT_ATTR_BIOSBOOT 1
57 1.1 martin #define GPT_ATTR_BOOTME 2
58 1.1 martin #define GPT_ATTR_BOOTONCE 4
59 1.1 martin #define GPT_ATTR_BOOTFAILED 8
60 1.1 martin #define GPT_ATTR_NOBLOCKIO 16
61 1.1 martin #define GPT_ATTR_REQUIRED 32
62 1.1 martin
63 1.1 martin /* when we don't care for BIOS or UEFI boot, use the combined boot flags */
64 1.1 martin #define GPT_ATTR_BOOT (GPT_ATTR_BIOSBOOT|GPT_ATTR_BOOTME)
65 1.1 martin
66 1.1 martin struct gpt_attr_desc {
67 1.1 martin const char *name;
68 1.1 martin uint flag;
69 1.1 martin };
70 1.1 martin static const struct gpt_attr_desc gpt_avail_attrs[] = {
71 1.1 martin { "biosboot", GPT_ATTR_BIOSBOOT },
72 1.1 martin { "bootme", GPT_ATTR_BOOTME },
73 1.1 martin { "bootonce", GPT_ATTR_BOOTONCE },
74 1.1 martin { "bootfailed", GPT_ATTR_BOOTFAILED },
75 1.1 martin { "noblockio", GPT_ATTR_NOBLOCKIO },
76 1.1 martin { "required", GPT_ATTR_REQUIRED },
77 1.1 martin { NULL, 0 }
78 1.1 martin };
79 1.1 martin
80 1.1 martin struct gpt_ptype_desc {
81 1.1 martin struct part_type_desc gent;
82 1.1 martin char tid[GUID_STR_LEN];
83 1.1 martin uint fsflags, default_fs_type;
84 1.1 martin };
85 1.1 martin
86 1.1 martin static const
87 1.1 martin struct {
88 1.1 martin const char *name;
89 1.1 martin uint fstype;
90 1.1 martin enum part_type ptype;
91 1.1 martin uint fsflags;
92 1.1 martin } gpt_fs_types[] = {
93 1.1 martin { .name = "ffs", .fstype = FS_BSDFFS, .ptype = PT_root,
94 1.1 martin .fsflags = GLM_LIKELY_FFS },
95 1.1 martin { .name = "swap", .fstype = FS_SWAP, .ptype = PT_swap },
96 1.1 martin { .name = "windows", .fstype = FS_MSDOS, .ptype = PT_FAT,
97 1.1 martin .fsflags = GLM_MAYBE_FAT32|GLM_MAYBE_NTFS },
98 1.1 martin { .name = "windows", .fstype = FS_NTFS, .ptype = PT_FAT,
99 1.1 martin .fsflags = GLM_MAYBE_FAT32|GLM_MAYBE_NTFS },
100 1.1 martin { .name = "efi", .fstype = FS_MSDOS, .ptype = PT_EFI_SYSTEM,
101 1.1 martin .fsflags = GLM_MAYBE_FAT32 },
102 1.1 martin { .name = "bios", .fstype = FS_MSDOS, .ptype = PT_FAT,
103 1.1 martin .fsflags = GLM_MAYBE_FAT32 },
104 1.1 martin { .name = "lfs", .fstype = FS_BSDLFS, .ptype = PT_root },
105 1.1 martin { .name = "linux-data", .fstype = FS_EX2FS, .ptype = PT_root },
106 1.1 martin { .name = "apple", .fstype = FS_HFS, .ptype = PT_unknown },
107 1.1 martin { .name = "ccd", .fstype = FS_CCD, .ptype = PT_unknown },
108 1.1 martin { .name = "cgd", .fstype = FS_CGD, .ptype = PT_unknown },
109 1.1 martin { .name = "raid", .fstype = FS_RAID, .ptype = PT_root },
110 1.1 martin { .name = "vmcore", .fstype = FS_VMKCORE, .ptype = PT_unknown },
111 1.1 martin { .name = "vmfs", .fstype = FS_VMFS, .ptype = PT_unknown },
112 1.1 martin { .name = "vmresered", .fstype = FS_VMWRESV, .ptype = PT_unknown }
113 1.1 martin };
114 1.1 martin
115 1.1 martin static size_t gpt_ptype_cnt;
116 1.1 martin static struct gpt_ptype_desc gpt_ptype_descs[GPT_PTYPE_MAX];
117 1.1 martin
118 1.1 martin /* similar to struct gpt_ent, but matching our needs */
119 1.1 martin struct gpt_part_entry {
120 1.1 martin const struct gpt_ptype_desc *gp_type;
121 1.1 martin char gp_id[GUID_STR_LEN]; /* partition guid as string */
122 1.1 martin daddr_t gp_start, gp_size;
123 1.1 martin uint gp_attr; /* various attribute bits */
124 1.1 martin char gp_label[GPT_LABEL_LEN]; /* user defined label */
125 1.1 martin char gp_dev_name[GPT_DEV_LEN]; /* name of wedge */
126 1.1 martin const char *last_mounted; /* last mounted if known */
127 1.1 martin uint fs_type, fs_sub_type; /* FS_* and maybe sub type */
128 1.1 martin uint gp_flags;
129 1.1 martin #define GPEF_ON_DISK 1 /* This entry exists on-disk */
130 1.1 martin #define GPEF_MODIFIED 2 /* this entry has been changed */
131 1.1 martin #define GPEF_WEDGE 4 /* wedge for this exists */
132 1.1 martin #define GPEF_RESIZED 8 /* size has changed */
133 1.1 martin struct gpt_part_entry *gp_next;
134 1.1 martin };
135 1.1 martin
136 1.1 martin static const struct gpt_ptype_desc *gpt_find_native_type(
137 1.1 martin const struct part_type_desc *gent);
138 1.1 martin static const struct gpt_ptype_desc *gpt_find_guid_type(const char*);
139 1.1 martin static bool
140 1.1 martin gpt_info_to_part(struct gpt_part_entry *p, const struct disk_part_info *info,
141 1.1 martin const char **err_msg);
142 1.1 martin
143 1.1 martin const struct disk_partitioning_scheme gpt_parts;
144 1.1 martin struct gpt_disk_partitions {
145 1.1 martin struct disk_partitions dp;
146 1.1 martin /*
147 1.1 martin * We keep a list of our current valid partitions, pointed
148 1.1 martin * to by "partitions".
149 1.1 martin * dp.num_part is the number of entries in "partitions".
150 1.1 martin * When partitions that have a representation on disk already
151 1.1 martin * are deleted, we move them to the "obsolete" list so we
152 1.1 martin * can issue the proper commands to remove it when writing back.
153 1.1 martin */
154 1.1 martin struct gpt_part_entry *partitions, /* current partitions */
155 1.1 martin *obsolete; /* deleted partitions */
156 1.1 martin size_t max_num_parts; /* how many entries max? */
157 1.1 martin size_t prologue, epilogue; /* number of sectors res. */
158 1.1 martin bool has_gpt; /* disk already has a GPT */
159 1.1 martin };
160 1.1 martin
161 1.1 martin /*
162 1.1 martin * Init global variables from MD details
163 1.1 martin */
164 1.1 martin static void
165 1.1 martin gpt_md_init(bool is_boot_disk, size_t *max_parts, size_t *head, size_t *tail)
166 1.1 martin {
167 1.1 martin size_t num;
168 1.1 martin
169 1.1 martin if (is_boot_disk) {
170 1.1 martin #ifdef MD_GPT_INITIAL_SIZE
171 1.1 martin #if MD_GPT_INITIAL_SIZE < 2*512
172 1.1 martin #error impossible small GPT prologue
173 1.1 martin #endif
174 1.1 martin num = ((MD_GPT_INITIAL_SIZE-(2*512))/512)*GPT_PARTS_PER_SEC;
175 1.1 martin #else
176 1.1 martin num = GPT_DEFAULT_MAX_PARTS;
177 1.1 martin #endif
178 1.1 martin } else {
179 1.1 martin num = GPT_DEFAULT_MAX_PARTS;
180 1.1 martin }
181 1.1 martin *max_parts = num;
182 1.1 martin *head = 2 + num/GPT_PARTS_PER_SEC;
183 1.1 martin *tail = 1 + num/GPT_PARTS_PER_SEC;
184 1.1 martin }
185 1.1 martin
186 1.1 martin /*
187 1.1 martin * Parse a part of "gpt show" output into a struct gpt_part_entry.
188 1.1 martin * Output is from "show -a" format if details = false, otherwise
189 1.1 martin * from details for a specific partition (show -i or show -b)
190 1.1 martin */
191 1.1 martin static void
192 1.1 martin gpt_add_info(struct gpt_part_entry *part, const char *tag, char *val,
193 1.1 martin bool details)
194 1.1 martin {
195 1.1 martin char *s, *e;
196 1.1 martin
197 1.1 martin if (details && strcmp(tag, "Start:") == 0) {
198 1.1 martin part->gp_start = strtouq(val, NULL, 10);
199 1.1 martin } else if (details && strcmp(tag, "Size:") == 0) {
200 1.1 martin part->gp_size = strtouq(val, NULL, 10);
201 1.1 martin } else if (details && strcmp(tag, "Type:") == 0) {
202 1.1 martin s = strchr(val, '(');
203 1.1 martin if (!s)
204 1.1 martin return;
205 1.1 martin e = strchr(s, ')');
206 1.1 martin if (!e)
207 1.1 martin return;
208 1.1 martin *e = 0;
209 1.1 martin part->gp_type = gpt_find_guid_type(s+1);
210 1.1 martin } else if (strcmp(tag, "TypeID:") == 0) {
211 1.1 martin part->gp_type = gpt_find_guid_type(val);
212 1.1 martin } else if (strcmp(tag, "GUID:") == 0) {
213 1.1 martin strlcpy(part->gp_id, val, sizeof(part->gp_id));
214 1.1 martin } else if (strcmp(tag, "Label:") == 0) {
215 1.1 martin if (strlen(val) > 0)
216 1.1 martin strlcpy(part->gp_label, val, sizeof(part->gp_label));
217 1.1 martin } else if (strcmp(tag, "Attributes:") == 0) {
218 1.1 martin char *n;
219 1.1 martin
220 1.1 martin while ((n = strsep(&val, ", ")) != NULL) {
221 1.1 martin if (*n == 0)
222 1.1 martin continue;
223 1.1 martin for (const struct gpt_attr_desc *p = gpt_avail_attrs;
224 1.1 martin p->name != NULL; p++) {
225 1.1 martin if (strcmp(p->name, n) == 0)
226 1.1 martin part->gp_attr |= p->flag;
227 1.1 martin }
228 1.1 martin }
229 1.1 martin }
230 1.1 martin }
231 1.1 martin
232 1.1 martin static struct disk_partitions *
233 1.1 martin gpt_read_from_disk(const char *dev, daddr_t start, daddr_t len)
234 1.1 martin {
235 1.1 martin char diskpath[MAXPATHLEN];
236 1.1 martin int fd;
237 1.1 martin
238 1.1 martin assert(start == 0);
239 1.1 martin assert(have_gpt);
240 1.1 martin
241 1.1 martin if (run_program(RUN_SILENT | RUN_ERROR_OK,
242 1.1 martin "gpt -rq header %s", dev) != 0)
243 1.1 martin return NULL;
244 1.1 martin
245 1.1 martin /* read the partitions */
246 1.1 martin int i;
247 1.1 martin unsigned int p_index;
248 1.1 martin daddr_t p_start = 0, p_size = 0, avail_start = 0, avail_size = 0,
249 1.1 martin disk_size = 0;
250 1.1 martin char *textbuf, *t, *tt, p_type[STRSIZE];
251 1.1 martin static const char regpart_prefix[] = "GPT part - ";
252 1.1 martin struct gpt_disk_partitions *parts;
253 1.1 martin struct gpt_part_entry *last = NULL, *add_to = NULL;
254 1.1 martin
255 1.1 martin if (collect(T_OUTPUT, &textbuf, "gpt -r show -a %s 2>/dev/null", dev)
256 1.1 martin < 1)
257 1.1 martin return NULL;
258 1.1 martin
259 1.1 martin /* parse output and create our list */
260 1.1 martin parts = calloc(1, sizeof(*parts));
261 1.1 martin if (parts == NULL)
262 1.1 martin return NULL;
263 1.1 martin
264 1.1 martin (void)strtok(textbuf, "\n"); /* ignore first line */
265 1.1 martin while ((t = strtok(NULL, "\n")) != NULL) {
266 1.1 martin i = 0; p_start = 0; p_size = 0; p_index = 0;
267 1.1 martin p_type[0] = 0;
268 1.1 martin while ((tt = strsep(&t, " \t")) != NULL) {
269 1.1 martin if (strlen(tt) == 0)
270 1.1 martin continue;
271 1.1 martin if (i == 0) {
272 1.1 martin if (add_to != NULL)
273 1.1 martin gpt_add_info(add_to, tt, t, false);
274 1.1 martin p_start = strtouq(tt, NULL, 10);
275 1.1 martin if (p_start == 0 && add_to != NULL)
276 1.1 martin break;
277 1.1 martin else
278 1.1 martin add_to = NULL;
279 1.1 martin }
280 1.1 martin if (i == 1)
281 1.1 martin p_size = strtouq(tt, NULL, 10);
282 1.1 martin if (i == 2)
283 1.1 martin p_index = strtouq(tt, NULL, 10);
284 1.1 martin if (i > 2 || (i == 2 && p_index == 0)) {
285 1.1 martin if (p_type[0])
286 1.1 martin strlcat(p_type, " ", STRSIZE);
287 1.1 martin strlcat(p_type, tt, STRSIZE);
288 1.1 martin }
289 1.1 martin i++;
290 1.1 martin }
291 1.1 martin
292 1.1 martin if (p_start == 0 || p_size == 0)
293 1.1 martin continue;
294 1.1 martin else if (strcmp(p_type, "Pri GPT table") == 0) {
295 1.1 martin avail_start = p_start + p_size;
296 1.1 martin parts->prologue = avail_start;
297 1.1 martin parts->epilogue = p_size + 1;
298 1.1 martin parts->max_num_parts = p_size * GPT_PARTS_PER_SEC;
299 1.1 martin } else if (strcmp(p_type, "Sec GPT table") == 0)
300 1.1 martin avail_size = p_start - avail_start;
301 1.1 martin else if(strcmp(p_type, "Sec GPT header") == 0)
302 1.1 martin disk_size = p_start + p_size;
303 1.1 martin else if (p_index == 0 && strlen(p_type) > 0)
304 1.1 martin /* Utilitary entry (PMBR, etc) */
305 1.1 martin continue;
306 1.1 martin else if (p_index == 0) {
307 1.1 martin /* Free space */
308 1.1 martin continue;
309 1.1 martin } else {
310 1.1 martin /* Usual partition */
311 1.1 martin tt = p_type;
312 1.1 martin if (strncmp(tt, regpart_prefix,
313 1.1 martin strlen(regpart_prefix)) == 0)
314 1.1 martin tt += strlen(regpart_prefix);
315 1.1 martin
316 1.1 martin /* Add to our linked list */
317 1.1 martin struct gpt_part_entry *np = calloc(1, sizeof(*np));
318 1.1 martin if (np == NULL)
319 1.1 martin break;
320 1.1 martin
321 1.1 martin strlcpy(np->gp_label, tt, sizeof(np->gp_label));
322 1.1 martin np->gp_start = p_start;
323 1.1 martin np->gp_size = p_size;
324 1.1 martin np->gp_flags |= GPEF_ON_DISK;
325 1.1 martin
326 1.1 martin if (last == NULL)
327 1.1 martin parts->partitions = np;
328 1.1 martin else
329 1.1 martin last->gp_next = np;
330 1.1 martin last = np;
331 1.1 martin add_to = np;
332 1.1 martin parts->dp.num_part++;
333 1.1 martin }
334 1.1 martin }
335 1.1 martin free(textbuf);
336 1.1 martin
337 1.2 martin /* If the GPT was not complete (e.g. truncated image), barf */
338 1.2 martin if (disk_size <= 0) {
339 1.2 martin free(parts);
340 1.2 martin return NULL;
341 1.2 martin }
342 1.2 martin
343 1.1 martin parts->dp.pscheme = &gpt_parts;
344 1.1 martin parts->dp.disk = dev;
345 1.1 martin parts->dp.disk_start = start;
346 1.1 martin parts->dp.disk_size = disk_size;
347 1.1 martin parts->dp.free_space = avail_size;
348 1.1 martin parts->has_gpt = true;
349 1.1 martin
350 1.1 martin fd = opendisk(parts->dp.disk, O_RDONLY, diskpath, sizeof(diskpath), 0);
351 1.1 martin for (struct gpt_part_entry *p = parts->partitions; p != NULL;
352 1.1 martin p = p->gp_next) {
353 1.1 martin #ifdef DEFAULT_UFS2
354 1.1 martin bool fs_is_default = false;
355 1.1 martin #endif
356 1.1 martin
357 1.1 martin if (p->gp_type->fsflags != 0) {
358 1.1 martin const char *lm = get_last_mounted(fd, p->gp_start,
359 1.1 martin &p->fs_type, &p->fs_sub_type, p->gp_type->fsflags);
360 1.1 martin if (lm != NULL && *lm != 0) {
361 1.1 martin p->last_mounted = strdup(lm);
362 1.1 martin } else {
363 1.1 martin p->fs_type = p->gp_type->default_fs_type;
364 1.1 martin #ifdef DEFAULT_UFS2
365 1.1 martin fs_is_default = true;
366 1.1 martin #endif
367 1.1 martin }
368 1.1 martin } else {
369 1.1 martin p->fs_type = p->gp_type->default_fs_type;
370 1.1 martin #ifdef DEFAULT_UFS2
371 1.1 martin fs_is_default = true;
372 1.1 martin #endif
373 1.1 martin }
374 1.1 martin #ifdef DEFAULT_UFS2
375 1.1 martin if (fs_is_default && p->fs_type == FS_BSDFFS)
376 1.1 martin p->fs_sub_type = 2;
377 1.1 martin #endif
378 1.1 martin
379 1.1 martin parts->dp.free_space -= p->gp_size;
380 1.1 martin }
381 1.1 martin close(fd);
382 1.1 martin
383 1.1 martin return &parts->dp;
384 1.1 martin }
385 1.1 martin
386 1.1 martin static struct disk_partitions *
387 1.1 martin gpt_create_new(const char *disk, daddr_t start, daddr_t len, daddr_t total,
388 1.1 martin bool is_boot_drive)
389 1.1 martin {
390 1.1 martin struct gpt_disk_partitions *parts;
391 1.1 martin
392 1.1 martin if (start != 0) {
393 1.1 martin assert(0);
394 1.1 martin return NULL;
395 1.1 martin }
396 1.1 martin
397 1.1 martin parts = calloc(sizeof(*parts), 1);
398 1.1 martin if (!parts)
399 1.1 martin return NULL;
400 1.1 martin
401 1.1 martin parts->dp.pscheme = &gpt_parts;
402 1.1 martin parts->dp.disk = disk;
403 1.1 martin
404 1.1 martin gpt_md_init(is_boot_drive, &parts->max_num_parts, &parts->prologue,
405 1.1 martin &parts->epilogue);
406 1.1 martin
407 1.1 martin parts->dp.disk_start = start;
408 1.1 martin parts->dp.disk_size = len;
409 1.1 martin parts->dp.free_space = len - start - parts->prologue - parts->epilogue;
410 1.1 martin parts->has_gpt = false;
411 1.1 martin
412 1.1 martin return &parts->dp;
413 1.1 martin }
414 1.1 martin
415 1.1 martin static bool
416 1.1 martin gpt_get_part_info(const struct disk_partitions *arg, part_id id,
417 1.1 martin struct disk_part_info *info)
418 1.1 martin {
419 1.1 martin const struct gpt_disk_partitions *parts =
420 1.1 martin (const struct gpt_disk_partitions*)arg;
421 1.1 martin const struct gpt_part_entry *p = parts->partitions;
422 1.1 martin part_id no;
423 1.1 martin
424 1.1 martin for (no = 0; p != NULL && no < id; no++)
425 1.1 martin p = p->gp_next;
426 1.1 martin
427 1.1 martin if (no != id || p == NULL)
428 1.1 martin return false;
429 1.1 martin
430 1.1 martin memset(info, 0, sizeof(*info));
431 1.1 martin info->start = p->gp_start;
432 1.1 martin info->size = p->gp_size;
433 1.1 martin if (p->gp_type)
434 1.1 martin info->nat_type = &p->gp_type->gent;
435 1.1 martin info->last_mounted = p->last_mounted;
436 1.1 martin info->fs_type = p->fs_type;
437 1.1 martin info->fs_sub_type = p->fs_sub_type;
438 1.1 martin
439 1.1 martin return true;
440 1.1 martin }
441 1.1 martin
442 1.1 martin static bool
443 1.1 martin gpt_get_part_attr_str(const struct disk_partitions *arg, part_id id,
444 1.1 martin char *str, size_t avail_space)
445 1.1 martin {
446 1.1 martin const struct gpt_disk_partitions *parts =
447 1.1 martin (const struct gpt_disk_partitions*)arg;
448 1.1 martin const struct gpt_part_entry *p = parts->partitions;
449 1.1 martin part_id no;
450 1.1 martin static const char *flags = NULL;
451 1.1 martin
452 1.1 martin for (no = 0; p != NULL && no < id; no++)
453 1.1 martin p = p->gp_next;
454 1.1 martin
455 1.1 martin if (no != id || p == NULL)
456 1.1 martin return false;
457 1.1 martin
458 1.1 martin if (flags == NULL)
459 1.1 martin flags = msg_string(MSG_gpt_flags);
460 1.1 martin
461 1.1 martin if (avail_space < 2)
462 1.1 martin return false;
463 1.1 martin
464 1.1 martin if (p->gp_attr & GPT_ATTR_BOOT)
465 1.1 martin *str++ = flags[0];
466 1.1 martin *str = 0;
467 1.1 martin
468 1.1 martin return true;
469 1.1 martin }
470 1.1 martin
471 1.1 martin /*
472 1.1 martin * Find insert position and check for duplicates.
473 1.1 martin * If all goes well, insert the new "entry" in the "list".
474 1.1 martin * If there are collisions, report "no free space".
475 1.1 martin * We keep all lists sorted by start sector number,
476 1.1 martin */
477 1.1 martin static bool
478 1.1 martin gpt_insert_part_into_list(struct gpt_disk_partitions *parts,
479 1.1 martin struct gpt_part_entry **list,
480 1.1 martin struct gpt_part_entry *entry, const char **err_msg)
481 1.1 martin {
482 1.1 martin struct gpt_part_entry *p, *last;
483 1.1 martin
484 1.1 martin /* find the first entry past the new one (if any) */
485 1.1 martin for (last = NULL, p = *list; p != NULL; last = p, p = p->gp_next) {
486 1.1 martin if (p->gp_start > entry->gp_start)
487 1.1 martin break;
488 1.1 martin }
489 1.1 martin
490 1.1 martin /* check if last partition overlaps with new one */
491 1.1 martin if (last) {
492 1.1 martin if (last->gp_start + last->gp_size > entry->gp_start) {
493 1.1 martin if (err_msg)
494 1.1 martin *err_msg = msg_string(MSG_No_free_space);
495 1.1 martin return false;
496 1.1 martin }
497 1.1 martin }
498 1.1 martin
499 1.1 martin if (p == NULL) {
500 1.1 martin entry->gp_next = NULL;
501 1.1 martin if (last != NULL) {
502 1.1 martin last->gp_next = entry;
503 1.1 martin }
504 1.1 martin } else {
505 1.1 martin /* check if new entry overlaps with next */
506 1.1 martin if (entry->gp_start + entry->gp_size > p->gp_start) {
507 1.1 martin if (err_msg)
508 1.1 martin *err_msg = msg_string(MSG_No_free_space);
509 1.1 martin return false;
510 1.1 martin }
511 1.1 martin
512 1.1 martin entry->gp_next = p;
513 1.1 martin if (last != NULL)
514 1.1 martin last->gp_next = entry;
515 1.1 martin else
516 1.1 martin *list = entry;
517 1.1 martin }
518 1.1 martin if (*list == NULL)
519 1.1 martin *list = entry;
520 1.1 martin
521 1.1 martin return true;
522 1.1 martin }
523 1.1 martin
524 1.1 martin static bool
525 1.1 martin gpt_set_part_info(struct disk_partitions *arg, part_id id,
526 1.1 martin const struct disk_part_info *info, const char **err_msg)
527 1.1 martin {
528 1.1 martin struct gpt_disk_partitions *parts =
529 1.1 martin (struct gpt_disk_partitions*)arg;
530 1.1 martin struct gpt_part_entry *p = parts->partitions, *n;
531 1.1 martin part_id no;
532 1.1 martin daddr_t lendiff;
533 1.1 martin
534 1.1 martin for (no = 0; p != NULL && no < id; no++)
535 1.1 martin p = p->gp_next;
536 1.1 martin
537 1.1 martin if (no != id || p == NULL)
538 1.1 martin return false;
539 1.1 martin
540 1.1 martin if ((p->gp_flags & GPEF_ON_DISK)) {
541 1.1 martin if (info->start != p->gp_start) {
542 1.1 martin /* partition moved, we need to delete and re-add */
543 1.1 martin n = calloc(1, sizeof(*n));
544 1.1 martin if (n == NULL) {
545 1.1 martin if (err_msg)
546 1.1 martin *err_msg = err_outofmem;
547 1.1 martin return false;
548 1.1 martin }
549 1.1 martin *n = *p;
550 1.1 martin p->gp_flags &= ~GPEF_ON_DISK;
551 1.1 martin if (!gpt_insert_part_into_list(parts, &parts->obsolete,
552 1.1 martin n, err_msg))
553 1.1 martin return false;
554 1.1 martin } else if (info->size != p->gp_size) {
555 1.1 martin p->gp_flags |= GPEF_RESIZED;
556 1.1 martin }
557 1.1 martin }
558 1.1 martin
559 1.1 martin p->gp_flags |= GPEF_MODIFIED;
560 1.1 martin
561 1.1 martin lendiff = info->size - p->gp_size;
562 1.1 martin parts->dp.free_space -= lendiff;
563 1.1 martin return gpt_info_to_part(p, info, err_msg);
564 1.1 martin }
565 1.1 martin
566 1.1 martin static size_t
567 1.1 martin gpt_get_free_spaces_internal(const struct gpt_disk_partitions *parts,
568 1.1 martin struct disk_part_free_space *result, size_t max_num_result,
569 1.1 martin daddr_t min_space_size, daddr_t align, daddr_t start, daddr_t ignore)
570 1.1 martin {
571 1.1 martin size_t cnt = 0;
572 1.1 martin daddr_t s, e, from, size, end_of_disk;
573 1.1 martin struct gpt_part_entry *p;
574 1.1 martin
575 1.1 martin if (align > 1)
576 1.1 martin start = max(roundup(start, align), align);
577 1.1 martin if (start < 0 || start < (daddr_t)parts->prologue)
578 1.1 martin start = parts->prologue;
579 1.1 martin if (parts->dp.disk_start != 0 && parts->dp.disk_start > start)
580 1.1 martin start = parts->dp.disk_start;
581 1.1 martin if (min_space_size < 1)
582 1.1 martin min_space_size = 1;
583 1.1 martin end_of_disk = parts->dp.disk_start + parts->dp.disk_size
584 1.1 martin - parts->epilogue;
585 1.1 martin from = start;
586 1.1 martin while (from < end_of_disk && cnt < max_num_result) {
587 1.1 martin again:
588 1.1 martin size = parts->dp.disk_start + parts->dp.disk_size - from;
589 1.1 martin start = from;
590 1.1 martin if (start + size > end_of_disk)
591 1.1 martin size = end_of_disk - start;
592 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
593 1.1 martin s = p->gp_start;
594 1.1 martin e = p->gp_size + s;
595 1.1 martin if (s == ignore)
596 1.1 martin continue;
597 1.1 martin if (e < from)
598 1.1 martin continue;
599 1.1 martin if (s <= from && e > from) {
600 1.1 martin if (e - 1 >= end_of_disk)
601 1.1 martin return cnt;
602 1.1 martin from = e + 1;
603 1.1 martin if (align > 1) {
604 1.1 martin from = max(roundup(from, align), align);
605 1.1 martin if (from >= end_of_disk) {
606 1.1 martin size = 0;
607 1.1 martin break;
608 1.1 martin }
609 1.1 martin }
610 1.1 martin goto again;
611 1.1 martin }
612 1.1 martin if (s > from && s - from < size) {
613 1.1 martin size = s - from;
614 1.1 martin }
615 1.1 martin }
616 1.1 martin if (size >= min_space_size) {
617 1.1 martin result->start = start;
618 1.1 martin result->size = size;
619 1.1 martin result++;
620 1.1 martin cnt++;
621 1.1 martin }
622 1.1 martin from += size + 1;
623 1.1 martin if (align > 1)
624 1.1 martin from = max(roundup(from, align), align);
625 1.1 martin }
626 1.1 martin
627 1.1 martin return cnt;
628 1.1 martin }
629 1.1 martin
630 1.1 martin static daddr_t
631 1.1 martin gpt_max_free_space_at(const struct disk_partitions *arg, daddr_t start)
632 1.1 martin {
633 1.1 martin const struct gpt_disk_partitions *parts =
634 1.1 martin (const struct gpt_disk_partitions*)arg;
635 1.1 martin struct disk_part_free_space space;
636 1.1 martin
637 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, 1, 0,
638 1.1 martin start, start) == 1)
639 1.1 martin return space.size;
640 1.1 martin
641 1.1 martin return 0;
642 1.1 martin }
643 1.1 martin
644 1.1 martin static size_t
645 1.1 martin gpt_get_free_spaces(const struct disk_partitions *arg,
646 1.1 martin struct disk_part_free_space *result, size_t max_num_result,
647 1.1 martin daddr_t min_space_size, daddr_t align, daddr_t start,
648 1.1 martin daddr_t ignore)
649 1.1 martin {
650 1.1 martin const struct gpt_disk_partitions *parts =
651 1.1 martin (const struct gpt_disk_partitions*)arg;
652 1.1 martin
653 1.1 martin return gpt_get_free_spaces_internal(parts, result,
654 1.1 martin max_num_result, min_space_size, align, start, ignore);
655 1.1 martin }
656 1.1 martin
657 1.1 martin
658 1.1 martin static bool
659 1.1 martin gpt_adapt(const struct disk_partitions *arg,
660 1.1 martin const struct disk_part_info *src, struct disk_part_info *dest)
661 1.1 martin {
662 1.1 martin /* slightly simplistic, enhance when needed */
663 1.1 martin memcpy(dest, src, sizeof(*dest));
664 1.1 martin
665 1.1 martin if (src->nat_type == NULL)
666 1.1 martin return false;
667 1.1 martin
668 1.1 martin dest->nat_type = arg->pscheme->get_generic_part_type(
669 1.1 martin src->nat_type->generic_ptype);
670 1.1 martin if (dest->nat_type == NULL)
671 1.1 martin dest->nat_type = arg->pscheme->get_generic_part_type(
672 1.1 martin PT_unknown);
673 1.1 martin
674 1.1 martin return true;
675 1.1 martin }
676 1.1 martin
677 1.1 martin static void
678 1.1 martin gpt_match_ptype(const char *name, struct gpt_ptype_desc *t)
679 1.1 martin {
680 1.1 martin size_t i;
681 1.1 martin
682 1.1 martin for (i = 0; i < __arraycount(gpt_fs_types); i++) {
683 1.1 martin if (strcmp(name, gpt_fs_types[i].name) == 0) {
684 1.1 martin t->gent.generic_ptype = gpt_fs_types[i].ptype;
685 1.1 martin t->fsflags = gpt_fs_types[i].fsflags;
686 1.1 martin t->default_fs_type = gpt_fs_types[i].fstype;
687 1.1 martin return;
688 1.1 martin }
689 1.1 martin }
690 1.1 martin
691 1.1 martin t->gent.generic_ptype = PT_unknown;
692 1.1 martin t->fsflags = 0;
693 1.1 martin t->default_fs_type = FS_BSDFFS;
694 1.1 martin }
695 1.1 martin
696 1.1 martin static void
697 1.1 martin gpt_internal_add_ptype(const char *uid, const char *name, const char *desc)
698 1.1 martin {
699 1.1 martin strlcpy(gpt_ptype_descs[gpt_ptype_cnt].tid, uid,
700 1.1 martin sizeof(gpt_ptype_descs[gpt_ptype_cnt].tid));
701 1.1 martin gpt_ptype_descs[gpt_ptype_cnt].gent.short_desc = name;
702 1.1 martin gpt_ptype_descs[gpt_ptype_cnt].gent.description = desc;
703 1.1 martin gpt_match_ptype(name, &gpt_ptype_descs[gpt_ptype_cnt]);
704 1.1 martin gpt_ptype_cnt++;
705 1.1 martin }
706 1.1 martin
707 1.1 martin static void
708 1.1 martin gpt_init_ptypes(void)
709 1.1 martin {
710 1.1 martin if (gpt_ptype_cnt == 0)
711 1.1 martin gpt_uuid_query(gpt_internal_add_ptype);
712 1.1 martin }
713 1.1 martin
714 1.1 martin static size_t
715 1.1 martin gpt_type_count(void)
716 1.1 martin {
717 1.1 martin if (gpt_ptype_cnt == 0)
718 1.1 martin gpt_init_ptypes();
719 1.1 martin
720 1.1 martin return gpt_ptype_cnt;
721 1.1 martin }
722 1.1 martin
723 1.1 martin static const struct part_type_desc *
724 1.1 martin gpt_get_ptype(size_t ndx)
725 1.1 martin {
726 1.1 martin if (gpt_ptype_cnt == 0)
727 1.1 martin gpt_init_ptypes();
728 1.1 martin
729 1.1 martin if (ndx >= gpt_ptype_cnt)
730 1.1 martin return NULL;
731 1.1 martin
732 1.1 martin return &gpt_ptype_descs[ndx].gent;
733 1.1 martin }
734 1.1 martin
735 1.1 martin static const struct part_type_desc *
736 1.1 martin gpt_get_generic_type(enum part_type gent)
737 1.1 martin {
738 1.1 martin if (gpt_ptype_cnt == 0)
739 1.1 martin gpt_init_ptypes();
740 1.1 martin
741 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
742 1.1 martin if (gpt_ptype_descs[i].gent.generic_ptype == gent)
743 1.1 martin return &gpt_ptype_descs[i].gent;
744 1.1 martin
745 1.1 martin return NULL;
746 1.1 martin }
747 1.1 martin
748 1.1 martin static const struct gpt_ptype_desc *
749 1.1 martin gpt_find_native_type(const struct part_type_desc *gent)
750 1.1 martin {
751 1.1 martin if (gpt_ptype_cnt == 0)
752 1.1 martin gpt_init_ptypes();
753 1.1 martin
754 1.1 martin if (gent == NULL)
755 1.1 martin return NULL;
756 1.1 martin
757 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
758 1.1 martin if (gent == &gpt_ptype_descs[i].gent)
759 1.1 martin return &gpt_ptype_descs[i];
760 1.1 martin
761 1.1 martin gent = gpt_get_generic_type(gent->generic_ptype);
762 1.1 martin if (gent == NULL)
763 1.1 martin return NULL;
764 1.1 martin
765 1.1 martin /* this can not recurse deeper than once, we would not have found a
766 1.1 martin * generic type a few lines above if it would. */
767 1.1 martin return gpt_find_native_type(gent);
768 1.1 martin }
769 1.1 martin
770 1.1 martin static const struct gpt_ptype_desc *
771 1.1 martin gpt_find_guid_type(const char *uid)
772 1.1 martin {
773 1.1 martin if (gpt_ptype_cnt == 0)
774 1.1 martin gpt_init_ptypes();
775 1.1 martin
776 1.1 martin if (uid == NULL || uid[0] == 0)
777 1.1 martin return NULL;
778 1.1 martin
779 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
780 1.1 martin if (strcmp(gpt_ptype_descs[i].tid, uid) == 0)
781 1.1 martin return &gpt_ptype_descs[i];
782 1.1 martin
783 1.1 martin return NULL;
784 1.1 martin }
785 1.1 martin
786 1.1 martin static const struct part_type_desc *
787 1.1 martin gpt_find_type(const char *desc)
788 1.1 martin {
789 1.1 martin if (gpt_ptype_cnt == 0)
790 1.1 martin gpt_init_ptypes();
791 1.1 martin
792 1.1 martin if (desc == NULL || desc[0] == 0)
793 1.1 martin return NULL;
794 1.1 martin
795 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
796 1.1 martin if (strcmp(gpt_ptype_descs[i].gent.short_desc, desc) == 0)
797 1.1 martin return &gpt_ptype_descs[i].gent;
798 1.1 martin
799 1.1 martin return NULL;
800 1.1 martin }
801 1.1 martin
802 1.1 martin static const struct part_type_desc *
803 1.1 martin gpt_get_fs_part_type(unsigned fstype, unsigned fs_sub_type)
804 1.1 martin {
805 1.1 martin size_t i;
806 1.1 martin
807 1.1 martin for (i = 0; i < __arraycount(gpt_fs_types); i++)
808 1.1 martin if (fstype == gpt_fs_types[i].fstype)
809 1.1 martin return gpt_find_type(gpt_fs_types[i].name);
810 1.1 martin
811 1.1 martin return gpt_get_generic_type(PT_root);
812 1.1 martin }
813 1.1 martin
814 1.1 martin static daddr_t
815 1.1 martin gpt_get_part_alignment(const struct disk_partitions *parts)
816 1.1 martin {
817 1.1 martin
818 1.1 martin assert(parts->disk_size > 0);
819 1.1 martin if (parts->disk_size < 0)
820 1.1 martin return 1;
821 1.1 martin
822 1.1 martin /* Use 1MB offset/alignemnt for large (>128GB) disks */
823 1.1 martin if (parts->disk_size > HUGE_DISK_SIZE)
824 1.1 martin return 2048;
825 1.1 martin else if (parts->disk_size > TINY_DISK_SIZE)
826 1.1 martin return 64;
827 1.1 martin else
828 1.1 martin return 4;
829 1.1 martin }
830 1.1 martin
831 1.1 martin static bool
832 1.1 martin gpt_can_add_partition(const struct disk_partitions *arg)
833 1.1 martin {
834 1.1 martin const struct gpt_disk_partitions *parts =
835 1.1 martin (const struct gpt_disk_partitions*)arg;
836 1.1 martin struct disk_part_free_space space;
837 1.1 martin daddr_t align;
838 1.1 martin
839 1.1 martin if (parts->dp.num_part >= parts->max_num_parts)
840 1.1 martin return false;
841 1.1 martin
842 1.1 martin align = gpt_get_part_alignment(arg);
843 1.1 martin if (parts->dp.free_space <= align)
844 1.1 martin return false;
845 1.1 martin
846 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, align, align,
847 1.1 martin 0, -1) < 1)
848 1.1 martin return false;
849 1.1 martin
850 1.1 martin return true;
851 1.1 martin }
852 1.1 martin
853 1.1 martin static bool
854 1.1 martin gpt_info_to_part(struct gpt_part_entry *p, const struct disk_part_info *info,
855 1.1 martin const char **err_msg)
856 1.1 martin {
857 1.1 martin p->gp_type = gpt_find_native_type(info->nat_type);
858 1.1 martin p->gp_start = info->start;
859 1.1 martin p->gp_size = info->size;
860 1.1 martin if (info->last_mounted != NULL && info->last_mounted !=
861 1.1 martin p->last_mounted) {
862 1.1 martin free(__UNCONST(p->last_mounted));
863 1.1 martin p->last_mounted = strdup(info->last_mounted);
864 1.1 martin }
865 1.1 martin p->fs_type = info->fs_type;
866 1.1 martin p->fs_sub_type = info->fs_sub_type;
867 1.1 martin
868 1.1 martin return true;
869 1.1 martin }
870 1.1 martin
871 1.1 martin static part_id
872 1.1 martin gpt_add_part(struct disk_partitions *arg,
873 1.1 martin const struct disk_part_info *info, const char **err_msg)
874 1.1 martin {
875 1.1 martin struct gpt_disk_partitions *parts =
876 1.1 martin (struct gpt_disk_partitions*)arg;
877 1.1 martin struct disk_part_free_space space;
878 1.1 martin struct disk_part_info data = *info;
879 1.1 martin struct gpt_part_entry *p;
880 1.1 martin bool ok;
881 1.1 martin
882 1.1 martin if (err_msg != NULL)
883 1.1 martin *err_msg = NULL;
884 1.1 martin
885 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, 1, 1,
886 1.1 martin info->start, -1) < 1) {
887 1.1 martin if (err_msg)
888 1.1 martin *err_msg = msg_string(MSG_No_free_space);
889 1.1 martin return NO_PART;
890 1.1 martin }
891 1.1 martin if (parts->dp.num_part >= parts->max_num_parts) {
892 1.1 martin if (err_msg)
893 1.1 martin *err_msg = msg_string(MSG_err_too_many_partitions);
894 1.1 martin return NO_PART;
895 1.1 martin }
896 1.1 martin
897 1.1 martin if (data.size > space.size)
898 1.1 martin data.size = space.size;
899 1.1 martin
900 1.1 martin p = calloc(1, sizeof(*p));
901 1.1 martin if (p == NULL) {
902 1.1 martin if (err_msg != NULL)
903 1.1 martin *err_msg = INTERNAL_ERROR;
904 1.1 martin return NO_PART;
905 1.1 martin }
906 1.1 martin if (!gpt_info_to_part(p, &data, err_msg)) {
907 1.1 martin free(p);
908 1.1 martin return NO_PART;
909 1.1 martin }
910 1.1 martin p->gp_flags |= GPEF_MODIFIED;
911 1.1 martin ok = gpt_insert_part_into_list(parts, &parts->partitions, p, err_msg);
912 1.1 martin if (ok) {
913 1.1 martin parts->dp.num_part++;
914 1.1 martin parts->dp.free_space -= p->gp_size;
915 1.1 martin return parts->dp.num_part-1;
916 1.1 martin } else {
917 1.1 martin free(p);
918 1.1 martin return NO_PART;
919 1.1 martin }
920 1.1 martin }
921 1.1 martin
922 1.1 martin static bool
923 1.1 martin gpt_delete_partition(struct disk_partitions *arg, part_id id,
924 1.1 martin const char **err_msg)
925 1.1 martin {
926 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
927 1.1 martin struct gpt_part_entry *p, *last = NULL;
928 1.1 martin part_id i;
929 1.1 martin bool res;
930 1.1 martin
931 1.1 martin if (parts->dp.num_part == 0)
932 1.1 martin return false;
933 1.1 martin
934 1.1 martin for (i = 0, p = parts->partitions;
935 1.1 martin i != id && i < parts->dp.num_part && p != NULL;
936 1.1 martin i++, p = p->gp_next)
937 1.1 martin last = p;
938 1.1 martin
939 1.1 martin if (p == NULL) {
940 1.1 martin if (err_msg)
941 1.1 martin *err_msg = INTERNAL_ERROR;
942 1.1 martin return false;
943 1.1 martin }
944 1.1 martin
945 1.1 martin if (last == NULL)
946 1.1 martin parts->partitions = p->gp_next;
947 1.1 martin else
948 1.1 martin last->gp_next = p->gp_next;
949 1.1 martin
950 1.1 martin res = true;
951 1.1 martin if (p->gp_flags & GPEF_ON_DISK) {
952 1.1 martin if (!gpt_insert_part_into_list(parts, &parts->obsolete,
953 1.1 martin p, err_msg))
954 1.1 martin res = false;
955 1.1 martin } else {
956 1.1 martin free(p);
957 1.1 martin }
958 1.1 martin
959 1.1 martin if (res) {
960 1.1 martin parts->dp.num_part--;
961 1.1 martin parts->dp.free_space += p->gp_size;
962 1.1 martin }
963 1.1 martin
964 1.1 martin return res;
965 1.1 martin }
966 1.1 martin
967 1.1 martin static bool
968 1.1 martin gpt_delete_all_partitions(struct disk_partitions *arg)
969 1.1 martin {
970 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
971 1.1 martin
972 1.1 martin while (parts->dp.num_part > 0) {
973 1.1 martin if (!gpt_delete_partition(&parts->dp, 0, NULL))
974 1.1 martin return false;
975 1.1 martin }
976 1.1 martin
977 1.1 martin return true;
978 1.1 martin }
979 1.1 martin
980 1.1 martin static bool
981 1.1 martin gpt_read_part(const char *disk, daddr_t start, struct gpt_part_entry *p)
982 1.1 martin {
983 1.1 martin char *textbuf, *t, *tt;
984 1.1 martin static const char expected_hdr[] = "Details for index ";
985 1.1 martin
986 1.1 martin /* run gpt show for this partition */
987 1.1 martin if (collect(T_OUTPUT, &textbuf,
988 1.1 martin "gpt -r show -b %" PRIu64 " %s 2>/dev/null", start, disk) < 1)
989 1.1 martin return false;
990 1.1 martin
991 1.1 martin /*
992 1.1 martin * gpt show should respond with single partition details, but will
993 1.1 martin * fall back to "show -a" output if something is wrong
994 1.1 martin */
995 1.1 martin t = strtok(textbuf, "\n"); /* first line is special */
996 1.1 martin if (strncmp(t, expected_hdr, sizeof(expected_hdr)-1) != 0) {
997 1.1 martin free(textbuf);
998 1.1 martin return false;
999 1.1 martin }
1000 1.1 martin
1001 1.1 martin /* parse output into "old" */
1002 1.1 martin while ((t = strtok(NULL, "\n")) != NULL) {
1003 1.1 martin tt = strsep(&t, " \t");
1004 1.1 martin if (strlen(tt) == 0)
1005 1.1 martin continue;
1006 1.1 martin gpt_add_info(p, tt, t, true);
1007 1.1 martin }
1008 1.1 martin free(textbuf);
1009 1.1 martin
1010 1.1 martin return true;
1011 1.1 martin }
1012 1.1 martin
1013 1.1 martin static bool
1014 1.1 martin gpt_apply_attr(const char *disk, const char *cmd, off_t start, uint todo)
1015 1.1 martin {
1016 1.1 martin size_t i;
1017 1.1 martin char attr_str[STRSIZE];
1018 1.1 martin
1019 1.1 martin if (todo == 0)
1020 1.1 martin return true;
1021 1.1 martin
1022 1.1 martin strcpy(attr_str, "-a ");
1023 1.1 martin for (i = 0; todo != 0; i++) {
1024 1.1 martin if (!(gpt_avail_attrs[i].flag & todo))
1025 1.1 martin continue;
1026 1.1 martin todo &= ~gpt_avail_attrs[i].flag;
1027 1.1 martin if (attr_str[0])
1028 1.1 martin strlcat(attr_str, ",",
1029 1.1 martin sizeof(attr_str));
1030 1.1 martin strlcat(attr_str,
1031 1.1 martin gpt_avail_attrs[i].name,
1032 1.1 martin sizeof(attr_str));
1033 1.1 martin }
1034 1.1 martin if (run_program(RUN_SILENT,
1035 1.1 martin "gpt %s %s -b %" PRIu64 " %s", cmd, attr_str, start, disk) != 0)
1036 1.1 martin return false;
1037 1.1 martin return true;
1038 1.1 martin }
1039 1.1 martin
1040 1.1 martin /*
1041 1.1 martin * Modify an existing on-disk partition.
1042 1.1 martin * Start and size can not be changed here, caller needs to deal
1043 1.1 martin * with that kind of changes upfront.
1044 1.1 martin */
1045 1.1 martin static bool
1046 1.1 martin gpt_modify_part(const char *disk, struct gpt_part_entry *p)
1047 1.1 martin {
1048 1.1 martin struct gpt_part_entry old;
1049 1.1 martin uint todo_set, todo_unset;
1050 1.1 martin
1051 1.1 martin /*
1052 1.1 martin * Query current on-disk state
1053 1.1 martin */
1054 1.1 martin memset(&old, 0, sizeof old);
1055 1.1 martin if (!gpt_read_part(disk, p->gp_start, &old))
1056 1.1 martin return false;
1057 1.1 martin
1058 1.1 martin /* Reject unsupported changes */
1059 1.1 martin if (old.gp_start != p->gp_start || old.gp_size != p->gp_size)
1060 1.1 martin return false;
1061 1.1 martin
1062 1.1 martin /*
1063 1.1 martin * GUID should never change, but the internal copy
1064 1.1 martin * may not yet know it.
1065 1.1 martin */
1066 1.1 martin strcpy(p->gp_id, old.gp_id);
1067 1.1 martin
1068 1.1 martin /* Check type */
1069 1.1 martin if (p->gp_type != old.gp_type) {
1070 1.1 martin if (run_program(RUN_SILENT,
1071 1.1 martin "gpt label -b %" PRIu64 " -T %s %s",
1072 1.1 martin p->gp_start, p->gp_type->tid, disk) != 0)
1073 1.1 martin return false;
1074 1.1 martin }
1075 1.1 martin
1076 1.1 martin /* Check label */
1077 1.1 martin if (strcmp(p->gp_label, old.gp_label) != 0) {
1078 1.1 martin if (run_program(RUN_SILENT,
1079 1.1 martin "gpt label -b %" PRIu64 " -l %s %s",
1080 1.1 martin p->gp_start, p->gp_label, disk) != 0)
1081 1.1 martin return false;
1082 1.1 martin }
1083 1.1 martin
1084 1.1 martin /* Check attributes */
1085 1.1 martin if (p->gp_attr != old.gp_attr) {
1086 1.1 martin if (p->gp_attr == 0) {
1087 1.1 martin if (run_program(RUN_SILENT,
1088 1.1 martin "gpt set -N -b %" PRIu64 " %s",
1089 1.1 martin p->gp_start, disk) != 0)
1090 1.1 martin return false;
1091 1.1 martin } else {
1092 1.1 martin todo_set = (p->gp_attr ^ old.gp_attr) & p->gp_attr;
1093 1.1 martin todo_unset = (p->gp_attr ^ old.gp_attr) & old.gp_attr;
1094 1.1 martin if (!gpt_apply_attr(disk, "unset", p->gp_start,
1095 1.1 martin todo_unset))
1096 1.1 martin return false;
1097 1.1 martin if (!gpt_apply_attr(disk, "set", p->gp_start,
1098 1.1 martin todo_set))
1099 1.1 martin return false;
1100 1.1 martin }
1101 1.1 martin }
1102 1.1 martin
1103 1.1 martin return true;
1104 1.1 martin }
1105 1.1 martin
1106 1.1 martin /*
1107 1.1 martin * verbatim copy from sys/dev/dkwedge/dkwedge_bsdlabel.c:
1108 1.1 martin * map FS_* to wedge strings
1109 1.1 martin */
1110 1.1 martin static const char *
1111 1.1 martin bsdlabel_fstype_to_str(uint8_t fstype)
1112 1.1 martin {
1113 1.1 martin const char *str;
1114 1.1 martin
1115 1.1 martin /*
1116 1.1 martin * For each type known to FSTYPE_DEFN (from <sys/disklabel.h>),
1117 1.1 martin * a suitable case branch will convert the type number to a string.
1118 1.1 martin */
1119 1.1 martin switch (fstype) {
1120 1.1 martin #define FSTYPE_TO_STR_CASE(tag, number, name, fsck, mount) \
1121 1.1 martin case __CONCAT(FS_,tag): str = __CONCAT(DKW_PTYPE_,tag); break;
1122 1.1 martin FSTYPE_DEFN(FSTYPE_TO_STR_CASE)
1123 1.1 martin #undef FSTYPE_TO_STR_CASE
1124 1.1 martin default: str = NULL; break;
1125 1.1 martin }
1126 1.1 martin
1127 1.1 martin return (str);
1128 1.1 martin }
1129 1.1 martin
1130 1.1 martin static bool
1131 1.1 martin gpt_add_wedge(const char *disk, struct gpt_part_entry *p)
1132 1.1 martin {
1133 1.1 martin struct dkwedge_info dkw;
1134 1.1 martin const char *tname;
1135 1.1 martin char diskpath[MAXPATHLEN];
1136 1.1 martin int fd;
1137 1.1 martin
1138 1.1 martin memset(&dkw, 0, sizeof(dkw));
1139 1.1 martin tname = bsdlabel_fstype_to_str(p->fs_type);
1140 1.1 martin if (tname)
1141 1.1 martin strlcpy(dkw.dkw_ptype, tname, sizeof(dkw.dkw_ptype));
1142 1.1 martin
1143 1.1 martin strlcpy((char*)&dkw.dkw_wname, p->gp_id, sizeof(dkw.dkw_wname));
1144 1.1 martin dkw.dkw_offset = p->gp_start;
1145 1.1 martin dkw.dkw_size = p->gp_size;
1146 1.1 martin
1147 1.1 martin fd = opendisk(disk, O_RDWR, diskpath, sizeof(diskpath), 0);
1148 1.1 martin if (fd < 0)
1149 1.1 martin return false;
1150 1.1 martin if (ioctl(fd, DIOCAWEDGE, &dkw) == -1) {
1151 1.1 martin close(fd);
1152 1.1 martin return false;
1153 1.1 martin }
1154 1.1 martin close(fd);
1155 1.1 martin
1156 1.1 martin strlcpy(p->gp_dev_name, dkw.dkw_devname, sizeof(p->gp_dev_name));
1157 1.1 martin p->gp_flags |= GPEF_WEDGE;
1158 1.1 martin return true;
1159 1.1 martin }
1160 1.1 martin
1161 1.1 martin static bool
1162 1.1 martin gpt_get_part_device(const struct disk_partitions *arg,
1163 1.1 martin part_id id, char *devname, size_t max_devname_len, int *part,
1164 1.1 martin enum dev_name_usage usage, bool with_path)
1165 1.1 martin {
1166 1.1 martin const struct gpt_disk_partitions *parts =
1167 1.1 martin (const struct gpt_disk_partitions*)arg;
1168 1.1 martin struct gpt_part_entry *p = parts->partitions;
1169 1.1 martin part_id no;
1170 1.1 martin
1171 1.1 martin
1172 1.1 martin for (no = 0; p != NULL && no < id; no++)
1173 1.1 martin p = p->gp_next;
1174 1.1 martin
1175 1.1 martin if (no != id || p == NULL)
1176 1.1 martin return false;
1177 1.1 martin
1178 1.1 martin if (part)
1179 1.1 martin *part = -1;
1180 1.1 martin
1181 1.1 martin if (!(p->gp_flags & GPEF_WEDGE) &&
1182 1.1 martin (usage == plain_name || usage == raw_dev_name))
1183 1.1 martin gpt_add_wedge(arg->disk, p);
1184 1.1 martin
1185 1.1 martin switch (usage) {
1186 1.1 martin case logical_name:
1187 1.1 martin if (p->gp_label[0] != 0)
1188 1.1 martin snprintf(devname, max_devname_len,
1189 1.1 martin "NAME=%s", p->gp_label);
1190 1.1 martin else
1191 1.1 martin snprintf(devname, max_devname_len,
1192 1.1 martin "NAME=%s", p->gp_id);
1193 1.1 martin break;
1194 1.1 martin case plain_name:
1195 1.1 martin assert(p->gp_flags & GPEF_WEDGE);
1196 1.1 martin if (with_path)
1197 1.1 martin snprintf(devname, max_devname_len, _PATH_DEV "%s",
1198 1.1 martin p->gp_dev_name);
1199 1.1 martin else
1200 1.1 martin strlcpy(devname, p->gp_dev_name, max_devname_len);
1201 1.1 martin break;
1202 1.1 martin case raw_dev_name:
1203 1.1 martin assert(p->gp_flags & GPEF_WEDGE);
1204 1.1 martin if (with_path)
1205 1.1 martin snprintf(devname, max_devname_len, _PATH_DEV "r%s",
1206 1.1 martin p->gp_dev_name);
1207 1.1 martin else
1208 1.1 martin snprintf(devname, max_devname_len, "r%s",
1209 1.1 martin p->gp_dev_name);
1210 1.1 martin break;
1211 1.1 martin default:
1212 1.1 martin return false;
1213 1.1 martin }
1214 1.1 martin
1215 1.1 martin return true;
1216 1.1 martin }
1217 1.1 martin
1218 1.1 martin static bool
1219 1.1 martin gpt_write_to_disk(struct disk_partitions *arg)
1220 1.1 martin {
1221 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
1222 1.1 martin struct gpt_part_entry *p, *n;
1223 1.1 martin char label_arg[sizeof(p->gp_label) + 4];
1224 1.1 martin char diskpath[MAXPATHLEN];
1225 1.1 martin int fd, bits = 0;
1226 1.1 martin bool root_is_new = false, efi_is_new = false;
1227 1.1 martin part_id root_id = NO_PART, efi_id = NO_PART, pno;
1228 1.1 martin
1229 1.1 martin /*
1230 1.1 martin * Remove all wedges on this disk - they may become invalid and we
1231 1.1 martin * have no easy way to associate them with the partitioning data.
1232 1.1 martin * Instead we will explicitly request creation of wedges on demand
1233 1.1 martin * later.
1234 1.1 martin */
1235 1.1 martin fd = opendisk(arg->disk, O_RDWR, diskpath, sizeof(diskpath), 0);
1236 1.1 martin if (fd < 0)
1237 1.1 martin return false;
1238 1.1 martin if (ioctl(fd, DIOCRMWEDGES, &bits) == -1)
1239 1.1 martin return false;
1240 1.1 martin close(fd);
1241 1.1 martin
1242 1.1 martin /*
1243 1.1 martin * Mark all partitions as "have no wedge yet". While there,
1244 1.1 martin * collect first root and efi partition (if available)
1245 1.1 martin */
1246 1.1 martin for (pno = 0, p = parts->partitions; p != NULL; p = p->gp_next, pno++) {
1247 1.1 martin p->gp_flags &= ~GPEF_WEDGE;
1248 1.1 martin if (root_id == NO_PART) {
1249 1.1 martin if (p->gp_type->gent.generic_ptype == PT_root &&
1250 1.1 martin p->gp_start == pm->ptstart) {
1251 1.1 martin root_id = pno;
1252 1.1 martin root_is_new = !(p->gp_flags & GPEF_ON_DISK);
1253 1.1 martin } else if (efi_id == NO_PART &&
1254 1.1 martin p->gp_type->gent.generic_ptype == PT_EFI_SYSTEM) {
1255 1.1 martin efi_id = pno;
1256 1.1 martin efi_is_new = !(p->gp_flags & GPEF_ON_DISK);
1257 1.1 martin }
1258 1.1 martin }
1259 1.1 martin }
1260 1.1 martin
1261 1.1 martin /*
1262 1.1 martin * If no GPT on disk yet, create it.
1263 1.1 martin */
1264 1.1 martin if (!parts->has_gpt) {
1265 1.1 martin char limit[30];
1266 1.1 martin
1267 1.1 martin if (parts->max_num_parts > 0)
1268 1.1 martin sprintf(limit, "-p %zu", parts->max_num_parts);
1269 1.1 martin else
1270 1.1 martin limit[0] = 0;
1271 1.1 martin if (run_program(RUN_SILENT, "gpt create %s %s",
1272 1.1 martin limit, parts->dp.disk))
1273 1.1 martin return false;
1274 1.1 martin parts->has_gpt = true;
1275 1.1 martin }
1276 1.1 martin
1277 1.1 martin /*
1278 1.1 martin * Delete all old partitions
1279 1.1 martin */
1280 1.1 martin for (p = parts->obsolete; p != NULL; p = n) {
1281 1.1 martin run_program(RUN_SILENT, "gpt -n remove -b %" PRIu64 " %s",
1282 1.1 martin p->gp_start, arg->disk);
1283 1.1 martin n = p->gp_next;
1284 1.1 martin free(p);
1285 1.1 martin }
1286 1.1 martin parts->obsolete = NULL;
1287 1.1 martin
1288 1.1 martin /*
1289 1.1 martin * Modify existing but changed partitions
1290 1.1 martin */
1291 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
1292 1.1 martin if (!(p->gp_flags & GPEF_ON_DISK))
1293 1.1 martin continue;
1294 1.1 martin
1295 1.1 martin if (p->gp_flags & GPEF_RESIZED) {
1296 1.1 martin run_program(RUN_SILENT,
1297 1.1 martin "gpt -n resize -b %" PRIu64 " -s %" PRIu64 "s %s",
1298 1.1 martin p->gp_start, p->gp_size, arg->disk);
1299 1.1 martin p->gp_flags &= ~GPEF_RESIZED;
1300 1.1 martin }
1301 1.1 martin
1302 1.1 martin if (!(p->gp_flags & GPEF_MODIFIED))
1303 1.1 martin continue;
1304 1.1 martin
1305 1.1 martin if (!gpt_modify_part(parts->dp.disk, p))
1306 1.1 martin return false;
1307 1.1 martin }
1308 1.1 martin
1309 1.1 martin /*
1310 1.1 martin * Add new partitions
1311 1.1 martin */
1312 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
1313 1.1 martin if (p->gp_flags & GPEF_ON_DISK)
1314 1.1 martin continue;
1315 1.1 martin if (!(p->gp_flags & GPEF_MODIFIED))
1316 1.1 martin continue;
1317 1.1 martin
1318 1.1 martin if (p->gp_label[0] == 0)
1319 1.1 martin label_arg[0] = 0;
1320 1.1 martin else
1321 1.1 martin sprintf(label_arg, "-l %s", p->gp_label);
1322 1.1 martin
1323 1.1 martin run_program(RUN_SILENT,
1324 1.1 martin "gpt -n add -b %" PRIu64 " -s %" PRIu64 "s -t %s %s %s",
1325 1.1 martin p->gp_start, p->gp_size, p->gp_type->tid,
1326 1.1 martin label_arg, arg->disk);
1327 1.1 martin gpt_apply_attr(arg->disk, "set", p->gp_start, p->gp_attr);
1328 1.1 martin gpt_read_part(arg->disk, p->gp_start, p);
1329 1.1 martin p->gp_flags |= GPEF_ON_DISK;
1330 1.1 martin }
1331 1.1 martin
1332 1.1 martin /*
1333 1.1 martin * Additional MD bootloader magic...
1334 1.1 martin */
1335 1.1 martin if (!md_gpt_post_write(&parts->dp, root_id, root_is_new, efi_id,
1336 1.1 martin efi_is_new))
1337 1.1 martin return false;
1338 1.1 martin
1339 1.1 martin return true;
1340 1.1 martin }
1341 1.1 martin
1342 1.1 martin bool
1343 1.1 martin gpt_parts_check(void)
1344 1.1 martin {
1345 1.1 martin
1346 1.1 martin check_available_binaries();
1347 1.1 martin
1348 1.1 martin return have_gpt && have_dk;
1349 1.1 martin }
1350 1.1 martin
1351 1.1 martin static void
1352 1.1 martin gpt_free(struct disk_partitions *arg)
1353 1.1 martin {
1354 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
1355 1.1 martin struct gpt_part_entry *p, *n;
1356 1.1 martin
1357 1.1 martin assert(parts != NULL);
1358 1.1 martin for (p = parts->partitions; p != NULL; p = n) {
1359 1.1 martin free(__UNCONST(p->last_mounted));
1360 1.1 martin n = p->gp_next;
1361 1.1 martin free(p);
1362 1.1 martin }
1363 1.1 martin free(parts);
1364 1.1 martin }
1365 1.1 martin
1366 1.1 martin static bool
1367 1.1 martin gpt_custom_attribute_writable(const struct disk_partitions *arg,
1368 1.1 martin part_id ptn, size_t attr_no)
1369 1.1 martin {
1370 1.1 martin const struct gpt_disk_partitions *parts =
1371 1.1 martin (const struct gpt_disk_partitions*)arg;
1372 1.1 martin size_t i;
1373 1.1 martin struct gpt_part_entry *p;
1374 1.1 martin
1375 1.1 martin if (attr_no >= arg->pscheme->custom_attribute_count)
1376 1.1 martin return false;
1377 1.1 martin
1378 1.1 martin const msg label = arg->pscheme->custom_attributes[attr_no].label;
1379 1.1 martin
1380 1.1 martin /* we can not edit the uuid attribute */
1381 1.1 martin if (label == MSG_ptn_uuid)
1382 1.1 martin return false;
1383 1.1 martin
1384 1.1 martin /* the label is always editable */
1385 1.1 martin if (label == MSG_ptn_label)
1386 1.1 martin return true;
1387 1.1 martin
1388 1.1 martin /* the GPT type is read only */
1389 1.1 martin if (label == MSG_ptn_gpt_type)
1390 1.1 martin return false;
1391 1.1 martin
1392 1.1 martin /* BOOTME makes no sense on swap partitions */
1393 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1394 1.1 martin if (i == ptn)
1395 1.1 martin break;
1396 1.1 martin
1397 1.1 martin if (p == NULL)
1398 1.1 martin return false;
1399 1.1 martin
1400 1.1 martin if (p->fs_type == FS_SWAP || p->gp_type->gent.generic_ptype == PT_swap)
1401 1.1 martin return false;
1402 1.1 martin
1403 1.1 martin return true;
1404 1.1 martin }
1405 1.1 martin
1406 1.1 martin static bool
1407 1.1 martin gpt_format_custom_attribute(const struct disk_partitions *arg,
1408 1.1 martin part_id ptn, size_t attr_no, const struct disk_part_info *info,
1409 1.1 martin char *out, size_t out_space)
1410 1.1 martin {
1411 1.1 martin const struct gpt_disk_partitions *parts =
1412 1.1 martin (const struct gpt_disk_partitions*)arg;
1413 1.1 martin size_t i;
1414 1.1 martin struct gpt_part_entry *p, data;
1415 1.1 martin
1416 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1417 1.1 martin if (i == ptn)
1418 1.1 martin break;
1419 1.1 martin
1420 1.1 martin if (p == NULL)
1421 1.1 martin return false;
1422 1.1 martin
1423 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1424 1.1 martin return false;
1425 1.1 martin
1426 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1427 1.1 martin
1428 1.1 martin if (info != NULL) {
1429 1.1 martin data = *p;
1430 1.1 martin gpt_info_to_part(&data, info, NULL);
1431 1.1 martin p = &data;
1432 1.1 martin }
1433 1.1 martin
1434 1.1 martin if (label == MSG_ptn_label)
1435 1.1 martin strlcpy(out, p->gp_label, out_space);
1436 1.1 martin else if (label == MSG_ptn_uuid)
1437 1.1 martin strlcpy(out, p->gp_id, out_space);
1438 1.1 martin else if (label == MSG_ptn_gpt_type)
1439 1.1 martin strlcpy(out, p->gp_type->gent.description, out_space);
1440 1.1 martin else if (label == MSG_ptn_boot)
1441 1.1 martin strlcpy(out, msg_string(p->gp_attr & GPT_ATTR_BOOT ?
1442 1.1 martin MSG_Yes : MSG_No), out_space);
1443 1.1 martin else
1444 1.1 martin return false;
1445 1.1 martin
1446 1.1 martin return true;
1447 1.1 martin }
1448 1.1 martin
1449 1.1 martin static bool
1450 1.1 martin gpt_custom_attribute_toggle(struct disk_partitions *arg,
1451 1.1 martin part_id ptn, size_t attr_no)
1452 1.1 martin {
1453 1.1 martin const struct gpt_disk_partitions *parts =
1454 1.1 martin (const struct gpt_disk_partitions*)arg;
1455 1.1 martin size_t i;
1456 1.1 martin struct gpt_part_entry *p;
1457 1.1 martin
1458 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1459 1.1 martin if (i == ptn)
1460 1.1 martin break;
1461 1.1 martin
1462 1.1 martin if (p == NULL)
1463 1.1 martin return false;
1464 1.1 martin
1465 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1466 1.1 martin return false;
1467 1.1 martin
1468 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1469 1.1 martin if (label != MSG_ptn_boot)
1470 1.1 martin return false;
1471 1.1 martin
1472 1.1 martin if (p->gp_attr & GPT_ATTR_BOOT) {
1473 1.1 martin p->gp_attr &= ~GPT_ATTR_BOOT;
1474 1.1 martin } else {
1475 1.1 martin for (i = 0, p = parts->partitions; p != NULL;
1476 1.1 martin i++, p = p->gp_next)
1477 1.1 martin if (i == ptn)
1478 1.1 martin p->gp_attr |= GPT_ATTR_BOOT;
1479 1.1 martin else
1480 1.1 martin p->gp_attr &= ~GPT_ATTR_BOOT;
1481 1.1 martin }
1482 1.1 martin return true;
1483 1.1 martin }
1484 1.1 martin
1485 1.1 martin static bool
1486 1.1 martin gpt_custom_attribute_set_str(struct disk_partitions *arg,
1487 1.1 martin part_id ptn, size_t attr_no, const char *new_val)
1488 1.1 martin {
1489 1.1 martin const struct gpt_disk_partitions *parts =
1490 1.1 martin (const struct gpt_disk_partitions*)arg;
1491 1.1 martin size_t i;
1492 1.1 martin struct gpt_part_entry *p;
1493 1.1 martin
1494 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1495 1.1 martin if (i == ptn)
1496 1.1 martin break;
1497 1.1 martin
1498 1.1 martin if (p == NULL)
1499 1.1 martin return false;
1500 1.1 martin
1501 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1502 1.1 martin return false;
1503 1.1 martin
1504 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1505 1.1 martin
1506 1.1 martin if (label != MSG_ptn_label)
1507 1.1 martin return false;
1508 1.1 martin
1509 1.1 martin strlcpy(p->gp_label, new_val, sizeof(p->gp_label));
1510 1.1 martin return true;
1511 1.1 martin }
1512 1.1 martin
1513 1.1 martin static bool
1514 1.1 martin gpt_have_boot_support(const char *disk)
1515 1.1 martin {
1516 1.1 martin #ifdef HAVE_GPT_BOOT
1517 1.1 martin return true;
1518 1.1 martin #else
1519 1.1 martin return false;
1520 1.1 martin #endif
1521 1.1 martin }
1522 1.1 martin
1523 1.1 martin const struct disk_part_custom_attribute gpt_custom_attrs[] = {
1524 1.1 martin { .label = MSG_ptn_label, .type = pet_str },
1525 1.1 martin { .label = MSG_ptn_uuid, .type = pet_str },
1526 1.1 martin { .label = MSG_ptn_gpt_type, .type = pet_str },
1527 1.1 martin { .label = MSG_ptn_boot, .type = pet_bool },
1528 1.1 martin };
1529 1.1 martin
1530 1.1 martin const struct disk_partitioning_scheme
1531 1.1 martin gpt_parts = {
1532 1.1 martin .name = MSG_parttype_gpt,
1533 1.1 martin .short_name = MSG_parttype_gpt_short,
1534 1.1 martin .part_flag_desc = MSG_gpt_flag_desc,
1535 1.1 martin .custom_attribute_count = __arraycount(gpt_custom_attrs),
1536 1.1 martin .custom_attributes = gpt_custom_attrs,
1537 1.1 martin .get_part_types_count = gpt_type_count,
1538 1.1 martin .get_part_type = gpt_get_ptype,
1539 1.1 martin .get_generic_part_type = gpt_get_generic_type,
1540 1.1 martin .get_fs_part_type = gpt_get_fs_part_type,
1541 1.1 martin .get_part_alignment = gpt_get_part_alignment,
1542 1.1 martin .read_from_disk = gpt_read_from_disk,
1543 1.1 martin .create_new_for_disk = gpt_create_new,
1544 1.1 martin .have_boot_support = gpt_have_boot_support,
1545 1.1 martin .can_add_partition = gpt_can_add_partition,
1546 1.1 martin .custom_attribute_writable = gpt_custom_attribute_writable,
1547 1.1 martin .format_custom_attribute = gpt_format_custom_attribute,
1548 1.1 martin .custom_attribute_toggle = gpt_custom_attribute_toggle,
1549 1.1 martin .custom_attribute_set_str = gpt_custom_attribute_set_str,
1550 1.1 martin .get_part_device = gpt_get_part_device,
1551 1.1 martin .max_free_space_at = gpt_max_free_space_at,
1552 1.1 martin .get_free_spaces = gpt_get_free_spaces,
1553 1.1 martin .adapt_foreign_part_info = gpt_adapt,
1554 1.1 martin .get_part_info = gpt_get_part_info,
1555 1.1 martin .get_part_attr_str = gpt_get_part_attr_str,
1556 1.1 martin .set_part_info = gpt_set_part_info,
1557 1.1 martin .add_partition = gpt_add_part,
1558 1.1 martin .delete_all_partitions = gpt_delete_all_partitions,
1559 1.1 martin .delete_partition = gpt_delete_partition,
1560 1.1 martin .write_to_disk = gpt_write_to_disk,
1561 1.1 martin .free = gpt_free,
1562 1.1 martin };
1563