xref: /freebsd/usr.sbin/bsdinstall/partedit/gpart_ops.c (revision 7791ecf04b48a0c365b003447f479ec890115dfc)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2011 Nathan Whitehorn
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  *
28  * $FreeBSD$
29  */
30 
31 #include <sys/param.h>
32 #include <sys/stat.h>
33 #include <errno.h>
34 #include <fcntl.h>
35 #include <libutil.h>
36 #include <inttypes.h>
37 
38 #include <libgeom.h>
39 #include <dialog.h>
40 #include <dlg_keys.h>
41 
42 #include "partedit.h"
43 
44 #define GPART_FLAGS "x" /* Do not commit changes by default */
45 
46 static void
47 gpart_show_error(const char *title, const char *explanation, const char *errstr)
48 {
49 	char *errmsg;
50 	char message[512];
51 	int error;
52 
53 	if (explanation == NULL)
54 		explanation = "";
55 
56 	error = strtol(errstr, &errmsg, 0);
57 	if (errmsg != errstr) {
58 		while (errmsg[0] == ' ')
59 			errmsg++;
60 		if (errmsg[0] != '\0')
61 			sprintf(message, "%s%s. %s", explanation,
62 			    strerror(error), errmsg);
63 		else
64 			sprintf(message, "%s%s", explanation, strerror(error));
65 	} else {
66 		sprintf(message, "%s%s", explanation, errmsg);
67 	}
68 
69 	dialog_msgbox(title, message, 0, 0, TRUE);
70 }
71 
72 static int
73 scheme_supports_labels(const char *scheme)
74 {
75 	if (strcmp(scheme, "APM") == 0)
76 		return (1);
77 	if (strcmp(scheme, "GPT") == 0)
78 		return (1);
79 
80 	return (0);
81 }
82 
83 static void
84 newfs_command(const char *fstype, char *command, int use_default)
85 {
86 	if (strcmp(fstype, "freebsd-ufs") == 0) {
87 		int i;
88 		DIALOG_LISTITEM items[] = {
89 			{"UFS1", "UFS Version 1",
90 			    "Use version 1 of the UFS file system instead "
91 			    "of version 2 (not recommended)", 0 },
92 			{"SU", "Softupdates",
93 			    "Enable softupdates (default)", 1 },
94 			{"SUJ", "Softupdates journaling",
95 			    "Enable file system journaling (default - "
96 			    "turn off for SSDs)", 1 },
97 			{"TRIM", "Enable SSD TRIM support",
98 			    "Enable TRIM support, useful on solid-state drives",
99 			    0 },
100 		};
101 
102 		if (!use_default) {
103 			int choice;
104 			choice = dlg_checklist("UFS Options", "", 0, 0, 0,
105 			    nitems(items), items, NULL,
106 			    FLAG_CHECK, &i);
107 			if (choice == 1) /* Cancel */
108 				return;
109 		}
110 
111 		strcpy(command, "newfs ");
112 		for (i = 0; i < (int)nitems(items); i++) {
113 			if (items[i].state == 0)
114 				continue;
115 			if (strcmp(items[i].name, "UFS1") == 0)
116 				strcat(command, "-O1 ");
117 			else if (strcmp(items[i].name, "SU") == 0)
118 				strcat(command, "-U ");
119 			else if (strcmp(items[i].name, "SUJ") == 0)
120 				strcat(command, "-j ");
121 			else if (strcmp(items[i].name, "TRIM") == 0)
122 				strcat(command, "-t ");
123 		}
124 	} else if (strcmp(fstype, "freebsd-zfs") == 0) {
125 		int i;
126 		DIALOG_LISTITEM items[] = {
127 			{"fletcher4", "checksum algorithm: fletcher4",
128 			    "Use fletcher4 for data integrity checking. "
129 			    "(default)", 1 },
130 			{"fletcher2", "checksum algorithm: fletcher2",
131 			    "Use fletcher2 for data integrity checking. "
132 			    "(not recommended)", 0 },
133 			{"sha256", "checksum algorithm: sha256",
134 			    "Use sha256 for data integrity checking. "
135 			    "(not recommended)", 0 },
136 			{"atime", "Update atimes for files",
137 			    "Disable atime update", 0 },
138 		};
139 
140 		if (!use_default) {
141 			int choice;
142 			choice = dlg_checklist("ZFS Options", "", 0, 0, 0,
143 			    nitems(items), items, NULL,
144 			    FLAG_CHECK, &i);
145 			if (choice == 1) /* Cancel */
146 				return;
147 		}
148 
149 		strcpy(command, "zpool create -f -m none ");
150 		if (getenv("BSDINSTALL_TMPBOOT") != NULL) {
151 			char zfsboot_path[MAXPATHLEN];
152 			snprintf(zfsboot_path, sizeof(zfsboot_path), "%s/zfs",
153 			    getenv("BSDINSTALL_TMPBOOT"));
154 			mkdir(zfsboot_path, S_IRWXU | S_IRGRP | S_IXGRP |
155 			    S_IROTH | S_IXOTH);
156 			sprintf(command, "%s -o cachefile=%s/zpool.cache ",
157 			    command, zfsboot_path);
158 		}
159 		for (i = 0; i < (int)nitems(items); i++) {
160 			if (items[i].state == 0)
161 				continue;
162 			if (strcmp(items[i].name, "fletcher4") == 0)
163 				strcat(command, "-O checksum=fletcher4 ");
164 			else if (strcmp(items[i].name, "fletcher2") == 0)
165 				strcat(command, "-O checksum=fletcher2 ");
166 			else if (strcmp(items[i].name, "sha256") == 0)
167 				strcat(command, "-O checksum=sha256 ");
168 			else if (strcmp(items[i].name, "atime") == 0)
169 				strcat(command, "-O atime=off ");
170 		}
171 	} else if (strcmp(fstype, "fat32") == 0 || strcmp(fstype, "efi") == 0 ||
172 	     strcmp(fstype, "ms-basic-data") == 0) {
173 		int i;
174 		DIALOG_LISTITEM items[] = {
175 			{"FAT32", "FAT Type 32",
176 			    "Create a FAT32 filesystem (default)", 1 },
177 			{"FAT16", "FAT Type 16",
178 			    "Create a FAT16 filesystem", 0 },
179 			{"FAT12", "FAT Type 12",
180 			    "Create a FAT12 filesystem", 0 },
181 		};
182 
183 		if (!use_default) {
184 			int choice;
185 			choice = dlg_checklist("FAT Options", "", 0, 0, 0,
186 			    nitems(items), items, NULL,
187 			    FLAG_RADIO, &i);
188 			if (choice == 1) /* Cancel */
189 				return;
190 		}
191 
192 		strcpy(command, "newfs_msdos ");
193 		for (i = 0; i < (int)nitems(items); i++) {
194 			if (items[i].state == 0)
195 				continue;
196 			if (strcmp(items[i].name, "FAT32") == 0)
197 				strcat(command, "-F 32 -c 1");
198 			else if (strcmp(items[i].name, "FAT16") == 0)
199 				strcat(command, "-F 16 ");
200 			else if (strcmp(items[i].name, "FAT12") == 0)
201 				strcat(command, "-F 12 ");
202 		}
203 	} else {
204 		if (!use_default)
205 			dialog_msgbox("Error", "No configurable options exist "
206 			    "for this filesystem.", 0, 0, TRUE);
207 		command[0] = '\0';
208 	}
209 }
210 
211 const char *
212 choose_part_type(const char *def_scheme)
213 {
214 	int cancel, choice;
215 	const char *scheme = NULL;
216 
217 	DIALOG_LISTITEM items[] = {
218 		{"APM", "Apple Partition Map",
219 		    "Bootable on PowerPC Apple Hardware", 0 },
220 		{"BSD", "BSD Labels",
221 		    "Bootable on most x86 systems", 0 },
222 		{"GPT", "GUID Partition Table",
223 		    "Bootable on most x86 systems and EFI aware ARM64", 0 },
224 		{"MBR", "DOS Partitions",
225 		    "Bootable on most x86 systems", 0 },
226 	};
227 
228 parttypemenu:
229 	dialog_vars.default_item = __DECONST(char *, def_scheme);
230 	cancel = dlg_menu("Partition Scheme",
231 	    "Select a partition scheme for this volume:", 0, 0, 0,
232 	    nitems(items), items, &choice, NULL);
233 	dialog_vars.default_item = NULL;
234 
235 	if (cancel)
236 		return NULL;
237 
238 	if (!is_scheme_bootable(items[choice].name)) {
239 		char message[512];
240 		sprintf(message, "This partition scheme (%s) is not "
241 		    "bootable on this platform. Are you sure you want "
242 		    "to proceed?", items[choice].name);
243 		dialog_vars.defaultno = TRUE;
244 		cancel = dialog_yesno("Warning", message, 0, 0);
245 		dialog_vars.defaultno = FALSE;
246 		if (cancel) /* cancel */
247 			goto parttypemenu;
248 	}
249 
250 	scheme = items[choice].name;
251 
252 	return scheme;
253 }
254 
255 int
256 gpart_partition(const char *lg_name, const char *scheme)
257 {
258 	int cancel;
259 	struct gctl_req *r;
260 	const char *errstr;
261 
262 schememenu:
263 	if (scheme == NULL) {
264 		scheme = choose_part_type(default_scheme());
265 
266 		if (scheme == NULL)
267 			return (-1);
268 
269 		if (!is_scheme_bootable(scheme)) {
270 			char message[512];
271 			sprintf(message, "This partition scheme (%s) is not "
272 			    "bootable on this platform. Are you sure you want "
273 			    "to proceed?", scheme);
274 			dialog_vars.defaultno = TRUE;
275 			cancel = dialog_yesno("Warning", message, 0, 0);
276 			dialog_vars.defaultno = FALSE;
277 			if (cancel) { /* cancel */
278 				/* Reset scheme so user can choose another */
279 				scheme = NULL;
280 				goto schememenu;
281 			}
282 		}
283 	}
284 
285 	r = gctl_get_handle();
286 	gctl_ro_param(r, "class", -1, "PART");
287 	gctl_ro_param(r, "arg0", -1, lg_name);
288 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
289 	gctl_ro_param(r, "scheme", -1, scheme);
290 	gctl_ro_param(r, "verb", -1, "create");
291 
292 	errstr = gctl_issue(r);
293 	if (errstr != NULL && errstr[0] != '\0') {
294 		gpart_show_error("Error", NULL, errstr);
295 		gctl_free(r);
296 		scheme = NULL;
297 		goto schememenu;
298 	}
299 	gctl_free(r);
300 
301 	if (bootcode_path(scheme) != NULL)
302 		get_part_metadata(lg_name, 1)->bootcode = 1;
303 	return (0);
304 }
305 
306 static void
307 gpart_activate(struct gprovider *pp)
308 {
309 	struct gconfig *gc;
310 	struct gctl_req *r;
311 	const char *errstr, *scheme;
312 	const char *attribute = NULL;
313 	intmax_t idx;
314 
315 	/*
316 	 * Some partition schemes need this partition to be marked 'active'
317 	 * for it to be bootable.
318 	 */
319 	LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) {
320 		if (strcmp(gc->lg_name, "scheme") == 0) {
321 			scheme = gc->lg_val;
322 			break;
323 		}
324 	}
325 
326 	if (strcmp(scheme, "MBR") == 0 || strcmp(scheme, "EBR") == 0)
327 		attribute = "active";
328 	else
329 		return;
330 
331 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
332 		if (strcmp(gc->lg_name, "index") == 0) {
333 			idx = atoi(gc->lg_val);
334 			break;
335 		}
336 	}
337 
338 	r = gctl_get_handle();
339 	gctl_ro_param(r, "class", -1, "PART");
340 	gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name);
341 	gctl_ro_param(r, "verb", -1, "set");
342 	gctl_ro_param(r, "attrib", -1, attribute);
343 	gctl_ro_param(r, "index", sizeof(idx), &idx);
344 
345 	errstr = gctl_issue(r);
346 	if (errstr != NULL && errstr[0] != '\0')
347 		gpart_show_error("Error", "Error marking partition active:",
348 		    errstr);
349 	gctl_free(r);
350 }
351 
352 void
353 gpart_set_root(const char *lg_name, const char *attribute)
354 {
355 	struct gctl_req *r;
356 	const char *errstr;
357 
358 	r = gctl_get_handle();
359 	gctl_ro_param(r, "class", -1, "PART");
360 	gctl_ro_param(r, "arg0", -1, lg_name);
361 	gctl_ro_param(r, "flags", -1, "C");
362 	gctl_ro_param(r, "verb", -1, "set");
363 	gctl_ro_param(r, "attrib", -1, attribute);
364 
365 	errstr = gctl_issue(r);
366 	if (errstr != NULL && errstr[0] != '\0')
367 		gpart_show_error("Error", "Error setting parameter on disk:",
368 		    errstr);
369 	gctl_free(r);
370 }
371 
372 static void
373 gpart_bootcode(struct ggeom *gp)
374 {
375 	const char *bootcode;
376 	struct gconfig *gc;
377 	struct gctl_req *r;
378 	const char *errstr, *scheme;
379 	uint8_t *boot;
380 	size_t bootsize, bytes;
381 	int bootfd;
382 
383 	/*
384 	 * Write default bootcode to the newly partitioned disk, if that
385 	 * applies on this platform.
386 	 */
387 	LIST_FOREACH(gc, &gp->lg_config, lg_config) {
388 		if (strcmp(gc->lg_name, "scheme") == 0) {
389 			scheme = gc->lg_val;
390 			break;
391 		}
392 	}
393 
394 	bootcode = bootcode_path(scheme);
395 	if (bootcode == NULL)
396 		return;
397 
398 	bootfd = open(bootcode, O_RDONLY);
399 	if (bootfd < 0) {
400 		dialog_msgbox("Bootcode Error", strerror(errno), 0, 0,
401 		    TRUE);
402 		return;
403 	}
404 
405 	bootsize = lseek(bootfd, 0, SEEK_END);
406 	boot = malloc(bootsize);
407 	lseek(bootfd, 0, SEEK_SET);
408 	bytes = 0;
409 	while (bytes < bootsize)
410 		bytes += read(bootfd, boot + bytes, bootsize - bytes);
411 	close(bootfd);
412 
413 	r = gctl_get_handle();
414 	gctl_ro_param(r, "class", -1, "PART");
415 	gctl_ro_param(r, "arg0", -1, gp->lg_name);
416 	gctl_ro_param(r, "verb", -1, "bootcode");
417 	gctl_ro_param(r, "bootcode", bootsize, boot);
418 
419 	errstr = gctl_issue(r);
420 	if (errstr != NULL && errstr[0] != '\0')
421 		gpart_show_error("Bootcode Error", NULL, errstr);
422 	gctl_free(r);
423 	free(boot);
424 }
425 
426 static void
427 gpart_partcode(struct gprovider *pp, const char *fstype)
428 {
429 	struct gconfig *gc;
430 	const char *scheme;
431 	const char *indexstr;
432 	char message[255], command[255];
433 
434 	LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) {
435 		if (strcmp(gc->lg_name, "scheme") == 0) {
436 			scheme = gc->lg_val;
437 			break;
438 		}
439 	}
440 
441 	/* Make sure this partition scheme needs partcode on this platform */
442 	if (partcode_path(scheme, fstype) == NULL)
443 		return;
444 
445 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
446 		if (strcmp(gc->lg_name, "index") == 0) {
447 			indexstr = gc->lg_val;
448 			break;
449 		}
450 	}
451 
452 	/* Shell out to gpart for partcode for now */
453 	sprintf(command, "gpart bootcode -p %s -i %s %s",
454 	    partcode_path(scheme, fstype), indexstr, pp->lg_geom->lg_name);
455 	if (system(command) != 0) {
456 		sprintf(message, "Error installing partcode on partition %s",
457 		    pp->lg_name);
458 		dialog_msgbox("Error", message, 0, 0, TRUE);
459 	}
460 }
461 
462 void
463 gpart_destroy(struct ggeom *lg_geom)
464 {
465 	struct gctl_req *r;
466 	struct gprovider *pp;
467 	const char *errstr;
468 	int force = 1;
469 
470 	/* Delete all child metadata */
471 	LIST_FOREACH(pp, &lg_geom->lg_provider, lg_provider)
472 		gpart_delete(pp);
473 
474 	/* Revert any local changes to get this geom into a pristine state */
475 	r = gctl_get_handle();
476 	gctl_ro_param(r, "class", -1, "PART");
477 	gctl_ro_param(r, "arg0", -1, lg_geom->lg_name);
478 	gctl_ro_param(r, "verb", -1, "undo");
479 	gctl_issue(r); /* Ignore errors -- these are non-fatal */
480 	gctl_free(r);
481 
482 	/* Now destroy the geom itself */
483 	r = gctl_get_handle();
484 	gctl_ro_param(r, "class", -1, "PART");
485 	gctl_ro_param(r, "arg0", -1, lg_geom->lg_name);
486 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
487 	gctl_ro_param(r, "force", sizeof(force), &force);
488 	gctl_ro_param(r, "verb", -1, "destroy");
489 	errstr = gctl_issue(r);
490 	if (errstr != NULL && errstr[0] != '\0') {
491 		/*
492 		 * Check if we reverted away the existence of the geom
493 		 * altogether. Show all other errors to the user.
494 		 */
495 		if (strtol(errstr, NULL, 0) != EINVAL)
496 			gpart_show_error("Error", NULL, errstr);
497 	}
498 	gctl_free(r);
499 
500 	/* And any metadata associated with the partition scheme itself */
501 	delete_part_metadata(lg_geom->lg_name);
502 }
503 
504 void
505 gpart_edit(struct gprovider *pp)
506 {
507 	struct gctl_req *r;
508 	struct gconfig *gc;
509 	struct gconsumer *cp;
510 	struct ggeom *geom;
511 	const char *errstr, *oldtype, *scheme;
512 	struct partition_metadata *md;
513 	char sizestr[32];
514 	char newfs[255];
515 	intmax_t idx;
516 	int hadlabel, choice, junk, nitems;
517 	unsigned i;
518 
519 	DIALOG_FORMITEM items[] = {
520 		{0, "Type:", 5, 0, 0, FALSE, "", 11, 0, 12, 15, 0,
521 		    FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, "
522 		    "freebsd-swap)", FALSE},
523 		{0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 0, 0,
524 		    FALSE, "Partition size. Append K, M, G for kilobytes, "
525 		    "megabytes or gigabytes.", FALSE},
526 		{0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0,
527 		    FALSE, "Path at which to mount this partition (leave blank "
528 		    "for swap, set to / for root filesystem)", FALSE},
529 		{0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE,
530 		    "Partition name. Not all partition schemes support this.",
531 		    FALSE},
532 	};
533 
534 	/*
535 	 * Find the PART geom we are manipulating. This may be a consumer of
536 	 * this provider, or its parent. Check the consumer case first.
537 	 */
538 	geom = NULL;
539 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
540 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
541 			/* Check for zombie geoms, treating them as blank */
542 			scheme = NULL;
543 			LIST_FOREACH(gc, &cp->lg_geom->lg_config, lg_config) {
544 				if (strcmp(gc->lg_name, "scheme") == 0) {
545 					scheme = gc->lg_val;
546 					break;
547 				}
548 			}
549 			if (scheme == NULL || strcmp(scheme, "(none)") == 0) {
550 				gpart_partition(cp->lg_geom->lg_name, NULL);
551 				return;
552 			}
553 
554 			/* If this is a nested partition, edit as usual */
555 			if (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
556 				break;
557 
558 			/* Destroy the geom and all sub-partitions */
559 			gpart_destroy(cp->lg_geom);
560 
561 			/* Now re-partition and return */
562 			gpart_partition(cp->lg_geom->lg_name, NULL);
563 			return;
564 		}
565 
566 	if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
567 		geom = pp->lg_geom;
568 
569 	if (geom == NULL) {
570 		/* Disk not partitioned, so partition it */
571 		gpart_partition(pp->lg_name, NULL);
572 		return;
573 	}
574 
575 	LIST_FOREACH(gc, &geom->lg_config, lg_config) {
576 		if (strcmp(gc->lg_name, "scheme") == 0) {
577 			scheme = gc->lg_val;
578 			break;
579 		}
580 	}
581 
582 	nitems = scheme_supports_labels(scheme) ? 4 : 3;
583 
584 	/* Edit editable parameters of a partition */
585 	hadlabel = 0;
586 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
587 		if (strcmp(gc->lg_name, "type") == 0) {
588 			oldtype = gc->lg_val;
589 			items[0].text = gc->lg_val;
590 		}
591 		if (strcmp(gc->lg_name, "label") == 0 && gc->lg_val != NULL) {
592 			hadlabel = 1;
593 			items[3].text = gc->lg_val;
594 		}
595 		if (strcmp(gc->lg_name, "index") == 0)
596 			idx = atoi(gc->lg_val);
597 	}
598 
599 	TAILQ_FOREACH(md, &part_metadata, metadata) {
600 		if (md->name != NULL && strcmp(md->name, pp->lg_name) == 0) {
601 			if (md->fstab != NULL)
602 				items[2].text = md->fstab->fs_file;
603 			break;
604 		}
605 	}
606 
607 	humanize_number(sizestr, 7, pp->lg_mediasize, "B", HN_AUTOSCALE,
608 	    HN_NOSPACE | HN_DECIMAL);
609 	items[1].text = sizestr;
610 
611 editpart:
612 	choice = dlg_form("Edit Partition", "", 0, 0, 0, nitems, items, &junk);
613 
614 	if (choice) /* Cancel pressed */
615 		goto endedit;
616 
617 	/* If this is the root partition, check that this fs is bootable */
618 	if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme,
619 	    items[0].text)) {
620 		char message[512];
621 		sprintf(message, "This file system (%s) is not bootable "
622 		    "on this system. Are you sure you want to proceed?",
623 		    items[0].text);
624 		dialog_vars.defaultno = TRUE;
625 		choice = dialog_yesno("Warning", message, 0, 0);
626 		dialog_vars.defaultno = FALSE;
627 		if (choice == 1) /* cancel */
628 			goto editpart;
629 	}
630 
631 	/* Check if the label has a / in it */
632 	if (strchr(items[3].text, '/') != NULL) {
633 		dialog_msgbox("Error", "Label contains a /, which is not an "
634 		    "allowed character.", 0, 0, TRUE);
635 		goto editpart;
636 	}
637 
638 	r = gctl_get_handle();
639 	gctl_ro_param(r, "class", -1, "PART");
640 	gctl_ro_param(r, "arg0", -1, geom->lg_name);
641 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
642 	gctl_ro_param(r, "verb", -1, "modify");
643 	gctl_ro_param(r, "index", sizeof(idx), &idx);
644 	if (hadlabel || items[3].text[0] != '\0')
645 		gctl_ro_param(r, "label", -1, items[3].text);
646 	gctl_ro_param(r, "type", -1, items[0].text);
647 	errstr = gctl_issue(r);
648 	if (errstr != NULL && errstr[0] != '\0') {
649 		gpart_show_error("Error", NULL, errstr);
650 		gctl_free(r);
651 		goto editpart;
652 	}
653 	gctl_free(r);
654 
655 	newfs_command(items[0].text, newfs, 1);
656 	set_default_part_metadata(pp->lg_name, scheme, items[0].text,
657 	    items[2].text, (strcmp(oldtype, items[0].text) != 0) ?
658 	    newfs : NULL);
659 
660 endedit:
661 	if (strcmp(oldtype, items[0].text) != 0 && cp != NULL)
662 		gpart_destroy(cp->lg_geom);
663 	if (strcmp(oldtype, items[0].text) != 0 && strcmp(items[0].text,
664 	    "freebsd") == 0)
665 		gpart_partition(pp->lg_name, "BSD");
666 
667 	for (i = 0; i < nitems(items); i++)
668 		if (items[i].text_free)
669 			free(items[i].text);
670 }
671 
672 void
673 set_default_part_metadata(const char *name, const char *scheme,
674     const char *type, const char *mountpoint, const char *newfs)
675 {
676 	struct partition_metadata *md;
677 	char *zpool_name = NULL;
678 	const char *default_bootmount = NULL;
679 	int i;
680 
681 	/* Set part metadata */
682 	md = get_part_metadata(name, 1);
683 
684 	if (newfs) {
685 		if (md->newfs != NULL) {
686 			free(md->newfs);
687 			md->newfs = NULL;
688 		}
689 
690 		if (newfs != NULL && newfs[0] != '\0') {
691 			md->newfs = malloc(strlen(newfs) + strlen(" /dev/") +
692 			    strlen(mountpoint) + 5 + strlen(name) + 1);
693 			if (strcmp("freebsd-zfs", type) == 0) {
694 				zpool_name = strdup((strlen(mountpoint) == 1) ?
695 				    "root" : &mountpoint[1]);
696 				for (i = 0; zpool_name[i] != 0; i++)
697 					if (!isalnum(zpool_name[i]))
698 						zpool_name[i] = '_';
699 				sprintf(md->newfs, "%s %s /dev/%s", newfs,
700 				    zpool_name, name);
701 			} else {
702 				sprintf(md->newfs, "%s /dev/%s", newfs, name);
703 			}
704 		}
705 	}
706 
707 	if (strcmp(type, "freebsd-swap") == 0)
708 		mountpoint = "none";
709 	if (strcmp(type, bootpart_type(scheme, &default_bootmount)) == 0) {
710 		if (default_bootmount == NULL) {
711 
712 			int fd = openat(tmpdfd, "bsdinstall-esps",
713 			    O_CREAT | O_WRONLY | O_APPEND, 0600);
714 			if (fd > 0) {
715 				write(fd, md->name, strlen(md->name));
716 				write(fd, "\n", 1);
717 				close(fd);
718 			}
719 
720 			md->bootcode = 1;
721 		}
722 		else if (mountpoint == NULL || strlen(mountpoint) == 0)
723 			mountpoint = default_bootmount;
724 	}
725 
726 	if (mountpoint == NULL || mountpoint[0] == '\0') {
727 		if (md->fstab != NULL) {
728 			free(md->fstab->fs_spec);
729 			free(md->fstab->fs_file);
730 			free(md->fstab->fs_vfstype);
731 			free(md->fstab->fs_mntops);
732 			free(md->fstab->fs_type);
733 			free(md->fstab);
734 			md->fstab = NULL;
735 		}
736 	} else {
737 		if (md->fstab == NULL) {
738 			md->fstab = malloc(sizeof(struct fstab));
739 		} else {
740 			free(md->fstab->fs_spec);
741 			free(md->fstab->fs_file);
742 			free(md->fstab->fs_vfstype);
743 			free(md->fstab->fs_mntops);
744 			free(md->fstab->fs_type);
745 		}
746 		if (strcmp("freebsd-zfs", type) == 0) {
747 			md->fstab->fs_spec = strdup(zpool_name);
748 		} else {
749 			md->fstab->fs_spec = malloc(strlen(name) +
750 			    strlen("/dev/") + 1);
751 			sprintf(md->fstab->fs_spec, "/dev/%s", name);
752 		}
753 		md->fstab->fs_file = strdup(mountpoint);
754 		/* Get VFS from text after freebsd-, if possible */
755 		if (strncmp("freebsd-", type, 8) == 0)
756 			md->fstab->fs_vfstype = strdup(&type[8]);
757 		else if (strcmp("fat32", type) == 0 || strcmp("efi", type) == 0
758 	     	    || strcmp("ms-basic-data", type) == 0)
759 			md->fstab->fs_vfstype = strdup("msdosfs");
760 		else
761 			md->fstab->fs_vfstype = strdup(type); /* Guess */
762 		if (strcmp(type, "freebsd-swap") == 0) {
763 			md->fstab->fs_type = strdup(FSTAB_SW);
764 			md->fstab->fs_freq = 0;
765 			md->fstab->fs_passno = 0;
766 		} else if (strcmp(type, "freebsd-zfs") == 0) {
767 			md->fstab->fs_type = strdup(FSTAB_RW);
768 			md->fstab->fs_freq = 0;
769 			md->fstab->fs_passno = 0;
770 		} else {
771 			md->fstab->fs_type = strdup(FSTAB_RW);
772 			if (strcmp(mountpoint, "/") == 0) {
773 				md->fstab->fs_freq = 1;
774 				md->fstab->fs_passno = 1;
775 			} else {
776 				md->fstab->fs_freq = 2;
777 				md->fstab->fs_passno = 2;
778 			}
779 		}
780 		md->fstab->fs_mntops = strdup(md->fstab->fs_type);
781 	}
782 
783 	if (zpool_name != NULL)
784 		free(zpool_name);
785 }
786 
787 static
788 int part_compare(const void *xa, const void *xb)
789 {
790 	struct gprovider **a = (struct gprovider **)xa;
791 	struct gprovider **b = (struct gprovider **)xb;
792 	intmax_t astart, bstart;
793 	struct gconfig *gc;
794 
795 	astart = bstart = 0;
796 	LIST_FOREACH(gc, &(*a)->lg_config, lg_config)
797 		if (strcmp(gc->lg_name, "start") == 0) {
798 			astart = strtoimax(gc->lg_val, NULL, 0);
799 			break;
800 		}
801 	LIST_FOREACH(gc, &(*b)->lg_config, lg_config)
802 		if (strcmp(gc->lg_name, "start") == 0) {
803 			bstart = strtoimax(gc->lg_val, NULL, 0);
804 			break;
805 		}
806 
807 	if (astart < bstart)
808 		return -1;
809 	else if (astart > bstart)
810 		return 1;
811 	else
812 		return 0;
813 }
814 
815 intmax_t
816 gpart_max_free(struct ggeom *geom, intmax_t *npartstart)
817 {
818 	struct gconfig *gc;
819 	struct gprovider *pp, **providers;
820 	intmax_t sectorsize, stripesize, offset;
821 	intmax_t lastend;
822 	intmax_t start, end;
823 	intmax_t maxsize, maxstart;
824 	intmax_t partstart, partend;
825 	int i, nparts;
826 
827 	/* Now get the maximum free size and free start */
828 	start = end = 0;
829 	LIST_FOREACH(gc, &geom->lg_config, lg_config) {
830 		if (strcmp(gc->lg_name, "first") == 0)
831 			start = strtoimax(gc->lg_val, NULL, 0);
832 		if (strcmp(gc->lg_name, "last") == 0)
833 			end = strtoimax(gc->lg_val, NULL, 0);
834 	}
835 
836 	i = nparts = 0;
837 	LIST_FOREACH(pp, &geom->lg_provider, lg_provider)
838 		nparts++;
839 	providers = calloc(nparts, sizeof(providers[0]));
840 	LIST_FOREACH(pp, &geom->lg_provider, lg_provider)
841 		providers[i++] = pp;
842 	qsort(providers, nparts, sizeof(providers[0]), part_compare);
843 
844 	lastend = start - 1;
845 	maxsize = 0;
846 	for (i = 0; i < nparts; i++) {
847 		pp = providers[i];
848 
849 		LIST_FOREACH(gc, &pp->lg_config, lg_config) {
850 			if (strcmp(gc->lg_name, "start") == 0)
851 				partstart = strtoimax(gc->lg_val, NULL, 0);
852 			if (strcmp(gc->lg_name, "end") == 0)
853 				partend = strtoimax(gc->lg_val, NULL, 0);
854 		}
855 
856 		if (partstart - lastend > maxsize) {
857 			maxsize = partstart - lastend - 1;
858 			maxstart = lastend + 1;
859 		}
860 
861 		lastend = partend;
862 	}
863 
864 	if (end - lastend > maxsize) {
865 		maxsize = end - lastend;
866 		maxstart = lastend + 1;
867 	}
868 
869 	pp = LIST_FIRST(&geom->lg_consumer)->lg_provider;
870 
871 	/*
872 	 * Round the start and size of the largest available space up to
873 	 * the nearest multiple of the adjusted stripe size.
874 	 *
875 	 * The adjusted stripe size is the least common multiple of the
876 	 * actual stripe size, or the sector size if no stripe size was
877 	 * reported, and 4096.  The reason for this is that contemporary
878 	 * disks often have 4096-byte physical sectors but report 512
879 	 * bytes instead for compatibility with older / broken operating
880 	 * systems and BIOSes.  For the same reasons, virtualized storage
881 	 * may also report a 512-byte stripe size, or none at all.
882 	 */
883 	sectorsize = pp->lg_sectorsize;
884 	if ((stripesize = pp->lg_stripesize) == 0)
885 		stripesize = sectorsize;
886 	while (stripesize % 4096 != 0)
887 		stripesize *= 2;
888 	if ((offset = maxstart * sectorsize % stripesize) != 0) {
889 		offset = (stripesize - offset) / sectorsize;
890 		maxstart += offset;
891 		maxsize -= offset;
892 	}
893 
894 	if (npartstart != NULL)
895 		*npartstart = maxstart;
896 
897 	return (maxsize);
898 }
899 
900 static size_t
901 add_boot_partition(struct ggeom *geom, struct gprovider *pp,
902     const char *scheme, int interactive)
903 {
904 	struct gconfig *gc;
905 	struct gprovider *ppi;
906 	int choice;
907 
908 	/* Check for existing freebsd-boot partition */
909 	LIST_FOREACH(ppi, &geom->lg_provider, lg_provider) {
910 		struct partition_metadata *md;
911 		const char *bootmount = NULL;
912 
913 		LIST_FOREACH(gc, &ppi->lg_config, lg_config)
914 			if (strcmp(gc->lg_name, "type") == 0)
915 				break;
916 		if (gc == NULL)
917 			continue;
918 		if (strcmp(gc->lg_val, bootpart_type(scheme, &bootmount)) != 0)
919 			continue;
920 
921 		/*
922 		 * If the boot partition is not mountable and needs partcode,
923 		 * but doesn't have it, it doesn't satisfy our requirements.
924 		 */
925 		md = get_part_metadata(ppi->lg_name, 0);
926 		if (bootmount == NULL && (md == NULL || !md->bootcode))
927 			continue;
928 
929 		/* If it is mountable, but mounted somewhere else, remount */
930 		if (bootmount != NULL && md != NULL && md->fstab != NULL
931 		    && strlen(md->fstab->fs_file) > 0
932 		    && strcmp(md->fstab->fs_file, bootmount) != 0)
933 			continue;
934 
935 		/* If it is mountable, but mountpoint is not set, mount it */
936 		if (bootmount != NULL && md == NULL)
937 			set_default_part_metadata(ppi->lg_name, scheme,
938 			    gc->lg_val, bootmount, NULL);
939 
940 		/* Looks good at this point, no added data needed */
941 		return (0);
942 	}
943 
944 	if (interactive)
945 		choice = dialog_yesno("Boot Partition",
946 		    "This partition scheme requires a boot partition "
947 		    "for the disk to be bootable. Would you like to "
948 		    "make one now?", 0, 0);
949 	else
950 		choice = 0;
951 
952 	if (choice == 0) { /* yes */
953 		struct partition_metadata *md;
954 		const char *bootmount = NULL;
955 		char *bootpartname = NULL;
956 		char sizestr[7];
957 
958 		humanize_number(sizestr, 7,
959 		    bootpart_size(scheme), "B", HN_AUTOSCALE,
960 		    HN_NOSPACE | HN_DECIMAL);
961 
962 		gpart_create(pp, bootpart_type(scheme, &bootmount),
963 		    sizestr, bootmount, &bootpartname, 0);
964 
965 		if (bootpartname == NULL) /* Error reported to user already */
966 			return 0;
967 
968 		/* If the part is not mountable, make sure newfs isn't set */
969 		if (bootmount == NULL) {
970 			md = get_part_metadata(bootpartname, 0);
971 			if (md != NULL && md->newfs != NULL) {
972 				free(md->newfs);
973 				md->newfs = NULL;
974 			}
975 		}
976 
977 		free(bootpartname);
978 
979 		return (bootpart_size(scheme));
980 	}
981 
982 	return (0);
983 }
984 
985 void
986 gpart_create(struct gprovider *pp, const char *default_type,
987     const char *default_size, const char *default_mountpoint,
988     char **partname, int interactive)
989 {
990 	struct gctl_req *r;
991 	struct gconfig *gc;
992 	struct gconsumer *cp;
993 	struct ggeom *geom;
994 	const char *errstr, *scheme;
995 	char sizestr[32], startstr[32], output[64], *newpartname;
996 	char newfs[255], options_fstype[64];
997 	intmax_t maxsize, size, sector, firstfree, stripe;
998 	uint64_t bytes;
999 	int nitems, choice, junk;
1000 	unsigned i;
1001 
1002 	DIALOG_FORMITEM items[] = {
1003 		{0, "Type:", 5, 0, 0, FALSE, "freebsd-ufs", 11, 0, 12, 15, 0,
1004 		    FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, "
1005 		    "freebsd-swap)", FALSE},
1006 		{0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 15, 0,
1007 		    FALSE, "Partition size. Append K, M, G for kilobytes, "
1008 		    "megabytes or gigabytes.", FALSE},
1009 		{0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0,
1010 		    FALSE, "Path at which to mount partition (blank for "
1011 		    "swap, / for root filesystem)", FALSE},
1012 		{0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE,
1013 		    "Partition name. Not all partition schemes support this.",
1014 		    FALSE},
1015 	};
1016 
1017 	if (partname != NULL)
1018 		*partname = NULL;
1019 
1020 	/* Record sector and stripe sizes */
1021 	sector = pp->lg_sectorsize;
1022 	stripe = pp->lg_stripesize;
1023 
1024 	/*
1025 	 * Find the PART geom we are manipulating. This may be a consumer of
1026 	 * this provider, or its parent. Check the consumer case first.
1027 	 */
1028 	geom = NULL;
1029 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
1030 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
1031 			geom = cp->lg_geom;
1032 			break;
1033 		}
1034 
1035 	if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
1036 		geom = pp->lg_geom;
1037 
1038 	/* Now get the partition scheme */
1039 	scheme = NULL;
1040 	if (geom != NULL) {
1041 		LIST_FOREACH(gc, &geom->lg_config, lg_config)
1042 			if (strcmp(gc->lg_name, "scheme") == 0)
1043 				scheme = gc->lg_val;
1044 	}
1045 
1046 	if (geom == NULL || scheme == NULL || strcmp(scheme, "(none)") == 0) {
1047 		if (gpart_partition(pp->lg_name, NULL) == 0)
1048 			dialog_msgbox("",
1049 			    "The partition table has been successfully created."
1050 			    " Please press Create again to create partitions.",
1051 			    0, 0, TRUE);
1052 
1053 		return;
1054 	}
1055 
1056 	/*
1057 	 * If we still don't have a geom, either the user has
1058 	 * canceled partitioning or there has been an error which has already
1059 	 * been displayed, so bail.
1060 	 */
1061 	if (geom == NULL)
1062 		return;
1063 
1064 	maxsize = size = gpart_max_free(geom, &firstfree);
1065 	if (size <= 0) {
1066 		dialog_msgbox("Error", "No free space left on device.", 0, 0,
1067 		    TRUE);
1068 		return;
1069 	}
1070 
1071 	humanize_number(sizestr, 7, size*sector, "B", HN_AUTOSCALE,
1072 	    HN_NOSPACE | HN_DECIMAL);
1073 	items[1].text = sizestr;
1074 
1075 	/* Special-case the MBR default type for nested partitions */
1076 	if (strcmp(scheme, "MBR") == 0) {
1077 		items[0].text = "freebsd";
1078 		items[0].help = "Filesystem type (e.g. freebsd, fat32)";
1079 	}
1080 
1081 	nitems = scheme_supports_labels(scheme) ? 4 : 3;
1082 
1083 	if (default_type != NULL)
1084 		items[0].text = (char *)default_type;
1085 	if (default_size != NULL)
1086 		items[1].text = (char *)default_size;
1087 	if (default_mountpoint != NULL)
1088 		items[2].text = (char *)default_mountpoint;
1089 
1090 	/* Default options */
1091 	strncpy(options_fstype, items[0].text,
1092 	    sizeof(options_fstype));
1093 	newfs_command(options_fstype, newfs, 1);
1094 addpartform:
1095 	if (interactive) {
1096 		dialog_vars.extra_label = "Options";
1097 		dialog_vars.extra_button = TRUE;
1098 		choice = dlg_form("Add Partition", "", 0, 0, 0, nitems,
1099 		    items, &junk);
1100 		dialog_vars.extra_button = FALSE;
1101 		switch (choice) {
1102 		case 0: /* OK */
1103 			break;
1104 		case 1: /* Cancel */
1105 			return;
1106 		case 3: /* Options */
1107 			strncpy(options_fstype, items[0].text,
1108 			    sizeof(options_fstype));
1109 			newfs_command(options_fstype, newfs, 0);
1110 			goto addpartform;
1111 		}
1112 	}
1113 
1114 	/*
1115 	 * If the user changed the fs type after specifying options, undo
1116 	 * their choices in favor of the new filesystem's defaults.
1117 	 */
1118 	if (strcmp(options_fstype, items[0].text) != 0) {
1119 		strncpy(options_fstype, items[0].text, sizeof(options_fstype));
1120 		newfs_command(options_fstype, newfs, 1);
1121 	}
1122 
1123 	size = maxsize;
1124 	if (strlen(items[1].text) > 0) {
1125 		if (expand_number(items[1].text, &bytes) != 0) {
1126 			char error[512];
1127 
1128 			sprintf(error, "Invalid size: %s\n", strerror(errno));
1129 			dialog_msgbox("Error", error, 0, 0, TRUE);
1130 			goto addpartform;
1131 		}
1132 		size = MIN((intmax_t)(bytes/sector), maxsize);
1133 	}
1134 
1135 	/* Check if the label has a / in it */
1136 	if (strchr(items[3].text, '/') != NULL) {
1137 		dialog_msgbox("Error", "Label contains a /, which is not an "
1138 		    "allowed character.", 0, 0, TRUE);
1139 		goto addpartform;
1140 	}
1141 
1142 	/* Warn if no mountpoint set */
1143 	if (strcmp(items[0].text, "freebsd-ufs") == 0 &&
1144 	    items[2].text[0] != '/') {
1145 		choice = 0;
1146 		if (interactive) {
1147 			dialog_vars.defaultno = TRUE;
1148 			choice = dialog_yesno("Warning",
1149 			    "This partition does not have a valid mountpoint "
1150 			    "(for the partition from which you intend to boot the "
1151 			    "operating system, the mountpoint should be /). Are you "
1152 			    "sure you want to continue?"
1153 			, 0, 0);
1154 			dialog_vars.defaultno = FALSE;
1155 		}
1156 		if (choice == 1) /* cancel */
1157 			goto addpartform;
1158 	}
1159 
1160 	/*
1161 	 * Error if this scheme needs nested partitions, this is one, and
1162 	 * a mountpoint was set.
1163 	 */
1164 	if (strcmp(items[0].text, "freebsd") == 0 &&
1165 	    strlen(items[2].text) > 0) {
1166 		dialog_msgbox("Error", "Partitions of type \"freebsd\" are "
1167 		    "nested BSD-type partition schemes and cannot have "
1168 		    "mountpoints. After creating one, select it and press "
1169 		    "Create again to add the actual file systems.", 0, 0, TRUE);
1170 		goto addpartform;
1171 	}
1172 
1173 	/* If this is the root partition, check that this scheme is bootable */
1174 	if (strcmp(items[2].text, "/") == 0 && !is_scheme_bootable(scheme)) {
1175 		char message[512];
1176 		sprintf(message, "This partition scheme (%s) is not bootable "
1177 		    "on this platform. Are you sure you want to proceed?",
1178 		    scheme);
1179 		dialog_vars.defaultno = TRUE;
1180 		choice = dialog_yesno("Warning", message, 0, 0);
1181 		dialog_vars.defaultno = FALSE;
1182 		if (choice == 1) /* cancel */
1183 			goto addpartform;
1184 	}
1185 
1186 	/* If this is the root partition, check that this fs is bootable */
1187 	if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme,
1188 	    items[0].text)) {
1189 		char message[512];
1190 		sprintf(message, "This file system (%s) is not bootable "
1191 		    "on this system. Are you sure you want to proceed?",
1192 		    items[0].text);
1193 		dialog_vars.defaultno = TRUE;
1194 		choice = dialog_yesno("Warning", message, 0, 0);
1195 		dialog_vars.defaultno = FALSE;
1196 		if (choice == 1) /* cancel */
1197 			goto addpartform;
1198 	}
1199 
1200 	/*
1201 	 * If this is the root partition, and we need a boot partition, ask
1202 	 * the user to add one.
1203 	 */
1204 
1205 	if ((strcmp(items[0].text, "freebsd") == 0 ||
1206 	    strcmp(items[2].text, "/") == 0) && bootpart_size(scheme) > 0) {
1207 		size_t bytes = add_boot_partition(geom, pp, scheme,
1208 		    interactive);
1209 
1210 		/* Now adjust the part we are really adding forward */
1211 		if (bytes > 0) {
1212 			firstfree += bytes / sector;
1213 			size -= (bytes + stripe)/sector;
1214 			if (stripe > 0 && (firstfree*sector % stripe) != 0)
1215 				firstfree += (stripe - ((firstfree*sector) %
1216 				    stripe)) / sector;
1217 		}
1218 	}
1219 
1220 	r = gctl_get_handle();
1221 	gctl_ro_param(r, "class", -1, "PART");
1222 	gctl_ro_param(r, "arg0", -1, geom->lg_name);
1223 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
1224 	gctl_ro_param(r, "verb", -1, "add");
1225 
1226 	gctl_ro_param(r, "type", -1, items[0].text);
1227 	snprintf(sizestr, sizeof(sizestr), "%jd", size);
1228 	gctl_ro_param(r, "size", -1, sizestr);
1229 	snprintf(startstr, sizeof(startstr), "%jd", firstfree);
1230 	gctl_ro_param(r, "start", -1, startstr);
1231 	if (items[3].text[0] != '\0')
1232 		gctl_ro_param(r, "label", -1, items[3].text);
1233 	gctl_rw_param(r, "output", sizeof(output), output);
1234 	errstr = gctl_issue(r);
1235 	if (errstr != NULL && errstr[0] != '\0') {
1236 		gpart_show_error("Error", NULL, errstr);
1237 		gctl_free(r);
1238 		goto addpartform;
1239 	}
1240 	newpartname = strtok(output, " ");
1241 	gctl_free(r);
1242 
1243 	/*
1244 	 * Try to destroy any geom that gpart picked up already here from
1245 	 * dirty blocks.
1246 	 */
1247 	r = gctl_get_handle();
1248 	gctl_ro_param(r, "class", -1, "PART");
1249 	gctl_ro_param(r, "arg0", -1, newpartname);
1250 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
1251 	junk = 1;
1252 	gctl_ro_param(r, "force", sizeof(junk), &junk);
1253 	gctl_ro_param(r, "verb", -1, "destroy");
1254 	gctl_issue(r); /* Error usually expected and non-fatal */
1255 	gctl_free(r);
1256 
1257 
1258 	if (strcmp(items[0].text, "freebsd") == 0)
1259 		gpart_partition(newpartname, "BSD");
1260 	else
1261 		set_default_part_metadata(newpartname, scheme,
1262 		    items[0].text, items[2].text, newfs);
1263 
1264 	for (i = 0; i < nitems(items); i++)
1265 		if (items[i].text_free)
1266 			free(items[i].text);
1267 
1268 	if (partname != NULL)
1269 		*partname = strdup(newpartname);
1270 }
1271 
1272 void
1273 gpart_delete(struct gprovider *pp)
1274 {
1275 	struct gconfig *gc;
1276 	struct ggeom *geom;
1277 	struct gconsumer *cp;
1278 	struct gctl_req *r;
1279 	const char *errstr;
1280 	intmax_t idx;
1281 	int is_partition;
1282 
1283 	/* Is it a partition? */
1284 	is_partition = (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0);
1285 
1286 	/* Find out if this is the root of a gpart geom */
1287 	geom = NULL;
1288 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
1289 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
1290 			geom = cp->lg_geom;
1291 			break;
1292 		}
1293 
1294 	/* If so, destroy all children */
1295 	if (geom != NULL) {
1296 		gpart_destroy(geom);
1297 
1298 		/* If this is a partition, revert it, so it can be deleted */
1299 		if (is_partition) {
1300 			r = gctl_get_handle();
1301 			gctl_ro_param(r, "class", -1, "PART");
1302 			gctl_ro_param(r, "arg0", -1, geom->lg_name);
1303 			gctl_ro_param(r, "verb", -1, "undo");
1304 			gctl_issue(r); /* Ignore non-fatal errors */
1305 			gctl_free(r);
1306 		}
1307 	}
1308 
1309 	/*
1310 	 * If this is not a partition, see if that is a problem, complain if
1311 	 * necessary, and return always, since we need not do anything further,
1312 	 * error or no.
1313 	 */
1314 	if (!is_partition) {
1315 		if (geom == NULL)
1316 			dialog_msgbox("Error",
1317 			    "Only partitions can be deleted.", 0, 0, TRUE);
1318 		return;
1319 	}
1320 
1321 	r = gctl_get_handle();
1322 	gctl_ro_param(r, "class", -1, pp->lg_geom->lg_class->lg_name);
1323 	gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name);
1324 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
1325 	gctl_ro_param(r, "verb", -1, "delete");
1326 
1327 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
1328 		if (strcmp(gc->lg_name, "index") == 0) {
1329 			idx = atoi(gc->lg_val);
1330 			gctl_ro_param(r, "index", sizeof(idx), &idx);
1331 			break;
1332 		}
1333 	}
1334 
1335 	errstr = gctl_issue(r);
1336 	if (errstr != NULL && errstr[0] != '\0') {
1337 		gpart_show_error("Error", NULL, errstr);
1338 		gctl_free(r);
1339 		return;
1340 	}
1341 
1342 	gctl_free(r);
1343 
1344 	delete_part_metadata(pp->lg_name);
1345 }
1346 
1347 void
1348 gpart_revert_all(struct gmesh *mesh)
1349 {
1350 	struct gclass *classp;
1351 	struct gconfig *gc;
1352 	struct ggeom *gp;
1353 	struct gctl_req *r;
1354 	const char *modified;
1355 
1356 	LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
1357 		if (strcmp(classp->lg_name, "PART") == 0)
1358 			break;
1359 	}
1360 
1361 	if (strcmp(classp->lg_name, "PART") != 0) {
1362 		dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE);
1363 		return;
1364 	}
1365 
1366 	LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
1367 		modified = "true"; /* XXX: If we don't know (kernel too old),
1368 				    * assume there are modifications. */
1369 		LIST_FOREACH(gc, &gp->lg_config, lg_config) {
1370 			if (strcmp(gc->lg_name, "modified") == 0) {
1371 				modified = gc->lg_val;
1372 				break;
1373 			}
1374 		}
1375 
1376 		if (strcmp(modified, "false") == 0)
1377 			continue;
1378 
1379 		r = gctl_get_handle();
1380 		gctl_ro_param(r, "class", -1, "PART");
1381 		gctl_ro_param(r, "arg0", -1, gp->lg_name);
1382 		gctl_ro_param(r, "verb", -1, "undo");
1383 
1384 		gctl_issue(r);
1385 		gctl_free(r);
1386 	}
1387 }
1388 
1389 void
1390 gpart_commit(struct gmesh *mesh)
1391 {
1392 	struct partition_metadata *md;
1393 	struct gclass *classp;
1394 	struct ggeom *gp;
1395 	struct gconfig *gc;
1396 	struct gconsumer *cp;
1397 	struct gprovider *pp;
1398 	struct gctl_req *r;
1399 	const char *errstr;
1400 	const char *modified;
1401 	const char *rootfs;
1402 
1403 	LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
1404 		if (strcmp(classp->lg_name, "PART") == 0)
1405 			break;
1406 	}
1407 
1408 	/* Figure out what filesystem / uses */
1409 	rootfs = "ufs"; /* Assume ufs if nothing else present */
1410 	TAILQ_FOREACH(md, &part_metadata, metadata) {
1411 		if (md->fstab != NULL && strcmp(md->fstab->fs_file, "/") == 0) {
1412 			rootfs = md->fstab->fs_vfstype;
1413 			break;
1414 		}
1415 	}
1416 
1417 	if (strcmp(classp->lg_name, "PART") != 0) {
1418 		dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE);
1419 		return;
1420 	}
1421 
1422 	LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
1423 		modified = "true"; /* XXX: If we don't know (kernel too old),
1424 				    * assume there are modifications. */
1425 		LIST_FOREACH(gc, &gp->lg_config, lg_config) {
1426 			if (strcmp(gc->lg_name, "modified") == 0) {
1427 				modified = gc->lg_val;
1428 				break;
1429 			}
1430 		}
1431 
1432 		if (strcmp(modified, "false") == 0)
1433 			continue;
1434 
1435 		/* Add bootcode if necessary, before the commit */
1436 		md = get_part_metadata(gp->lg_name, 0);
1437 		if (md != NULL && md->bootcode)
1438 			gpart_bootcode(gp);
1439 
1440 		/* Now install partcode on its partitions, if necessary */
1441 		LIST_FOREACH(pp, &gp->lg_provider, lg_provider) {
1442 			md = get_part_metadata(pp->lg_name, 0);
1443 			if (md == NULL || !md->bootcode)
1444 				continue;
1445 
1446 			/* Mark this partition active if that's required */
1447 			gpart_activate(pp);
1448 
1449 			/* Check if the partition has sub-partitions */
1450 			LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
1451 				if (strcmp(cp->lg_geom->lg_class->lg_name,
1452 				    "PART") == 0)
1453 					break;
1454 
1455 			if (cp == NULL) /* No sub-partitions */
1456 				gpart_partcode(pp, rootfs);
1457 		}
1458 
1459 		r = gctl_get_handle();
1460 		gctl_ro_param(r, "class", -1, "PART");
1461 		gctl_ro_param(r, "arg0", -1, gp->lg_name);
1462 		gctl_ro_param(r, "verb", -1, "commit");
1463 
1464 		errstr = gctl_issue(r);
1465 		if (errstr != NULL && errstr[0] != '\0')
1466 			gpart_show_error("Error", NULL, errstr);
1467 		gctl_free(r);
1468 	}
1469 }
1470 
1471