1 /*- 2 * Copyright (c) 2011 Nathan Whitehorn 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD$ 27 */ 28 29 #include <sys/param.h> 30 #include <errno.h> 31 #include <libutil.h> 32 #include <inttypes.h> 33 34 #include <libgeom.h> 35 #include <dialog.h> 36 #include <dlg_keys.h> 37 38 #include "partedit.h" 39 40 #define MIN_FREE_SPACE (1024*1024*1024) /* 1 GB */ 41 #define SWAP_SIZE(available) MIN(available/20, 4*1024*1024*1024LL) 42 43 static char *boot_disk(struct gmesh *mesh); 44 static char *wizard_partition(struct gmesh *mesh, const char *disk); 45 static int wizard_makeparts(struct gmesh *mesh, const char *disk); 46 47 int 48 part_wizard(void) { 49 int error; 50 struct gmesh mesh; 51 char *disk, *schemeroot; 52 53 startwizard: 54 error = geom_gettree(&mesh); 55 56 dlg_put_backtitle(); 57 error = geom_gettree(&mesh); 58 disk = boot_disk(&mesh); 59 if (disk == NULL) 60 return (1); 61 62 dlg_clear(); 63 dlg_put_backtitle(); 64 schemeroot = wizard_partition(&mesh, disk); 65 free(disk); 66 if (schemeroot == NULL) 67 return (1); 68 69 geom_deletetree(&mesh); 70 dlg_clear(); 71 dlg_put_backtitle(); 72 error = geom_gettree(&mesh); 73 74 error = wizard_makeparts(&mesh, schemeroot); 75 if (error) 76 goto startwizard; 77 free(schemeroot); 78 79 geom_deletetree(&mesh); 80 81 return (0); 82 } 83 84 static char * 85 boot_disk(struct gmesh *mesh) 86 { 87 struct gclass *classp; 88 struct gconfig *gc; 89 struct ggeom *gp; 90 struct gprovider *pp; 91 DIALOG_LISTITEM *disks = NULL; 92 const char *type, *desc; 93 char diskdesc[512]; 94 char *chosen; 95 int i, err, selected, n = 0; 96 97 LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 98 if (strcmp(classp->lg_name, "DISK") != 0 && 99 strcmp(classp->lg_name, "RAID") != 0 && 100 strcmp(classp->lg_name, "MD") != 0) 101 continue; 102 103 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 104 if (LIST_EMPTY(&gp->lg_provider)) 105 continue; 106 107 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 108 desc = type = NULL; 109 LIST_FOREACH(gc, &pp->lg_config, lg_config) { 110 if (strcmp(gc->lg_name, "type") == 0) 111 type = gc->lg_val; 112 if (strcmp(gc->lg_name, "descr") == 0) 113 desc = gc->lg_val; 114 } 115 116 /* Skip swap-backed md and WORM devices */ 117 if (strcmp(classp->lg_name, "MD") == 0 && 118 type != NULL && strcmp(type, "swap") == 0) 119 continue; 120 if (strncmp(pp->lg_name, "cd", 2) == 0) 121 continue; 122 123 disks = realloc(disks, (++n)*sizeof(disks[0])); 124 disks[n-1].name = pp->lg_name; 125 humanize_number(diskdesc, 7, pp->lg_mediasize, 126 "B", HN_AUTOSCALE, HN_DECIMAL); 127 if (strncmp(pp->lg_name, "ad", 2) == 0) 128 strcat(diskdesc, " ATA Hard Disk"); 129 else if (strncmp(pp->lg_name, "md", 2) == 0) 130 strcat(diskdesc, " Memory Disk"); 131 else 132 strcat(diskdesc, " Disk"); 133 134 if (desc != NULL) 135 snprintf(diskdesc, sizeof(diskdesc), 136 "%s <%s>", diskdesc, desc); 137 138 disks[n-1].text = strdup(diskdesc); 139 disks[n-1].help = NULL; 140 disks[n-1].state = 0; 141 } 142 } 143 } 144 145 if (n > 1) { 146 err = dlg_menu("Partitioning", 147 "Select the disk on which to install FreeBSD.", 0, 0, 0, 148 n, disks, &selected, NULL); 149 150 chosen = (err == 0) ? strdup(disks[selected].name) : NULL; 151 } else if (n == 1) { 152 chosen = strdup(disks[0].name); 153 } else { 154 chosen = NULL; 155 } 156 157 for (i = 0; i < n; i++) 158 free(disks[i].text); 159 160 return (chosen); 161 } 162 163 static struct gprovider * 164 provider_for_name(struct gmesh *mesh, const char *name) 165 { 166 struct gclass *classp; 167 struct gprovider *pp = NULL; 168 struct ggeom *gp; 169 170 LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 171 if (strcmp(classp->lg_name, "DISK") != 0 && 172 strcmp(classp->lg_name, "PART") != 0 && 173 strcmp(classp->lg_name, "RAID") != 0 && 174 strcmp(classp->lg_name, "MD") != 0) 175 continue; 176 177 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 178 if (LIST_EMPTY(&gp->lg_provider)) 179 continue; 180 181 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) 182 if (strcmp(pp->lg_name, name) == 0) 183 break; 184 185 if (pp != NULL) break; 186 } 187 188 if (pp != NULL) break; 189 } 190 191 return (pp); 192 } 193 194 static char * 195 wizard_partition(struct gmesh *mesh, const char *disk) 196 { 197 struct gclass *classp; 198 struct ggeom *gpart = NULL; 199 struct gconfig *gc; 200 char message[512]; 201 const char *scheme = NULL; 202 char *retval = NULL; 203 int choice; 204 205 LIST_FOREACH(classp, &mesh->lg_class, lg_class) 206 if (strcmp(classp->lg_name, "PART") == 0) 207 break; 208 209 if (classp != NULL) { 210 LIST_FOREACH(gpart, &classp->lg_geom, lg_geom) 211 if (strcmp(gpart->lg_name, disk) == 0) 212 break; 213 } 214 215 if (gpart != NULL) { 216 LIST_FOREACH(gc, &gpart->lg_config, lg_config) { 217 if (strcmp(gc->lg_name, "scheme") == 0) { 218 scheme = gc->lg_val; 219 break; 220 } 221 } 222 } 223 224 /* Treat uncommitted scheme deletions as no scheme */ 225 if (scheme != NULL && strcmp(scheme, "(none)") == 0) 226 scheme = NULL; 227 228 query: 229 dialog_vars.yes_label = "Entire Disk"; 230 dialog_vars.no_label = "Partition"; 231 if (gpart != NULL) 232 dialog_vars.defaultno = TRUE; 233 234 snprintf(message, sizeof(message), "Would you like to use this entire " 235 "disk (%s) for FreeBSD or partition it to share it with other " 236 "operating systems? Using the entire disk will erase any data " 237 "currently stored there.", disk); 238 choice = dialog_yesno("Partition", message, 0, 0); 239 240 dialog_vars.yes_label = NULL; 241 dialog_vars.no_label = NULL; 242 dialog_vars.defaultno = FALSE; 243 244 if (choice == 1 && scheme != NULL && !is_scheme_bootable(scheme)) { 245 char warning[512]; 246 int subchoice; 247 248 sprintf(warning, "The existing partition scheme on this " 249 "disk (%s) is not bootable on this platform. To install " 250 "FreeBSD, it must be repartitioned. This will destroy all " 251 "data on the disk. Are you sure you want to proceed?", 252 scheme); 253 subchoice = dialog_yesno("Non-bootable Disk", warning, 0, 0); 254 if (subchoice != 0) 255 goto query; 256 257 gpart_destroy(gpart); 258 gpart_partition(disk, default_scheme()); 259 scheme = default_scheme(); 260 } 261 262 if (scheme == NULL || choice == 0) { 263 if (gpart != NULL && scheme != NULL) { 264 /* Erase partitioned disk */ 265 choice = dialog_yesno("Confirmation", "This will erase " 266 "the disk. Are you sure you want to proceed?", 0, 0); 267 if (choice != 0) 268 goto query; 269 270 gpart_destroy(gpart); 271 } 272 273 gpart_partition(disk, default_scheme()); 274 scheme = default_scheme(); 275 } 276 277 if (strcmp(scheme, "PC98") == 0 || strcmp(scheme, "MBR") == 0) { 278 struct gmesh submesh; 279 geom_gettree(&submesh); 280 gpart_create(provider_for_name(&submesh, disk), 281 "freebsd", NULL, NULL, &retval, 282 choice /* Non-interactive for "Entire Disk" */); 283 geom_deletetree(&submesh); 284 } else { 285 retval = strdup(disk); 286 } 287 288 return (retval); 289 } 290 291 static int 292 wizard_makeparts(struct gmesh *mesh, const char *disk) 293 { 294 struct gmesh submesh; 295 struct gclass *classp; 296 struct ggeom *gp; 297 struct gprovider *pp; 298 intmax_t swapsize, available; 299 char swapsizestr[10], rootsizestr[10]; 300 int retval; 301 302 LIST_FOREACH(classp, &mesh->lg_class, lg_class) 303 if (strcmp(classp->lg_name, "PART") == 0) 304 break; 305 306 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) 307 if (strcmp(gp->lg_name, disk) == 0) 308 break; 309 310 pp = provider_for_name(mesh, disk); 311 312 available = gpart_max_free(gp, NULL)*pp->lg_sectorsize; 313 if (available < MIN_FREE_SPACE) { 314 char availablestr[10], neededstr[10], message[512]; 315 humanize_number(availablestr, 7, available, "B", HN_AUTOSCALE, 316 HN_DECIMAL); 317 humanize_number(neededstr, 7, MIN_FREE_SPACE, "B", HN_AUTOSCALE, 318 HN_DECIMAL); 319 sprintf(message, "There is not enough free space on %s to " 320 "install FreeBSD (%s free, %s required). Would you like " 321 "to choose another disk or to open the partition editor?", 322 disk, availablestr, neededstr); 323 324 dialog_vars.yes_label = "Another Disk"; 325 dialog_vars.no_label = "Editor"; 326 retval = dialog_yesno("Warning", message, 0, 0); 327 dialog_vars.yes_label = NULL; 328 dialog_vars.no_label = NULL; 329 330 return (!retval); /* Editor -> return 0 */ 331 } 332 333 swapsize = SWAP_SIZE(available); 334 humanize_number(swapsizestr, 7, swapsize, "B", HN_AUTOSCALE, 335 HN_NOSPACE | HN_DECIMAL); 336 humanize_number(rootsizestr, 7, available - swapsize - 1024*1024, 337 "B", HN_AUTOSCALE, HN_NOSPACE | HN_DECIMAL); 338 339 geom_gettree(&submesh); 340 pp = provider_for_name(&submesh, disk); 341 gpart_create(pp, "freebsd-ufs", rootsizestr, "/", NULL, 0); 342 geom_deletetree(&submesh); 343 344 geom_gettree(&submesh); 345 pp = provider_for_name(&submesh, disk); 346 gpart_create(pp, "freebsd-swap", swapsizestr, NULL, NULL, 0); 347 geom_deletetree(&submesh); 348 349 return (0); 350 } 351 352