12118f387SNathan Whitehorn /*- 22118f387SNathan Whitehorn * Copyright (c) 2011 Nathan Whitehorn 32118f387SNathan Whitehorn * All rights reserved. 42118f387SNathan Whitehorn * 52118f387SNathan Whitehorn * Redistribution and use in source and binary forms, with or without 62118f387SNathan Whitehorn * modification, are permitted provided that the following conditions 72118f387SNathan Whitehorn * are met: 82118f387SNathan Whitehorn * 1. Redistributions of source code must retain the above copyright 92118f387SNathan Whitehorn * notice, this list of conditions and the following disclaimer. 102118f387SNathan Whitehorn * 2. Redistributions in binary form must reproduce the above copyright 112118f387SNathan Whitehorn * notice, this list of conditions and the following disclaimer in the 122118f387SNathan Whitehorn * documentation and/or other materials provided with the distribution. 132118f387SNathan Whitehorn * 142118f387SNathan Whitehorn * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 152118f387SNathan Whitehorn * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 162118f387SNathan Whitehorn * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 172118f387SNathan Whitehorn * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 182118f387SNathan Whitehorn * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 192118f387SNathan Whitehorn * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 202118f387SNathan Whitehorn * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 212118f387SNathan Whitehorn * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 222118f387SNathan Whitehorn * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 232118f387SNathan Whitehorn * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 242118f387SNathan Whitehorn * SUCH DAMAGE. 252118f387SNathan Whitehorn * 262118f387SNathan Whitehorn * $FreeBSD$ 272118f387SNathan Whitehorn */ 282118f387SNathan Whitehorn 292118f387SNathan Whitehorn #include <sys/param.h> 306e15678aSNathan Whitehorn #include <sys/stat.h> 312118f387SNathan Whitehorn #include <errno.h> 322118f387SNathan Whitehorn #include <libutil.h> 332118f387SNathan Whitehorn #include <inttypes.h> 342118f387SNathan Whitehorn 352118f387SNathan Whitehorn #include <libgeom.h> 362118f387SNathan Whitehorn #include <dialog.h> 372118f387SNathan Whitehorn #include <dlg_keys.h> 382118f387SNathan Whitehorn 392118f387SNathan Whitehorn #include "partedit.h" 402118f387SNathan Whitehorn 412118f387SNathan Whitehorn #define GPART_FLAGS "x" /* Do not commit changes by default */ 422118f387SNathan Whitehorn 432118f387SNathan Whitehorn static void 442118f387SNathan Whitehorn gpart_show_error(const char *title, const char *explanation, const char *errstr) 452118f387SNathan Whitehorn { 462118f387SNathan Whitehorn char *errmsg; 472118f387SNathan Whitehorn char message[512]; 482118f387SNathan Whitehorn int error; 492118f387SNathan Whitehorn 502118f387SNathan Whitehorn if (explanation == NULL) 512118f387SNathan Whitehorn explanation = ""; 522118f387SNathan Whitehorn 532118f387SNathan Whitehorn error = strtol(errstr, &errmsg, 0); 542118f387SNathan Whitehorn if (errmsg != errstr) { 552118f387SNathan Whitehorn while (errmsg[0] == ' ') 562118f387SNathan Whitehorn errmsg++; 572118f387SNathan Whitehorn if (errmsg[0] != '\0') 582118f387SNathan Whitehorn sprintf(message, "%s%s. %s", explanation, 592118f387SNathan Whitehorn strerror(error), errmsg); 602118f387SNathan Whitehorn else 612118f387SNathan Whitehorn sprintf(message, "%s%s", explanation, strerror(error)); 622118f387SNathan Whitehorn } else { 632118f387SNathan Whitehorn sprintf(message, "%s%s", explanation, errmsg); 642118f387SNathan Whitehorn } 652118f387SNathan Whitehorn 662118f387SNathan Whitehorn dialog_msgbox(title, message, 0, 0, TRUE); 672118f387SNathan Whitehorn } 682118f387SNathan Whitehorn 69c67f41d0SNathan Whitehorn static int 70c67f41d0SNathan Whitehorn scheme_supports_labels(const char *scheme) 71c67f41d0SNathan Whitehorn { 72c67f41d0SNathan Whitehorn if (strcmp(scheme, "APM") == 0) 73c67f41d0SNathan Whitehorn return (1); 74c67f41d0SNathan Whitehorn if (strcmp(scheme, "GPT") == 0) 75c67f41d0SNathan Whitehorn return (1); 76c67f41d0SNathan Whitehorn 77c67f41d0SNathan Whitehorn return (0); 78c67f41d0SNathan Whitehorn } 79c67f41d0SNathan Whitehorn 8050de5d07SNathan Whitehorn static void 8150de5d07SNathan Whitehorn newfs_command(const char *fstype, char *command, int use_default) 8250de5d07SNathan Whitehorn { 8350de5d07SNathan Whitehorn if (strcmp(fstype, "freebsd-ufs") == 0) { 8450de5d07SNathan Whitehorn int i; 8550de5d07SNathan Whitehorn DIALOG_LISTITEM items[] = { 8650de5d07SNathan Whitehorn {"UFS1", "UFS Version 1", 8750de5d07SNathan Whitehorn "Use version 1 of the UFS file system instead " 8850de5d07SNathan Whitehorn "of version 2 (not recommended)", 0 }, 8950de5d07SNathan Whitehorn {"SU", "Softupdates", 9050de5d07SNathan Whitehorn "Enable softupdates (default)", 1 }, 9150de5d07SNathan Whitehorn {"SUJ", "Softupdates journaling", 9250de5d07SNathan Whitehorn "Enable file system journaling (default - " 9350de5d07SNathan Whitehorn "turn off for SSDs)", 1 }, 9450de5d07SNathan Whitehorn {"TRIM", "Enable SSD TRIM support", 9550de5d07SNathan Whitehorn "Enable TRIM support, useful on solid-state drives", 9650de5d07SNathan Whitehorn 0 }, 9750de5d07SNathan Whitehorn }; 9850de5d07SNathan Whitehorn 9950de5d07SNathan Whitehorn if (!use_default) { 10050de5d07SNathan Whitehorn int choice; 10150de5d07SNathan Whitehorn choice = dlg_checklist("UFS Options", "", 0, 0, 0, 10263128851SMarcelo Araujo nitems(items), items, NULL, 10350de5d07SNathan Whitehorn FLAG_CHECK, &i); 10450de5d07SNathan Whitehorn if (choice == 1) /* Cancel */ 10550de5d07SNathan Whitehorn return; 10650de5d07SNathan Whitehorn } 10750de5d07SNathan Whitehorn 10850de5d07SNathan Whitehorn strcpy(command, "newfs "); 10963128851SMarcelo Araujo for (i = 0; i < (int)nitems(items); i++) { 11050de5d07SNathan Whitehorn if (items[i].state == 0) 11150de5d07SNathan Whitehorn continue; 11250de5d07SNathan Whitehorn if (strcmp(items[i].name, "UFS1") == 0) 11350de5d07SNathan Whitehorn strcat(command, "-O1 "); 11450de5d07SNathan Whitehorn else if (strcmp(items[i].name, "SU") == 0) 11550de5d07SNathan Whitehorn strcat(command, "-U "); 11650de5d07SNathan Whitehorn else if (strcmp(items[i].name, "SUJ") == 0) 11750de5d07SNathan Whitehorn strcat(command, "-j "); 11850de5d07SNathan Whitehorn else if (strcmp(items[i].name, "TRIM") == 0) 11950de5d07SNathan Whitehorn strcat(command, "-t "); 12050de5d07SNathan Whitehorn } 1216e15678aSNathan Whitehorn } else if (strcmp(fstype, "freebsd-zfs") == 0) { 1226e15678aSNathan Whitehorn int i; 1236e15678aSNathan Whitehorn DIALOG_LISTITEM items[] = { 1246e15678aSNathan Whitehorn {"fletcher4", "checksum algorithm: fletcher4", 1256e15678aSNathan Whitehorn "Use fletcher4 for data integrity checking. " 1266e15678aSNathan Whitehorn "(default)", 1 }, 1276e15678aSNathan Whitehorn {"fletcher2", "checksum algorithm: fletcher2", 1286e15678aSNathan Whitehorn "Use fletcher2 for data integrity checking. " 1296e15678aSNathan Whitehorn "(not recommended)", 0 }, 1306e15678aSNathan Whitehorn {"sha256", "checksum algorithm: sha256", 1316e15678aSNathan Whitehorn "Use sha256 for data integrity checking. " 1326e15678aSNathan Whitehorn "(not recommended)", 0 }, 1336e15678aSNathan Whitehorn {"atime", "Update atimes for files", 1346e15678aSNathan Whitehorn "Disable atime update", 0 }, 1356e15678aSNathan Whitehorn }; 1366e15678aSNathan Whitehorn 1376e15678aSNathan Whitehorn if (!use_default) { 1386e15678aSNathan Whitehorn int choice; 1396e15678aSNathan Whitehorn choice = dlg_checklist("ZFS Options", "", 0, 0, 0, 14063128851SMarcelo Araujo nitems(items), items, NULL, 1416e15678aSNathan Whitehorn FLAG_CHECK, &i); 1426e15678aSNathan Whitehorn if (choice == 1) /* Cancel */ 1436e15678aSNathan Whitehorn return; 1446e15678aSNathan Whitehorn } 1456e15678aSNathan Whitehorn 1466e15678aSNathan Whitehorn strcpy(command, "zpool create -f -m none "); 1476e15678aSNathan Whitehorn if (getenv("BSDINSTALL_TMPBOOT") != NULL) { 1486e15678aSNathan Whitehorn char zfsboot_path[MAXPATHLEN]; 1496c546e77SConrad Meyer snprintf(zfsboot_path, sizeof(zfsboot_path), "%s/zfs", 1506e15678aSNathan Whitehorn getenv("BSDINSTALL_TMPBOOT")); 1516e15678aSNathan Whitehorn mkdir(zfsboot_path, S_IRWXU | S_IRGRP | S_IXGRP | 1526e15678aSNathan Whitehorn S_IROTH | S_IXOTH); 1536e15678aSNathan Whitehorn sprintf(command, "%s -o cachefile=%s/zpool.cache ", 1546e15678aSNathan Whitehorn command, zfsboot_path); 1556e15678aSNathan Whitehorn } 15663128851SMarcelo Araujo for (i = 0; i < (int)nitems(items); i++) { 1576e15678aSNathan Whitehorn if (items[i].state == 0) 1586e15678aSNathan Whitehorn continue; 1596e15678aSNathan Whitehorn if (strcmp(items[i].name, "fletcher4") == 0) 1606e15678aSNathan Whitehorn strcat(command, "-O checksum=fletcher4 "); 1616e15678aSNathan Whitehorn else if (strcmp(items[i].name, "fletcher2") == 0) 1626e15678aSNathan Whitehorn strcat(command, "-O checksum=fletcher2 "); 1636e15678aSNathan Whitehorn else if (strcmp(items[i].name, "sha256") == 0) 1646e15678aSNathan Whitehorn strcat(command, "-O checksum=sha256 "); 1656e15678aSNathan Whitehorn else if (strcmp(items[i].name, "atime") == 0) 1666e15678aSNathan Whitehorn strcat(command, "-O atime=off "); 1676e15678aSNathan Whitehorn } 16850de5d07SNathan Whitehorn } else if (strcmp(fstype, "fat32") == 0 || strcmp(fstype, "efi") == 0) { 16950de5d07SNathan Whitehorn int i; 17050de5d07SNathan Whitehorn DIALOG_LISTITEM items[] = { 17150de5d07SNathan Whitehorn {"FAT32", "FAT Type 32", 17250de5d07SNathan Whitehorn "Create a FAT32 filesystem (default)", 1 }, 17350de5d07SNathan Whitehorn {"FAT16", "FAT Type 16", 17450de5d07SNathan Whitehorn "Create a FAT16 filesystem", 0 }, 17550de5d07SNathan Whitehorn {"FAT12", "FAT Type 12", 17650de5d07SNathan Whitehorn "Create a FAT12 filesystem", 0 }, 17750de5d07SNathan Whitehorn }; 17850de5d07SNathan Whitehorn 17950de5d07SNathan Whitehorn if (!use_default) { 18050de5d07SNathan Whitehorn int choice; 18150de5d07SNathan Whitehorn choice = dlg_checklist("FAT Options", "", 0, 0, 0, 18263128851SMarcelo Araujo nitems(items), items, NULL, 18350de5d07SNathan Whitehorn FLAG_RADIO, &i); 18450de5d07SNathan Whitehorn if (choice == 1) /* Cancel */ 18550de5d07SNathan Whitehorn return; 18650de5d07SNathan Whitehorn } 18750de5d07SNathan Whitehorn 18850de5d07SNathan Whitehorn strcpy(command, "newfs_msdos "); 18963128851SMarcelo Araujo for (i = 0; i < (int)nitems(items); i++) { 19050de5d07SNathan Whitehorn if (items[i].state == 0) 19150de5d07SNathan Whitehorn continue; 19250de5d07SNathan Whitehorn if (strcmp(items[i].name, "FAT32") == 0) 19350de5d07SNathan Whitehorn strcat(command, "-F 32 "); 19450de5d07SNathan Whitehorn else if (strcmp(items[i].name, "FAT16") == 0) 19550de5d07SNathan Whitehorn strcat(command, "-F 16 "); 196ae2e7abfSNathan Whitehorn else if (strcmp(items[i].name, "FAT12") == 0) 19750de5d07SNathan Whitehorn strcat(command, "-F 12 "); 19850de5d07SNathan Whitehorn } 19950de5d07SNathan Whitehorn } else { 20050de5d07SNathan Whitehorn if (!use_default) 20150de5d07SNathan Whitehorn dialog_msgbox("Error", "No configurable options exist " 20250de5d07SNathan Whitehorn "for this filesystem.", 0, 0, TRUE); 20350de5d07SNathan Whitehorn command[0] = '\0'; 20450de5d07SNathan Whitehorn } 20550de5d07SNathan Whitehorn } 20650de5d07SNathan Whitehorn 2077059fa6fSAllan Jude const char * 2087059fa6fSAllan Jude choose_part_type(const char *def_scheme) 2092118f387SNathan Whitehorn { 2102118f387SNathan Whitehorn int cancel, choice; 2117059fa6fSAllan Jude const char *scheme = NULL; 2122118f387SNathan Whitehorn 2132118f387SNathan Whitehorn DIALOG_LISTITEM items[] = { 2142118f387SNathan Whitehorn {"APM", "Apple Partition Map", 2152118f387SNathan Whitehorn "Bootable on PowerPC Apple Hardware", 0 }, 2162118f387SNathan Whitehorn {"BSD", "BSD Labels", 2172118f387SNathan Whitehorn "Bootable on most x86 systems", 0 }, 2182118f387SNathan Whitehorn {"GPT", "GUID Partition Table", 2190fbcaf76SWojciech Macek "Bootable on most x86 systems and EFI aware ARM64", 0 }, 2202118f387SNathan Whitehorn {"MBR", "DOS Partitions", 2212118f387SNathan Whitehorn "Bootable on most x86 systems", 0 }, 2222118f387SNathan Whitehorn {"VTOC8", "Sun VTOC8 Partition Table", 2232118f387SNathan Whitehorn "Bootable on Sun SPARC systems", 0 }, 2242118f387SNathan Whitehorn }; 2252118f387SNathan Whitehorn 2267059fa6fSAllan Jude parttypemenu: 2277059fa6fSAllan Jude dialog_vars.default_item = __DECONST(char *, def_scheme); 2282118f387SNathan Whitehorn cancel = dlg_menu("Partition Scheme", 2292118f387SNathan Whitehorn "Select a partition scheme for this volume:", 0, 0, 0, 23063128851SMarcelo Araujo nitems(items), items, &choice, NULL); 2312118f387SNathan Whitehorn dialog_vars.default_item = NULL; 2322118f387SNathan Whitehorn 2332118f387SNathan Whitehorn if (cancel) 2347059fa6fSAllan Jude return NULL; 2352118f387SNathan Whitehorn 2362118f387SNathan Whitehorn if (!is_scheme_bootable(items[choice].name)) { 2372118f387SNathan Whitehorn char message[512]; 2382118f387SNathan Whitehorn sprintf(message, "This partition scheme (%s) is not " 2392118f387SNathan Whitehorn "bootable on this platform. Are you sure you want " 2402118f387SNathan Whitehorn "to proceed?", items[choice].name); 2412118f387SNathan Whitehorn dialog_vars.defaultno = TRUE; 2422118f387SNathan Whitehorn cancel = dialog_yesno("Warning", message, 0, 0); 2432118f387SNathan Whitehorn dialog_vars.defaultno = FALSE; 2442118f387SNathan Whitehorn if (cancel) /* cancel */ 2457059fa6fSAllan Jude goto parttypemenu; 2462118f387SNathan Whitehorn } 2472118f387SNathan Whitehorn 2482118f387SNathan Whitehorn scheme = items[choice].name; 2497059fa6fSAllan Jude 2507059fa6fSAllan Jude return scheme; 2517059fa6fSAllan Jude } 2527059fa6fSAllan Jude 2537059fa6fSAllan Jude int 2547059fa6fSAllan Jude gpart_partition(const char *lg_name, const char *scheme) 2557059fa6fSAllan Jude { 2567059fa6fSAllan Jude int cancel; 2577059fa6fSAllan Jude struct gctl_req *r; 2587059fa6fSAllan Jude const char *errstr; 2597059fa6fSAllan Jude 2607059fa6fSAllan Jude schememenu: 2617059fa6fSAllan Jude if (scheme == NULL) { 2627059fa6fSAllan Jude scheme = choose_part_type(default_scheme()); 2637059fa6fSAllan Jude 2647059fa6fSAllan Jude if (scheme == NULL) 2657059fa6fSAllan Jude return (-1); 2667059fa6fSAllan Jude 2677059fa6fSAllan Jude if (!is_scheme_bootable(scheme)) { 2687059fa6fSAllan Jude char message[512]; 2697059fa6fSAllan Jude sprintf(message, "This partition scheme (%s) is not " 2707059fa6fSAllan Jude "bootable on this platform. Are you sure you want " 2717059fa6fSAllan Jude "to proceed?", scheme); 2727059fa6fSAllan Jude dialog_vars.defaultno = TRUE; 2737059fa6fSAllan Jude cancel = dialog_yesno("Warning", message, 0, 0); 2747059fa6fSAllan Jude dialog_vars.defaultno = FALSE; 2757059fa6fSAllan Jude if (cancel) { /* cancel */ 2767059fa6fSAllan Jude /* Reset scheme so user can choose another */ 2777059fa6fSAllan Jude scheme = NULL; 2787059fa6fSAllan Jude goto schememenu; 2797059fa6fSAllan Jude } 2807059fa6fSAllan Jude } 2812118f387SNathan Whitehorn } 2822118f387SNathan Whitehorn 2832118f387SNathan Whitehorn r = gctl_get_handle(); 2842118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 2852118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, lg_name); 2862118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 2872118f387SNathan Whitehorn gctl_ro_param(r, "scheme", -1, scheme); 2882118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "create"); 2892118f387SNathan Whitehorn 2902118f387SNathan Whitehorn errstr = gctl_issue(r); 2912118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') { 2922118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 2932118f387SNathan Whitehorn gctl_free(r); 2942118f387SNathan Whitehorn scheme = NULL; 2952118f387SNathan Whitehorn goto schememenu; 2962118f387SNathan Whitehorn } 2972118f387SNathan Whitehorn gctl_free(r); 2982118f387SNathan Whitehorn 2992118f387SNathan Whitehorn if (bootcode_path(scheme) != NULL) 3002118f387SNathan Whitehorn get_part_metadata(lg_name, 1)->bootcode = 1; 3012118f387SNathan Whitehorn return (0); 3022118f387SNathan Whitehorn } 3032118f387SNathan Whitehorn 3042118f387SNathan Whitehorn static void 3052118f387SNathan Whitehorn gpart_activate(struct gprovider *pp) 3062118f387SNathan Whitehorn { 3072118f387SNathan Whitehorn struct gconfig *gc; 3082118f387SNathan Whitehorn struct gctl_req *r; 3092118f387SNathan Whitehorn const char *errstr, *scheme; 3102118f387SNathan Whitehorn const char *attribute = NULL; 3112118f387SNathan Whitehorn intmax_t idx; 3122118f387SNathan Whitehorn 3132118f387SNathan Whitehorn /* 3142118f387SNathan Whitehorn * Some partition schemes need this partition to be marked 'active' 3152118f387SNathan Whitehorn * for it to be bootable. 3162118f387SNathan Whitehorn */ 3172118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) { 3182118f387SNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) { 3192118f387SNathan Whitehorn scheme = gc->lg_val; 3202118f387SNathan Whitehorn break; 3212118f387SNathan Whitehorn } 3222118f387SNathan Whitehorn } 3232118f387SNathan Whitehorn 3242b375b4eSYoshihiro Takahashi if (strcmp(scheme, "MBR") == 0 || strcmp(scheme, "EBR") == 0) 3252118f387SNathan Whitehorn attribute = "active"; 3262118f387SNathan Whitehorn else 3272118f387SNathan Whitehorn return; 3282118f387SNathan Whitehorn 3292118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) { 3302118f387SNathan Whitehorn if (strcmp(gc->lg_name, "index") == 0) { 3312118f387SNathan Whitehorn idx = atoi(gc->lg_val); 3322118f387SNathan Whitehorn break; 3332118f387SNathan Whitehorn } 3342118f387SNathan Whitehorn } 3352118f387SNathan Whitehorn 3362118f387SNathan Whitehorn r = gctl_get_handle(); 3372118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 3382118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name); 3392118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "set"); 3402118f387SNathan Whitehorn gctl_ro_param(r, "attrib", -1, attribute); 3412118f387SNathan Whitehorn gctl_ro_param(r, "index", sizeof(idx), &idx); 3422118f387SNathan Whitehorn 3432118f387SNathan Whitehorn errstr = gctl_issue(r); 3442118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') 3452118f387SNathan Whitehorn gpart_show_error("Error", "Error marking partition active:", 3462118f387SNathan Whitehorn errstr); 3472118f387SNathan Whitehorn gctl_free(r); 3482118f387SNathan Whitehorn } 3492118f387SNathan Whitehorn 3507059fa6fSAllan Jude void 3517059fa6fSAllan Jude gpart_set_root(const char *lg_name, const char *attribute) 3527059fa6fSAllan Jude { 3537059fa6fSAllan Jude struct gctl_req *r; 3547059fa6fSAllan Jude const char *errstr; 3557059fa6fSAllan Jude 3567059fa6fSAllan Jude r = gctl_get_handle(); 3577059fa6fSAllan Jude gctl_ro_param(r, "class", -1, "PART"); 3587059fa6fSAllan Jude gctl_ro_param(r, "arg0", -1, lg_name); 3597059fa6fSAllan Jude gctl_ro_param(r, "flags", -1, "C"); 3607059fa6fSAllan Jude gctl_ro_param(r, "verb", -1, "set"); 3617059fa6fSAllan Jude gctl_ro_param(r, "attrib", -1, attribute); 3627059fa6fSAllan Jude 3637059fa6fSAllan Jude errstr = gctl_issue(r); 3647059fa6fSAllan Jude if (errstr != NULL && errstr[0] != '\0') 3657059fa6fSAllan Jude gpart_show_error("Error", "Error setting parameter on disk:", 3667059fa6fSAllan Jude errstr); 3677059fa6fSAllan Jude gctl_free(r); 3687059fa6fSAllan Jude } 3697059fa6fSAllan Jude 3702118f387SNathan Whitehorn static void 3712118f387SNathan Whitehorn gpart_bootcode(struct ggeom *gp) 3722118f387SNathan Whitehorn { 3732118f387SNathan Whitehorn const char *bootcode; 3742118f387SNathan Whitehorn struct gconfig *gc; 3752118f387SNathan Whitehorn struct gctl_req *r; 3762118f387SNathan Whitehorn const char *errstr, *scheme; 3772118f387SNathan Whitehorn uint8_t *boot; 3782118f387SNathan Whitehorn size_t bootsize, bytes; 3792118f387SNathan Whitehorn int bootfd; 3802118f387SNathan Whitehorn 3812118f387SNathan Whitehorn /* 3822118f387SNathan Whitehorn * Write default bootcode to the newly partitioned disk, if that 3832118f387SNathan Whitehorn * applies on this platform. 3842118f387SNathan Whitehorn */ 3852118f387SNathan Whitehorn LIST_FOREACH(gc, &gp->lg_config, lg_config) { 3862118f387SNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) { 3872118f387SNathan Whitehorn scheme = gc->lg_val; 3882118f387SNathan Whitehorn break; 3892118f387SNathan Whitehorn } 3902118f387SNathan Whitehorn } 3912118f387SNathan Whitehorn 3922118f387SNathan Whitehorn bootcode = bootcode_path(scheme); 3932118f387SNathan Whitehorn if (bootcode == NULL) 3942118f387SNathan Whitehorn return; 3952118f387SNathan Whitehorn 3962118f387SNathan Whitehorn bootfd = open(bootcode, O_RDONLY); 397c725e3efSKevin Lo if (bootfd < 0) { 3982118f387SNathan Whitehorn dialog_msgbox("Bootcode Error", strerror(errno), 0, 0, 3992118f387SNathan Whitehorn TRUE); 4002118f387SNathan Whitehorn return; 4012118f387SNathan Whitehorn } 4022118f387SNathan Whitehorn 4032118f387SNathan Whitehorn bootsize = lseek(bootfd, 0, SEEK_END); 4042118f387SNathan Whitehorn boot = malloc(bootsize); 4052118f387SNathan Whitehorn lseek(bootfd, 0, SEEK_SET); 4062118f387SNathan Whitehorn bytes = 0; 4072118f387SNathan Whitehorn while (bytes < bootsize) 4082118f387SNathan Whitehorn bytes += read(bootfd, boot + bytes, bootsize - bytes); 4092118f387SNathan Whitehorn close(bootfd); 4102118f387SNathan Whitehorn 4112118f387SNathan Whitehorn r = gctl_get_handle(); 4122118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 4132118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, gp->lg_name); 4142118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "bootcode"); 4152118f387SNathan Whitehorn gctl_ro_param(r, "bootcode", bootsize, boot); 4162118f387SNathan Whitehorn 4172118f387SNathan Whitehorn errstr = gctl_issue(r); 4182118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') 4192118f387SNathan Whitehorn gpart_show_error("Bootcode Error", NULL, errstr); 4202118f387SNathan Whitehorn gctl_free(r); 4212118f387SNathan Whitehorn free(boot); 4222118f387SNathan Whitehorn } 4232118f387SNathan Whitehorn 4242118f387SNathan Whitehorn static void 4256e15678aSNathan Whitehorn gpart_partcode(struct gprovider *pp, const char *fstype) 4262118f387SNathan Whitehorn { 4272118f387SNathan Whitehorn struct gconfig *gc; 4282118f387SNathan Whitehorn const char *scheme; 4292118f387SNathan Whitehorn const char *indexstr; 4302118f387SNathan Whitehorn char message[255], command[255]; 4312118f387SNathan Whitehorn 4322118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_geom->lg_config, lg_config) { 4332118f387SNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) { 4342118f387SNathan Whitehorn scheme = gc->lg_val; 4352118f387SNathan Whitehorn break; 4362118f387SNathan Whitehorn } 4372118f387SNathan Whitehorn } 4382118f387SNathan Whitehorn 4392118f387SNathan Whitehorn /* Make sure this partition scheme needs partcode on this platform */ 4406e15678aSNathan Whitehorn if (partcode_path(scheme, fstype) == NULL) 4412118f387SNathan Whitehorn return; 4422118f387SNathan Whitehorn 4432118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) { 4442118f387SNathan Whitehorn if (strcmp(gc->lg_name, "index") == 0) { 4452118f387SNathan Whitehorn indexstr = gc->lg_val; 4462118f387SNathan Whitehorn break; 4472118f387SNathan Whitehorn } 4482118f387SNathan Whitehorn } 4492118f387SNathan Whitehorn 4502118f387SNathan Whitehorn /* Shell out to gpart for partcode for now */ 4512118f387SNathan Whitehorn sprintf(command, "gpart bootcode -p %s -i %s %s", 4526e15678aSNathan Whitehorn partcode_path(scheme, fstype), indexstr, pp->lg_geom->lg_name); 4532118f387SNathan Whitehorn if (system(command) != 0) { 4542118f387SNathan Whitehorn sprintf(message, "Error installing partcode on partition %s", 4552118f387SNathan Whitehorn pp->lg_name); 4562118f387SNathan Whitehorn dialog_msgbox("Error", message, 0, 0, TRUE); 4572118f387SNathan Whitehorn } 4582118f387SNathan Whitehorn } 4592118f387SNathan Whitehorn 4602118f387SNathan Whitehorn void 461f36a5e0fSNathan Whitehorn gpart_destroy(struct ggeom *lg_geom) 4622118f387SNathan Whitehorn { 4632118f387SNathan Whitehorn struct gctl_req *r; 464f36a5e0fSNathan Whitehorn struct gprovider *pp; 4652118f387SNathan Whitehorn const char *errstr; 466f36a5e0fSNathan Whitehorn int force = 1; 4672118f387SNathan Whitehorn 468f36a5e0fSNathan Whitehorn /* Delete all child metadata */ 4692118f387SNathan Whitehorn LIST_FOREACH(pp, &lg_geom->lg_provider, lg_provider) 4702118f387SNathan Whitehorn gpart_delete(pp); 4712118f387SNathan Whitehorn 472f36a5e0fSNathan Whitehorn /* Revert any local changes to get this geom into a pristine state */ 473f36a5e0fSNathan Whitehorn r = gctl_get_handle(); 474f36a5e0fSNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 475f36a5e0fSNathan Whitehorn gctl_ro_param(r, "arg0", -1, lg_geom->lg_name); 476f36a5e0fSNathan Whitehorn gctl_ro_param(r, "verb", -1, "undo"); 477f36a5e0fSNathan Whitehorn gctl_issue(r); /* Ignore errors -- these are non-fatal */ 478f36a5e0fSNathan Whitehorn gctl_free(r); 479f36a5e0fSNathan Whitehorn 4802118f387SNathan Whitehorn /* Now destroy the geom itself */ 4812118f387SNathan Whitehorn r = gctl_get_handle(); 4822118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 4832118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, lg_geom->lg_name); 4842118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 485f36a5e0fSNathan Whitehorn gctl_ro_param(r, "force", sizeof(force), &force); 4862118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "destroy"); 4872118f387SNathan Whitehorn errstr = gctl_issue(r); 488987415b1SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') { 489987415b1SNathan Whitehorn /* 490987415b1SNathan Whitehorn * Check if we reverted away the existence of the geom 491987415b1SNathan Whitehorn * altogether. Show all other errors to the user. 492987415b1SNathan Whitehorn */ 493987415b1SNathan Whitehorn if (strtol(errstr, NULL, 0) != EINVAL) 4942118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 495987415b1SNathan Whitehorn } 4962118f387SNathan Whitehorn gctl_free(r); 4972118f387SNathan Whitehorn 4982118f387SNathan Whitehorn /* And any metadata associated with the partition scheme itself */ 4992118f387SNathan Whitehorn delete_part_metadata(lg_geom->lg_name); 5002118f387SNathan Whitehorn } 5012118f387SNathan Whitehorn 5022118f387SNathan Whitehorn void 5032118f387SNathan Whitehorn gpart_edit(struct gprovider *pp) 5042118f387SNathan Whitehorn { 5052118f387SNathan Whitehorn struct gctl_req *r; 5062118f387SNathan Whitehorn struct gconfig *gc; 5072118f387SNathan Whitehorn struct gconsumer *cp; 5082118f387SNathan Whitehorn struct ggeom *geom; 5092118f387SNathan Whitehorn const char *errstr, *oldtype, *scheme; 5102118f387SNathan Whitehorn struct partition_metadata *md; 5112118f387SNathan Whitehorn char sizestr[32]; 5126e15678aSNathan Whitehorn char newfs[255]; 5132118f387SNathan Whitehorn intmax_t idx; 5142118f387SNathan Whitehorn int hadlabel, choice, junk, nitems; 5152118f387SNathan Whitehorn unsigned i; 5162118f387SNathan Whitehorn 5172118f387SNathan Whitehorn DIALOG_FORMITEM items[] = { 5182118f387SNathan Whitehorn {0, "Type:", 5, 0, 0, FALSE, "", 11, 0, 12, 15, 0, 5196e15678aSNathan Whitehorn FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, " 5206e15678aSNathan Whitehorn "freebsd-swap)", FALSE}, 5212118f387SNathan Whitehorn {0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 0, 0, 5222118f387SNathan Whitehorn FALSE, "Partition size. Append K, M, G for kilobytes, " 5232118f387SNathan Whitehorn "megabytes or gigabytes.", FALSE}, 5242118f387SNathan Whitehorn {0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0, 5252118f387SNathan Whitehorn FALSE, "Path at which to mount this partition (leave blank " 5262118f387SNathan Whitehorn "for swap, set to / for root filesystem)", FALSE}, 5272118f387SNathan Whitehorn {0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE, 5282118f387SNathan Whitehorn "Partition name. Not all partition schemes support this.", 5292118f387SNathan Whitehorn FALSE}, 5302118f387SNathan Whitehorn }; 5312118f387SNathan Whitehorn 5322118f387SNathan Whitehorn /* 5332118f387SNathan Whitehorn * Find the PART geom we are manipulating. This may be a consumer of 5342118f387SNathan Whitehorn * this provider, or its parent. Check the consumer case first. 5352118f387SNathan Whitehorn */ 5362118f387SNathan Whitehorn geom = NULL; 5372118f387SNathan Whitehorn LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers) 5382118f387SNathan Whitehorn if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) { 539f36a5e0fSNathan Whitehorn /* Check for zombie geoms, treating them as blank */ 540f36a5e0fSNathan Whitehorn scheme = NULL; 541f36a5e0fSNathan Whitehorn LIST_FOREACH(gc, &cp->lg_geom->lg_config, lg_config) { 542f36a5e0fSNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) { 543f36a5e0fSNathan Whitehorn scheme = gc->lg_val; 544f36a5e0fSNathan Whitehorn break; 545f36a5e0fSNathan Whitehorn } 546f36a5e0fSNathan Whitehorn } 547f36a5e0fSNathan Whitehorn if (scheme == NULL || strcmp(scheme, "(none)") == 0) { 548f36a5e0fSNathan Whitehorn gpart_partition(cp->lg_geom->lg_name, NULL); 5492118f387SNathan Whitehorn return; 550f36a5e0fSNathan Whitehorn } 5512118f387SNathan Whitehorn 552987415b1SNathan Whitehorn /* If this is a nested partition, edit as usual */ 553987415b1SNathan Whitehorn if (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0) 554987415b1SNathan Whitehorn break; 555987415b1SNathan Whitehorn 5562118f387SNathan Whitehorn /* Destroy the geom and all sub-partitions */ 557f36a5e0fSNathan Whitehorn gpart_destroy(cp->lg_geom); 5582118f387SNathan Whitehorn 5592118f387SNathan Whitehorn /* Now re-partition and return */ 5602118f387SNathan Whitehorn gpart_partition(cp->lg_geom->lg_name, NULL); 5612118f387SNathan Whitehorn return; 5622118f387SNathan Whitehorn } 5632118f387SNathan Whitehorn 5642118f387SNathan Whitehorn if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0) 5652118f387SNathan Whitehorn geom = pp->lg_geom; 5662118f387SNathan Whitehorn 5672118f387SNathan Whitehorn if (geom == NULL) { 5682118f387SNathan Whitehorn /* Disk not partitioned, so partition it */ 569bcc25b7eSNathan Whitehorn gpart_partition(pp->lg_name, NULL); 5702118f387SNathan Whitehorn return; 5712118f387SNathan Whitehorn } 5722118f387SNathan Whitehorn 5732118f387SNathan Whitehorn LIST_FOREACH(gc, &geom->lg_config, lg_config) { 5742118f387SNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) { 5752118f387SNathan Whitehorn scheme = gc->lg_val; 5762118f387SNathan Whitehorn break; 5772118f387SNathan Whitehorn } 5782118f387SNathan Whitehorn } 5792118f387SNathan Whitehorn 580c67f41d0SNathan Whitehorn nitems = scheme_supports_labels(scheme) ? 4 : 3; 5812118f387SNathan Whitehorn 5822118f387SNathan Whitehorn /* Edit editable parameters of a partition */ 5832118f387SNathan Whitehorn hadlabel = 0; 5842118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) { 5852118f387SNathan Whitehorn if (strcmp(gc->lg_name, "type") == 0) { 5862118f387SNathan Whitehorn oldtype = gc->lg_val; 5872118f387SNathan Whitehorn items[0].text = gc->lg_val; 5882118f387SNathan Whitehorn } 5892118f387SNathan Whitehorn if (strcmp(gc->lg_name, "label") == 0 && gc->lg_val != NULL) { 5902118f387SNathan Whitehorn hadlabel = 1; 5912118f387SNathan Whitehorn items[3].text = gc->lg_val; 5922118f387SNathan Whitehorn } 5932118f387SNathan Whitehorn if (strcmp(gc->lg_name, "index") == 0) 5942118f387SNathan Whitehorn idx = atoi(gc->lg_val); 5952118f387SNathan Whitehorn } 5962118f387SNathan Whitehorn 5972118f387SNathan Whitehorn TAILQ_FOREACH(md, &part_metadata, metadata) { 5982118f387SNathan Whitehorn if (md->name != NULL && strcmp(md->name, pp->lg_name) == 0) { 5992118f387SNathan Whitehorn if (md->fstab != NULL) 6002118f387SNathan Whitehorn items[2].text = md->fstab->fs_file; 6012118f387SNathan Whitehorn break; 6022118f387SNathan Whitehorn } 6032118f387SNathan Whitehorn } 6042118f387SNathan Whitehorn 6052118f387SNathan Whitehorn humanize_number(sizestr, 7, pp->lg_mediasize, "B", HN_AUTOSCALE, 6062118f387SNathan Whitehorn HN_NOSPACE | HN_DECIMAL); 6072118f387SNathan Whitehorn items[1].text = sizestr; 6082118f387SNathan Whitehorn 6092118f387SNathan Whitehorn editpart: 6102118f387SNathan Whitehorn choice = dlg_form("Edit Partition", "", 0, 0, 0, nitems, items, &junk); 6112118f387SNathan Whitehorn 6122118f387SNathan Whitehorn if (choice) /* Cancel pressed */ 613987415b1SNathan Whitehorn goto endedit; 6142118f387SNathan Whitehorn 615e24e98d0SWojciech Macek /* If this is the root partition, check that this fs is bootable */ 616e24e98d0SWojciech Macek if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme, 617e24e98d0SWojciech Macek items[0].text)) { 618e24e98d0SWojciech Macek char message[512]; 619e24e98d0SWojciech Macek sprintf(message, "This file system (%s) is not bootable " 620e24e98d0SWojciech Macek "on this system. Are you sure you want to proceed?", 621e24e98d0SWojciech Macek items[0].text); 622e24e98d0SWojciech Macek dialog_vars.defaultno = TRUE; 623e24e98d0SWojciech Macek choice = dialog_yesno("Warning", message, 0, 0); 624e24e98d0SWojciech Macek dialog_vars.defaultno = FALSE; 625e24e98d0SWojciech Macek if (choice == 1) /* cancel */ 626e24e98d0SWojciech Macek goto editpart; 627e24e98d0SWojciech Macek } 628e24e98d0SWojciech Macek 6292118f387SNathan Whitehorn /* Check if the label has a / in it */ 6302118f387SNathan Whitehorn if (strchr(items[3].text, '/') != NULL) { 6312118f387SNathan Whitehorn dialog_msgbox("Error", "Label contains a /, which is not an " 6322118f387SNathan Whitehorn "allowed character.", 0, 0, TRUE); 6332118f387SNathan Whitehorn goto editpart; 6342118f387SNathan Whitehorn } 6352118f387SNathan Whitehorn 6362118f387SNathan Whitehorn r = gctl_get_handle(); 6372118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 6382118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, geom->lg_name); 6392118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 6402118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "modify"); 6412118f387SNathan Whitehorn gctl_ro_param(r, "index", sizeof(idx), &idx); 6422118f387SNathan Whitehorn if (hadlabel || items[3].text[0] != '\0') 6432118f387SNathan Whitehorn gctl_ro_param(r, "label", -1, items[3].text); 6442118f387SNathan Whitehorn gctl_ro_param(r, "type", -1, items[0].text); 6452118f387SNathan Whitehorn errstr = gctl_issue(r); 6462118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') { 6472118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 6482118f387SNathan Whitehorn gctl_free(r); 6492118f387SNathan Whitehorn goto editpart; 6502118f387SNathan Whitehorn } 6512118f387SNathan Whitehorn gctl_free(r); 6522118f387SNathan Whitehorn 65350de5d07SNathan Whitehorn newfs_command(items[0].text, newfs, 1); 6542118f387SNathan Whitehorn set_default_part_metadata(pp->lg_name, scheme, items[0].text, 65550de5d07SNathan Whitehorn items[2].text, (strcmp(oldtype, items[0].text) != 0) ? 65650de5d07SNathan Whitehorn newfs : NULL); 6572118f387SNathan Whitehorn 658987415b1SNathan Whitehorn endedit: 659987415b1SNathan Whitehorn if (strcmp(oldtype, items[0].text) != 0 && cp != NULL) 660987415b1SNathan Whitehorn gpart_destroy(cp->lg_geom); 661987415b1SNathan Whitehorn if (strcmp(oldtype, items[0].text) != 0 && strcmp(items[0].text, 662987415b1SNathan Whitehorn "freebsd") == 0) 663987415b1SNathan Whitehorn gpart_partition(pp->lg_name, "BSD"); 664987415b1SNathan Whitehorn 66563128851SMarcelo Araujo for (i = 0; i < nitems(items); i++) 6662118f387SNathan Whitehorn if (items[i].text_free) 6672118f387SNathan Whitehorn free(items[i].text); 6682118f387SNathan Whitehorn } 6692118f387SNathan Whitehorn 6702118f387SNathan Whitehorn void 6712118f387SNathan Whitehorn set_default_part_metadata(const char *name, const char *scheme, 67250de5d07SNathan Whitehorn const char *type, const char *mountpoint, const char *newfs) 6732118f387SNathan Whitehorn { 6742118f387SNathan Whitehorn struct partition_metadata *md; 6756e15678aSNathan Whitehorn char *zpool_name = NULL; 6766e15678aSNathan Whitehorn int i; 6772118f387SNathan Whitehorn 6782118f387SNathan Whitehorn /* Set part metadata */ 6792118f387SNathan Whitehorn md = get_part_metadata(name, 1); 6802118f387SNathan Whitehorn 6812118f387SNathan Whitehorn if (newfs) { 6822118f387SNathan Whitehorn if (md->newfs != NULL) { 6832118f387SNathan Whitehorn free(md->newfs); 6842118f387SNathan Whitehorn md->newfs = NULL; 6852118f387SNathan Whitehorn } 6862118f387SNathan Whitehorn 68750de5d07SNathan Whitehorn if (newfs != NULL && newfs[0] != '\0') { 68850de5d07SNathan Whitehorn md->newfs = malloc(strlen(newfs) + strlen(" /dev/") + 6896e15678aSNathan Whitehorn strlen(mountpoint) + 5 + strlen(name) + 1); 6906e15678aSNathan Whitehorn if (strcmp("freebsd-zfs", type) == 0) { 6916e15678aSNathan Whitehorn zpool_name = strdup((strlen(mountpoint) == 1) ? 6926e15678aSNathan Whitehorn "root" : &mountpoint[1]); 6936e15678aSNathan Whitehorn for (i = 0; zpool_name[i] != 0; i++) 6946e15678aSNathan Whitehorn if (!isalnum(zpool_name[i])) 6956e15678aSNathan Whitehorn zpool_name[i] = '_'; 6966e15678aSNathan Whitehorn sprintf(md->newfs, "%s %s /dev/%s", newfs, 6976e15678aSNathan Whitehorn zpool_name, name); 6986e15678aSNathan Whitehorn } else { 69950de5d07SNathan Whitehorn sprintf(md->newfs, "%s /dev/%s", newfs, name); 7002118f387SNathan Whitehorn } 7012118f387SNathan Whitehorn } 7026e15678aSNathan Whitehorn } 7032118f387SNathan Whitehorn 7042118f387SNathan Whitehorn if (strcmp(type, "freebsd-swap") == 0) 7052118f387SNathan Whitehorn mountpoint = "none"; 7066b446ed5SNathan Whitehorn if (strcmp(type, bootpart_type(scheme)) == 0) 7072118f387SNathan Whitehorn md->bootcode = 1; 7082118f387SNathan Whitehorn 7096e15678aSNathan Whitehorn /* VTOC8 needs partcode at the start of partitions */ 7106e15678aSNathan Whitehorn if (strcmp(scheme, "VTOC8") == 0 && (strcmp(type, "freebsd-ufs") == 0 7116e15678aSNathan Whitehorn || strcmp(type, "freebsd-zfs") == 0)) 7122118f387SNathan Whitehorn md->bootcode = 1; 7132118f387SNathan Whitehorn 7142118f387SNathan Whitehorn if (mountpoint == NULL || mountpoint[0] == '\0') { 7152118f387SNathan Whitehorn if (md->fstab != NULL) { 7162118f387SNathan Whitehorn free(md->fstab->fs_spec); 7172118f387SNathan Whitehorn free(md->fstab->fs_file); 7182118f387SNathan Whitehorn free(md->fstab->fs_vfstype); 7192118f387SNathan Whitehorn free(md->fstab->fs_mntops); 7202118f387SNathan Whitehorn free(md->fstab->fs_type); 7212118f387SNathan Whitehorn free(md->fstab); 7222118f387SNathan Whitehorn md->fstab = NULL; 7232118f387SNathan Whitehorn } 7242118f387SNathan Whitehorn } else { 7252118f387SNathan Whitehorn if (md->fstab == NULL) { 7262118f387SNathan Whitehorn md->fstab = malloc(sizeof(struct fstab)); 7272118f387SNathan Whitehorn } else { 7282118f387SNathan Whitehorn free(md->fstab->fs_spec); 7292118f387SNathan Whitehorn free(md->fstab->fs_file); 7302118f387SNathan Whitehorn free(md->fstab->fs_vfstype); 7312118f387SNathan Whitehorn free(md->fstab->fs_mntops); 7322118f387SNathan Whitehorn free(md->fstab->fs_type); 7332118f387SNathan Whitehorn } 7346e15678aSNathan Whitehorn if (strcmp("freebsd-zfs", type) == 0) { 7356e15678aSNathan Whitehorn md->fstab->fs_spec = strdup(zpool_name); 7366e15678aSNathan Whitehorn } else { 7376e15678aSNathan Whitehorn md->fstab->fs_spec = malloc(strlen(name) + 7386e15678aSNathan Whitehorn strlen("/dev/") + 1); 7392118f387SNathan Whitehorn sprintf(md->fstab->fs_spec, "/dev/%s", name); 7406e15678aSNathan Whitehorn } 7412118f387SNathan Whitehorn md->fstab->fs_file = strdup(mountpoint); 7422118f387SNathan Whitehorn /* Get VFS from text after freebsd-, if possible */ 74350de5d07SNathan Whitehorn if (strncmp("freebsd-", type, 8) == 0) 7442118f387SNathan Whitehorn md->fstab->fs_vfstype = strdup(&type[8]); 74550de5d07SNathan Whitehorn else if (strcmp("fat32", type) == 0 || strcmp("efi", type) == 0) 74650de5d07SNathan Whitehorn md->fstab->fs_vfstype = strdup("msdosfs"); 7472118f387SNathan Whitehorn else 7482118f387SNathan Whitehorn md->fstab->fs_vfstype = strdup(type); /* Guess */ 7492118f387SNathan Whitehorn if (strcmp(type, "freebsd-swap") == 0) { 7502118f387SNathan Whitehorn md->fstab->fs_type = strdup(FSTAB_SW); 7512118f387SNathan Whitehorn md->fstab->fs_freq = 0; 7522118f387SNathan Whitehorn md->fstab->fs_passno = 0; 7536e15678aSNathan Whitehorn } else if (strcmp(type, "freebsd-zfs") == 0) { 7546e15678aSNathan Whitehorn md->fstab->fs_type = strdup(FSTAB_RW); 7556e15678aSNathan Whitehorn md->fstab->fs_freq = 0; 7566e15678aSNathan Whitehorn md->fstab->fs_passno = 0; 7572118f387SNathan Whitehorn } else { 7582118f387SNathan Whitehorn md->fstab->fs_type = strdup(FSTAB_RW); 7592118f387SNathan Whitehorn if (strcmp(mountpoint, "/") == 0) { 7602118f387SNathan Whitehorn md->fstab->fs_freq = 1; 7612118f387SNathan Whitehorn md->fstab->fs_passno = 1; 7622118f387SNathan Whitehorn } else { 7632118f387SNathan Whitehorn md->fstab->fs_freq = 2; 7642118f387SNathan Whitehorn md->fstab->fs_passno = 2; 7652118f387SNathan Whitehorn } 7662118f387SNathan Whitehorn } 7672118f387SNathan Whitehorn md->fstab->fs_mntops = strdup(md->fstab->fs_type); 7682118f387SNathan Whitehorn } 7696e15678aSNathan Whitehorn 7706e15678aSNathan Whitehorn if (zpool_name != NULL) 7716e15678aSNathan Whitehorn free(zpool_name); 7722118f387SNathan Whitehorn } 7732118f387SNathan Whitehorn 7742118f387SNathan Whitehorn static 7752118f387SNathan Whitehorn int part_compare(const void *xa, const void *xb) 7762118f387SNathan Whitehorn { 7772118f387SNathan Whitehorn struct gprovider **a = (struct gprovider **)xa; 7782118f387SNathan Whitehorn struct gprovider **b = (struct gprovider **)xb; 7792118f387SNathan Whitehorn intmax_t astart, bstart; 7802118f387SNathan Whitehorn struct gconfig *gc; 7812118f387SNathan Whitehorn 7822118f387SNathan Whitehorn astart = bstart = 0; 7832118f387SNathan Whitehorn LIST_FOREACH(gc, &(*a)->lg_config, lg_config) 7842118f387SNathan Whitehorn if (strcmp(gc->lg_name, "start") == 0) { 7852118f387SNathan Whitehorn astart = strtoimax(gc->lg_val, NULL, 0); 7862118f387SNathan Whitehorn break; 7872118f387SNathan Whitehorn } 7882118f387SNathan Whitehorn LIST_FOREACH(gc, &(*b)->lg_config, lg_config) 7892118f387SNathan Whitehorn if (strcmp(gc->lg_name, "start") == 0) { 7902118f387SNathan Whitehorn bstart = strtoimax(gc->lg_val, NULL, 0); 7912118f387SNathan Whitehorn break; 7922118f387SNathan Whitehorn } 7932118f387SNathan Whitehorn 7942118f387SNathan Whitehorn if (astart < bstart) 7952118f387SNathan Whitehorn return -1; 7962118f387SNathan Whitehorn else if (astart > bstart) 7972118f387SNathan Whitehorn return 1; 7982118f387SNathan Whitehorn else 7992118f387SNathan Whitehorn return 0; 8002118f387SNathan Whitehorn } 8012118f387SNathan Whitehorn 8022118f387SNathan Whitehorn intmax_t 8032118f387SNathan Whitehorn gpart_max_free(struct ggeom *geom, intmax_t *npartstart) 8042118f387SNathan Whitehorn { 8052118f387SNathan Whitehorn struct gconfig *gc; 8062118f387SNathan Whitehorn struct gprovider *pp, **providers; 80747ead00dSDag-Erling Smørgrav intmax_t sectorsize, stripesize, offset; 8082118f387SNathan Whitehorn intmax_t lastend; 8092118f387SNathan Whitehorn intmax_t start, end; 8102118f387SNathan Whitehorn intmax_t maxsize, maxstart; 8112118f387SNathan Whitehorn intmax_t partstart, partend; 8122118f387SNathan Whitehorn int i, nparts; 8132118f387SNathan Whitehorn 8142118f387SNathan Whitehorn /* Now get the maximum free size and free start */ 8152118f387SNathan Whitehorn start = end = 0; 8162118f387SNathan Whitehorn LIST_FOREACH(gc, &geom->lg_config, lg_config) { 8172118f387SNathan Whitehorn if (strcmp(gc->lg_name, "first") == 0) 8182118f387SNathan Whitehorn start = strtoimax(gc->lg_val, NULL, 0); 8192118f387SNathan Whitehorn if (strcmp(gc->lg_name, "last") == 0) 8202118f387SNathan Whitehorn end = strtoimax(gc->lg_val, NULL, 0); 8212118f387SNathan Whitehorn } 8222118f387SNathan Whitehorn 8232118f387SNathan Whitehorn i = nparts = 0; 8242118f387SNathan Whitehorn LIST_FOREACH(pp, &geom->lg_provider, lg_provider) 8252118f387SNathan Whitehorn nparts++; 8262118f387SNathan Whitehorn providers = calloc(nparts, sizeof(providers[0])); 8272118f387SNathan Whitehorn LIST_FOREACH(pp, &geom->lg_provider, lg_provider) 8282118f387SNathan Whitehorn providers[i++] = pp; 8292118f387SNathan Whitehorn qsort(providers, nparts, sizeof(providers[0]), part_compare); 8302118f387SNathan Whitehorn 8312118f387SNathan Whitehorn lastend = start - 1; 8322118f387SNathan Whitehorn maxsize = 0; 8332118f387SNathan Whitehorn for (i = 0; i < nparts; i++) { 8342118f387SNathan Whitehorn pp = providers[i]; 8352118f387SNathan Whitehorn 8362118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) { 8372118f387SNathan Whitehorn if (strcmp(gc->lg_name, "start") == 0) 8382118f387SNathan Whitehorn partstart = strtoimax(gc->lg_val, NULL, 0); 8392118f387SNathan Whitehorn if (strcmp(gc->lg_name, "end") == 0) 8402118f387SNathan Whitehorn partend = strtoimax(gc->lg_val, NULL, 0); 8412118f387SNathan Whitehorn } 8422118f387SNathan Whitehorn 8432118f387SNathan Whitehorn if (partstart - lastend > maxsize) { 8442118f387SNathan Whitehorn maxsize = partstart - lastend - 1; 8452118f387SNathan Whitehorn maxstart = lastend + 1; 8462118f387SNathan Whitehorn } 8472118f387SNathan Whitehorn 8482118f387SNathan Whitehorn lastend = partend; 8492118f387SNathan Whitehorn } 8502118f387SNathan Whitehorn 8512118f387SNathan Whitehorn if (end - lastend > maxsize) { 8522118f387SNathan Whitehorn maxsize = end - lastend - 1; 8532118f387SNathan Whitehorn maxstart = lastend + 1; 8542118f387SNathan Whitehorn } 8552118f387SNathan Whitehorn 8562118f387SNathan Whitehorn pp = LIST_FIRST(&geom->lg_consumer)->lg_provider; 8572118f387SNathan Whitehorn 85847ead00dSDag-Erling Smørgrav /* 85947ead00dSDag-Erling Smørgrav * Round the start and size of the largest available space up to 86047ead00dSDag-Erling Smørgrav * the nearest multiple of the adjusted stripe size. 86147ead00dSDag-Erling Smørgrav * 86247ead00dSDag-Erling Smørgrav * The adjusted stripe size is the least common multiple of the 86347ead00dSDag-Erling Smørgrav * actual stripe size, or the sector size if no stripe size was 86447ead00dSDag-Erling Smørgrav * reported, and 4096. The reason for this is that contemporary 86547ead00dSDag-Erling Smørgrav * disks often have 4096-byte physical sectors but report 512 86647ead00dSDag-Erling Smørgrav * bytes instead for compatibility with older / broken operating 86747ead00dSDag-Erling Smørgrav * systems and BIOSes. For the same reasons, virtualized storage 86847ead00dSDag-Erling Smørgrav * may also report a 512-byte stripe size, or none at all. 86947ead00dSDag-Erling Smørgrav */ 87047ead00dSDag-Erling Smørgrav sectorsize = pp->lg_sectorsize; 87147ead00dSDag-Erling Smørgrav if ((stripesize = pp->lg_stripesize) == 0) 87247ead00dSDag-Erling Smørgrav stripesize = sectorsize; 87347ead00dSDag-Erling Smørgrav while (stripesize % 4096 != 0) 87447ead00dSDag-Erling Smørgrav stripesize *= 2; 87547ead00dSDag-Erling Smørgrav if ((offset = maxstart * sectorsize % stripesize) != 0) { 87647ead00dSDag-Erling Smørgrav offset = (stripesize - offset) / sectorsize; 8772118f387SNathan Whitehorn maxstart += offset; 8782118f387SNathan Whitehorn maxsize -= offset; 8792118f387SNathan Whitehorn } 8802118f387SNathan Whitehorn 8812118f387SNathan Whitehorn if (npartstart != NULL) 8822118f387SNathan Whitehorn *npartstart = maxstart; 8832118f387SNathan Whitehorn 8842118f387SNathan Whitehorn return (maxsize); 8852118f387SNathan Whitehorn } 8862118f387SNathan Whitehorn 8872118f387SNathan Whitehorn void 8882118f387SNathan Whitehorn gpart_create(struct gprovider *pp, char *default_type, char *default_size, 8892118f387SNathan Whitehorn char *default_mountpoint, char **partname, int interactive) 8902118f387SNathan Whitehorn { 8912118f387SNathan Whitehorn struct gctl_req *r; 8922118f387SNathan Whitehorn struct gconfig *gc; 8932118f387SNathan Whitehorn struct gconsumer *cp; 8942118f387SNathan Whitehorn struct ggeom *geom; 8952118f387SNathan Whitehorn const char *errstr, *scheme; 8967899c6c1SNathan Whitehorn char sizestr[32], startstr[32], output[64], *newpartname; 8976e15678aSNathan Whitehorn char newfs[255], options_fstype[64]; 8982118f387SNathan Whitehorn intmax_t maxsize, size, sector, firstfree, stripe; 8992118f387SNathan Whitehorn uint64_t bytes; 9002118f387SNathan Whitehorn int nitems, choice, junk; 9012118f387SNathan Whitehorn unsigned i; 9022118f387SNathan Whitehorn 9032118f387SNathan Whitehorn DIALOG_FORMITEM items[] = { 9042118f387SNathan Whitehorn {0, "Type:", 5, 0, 0, FALSE, "freebsd-ufs", 11, 0, 12, 15, 0, 9056e15678aSNathan Whitehorn FALSE, "Filesystem type (e.g. freebsd-ufs, freebsd-zfs, " 9066e15678aSNathan Whitehorn "freebsd-swap)", FALSE}, 9072118f387SNathan Whitehorn {0, "Size:", 5, 1, 0, FALSE, "", 11, 1, 12, 15, 0, 9082118f387SNathan Whitehorn FALSE, "Partition size. Append K, M, G for kilobytes, " 9092118f387SNathan Whitehorn "megabytes or gigabytes.", FALSE}, 9102118f387SNathan Whitehorn {0, "Mountpoint:", 11, 2, 0, FALSE, "", 11, 2, 12, 15, 0, 9112118f387SNathan Whitehorn FALSE, "Path at which to mount partition (blank for " 9122118f387SNathan Whitehorn "swap, / for root filesystem)", FALSE}, 9132118f387SNathan Whitehorn {0, "Label:", 7, 3, 0, FALSE, "", 11, 3, 12, 15, 0, FALSE, 9142118f387SNathan Whitehorn "Partition name. Not all partition schemes support this.", 9152118f387SNathan Whitehorn FALSE}, 9162118f387SNathan Whitehorn }; 9172118f387SNathan Whitehorn 9182118f387SNathan Whitehorn if (partname != NULL) 9192118f387SNathan Whitehorn *partname = NULL; 9202118f387SNathan Whitehorn 9212118f387SNathan Whitehorn /* Record sector and stripe sizes */ 9222118f387SNathan Whitehorn sector = pp->lg_sectorsize; 9232118f387SNathan Whitehorn stripe = pp->lg_stripesize; 9242118f387SNathan Whitehorn 9252118f387SNathan Whitehorn /* 9262118f387SNathan Whitehorn * Find the PART geom we are manipulating. This may be a consumer of 9272118f387SNathan Whitehorn * this provider, or its parent. Check the consumer case first. 9282118f387SNathan Whitehorn */ 9292118f387SNathan Whitehorn geom = NULL; 9302118f387SNathan Whitehorn LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers) 9312118f387SNathan Whitehorn if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) { 9322118f387SNathan Whitehorn geom = cp->lg_geom; 9332118f387SNathan Whitehorn break; 9342118f387SNathan Whitehorn } 9352118f387SNathan Whitehorn 9362118f387SNathan Whitehorn if (geom == NULL && strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0) 9372118f387SNathan Whitehorn geom = pp->lg_geom; 9382118f387SNathan Whitehorn 9392118f387SNathan Whitehorn /* Now get the partition scheme */ 9402118f387SNathan Whitehorn scheme = NULL; 9412118f387SNathan Whitehorn if (geom != NULL) { 9422118f387SNathan Whitehorn LIST_FOREACH(gc, &geom->lg_config, lg_config) 9432118f387SNathan Whitehorn if (strcmp(gc->lg_name, "scheme") == 0) 9442118f387SNathan Whitehorn scheme = gc->lg_val; 9452118f387SNathan Whitehorn } 9462118f387SNathan Whitehorn 9472118f387SNathan Whitehorn if (geom == NULL || scheme == NULL || strcmp(scheme, "(none)") == 0) { 948bcc25b7eSNathan Whitehorn if (gpart_partition(pp->lg_name, NULL) == 0) 9492118f387SNathan Whitehorn dialog_msgbox("", 9502118f387SNathan Whitehorn "The partition table has been successfully created." 9512118f387SNathan Whitehorn " Please press Create again to create partitions.", 9522118f387SNathan Whitehorn 0, 0, TRUE); 9532118f387SNathan Whitehorn 9542118f387SNathan Whitehorn return; 9552118f387SNathan Whitehorn } 9562118f387SNathan Whitehorn 9572118f387SNathan Whitehorn /* 9582118f387SNathan Whitehorn * If we still don't have a geom, either the user has 9592118f387SNathan Whitehorn * canceled partitioning or there has been an error which has already 9602118f387SNathan Whitehorn * been displayed, so bail. 9612118f387SNathan Whitehorn */ 9622118f387SNathan Whitehorn if (geom == NULL) 9632118f387SNathan Whitehorn return; 9642118f387SNathan Whitehorn 9652118f387SNathan Whitehorn maxsize = size = gpart_max_free(geom, &firstfree); 9662118f387SNathan Whitehorn if (size <= 0) { 9672118f387SNathan Whitehorn dialog_msgbox("Error", "No free space left on device.", 0, 0, 9682118f387SNathan Whitehorn TRUE); 9692118f387SNathan Whitehorn return; 9702118f387SNathan Whitehorn } 9712118f387SNathan Whitehorn 9722118f387SNathan Whitehorn humanize_number(sizestr, 7, size*sector, "B", HN_AUTOSCALE, 9732118f387SNathan Whitehorn HN_NOSPACE | HN_DECIMAL); 9742118f387SNathan Whitehorn items[1].text = sizestr; 9752118f387SNathan Whitehorn 9762118f387SNathan Whitehorn /* Special-case the MBR default type for nested partitions */ 9772b375b4eSYoshihiro Takahashi if (strcmp(scheme, "MBR") == 0) { 9782118f387SNathan Whitehorn items[0].text = "freebsd"; 979bcc25b7eSNathan Whitehorn items[0].help = "Filesystem type (e.g. freebsd, fat32)"; 980bcc25b7eSNathan Whitehorn } 9812118f387SNathan Whitehorn 982c67f41d0SNathan Whitehorn nitems = scheme_supports_labels(scheme) ? 4 : 3; 9832118f387SNathan Whitehorn 9842118f387SNathan Whitehorn if (default_type != NULL) 9852118f387SNathan Whitehorn items[0].text = default_type; 9862118f387SNathan Whitehorn if (default_size != NULL) 9872118f387SNathan Whitehorn items[1].text = default_size; 9882118f387SNathan Whitehorn if (default_mountpoint != NULL) 9892118f387SNathan Whitehorn items[2].text = default_mountpoint; 9902118f387SNathan Whitehorn 99150de5d07SNathan Whitehorn /* Default options */ 99250de5d07SNathan Whitehorn strncpy(options_fstype, items[0].text, 99350de5d07SNathan Whitehorn sizeof(options_fstype)); 99450de5d07SNathan Whitehorn newfs_command(options_fstype, newfs, 1); 9952118f387SNathan Whitehorn addpartform: 9962118f387SNathan Whitehorn if (interactive) { 99750de5d07SNathan Whitehorn dialog_vars.extra_label = "Options"; 99850de5d07SNathan Whitehorn dialog_vars.extra_button = TRUE; 9992118f387SNathan Whitehorn choice = dlg_form("Add Partition", "", 0, 0, 0, nitems, 10002118f387SNathan Whitehorn items, &junk); 100150de5d07SNathan Whitehorn dialog_vars.extra_button = FALSE; 100250de5d07SNathan Whitehorn switch (choice) { 100350de5d07SNathan Whitehorn case 0: /* OK */ 100450de5d07SNathan Whitehorn break; 100550de5d07SNathan Whitehorn case 1: /* Cancel */ 10062118f387SNathan Whitehorn return; 100750de5d07SNathan Whitehorn case 3: /* Options */ 100850de5d07SNathan Whitehorn strncpy(options_fstype, items[0].text, 100950de5d07SNathan Whitehorn sizeof(options_fstype)); 101050de5d07SNathan Whitehorn newfs_command(options_fstype, newfs, 0); 101150de5d07SNathan Whitehorn goto addpartform; 101250de5d07SNathan Whitehorn } 101350de5d07SNathan Whitehorn } 101450de5d07SNathan Whitehorn 101550de5d07SNathan Whitehorn /* 101650de5d07SNathan Whitehorn * If the user changed the fs type after specifying options, undo 101750de5d07SNathan Whitehorn * their choices in favor of the new filesystem's defaults. 101850de5d07SNathan Whitehorn */ 10190d49c6d5SNathan Whitehorn if (strcmp(options_fstype, items[0].text) != 0) { 102050de5d07SNathan Whitehorn strncpy(options_fstype, items[0].text, sizeof(options_fstype)); 102150de5d07SNathan Whitehorn newfs_command(options_fstype, newfs, 1); 10222118f387SNathan Whitehorn } 10232118f387SNathan Whitehorn 10242118f387SNathan Whitehorn size = maxsize; 10252118f387SNathan Whitehorn if (strlen(items[1].text) > 0) { 10262118f387SNathan Whitehorn if (expand_number(items[1].text, &bytes) != 0) { 10272118f387SNathan Whitehorn char error[512]; 10282118f387SNathan Whitehorn 10292118f387SNathan Whitehorn sprintf(error, "Invalid size: %s\n", strerror(errno)); 10302118f387SNathan Whitehorn dialog_msgbox("Error", error, 0, 0, TRUE); 10312118f387SNathan Whitehorn goto addpartform; 10322118f387SNathan Whitehorn } 10332118f387SNathan Whitehorn size = MIN((intmax_t)(bytes/sector), maxsize); 10342118f387SNathan Whitehorn } 10352118f387SNathan Whitehorn 10362118f387SNathan Whitehorn /* Check if the label has a / in it */ 10372118f387SNathan Whitehorn if (strchr(items[3].text, '/') != NULL) { 10382118f387SNathan Whitehorn dialog_msgbox("Error", "Label contains a /, which is not an " 10392118f387SNathan Whitehorn "allowed character.", 0, 0, TRUE); 10402118f387SNathan Whitehorn goto addpartform; 10412118f387SNathan Whitehorn } 10422118f387SNathan Whitehorn 10432118f387SNathan Whitehorn /* Warn if no mountpoint set */ 10442118f387SNathan Whitehorn if (strcmp(items[0].text, "freebsd-ufs") == 0 && 10452118f387SNathan Whitehorn items[2].text[0] != '/') { 10462118f387SNathan Whitehorn dialog_vars.defaultno = TRUE; 10472118f387SNathan Whitehorn choice = dialog_yesno("Warning", 10482118f387SNathan Whitehorn "This partition does not have a valid mountpoint " 10492118f387SNathan Whitehorn "(for the partition from which you intend to boot the " 10502118f387SNathan Whitehorn "operating system, the mountpoint should be /). Are you " 10512118f387SNathan Whitehorn "sure you want to continue?" 10522118f387SNathan Whitehorn , 0, 0); 10532118f387SNathan Whitehorn dialog_vars.defaultno = FALSE; 10542118f387SNathan Whitehorn if (choice == 1) /* cancel */ 10552118f387SNathan Whitehorn goto addpartform; 10562118f387SNathan Whitehorn } 10572118f387SNathan Whitehorn 1058a780c996SNathan Whitehorn /* 1059a780c996SNathan Whitehorn * Error if this scheme needs nested partitions, this is one, and 1060a780c996SNathan Whitehorn * a mountpoint was set. 1061a780c996SNathan Whitehorn */ 1062a780c996SNathan Whitehorn if (strcmp(items[0].text, "freebsd") == 0 && 1063a780c996SNathan Whitehorn strlen(items[2].text) > 0) { 1064a780c996SNathan Whitehorn dialog_msgbox("Error", "Partitions of type \"freebsd\" are " 1065a780c996SNathan Whitehorn "nested BSD-type partition schemes and cannot have " 1066a780c996SNathan Whitehorn "mountpoints. After creating one, select it and press " 1067a780c996SNathan Whitehorn "Create again to add the actual file systems.", 0, 0, TRUE); 1068a780c996SNathan Whitehorn goto addpartform; 1069a780c996SNathan Whitehorn } 1070a780c996SNathan Whitehorn 10712118f387SNathan Whitehorn /* If this is the root partition, check that this scheme is bootable */ 10722118f387SNathan Whitehorn if (strcmp(items[2].text, "/") == 0 && !is_scheme_bootable(scheme)) { 10732118f387SNathan Whitehorn char message[512]; 10742118f387SNathan Whitehorn sprintf(message, "This partition scheme (%s) is not bootable " 10752118f387SNathan Whitehorn "on this platform. Are you sure you want to proceed?", 10762118f387SNathan Whitehorn scheme); 10772118f387SNathan Whitehorn dialog_vars.defaultno = TRUE; 10782118f387SNathan Whitehorn choice = dialog_yesno("Warning", message, 0, 0); 10792118f387SNathan Whitehorn dialog_vars.defaultno = FALSE; 10802118f387SNathan Whitehorn if (choice == 1) /* cancel */ 10812118f387SNathan Whitehorn goto addpartform; 10822118f387SNathan Whitehorn } 10832118f387SNathan Whitehorn 10846e15678aSNathan Whitehorn /* If this is the root partition, check that this fs is bootable */ 10856e15678aSNathan Whitehorn if (strcmp(items[2].text, "/") == 0 && !is_fs_bootable(scheme, 10866e15678aSNathan Whitehorn items[0].text)) { 10876e15678aSNathan Whitehorn char message[512]; 10886e15678aSNathan Whitehorn sprintf(message, "This file system (%s) is not bootable " 10896e15678aSNathan Whitehorn "on this system. Are you sure you want to proceed?", 10906e15678aSNathan Whitehorn items[0].text); 10916e15678aSNathan Whitehorn dialog_vars.defaultno = TRUE; 10926e15678aSNathan Whitehorn choice = dialog_yesno("Warning", message, 0, 0); 10936e15678aSNathan Whitehorn dialog_vars.defaultno = FALSE; 10946e15678aSNathan Whitehorn if (choice == 1) /* cancel */ 10956e15678aSNathan Whitehorn goto addpartform; 10966e15678aSNathan Whitehorn } 10976e15678aSNathan Whitehorn 10982118f387SNathan Whitehorn /* 10992118f387SNathan Whitehorn * If this is the root partition, and we need a boot partition, ask 11002118f387SNathan Whitehorn * the user to add one. 11012118f387SNathan Whitehorn */ 1102a780c996SNathan Whitehorn 1103a780c996SNathan Whitehorn /* Check for existing freebsd-boot partition */ 1104a780c996SNathan Whitehorn LIST_FOREACH(pp, &geom->lg_provider, lg_provider) { 1105a780c996SNathan Whitehorn struct partition_metadata *md; 1106a780c996SNathan Whitehorn md = get_part_metadata(pp->lg_name, 0); 1107a780c996SNathan Whitehorn if (md == NULL || !md->bootcode) 1108a780c996SNathan Whitehorn continue; 1109a780c996SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) 1110a780c996SNathan Whitehorn if (strcmp(gc->lg_name, "type") == 0) 1111a780c996SNathan Whitehorn break; 11126b446ed5SNathan Whitehorn if (gc != NULL && strcmp(gc->lg_val, 11136b446ed5SNathan Whitehorn bootpart_type(scheme)) == 0) 1114a780c996SNathan Whitehorn break; 1115a780c996SNathan Whitehorn } 1116a780c996SNathan Whitehorn 1117a780c996SNathan Whitehorn /* If there isn't one, and we need one, ask */ 11188d795806SNathan Whitehorn if ((strcmp(items[0].text, "freebsd") == 0 || 11198d795806SNathan Whitehorn strcmp(items[2].text, "/") == 0) && bootpart_size(scheme) > 0 && 1120a780c996SNathan Whitehorn pp == NULL) { 11212118f387SNathan Whitehorn if (interactive) 11222118f387SNathan Whitehorn choice = dialog_yesno("Boot Partition", 11232118f387SNathan Whitehorn "This partition scheme requires a boot partition " 11242118f387SNathan Whitehorn "for the disk to be bootable. Would you like to " 11252118f387SNathan Whitehorn "make one now?", 0, 0); 11262118f387SNathan Whitehorn else 11272118f387SNathan Whitehorn choice = 0; 11282118f387SNathan Whitehorn 11292118f387SNathan Whitehorn if (choice == 0) { /* yes */ 11302118f387SNathan Whitehorn r = gctl_get_handle(); 11312118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 11322118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, geom->lg_name); 11332118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 11342118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "add"); 11356b446ed5SNathan Whitehorn gctl_ro_param(r, "type", -1, bootpart_type(scheme)); 11362118f387SNathan Whitehorn snprintf(sizestr, sizeof(sizestr), "%jd", 11372118f387SNathan Whitehorn bootpart_size(scheme) / sector); 11382118f387SNathan Whitehorn gctl_ro_param(r, "size", -1, sizestr); 11392118f387SNathan Whitehorn snprintf(startstr, sizeof(startstr), "%jd", firstfree); 11402118f387SNathan Whitehorn gctl_ro_param(r, "start", -1, startstr); 11412118f387SNathan Whitehorn gctl_rw_param(r, "output", sizeof(output), output); 11422118f387SNathan Whitehorn errstr = gctl_issue(r); 11432118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') 11442118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 11452118f387SNathan Whitehorn gctl_free(r); 11462118f387SNathan Whitehorn 11472118f387SNathan Whitehorn get_part_metadata(strtok(output, " "), 1)->bootcode = 1; 11482118f387SNathan Whitehorn 11492118f387SNathan Whitehorn /* Now adjust the part we are really adding forward */ 11502118f387SNathan Whitehorn firstfree += bootpart_size(scheme) / sector; 11512118f387SNathan Whitehorn size -= (bootpart_size(scheme) + stripe)/sector; 11522118f387SNathan Whitehorn if (stripe > 0 && (firstfree*sector % stripe) != 0) 11532118f387SNathan Whitehorn firstfree += (stripe - ((firstfree*sector) % 11542118f387SNathan Whitehorn stripe)) / sector; 11552118f387SNathan Whitehorn } 11562118f387SNathan Whitehorn } 11572118f387SNathan Whitehorn 11582118f387SNathan Whitehorn r = gctl_get_handle(); 11592118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 11602118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, geom->lg_name); 11612118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 11622118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "add"); 11632118f387SNathan Whitehorn 11642118f387SNathan Whitehorn gctl_ro_param(r, "type", -1, items[0].text); 11652118f387SNathan Whitehorn snprintf(sizestr, sizeof(sizestr), "%jd", size); 11662118f387SNathan Whitehorn gctl_ro_param(r, "size", -1, sizestr); 11672118f387SNathan Whitehorn snprintf(startstr, sizeof(startstr), "%jd", firstfree); 11682118f387SNathan Whitehorn gctl_ro_param(r, "start", -1, startstr); 11692118f387SNathan Whitehorn if (items[3].text[0] != '\0') 11702118f387SNathan Whitehorn gctl_ro_param(r, "label", -1, items[3].text); 11712118f387SNathan Whitehorn gctl_rw_param(r, "output", sizeof(output), output); 11722118f387SNathan Whitehorn errstr = gctl_issue(r); 11732118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') { 11742118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 11752118f387SNathan Whitehorn gctl_free(r); 11762118f387SNathan Whitehorn goto addpartform; 11772118f387SNathan Whitehorn } 11787899c6c1SNathan Whitehorn newpartname = strtok(output, " "); 11797899c6c1SNathan Whitehorn gctl_free(r); 11807899c6c1SNathan Whitehorn 11817899c6c1SNathan Whitehorn /* 11827899c6c1SNathan Whitehorn * Try to destroy any geom that gpart picked up already here from 11837899c6c1SNathan Whitehorn * dirty blocks. 11847899c6c1SNathan Whitehorn */ 11857899c6c1SNathan Whitehorn r = gctl_get_handle(); 11867899c6c1SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 11877899c6c1SNathan Whitehorn gctl_ro_param(r, "arg0", -1, newpartname); 11887899c6c1SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 11897899c6c1SNathan Whitehorn junk = 1; 11907899c6c1SNathan Whitehorn gctl_ro_param(r, "force", sizeof(junk), &junk); 11917899c6c1SNathan Whitehorn gctl_ro_param(r, "verb", -1, "destroy"); 11927899c6c1SNathan Whitehorn gctl_issue(r); /* Error usually expected and non-fatal */ 11937899c6c1SNathan Whitehorn gctl_free(r); 11942118f387SNathan Whitehorn 11956b446ed5SNathan Whitehorn if (strcmp(items[0].text, bootpart_type(scheme)) == 0) 11967899c6c1SNathan Whitehorn get_part_metadata(newpartname, 1)->bootcode = 1; 11972118f387SNathan Whitehorn else if (strcmp(items[0].text, "freebsd") == 0) 11987899c6c1SNathan Whitehorn gpart_partition(newpartname, "BSD"); 11992118f387SNathan Whitehorn else 12007899c6c1SNathan Whitehorn set_default_part_metadata(newpartname, scheme, 120150de5d07SNathan Whitehorn items[0].text, items[2].text, newfs); 12022118f387SNathan Whitehorn 120363128851SMarcelo Araujo for (i = 0; i < nitems(items); i++) 12042118f387SNathan Whitehorn if (items[i].text_free) 12052118f387SNathan Whitehorn free(items[i].text); 12062118f387SNathan Whitehorn 12072118f387SNathan Whitehorn if (partname != NULL) 12087899c6c1SNathan Whitehorn *partname = strdup(newpartname); 12092118f387SNathan Whitehorn } 12102118f387SNathan Whitehorn 12112118f387SNathan Whitehorn void 12122118f387SNathan Whitehorn gpart_delete(struct gprovider *pp) 12132118f387SNathan Whitehorn { 12142118f387SNathan Whitehorn struct gconfig *gc; 12152118f387SNathan Whitehorn struct ggeom *geom; 12162118f387SNathan Whitehorn struct gconsumer *cp; 12172118f387SNathan Whitehorn struct gctl_req *r; 12182118f387SNathan Whitehorn const char *errstr; 12192118f387SNathan Whitehorn intmax_t idx; 1220f36a5e0fSNathan Whitehorn int is_partition; 12212118f387SNathan Whitehorn 12222118f387SNathan Whitehorn /* Is it a partition? */ 12232118f387SNathan Whitehorn is_partition = (strcmp(pp->lg_geom->lg_class->lg_name, "PART") == 0); 12242118f387SNathan Whitehorn 12252118f387SNathan Whitehorn /* Find out if this is the root of a gpart geom */ 12262118f387SNathan Whitehorn geom = NULL; 12272118f387SNathan Whitehorn LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers) 12282118f387SNathan Whitehorn if (strcmp(cp->lg_geom->lg_class->lg_name, "PART") == 0) { 12292118f387SNathan Whitehorn geom = cp->lg_geom; 12302118f387SNathan Whitehorn break; 12312118f387SNathan Whitehorn } 12322118f387SNathan Whitehorn 1233f36a5e0fSNathan Whitehorn /* If so, destroy all children */ 12342118f387SNathan Whitehorn if (geom != NULL) { 1235f36a5e0fSNathan Whitehorn gpart_destroy(geom); 1236f36a5e0fSNathan Whitehorn 1237f36a5e0fSNathan Whitehorn /* If this is a partition, revert it, so it can be deleted */ 12382118f387SNathan Whitehorn if (is_partition) { 1239f36a5e0fSNathan Whitehorn r = gctl_get_handle(); 1240f36a5e0fSNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 1241f36a5e0fSNathan Whitehorn gctl_ro_param(r, "arg0", -1, geom->lg_name); 1242f36a5e0fSNathan Whitehorn gctl_ro_param(r, "verb", -1, "undo"); 1243f36a5e0fSNathan Whitehorn gctl_issue(r); /* Ignore non-fatal errors */ 1244f36a5e0fSNathan Whitehorn gctl_free(r); 12452118f387SNathan Whitehorn } 12462118f387SNathan Whitehorn } 12472118f387SNathan Whitehorn 12482118f387SNathan Whitehorn /* 12492118f387SNathan Whitehorn * If this is not a partition, see if that is a problem, complain if 12502118f387SNathan Whitehorn * necessary, and return always, since we need not do anything further, 12512118f387SNathan Whitehorn * error or no. 12522118f387SNathan Whitehorn */ 12532118f387SNathan Whitehorn if (!is_partition) { 12542118f387SNathan Whitehorn if (geom == NULL) 12552118f387SNathan Whitehorn dialog_msgbox("Error", 12562118f387SNathan Whitehorn "Only partitions can be deleted.", 0, 0, TRUE); 12572118f387SNathan Whitehorn return; 12582118f387SNathan Whitehorn } 12592118f387SNathan Whitehorn 12602118f387SNathan Whitehorn r = gctl_get_handle(); 12612118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, pp->lg_geom->lg_class->lg_name); 12622118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, pp->lg_geom->lg_name); 12632118f387SNathan Whitehorn gctl_ro_param(r, "flags", -1, GPART_FLAGS); 12642118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "delete"); 12652118f387SNathan Whitehorn 12662118f387SNathan Whitehorn LIST_FOREACH(gc, &pp->lg_config, lg_config) { 12672118f387SNathan Whitehorn if (strcmp(gc->lg_name, "index") == 0) { 12682118f387SNathan Whitehorn idx = atoi(gc->lg_val); 12692118f387SNathan Whitehorn gctl_ro_param(r, "index", sizeof(idx), &idx); 12702118f387SNathan Whitehorn break; 12712118f387SNathan Whitehorn } 12722118f387SNathan Whitehorn } 12732118f387SNathan Whitehorn 12742118f387SNathan Whitehorn errstr = gctl_issue(r); 12752118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') { 12762118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 12772118f387SNathan Whitehorn gctl_free(r); 12782118f387SNathan Whitehorn return; 12792118f387SNathan Whitehorn } 12802118f387SNathan Whitehorn 12812118f387SNathan Whitehorn gctl_free(r); 12822118f387SNathan Whitehorn 12832118f387SNathan Whitehorn delete_part_metadata(pp->lg_name); 12842118f387SNathan Whitehorn } 12852118f387SNathan Whitehorn 12862118f387SNathan Whitehorn void 12872118f387SNathan Whitehorn gpart_revert_all(struct gmesh *mesh) 12882118f387SNathan Whitehorn { 12892118f387SNathan Whitehorn struct gclass *classp; 12902118f387SNathan Whitehorn struct gconfig *gc; 12912118f387SNathan Whitehorn struct ggeom *gp; 12922118f387SNathan Whitehorn struct gctl_req *r; 12932118f387SNathan Whitehorn const char *modified; 12942118f387SNathan Whitehorn 12952118f387SNathan Whitehorn LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 12962118f387SNathan Whitehorn if (strcmp(classp->lg_name, "PART") == 0) 12972118f387SNathan Whitehorn break; 12982118f387SNathan Whitehorn } 12992118f387SNathan Whitehorn 13002118f387SNathan Whitehorn if (strcmp(classp->lg_name, "PART") != 0) { 13012118f387SNathan Whitehorn dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE); 13022118f387SNathan Whitehorn return; 13032118f387SNathan Whitehorn } 13042118f387SNathan Whitehorn 13052118f387SNathan Whitehorn LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 13062118f387SNathan Whitehorn modified = "true"; /* XXX: If we don't know (kernel too old), 13072118f387SNathan Whitehorn * assume there are modifications. */ 13082118f387SNathan Whitehorn LIST_FOREACH(gc, &gp->lg_config, lg_config) { 13092118f387SNathan Whitehorn if (strcmp(gc->lg_name, "modified") == 0) { 13102118f387SNathan Whitehorn modified = gc->lg_val; 13112118f387SNathan Whitehorn break; 13122118f387SNathan Whitehorn } 13132118f387SNathan Whitehorn } 13142118f387SNathan Whitehorn 13152118f387SNathan Whitehorn if (strcmp(modified, "false") == 0) 13162118f387SNathan Whitehorn continue; 13172118f387SNathan Whitehorn 13182118f387SNathan Whitehorn r = gctl_get_handle(); 13192118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 13202118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, gp->lg_name); 13212118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "undo"); 13222118f387SNathan Whitehorn 13237899c6c1SNathan Whitehorn gctl_issue(r); 13242118f387SNathan Whitehorn gctl_free(r); 13252118f387SNathan Whitehorn } 13262118f387SNathan Whitehorn } 13272118f387SNathan Whitehorn 13282118f387SNathan Whitehorn void 13292118f387SNathan Whitehorn gpart_commit(struct gmesh *mesh) 13302118f387SNathan Whitehorn { 13312118f387SNathan Whitehorn struct partition_metadata *md; 13322118f387SNathan Whitehorn struct gclass *classp; 13332118f387SNathan Whitehorn struct ggeom *gp; 13342118f387SNathan Whitehorn struct gconfig *gc; 13352118f387SNathan Whitehorn struct gconsumer *cp; 13362118f387SNathan Whitehorn struct gprovider *pp; 13372118f387SNathan Whitehorn struct gctl_req *r; 13382118f387SNathan Whitehorn const char *errstr; 13392118f387SNathan Whitehorn const char *modified; 13406e15678aSNathan Whitehorn const char *rootfs; 13412118f387SNathan Whitehorn 13422118f387SNathan Whitehorn LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 13432118f387SNathan Whitehorn if (strcmp(classp->lg_name, "PART") == 0) 13442118f387SNathan Whitehorn break; 13452118f387SNathan Whitehorn } 13462118f387SNathan Whitehorn 13476e15678aSNathan Whitehorn /* Figure out what filesystem / uses */ 13486e15678aSNathan Whitehorn rootfs = "ufs"; /* Assume ufs if nothing else present */ 13496e15678aSNathan Whitehorn TAILQ_FOREACH(md, &part_metadata, metadata) { 13506e15678aSNathan Whitehorn if (md->fstab != NULL && strcmp(md->fstab->fs_file, "/") == 0) { 13516e15678aSNathan Whitehorn rootfs = md->fstab->fs_vfstype; 13526e15678aSNathan Whitehorn break; 13536e15678aSNathan Whitehorn } 13546e15678aSNathan Whitehorn } 13556e15678aSNathan Whitehorn 13562118f387SNathan Whitehorn if (strcmp(classp->lg_name, "PART") != 0) { 13572118f387SNathan Whitehorn dialog_msgbox("Error", "gpart not found!", 0, 0, TRUE); 13582118f387SNathan Whitehorn return; 13592118f387SNathan Whitehorn } 13602118f387SNathan Whitehorn 13612118f387SNathan Whitehorn LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 13622118f387SNathan Whitehorn modified = "true"; /* XXX: If we don't know (kernel too old), 13632118f387SNathan Whitehorn * assume there are modifications. */ 13642118f387SNathan Whitehorn LIST_FOREACH(gc, &gp->lg_config, lg_config) { 13652118f387SNathan Whitehorn if (strcmp(gc->lg_name, "modified") == 0) { 13662118f387SNathan Whitehorn modified = gc->lg_val; 13672118f387SNathan Whitehorn break; 13682118f387SNathan Whitehorn } 13692118f387SNathan Whitehorn } 13702118f387SNathan Whitehorn 13712118f387SNathan Whitehorn if (strcmp(modified, "false") == 0) 13722118f387SNathan Whitehorn continue; 13732118f387SNathan Whitehorn 13742118f387SNathan Whitehorn /* Add bootcode if necessary, before the commit */ 13752118f387SNathan Whitehorn md = get_part_metadata(gp->lg_name, 0); 13762118f387SNathan Whitehorn if (md != NULL && md->bootcode) 13772118f387SNathan Whitehorn gpart_bootcode(gp); 13782118f387SNathan Whitehorn 13792118f387SNathan Whitehorn /* Now install partcode on its partitions, if necessary */ 13802118f387SNathan Whitehorn LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 13812118f387SNathan Whitehorn md = get_part_metadata(pp->lg_name, 0); 13822118f387SNathan Whitehorn if (md == NULL || !md->bootcode) 13832118f387SNathan Whitehorn continue; 13842118f387SNathan Whitehorn 13852118f387SNathan Whitehorn /* Mark this partition active if that's required */ 13862118f387SNathan Whitehorn gpart_activate(pp); 13872118f387SNathan Whitehorn 13882118f387SNathan Whitehorn /* Check if the partition has sub-partitions */ 13892118f387SNathan Whitehorn LIST_FOREACH(cp, &pp->lg_consumers, lg_consumers) 13902118f387SNathan Whitehorn if (strcmp(cp->lg_geom->lg_class->lg_name, 13912118f387SNathan Whitehorn "PART") == 0) 13922118f387SNathan Whitehorn break; 13932118f387SNathan Whitehorn 13942118f387SNathan Whitehorn if (cp == NULL) /* No sub-partitions */ 13956e15678aSNathan Whitehorn gpart_partcode(pp, rootfs); 13962118f387SNathan Whitehorn } 13972118f387SNathan Whitehorn 13982118f387SNathan Whitehorn r = gctl_get_handle(); 13992118f387SNathan Whitehorn gctl_ro_param(r, "class", -1, "PART"); 14002118f387SNathan Whitehorn gctl_ro_param(r, "arg0", -1, gp->lg_name); 14012118f387SNathan Whitehorn gctl_ro_param(r, "verb", -1, "commit"); 14022118f387SNathan Whitehorn 14032118f387SNathan Whitehorn errstr = gctl_issue(r); 14042118f387SNathan Whitehorn if (errstr != NULL && errstr[0] != '\0') 14052118f387SNathan Whitehorn gpart_show_error("Error", NULL, errstr); 14062118f387SNathan Whitehorn gctl_free(r); 14072118f387SNathan Whitehorn } 14082118f387SNathan Whitehorn } 14092118f387SNathan Whitehorn 1410