gpt.c revision 1.5 1 1.5 martin /* $NetBSD: gpt.c,v 1.5 2019/07/28 13:17:46 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.5 martin if (p->gp_type != NULL) {
358 1.5 martin
359 1.5 martin if (p->gp_type->fsflags != 0) {
360 1.5 martin const char *lm = get_last_mounted(fd,
361 1.5 martin p->gp_start, &p->fs_type,
362 1.5 martin &p->fs_sub_type, p->gp_type->fsflags);
363 1.5 martin if (lm != NULL && *lm != 0) {
364 1.5 martin char *path = strdup(lm);
365 1.5 martin canonicalize_last_mounted(path);
366 1.5 martin p->last_mounted = path;
367 1.5 martin } else {
368 1.5 martin p->fs_type = p->gp_type->
369 1.5 martin default_fs_type;
370 1.5 martin #ifdef DEFAULT_UFS2
371 1.5 martin fs_is_default = true;
372 1.5 martin #endif
373 1.5 martin }
374 1.1 martin } else {
375 1.1 martin p->fs_type = p->gp_type->default_fs_type;
376 1.1 martin #ifdef DEFAULT_UFS2
377 1.1 martin fs_is_default = true;
378 1.1 martin #endif
379 1.1 martin }
380 1.1 martin #ifdef DEFAULT_UFS2
381 1.5 martin if (fs_is_default && p->fs_type == FS_BSDFFS)
382 1.5 martin p->fs_sub_type = 2;
383 1.1 martin #endif
384 1.1 martin }
385 1.1 martin
386 1.1 martin parts->dp.free_space -= p->gp_size;
387 1.1 martin }
388 1.1 martin close(fd);
389 1.1 martin
390 1.1 martin return &parts->dp;
391 1.1 martin }
392 1.1 martin
393 1.1 martin static struct disk_partitions *
394 1.1 martin gpt_create_new(const char *disk, daddr_t start, daddr_t len, daddr_t total,
395 1.1 martin bool is_boot_drive)
396 1.1 martin {
397 1.1 martin struct gpt_disk_partitions *parts;
398 1.1 martin
399 1.1 martin if (start != 0) {
400 1.1 martin assert(0);
401 1.1 martin return NULL;
402 1.1 martin }
403 1.1 martin
404 1.3 martin parts = calloc(1, sizeof(*parts));
405 1.1 martin if (!parts)
406 1.1 martin return NULL;
407 1.1 martin
408 1.1 martin parts->dp.pscheme = &gpt_parts;
409 1.1 martin parts->dp.disk = disk;
410 1.1 martin
411 1.1 martin gpt_md_init(is_boot_drive, &parts->max_num_parts, &parts->prologue,
412 1.1 martin &parts->epilogue);
413 1.1 martin
414 1.1 martin parts->dp.disk_start = start;
415 1.1 martin parts->dp.disk_size = len;
416 1.1 martin parts->dp.free_space = len - start - parts->prologue - parts->epilogue;
417 1.1 martin parts->has_gpt = false;
418 1.1 martin
419 1.1 martin return &parts->dp;
420 1.1 martin }
421 1.1 martin
422 1.1 martin static bool
423 1.1 martin gpt_get_part_info(const struct disk_partitions *arg, part_id id,
424 1.1 martin struct disk_part_info *info)
425 1.1 martin {
426 1.5 martin static const struct part_type_desc gpt_unknown_type =
427 1.5 martin { .generic_ptype = PT_undef,
428 1.5 martin .short_desc = "<unknown>" };
429 1.1 martin const struct gpt_disk_partitions *parts =
430 1.1 martin (const struct gpt_disk_partitions*)arg;
431 1.1 martin const struct gpt_part_entry *p = parts->partitions;
432 1.1 martin part_id no;
433 1.1 martin
434 1.1 martin for (no = 0; p != NULL && no < id; no++)
435 1.1 martin p = p->gp_next;
436 1.1 martin
437 1.1 martin if (no != id || p == NULL)
438 1.1 martin return false;
439 1.1 martin
440 1.1 martin memset(info, 0, sizeof(*info));
441 1.1 martin info->start = p->gp_start;
442 1.1 martin info->size = p->gp_size;
443 1.1 martin if (p->gp_type)
444 1.1 martin info->nat_type = &p->gp_type->gent;
445 1.5 martin else
446 1.5 martin info->nat_type = &gpt_unknown_type;
447 1.1 martin info->last_mounted = p->last_mounted;
448 1.1 martin info->fs_type = p->fs_type;
449 1.1 martin info->fs_sub_type = p->fs_sub_type;
450 1.1 martin
451 1.1 martin return true;
452 1.1 martin }
453 1.1 martin
454 1.1 martin static bool
455 1.1 martin gpt_get_part_attr_str(const struct disk_partitions *arg, part_id id,
456 1.1 martin char *str, size_t avail_space)
457 1.1 martin {
458 1.1 martin const struct gpt_disk_partitions *parts =
459 1.1 martin (const struct gpt_disk_partitions*)arg;
460 1.1 martin const struct gpt_part_entry *p = parts->partitions;
461 1.1 martin part_id no;
462 1.1 martin static const char *flags = NULL;
463 1.1 martin
464 1.1 martin for (no = 0; p != NULL && no < id; no++)
465 1.1 martin p = p->gp_next;
466 1.1 martin
467 1.1 martin if (no != id || p == NULL)
468 1.1 martin return false;
469 1.1 martin
470 1.1 martin if (flags == NULL)
471 1.1 martin flags = msg_string(MSG_gpt_flags);
472 1.1 martin
473 1.1 martin if (avail_space < 2)
474 1.1 martin return false;
475 1.1 martin
476 1.1 martin if (p->gp_attr & GPT_ATTR_BOOT)
477 1.1 martin *str++ = flags[0];
478 1.1 martin *str = 0;
479 1.1 martin
480 1.1 martin return true;
481 1.1 martin }
482 1.1 martin
483 1.1 martin /*
484 1.1 martin * Find insert position and check for duplicates.
485 1.1 martin * If all goes well, insert the new "entry" in the "list".
486 1.1 martin * If there are collisions, report "no free space".
487 1.1 martin * We keep all lists sorted by start sector number,
488 1.1 martin */
489 1.1 martin static bool
490 1.1 martin gpt_insert_part_into_list(struct gpt_disk_partitions *parts,
491 1.1 martin struct gpt_part_entry **list,
492 1.1 martin struct gpt_part_entry *entry, const char **err_msg)
493 1.1 martin {
494 1.1 martin struct gpt_part_entry *p, *last;
495 1.1 martin
496 1.1 martin /* find the first entry past the new one (if any) */
497 1.1 martin for (last = NULL, p = *list; p != NULL; last = p, p = p->gp_next) {
498 1.1 martin if (p->gp_start > entry->gp_start)
499 1.1 martin break;
500 1.1 martin }
501 1.1 martin
502 1.1 martin /* check if last partition overlaps with new one */
503 1.1 martin if (last) {
504 1.1 martin if (last->gp_start + last->gp_size > entry->gp_start) {
505 1.1 martin if (err_msg)
506 1.1 martin *err_msg = msg_string(MSG_No_free_space);
507 1.1 martin return false;
508 1.1 martin }
509 1.1 martin }
510 1.1 martin
511 1.1 martin if (p == NULL) {
512 1.1 martin entry->gp_next = NULL;
513 1.1 martin if (last != NULL) {
514 1.1 martin last->gp_next = entry;
515 1.1 martin }
516 1.1 martin } else {
517 1.1 martin /* check if new entry overlaps with next */
518 1.1 martin if (entry->gp_start + entry->gp_size > p->gp_start) {
519 1.1 martin if (err_msg)
520 1.1 martin *err_msg = msg_string(MSG_No_free_space);
521 1.1 martin return false;
522 1.1 martin }
523 1.1 martin
524 1.1 martin entry->gp_next = p;
525 1.1 martin if (last != NULL)
526 1.1 martin last->gp_next = entry;
527 1.1 martin else
528 1.1 martin *list = entry;
529 1.1 martin }
530 1.1 martin if (*list == NULL)
531 1.1 martin *list = entry;
532 1.1 martin
533 1.1 martin return true;
534 1.1 martin }
535 1.1 martin
536 1.1 martin static bool
537 1.1 martin gpt_set_part_info(struct disk_partitions *arg, part_id id,
538 1.1 martin const struct disk_part_info *info, const char **err_msg)
539 1.1 martin {
540 1.1 martin struct gpt_disk_partitions *parts =
541 1.1 martin (struct gpt_disk_partitions*)arg;
542 1.1 martin struct gpt_part_entry *p = parts->partitions, *n;
543 1.1 martin part_id no;
544 1.1 martin daddr_t lendiff;
545 1.1 martin
546 1.1 martin for (no = 0; p != NULL && no < id; no++)
547 1.1 martin p = p->gp_next;
548 1.1 martin
549 1.1 martin if (no != id || p == NULL)
550 1.1 martin return false;
551 1.1 martin
552 1.1 martin if ((p->gp_flags & GPEF_ON_DISK)) {
553 1.1 martin if (info->start != p->gp_start) {
554 1.1 martin /* partition moved, we need to delete and re-add */
555 1.1 martin n = calloc(1, sizeof(*n));
556 1.1 martin if (n == NULL) {
557 1.1 martin if (err_msg)
558 1.1 martin *err_msg = err_outofmem;
559 1.1 martin return false;
560 1.1 martin }
561 1.1 martin *n = *p;
562 1.1 martin p->gp_flags &= ~GPEF_ON_DISK;
563 1.1 martin if (!gpt_insert_part_into_list(parts, &parts->obsolete,
564 1.1 martin n, err_msg))
565 1.1 martin return false;
566 1.1 martin } else if (info->size != p->gp_size) {
567 1.1 martin p->gp_flags |= GPEF_RESIZED;
568 1.1 martin }
569 1.1 martin }
570 1.1 martin
571 1.1 martin p->gp_flags |= GPEF_MODIFIED;
572 1.1 martin
573 1.1 martin lendiff = info->size - p->gp_size;
574 1.1 martin parts->dp.free_space -= lendiff;
575 1.1 martin return gpt_info_to_part(p, info, err_msg);
576 1.1 martin }
577 1.1 martin
578 1.1 martin static size_t
579 1.1 martin gpt_get_free_spaces_internal(const struct gpt_disk_partitions *parts,
580 1.1 martin struct disk_part_free_space *result, size_t max_num_result,
581 1.1 martin daddr_t min_space_size, daddr_t align, daddr_t start, daddr_t ignore)
582 1.1 martin {
583 1.1 martin size_t cnt = 0;
584 1.1 martin daddr_t s, e, from, size, end_of_disk;
585 1.1 martin struct gpt_part_entry *p;
586 1.1 martin
587 1.1 martin if (align > 1)
588 1.1 martin start = max(roundup(start, align), align);
589 1.1 martin if (start < 0 || start < (daddr_t)parts->prologue)
590 1.1 martin start = parts->prologue;
591 1.1 martin if (parts->dp.disk_start != 0 && parts->dp.disk_start > start)
592 1.1 martin start = parts->dp.disk_start;
593 1.1 martin if (min_space_size < 1)
594 1.1 martin min_space_size = 1;
595 1.1 martin end_of_disk = parts->dp.disk_start + parts->dp.disk_size
596 1.1 martin - parts->epilogue;
597 1.1 martin from = start;
598 1.1 martin while (from < end_of_disk && cnt < max_num_result) {
599 1.1 martin again:
600 1.1 martin size = parts->dp.disk_start + parts->dp.disk_size - from;
601 1.1 martin start = from;
602 1.1 martin if (start + size > end_of_disk)
603 1.1 martin size = end_of_disk - start;
604 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
605 1.1 martin s = p->gp_start;
606 1.1 martin e = p->gp_size + s;
607 1.1 martin if (s == ignore)
608 1.1 martin continue;
609 1.1 martin if (e < from)
610 1.1 martin continue;
611 1.1 martin if (s <= from && e > from) {
612 1.1 martin if (e - 1 >= end_of_disk)
613 1.1 martin return cnt;
614 1.1 martin from = e + 1;
615 1.1 martin if (align > 1) {
616 1.1 martin from = max(roundup(from, align), align);
617 1.1 martin if (from >= end_of_disk) {
618 1.1 martin size = 0;
619 1.1 martin break;
620 1.1 martin }
621 1.1 martin }
622 1.1 martin goto again;
623 1.1 martin }
624 1.1 martin if (s > from && s - from < size) {
625 1.1 martin size = s - from;
626 1.1 martin }
627 1.1 martin }
628 1.1 martin if (size >= min_space_size) {
629 1.1 martin result->start = start;
630 1.1 martin result->size = size;
631 1.1 martin result++;
632 1.1 martin cnt++;
633 1.1 martin }
634 1.1 martin from += size + 1;
635 1.1 martin if (align > 1)
636 1.1 martin from = max(roundup(from, align), align);
637 1.1 martin }
638 1.1 martin
639 1.1 martin return cnt;
640 1.1 martin }
641 1.1 martin
642 1.1 martin static daddr_t
643 1.1 martin gpt_max_free_space_at(const struct disk_partitions *arg, daddr_t start)
644 1.1 martin {
645 1.1 martin const struct gpt_disk_partitions *parts =
646 1.1 martin (const struct gpt_disk_partitions*)arg;
647 1.1 martin struct disk_part_free_space space;
648 1.1 martin
649 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, 1, 0,
650 1.1 martin start, start) == 1)
651 1.1 martin return space.size;
652 1.1 martin
653 1.1 martin return 0;
654 1.1 martin }
655 1.1 martin
656 1.1 martin static size_t
657 1.1 martin gpt_get_free_spaces(const struct disk_partitions *arg,
658 1.1 martin struct disk_part_free_space *result, size_t max_num_result,
659 1.1 martin daddr_t min_space_size, daddr_t align, daddr_t start,
660 1.1 martin daddr_t ignore)
661 1.1 martin {
662 1.1 martin const struct gpt_disk_partitions *parts =
663 1.1 martin (const struct gpt_disk_partitions*)arg;
664 1.1 martin
665 1.1 martin return gpt_get_free_spaces_internal(parts, result,
666 1.1 martin max_num_result, min_space_size, align, start, ignore);
667 1.1 martin }
668 1.1 martin
669 1.1 martin
670 1.1 martin static bool
671 1.1 martin gpt_adapt(const struct disk_partitions *arg,
672 1.1 martin const struct disk_part_info *src, struct disk_part_info *dest)
673 1.1 martin {
674 1.1 martin /* slightly simplistic, enhance when needed */
675 1.1 martin memcpy(dest, src, sizeof(*dest));
676 1.1 martin
677 1.1 martin if (src->nat_type == NULL)
678 1.1 martin return false;
679 1.1 martin
680 1.1 martin dest->nat_type = arg->pscheme->get_generic_part_type(
681 1.1 martin src->nat_type->generic_ptype);
682 1.1 martin if (dest->nat_type == NULL)
683 1.1 martin dest->nat_type = arg->pscheme->get_generic_part_type(
684 1.1 martin PT_unknown);
685 1.1 martin
686 1.1 martin return true;
687 1.1 martin }
688 1.1 martin
689 1.1 martin static void
690 1.1 martin gpt_match_ptype(const char *name, struct gpt_ptype_desc *t)
691 1.1 martin {
692 1.1 martin size_t i;
693 1.1 martin
694 1.1 martin for (i = 0; i < __arraycount(gpt_fs_types); i++) {
695 1.1 martin if (strcmp(name, gpt_fs_types[i].name) == 0) {
696 1.1 martin t->gent.generic_ptype = gpt_fs_types[i].ptype;
697 1.1 martin t->fsflags = gpt_fs_types[i].fsflags;
698 1.1 martin t->default_fs_type = gpt_fs_types[i].fstype;
699 1.1 martin return;
700 1.1 martin }
701 1.1 martin }
702 1.1 martin
703 1.1 martin t->gent.generic_ptype = PT_unknown;
704 1.1 martin t->fsflags = 0;
705 1.1 martin t->default_fs_type = FS_BSDFFS;
706 1.1 martin }
707 1.1 martin
708 1.1 martin static void
709 1.1 martin gpt_internal_add_ptype(const char *uid, const char *name, const char *desc)
710 1.1 martin {
711 1.1 martin strlcpy(gpt_ptype_descs[gpt_ptype_cnt].tid, uid,
712 1.1 martin sizeof(gpt_ptype_descs[gpt_ptype_cnt].tid));
713 1.1 martin gpt_ptype_descs[gpt_ptype_cnt].gent.short_desc = name;
714 1.1 martin gpt_ptype_descs[gpt_ptype_cnt].gent.description = desc;
715 1.1 martin gpt_match_ptype(name, &gpt_ptype_descs[gpt_ptype_cnt]);
716 1.1 martin gpt_ptype_cnt++;
717 1.1 martin }
718 1.1 martin
719 1.1 martin static void
720 1.1 martin gpt_init_ptypes(void)
721 1.1 martin {
722 1.1 martin if (gpt_ptype_cnt == 0)
723 1.1 martin gpt_uuid_query(gpt_internal_add_ptype);
724 1.1 martin }
725 1.1 martin
726 1.1 martin static size_t
727 1.1 martin gpt_type_count(void)
728 1.1 martin {
729 1.1 martin if (gpt_ptype_cnt == 0)
730 1.1 martin gpt_init_ptypes();
731 1.1 martin
732 1.1 martin return gpt_ptype_cnt;
733 1.1 martin }
734 1.1 martin
735 1.1 martin static const struct part_type_desc *
736 1.1 martin gpt_get_ptype(size_t ndx)
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 if (ndx >= gpt_ptype_cnt)
742 1.1 martin return NULL;
743 1.1 martin
744 1.1 martin return &gpt_ptype_descs[ndx].gent;
745 1.1 martin }
746 1.1 martin
747 1.1 martin static const struct part_type_desc *
748 1.1 martin gpt_get_generic_type(enum part_type gent)
749 1.1 martin {
750 1.1 martin if (gpt_ptype_cnt == 0)
751 1.1 martin gpt_init_ptypes();
752 1.1 martin
753 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
754 1.1 martin if (gpt_ptype_descs[i].gent.generic_ptype == gent)
755 1.1 martin return &gpt_ptype_descs[i].gent;
756 1.1 martin
757 1.1 martin return NULL;
758 1.1 martin }
759 1.1 martin
760 1.1 martin static const struct gpt_ptype_desc *
761 1.1 martin gpt_find_native_type(const struct part_type_desc *gent)
762 1.1 martin {
763 1.1 martin if (gpt_ptype_cnt == 0)
764 1.1 martin gpt_init_ptypes();
765 1.1 martin
766 1.1 martin if (gent == NULL)
767 1.1 martin return NULL;
768 1.1 martin
769 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
770 1.1 martin if (gent == &gpt_ptype_descs[i].gent)
771 1.1 martin return &gpt_ptype_descs[i];
772 1.1 martin
773 1.1 martin gent = gpt_get_generic_type(gent->generic_ptype);
774 1.1 martin if (gent == NULL)
775 1.1 martin return NULL;
776 1.1 martin
777 1.1 martin /* this can not recurse deeper than once, we would not have found a
778 1.1 martin * generic type a few lines above if it would. */
779 1.1 martin return gpt_find_native_type(gent);
780 1.1 martin }
781 1.1 martin
782 1.1 martin static const struct gpt_ptype_desc *
783 1.1 martin gpt_find_guid_type(const char *uid)
784 1.1 martin {
785 1.1 martin if (gpt_ptype_cnt == 0)
786 1.1 martin gpt_init_ptypes();
787 1.1 martin
788 1.1 martin if (uid == NULL || uid[0] == 0)
789 1.1 martin return NULL;
790 1.1 martin
791 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
792 1.1 martin if (strcmp(gpt_ptype_descs[i].tid, uid) == 0)
793 1.1 martin return &gpt_ptype_descs[i];
794 1.1 martin
795 1.1 martin return NULL;
796 1.1 martin }
797 1.1 martin
798 1.1 martin static const struct part_type_desc *
799 1.1 martin gpt_find_type(const char *desc)
800 1.1 martin {
801 1.1 martin if (gpt_ptype_cnt == 0)
802 1.1 martin gpt_init_ptypes();
803 1.1 martin
804 1.1 martin if (desc == NULL || desc[0] == 0)
805 1.1 martin return NULL;
806 1.1 martin
807 1.1 martin for (size_t i = 0; i < gpt_ptype_cnt; i++)
808 1.1 martin if (strcmp(gpt_ptype_descs[i].gent.short_desc, desc) == 0)
809 1.1 martin return &gpt_ptype_descs[i].gent;
810 1.1 martin
811 1.1 martin return NULL;
812 1.1 martin }
813 1.1 martin
814 1.1 martin static const struct part_type_desc *
815 1.1 martin gpt_get_fs_part_type(unsigned fstype, unsigned fs_sub_type)
816 1.1 martin {
817 1.1 martin size_t i;
818 1.1 martin
819 1.1 martin for (i = 0; i < __arraycount(gpt_fs_types); i++)
820 1.1 martin if (fstype == gpt_fs_types[i].fstype)
821 1.1 martin return gpt_find_type(gpt_fs_types[i].name);
822 1.1 martin
823 1.1 martin return gpt_get_generic_type(PT_root);
824 1.1 martin }
825 1.1 martin
826 1.1 martin static daddr_t
827 1.1 martin gpt_get_part_alignment(const struct disk_partitions *parts)
828 1.1 martin {
829 1.1 martin
830 1.1 martin assert(parts->disk_size > 0);
831 1.1 martin if (parts->disk_size < 0)
832 1.1 martin return 1;
833 1.1 martin
834 1.1 martin /* Use 1MB offset/alignemnt for large (>128GB) disks */
835 1.1 martin if (parts->disk_size > HUGE_DISK_SIZE)
836 1.1 martin return 2048;
837 1.1 martin else if (parts->disk_size > TINY_DISK_SIZE)
838 1.1 martin return 64;
839 1.1 martin else
840 1.1 martin return 4;
841 1.1 martin }
842 1.1 martin
843 1.1 martin static bool
844 1.1 martin gpt_can_add_partition(const struct disk_partitions *arg)
845 1.1 martin {
846 1.1 martin const struct gpt_disk_partitions *parts =
847 1.1 martin (const struct gpt_disk_partitions*)arg;
848 1.1 martin struct disk_part_free_space space;
849 1.1 martin daddr_t align;
850 1.1 martin
851 1.1 martin if (parts->dp.num_part >= parts->max_num_parts)
852 1.1 martin return false;
853 1.1 martin
854 1.1 martin align = gpt_get_part_alignment(arg);
855 1.1 martin if (parts->dp.free_space <= align)
856 1.1 martin return false;
857 1.1 martin
858 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, align, align,
859 1.1 martin 0, -1) < 1)
860 1.1 martin return false;
861 1.1 martin
862 1.1 martin return true;
863 1.1 martin }
864 1.1 martin
865 1.1 martin static bool
866 1.1 martin gpt_info_to_part(struct gpt_part_entry *p, const struct disk_part_info *info,
867 1.1 martin const char **err_msg)
868 1.1 martin {
869 1.1 martin p->gp_type = gpt_find_native_type(info->nat_type);
870 1.1 martin p->gp_start = info->start;
871 1.1 martin p->gp_size = info->size;
872 1.1 martin if (info->last_mounted != NULL && info->last_mounted !=
873 1.1 martin p->last_mounted) {
874 1.1 martin free(__UNCONST(p->last_mounted));
875 1.1 martin p->last_mounted = strdup(info->last_mounted);
876 1.1 martin }
877 1.1 martin p->fs_type = info->fs_type;
878 1.1 martin p->fs_sub_type = info->fs_sub_type;
879 1.1 martin
880 1.1 martin return true;
881 1.1 martin }
882 1.1 martin
883 1.1 martin static part_id
884 1.1 martin gpt_add_part(struct disk_partitions *arg,
885 1.1 martin const struct disk_part_info *info, const char **err_msg)
886 1.1 martin {
887 1.1 martin struct gpt_disk_partitions *parts =
888 1.1 martin (struct gpt_disk_partitions*)arg;
889 1.1 martin struct disk_part_free_space space;
890 1.1 martin struct disk_part_info data = *info;
891 1.1 martin struct gpt_part_entry *p;
892 1.1 martin bool ok;
893 1.1 martin
894 1.1 martin if (err_msg != NULL)
895 1.1 martin *err_msg = NULL;
896 1.1 martin
897 1.1 martin if (gpt_get_free_spaces_internal(parts, &space, 1, 1, 1,
898 1.1 martin info->start, -1) < 1) {
899 1.1 martin if (err_msg)
900 1.1 martin *err_msg = msg_string(MSG_No_free_space);
901 1.1 martin return NO_PART;
902 1.1 martin }
903 1.1 martin if (parts->dp.num_part >= parts->max_num_parts) {
904 1.1 martin if (err_msg)
905 1.1 martin *err_msg = msg_string(MSG_err_too_many_partitions);
906 1.1 martin return NO_PART;
907 1.1 martin }
908 1.1 martin
909 1.1 martin if (data.size > space.size)
910 1.1 martin data.size = space.size;
911 1.1 martin
912 1.1 martin p = calloc(1, sizeof(*p));
913 1.1 martin if (p == NULL) {
914 1.1 martin if (err_msg != NULL)
915 1.1 martin *err_msg = INTERNAL_ERROR;
916 1.1 martin return NO_PART;
917 1.1 martin }
918 1.1 martin if (!gpt_info_to_part(p, &data, err_msg)) {
919 1.1 martin free(p);
920 1.1 martin return NO_PART;
921 1.1 martin }
922 1.1 martin p->gp_flags |= GPEF_MODIFIED;
923 1.1 martin ok = gpt_insert_part_into_list(parts, &parts->partitions, p, err_msg);
924 1.1 martin if (ok) {
925 1.1 martin parts->dp.num_part++;
926 1.1 martin parts->dp.free_space -= p->gp_size;
927 1.1 martin return parts->dp.num_part-1;
928 1.1 martin } else {
929 1.1 martin free(p);
930 1.1 martin return NO_PART;
931 1.1 martin }
932 1.1 martin }
933 1.1 martin
934 1.1 martin static bool
935 1.1 martin gpt_delete_partition(struct disk_partitions *arg, part_id id,
936 1.1 martin const char **err_msg)
937 1.1 martin {
938 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
939 1.1 martin struct gpt_part_entry *p, *last = NULL;
940 1.1 martin part_id i;
941 1.1 martin bool res;
942 1.1 martin
943 1.1 martin if (parts->dp.num_part == 0)
944 1.1 martin return false;
945 1.1 martin
946 1.1 martin for (i = 0, p = parts->partitions;
947 1.1 martin i != id && i < parts->dp.num_part && p != NULL;
948 1.1 martin i++, p = p->gp_next)
949 1.1 martin last = p;
950 1.1 martin
951 1.1 martin if (p == NULL) {
952 1.1 martin if (err_msg)
953 1.1 martin *err_msg = INTERNAL_ERROR;
954 1.1 martin return false;
955 1.1 martin }
956 1.1 martin
957 1.1 martin if (last == NULL)
958 1.1 martin parts->partitions = p->gp_next;
959 1.1 martin else
960 1.1 martin last->gp_next = p->gp_next;
961 1.1 martin
962 1.1 martin res = true;
963 1.1 martin if (p->gp_flags & GPEF_ON_DISK) {
964 1.1 martin if (!gpt_insert_part_into_list(parts, &parts->obsolete,
965 1.1 martin p, err_msg))
966 1.1 martin res = false;
967 1.1 martin } else {
968 1.1 martin free(p);
969 1.1 martin }
970 1.1 martin
971 1.1 martin if (res) {
972 1.1 martin parts->dp.num_part--;
973 1.1 martin parts->dp.free_space += p->gp_size;
974 1.1 martin }
975 1.1 martin
976 1.1 martin return res;
977 1.1 martin }
978 1.1 martin
979 1.1 martin static bool
980 1.1 martin gpt_delete_all_partitions(struct disk_partitions *arg)
981 1.1 martin {
982 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
983 1.1 martin
984 1.1 martin while (parts->dp.num_part > 0) {
985 1.1 martin if (!gpt_delete_partition(&parts->dp, 0, NULL))
986 1.1 martin return false;
987 1.1 martin }
988 1.1 martin
989 1.1 martin return true;
990 1.1 martin }
991 1.1 martin
992 1.1 martin static bool
993 1.1 martin gpt_read_part(const char *disk, daddr_t start, struct gpt_part_entry *p)
994 1.1 martin {
995 1.1 martin char *textbuf, *t, *tt;
996 1.1 martin static const char expected_hdr[] = "Details for index ";
997 1.1 martin
998 1.1 martin /* run gpt show for this partition */
999 1.1 martin if (collect(T_OUTPUT, &textbuf,
1000 1.1 martin "gpt -r show -b %" PRIu64 " %s 2>/dev/null", start, disk) < 1)
1001 1.1 martin return false;
1002 1.1 martin
1003 1.1 martin /*
1004 1.1 martin * gpt show should respond with single partition details, but will
1005 1.1 martin * fall back to "show -a" output if something is wrong
1006 1.1 martin */
1007 1.1 martin t = strtok(textbuf, "\n"); /* first line is special */
1008 1.1 martin if (strncmp(t, expected_hdr, sizeof(expected_hdr)-1) != 0) {
1009 1.1 martin free(textbuf);
1010 1.1 martin return false;
1011 1.1 martin }
1012 1.1 martin
1013 1.1 martin /* parse output into "old" */
1014 1.1 martin while ((t = strtok(NULL, "\n")) != NULL) {
1015 1.1 martin tt = strsep(&t, " \t");
1016 1.1 martin if (strlen(tt) == 0)
1017 1.1 martin continue;
1018 1.1 martin gpt_add_info(p, tt, t, true);
1019 1.1 martin }
1020 1.1 martin free(textbuf);
1021 1.1 martin
1022 1.1 martin return true;
1023 1.1 martin }
1024 1.1 martin
1025 1.1 martin static bool
1026 1.1 martin gpt_apply_attr(const char *disk, const char *cmd, off_t start, uint todo)
1027 1.1 martin {
1028 1.1 martin size_t i;
1029 1.1 martin char attr_str[STRSIZE];
1030 1.1 martin
1031 1.1 martin if (todo == 0)
1032 1.1 martin return true;
1033 1.1 martin
1034 1.1 martin strcpy(attr_str, "-a ");
1035 1.1 martin for (i = 0; todo != 0; i++) {
1036 1.1 martin if (!(gpt_avail_attrs[i].flag & todo))
1037 1.1 martin continue;
1038 1.1 martin todo &= ~gpt_avail_attrs[i].flag;
1039 1.1 martin if (attr_str[0])
1040 1.1 martin strlcat(attr_str, ",",
1041 1.1 martin sizeof(attr_str));
1042 1.1 martin strlcat(attr_str,
1043 1.1 martin gpt_avail_attrs[i].name,
1044 1.1 martin sizeof(attr_str));
1045 1.1 martin }
1046 1.1 martin if (run_program(RUN_SILENT,
1047 1.1 martin "gpt %s %s -b %" PRIu64 " %s", cmd, attr_str, start, disk) != 0)
1048 1.1 martin return false;
1049 1.1 martin return true;
1050 1.1 martin }
1051 1.1 martin
1052 1.1 martin /*
1053 1.1 martin * Modify an existing on-disk partition.
1054 1.1 martin * Start and size can not be changed here, caller needs to deal
1055 1.1 martin * with that kind of changes upfront.
1056 1.1 martin */
1057 1.1 martin static bool
1058 1.1 martin gpt_modify_part(const char *disk, struct gpt_part_entry *p)
1059 1.1 martin {
1060 1.1 martin struct gpt_part_entry old;
1061 1.1 martin uint todo_set, todo_unset;
1062 1.1 martin
1063 1.1 martin /*
1064 1.1 martin * Query current on-disk state
1065 1.1 martin */
1066 1.1 martin memset(&old, 0, sizeof old);
1067 1.1 martin if (!gpt_read_part(disk, p->gp_start, &old))
1068 1.1 martin return false;
1069 1.1 martin
1070 1.1 martin /* Reject unsupported changes */
1071 1.1 martin if (old.gp_start != p->gp_start || old.gp_size != p->gp_size)
1072 1.1 martin return false;
1073 1.1 martin
1074 1.1 martin /*
1075 1.1 martin * GUID should never change, but the internal copy
1076 1.1 martin * may not yet know it.
1077 1.1 martin */
1078 1.1 martin strcpy(p->gp_id, old.gp_id);
1079 1.1 martin
1080 1.1 martin /* Check type */
1081 1.1 martin if (p->gp_type != old.gp_type) {
1082 1.1 martin if (run_program(RUN_SILENT,
1083 1.1 martin "gpt label -b %" PRIu64 " -T %s %s",
1084 1.1 martin p->gp_start, p->gp_type->tid, disk) != 0)
1085 1.1 martin return false;
1086 1.1 martin }
1087 1.1 martin
1088 1.1 martin /* Check label */
1089 1.1 martin if (strcmp(p->gp_label, old.gp_label) != 0) {
1090 1.1 martin if (run_program(RUN_SILENT,
1091 1.1 martin "gpt label -b %" PRIu64 " -l %s %s",
1092 1.1 martin p->gp_start, p->gp_label, disk) != 0)
1093 1.1 martin return false;
1094 1.1 martin }
1095 1.1 martin
1096 1.1 martin /* Check attributes */
1097 1.1 martin if (p->gp_attr != old.gp_attr) {
1098 1.1 martin if (p->gp_attr == 0) {
1099 1.1 martin if (run_program(RUN_SILENT,
1100 1.1 martin "gpt set -N -b %" PRIu64 " %s",
1101 1.1 martin p->gp_start, disk) != 0)
1102 1.1 martin return false;
1103 1.1 martin } else {
1104 1.1 martin todo_set = (p->gp_attr ^ old.gp_attr) & p->gp_attr;
1105 1.1 martin todo_unset = (p->gp_attr ^ old.gp_attr) & old.gp_attr;
1106 1.1 martin if (!gpt_apply_attr(disk, "unset", p->gp_start,
1107 1.1 martin todo_unset))
1108 1.1 martin return false;
1109 1.1 martin if (!gpt_apply_attr(disk, "set", p->gp_start,
1110 1.1 martin todo_set))
1111 1.1 martin return false;
1112 1.1 martin }
1113 1.1 martin }
1114 1.1 martin
1115 1.1 martin return true;
1116 1.1 martin }
1117 1.1 martin
1118 1.1 martin /*
1119 1.1 martin * verbatim copy from sys/dev/dkwedge/dkwedge_bsdlabel.c:
1120 1.1 martin * map FS_* to wedge strings
1121 1.1 martin */
1122 1.1 martin static const char *
1123 1.1 martin bsdlabel_fstype_to_str(uint8_t fstype)
1124 1.1 martin {
1125 1.1 martin const char *str;
1126 1.1 martin
1127 1.1 martin /*
1128 1.1 martin * For each type known to FSTYPE_DEFN (from <sys/disklabel.h>),
1129 1.1 martin * a suitable case branch will convert the type number to a string.
1130 1.1 martin */
1131 1.1 martin switch (fstype) {
1132 1.1 martin #define FSTYPE_TO_STR_CASE(tag, number, name, fsck, mount) \
1133 1.1 martin case __CONCAT(FS_,tag): str = __CONCAT(DKW_PTYPE_,tag); break;
1134 1.1 martin FSTYPE_DEFN(FSTYPE_TO_STR_CASE)
1135 1.1 martin #undef FSTYPE_TO_STR_CASE
1136 1.1 martin default: str = NULL; break;
1137 1.1 martin }
1138 1.1 martin
1139 1.1 martin return (str);
1140 1.1 martin }
1141 1.1 martin
1142 1.1 martin static bool
1143 1.1 martin gpt_add_wedge(const char *disk, struct gpt_part_entry *p)
1144 1.1 martin {
1145 1.1 martin struct dkwedge_info dkw;
1146 1.1 martin const char *tname;
1147 1.1 martin char diskpath[MAXPATHLEN];
1148 1.1 martin int fd;
1149 1.1 martin
1150 1.1 martin memset(&dkw, 0, sizeof(dkw));
1151 1.1 martin tname = bsdlabel_fstype_to_str(p->fs_type);
1152 1.1 martin if (tname)
1153 1.1 martin strlcpy(dkw.dkw_ptype, tname, sizeof(dkw.dkw_ptype));
1154 1.1 martin
1155 1.1 martin strlcpy((char*)&dkw.dkw_wname, p->gp_id, sizeof(dkw.dkw_wname));
1156 1.1 martin dkw.dkw_offset = p->gp_start;
1157 1.1 martin dkw.dkw_size = p->gp_size;
1158 1.1 martin
1159 1.1 martin fd = opendisk(disk, O_RDWR, diskpath, sizeof(diskpath), 0);
1160 1.1 martin if (fd < 0)
1161 1.1 martin return false;
1162 1.1 martin if (ioctl(fd, DIOCAWEDGE, &dkw) == -1) {
1163 1.1 martin close(fd);
1164 1.1 martin return false;
1165 1.1 martin }
1166 1.1 martin close(fd);
1167 1.1 martin
1168 1.1 martin strlcpy(p->gp_dev_name, dkw.dkw_devname, sizeof(p->gp_dev_name));
1169 1.1 martin p->gp_flags |= GPEF_WEDGE;
1170 1.1 martin return true;
1171 1.1 martin }
1172 1.1 martin
1173 1.1 martin static bool
1174 1.1 martin gpt_get_part_device(const struct disk_partitions *arg,
1175 1.1 martin part_id id, char *devname, size_t max_devname_len, int *part,
1176 1.1 martin enum dev_name_usage usage, bool with_path)
1177 1.1 martin {
1178 1.1 martin const struct gpt_disk_partitions *parts =
1179 1.1 martin (const struct gpt_disk_partitions*)arg;
1180 1.1 martin struct gpt_part_entry *p = parts->partitions;
1181 1.1 martin part_id no;
1182 1.1 martin
1183 1.1 martin
1184 1.1 martin for (no = 0; p != NULL && no < id; no++)
1185 1.1 martin p = p->gp_next;
1186 1.1 martin
1187 1.1 martin if (no != id || p == NULL)
1188 1.1 martin return false;
1189 1.1 martin
1190 1.1 martin if (part)
1191 1.1 martin *part = -1;
1192 1.1 martin
1193 1.1 martin if (!(p->gp_flags & GPEF_WEDGE) &&
1194 1.1 martin (usage == plain_name || usage == raw_dev_name))
1195 1.1 martin gpt_add_wedge(arg->disk, p);
1196 1.1 martin
1197 1.1 martin switch (usage) {
1198 1.1 martin case logical_name:
1199 1.1 martin if (p->gp_label[0] != 0)
1200 1.1 martin snprintf(devname, max_devname_len,
1201 1.1 martin "NAME=%s", p->gp_label);
1202 1.1 martin else
1203 1.1 martin snprintf(devname, max_devname_len,
1204 1.1 martin "NAME=%s", p->gp_id);
1205 1.1 martin break;
1206 1.1 martin case plain_name:
1207 1.1 martin assert(p->gp_flags & GPEF_WEDGE);
1208 1.1 martin if (with_path)
1209 1.1 martin snprintf(devname, max_devname_len, _PATH_DEV "%s",
1210 1.1 martin p->gp_dev_name);
1211 1.1 martin else
1212 1.1 martin strlcpy(devname, p->gp_dev_name, max_devname_len);
1213 1.1 martin break;
1214 1.1 martin case raw_dev_name:
1215 1.1 martin assert(p->gp_flags & GPEF_WEDGE);
1216 1.1 martin if (with_path)
1217 1.1 martin snprintf(devname, max_devname_len, _PATH_DEV "r%s",
1218 1.1 martin p->gp_dev_name);
1219 1.1 martin else
1220 1.1 martin snprintf(devname, max_devname_len, "r%s",
1221 1.1 martin p->gp_dev_name);
1222 1.1 martin break;
1223 1.1 martin default:
1224 1.1 martin return false;
1225 1.1 martin }
1226 1.1 martin
1227 1.1 martin return true;
1228 1.1 martin }
1229 1.1 martin
1230 1.1 martin static bool
1231 1.1 martin gpt_write_to_disk(struct disk_partitions *arg)
1232 1.1 martin {
1233 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
1234 1.1 martin struct gpt_part_entry *p, *n;
1235 1.1 martin char label_arg[sizeof(p->gp_label) + 4];
1236 1.1 martin char diskpath[MAXPATHLEN];
1237 1.1 martin int fd, bits = 0;
1238 1.1 martin bool root_is_new = false, efi_is_new = false;
1239 1.1 martin part_id root_id = NO_PART, efi_id = NO_PART, pno;
1240 1.1 martin
1241 1.1 martin /*
1242 1.1 martin * Remove all wedges on this disk - they may become invalid and we
1243 1.1 martin * have no easy way to associate them with the partitioning data.
1244 1.1 martin * Instead we will explicitly request creation of wedges on demand
1245 1.1 martin * later.
1246 1.1 martin */
1247 1.1 martin fd = opendisk(arg->disk, O_RDWR, diskpath, sizeof(diskpath), 0);
1248 1.1 martin if (fd < 0)
1249 1.1 martin return false;
1250 1.1 martin if (ioctl(fd, DIOCRMWEDGES, &bits) == -1)
1251 1.1 martin return false;
1252 1.1 martin close(fd);
1253 1.1 martin
1254 1.1 martin /*
1255 1.1 martin * Mark all partitions as "have no wedge yet". While there,
1256 1.1 martin * collect first root and efi partition (if available)
1257 1.1 martin */
1258 1.1 martin for (pno = 0, p = parts->partitions; p != NULL; p = p->gp_next, pno++) {
1259 1.1 martin p->gp_flags &= ~GPEF_WEDGE;
1260 1.5 martin if (root_id == NO_PART && p->gp_type != NULL) {
1261 1.1 martin if (p->gp_type->gent.generic_ptype == PT_root &&
1262 1.1 martin p->gp_start == pm->ptstart) {
1263 1.1 martin root_id = pno;
1264 1.1 martin root_is_new = !(p->gp_flags & GPEF_ON_DISK);
1265 1.1 martin } else if (efi_id == NO_PART &&
1266 1.1 martin p->gp_type->gent.generic_ptype == PT_EFI_SYSTEM) {
1267 1.1 martin efi_id = pno;
1268 1.1 martin efi_is_new = !(p->gp_flags & GPEF_ON_DISK);
1269 1.1 martin }
1270 1.1 martin }
1271 1.1 martin }
1272 1.1 martin
1273 1.1 martin /*
1274 1.1 martin * If no GPT on disk yet, create it.
1275 1.1 martin */
1276 1.1 martin if (!parts->has_gpt) {
1277 1.1 martin char limit[30];
1278 1.1 martin
1279 1.1 martin if (parts->max_num_parts > 0)
1280 1.1 martin sprintf(limit, "-p %zu", parts->max_num_parts);
1281 1.1 martin else
1282 1.1 martin limit[0] = 0;
1283 1.1 martin if (run_program(RUN_SILENT, "gpt create %s %s",
1284 1.1 martin limit, parts->dp.disk))
1285 1.1 martin return false;
1286 1.1 martin parts->has_gpt = true;
1287 1.1 martin }
1288 1.1 martin
1289 1.1 martin /*
1290 1.1 martin * Delete all old partitions
1291 1.1 martin */
1292 1.1 martin for (p = parts->obsolete; p != NULL; p = n) {
1293 1.1 martin run_program(RUN_SILENT, "gpt -n remove -b %" PRIu64 " %s",
1294 1.1 martin p->gp_start, arg->disk);
1295 1.1 martin n = p->gp_next;
1296 1.1 martin free(p);
1297 1.1 martin }
1298 1.1 martin parts->obsolete = NULL;
1299 1.1 martin
1300 1.1 martin /*
1301 1.1 martin * Modify existing but changed partitions
1302 1.1 martin */
1303 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
1304 1.1 martin if (!(p->gp_flags & GPEF_ON_DISK))
1305 1.1 martin continue;
1306 1.1 martin
1307 1.1 martin if (p->gp_flags & GPEF_RESIZED) {
1308 1.1 martin run_program(RUN_SILENT,
1309 1.1 martin "gpt -n resize -b %" PRIu64 " -s %" PRIu64 "s %s",
1310 1.1 martin p->gp_start, p->gp_size, arg->disk);
1311 1.1 martin p->gp_flags &= ~GPEF_RESIZED;
1312 1.1 martin }
1313 1.1 martin
1314 1.1 martin if (!(p->gp_flags & GPEF_MODIFIED))
1315 1.1 martin continue;
1316 1.1 martin
1317 1.1 martin if (!gpt_modify_part(parts->dp.disk, p))
1318 1.1 martin return false;
1319 1.1 martin }
1320 1.1 martin
1321 1.1 martin /*
1322 1.1 martin * Add new partitions
1323 1.1 martin */
1324 1.1 martin for (p = parts->partitions; p != NULL; p = p->gp_next) {
1325 1.1 martin if (p->gp_flags & GPEF_ON_DISK)
1326 1.1 martin continue;
1327 1.1 martin if (!(p->gp_flags & GPEF_MODIFIED))
1328 1.1 martin continue;
1329 1.1 martin
1330 1.1 martin if (p->gp_label[0] == 0)
1331 1.1 martin label_arg[0] = 0;
1332 1.1 martin else
1333 1.1 martin sprintf(label_arg, "-l %s", p->gp_label);
1334 1.1 martin
1335 1.5 martin if (p->gp_type != NULL)
1336 1.5 martin run_program(RUN_SILENT,
1337 1.5 martin "gpt -n add -b %" PRIu64 " -s %" PRIu64
1338 1.5 martin "s -t %s %s %s",
1339 1.5 martin p->gp_start, p->gp_size, p->gp_type->tid,
1340 1.5 martin label_arg, arg->disk);
1341 1.5 martin else
1342 1.5 martin run_program(RUN_SILENT,
1343 1.5 martin "gpt -n add -b %" PRIu64 " -s %" PRIu64
1344 1.5 martin "s %s %s",
1345 1.5 martin p->gp_start, p->gp_size, label_arg, arg->disk);
1346 1.1 martin gpt_apply_attr(arg->disk, "set", p->gp_start, p->gp_attr);
1347 1.1 martin gpt_read_part(arg->disk, p->gp_start, p);
1348 1.1 martin p->gp_flags |= GPEF_ON_DISK;
1349 1.1 martin }
1350 1.1 martin
1351 1.1 martin /*
1352 1.1 martin * Additional MD bootloader magic...
1353 1.1 martin */
1354 1.1 martin if (!md_gpt_post_write(&parts->dp, root_id, root_is_new, efi_id,
1355 1.1 martin efi_is_new))
1356 1.1 martin return false;
1357 1.1 martin
1358 1.1 martin return true;
1359 1.1 martin }
1360 1.1 martin
1361 1.1 martin bool
1362 1.1 martin gpt_parts_check(void)
1363 1.1 martin {
1364 1.1 martin
1365 1.1 martin check_available_binaries();
1366 1.1 martin
1367 1.1 martin return have_gpt && have_dk;
1368 1.1 martin }
1369 1.1 martin
1370 1.1 martin static void
1371 1.1 martin gpt_free(struct disk_partitions *arg)
1372 1.1 martin {
1373 1.1 martin struct gpt_disk_partitions *parts = (struct gpt_disk_partitions*)arg;
1374 1.1 martin struct gpt_part_entry *p, *n;
1375 1.1 martin
1376 1.1 martin assert(parts != NULL);
1377 1.1 martin for (p = parts->partitions; p != NULL; p = n) {
1378 1.1 martin free(__UNCONST(p->last_mounted));
1379 1.1 martin n = p->gp_next;
1380 1.1 martin free(p);
1381 1.1 martin }
1382 1.1 martin free(parts);
1383 1.1 martin }
1384 1.1 martin
1385 1.1 martin static bool
1386 1.1 martin gpt_custom_attribute_writable(const struct disk_partitions *arg,
1387 1.1 martin part_id ptn, size_t attr_no)
1388 1.1 martin {
1389 1.1 martin const struct gpt_disk_partitions *parts =
1390 1.1 martin (const struct gpt_disk_partitions*)arg;
1391 1.1 martin size_t i;
1392 1.1 martin struct gpt_part_entry *p;
1393 1.1 martin
1394 1.1 martin if (attr_no >= arg->pscheme->custom_attribute_count)
1395 1.1 martin return false;
1396 1.1 martin
1397 1.1 martin const msg label = arg->pscheme->custom_attributes[attr_no].label;
1398 1.1 martin
1399 1.1 martin /* we can not edit the uuid attribute */
1400 1.1 martin if (label == MSG_ptn_uuid)
1401 1.1 martin return false;
1402 1.1 martin
1403 1.1 martin /* the label is always editable */
1404 1.1 martin if (label == MSG_ptn_label)
1405 1.1 martin return true;
1406 1.1 martin
1407 1.1 martin /* the GPT type is read only */
1408 1.1 martin if (label == MSG_ptn_gpt_type)
1409 1.1 martin return false;
1410 1.1 martin
1411 1.1 martin /* BOOTME makes no sense on swap partitions */
1412 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1413 1.1 martin if (i == ptn)
1414 1.1 martin break;
1415 1.1 martin
1416 1.1 martin if (p == NULL)
1417 1.1 martin return false;
1418 1.1 martin
1419 1.5 martin if (p->fs_type == FS_SWAP ||
1420 1.5 martin (p->gp_type != NULL && p->gp_type->gent.generic_ptype == PT_swap))
1421 1.1 martin return false;
1422 1.1 martin
1423 1.1 martin return true;
1424 1.1 martin }
1425 1.1 martin
1426 1.1 martin static bool
1427 1.1 martin gpt_format_custom_attribute(const struct disk_partitions *arg,
1428 1.1 martin part_id ptn, size_t attr_no, const struct disk_part_info *info,
1429 1.1 martin char *out, size_t out_space)
1430 1.1 martin {
1431 1.1 martin const struct gpt_disk_partitions *parts =
1432 1.1 martin (const struct gpt_disk_partitions*)arg;
1433 1.1 martin size_t i;
1434 1.1 martin struct gpt_part_entry *p, data;
1435 1.1 martin
1436 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1437 1.1 martin if (i == ptn)
1438 1.1 martin break;
1439 1.1 martin
1440 1.1 martin if (p == NULL)
1441 1.1 martin return false;
1442 1.1 martin
1443 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1444 1.1 martin return false;
1445 1.1 martin
1446 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1447 1.1 martin
1448 1.1 martin if (info != NULL) {
1449 1.1 martin data = *p;
1450 1.1 martin gpt_info_to_part(&data, info, NULL);
1451 1.1 martin p = &data;
1452 1.1 martin }
1453 1.1 martin
1454 1.1 martin if (label == MSG_ptn_label)
1455 1.1 martin strlcpy(out, p->gp_label, out_space);
1456 1.1 martin else if (label == MSG_ptn_uuid)
1457 1.1 martin strlcpy(out, p->gp_id, out_space);
1458 1.5 martin else if (label == MSG_ptn_gpt_type) {
1459 1.5 martin if (p->gp_type != NULL)
1460 1.5 martin strlcpy(out, p->gp_type->gent.description, out_space);
1461 1.5 martin else if (out_space > 1)
1462 1.5 martin out[0] = 0;
1463 1.5 martin } else if (label == MSG_ptn_boot)
1464 1.1 martin strlcpy(out, msg_string(p->gp_attr & GPT_ATTR_BOOT ?
1465 1.1 martin MSG_Yes : MSG_No), out_space);
1466 1.1 martin else
1467 1.1 martin return false;
1468 1.1 martin
1469 1.1 martin return true;
1470 1.1 martin }
1471 1.1 martin
1472 1.1 martin static bool
1473 1.1 martin gpt_custom_attribute_toggle(struct disk_partitions *arg,
1474 1.1 martin part_id ptn, size_t attr_no)
1475 1.1 martin {
1476 1.1 martin const struct gpt_disk_partitions *parts =
1477 1.1 martin (const struct gpt_disk_partitions*)arg;
1478 1.1 martin size_t i;
1479 1.1 martin struct gpt_part_entry *p;
1480 1.1 martin
1481 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1482 1.1 martin if (i == ptn)
1483 1.1 martin break;
1484 1.1 martin
1485 1.1 martin if (p == NULL)
1486 1.1 martin return false;
1487 1.1 martin
1488 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1489 1.1 martin return false;
1490 1.1 martin
1491 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1492 1.1 martin if (label != MSG_ptn_boot)
1493 1.1 martin return false;
1494 1.1 martin
1495 1.1 martin if (p->gp_attr & GPT_ATTR_BOOT) {
1496 1.1 martin p->gp_attr &= ~GPT_ATTR_BOOT;
1497 1.1 martin } else {
1498 1.1 martin for (i = 0, p = parts->partitions; p != NULL;
1499 1.1 martin i++, p = p->gp_next)
1500 1.1 martin if (i == ptn)
1501 1.1 martin p->gp_attr |= GPT_ATTR_BOOT;
1502 1.1 martin else
1503 1.1 martin p->gp_attr &= ~GPT_ATTR_BOOT;
1504 1.1 martin }
1505 1.1 martin return true;
1506 1.1 martin }
1507 1.1 martin
1508 1.1 martin static bool
1509 1.1 martin gpt_custom_attribute_set_str(struct disk_partitions *arg,
1510 1.1 martin part_id ptn, size_t attr_no, const char *new_val)
1511 1.1 martin {
1512 1.1 martin const struct gpt_disk_partitions *parts =
1513 1.1 martin (const struct gpt_disk_partitions*)arg;
1514 1.1 martin size_t i;
1515 1.1 martin struct gpt_part_entry *p;
1516 1.1 martin
1517 1.1 martin for (i = 0, p = parts->partitions; p != NULL; i++, p = p->gp_next)
1518 1.1 martin if (i == ptn)
1519 1.1 martin break;
1520 1.1 martin
1521 1.1 martin if (p == NULL)
1522 1.1 martin return false;
1523 1.1 martin
1524 1.1 martin if (attr_no >= parts->dp.pscheme->custom_attribute_count)
1525 1.1 martin return false;
1526 1.1 martin
1527 1.1 martin const msg label = parts->dp.pscheme->custom_attributes[attr_no].label;
1528 1.1 martin
1529 1.1 martin if (label != MSG_ptn_label)
1530 1.1 martin return false;
1531 1.1 martin
1532 1.1 martin strlcpy(p->gp_label, new_val, sizeof(p->gp_label));
1533 1.1 martin return true;
1534 1.1 martin }
1535 1.1 martin
1536 1.1 martin static bool
1537 1.1 martin gpt_have_boot_support(const char *disk)
1538 1.1 martin {
1539 1.1 martin #ifdef HAVE_GPT_BOOT
1540 1.1 martin return true;
1541 1.1 martin #else
1542 1.1 martin return false;
1543 1.1 martin #endif
1544 1.1 martin }
1545 1.1 martin
1546 1.1 martin const struct disk_part_custom_attribute gpt_custom_attrs[] = {
1547 1.1 martin { .label = MSG_ptn_label, .type = pet_str },
1548 1.1 martin { .label = MSG_ptn_uuid, .type = pet_str },
1549 1.1 martin { .label = MSG_ptn_gpt_type, .type = pet_str },
1550 1.1 martin { .label = MSG_ptn_boot, .type = pet_bool },
1551 1.1 martin };
1552 1.1 martin
1553 1.1 martin const struct disk_partitioning_scheme
1554 1.1 martin gpt_parts = {
1555 1.1 martin .name = MSG_parttype_gpt,
1556 1.1 martin .short_name = MSG_parttype_gpt_short,
1557 1.1 martin .part_flag_desc = MSG_gpt_flag_desc,
1558 1.1 martin .custom_attribute_count = __arraycount(gpt_custom_attrs),
1559 1.1 martin .custom_attributes = gpt_custom_attrs,
1560 1.1 martin .get_part_types_count = gpt_type_count,
1561 1.1 martin .get_part_type = gpt_get_ptype,
1562 1.1 martin .get_generic_part_type = gpt_get_generic_type,
1563 1.1 martin .get_fs_part_type = gpt_get_fs_part_type,
1564 1.1 martin .get_part_alignment = gpt_get_part_alignment,
1565 1.1 martin .read_from_disk = gpt_read_from_disk,
1566 1.1 martin .create_new_for_disk = gpt_create_new,
1567 1.1 martin .have_boot_support = gpt_have_boot_support,
1568 1.1 martin .can_add_partition = gpt_can_add_partition,
1569 1.1 martin .custom_attribute_writable = gpt_custom_attribute_writable,
1570 1.1 martin .format_custom_attribute = gpt_format_custom_attribute,
1571 1.1 martin .custom_attribute_toggle = gpt_custom_attribute_toggle,
1572 1.1 martin .custom_attribute_set_str = gpt_custom_attribute_set_str,
1573 1.1 martin .get_part_device = gpt_get_part_device,
1574 1.1 martin .max_free_space_at = gpt_max_free_space_at,
1575 1.1 martin .get_free_spaces = gpt_get_free_spaces,
1576 1.1 martin .adapt_foreign_part_info = gpt_adapt,
1577 1.1 martin .get_part_info = gpt_get_part_info,
1578 1.1 martin .get_part_attr_str = gpt_get_part_attr_str,
1579 1.1 martin .set_part_info = gpt_set_part_info,
1580 1.1 martin .add_partition = gpt_add_part,
1581 1.1 martin .delete_all_partitions = gpt_delete_all_partitions,
1582 1.1 martin .delete_partition = gpt_delete_partition,
1583 1.1 martin .write_to_disk = gpt_write_to_disk,
1584 1.1 martin .free = gpt_free,
1585 1.1 martin };
1586