xref: /freebsd/usr.sbin/bsdinstall/partedit/gpart_ops.c (revision 2117cdd4b4ebda7db0270c56f5a3cbdd37ec8cfc)
12118f387SNathan Whitehorn /*-
21de7b4b8SPedro F. Giffuni  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
31de7b4b8SPedro F. Giffuni  *
42118f387SNathan Whitehorn  * Copyright (c) 2011 Nathan Whitehorn
52118f387SNathan Whitehorn  * All rights reserved.
62118f387SNathan Whitehorn  *
72118f387SNathan Whitehorn  * Redistribution and use in source and binary forms, with or without
82118f387SNathan Whitehorn  * modification, are permitted provided that the following conditions
92118f387SNathan Whitehorn  * are met:
102118f387SNathan Whitehorn  * 1. Redistributions of source code must retain the above copyright
112118f387SNathan Whitehorn  *    notice, this list of conditions and the following disclaimer.
122118f387SNathan Whitehorn  * 2. Redistributions in binary form must reproduce the above copyright
132118f387SNathan Whitehorn  *    notice, this list of conditions and the following disclaimer in the
142118f387SNathan Whitehorn  *    documentation and/or other materials provided with the distribution.
152118f387SNathan Whitehorn  *
162118f387SNathan Whitehorn  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
172118f387SNathan Whitehorn  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
182118f387SNathan Whitehorn  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
192118f387SNathan Whitehorn  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
202118f387SNathan Whitehorn  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
212118f387SNathan Whitehorn  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
222118f387SNathan Whitehorn  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
232118f387SNathan Whitehorn  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
242118f387SNathan Whitehorn  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
252118f387SNathan Whitehorn  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
262118f387SNathan Whitehorn  * SUCH DAMAGE.
272118f387SNathan Whitehorn  *
282118f387SNathan Whitehorn  * $FreeBSD$
292118f387SNathan Whitehorn  */
302118f387SNathan Whitehorn 
312118f387SNathan Whitehorn #include <sys/param.h>
326e15678aSNathan Whitehorn #include <sys/stat.h>
332118f387SNathan Whitehorn #include <errno.h>
34db8b5613SRebecca Cran #include <fcntl.h>
352118f387SNathan Whitehorn #include <libutil.h>
362118f387SNathan Whitehorn #include <inttypes.h>
372118f387SNathan Whitehorn 
382118f387SNathan Whitehorn #include <libgeom.h>
392118f387SNathan Whitehorn #include <dialog.h>
402118f387SNathan Whitehorn #include <dlg_keys.h>
412118f387SNathan Whitehorn 
422118f387SNathan Whitehorn #include "partedit.h"
432118f387SNathan Whitehorn 
442118f387SNathan Whitehorn #define GPART_FLAGS "x" /* Do not commit changes by default */
452118f387SNathan Whitehorn 
462118f387SNathan Whitehorn static void
472118f387SNathan Whitehorn gpart_show_error(const char *title, const char *explanation, const char *errstr)
482118f387SNathan Whitehorn {
492118f387SNathan Whitehorn 	char *errmsg;
502118f387SNathan Whitehorn 	char message[512];
512118f387SNathan Whitehorn 	int error;
522118f387SNathan Whitehorn 
532118f387SNathan Whitehorn 	if (explanation == NULL)
542118f387SNathan Whitehorn 		explanation = "";
552118f387SNathan Whitehorn 
562118f387SNathan Whitehorn 	error = strtol(errstr, &errmsg, 0);
572118f387SNathan Whitehorn 	if (errmsg != errstr) {
582118f387SNathan Whitehorn 		while (errmsg[0] == ' ')
592118f387SNathan Whitehorn 			errmsg++;
602118f387SNathan Whitehorn 		if (errmsg[0] != '\0')
612118f387SNathan Whitehorn 			sprintf(message, "%s%s. %s", explanation,
622118f387SNathan Whitehorn 			    strerror(error), errmsg);
632118f387SNathan Whitehorn 		else
642118f387SNathan Whitehorn 			sprintf(message, "%s%s", explanation, strerror(error));
652118f387SNathan Whitehorn 	} else {
662118f387SNathan Whitehorn 		sprintf(message, "%s%s", explanation, errmsg);
672118f387SNathan Whitehorn 	}
682118f387SNathan Whitehorn 
692118f387SNathan Whitehorn 	dialog_msgbox(title, message, 0, 0, TRUE);
702118f387SNathan Whitehorn }
712118f387SNathan Whitehorn 
72c67f41d0SNathan Whitehorn static int
73c67f41d0SNathan Whitehorn scheme_supports_labels(const char *scheme)
74c67f41d0SNathan Whitehorn {
75c67f41d0SNathan Whitehorn 	if (strcmp(scheme, "APM") == 0)
76c67f41d0SNathan Whitehorn 		return (1);
77c67f41d0SNathan Whitehorn 	if (strcmp(scheme, "GPT") == 0)
78c67f41d0SNathan Whitehorn 		return (1);
79c67f41d0SNathan Whitehorn 
80c67f41d0SNathan Whitehorn 	return (0);
81c67f41d0SNathan Whitehorn }
82c67f41d0SNathan Whitehorn 
8350de5d07SNathan Whitehorn static void
8450de5d07SNathan Whitehorn newfs_command(const char *fstype, char *command, int use_default)
8550de5d07SNathan Whitehorn {
8650de5d07SNathan Whitehorn 	if (strcmp(fstype, "freebsd-ufs") == 0) {
8750de5d07SNathan Whitehorn 		int i;
8850de5d07SNathan Whitehorn 		DIALOG_LISTITEM items[] = {
8950de5d07SNathan Whitehorn 			{"UFS1", "UFS Version 1",
9050de5d07SNathan Whitehorn 			    "Use version 1 of the UFS file system instead "
9150de5d07SNathan Whitehorn 			    "of version 2 (not recommended)", 0 },
9250de5d07SNathan Whitehorn 			{"SU", "Softupdates",
9350de5d07SNathan Whitehorn 			    "Enable softupdates (default)", 1 },
9450de5d07SNathan Whitehorn 			{"SUJ", "Softupdates journaling",
954a8b3e41SEd Maste 			    "Enable file system journaling (default - "
964a8b3e41SEd Maste 			    "turn off for SSDs)", 1 },
9750de5d07SNathan Whitehorn 			{"TRIM", "Enable SSD TRIM support",
9850de5d07SNathan Whitehorn 			    "Enable TRIM support, useful on solid-state drives",
9950de5d07SNathan Whitehorn 			    0 },
10050de5d07SNathan Whitehorn 		};
10150de5d07SNathan Whitehorn 
10250de5d07SNathan Whitehorn 		if (!use_default) {
10350de5d07SNathan Whitehorn 			int choice;
10450de5d07SNathan Whitehorn 			choice = dlg_checklist("UFS Options", "", 0, 0, 0,
10563128851SMarcelo Araujo 			    nitems(items), items, NULL,
10650de5d07SNathan Whitehorn 			    FLAG_CHECK, &i);
10750de5d07SNathan Whitehorn 			if (choice == 1) /* Cancel */
10850de5d07SNathan Whitehorn 				return;
10950de5d07SNathan Whitehorn 		}
11050de5d07SNathan Whitehorn 
11150de5d07SNathan Whitehorn 		strcpy(command, "newfs ");
11263128851SMarcelo Araujo 		for (i = 0; i < (int)nitems(items); i++) {
11350de5d07SNathan Whitehorn 			if (items[i].state == 0)
11450de5d07SNathan Whitehorn 				continue;
11550de5d07SNathan Whitehorn 			if (strcmp(items[i].name, "UFS1") == 0)
11650de5d07SNathan Whitehorn 				strcat(command, "-O1 ");
11750de5d07SNathan Whitehorn 			else if (strcmp(items[i].name, "SU") == 0)
11850de5d07SNathan Whitehorn 				strcat(command, "-U ");
11950de5d07SNathan Whitehorn 			else if (strcmp(items[i].name, "SUJ") == 0)
12050de5d07SNathan Whitehorn 				strcat(command, "-j ");
12150de5d07SNathan Whitehorn 			else if (strcmp(items[i].name, "TRIM") == 0)
12250de5d07SNathan Whitehorn 				strcat(command, "-t ");
12350de5d07SNathan Whitehorn 		}
1246e15678aSNathan Whitehorn 	} else if (strcmp(fstype, "freebsd-zfs") == 0) {
1256e15678aSNathan Whitehorn 		int i;
1266e15678aSNathan Whitehorn 		DIALOG_LISTITEM items[] = {
1276e15678aSNathan Whitehorn 			{"fletcher4", "checksum algorithm: fletcher4",
1286e15678aSNathan Whitehorn 			    "Use fletcher4 for data integrity checking. "
1296e15678aSNathan Whitehorn 			    "(default)", 1 },
1306e15678aSNathan Whitehorn 			{"fletcher2", "checksum algorithm: fletcher2",
1316e15678aSNathan Whitehorn 			    "Use fletcher2 for data integrity checking. "
1326e15678aSNathan Whitehorn 			    "(not recommended)", 0 },
1336e15678aSNathan Whitehorn 			{"sha256", "checksum algorithm: sha256",
1346e15678aSNathan Whitehorn 			    "Use sha256 for data integrity checking. "
1356e15678aSNathan Whitehorn 			    "(not recommended)", 0 },
1366e15678aSNathan Whitehorn 			{"atime", "Update atimes for files",
1376e15678aSNathan Whitehorn 			    "Disable atime update", 0 },
1386e15678aSNathan Whitehorn 		};
1396e15678aSNathan Whitehorn 
1406e15678aSNathan Whitehorn 		if (!use_default) {
1416e15678aSNathan Whitehorn 			int choice;
1426e15678aSNathan Whitehorn 			choice = dlg_checklist("ZFS Options", "", 0, 0, 0,
14363128851SMarcelo Araujo 			    nitems(items), items, NULL,
1446e15678aSNathan Whitehorn 			    FLAG_CHECK, &i);
1456e15678aSNathan Whitehorn 			if (choice == 1) /* Cancel */
1466e15678aSNathan Whitehorn 				return;
1476e15678aSNathan Whitehorn 		}
1486e15678aSNathan Whitehorn 
1496e15678aSNathan Whitehorn 		strcpy(command, "zpool create -f -m none ");
1506e15678aSNathan Whitehorn 		if (getenv("BSDINSTALL_TMPBOOT") != NULL) {
1516e15678aSNathan Whitehorn 			char zfsboot_path[MAXPATHLEN];
1526c546e77SConrad Meyer 			snprintf(zfsboot_path, sizeof(zfsboot_path), "%s/zfs",
1536e15678aSNathan Whitehorn 			    getenv("BSDINSTALL_TMPBOOT"));
1546e15678aSNathan Whitehorn 			mkdir(zfsboot_path, S_IRWXU | S_IRGRP | S_IXGRP |
1556e15678aSNathan Whitehorn 			    S_IROTH | S_IXOTH);
1566e15678aSNathan Whitehorn 			sprintf(command, "%s -o cachefile=%s/zpool.cache ",
1576e15678aSNathan Whitehorn 			    command, zfsboot_path);
1586e15678aSNathan Whitehorn 		}
15963128851SMarcelo Araujo 		for (i = 0; i < (int)nitems(items); i++) {
1606e15678aSNathan Whitehorn 			if (items[i].state == 0)
1616e15678aSNathan Whitehorn 				continue;
1626e15678aSNathan Whitehorn 			if (strcmp(items[i].name, "fletcher4") == 0)
1636e15678aSNathan Whitehorn 				strcat(command, "-O checksum=fletcher4 ");
1646e15678aSNathan Whitehorn 			else if (strcmp(items[i].name, "fletcher2") == 0)
1656e15678aSNathan Whitehorn 				strcat(command, "-O checksum=fletcher2 ");
1666e15678aSNathan Whitehorn 			else if (strcmp(items[i].name, "sha256") == 0)
1676e15678aSNathan Whitehorn 				strcat(command, "-O checksum=sha256 ");
1686e15678aSNathan Whitehorn 			else if (strcmp(items[i].name, "atime") == 0)
1696e15678aSNathan Whitehorn 				strcat(command, "-O atime=off ");
1706e15678aSNathan Whitehorn 		}
1719f7602f0SNathan Whitehorn 	} else if (strcmp(fstype, "fat32") == 0 || strcmp(fstype, "efi") == 0 ||
1729f7602f0SNathan Whitehorn 	     strcmp(fstype, "ms-basic-data") == 0) {
17350de5d07SNathan Whitehorn 		int i;
17450de5d07SNathan Whitehorn 		DIALOG_LISTITEM items[] = {
17550de5d07SNathan Whitehorn 			{"FAT32", "FAT Type 32",
17650de5d07SNathan Whitehorn 			    "Create a FAT32 filesystem (default)", 1 },
17750de5d07SNathan Whitehorn 			{"FAT16", "FAT Type 16",
17850de5d07SNathan Whitehorn 			    "Create a FAT16 filesystem", 0 },
17950de5d07SNathan Whitehorn 			{"FAT12", "FAT Type 12",
18050de5d07SNathan Whitehorn 			    "Create a FAT12 filesystem", 0 },
18150de5d07SNathan Whitehorn 		};
18250de5d07SNathan Whitehorn 
18350de5d07SNathan Whitehorn 		if (!use_default) {
18450de5d07SNathan Whitehorn 			int choice;
18550de5d07SNathan Whitehorn 			choice = dlg_checklist("FAT Options", "", 0, 0, 0,
18663128851SMarcelo Araujo 			    nitems(items), items, NULL,
18750de5d07SNathan Whitehorn 			    FLAG_RADIO, &i);
18850de5d07SNathan Whitehorn 			if (choice == 1) /* Cancel */
18950de5d07SNathan Whitehorn 				return;
19050de5d07SNathan Whitehorn 		}
19150de5d07SNathan Whitehorn 
19250de5d07SNathan Whitehorn 		strcpy(command, "newfs_msdos ");
19363128851SMarcelo Araujo 		for (i = 0; i < (int)nitems(items); i++) {
19450de5d07SNathan Whitehorn 			if (items[i].state == 0)
19550de5d07SNathan Whitehorn 				continue;
196db8b5613SRebecca Cran 			if (strcmp(items[i].name, "FAT32") == 0)
197db8b5613SRebecca Cran 				strcat(command, "-F 32 -c 1");
198db8b5613SRebecca Cran 			else if (strcmp(items[i].name, "FAT16") == 0)
19950de5d07SNathan Whitehorn 				strcat(command, "-F 16 ");
200ae2e7abfSNathan Whitehorn 			else if (strcmp(items[i].name, "FAT12") == 0)
20150de5d07SNathan Whitehorn 				strcat(command, "-F 12 ");
20250de5d07SNathan Whitehorn 		}
20350de5d07SNathan Whitehorn 	} else {
20450de5d07SNathan Whitehorn 		if (!use_default)
20550de5d07SNathan Whitehorn 			dialog_msgbox("Error", "No configurable options exist "
20650de5d07SNathan Whitehorn 			    "for this filesystem.", 0, 0, TRUE);
20750de5d07SNathan Whitehorn 		command[0] = '\0';
20850de5d07SNathan Whitehorn 	}
20950de5d07SNathan Whitehorn }
21050de5d07SNathan Whitehorn 
2117059fa6fSAllan Jude const char *
2127059fa6fSAllan Jude choose_part_type(const char *def_scheme)
2132118f387SNathan Whitehorn {
2142118f387SNathan Whitehorn 	int cancel, choice;
2157059fa6fSAllan Jude 	const char *scheme = NULL;
2162118f387SNathan Whitehorn 
2172118f387SNathan Whitehorn 	DIALOG_LISTITEM items[] = {
2182118f387SNathan Whitehorn 		{"APM", "Apple Partition Map",
2192118f387SNathan Whitehorn 		    "Bootable on PowerPC Apple Hardware", 0 },
2202118f387SNathan Whitehorn 		{"BSD", "BSD Labels",
2212118f387SNathan Whitehorn 		    "Bootable on most x86 systems", 0 },
2222118f387SNathan Whitehorn 		{"GPT", "GUID Partition Table",
2230fbcaf76SWojciech Macek 		    "Bootable on most x86 systems and EFI aware ARM64", 0 },
2242118f387SNathan Whitehorn 		{"MBR", "DOS Partitions",
2252118f387SNathan Whitehorn 		    "Bootable on most x86 systems", 0 },
2262118f387SNathan Whitehorn 	};
2272118f387SNathan Whitehorn 
2287059fa6fSAllan Jude parttypemenu:
2297059fa6fSAllan Jude 	dialog_vars.default_item = __DECONST(char *, def_scheme);
2302118f387SNathan Whitehorn 	cancel = dlg_menu("Partition Scheme",
2312118f387SNathan Whitehorn 	    "Select a partition scheme for this volume:", 0, 0, 0,
23263128851SMarcelo Araujo 	    nitems(items), items, &choice, NULL);
2332118f387SNathan Whitehorn 	dialog_vars.default_item = NULL;
2342118f387SNathan Whitehorn 
2352118f387SNathan Whitehorn 	if (cancel)
2367059fa6fSAllan Jude 		return NULL;
2372118f387SNathan Whitehorn 
2382118f387SNathan Whitehorn 	if (!is_scheme_bootable(items[choice].name)) {
2392118f387SNathan Whitehorn 		char message[512];
2402118f387SNathan Whitehorn 		sprintf(message, "This partition scheme (%s) is not "
2412118f387SNathan Whitehorn 		    "bootable on this platform. Are you sure you want "
2422118f387SNathan Whitehorn 		    "to proceed?", items[choice].name);
2432118f387SNathan Whitehorn 		dialog_vars.defaultno = TRUE;
2442118f387SNathan Whitehorn 		cancel = dialog_yesno("Warning", message, 0, 0);
2452118f387SNathan Whitehorn 		dialog_vars.defaultno = FALSE;
2462118f387SNathan Whitehorn 		if (cancel) /* cancel */
2477059fa6fSAllan Jude 			goto parttypemenu;
2482118f387SNathan Whitehorn 	}
2492118f387SNathan Whitehorn 
2502118f387SNathan Whitehorn 	scheme = items[choice].name;
2517059fa6fSAllan Jude 
2527059fa6fSAllan Jude 	return scheme;
2537059fa6fSAllan Jude }
2547059fa6fSAllan Jude 
2557059fa6fSAllan Jude int
2567059fa6fSAllan Jude gpart_partition(const char *lg_name, const char *scheme)
2577059fa6fSAllan Jude {
2587059fa6fSAllan Jude 	int cancel;
2597059fa6fSAllan Jude 	struct gctl_req *r;
2607059fa6fSAllan Jude 	const char *errstr;
2617059fa6fSAllan Jude 
2627059fa6fSAllan Jude schememenu:
2637059fa6fSAllan Jude 	if (scheme == NULL) {
2647059fa6fSAllan Jude 		scheme = choose_part_type(default_scheme());
2657059fa6fSAllan Jude 
2667059fa6fSAllan Jude 		if (scheme == NULL)
2677059fa6fSAllan Jude 			return (-1);
2687059fa6fSAllan Jude 
2697059fa6fSAllan Jude 		if (!is_scheme_bootable(scheme)) {
2707059fa6fSAllan Jude 			char message[512];
2717059fa6fSAllan Jude 			sprintf(message, "This partition scheme (%s) is not "
2727059fa6fSAllan Jude 			    "bootable on this platform. Are you sure you want "
2737059fa6fSAllan Jude 			    "to proceed?", scheme);
2747059fa6fSAllan Jude 			dialog_vars.defaultno = TRUE;
2757059fa6fSAllan Jude 			cancel = dialog_yesno("Warning", message, 0, 0);
2767059fa6fSAllan Jude 			dialog_vars.defaultno = FALSE;
2777059fa6fSAllan Jude 			if (cancel) { /* cancel */
2787059fa6fSAllan Jude 				/* Reset scheme so user can choose another */
2797059fa6fSAllan Jude 				scheme = NULL;
2807059fa6fSAllan Jude 				goto schememenu;
2817059fa6fSAllan Jude 			}
2827059fa6fSAllan Jude 		}
2832118f387SNathan Whitehorn 	}
2842118f387SNathan Whitehorn 
2852118f387SNathan Whitehorn 	r = gctl_get_handle();
2862118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
2872118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, lg_name);
2882118f387SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
2892118f387SNathan Whitehorn 	gctl_ro_param(r, "scheme", -1, scheme);
2902118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "create");
2912118f387SNathan Whitehorn 
2922118f387SNathan Whitehorn 	errstr = gctl_issue(r);
2932118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0') {
2942118f387SNathan Whitehorn 		gpart_show_error("Error", NULL, errstr);
2952118f387SNathan Whitehorn 		gctl_free(r);
2962118f387SNathan Whitehorn 		scheme = NULL;
2972118f387SNathan Whitehorn 		goto schememenu;
2982118f387SNathan Whitehorn 	}
2992118f387SNathan Whitehorn 	gctl_free(r);
3002118f387SNathan Whitehorn 
3012118f387SNathan Whitehorn 	if (bootcode_path(scheme) != NULL)
3022118f387SNathan Whitehorn 		get_part_metadata(lg_name, 1)->bootcode = 1;
3032118f387SNathan Whitehorn 	return (0);
3042118f387SNathan Whitehorn }
3052118f387SNathan Whitehorn 
3062118f387SNathan Whitehorn static void
3072118f387SNathan Whitehorn gpart_activate(struct gprovider *pp)
3082118f387SNathan Whitehorn {
3092118f387SNathan Whitehorn 	struct gconfig *gc;
3102118f387SNathan Whitehorn 	struct gctl_req *r;
3112118f387SNathan Whitehorn 	const char *errstr, *scheme;
3122118f387SNathan Whitehorn 	const char *attribute = NULL;
3132118f387SNathan Whitehorn 	intmax_t idx;
3142118f387SNathan Whitehorn 
3152118f387SNathan Whitehorn 	/*
3162118f387SNathan Whitehorn 	 * Some partition schemes need this partition to be marked 'active'
3172118f387SNathan Whitehorn 	 * for it to be bootable.
3182118f387SNathan Whitehorn 	 */
3192118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) {
3202118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "scheme") == 0) {
3212118f387SNathan Whitehorn 			scheme = gc->lg_val;
3222118f387SNathan Whitehorn 			break;
3232118f387SNathan Whitehorn 		}
3242118f387SNathan Whitehorn 	}
3252118f387SNathan Whitehorn 
3262b375b4eSYoshihiro Takahashi 	if (strcmp(scheme, "MBR") == 0 || strcmp(scheme, "EBR") == 0)
3272118f387SNathan Whitehorn 		attribute = "active";
3282118f387SNathan Whitehorn 	else
3292118f387SNathan Whitehorn 		return;
3302118f387SNathan Whitehorn 
3312118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
3322118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "index") == 0) {
3332118f387SNathan Whitehorn 			idx = atoi(gc->lg_val);
3342118f387SNathan Whitehorn 			break;
3352118f387SNathan Whitehorn 		}
3362118f387SNathan Whitehorn 	}
3372118f387SNathan Whitehorn 
3382118f387SNathan Whitehorn 	r = gctl_get_handle();
3392118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
3402118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name);
3412118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "set");
3422118f387SNathan Whitehorn 	gctl_ro_param(r, "attrib", -1, attribute);
3432118f387SNathan Whitehorn 	gctl_ro_param(r, "index", sizeof(idx), &idx);
3442118f387SNathan Whitehorn 
3452118f387SNathan Whitehorn 	errstr = gctl_issue(r);
3462118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0')
3472118f387SNathan Whitehorn 		gpart_show_error("Error", "Error marking partition active:",
3482118f387SNathan Whitehorn 		    errstr);
3492118f387SNathan Whitehorn 	gctl_free(r);
3502118f387SNathan Whitehorn }
3512118f387SNathan Whitehorn 
3527059fa6fSAllan Jude void
3537059fa6fSAllan Jude gpart_set_root(const char *lg_name, const char *attribute)
3547059fa6fSAllan Jude {
3557059fa6fSAllan Jude 	struct gctl_req *r;
3567059fa6fSAllan Jude 	const char *errstr;
3577059fa6fSAllan Jude 
3587059fa6fSAllan Jude 	r = gctl_get_handle();
3597059fa6fSAllan Jude 	gctl_ro_param(r, "class", -1, "PART");
3607059fa6fSAllan Jude 	gctl_ro_param(r, "arg0", -1, lg_name);
3617059fa6fSAllan Jude 	gctl_ro_param(r, "flags", -1, "C");
3627059fa6fSAllan Jude 	gctl_ro_param(r, "verb", -1, "set");
3637059fa6fSAllan Jude 	gctl_ro_param(r, "attrib", -1, attribute);
3647059fa6fSAllan Jude 
3657059fa6fSAllan Jude 	errstr = gctl_issue(r);
3667059fa6fSAllan Jude 	if (errstr != NULL && errstr[0] != '\0')
3677059fa6fSAllan Jude 		gpart_show_error("Error", "Error setting parameter on disk:",
3687059fa6fSAllan Jude 		    errstr);
3697059fa6fSAllan Jude 	gctl_free(r);
3707059fa6fSAllan Jude }
3717059fa6fSAllan Jude 
3722118f387SNathan Whitehorn static void
3732118f387SNathan Whitehorn gpart_bootcode(struct ggeom *gp)
3742118f387SNathan Whitehorn {
3752118f387SNathan Whitehorn 	const char *bootcode;
3762118f387SNathan Whitehorn 	struct gconfig *gc;
3772118f387SNathan Whitehorn 	struct gctl_req *r;
3782118f387SNathan Whitehorn 	const char *errstr, *scheme;
3792118f387SNathan Whitehorn 	uint8_t *boot;
3802118f387SNathan Whitehorn 	size_t bootsize, bytes;
3812118f387SNathan Whitehorn 	int bootfd;
3822118f387SNathan Whitehorn 
3832118f387SNathan Whitehorn 	/*
3842118f387SNathan Whitehorn 	 * Write default bootcode to the newly partitioned disk, if that
3852118f387SNathan Whitehorn 	 * applies on this platform.
3862118f387SNathan Whitehorn 	 */
3872118f387SNathan Whitehorn 	LIST_FOREACH(gc, &gp->lg_config, lg_config) {
3882118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "scheme") == 0) {
3892118f387SNathan Whitehorn 			scheme = gc->lg_val;
3902118f387SNathan Whitehorn 			break;
3912118f387SNathan Whitehorn 		}
3922118f387SNathan Whitehorn 	}
3932118f387SNathan Whitehorn 
3942118f387SNathan Whitehorn 	bootcode = bootcode_path(scheme);
3952118f387SNathan Whitehorn 	if (bootcode == NULL)
3962118f387SNathan Whitehorn 		return;
3972118f387SNathan Whitehorn 
3982118f387SNathan Whitehorn 	bootfd = open(bootcode, O_RDONLY);
399c725e3efSKevin Lo 	if (bootfd < 0) {
4002118f387SNathan Whitehorn 		dialog_msgbox("Bootcode Error", strerror(errno), 0, 0,
4012118f387SNathan Whitehorn 		    TRUE);
4022118f387SNathan Whitehorn 		return;
4032118f387SNathan Whitehorn 	}
4042118f387SNathan Whitehorn 
4052118f387SNathan Whitehorn 	bootsize = lseek(bootfd, 0, SEEK_END);
4062118f387SNathan Whitehorn 	boot = malloc(bootsize);
4072118f387SNathan Whitehorn 	lseek(bootfd, 0, SEEK_SET);
4082118f387SNathan Whitehorn 	bytes = 0;
4092118f387SNathan Whitehorn 	while (bytes < bootsize)
4102118f387SNathan Whitehorn 		bytes += read(bootfd, boot + bytes, bootsize - bytes);
4112118f387SNathan Whitehorn 	close(bootfd);
4122118f387SNathan Whitehorn 
4132118f387SNathan Whitehorn 	r = gctl_get_handle();
4142118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
4152118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, gp->lg_name);
4162118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "bootcode");
4172118f387SNathan Whitehorn 	gctl_ro_param(r, "bootcode", bootsize, boot);
4182118f387SNathan Whitehorn 
4192118f387SNathan Whitehorn 	errstr = gctl_issue(r);
4202118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0')
4212118f387SNathan Whitehorn 		gpart_show_error("Bootcode Error", NULL, errstr);
4222118f387SNathan Whitehorn 	gctl_free(r);
4232118f387SNathan Whitehorn 	free(boot);
4242118f387SNathan Whitehorn }
4252118f387SNathan Whitehorn 
4262118f387SNathan Whitehorn static void
4276e15678aSNathan Whitehorn gpart_partcode(struct gprovider *pp, const char *fstype)
4282118f387SNathan Whitehorn {
4292118f387SNathan Whitehorn 	struct gconfig *gc;
4302118f387SNathan Whitehorn 	const char *scheme;
4312118f387SNathan Whitehorn 	const char *indexstr;
4322118f387SNathan Whitehorn 	char message[255], command[255];
4332118f387SNathan Whitehorn 
4342118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) {
4352118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "scheme") == 0) {
4362118f387SNathan Whitehorn 			scheme = gc->lg_val;
4372118f387SNathan Whitehorn 			break;
4382118f387SNathan Whitehorn 		}
4392118f387SNathan Whitehorn 	}
4402118f387SNathan Whitehorn 
4412118f387SNathan Whitehorn 	/* Make sure this partition scheme needs partcode on this platform */
4426e15678aSNathan Whitehorn 	if (partcode_path(scheme, fstype) == NULL)
4432118f387SNathan Whitehorn 		return;
4442118f387SNathan Whitehorn 
4452118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
4462118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "index") == 0) {
4472118f387SNathan Whitehorn 			indexstr = gc->lg_val;
4482118f387SNathan Whitehorn 			break;
4492118f387SNathan Whitehorn 		}
4502118f387SNathan Whitehorn 	}
4512118f387SNathan Whitehorn 
4522118f387SNathan Whitehorn 	/* Shell out to gpart for partcode for now */
4532118f387SNathan Whitehorn 	sprintf(command, "gpart bootcode -p %s -i %s %s",
4546e15678aSNathan Whitehorn 	    partcode_path(scheme, fstype), indexstr, pp->lg_geom->lg_name);
4552118f387SNathan Whitehorn 	if (system(command) != 0) {
4562118f387SNathan Whitehorn 		sprintf(message, "Error installing partcode on partition %s",
4572118f387SNathan Whitehorn 		    pp->lg_name);
4582118f387SNathan Whitehorn 		dialog_msgbox("Error", message, 0, 0, TRUE);
4592118f387SNathan Whitehorn 	}
4602118f387SNathan Whitehorn }
4612118f387SNathan Whitehorn 
4622118f387SNathan Whitehorn void
463f36a5e0fSNathan Whitehorn gpart_destroy(struct ggeom *lg_geom)
4642118f387SNathan Whitehorn {
4652118f387SNathan Whitehorn 	struct gctl_req *r;
466f36a5e0fSNathan Whitehorn 	struct gprovider *pp;
4672118f387SNathan Whitehorn 	const char *errstr;
468f36a5e0fSNathan Whitehorn 	int force = 1;
4692118f387SNathan Whitehorn 
470f36a5e0fSNathan Whitehorn 	/* Delete all child metadata */
4712118f387SNathan Whitehorn 	LIST_FOREACH(pp, &lg_geom->lg_provider, lg_provider)
4722118f387SNathan Whitehorn 		gpart_delete(pp);
4732118f387SNathan Whitehorn 
474f36a5e0fSNathan Whitehorn 	/* Revert any local changes to get this geom into a pristine state */
475f36a5e0fSNathan Whitehorn 	r = gctl_get_handle();
476f36a5e0fSNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
477f36a5e0fSNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, lg_geom->lg_name);
478f36a5e0fSNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "undo");
479f36a5e0fSNathan Whitehorn 	gctl_issue(r); /* Ignore errors -- these are non-fatal */
480f36a5e0fSNathan Whitehorn 	gctl_free(r);
481f36a5e0fSNathan Whitehorn 
4822118f387SNathan Whitehorn 	/* Now destroy the geom itself */
4832118f387SNathan Whitehorn 	r = gctl_get_handle();
4842118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
4852118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, lg_geom->lg_name);
4862118f387SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
487f36a5e0fSNathan Whitehorn 	gctl_ro_param(r, "force", sizeof(force), &force);
4882118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "destroy");
4892118f387SNathan Whitehorn 	errstr = gctl_issue(r);
490987415b1SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0') {
491987415b1SNathan Whitehorn 		/*
492987415b1SNathan Whitehorn 		 * Check if we reverted away the existence of the geom
493987415b1SNathan Whitehorn 		 * altogether. Show all other errors to the user.
494987415b1SNathan Whitehorn 		 */
495987415b1SNathan Whitehorn 		if (strtol(errstr, NULL, 0) != EINVAL)
4962118f387SNathan Whitehorn 			gpart_show_error("Error", NULL, errstr);
497987415b1SNathan Whitehorn 	}
4982118f387SNathan Whitehorn 	gctl_free(r);
4992118f387SNathan Whitehorn 
5002118f387SNathan Whitehorn 	/* And any metadata associated with the partition scheme itself */
5012118f387SNathan Whitehorn 	delete_part_metadata(lg_geom->lg_name);
5022118f387SNathan Whitehorn }
5032118f387SNathan Whitehorn 
5042118f387SNathan Whitehorn void
5052118f387SNathan Whitehorn gpart_edit(struct gprovider *pp)
5062118f387SNathan Whitehorn {
5072118f387SNathan Whitehorn 	struct gctl_req *r;
5082118f387SNathan Whitehorn 	struct gconfig *gc;
5092118f387SNathan Whitehorn 	struct gconsumer *cp;
5102118f387SNathan Whitehorn 	struct ggeom *geom;
5112118f387SNathan Whitehorn 	const char *errstr, *oldtype, *scheme;
5122118f387SNathan Whitehorn 	struct partition_metadata *md;
5132118f387SNathan Whitehorn 	char sizestr[32];
5146e15678aSNathan Whitehorn 	char newfs[255];
5152118f387SNathan Whitehorn 	intmax_t idx;
5162118f387SNathan Whitehorn 	int hadlabel, choice, junk, nitems;
5172118f387SNathan Whitehorn 	unsigned i;
5182118f387SNathan Whitehorn 
5192118f387SNathan Whitehorn 	DIALOG_FORMITEM items[] = {
5202118f387SNathan Whitehorn 		{0, "Type:", 5, 0, 0, FALSE, "", 11, 0, 12, 15, 0,
5216e15678aSNathan Whitehorn 		    FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, "
5226e15678aSNathan Whitehorn 		    "freebsd-swap)", FALSE},
5232118f387SNathan Whitehorn 		{0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 0, 0,
5242118f387SNathan Whitehorn 		    FALSE, "Partition size. Append K, M, G for kilobytes, "
5252118f387SNathan Whitehorn 		    "megabytes or gigabytes.", FALSE},
5262118f387SNathan Whitehorn 		{0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0,
5272118f387SNathan Whitehorn 		    FALSE, "Path at which to mount this partition (leave blank "
5282118f387SNathan Whitehorn 		    "for swap, set to / for root filesystem)", FALSE},
5292118f387SNathan Whitehorn 		{0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE,
5302118f387SNathan Whitehorn 		    "Partition name. Not all partition schemes support this.",
5312118f387SNathan Whitehorn 		    FALSE},
5322118f387SNathan Whitehorn 	};
5332118f387SNathan Whitehorn 
5342118f387SNathan Whitehorn 	/*
5352118f387SNathan Whitehorn 	 * Find the PART geom we are manipulating. This may be a consumer of
5362118f387SNathan Whitehorn 	 * this provider, or its parent. Check the consumer case first.
5372118f387SNathan Whitehorn 	 */
5382118f387SNathan Whitehorn 	geom = NULL;
5392118f387SNathan Whitehorn 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
5402118f387SNathan Whitehorn 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
541f36a5e0fSNathan Whitehorn 			/* Check for zombie geoms, treating them as blank */
542f36a5e0fSNathan Whitehorn 			scheme = NULL;
543f36a5e0fSNathan Whitehorn 			LIST_FOREACH(gc, &cp->lg_geom->lg_config, lg_config) {
544f36a5e0fSNathan Whitehorn 				if (strcmp(gc->lg_name, "scheme") == 0) {
545f36a5e0fSNathan Whitehorn 					scheme = gc->lg_val;
546f36a5e0fSNathan Whitehorn 					break;
547f36a5e0fSNathan Whitehorn 				}
548f36a5e0fSNathan Whitehorn 			}
549f36a5e0fSNathan Whitehorn 			if (scheme == NULL || strcmp(scheme, "(none)") == 0) {
550f36a5e0fSNathan Whitehorn 				gpart_partition(cp->lg_geom->lg_name, NULL);
5512118f387SNathan Whitehorn 				return;
552f36a5e0fSNathan Whitehorn 			}
5532118f387SNathan Whitehorn 
554987415b1SNathan Whitehorn 			/* If this is a nested partition, edit as usual */
555987415b1SNathan Whitehorn 			if (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
556987415b1SNathan Whitehorn 				break;
557987415b1SNathan Whitehorn 
5582118f387SNathan Whitehorn 			/* Destroy the geom and all sub-partitions */
559f36a5e0fSNathan Whitehorn 			gpart_destroy(cp->lg_geom);
5602118f387SNathan Whitehorn 
5612118f387SNathan Whitehorn 			/* Now re-partition and return */
5622118f387SNathan Whitehorn 			gpart_partition(cp->lg_geom->lg_name, NULL);
5632118f387SNathan Whitehorn 			return;
5642118f387SNathan Whitehorn 		}
5652118f387SNathan Whitehorn 
5662118f387SNathan Whitehorn 	if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
5672118f387SNathan Whitehorn 		geom = pp->lg_geom;
5682118f387SNathan Whitehorn 
5692118f387SNathan Whitehorn 	if (geom == NULL) {
5702118f387SNathan Whitehorn 		/* Disk not partitioned, so partition it */
571bcc25b7eSNathan Whitehorn 		gpart_partition(pp->lg_name, NULL);
5722118f387SNathan Whitehorn 		return;
5732118f387SNathan Whitehorn 	}
5742118f387SNathan Whitehorn 
5752118f387SNathan Whitehorn 	LIST_FOREACH(gc, &geom->lg_config, lg_config) {
5762118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "scheme") == 0) {
5772118f387SNathan Whitehorn 			scheme = gc->lg_val;
5782118f387SNathan Whitehorn 			break;
5792118f387SNathan Whitehorn 		}
5802118f387SNathan Whitehorn 	}
5812118f387SNathan Whitehorn 
582c67f41d0SNathan Whitehorn 	nitems = scheme_supports_labels(scheme) ? 4 : 3;
5832118f387SNathan Whitehorn 
5842118f387SNathan Whitehorn 	/* Edit editable parameters of a partition */
5852118f387SNathan Whitehorn 	hadlabel = 0;
5862118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
5872118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "type") == 0) {
5882118f387SNathan Whitehorn 			oldtype = gc->lg_val;
5892118f387SNathan Whitehorn 			items[0].text = gc->lg_val;
5902118f387SNathan Whitehorn 		}
5912118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "label") == 0 && gc->lg_val != NULL) {
5922118f387SNathan Whitehorn 			hadlabel = 1;
5932118f387SNathan Whitehorn 			items[3].text = gc->lg_val;
5942118f387SNathan Whitehorn 		}
5952118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "index") == 0)
5962118f387SNathan Whitehorn 			idx = atoi(gc->lg_val);
5972118f387SNathan Whitehorn 	}
5982118f387SNathan Whitehorn 
5992118f387SNathan Whitehorn 	TAILQ_FOREACH(md, &part_metadata, metadata) {
6002118f387SNathan Whitehorn 		if (md->name != NULL && strcmp(md->name, pp->lg_name) == 0) {
6012118f387SNathan Whitehorn 			if (md->fstab != NULL)
6022118f387SNathan Whitehorn 				items[2].text = md->fstab->fs_file;
6032118f387SNathan Whitehorn 			break;
6042118f387SNathan Whitehorn 		}
6052118f387SNathan Whitehorn 	}
6062118f387SNathan Whitehorn 
6072118f387SNathan Whitehorn 	humanize_number(sizestr, 7, pp->lg_mediasize, "B", HN_AUTOSCALE,
6082118f387SNathan Whitehorn 	    HN_NOSPACE | HN_DECIMAL);
6092118f387SNathan Whitehorn 	items[1].text = sizestr;
6102118f387SNathan Whitehorn 
6112118f387SNathan Whitehorn editpart:
6122118f387SNathan Whitehorn 	choice = dlg_form("Edit Partition", "", 0, 0, 0, nitems, items, &junk);
6132118f387SNathan Whitehorn 
6142118f387SNathan Whitehorn 	if (choice) /* Cancel pressed */
615987415b1SNathan Whitehorn 		goto endedit;
6162118f387SNathan Whitehorn 
617e24e98d0SWojciech Macek 	/* If this is the root partition, check that this fs is bootable */
618e24e98d0SWojciech Macek 	if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme,
619e24e98d0SWojciech Macek 	    items[0].text)) {
620e24e98d0SWojciech Macek 		char message[512];
621e24e98d0SWojciech Macek 		sprintf(message, "This file system (%s) is not bootable "
622e24e98d0SWojciech Macek 		    "on this system. Are you sure you want to proceed?",
623e24e98d0SWojciech Macek 		    items[0].text);
624e24e98d0SWojciech Macek 		dialog_vars.defaultno = TRUE;
625e24e98d0SWojciech Macek 		choice = dialog_yesno("Warning", message, 0, 0);
626e24e98d0SWojciech Macek 		dialog_vars.defaultno = FALSE;
627e24e98d0SWojciech Macek 		if (choice == 1) /* cancel */
628e24e98d0SWojciech Macek 			goto editpart;
629e24e98d0SWojciech Macek 	}
630e24e98d0SWojciech Macek 
6312118f387SNathan Whitehorn 	/* Check if the label has a / in it */
6322118f387SNathan Whitehorn 	if (strchr(items[3].text, '/') != NULL) {
6332118f387SNathan Whitehorn 		dialog_msgbox("Error", "Label contains a /, which is not an "
6342118f387SNathan Whitehorn 		    "allowed character.", 0, 0, TRUE);
6352118f387SNathan Whitehorn 		goto editpart;
6362118f387SNathan Whitehorn 	}
6372118f387SNathan Whitehorn 
6382118f387SNathan Whitehorn 	r = gctl_get_handle();
6392118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
6402118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, geom->lg_name);
6412118f387SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
6422118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "modify");
6432118f387SNathan Whitehorn 	gctl_ro_param(r, "index", sizeof(idx), &idx);
6442118f387SNathan Whitehorn 	if (hadlabel || items[3].text[0] != '\0')
6452118f387SNathan Whitehorn 		gctl_ro_param(r, "label", -1, items[3].text);
6462118f387SNathan Whitehorn 	gctl_ro_param(r, "type", -1, items[0].text);
6472118f387SNathan Whitehorn 	errstr = gctl_issue(r);
6482118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0') {
6492118f387SNathan Whitehorn 		gpart_show_error("Error", NULL, errstr);
6502118f387SNathan Whitehorn 		gctl_free(r);
6512118f387SNathan Whitehorn 		goto editpart;
6522118f387SNathan Whitehorn 	}
6532118f387SNathan Whitehorn 	gctl_free(r);
6542118f387SNathan Whitehorn 
65550de5d07SNathan Whitehorn 	newfs_command(items[0].text, newfs, 1);
6562118f387SNathan Whitehorn 	set_default_part_metadata(pp->lg_name, scheme, items[0].text,
65750de5d07SNathan Whitehorn 	    items[2].text, (strcmp(oldtype, items[0].text) != 0) ?
65850de5d07SNathan Whitehorn 	    newfs : NULL);
6592118f387SNathan Whitehorn 
660987415b1SNathan Whitehorn endedit:
661987415b1SNathan Whitehorn 	if (strcmp(oldtype, items[0].text) != 0 && cp != NULL)
662987415b1SNathan Whitehorn 		gpart_destroy(cp->lg_geom);
663987415b1SNathan Whitehorn 	if (strcmp(oldtype, items[0].text) != 0 && strcmp(items[0].text,
664987415b1SNathan Whitehorn 	    "freebsd") == 0)
665987415b1SNathan Whitehorn 		gpart_partition(pp->lg_name, "BSD");
666987415b1SNathan Whitehorn 
66763128851SMarcelo Araujo 	for (i = 0; i < nitems(items); i++)
6682118f387SNathan Whitehorn 		if (items[i].text_free)
6692118f387SNathan Whitehorn 			free(items[i].text);
6702118f387SNathan Whitehorn }
6712118f387SNathan Whitehorn 
6722118f387SNathan Whitehorn void
6732118f387SNathan Whitehorn set_default_part_metadata(const char *name, const char *scheme,
67450de5d07SNathan Whitehorn     const char *type, const char *mountpoint, const char *newfs)
6752118f387SNathan Whitehorn {
6762118f387SNathan Whitehorn 	struct partition_metadata *md;
6776e15678aSNathan Whitehorn 	char *zpool_name = NULL;
67852f39da1SNathan Whitehorn 	const char *default_bootmount = NULL;
6796e15678aSNathan Whitehorn 	int i;
6802118f387SNathan Whitehorn 
6812118f387SNathan Whitehorn 	/* Set part metadata */
6822118f387SNathan Whitehorn 	md = get_part_metadata(name, 1);
6832118f387SNathan Whitehorn 
6842118f387SNathan Whitehorn 	if (newfs) {
6852118f387SNathan Whitehorn 		if (md->newfs != NULL) {
6862118f387SNathan Whitehorn 			free(md->newfs);
6872118f387SNathan Whitehorn 			md->newfs = NULL;
6882118f387SNathan Whitehorn 		}
6892118f387SNathan Whitehorn 
69050de5d07SNathan Whitehorn 		if (newfs != NULL && newfs[0] != '\0') {
69150de5d07SNathan Whitehorn 			md->newfs = malloc(strlen(newfs) + strlen(" /dev/") +
6926e15678aSNathan Whitehorn 			    strlen(mountpoint) + 5 + strlen(name) + 1);
6936e15678aSNathan Whitehorn 			if (strcmp("freebsd-zfs", type) == 0) {
6946e15678aSNathan Whitehorn 				zpool_name = strdup((strlen(mountpoint) == 1) ?
6956e15678aSNathan Whitehorn 				    "root" : &mountpoint[1]);
6966e15678aSNathan Whitehorn 				for (i = 0; zpool_name[i] != 0; i++)
6976e15678aSNathan Whitehorn 					if (!isalnum(zpool_name[i]))
6986e15678aSNathan Whitehorn 						zpool_name[i] = '_';
6996e15678aSNathan Whitehorn 				sprintf(md->newfs, "%s %s /dev/%s", newfs,
7006e15678aSNathan Whitehorn 				    zpool_name, name);
7016e15678aSNathan Whitehorn 			} else {
70250de5d07SNathan Whitehorn 				sprintf(md->newfs, "%s /dev/%s", newfs, name);
7032118f387SNathan Whitehorn 			}
7042118f387SNathan Whitehorn 		}
7056e15678aSNathan Whitehorn 	}
7062118f387SNathan Whitehorn 
7072118f387SNathan Whitehorn 	if (strcmp(type, "freebsd-swap") == 0)
7082118f387SNathan Whitehorn 		mountpoint = "none";
70952f39da1SNathan Whitehorn 	if (strcmp(type, bootpart_type(scheme, &default_bootmount)) == 0) {
7100b7472b3SNathan Whitehorn 		if (default_bootmount == NULL)
7112118f387SNathan Whitehorn 			md->bootcode = 1;
71252f39da1SNathan Whitehorn 		else if (mountpoint == NULL || strlen(mountpoint) == 0)
71352f39da1SNathan Whitehorn 			mountpoint = default_bootmount;
71452f39da1SNathan Whitehorn 	}
7152118f387SNathan Whitehorn 
7162118f387SNathan Whitehorn 	if (mountpoint == NULL || mountpoint[0] == '\0') {
7172118f387SNathan Whitehorn 		if (md->fstab != NULL) {
7182118f387SNathan Whitehorn 			free(md->fstab->fs_spec);
7192118f387SNathan Whitehorn 			free(md->fstab->fs_file);
7202118f387SNathan Whitehorn 			free(md->fstab->fs_vfstype);
7212118f387SNathan Whitehorn 			free(md->fstab->fs_mntops);
7222118f387SNathan Whitehorn 			free(md->fstab->fs_type);
7232118f387SNathan Whitehorn 			free(md->fstab);
7242118f387SNathan Whitehorn 			md->fstab = NULL;
7252118f387SNathan Whitehorn 		}
7262118f387SNathan Whitehorn 	} else {
7272118f387SNathan Whitehorn 		if (md->fstab == NULL) {
7282118f387SNathan Whitehorn 			md->fstab = malloc(sizeof(struct fstab));
7292118f387SNathan Whitehorn 		} else {
7302118f387SNathan Whitehorn 			free(md->fstab->fs_spec);
7312118f387SNathan Whitehorn 			free(md->fstab->fs_file);
7322118f387SNathan Whitehorn 			free(md->fstab->fs_vfstype);
7332118f387SNathan Whitehorn 			free(md->fstab->fs_mntops);
7342118f387SNathan Whitehorn 			free(md->fstab->fs_type);
7352118f387SNathan Whitehorn 		}
7366e15678aSNathan Whitehorn 		if (strcmp("freebsd-zfs", type) == 0) {
7376e15678aSNathan Whitehorn 			md->fstab->fs_spec = strdup(zpool_name);
7386e15678aSNathan Whitehorn 		} else {
7396e15678aSNathan Whitehorn 			md->fstab->fs_spec = malloc(strlen(name) +
7406e15678aSNathan Whitehorn 			    strlen("/dev/") + 1);
7412118f387SNathan Whitehorn 			sprintf(md->fstab->fs_spec, "/dev/%s", name);
7426e15678aSNathan Whitehorn 		}
7432118f387SNathan Whitehorn 		md->fstab->fs_file = strdup(mountpoint);
7442118f387SNathan Whitehorn 		/* Get VFS from text after freebsd-, if possible */
74550de5d07SNathan Whitehorn 		if (strncmp("freebsd-", type, 8) == 0)
7462118f387SNathan Whitehorn 			md->fstab->fs_vfstype = strdup(&type[8]);
7479f7602f0SNathan Whitehorn 		else if (strcmp("fat32", type) == 0 || strcmp("efi", type) == 0
7489f7602f0SNathan Whitehorn 	     	    || strcmp("ms-basic-data", type) == 0)
74950de5d07SNathan Whitehorn 			md->fstab->fs_vfstype = strdup("msdosfs");
7502118f387SNathan Whitehorn 		else
7512118f387SNathan Whitehorn 			md->fstab->fs_vfstype = strdup(type); /* Guess */
7522118f387SNathan Whitehorn 		if (strcmp(type, "freebsd-swap") == 0) {
7532118f387SNathan Whitehorn 			md->fstab->fs_type = strdup(FSTAB_SW);
7542118f387SNathan Whitehorn 			md->fstab->fs_freq = 0;
7552118f387SNathan Whitehorn 			md->fstab->fs_passno = 0;
7566e15678aSNathan Whitehorn 		} else if (strcmp(type, "freebsd-zfs") == 0) {
7576e15678aSNathan Whitehorn 			md->fstab->fs_type = strdup(FSTAB_RW);
7586e15678aSNathan Whitehorn 			md->fstab->fs_freq = 0;
7596e15678aSNathan Whitehorn 			md->fstab->fs_passno = 0;
7602118f387SNathan Whitehorn 		} else {
7612118f387SNathan Whitehorn 			md->fstab->fs_type = strdup(FSTAB_RW);
7622118f387SNathan Whitehorn 			if (strcmp(mountpoint, "/") == 0) {
7632118f387SNathan Whitehorn 				md->fstab->fs_freq = 1;
7642118f387SNathan Whitehorn 				md->fstab->fs_passno = 1;
7652118f387SNathan Whitehorn 			} else {
7662118f387SNathan Whitehorn 				md->fstab->fs_freq = 2;
7672118f387SNathan Whitehorn 				md->fstab->fs_passno = 2;
7682118f387SNathan Whitehorn 			}
7692118f387SNathan Whitehorn 		}
7702118f387SNathan Whitehorn 		md->fstab->fs_mntops = strdup(md->fstab->fs_type);
7712118f387SNathan Whitehorn 	}
7726e15678aSNathan Whitehorn 
7736e15678aSNathan Whitehorn 	if (zpool_name != NULL)
7746e15678aSNathan Whitehorn 		free(zpool_name);
7752118f387SNathan Whitehorn }
7762118f387SNathan Whitehorn 
7772118f387SNathan Whitehorn static
7782118f387SNathan Whitehorn int part_compare(const void *xa, const void *xb)
7792118f387SNathan Whitehorn {
7802118f387SNathan Whitehorn 	struct gprovider **a = (struct gprovider **)xa;
7812118f387SNathan Whitehorn 	struct gprovider **b = (struct gprovider **)xb;
7822118f387SNathan Whitehorn 	intmax_t astart, bstart;
7832118f387SNathan Whitehorn 	struct gconfig *gc;
7842118f387SNathan Whitehorn 
7852118f387SNathan Whitehorn 	astart = bstart = 0;
7862118f387SNathan Whitehorn 	LIST_FOREACH(gc, &(*a)->lg_config, lg_config)
7872118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "start") == 0) {
7882118f387SNathan Whitehorn 			astart = strtoimax(gc->lg_val, NULL, 0);
7892118f387SNathan Whitehorn 			break;
7902118f387SNathan Whitehorn 		}
7912118f387SNathan Whitehorn 	LIST_FOREACH(gc, &(*b)->lg_config, lg_config)
7922118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "start") == 0) {
7932118f387SNathan Whitehorn 			bstart = strtoimax(gc->lg_val, NULL, 0);
7942118f387SNathan Whitehorn 			break;
7952118f387SNathan Whitehorn 		}
7962118f387SNathan Whitehorn 
7972118f387SNathan Whitehorn 	if (astart < bstart)
7982118f387SNathan Whitehorn 		return -1;
7992118f387SNathan Whitehorn 	else if (astart > bstart)
8002118f387SNathan Whitehorn 		return 1;
8012118f387SNathan Whitehorn 	else
8022118f387SNathan Whitehorn 		return 0;
8032118f387SNathan Whitehorn }
8042118f387SNathan Whitehorn 
8052118f387SNathan Whitehorn intmax_t
8062118f387SNathan Whitehorn gpart_max_free(struct ggeom *geom, intmax_t *npartstart)
8072118f387SNathan Whitehorn {
8082118f387SNathan Whitehorn 	struct gconfig *gc;
8092118f387SNathan Whitehorn 	struct gprovider *pp, **providers;
81047ead00dSDag-Erling Smørgrav 	intmax_t sectorsize, stripesize, offset;
8112118f387SNathan Whitehorn 	intmax_t lastend;
8122118f387SNathan Whitehorn 	intmax_t start, end;
8132118f387SNathan Whitehorn 	intmax_t maxsize, maxstart;
8142118f387SNathan Whitehorn 	intmax_t partstart, partend;
8152118f387SNathan Whitehorn 	int i, nparts;
8162118f387SNathan Whitehorn 
8172118f387SNathan Whitehorn 	/* Now get the maximum free size and free start */
8182118f387SNathan Whitehorn 	start = end = 0;
8192118f387SNathan Whitehorn 	LIST_FOREACH(gc, &geom->lg_config, lg_config) {
8202118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "first") == 0)
8212118f387SNathan Whitehorn 			start = strtoimax(gc->lg_val, NULL, 0);
8222118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "last") == 0)
8232118f387SNathan Whitehorn 			end = strtoimax(gc->lg_val, NULL, 0);
8242118f387SNathan Whitehorn 	}
8252118f387SNathan Whitehorn 
8262118f387SNathan Whitehorn 	i = nparts = 0;
8272118f387SNathan Whitehorn 	LIST_FOREACH(pp, &geom->lg_provider, lg_provider)
8282118f387SNathan Whitehorn 		nparts++;
8292118f387SNathan Whitehorn 	providers = calloc(nparts, sizeof(providers[0]));
8302118f387SNathan Whitehorn 	LIST_FOREACH(pp, &geom->lg_provider, lg_provider)
8312118f387SNathan Whitehorn 		providers[i++] = pp;
8322118f387SNathan Whitehorn 	qsort(providers, nparts, sizeof(providers[0]), part_compare);
8332118f387SNathan Whitehorn 
8342118f387SNathan Whitehorn 	lastend = start - 1;
8352118f387SNathan Whitehorn 	maxsize = 0;
8362118f387SNathan Whitehorn 	for (i = 0; i < nparts; i++) {
8372118f387SNathan Whitehorn 		pp = providers[i];
8382118f387SNathan Whitehorn 
8392118f387SNathan Whitehorn 		LIST_FOREACH(gc, &pp->lg_config, lg_config) {
8402118f387SNathan Whitehorn 			if (strcmp(gc->lg_name, "start") == 0)
8412118f387SNathan Whitehorn 				partstart = strtoimax(gc->lg_val, NULL, 0);
8422118f387SNathan Whitehorn 			if (strcmp(gc->lg_name, "end") == 0)
8432118f387SNathan Whitehorn 				partend = strtoimax(gc->lg_val, NULL, 0);
8442118f387SNathan Whitehorn 		}
8452118f387SNathan Whitehorn 
8462118f387SNathan Whitehorn 		if (partstart - lastend > maxsize) {
8472118f387SNathan Whitehorn 			maxsize = partstart - lastend - 1;
8482118f387SNathan Whitehorn 			maxstart = lastend + 1;
8492118f387SNathan Whitehorn 		}
8502118f387SNathan Whitehorn 
8512118f387SNathan Whitehorn 		lastend = partend;
8522118f387SNathan Whitehorn 	}
8532118f387SNathan Whitehorn 
8542118f387SNathan Whitehorn 	if (end - lastend > maxsize) {
8558c7de243SNathan Whitehorn 		maxsize = end - lastend;
8562118f387SNathan Whitehorn 		maxstart = lastend + 1;
8572118f387SNathan Whitehorn 	}
8582118f387SNathan Whitehorn 
8592118f387SNathan Whitehorn 	pp = LIST_FIRST(&geom->lg_consumer)->lg_provider;
8602118f387SNathan Whitehorn 
86147ead00dSDag-Erling Smørgrav 	/*
86247ead00dSDag-Erling Smørgrav 	 * Round the start and size of the largest available space up to
86347ead00dSDag-Erling Smørgrav 	 * the nearest multiple of the adjusted stripe size.
86447ead00dSDag-Erling Smørgrav 	 *
86547ead00dSDag-Erling Smørgrav 	 * The adjusted stripe size is the least common multiple of the
86647ead00dSDag-Erling Smørgrav 	 * actual stripe size, or the sector size if no stripe size was
86747ead00dSDag-Erling Smørgrav 	 * reported, and 4096.  The reason for this is that contemporary
86847ead00dSDag-Erling Smørgrav 	 * disks often have 4096-byte physical sectors but report 512
86947ead00dSDag-Erling Smørgrav 	 * bytes instead for compatibility with older / broken operating
87047ead00dSDag-Erling Smørgrav 	 * systems and BIOSes.  For the same reasons, virtualized storage
87147ead00dSDag-Erling Smørgrav 	 * may also report a 512-byte stripe size, or none at all.
87247ead00dSDag-Erling Smørgrav 	 */
87347ead00dSDag-Erling Smørgrav 	sectorsize = pp->lg_sectorsize;
87447ead00dSDag-Erling Smørgrav 	if ((stripesize = pp->lg_stripesize) == 0)
87547ead00dSDag-Erling Smørgrav 		stripesize = sectorsize;
87647ead00dSDag-Erling Smørgrav 	while (stripesize % 4096 != 0)
87747ead00dSDag-Erling Smørgrav 		stripesize *= 2;
87847ead00dSDag-Erling Smørgrav 	if ((offset = maxstart * sectorsize % stripesize) != 0) {
87947ead00dSDag-Erling Smørgrav 		offset = (stripesize - offset) / sectorsize;
8802118f387SNathan Whitehorn 		maxstart += offset;
8812118f387SNathan Whitehorn 		maxsize -= offset;
8822118f387SNathan Whitehorn 	}
8832118f387SNathan Whitehorn 
8842118f387SNathan Whitehorn 	if (npartstart != NULL)
8852118f387SNathan Whitehorn 		*npartstart = maxstart;
8862118f387SNathan Whitehorn 
8872118f387SNathan Whitehorn 	return (maxsize);
8882118f387SNathan Whitehorn }
8892118f387SNathan Whitehorn 
89052f39da1SNathan Whitehorn static size_t
89152f39da1SNathan Whitehorn add_boot_partition(struct ggeom *geom, struct gprovider *pp,
89252f39da1SNathan Whitehorn     const char *scheme, int interactive)
89352f39da1SNathan Whitehorn {
89452f39da1SNathan Whitehorn 	struct gconfig *gc;
89552f39da1SNathan Whitehorn 	struct gprovider *ppi;
89652f39da1SNathan Whitehorn 	int choice;
89752f39da1SNathan Whitehorn 
89852f39da1SNathan Whitehorn 	/* Check for existing freebsd-boot partition */
89952f39da1SNathan Whitehorn 	LIST_FOREACH(ppi, &geom->lg_provider, lg_provider) {
90052f39da1SNathan Whitehorn 		struct partition_metadata *md;
90152f39da1SNathan Whitehorn 		const char *bootmount = NULL;
90252f39da1SNathan Whitehorn 
90352f39da1SNathan Whitehorn 		LIST_FOREACH(gc, &ppi->lg_config, lg_config)
90452f39da1SNathan Whitehorn 			if (strcmp(gc->lg_name, "type") == 0)
90552f39da1SNathan Whitehorn 				break;
90652f39da1SNathan Whitehorn 		if (gc == NULL)
90752f39da1SNathan Whitehorn 			continue;
90852f39da1SNathan Whitehorn 		if (strcmp(gc->lg_val, bootpart_type(scheme, &bootmount)) != 0)
90952f39da1SNathan Whitehorn 			continue;
91052f39da1SNathan Whitehorn 
91152f39da1SNathan Whitehorn 		/*
91252f39da1SNathan Whitehorn 		 * If the boot partition is not mountable and needs partcode,
91352f39da1SNathan Whitehorn 		 * but doesn't have it, it doesn't satisfy our requirements.
91452f39da1SNathan Whitehorn 		 */
91552f39da1SNathan Whitehorn 		md = get_part_metadata(ppi->lg_name, 0);
91652f39da1SNathan Whitehorn 		if (bootmount == NULL && (md == NULL || !md->bootcode))
91752f39da1SNathan Whitehorn 			continue;
91852f39da1SNathan Whitehorn 
91952f39da1SNathan Whitehorn 		/* If it is mountable, but mounted somewhere else, remount */
92052f39da1SNathan Whitehorn 		if (bootmount != NULL && md != NULL && md->fstab != NULL
92152f39da1SNathan Whitehorn 		    && strlen(md->fstab->fs_file) > 0
92252f39da1SNathan Whitehorn 		    && strcmp(md->fstab->fs_file, bootmount) != 0)
92352f39da1SNathan Whitehorn 			continue;
92452f39da1SNathan Whitehorn 
92552f39da1SNathan Whitehorn 		/* If it is mountable, but mountpoint is not set, mount it */
92652f39da1SNathan Whitehorn 		if (bootmount != NULL && md == NULL)
92752f39da1SNathan Whitehorn 			set_default_part_metadata(ppi->lg_name, scheme,
92852f39da1SNathan Whitehorn 			    gc->lg_val, bootmount, NULL);
92952f39da1SNathan Whitehorn 
93052f39da1SNathan Whitehorn 		/* Looks good at this point, no added data needed */
93152f39da1SNathan Whitehorn 		return (0);
93252f39da1SNathan Whitehorn 	}
93352f39da1SNathan Whitehorn 
93452f39da1SNathan Whitehorn 	if (interactive)
93552f39da1SNathan Whitehorn 		choice = dialog_yesno("Boot Partition",
93652f39da1SNathan Whitehorn 		    "This partition scheme requires a boot partition "
93752f39da1SNathan Whitehorn 		    "for the disk to be bootable. Would you like to "
93852f39da1SNathan Whitehorn 		    "make one now?", 0, 0);
93952f39da1SNathan Whitehorn 	else
94052f39da1SNathan Whitehorn 		choice = 0;
94152f39da1SNathan Whitehorn 
94252f39da1SNathan Whitehorn 	if (choice == 0) { /* yes */
9436134b186SNathan Whitehorn 		struct partition_metadata *md;
94452f39da1SNathan Whitehorn 		const char *bootmount = NULL;
9456134b186SNathan Whitehorn 		char *bootpartname = NULL;
94652f39da1SNathan Whitehorn 		char sizestr[7];
94752f39da1SNathan Whitehorn 
94852f39da1SNathan Whitehorn 		humanize_number(sizestr, 7,
94952f39da1SNathan Whitehorn 		    bootpart_size(scheme), "B", HN_AUTOSCALE,
95052f39da1SNathan Whitehorn 		    HN_NOSPACE | HN_DECIMAL);
95152f39da1SNathan Whitehorn 
95252f39da1SNathan Whitehorn 		gpart_create(pp, bootpart_type(scheme, &bootmount),
9536134b186SNathan Whitehorn 		    sizestr, bootmount, &bootpartname, 0);
9546134b186SNathan Whitehorn 
9556134b186SNathan Whitehorn 		if (bootpartname == NULL) /* Error reported to user already */
9566134b186SNathan Whitehorn 			return 0;
9576134b186SNathan Whitehorn 
9586134b186SNathan Whitehorn 		/* If the part is not mountable, make sure newfs isn't set */
9596134b186SNathan Whitehorn 		if (bootmount == NULL) {
9606134b186SNathan Whitehorn 			md = get_part_metadata(bootpartname, 0);
9616134b186SNathan Whitehorn 			if (md != NULL && md->newfs != NULL) {
9626134b186SNathan Whitehorn 				free(md->newfs);
9636134b186SNathan Whitehorn 				md->newfs = NULL;
9646134b186SNathan Whitehorn 			}
9656134b186SNathan Whitehorn 		}
9666134b186SNathan Whitehorn 
9676134b186SNathan Whitehorn 		free(bootpartname);
96852f39da1SNathan Whitehorn 
96952f39da1SNathan Whitehorn 		return (bootpart_size(scheme));
97052f39da1SNathan Whitehorn 	}
97152f39da1SNathan Whitehorn 
97252f39da1SNathan Whitehorn 	return (0);
97352f39da1SNathan Whitehorn }
97452f39da1SNathan Whitehorn 
9752118f387SNathan Whitehorn void
97652f39da1SNathan Whitehorn gpart_create(struct gprovider *pp, const char *default_type,
97752f39da1SNathan Whitehorn     const char *default_size, const char *default_mountpoint,
97852f39da1SNathan Whitehorn     char **partname, int interactive)
9792118f387SNathan Whitehorn {
9802118f387SNathan Whitehorn 	struct gctl_req *r;
9812118f387SNathan Whitehorn 	struct gconfig *gc;
9822118f387SNathan Whitehorn 	struct gconsumer *cp;
9832118f387SNathan Whitehorn 	struct ggeom *geom;
9842118f387SNathan Whitehorn 	const char *errstr, *scheme;
9857899c6c1SNathan Whitehorn 	char sizestr[32], startstr[32], output[64], *newpartname;
9866e15678aSNathan Whitehorn 	char newfs[255], options_fstype[64];
9872118f387SNathan Whitehorn 	intmax_t maxsize, size, sector, firstfree, stripe;
9882118f387SNathan Whitehorn 	uint64_t bytes;
9892118f387SNathan Whitehorn 	int nitems, choice, junk;
9902118f387SNathan Whitehorn 	unsigned i;
9912118f387SNathan Whitehorn 
9922118f387SNathan Whitehorn 	DIALOG_FORMITEM items[] = {
9932118f387SNathan Whitehorn 		{0, "Type:", 5, 0, 0, FALSE, "freebsd-ufs", 11, 0, 12, 15, 0,
9946e15678aSNathan Whitehorn 		    FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, "
9956e15678aSNathan Whitehorn 		    "freebsd-swap)", FALSE},
9962118f387SNathan Whitehorn 		{0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 15, 0,
9972118f387SNathan Whitehorn 		    FALSE, "Partition size. Append K, M, G for kilobytes, "
9982118f387SNathan Whitehorn 		    "megabytes or gigabytes.", FALSE},
9992118f387SNathan Whitehorn 		{0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0,
10002118f387SNathan Whitehorn 		    FALSE, "Path at which to mount partition (blank for "
10012118f387SNathan Whitehorn 		    "swap, / for root filesystem)", FALSE},
10022118f387SNathan Whitehorn 		{0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE,
10032118f387SNathan Whitehorn 		    "Partition name. Not all partition schemes support this.",
10042118f387SNathan Whitehorn 		    FALSE},
10052118f387SNathan Whitehorn 	};
10062118f387SNathan Whitehorn 
10072118f387SNathan Whitehorn 	if (partname != NULL)
10082118f387SNathan Whitehorn 		*partname = NULL;
10092118f387SNathan Whitehorn 
10102118f387SNathan Whitehorn 	/* Record sector and stripe sizes */
10112118f387SNathan Whitehorn 	sector = pp->lg_sectorsize;
10122118f387SNathan Whitehorn 	stripe = pp->lg_stripesize;
10132118f387SNathan Whitehorn 
10142118f387SNathan Whitehorn 	/*
10152118f387SNathan Whitehorn 	 * Find the PART geom we are manipulating. This may be a consumer of
10162118f387SNathan Whitehorn 	 * this provider, or its parent. Check the consumer case first.
10172118f387SNathan Whitehorn 	 */
10182118f387SNathan Whitehorn 	geom = NULL;
10192118f387SNathan Whitehorn 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
10202118f387SNathan Whitehorn 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
10212118f387SNathan Whitehorn 			geom = cp->lg_geom;
10222118f387SNathan Whitehorn 			break;
10232118f387SNathan Whitehorn 		}
10242118f387SNathan Whitehorn 
10252118f387SNathan Whitehorn 	if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0)
10262118f387SNathan Whitehorn 		geom = pp->lg_geom;
10272118f387SNathan Whitehorn 
10282118f387SNathan Whitehorn 	/* Now get the partition scheme */
10292118f387SNathan Whitehorn 	scheme = NULL;
10302118f387SNathan Whitehorn 	if (geom != NULL) {
10312118f387SNathan Whitehorn 		LIST_FOREACH(gc, &geom->lg_config, lg_config)
10322118f387SNathan Whitehorn 			if (strcmp(gc->lg_name, "scheme") == 0)
10332118f387SNathan Whitehorn 				scheme = gc->lg_val;
10342118f387SNathan Whitehorn 	}
10352118f387SNathan Whitehorn 
10362118f387SNathan Whitehorn 	if (geom == NULL || scheme == NULL || strcmp(scheme, "(none)") == 0) {
1037bcc25b7eSNathan Whitehorn 		if (gpart_partition(pp->lg_name, NULL) == 0)
10382118f387SNathan Whitehorn 			dialog_msgbox("",
10392118f387SNathan Whitehorn 			    "The partition table has been successfully created."
10402118f387SNathan Whitehorn 			    " Please press Create again to create partitions.",
10412118f387SNathan Whitehorn 			    0, 0, TRUE);
10422118f387SNathan Whitehorn 
10432118f387SNathan Whitehorn 		return;
10442118f387SNathan Whitehorn 	}
10452118f387SNathan Whitehorn 
10462118f387SNathan Whitehorn 	/*
10472118f387SNathan Whitehorn 	 * If we still don't have a geom, either the user has
10482118f387SNathan Whitehorn 	 * canceled partitioning or there has been an error which has already
10492118f387SNathan Whitehorn 	 * been displayed, so bail.
10502118f387SNathan Whitehorn 	 */
10512118f387SNathan Whitehorn 	if (geom == NULL)
10522118f387SNathan Whitehorn 		return;
10532118f387SNathan Whitehorn 
10542118f387SNathan Whitehorn 	maxsize = size = gpart_max_free(geom, &firstfree);
10552118f387SNathan Whitehorn 	if (size <= 0) {
10562118f387SNathan Whitehorn 		dialog_msgbox("Error", "No free space left on device.", 0, 0,
10572118f387SNathan Whitehorn 		    TRUE);
10582118f387SNathan Whitehorn 		return;
10592118f387SNathan Whitehorn 	}
10602118f387SNathan Whitehorn 
10612118f387SNathan Whitehorn 	humanize_number(sizestr, 7, size*sector, "B", HN_AUTOSCALE,
10622118f387SNathan Whitehorn 	    HN_NOSPACE | HN_DECIMAL);
10632118f387SNathan Whitehorn 	items[1].text = sizestr;
10642118f387SNathan Whitehorn 
10652118f387SNathan Whitehorn 	/* Special-case the MBR default type for nested partitions */
10662b375b4eSYoshihiro Takahashi 	if (strcmp(scheme, "MBR") == 0) {
10672118f387SNathan Whitehorn 		items[0].text = "freebsd";
1068bcc25b7eSNathan Whitehorn 		items[0].help = "Filesystem type (e.g. freebsd, fat32)";
1069bcc25b7eSNathan Whitehorn 	}
10702118f387SNathan Whitehorn 
1071c67f41d0SNathan Whitehorn 	nitems = scheme_supports_labels(scheme) ? 4 : 3;
10722118f387SNathan Whitehorn 
10732118f387SNathan Whitehorn 	if (default_type != NULL)
107452f39da1SNathan Whitehorn 		items[0].text = (char *)default_type;
10752118f387SNathan Whitehorn 	if (default_size != NULL)
107652f39da1SNathan Whitehorn 		items[1].text = (char *)default_size;
10772118f387SNathan Whitehorn 	if (default_mountpoint != NULL)
107852f39da1SNathan Whitehorn 		items[2].text = (char *)default_mountpoint;
10792118f387SNathan Whitehorn 
108050de5d07SNathan Whitehorn 	/* Default options */
108150de5d07SNathan Whitehorn 	strncpy(options_fstype, items[0].text,
108250de5d07SNathan Whitehorn 	    sizeof(options_fstype));
108350de5d07SNathan Whitehorn 	newfs_command(options_fstype, newfs, 1);
10842118f387SNathan Whitehorn addpartform:
10852118f387SNathan Whitehorn 	if (interactive) {
108650de5d07SNathan Whitehorn 		dialog_vars.extra_label = "Options";
108750de5d07SNathan Whitehorn 		dialog_vars.extra_button = TRUE;
10882118f387SNathan Whitehorn 		choice = dlg_form("Add Partition", "", 0, 0, 0, nitems,
10892118f387SNathan Whitehorn 		    items, &junk);
109050de5d07SNathan Whitehorn 		dialog_vars.extra_button = FALSE;
109150de5d07SNathan Whitehorn 		switch (choice) {
109250de5d07SNathan Whitehorn 		case 0: /* OK */
109350de5d07SNathan Whitehorn 			break;
109450de5d07SNathan Whitehorn 		case 1: /* Cancel */
10952118f387SNathan Whitehorn 			return;
109650de5d07SNathan Whitehorn 		case 3: /* Options */
109750de5d07SNathan Whitehorn 			strncpy(options_fstype, items[0].text,
109850de5d07SNathan Whitehorn 			    sizeof(options_fstype));
109950de5d07SNathan Whitehorn 			newfs_command(options_fstype, newfs, 0);
110050de5d07SNathan Whitehorn 			goto addpartform;
110150de5d07SNathan Whitehorn 		}
110250de5d07SNathan Whitehorn 	}
110350de5d07SNathan Whitehorn 
110450de5d07SNathan Whitehorn 	/*
110550de5d07SNathan Whitehorn 	 * If the user changed the fs type after specifying options, undo
110650de5d07SNathan Whitehorn 	 * their choices in favor of the new filesystem's defaults.
110750de5d07SNathan Whitehorn 	 */
11080d49c6d5SNathan Whitehorn 	if (strcmp(options_fstype, items[0].text) != 0) {
110950de5d07SNathan Whitehorn 		strncpy(options_fstype, items[0].text, sizeof(options_fstype));
111050de5d07SNathan Whitehorn 		newfs_command(options_fstype, newfs, 1);
11112118f387SNathan Whitehorn 	}
11122118f387SNathan Whitehorn 
11132118f387SNathan Whitehorn 	size = maxsize;
11142118f387SNathan Whitehorn 	if (strlen(items[1].text) > 0) {
11152118f387SNathan Whitehorn 		if (expand_number(items[1].text, &bytes) != 0) {
11162118f387SNathan Whitehorn 			char error[512];
11172118f387SNathan Whitehorn 
11182118f387SNathan Whitehorn 			sprintf(error, "Invalid size: %s\n", strerror(errno));
11192118f387SNathan Whitehorn 			dialog_msgbox("Error", error, 0, 0, TRUE);
11202118f387SNathan Whitehorn 			goto addpartform;
11212118f387SNathan Whitehorn 		}
11222118f387SNathan Whitehorn 		size = MIN((intmax_t)(bytes/sector), maxsize);
11232118f387SNathan Whitehorn 	}
11242118f387SNathan Whitehorn 
11252118f387SNathan Whitehorn 	/* Check if the label has a / in it */
11262118f387SNathan Whitehorn 	if (strchr(items[3].text, '/') != NULL) {
11272118f387SNathan Whitehorn 		dialog_msgbox("Error", "Label contains a /, which is not an "
11282118f387SNathan Whitehorn 		    "allowed character.", 0, 0, TRUE);
11292118f387SNathan Whitehorn 		goto addpartform;
11302118f387SNathan Whitehorn 	}
11312118f387SNathan Whitehorn 
11322118f387SNathan Whitehorn 	/* Warn if no mountpoint set */
11332118f387SNathan Whitehorn 	if (strcmp(items[0].text, "freebsd-ufs") == 0 &&
11342118f387SNathan Whitehorn 	    items[2].text[0] != '/') {
113540f0d8dcSRavi Pokala 		choice = 0;
113640f0d8dcSRavi Pokala 		if (interactive) {
11372118f387SNathan Whitehorn 			dialog_vars.defaultno = TRUE;
11382118f387SNathan Whitehorn 			choice = dialog_yesno("Warning",
11392118f387SNathan Whitehorn 			    "This partition does not have a valid mountpoint "
11402118f387SNathan Whitehorn 			    "(for the partition from which you intend to boot the "
11412118f387SNathan Whitehorn 			    "operating system, the mountpoint should be /). Are you "
11422118f387SNathan Whitehorn 			    "sure you want to continue?"
11432118f387SNathan Whitehorn 			, 0, 0);
11442118f387SNathan Whitehorn 			dialog_vars.defaultno = FALSE;
114540f0d8dcSRavi Pokala 		}
11462118f387SNathan Whitehorn 		if (choice == 1) /* cancel */
11472118f387SNathan Whitehorn 			goto addpartform;
11482118f387SNathan Whitehorn 	}
11492118f387SNathan Whitehorn 
1150a780c996SNathan Whitehorn 	/*
1151a780c996SNathan Whitehorn 	 * Error if this scheme needs nested partitions, this is one, and
1152a780c996SNathan Whitehorn 	 * a mountpoint was set.
1153a780c996SNathan Whitehorn 	 */
1154a780c996SNathan Whitehorn 	if (strcmp(items[0].text, "freebsd") == 0 &&
1155a780c996SNathan Whitehorn 	    strlen(items[2].text) > 0) {
1156a780c996SNathan Whitehorn 		dialog_msgbox("Error", "Partitions of type \"freebsd\" are "
1157a780c996SNathan Whitehorn 		    "nested BSD-type partition schemes and cannot have "
1158a780c996SNathan Whitehorn 		    "mountpoints. After creating one, select it and press "
1159a780c996SNathan Whitehorn 		    "Create again to add the actual file systems.", 0, 0, TRUE);
1160a780c996SNathan Whitehorn 		goto addpartform;
1161a780c996SNathan Whitehorn 	}
1162a780c996SNathan Whitehorn 
11632118f387SNathan Whitehorn 	/* If this is the root partition, check that this scheme is bootable */
11642118f387SNathan Whitehorn 	if (strcmp(items[2].text, "/") == 0 && !is_scheme_bootable(scheme)) {
11652118f387SNathan Whitehorn 		char message[512];
11662118f387SNathan Whitehorn 		sprintf(message, "This partition scheme (%s) is not bootable "
11672118f387SNathan Whitehorn 		    "on this platform. Are you sure you want to proceed?",
11682118f387SNathan Whitehorn 		    scheme);
11692118f387SNathan Whitehorn 		dialog_vars.defaultno = TRUE;
11702118f387SNathan Whitehorn 		choice = dialog_yesno("Warning", message, 0, 0);
11712118f387SNathan Whitehorn 		dialog_vars.defaultno = FALSE;
11722118f387SNathan Whitehorn 		if (choice == 1) /* cancel */
11732118f387SNathan Whitehorn 			goto addpartform;
11742118f387SNathan Whitehorn 	}
11752118f387SNathan Whitehorn 
11766e15678aSNathan Whitehorn 	/* If this is the root partition, check that this fs is bootable */
11776e15678aSNathan Whitehorn 	if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme,
11786e15678aSNathan Whitehorn 	    items[0].text)) {
11796e15678aSNathan Whitehorn 		char message[512];
11806e15678aSNathan Whitehorn 		sprintf(message, "This file system (%s) is not bootable "
11816e15678aSNathan Whitehorn 		    "on this system. Are you sure you want to proceed?",
11826e15678aSNathan Whitehorn 		    items[0].text);
11836e15678aSNathan Whitehorn 		dialog_vars.defaultno = TRUE;
11846e15678aSNathan Whitehorn 		choice = dialog_yesno("Warning", message, 0, 0);
11856e15678aSNathan Whitehorn 		dialog_vars.defaultno = FALSE;
11866e15678aSNathan Whitehorn 		if (choice == 1) /* cancel */
11876e15678aSNathan Whitehorn 			goto addpartform;
11886e15678aSNathan Whitehorn 	}
11896e15678aSNathan Whitehorn 
11902118f387SNathan Whitehorn 	/*
11912118f387SNathan Whitehorn 	 * If this is the root partition, and we need a boot partition, ask
11922118f387SNathan Whitehorn 	 * the user to add one.
11932118f387SNathan Whitehorn 	 */
1194a780c996SNathan Whitehorn 
11958d795806SNathan Whitehorn 	if ((strcmp(items[0].text, "freebsd") == 0 ||
119652f39da1SNathan Whitehorn 	    strcmp(items[2].text, "/") == 0) && bootpart_size(scheme) > 0) {
119752f39da1SNathan Whitehorn 		size_t bytes = add_boot_partition(geom, pp, scheme,
119852f39da1SNathan Whitehorn 		    interactive);
11992118f387SNathan Whitehorn 
12002118f387SNathan Whitehorn 		/* Now adjust the part we are really adding forward */
120152f39da1SNathan Whitehorn 		if (bytes > 0) {
120252f39da1SNathan Whitehorn 			firstfree += bytes / sector;
120352f39da1SNathan Whitehorn 			size -= (bytes + stripe)/sector;
12042118f387SNathan Whitehorn 			if (stripe > 0 && (firstfree*sector % stripe) != 0)
12052118f387SNathan Whitehorn 				firstfree += (stripe - ((firstfree*sector) %
12062118f387SNathan Whitehorn 				    stripe)) / sector;
12072118f387SNathan Whitehorn 		}
12082118f387SNathan Whitehorn 	}
12092118f387SNathan Whitehorn 
1210*2117cdd4SAlexander Motin 	output[0] = '\0';
1211*2117cdd4SAlexander Motin 
12122118f387SNathan Whitehorn 	r = gctl_get_handle();
12132118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
12142118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, geom->lg_name);
12152118f387SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
12162118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "add");
12172118f387SNathan Whitehorn 	gctl_ro_param(r, "type", -1, items[0].text);
12182118f387SNathan Whitehorn 	snprintf(sizestr, sizeof(sizestr), "%jd", size);
12192118f387SNathan Whitehorn 	gctl_ro_param(r, "size", -1, sizestr);
12202118f387SNathan Whitehorn 	snprintf(startstr, sizeof(startstr), "%jd", firstfree);
12212118f387SNathan Whitehorn 	gctl_ro_param(r, "start", -1, startstr);
12222118f387SNathan Whitehorn 	if (items[3].text[0] != '\0')
12232118f387SNathan Whitehorn 		gctl_ro_param(r, "label", -1, items[3].text);
1224*2117cdd4SAlexander Motin 	gctl_add_param(r, "output", sizeof(output), output,
1225*2117cdd4SAlexander Motin 	    GCTL_PARAM_WR | GCTL_PARAM_ASCII);
12262118f387SNathan Whitehorn 	errstr = gctl_issue(r);
12272118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0') {
12282118f387SNathan Whitehorn 		gpart_show_error("Error", NULL, errstr);
12292118f387SNathan Whitehorn 		gctl_free(r);
12302118f387SNathan Whitehorn 		goto addpartform;
12312118f387SNathan Whitehorn 	}
12327899c6c1SNathan Whitehorn 	newpartname = strtok(output, " ");
12337899c6c1SNathan Whitehorn 	gctl_free(r);
12347899c6c1SNathan Whitehorn 
12357899c6c1SNathan Whitehorn 	/*
12367899c6c1SNathan Whitehorn 	 * Try to destroy any geom that gpart picked up already here from
12377899c6c1SNathan Whitehorn 	 * dirty blocks.
12387899c6c1SNathan Whitehorn 	 */
12397899c6c1SNathan Whitehorn 	r = gctl_get_handle();
12407899c6c1SNathan Whitehorn 	gctl_ro_param(r, "class", -1, "PART");
12417899c6c1SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, newpartname);
12427899c6c1SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
12437899c6c1SNathan Whitehorn 	junk = 1;
12447899c6c1SNathan Whitehorn 	gctl_ro_param(r, "force", sizeof(junk), &junk);
12457899c6c1SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "destroy");
12467899c6c1SNathan Whitehorn 	gctl_issue(r); /* Error usually expected and non-fatal */
12477899c6c1SNathan Whitehorn 	gctl_free(r);
12482118f387SNathan Whitehorn 
124952f39da1SNathan Whitehorn 
125052f39da1SNathan Whitehorn 	if (strcmp(items[0].text, "freebsd") == 0)
12517899c6c1SNathan Whitehorn 		gpart_partition(newpartname, "BSD");
12522118f387SNathan Whitehorn 	else
12537899c6c1SNathan Whitehorn 		set_default_part_metadata(newpartname, scheme,
125450de5d07SNathan Whitehorn 		    items[0].text, items[2].text, newfs);
12552118f387SNathan Whitehorn 
125663128851SMarcelo Araujo 	for (i = 0; i < nitems(items); i++)
12572118f387SNathan Whitehorn 		if (items[i].text_free)
12582118f387SNathan Whitehorn 			free(items[i].text);
12592118f387SNathan Whitehorn 
12602118f387SNathan Whitehorn 	if (partname != NULL)
12617899c6c1SNathan Whitehorn 		*partname = strdup(newpartname);
12622118f387SNathan Whitehorn }
12632118f387SNathan Whitehorn 
12642118f387SNathan Whitehorn void
12652118f387SNathan Whitehorn gpart_delete(struct gprovider *pp)
12662118f387SNathan Whitehorn {
12672118f387SNathan Whitehorn 	struct gconfig *gc;
12682118f387SNathan Whitehorn 	struct ggeom *geom;
12692118f387SNathan Whitehorn 	struct gconsumer *cp;
12702118f387SNathan Whitehorn 	struct gctl_req *r;
12712118f387SNathan Whitehorn 	const char *errstr;
12722118f387SNathan Whitehorn 	intmax_t idx;
1273f36a5e0fSNathan Whitehorn 	int is_partition;
12742118f387SNathan Whitehorn 
12752118f387SNathan Whitehorn 	/* Is it a partition? */
12762118f387SNathan Whitehorn 	is_partition = (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0);
12772118f387SNathan Whitehorn 
12782118f387SNathan Whitehorn 	/* Find out if this is the root of a gpart geom */
12792118f387SNathan Whitehorn 	geom = NULL;
12802118f387SNathan Whitehorn 	LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
12812118f387SNathan Whitehorn 		if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) {
12822118f387SNathan Whitehorn 			geom = cp->lg_geom;
12832118f387SNathan Whitehorn 			break;
12842118f387SNathan Whitehorn 		}
12852118f387SNathan Whitehorn 
1286f36a5e0fSNathan Whitehorn 	/* If so, destroy all children */
12872118f387SNathan Whitehorn 	if (geom != NULL) {
1288f36a5e0fSNathan Whitehorn 		gpart_destroy(geom);
1289f36a5e0fSNathan Whitehorn 
1290f36a5e0fSNathan Whitehorn 		/* If this is a partition, revert it, so it can be deleted */
12912118f387SNathan Whitehorn 		if (is_partition) {
1292f36a5e0fSNathan Whitehorn 			r = gctl_get_handle();
1293f36a5e0fSNathan Whitehorn 			gctl_ro_param(r, "class", -1, "PART");
1294f36a5e0fSNathan Whitehorn 			gctl_ro_param(r, "arg0", -1, geom->lg_name);
1295f36a5e0fSNathan Whitehorn 			gctl_ro_param(r, "verb", -1, "undo");
1296f36a5e0fSNathan Whitehorn 			gctl_issue(r); /* Ignore non-fatal errors */
1297f36a5e0fSNathan Whitehorn 			gctl_free(r);
12982118f387SNathan Whitehorn 		}
12992118f387SNathan Whitehorn 	}
13002118f387SNathan Whitehorn 
13012118f387SNathan Whitehorn 	/*
13022118f387SNathan Whitehorn 	 * If this is not a partition, see if that is a problem, complain if
13032118f387SNathan Whitehorn 	 * necessary, and return always, since we need not do anything further,
13042118f387SNathan Whitehorn 	 * error or no.
13052118f387SNathan Whitehorn 	 */
13062118f387SNathan Whitehorn 	if (!is_partition) {
13072118f387SNathan Whitehorn 		if (geom == NULL)
13082118f387SNathan Whitehorn 			dialog_msgbox("Error",
13092118f387SNathan Whitehorn 			    "Only partitions can be deleted.", 0, 0, TRUE);
13102118f387SNathan Whitehorn 		return;
13112118f387SNathan Whitehorn 	}
13122118f387SNathan Whitehorn 
13132118f387SNathan Whitehorn 	r = gctl_get_handle();
13142118f387SNathan Whitehorn 	gctl_ro_param(r, "class", -1, pp->lg_geom->lg_class->lg_name);
13152118f387SNathan Whitehorn 	gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name);
13162118f387SNathan Whitehorn 	gctl_ro_param(r, "flags", -1, GPART_FLAGS);
13172118f387SNathan Whitehorn 	gctl_ro_param(r, "verb", -1, "delete");
13182118f387SNathan Whitehorn 
13192118f387SNathan Whitehorn 	LIST_FOREACH(gc, &pp->lg_config, lg_config) {
13202118f387SNathan Whitehorn 		if (strcmp(gc->lg_name, "index") == 0) {
13212118f387SNathan Whitehorn 			idx = atoi(gc->lg_val);
13222118f387SNathan Whitehorn 			gctl_ro_param(r, "index", sizeof(idx), &idx);
13232118f387SNathan Whitehorn 			break;
13242118f387SNathan Whitehorn 		}
13252118f387SNathan Whitehorn 	}
13262118f387SNathan Whitehorn 
13272118f387SNathan Whitehorn 	errstr = gctl_issue(r);
13282118f387SNathan Whitehorn 	if (errstr != NULL && errstr[0] != '\0') {
13292118f387SNathan Whitehorn 		gpart_show_error("Error", NULL, errstr);
13302118f387SNathan Whitehorn 		gctl_free(r);
13312118f387SNathan Whitehorn 		return;
13322118f387SNathan Whitehorn 	}
13332118f387SNathan Whitehorn 
13342118f387SNathan Whitehorn 	gctl_free(r);
13352118f387SNathan Whitehorn 
13362118f387SNathan Whitehorn 	delete_part_metadata(pp->lg_name);
13372118f387SNathan Whitehorn }
13382118f387SNathan Whitehorn 
13392118f387SNathan Whitehorn void
13402118f387SNathan Whitehorn gpart_revert_all(struct gmesh *mesh)
13412118f387SNathan Whitehorn {
13422118f387SNathan Whitehorn 	struct gclass *classp;
13432118f387SNathan Whitehorn 	struct gconfig *gc;
13442118f387SNathan Whitehorn 	struct ggeom *gp;
13452118f387SNathan Whitehorn 	struct gctl_req *r;
13462118f387SNathan Whitehorn 	const char *modified;
13472118f387SNathan Whitehorn 
13482118f387SNathan Whitehorn 	LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
13492118f387SNathan Whitehorn 		if (strcmp(classp->lg_name, "PART") == 0)
13502118f387SNathan Whitehorn 			break;
13512118f387SNathan Whitehorn 	}
13522118f387SNathan Whitehorn 
13532118f387SNathan Whitehorn 	if (strcmp(classp->lg_name, "PART") != 0) {
13542118f387SNathan Whitehorn 		dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE);
13552118f387SNathan Whitehorn 		return;
13562118f387SNathan Whitehorn 	}
13572118f387SNathan Whitehorn 
13582118f387SNathan Whitehorn 	LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
13592118f387SNathan Whitehorn 		modified = "true"; /* XXX: If we don't know (kernel too old),
13602118f387SNathan Whitehorn 				    * assume there are modifications. */
13612118f387SNathan Whitehorn 		LIST_FOREACH(gc, &gp->lg_config, lg_config) {
13622118f387SNathan Whitehorn 			if (strcmp(gc->lg_name, "modified") == 0) {
13632118f387SNathan Whitehorn 				modified = gc->lg_val;
13642118f387SNathan Whitehorn 				break;
13652118f387SNathan Whitehorn 			}
13662118f387SNathan Whitehorn 		}
13672118f387SNathan Whitehorn 
13682118f387SNathan Whitehorn 		if (strcmp(modified, "false") == 0)
13692118f387SNathan Whitehorn 			continue;
13702118f387SNathan Whitehorn 
13712118f387SNathan Whitehorn 		r = gctl_get_handle();
13722118f387SNathan Whitehorn 		gctl_ro_param(r, "class", -1, "PART");
13732118f387SNathan Whitehorn 		gctl_ro_param(r, "arg0", -1, gp->lg_name);
13742118f387SNathan Whitehorn 		gctl_ro_param(r, "verb", -1, "undo");
13752118f387SNathan Whitehorn 
13767899c6c1SNathan Whitehorn 		gctl_issue(r);
13772118f387SNathan Whitehorn 		gctl_free(r);
13782118f387SNathan Whitehorn 	}
13792118f387SNathan Whitehorn }
13802118f387SNathan Whitehorn 
13812118f387SNathan Whitehorn void
13822118f387SNathan Whitehorn gpart_commit(struct gmesh *mesh)
13832118f387SNathan Whitehorn {
13842118f387SNathan Whitehorn 	struct partition_metadata *md;
13852118f387SNathan Whitehorn 	struct gclass *classp;
13862118f387SNathan Whitehorn 	struct ggeom *gp;
13872118f387SNathan Whitehorn 	struct gconfig *gc;
13882118f387SNathan Whitehorn 	struct gconsumer *cp;
13892118f387SNathan Whitehorn 	struct gprovider *pp;
13902118f387SNathan Whitehorn 	struct gctl_req *r;
13912118f387SNathan Whitehorn 	const char *errstr;
13922118f387SNathan Whitehorn 	const char *modified;
13936e15678aSNathan Whitehorn 	const char *rootfs;
13942118f387SNathan Whitehorn 
13952118f387SNathan Whitehorn 	LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
13962118f387SNathan Whitehorn 		if (strcmp(classp->lg_name, "PART") == 0)
13972118f387SNathan Whitehorn 			break;
13982118f387SNathan Whitehorn 	}
13992118f387SNathan Whitehorn 
14006e15678aSNathan Whitehorn 	/* Figure out what filesystem / uses */
14016e15678aSNathan Whitehorn 	rootfs = "ufs"; /* Assume ufs if nothing else present */
14026e15678aSNathan Whitehorn 	TAILQ_FOREACH(md, &part_metadata, metadata) {
14036e15678aSNathan Whitehorn 		if (md->fstab != NULL && strcmp(md->fstab->fs_file, "/") == 0) {
14046e15678aSNathan Whitehorn 			rootfs = md->fstab->fs_vfstype;
14056e15678aSNathan Whitehorn 			break;
14066e15678aSNathan Whitehorn 		}
14076e15678aSNathan Whitehorn 	}
14086e15678aSNathan Whitehorn 
14092118f387SNathan Whitehorn 	if (strcmp(classp->lg_name, "PART") != 0) {
14102118f387SNathan Whitehorn 		dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE);
14112118f387SNathan Whitehorn 		return;
14122118f387SNathan Whitehorn 	}
14132118f387SNathan Whitehorn 
14142118f387SNathan Whitehorn 	LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
14152118f387SNathan Whitehorn 		modified = "true"; /* XXX: If we don't know (kernel too old),
14162118f387SNathan Whitehorn 				    * assume there are modifications. */
14172118f387SNathan Whitehorn 		LIST_FOREACH(gc, &gp->lg_config, lg_config) {
14182118f387SNathan Whitehorn 			if (strcmp(gc->lg_name, "modified") == 0) {
14192118f387SNathan Whitehorn 				modified = gc->lg_val;
14202118f387SNathan Whitehorn 				break;
14212118f387SNathan Whitehorn 			}
14222118f387SNathan Whitehorn 		}
14232118f387SNathan Whitehorn 
14242118f387SNathan Whitehorn 		if (strcmp(modified, "false") == 0)
14252118f387SNathan Whitehorn 			continue;
14262118f387SNathan Whitehorn 
14272118f387SNathan Whitehorn 		/* Add bootcode if necessary, before the commit */
14282118f387SNathan Whitehorn 		md = get_part_metadata(gp->lg_name, 0);
14292118f387SNathan Whitehorn 		if (md != NULL && md->bootcode)
14302118f387SNathan Whitehorn 			gpart_bootcode(gp);
14312118f387SNathan Whitehorn 
14322118f387SNathan Whitehorn 		/* Now install partcode on its partitions, if necessary */
14332118f387SNathan Whitehorn 		LIST_FOREACH(pp, &gp->lg_provider, lg_provider) {
14342118f387SNathan Whitehorn 			md = get_part_metadata(pp->lg_name, 0);
14352118f387SNathan Whitehorn 			if (md == NULL || !md->bootcode)
14362118f387SNathan Whitehorn 				continue;
14372118f387SNathan Whitehorn 
14382118f387SNathan Whitehorn 			/* Mark this partition active if that's required */
14392118f387SNathan Whitehorn 			gpart_activate(pp);
14402118f387SNathan Whitehorn 
14412118f387SNathan Whitehorn 			/* Check if the partition has sub-partitions */
14422118f387SNathan Whitehorn 			LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers)
14432118f387SNathan Whitehorn 				if (strcmp(cp->lg_geom->lg_class->lg_name,
14442118f387SNathan Whitehorn 				    "PART") == 0)
14452118f387SNathan Whitehorn 					break;
14462118f387SNathan Whitehorn 
14472118f387SNathan Whitehorn 			if (cp == NULL) /* No sub-partitions */
14486e15678aSNathan Whitehorn 				gpart_partcode(pp, rootfs);
14492118f387SNathan Whitehorn 		}
14502118f387SNathan Whitehorn 
14512118f387SNathan Whitehorn 		r = gctl_get_handle();
14522118f387SNathan Whitehorn 		gctl_ro_param(r, "class", -1, "PART");
14532118f387SNathan Whitehorn 		gctl_ro_param(r, "arg0", -1, gp->lg_name);
14542118f387SNathan Whitehorn 		gctl_ro_param(r, "verb", -1, "commit");
14552118f387SNathan Whitehorn 
14562118f387SNathan Whitehorn 		errstr = gctl_issue(r);
14572118f387SNathan Whitehorn 		if (errstr != NULL && errstr[0] != '\0')
14582118f387SNathan Whitehorn 			gpart_show_error("Error", NULL, errstr);
14592118f387SNathan Whitehorn 		gctl_free(r);
14602118f387SNathan Whitehorn 	}
14612118f387SNathan Whitehorn }
14622118f387SNathan Whitehorn 
1463