Home | History | Annotate | Line # | Download | only in sysinst
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