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