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 <sys/sysctl.h>
35 #include <string.h>
36
37 #include <libgeom.h>
38 #include <dialog.h>
39 #include <dlg_keys.h>
40
41 #include "partedit.h"
42
43 #define MIN_FREE_SPACE (1024*1024*1024) /* 1 GB */
44 #define SWAP_SIZE(available) MIN(available/20, 4*1024*1024*1024LL)
45
46 static char *boot_disk(struct gmesh *mesh);
47 static char *wizard_partition(struct gmesh *mesh, const char *disk);
48
49 int
part_wizard(const char * fsreq)50 part_wizard(const char *fsreq) {
51 int error;
52 struct gmesh mesh;
53 char *disk, *schemeroot;
54 const char *fstype;
55
56 if (fsreq != NULL)
57 fstype = fsreq;
58 else
59 fstype = "ufs";
60
61 startwizard:
62 error = geom_gettree(&mesh);
63
64 dlg_put_backtitle();
65 error = geom_gettree(&mesh);
66 disk = boot_disk(&mesh);
67 if (disk == NULL)
68 return (1);
69
70 dlg_clear();
71 dlg_put_backtitle();
72 schemeroot = wizard_partition(&mesh, disk);
73 free(disk);
74 if (schemeroot == NULL)
75 return (1);
76
77 geom_deletetree(&mesh);
78 dlg_clear();
79 dlg_put_backtitle();
80 error = geom_gettree(&mesh);
81
82 error = wizard_makeparts(&mesh, schemeroot, fstype, 1);
83 if (error)
84 goto startwizard;
85 free(schemeroot);
86
87 geom_deletetree(&mesh);
88
89 return (0);
90 }
91
92 static char *
boot_disk(struct gmesh * mesh)93 boot_disk(struct gmesh *mesh)
94 {
95 struct gclass *classp;
96 struct gconfig *gc;
97 struct ggeom *gp;
98 struct gprovider *pp;
99 DIALOG_LISTITEM *disks = NULL;
100 const char *type, *desc;
101 char diskdesc[512];
102 char *chosen;
103 int i, err, selected, n = 0;
104
105 LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
106 if (strcmp(classp->lg_name, "DISK") != 0 &&
107 strcmp(classp->lg_name, "RAID") != 0 &&
108 strcmp(classp->lg_name, "MD") != 0)
109 continue;
110
111 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
112 if (LIST_EMPTY(&gp->lg_provider))
113 continue;
114
115 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) {
116 desc = type = NULL;
117 LIST_FOREACH(gc, &pp->lg_config, lg_config) {
118 if (strcmp(gc->lg_name, "type") == 0)
119 type = gc->lg_val;
120 if (strcmp(gc->lg_name, "descr") == 0)
121 desc = gc->lg_val;
122 }
123
124 /* Skip swap-backed md and WORM devices */
125 if (strcmp(classp->lg_name, "MD") == 0 &&
126 type != NULL && strcmp(type, "swap") == 0)
127 continue;
128 if (strncmp(pp->lg_name, "cd", 2) == 0)
129 continue;
130
131 disks = realloc(disks, (++n)*sizeof(disks[0]));
132 disks[n-1].name = pp->lg_name;
133 humanize_number(diskdesc, 7, pp->lg_mediasize,
134 "B", HN_AUTOSCALE, HN_DECIMAL);
135 if (strncmp(pp->lg_name, "ad", 2) == 0)
136 strcat(diskdesc, " ATA Hard Disk");
137 else if (strncmp(pp->lg_name, "md", 2) == 0)
138 strcat(diskdesc, " Memory Disk");
139 else
140 strcat(diskdesc, " Disk");
141
142 if (desc != NULL)
143 snprintf(diskdesc, sizeof(diskdesc),
144 "%s <%s>", diskdesc, desc);
145
146 disks[n-1].text = strdup(diskdesc);
147 disks[n-1].help = NULL;
148 disks[n-1].state = 0;
149 }
150 }
151 }
152
153 if (n > 1) {
154 err = dlg_menu("Partitioning",
155 "Select the disk on which to install FreeBSD.", 0, 0, 0,
156 n, disks, &selected, NULL);
157
158 chosen = (err == 0) ? strdup(disks[selected].name) : NULL;
159 } else if (n == 1) {
160 chosen = strdup(disks[0].name);
161 } else {
162 chosen = NULL;
163 }
164
165 for (i = 0; i < n; i++)
166 free(disks[i].text);
167
168 return (chosen);
169 }
170
171 static struct gprovider *
provider_for_name(struct gmesh * mesh,const char * name)172 provider_for_name(struct gmesh *mesh, const char *name)
173 {
174 struct gclass *classp;
175 struct gprovider *pp = NULL;
176 struct ggeom *gp;
177
178 LIST_FOREACH(classp, &mesh->lg_class, lg_class) {
179 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) {
180 if (LIST_EMPTY(&gp->lg_provider))
181 continue;
182
183 LIST_FOREACH(pp, &gp->lg_provider, lg_provider)
184 if (strcmp(pp->lg_name, name) == 0)
185 break;
186
187 if (pp != NULL) break;
188 }
189
190 if (pp != NULL) break;
191 }
192
193 return (pp);
194 }
195
196 static char *
wizard_partition(struct gmesh * mesh,const char * disk)197 wizard_partition(struct gmesh *mesh, const char *disk)
198 {
199 struct gclass *classp;
200 struct ggeom *gpart = NULL;
201 struct gconfig *gc;
202 char message[512];
203 const char *scheme = NULL;
204 char *retval = NULL;
205 int choice;
206
207 LIST_FOREACH(classp, &mesh->lg_class, lg_class)
208 if (strcmp(classp->lg_name, "PART") == 0)
209 break;
210
211 if (classp != NULL) {
212 LIST_FOREACH(gpart, &classp->lg_geom, lg_geom)
213 if (strcmp(gpart->lg_name, disk) == 0)
214 break;
215 }
216
217 if (gpart != NULL) {
218 LIST_FOREACH(gc, &gpart->lg_config, lg_config) {
219 if (strcmp(gc->lg_name, "scheme") == 0) {
220 scheme = gc->lg_val;
221 break;
222 }
223 }
224 }
225
226 /* Treat uncommitted scheme deletions as no scheme */
227 if (scheme != NULL && strcmp(scheme, "(none)") == 0)
228 scheme = NULL;
229
230 query:
231 dialog_vars.yes_label = "Entire Disk";
232 dialog_vars.no_label = "Partition";
233 if (gpart != NULL)
234 dialog_vars.defaultno = TRUE;
235
236 snprintf(message, sizeof(message), "Would you like to use this entire "
237 "disk (%s) for FreeBSD or partition it to share it with other "
238 "operating systems? Using the entire disk will erase any data "
239 "currently stored there.", disk);
240 choice = dialog_yesno("Partition", message, 0, 0);
241
242 dialog_vars.yes_label = NULL;
243 dialog_vars.no_label = NULL;
244 dialog_vars.defaultno = FALSE;
245
246 if (choice == 1 && scheme != NULL && !is_scheme_bootable(scheme)) {
247 char warning[512];
248 int subchoice;
249
250 sprintf(warning, "The existing partition scheme on this "
251 "disk (%s) is not bootable on this platform. To install "
252 "FreeBSD, it must be repartitioned. This will destroy all "
253 "data on the disk. Are you sure you want to proceed?",
254 scheme);
255 subchoice = dialog_yesno("Non-bootable Disk", warning, 0, 0);
256 if (subchoice != 0)
257 goto query;
258
259 gpart_destroy(gpart);
260 scheme = choose_part_type(default_scheme());
261 if (scheme == NULL)
262 return NULL;
263 gpart_partition(disk, scheme);
264 }
265
266 if (scheme == NULL || choice == 0) {
267 if (gpart != NULL && scheme != NULL) {
268 /* Erase partitioned disk */
269 choice = dialog_yesno("Confirmation", "This will erase "
270 "the disk. Are you sure you want to proceed?", 0, 0);
271 if (choice != 0)
272 goto query;
273
274 gpart_destroy(gpart);
275 }
276
277 scheme = choose_part_type(default_scheme());
278 if (scheme == NULL)
279 return NULL;
280 gpart_partition(disk, scheme);
281 }
282
283 if (strcmp(scheme, "PC98") == 0 || strcmp(scheme, "MBR") == 0) {
284 struct gmesh submesh;
285 geom_gettree(&submesh);
286 gpart_create(provider_for_name(&submesh, disk),
287 "freebsd", NULL, NULL, &retval,
288 choice /* Non-interactive for "Entire Disk" */);
289 geom_deletetree(&submesh);
290 } else {
291 retval = strdup(disk);
292 }
293
294 return (retval);
295 }
296
297 int
wizard_makeparts(struct gmesh * mesh,const char * disk,const char * fstype,int interactive)298 wizard_makeparts(struct gmesh *mesh, const char *disk, const char *fstype, int interactive)
299 {
300 struct gmesh submesh;
301 struct gclass *classp;
302 struct ggeom *gp;
303 struct gprovider *pp;
304 intmax_t swapsize, available;
305 char swapsizestr[10], rootsizestr[10], *fsname;
306 char *fsnames[] = {"freebsd-ufs", "freebsd-zfs"};
307 int retval;
308
309 if (strcmp(fstype, "zfs") == 0) {
310 fsname = fsnames[1];
311 } else {
312 /* default to UFS */
313 fsname = fsnames[0];
314 }
315
316 LIST_FOREACH(classp, &mesh->lg_class, lg_class)
317 if (strcmp(classp->lg_name, "PART") == 0)
318 break;
319
320 LIST_FOREACH(gp, &classp->lg_geom, lg_geom)
321 if (strcmp(gp->lg_name, disk) == 0)
322 break;
323
324 pp = provider_for_name(mesh, disk);
325
326 available = gpart_max_free(gp, NULL)*pp->lg_sectorsize;
327 if (interactive && available < MIN_FREE_SPACE) {
328 char availablestr[10], neededstr[10], message[512];
329 humanize_number(availablestr, 7, available, "B", HN_AUTOSCALE,
330 HN_DECIMAL);
331 humanize_number(neededstr, 7, MIN_FREE_SPACE, "B", HN_AUTOSCALE,
332 HN_DECIMAL);
333 sprintf(message, "There is not enough free space on %s to "
334 "install FreeBSD (%s free, %s required). Would you like "
335 "to choose another disk or to open the partition editor?",
336 disk, availablestr, neededstr);
337
338 dialog_vars.yes_label = "Another Disk";
339 dialog_vars.no_label = "Editor";
340 retval = dialog_yesno("Warning", message, 0, 0);
341 dialog_vars.yes_label = NULL;
342 dialog_vars.no_label = NULL;
343
344 return (!retval); /* Editor -> return 0 */
345 }
346
347 swapsize = SWAP_SIZE(available);
348 humanize_number(swapsizestr, 7, swapsize, "B", HN_AUTOSCALE,
349 HN_NOSPACE | HN_DECIMAL);
350 humanize_number(rootsizestr, 7, available - swapsize - 1024*1024,
351 "B", HN_AUTOSCALE, HN_NOSPACE | HN_DECIMAL);
352
353 geom_gettree(&submesh);
354 pp = provider_for_name(&submesh, disk);
355 gpart_create(pp, fsname, rootsizestr, "/", NULL, 0);
356 geom_deletetree(&submesh);
357
358 geom_gettree(&submesh);
359 pp = provider_for_name(&submesh, disk);
360 gpart_create(pp, "freebsd-swap", swapsizestr, NULL, NULL, 0);
361 geom_deletetree(&submesh);
362
363 return (0);
364 }
365
366