bsddisklabel.c revision 1.29 1 1.29 martin /* $NetBSD: bsddisklabel.c,v 1.29 2019/10/25 12:24:34 martin Exp $ */
2 1.1 dholland
3 1.1 dholland /*
4 1.1 dholland * Copyright 1997 Piermont Information Systems Inc.
5 1.1 dholland * All rights reserved.
6 1.1 dholland *
7 1.1 dholland * Based on code written by Philip A. Nelson for Piermont Information
8 1.1 dholland * Systems Inc.
9 1.1 dholland *
10 1.1 dholland * Redistribution and use in source and binary forms, with or without
11 1.1 dholland * modification, are permitted provided that the following conditions
12 1.1 dholland * are met:
13 1.1 dholland * 1. Redistributions of source code must retain the above copyright
14 1.1 dholland * notice, this list of conditions and the following disclaimer.
15 1.1 dholland * 2. Redistributions in binary form must reproduce the above copyright
16 1.1 dholland * notice, this list of conditions and the following disclaimer in the
17 1.1 dholland * documentation and/or other materials provided with the distribution.
18 1.1 dholland * 3. The name of Piermont Information Systems Inc. may not be used to endorse
19 1.1 dholland * or promote products derived from this software without specific prior
20 1.1 dholland * written permission.
21 1.1 dholland *
22 1.1 dholland * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
23 1.1 dholland * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 1.1 dholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 1.1 dholland * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
26 1.1 dholland * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
27 1.1 dholland * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
28 1.1 dholland * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
29 1.1 dholland * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
30 1.1 dholland * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
31 1.1 dholland * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
32 1.1 dholland * THE POSSIBILITY OF SUCH DAMAGE.
33 1.1 dholland */
34 1.1 dholland
35 1.1 dholland /* bsddisklabel.c -- generate standard BSD disklabel */
36 1.1 dholland /* Included by appropriate arch/XXXX/md.c */
37 1.1 dholland
38 1.1 dholland #include <sys/param.h>
39 1.1 dholland #include <sys/sysctl.h>
40 1.1 dholland #include <sys/exec.h>
41 1.1 dholland #include <sys/utsname.h>
42 1.1 dholland #include <sys/types.h>
43 1.1 dholland #include <sys/stat.h>
44 1.1 dholland #include <machine/cpu.h>
45 1.9 martin #include <assert.h>
46 1.1 dholland #include <stdio.h>
47 1.1 dholland #include <stddef.h>
48 1.1 dholland #include <util.h>
49 1.1 dholland #include <dirent.h>
50 1.1 dholland #include "defs.h"
51 1.1 dholland #include "md.h"
52 1.5 martin #include "defsizes.h"
53 1.1 dholland #include "endian.h"
54 1.1 dholland #include "msg_defs.h"
55 1.1 dholland #include "menu_defs.h"
56 1.1 dholland
57 1.9 martin static size_t fill_ptn_menu(struct partition_usage_set *pset);
58 1.1 dholland
59 1.9 martin /*
60 1.9 martin * The default partition layout.
61 1.9 martin */
62 1.9 martin static const struct part_usage_info
63 1.9 martin default_parts_init[] =
64 1.9 martin {
65 1.9 martin /*
66 1.9 martin * Pretty complex setup for boot partitions.
67 1.9 martin * This is copy&pasted below, please keep in sync!
68 1.9 martin */
69 1.9 martin #ifdef PART_BOOT
70 1.9 martin { .size = PART_BOOT/512, /* PART_BOOT is in BYTE, not MB! */
71 1.9 martin #ifdef PART_BOOT_MOUNT
72 1.9 martin .mount = PART_BOOT_MOUNT,
73 1.9 martin .instflags = PUIINST_MOUNT|PUIINST_BOOT,
74 1.9 martin #else
75 1.18 martin .instflags = PUIINST_BOOT,
76 1.9 martin #endif
77 1.9 martin #ifdef PART_BOOT_TYPE
78 1.9 martin .fs_type = PART_BOOT_TYPE,
79 1.9 martin #if PART_BOOT_TYPE == FS_MSDOS
80 1.9 martin .flags = PUIFLAG_ADD_OUTER,
81 1.9 martin #endif
82 1.9 martin #endif
83 1.9 martin #ifdef PART_BOOT_SUBT
84 1.9 martin .fs_version = PART_BOOT_SUBT,
85 1.9 martin #endif
86 1.9 martin },
87 1.9 martin #endif
88 1.9 martin
89 1.9 martin /*
90 1.9 martin * Two more copies of above for _BOOT1 and _BOOT2, please
91 1.9 martin * keep in sync!
92 1.9 martin */
93 1.9 martin #ifdef PART_BOOT1
94 1.9 martin { .size = PART_BOOT1/512, /* PART_BOOT1 is in BYTE, not MB! */
95 1.9 martin #ifdef PART_BOOT1_MOUNT
96 1.9 martin .mount = PART_BOOT1_MOUNT,
97 1.9 martin .instflags = PUIINST_MOUNT|PUIINST_BOOT,
98 1.9 martin #else
99 1.9 martin .instflags = PUIINST_MOUNT|PUIINST_BOOT,
100 1.9 martin #endif
101 1.9 martin #ifdef PART_BOOT1_TYPE
102 1.9 martin .fs_type = PART_BOOT1_TYPE,
103 1.9 martin #if PART_BOOT1_TYPE == FS_MSDOS
104 1.9 martin .flags = PUIFLAG_ADD_OUTER,
105 1.9 martin #endif
106 1.9 martin #endif
107 1.9 martin #ifdef PART_BOOT1_SUBT
108 1.9 martin .fs_version = PART_BOOT1_SUBT,
109 1.9 martin #endif
110 1.9 martin },
111 1.9 martin #endif
112 1.9 martin #ifdef PART_BOOT2
113 1.9 martin { .size = PART_BOOT2/512, /* PART_BOOT2 is in BYTE, not MB! */
114 1.9 martin #ifdef PART_BOOT2_MOUNT
115 1.9 martin .mount = PART_BOOT2_MOUNT,
116 1.9 martin .instflags = PUIINST_MOUNT|PUIINST_BOOT,
117 1.9 martin #else
118 1.9 martin .instflags = PUIINST_MOUNT|PUIINST_BOOT,
119 1.9 martin #endif
120 1.9 martin #ifdef PART_BOOT2_TYPE
121 1.9 martin .fs_type = PART_BOOT2_TYPE,
122 1.9 martin #if PART_BOOT2_TYPE == FS_MSDOS
123 1.9 martin .flags = PUIFLAG_ADD_OUTER,
124 1.9 martin #endif
125 1.1 dholland #endif
126 1.9 martin #ifdef PART_BOOT2_SUBT
127 1.9 martin .fs_version = PART_BOOT1_SUBT,
128 1.1 dholland #endif
129 1.9 martin },
130 1.1 dholland #endif
131 1.1 dholland
132 1.9 martin { .size = DEFROOTSIZE*(MEG/512), .mount = "/", .type = PT_root,
133 1.9 martin .flags = PUIFLAG_EXTEND },
134 1.9 martin {
135 1.9 martin #if DEFSWAPSIZE > 0
136 1.9 martin .size = DEFSWAPSIZE*(MEG/512),
137 1.9 martin #endif
138 1.9 martin .type = PT_swap, .fs_type = FS_SWAP },
139 1.9 martin #ifdef HAVE_TMPFS
140 1.9 martin { .type = PT_root, .mount = "/tmp", .fs_type = FS_TMPFS,
141 1.9 martin .flags = PUIFLG_JUST_MOUNTPOINT },
142 1.9 martin #else
143 1.9 martin { .type = PT_root, .mount = "/tmp", .fs_type = FS_MFS,
144 1.9 martin .flags = PUIFLG_JUST_MOUNTPOINT },
145 1.9 martin #endif
146 1.9 martin { .def_size = DEFUSRSIZE*(MEG/512), .mount = "/usr", .type = PT_root },
147 1.9 martin { .def_size = DEFVARSIZE*(MEG/512), .mount = "/var", .type = PT_root },
148 1.9 martin };
149 1.9 martin
150 1.9 martin static const char size_separator[] =
151 1.9 martin "----------------------------------- - --------------------";
152 1.9 martin static char size_menu_title[STRSIZE];
153 1.9 martin static char size_menu_exit[MENUSTRSIZE];
154 1.9 martin
155 1.9 martin static void
156 1.9 martin set_pset_exit_str(struct partition_usage_set *pset)
157 1.9 martin {
158 1.9 martin char *str, num[25];
159 1.9 martin const char *args[2];
160 1.9 martin bool overrun;
161 1.9 martin daddr_t free_space = pset->cur_free_space;
162 1.9 martin
163 1.9 martin /* format exit string */
164 1.9 martin overrun = free_space < 0;
165 1.9 martin if (overrun)
166 1.9 martin free_space = -free_space;
167 1.9 martin
168 1.9 martin snprintf(num, sizeof(num), "%" PRIu64, free_space / sizemult);
169 1.9 martin args[0] = num;
170 1.9 martin args[1] = multname;
171 1.9 martin str = str_arg_subst(
172 1.9 martin msg_string(overrun ? MSG_fssizesbad : MSG_fssizesok),
173 1.9 martin 2, args);
174 1.9 martin strlcpy(size_menu_exit, str, sizeof(size_menu_exit));
175 1.9 martin free(str);
176 1.9 martin }
177 1.9 martin
178 1.9 martin static void
179 1.9 martin draw_size_menu_header(menudesc *m, void *arg)
180 1.9 martin {
181 1.9 martin struct partition_usage_set *pset = arg;
182 1.9 martin size_t i;
183 1.9 martin char col1[70], desc[MENUSTRSIZE];
184 1.9 martin bool need_ext = false, need_existing = false;
185 1.9 martin
186 1.9 martin msg_display(MSG_ptnsizes);
187 1.9 martin
188 1.9 martin for (i = 0; i < pset->num; i++) {
189 1.9 martin if (pset->infos[i].flags & PUIFLG_IS_OUTER)
190 1.9 martin need_ext = true;
191 1.9 martin else if (pset->infos[i].cur_part_id != NO_PART)
192 1.9 martin need_existing = true;
193 1.9 martin }
194 1.9 martin if (need_ext && need_existing)
195 1.9 martin snprintf(desc, sizeof desc, "%s, %s",
196 1.9 martin msg_string(MSG_ptnsizes_mark_existing),
197 1.9 martin msg_string(MSG_ptnsizes_mark_external));
198 1.9 martin else if (need_existing)
199 1.9 martin strlcpy(desc, msg_string(MSG_ptnsizes_mark_existing),
200 1.9 martin sizeof desc);
201 1.9 martin else if (need_ext)
202 1.9 martin strlcpy(desc, msg_string(MSG_ptnsizes_mark_external),
203 1.9 martin sizeof desc);
204 1.9 martin if (need_ext || need_existing) {
205 1.9 martin msg_printf("\n");
206 1.9 martin msg_display_add_subst(msg_string(MSG_ptnsizes_markers),
207 1.9 martin 1, &desc);
208 1.9 martin }
209 1.9 martin msg_printf("\n\n");
210 1.9 martin
211 1.9 martin /* update menu title */
212 1.9 martin snprintf(col1, sizeof col1, "%s (%s)", msg_string(MSG_ptnheaders_size),
213 1.9 martin multname);
214 1.9 martin snprintf(size_menu_title, sizeof size_menu_title,
215 1.9 martin " %-37.37s %s\n %s", col1,
216 1.9 martin msg_string(MSG_ptnheaders_filesystem), size_separator);
217 1.1 dholland }
218 1.1 dholland
219 1.9 martin static void
220 1.9 martin draw_size_menu_line(menudesc *m, int opt, void *arg)
221 1.1 dholland {
222 1.9 martin struct partition_usage_set *pset = arg;
223 1.1 dholland daddr_t size;
224 1.9 martin char psize[38], inc_free[16], flag, swap[40];
225 1.9 martin const char *mount;
226 1.9 martin bool free_mount = false;
227 1.1 dholland
228 1.9 martin if (opt < 0 || (size_t)opt >= pset->num)
229 1.1 dholland return;
230 1.9 martin
231 1.9 martin inc_free[0] = 0;
232 1.9 martin if ((pset->infos[opt].flags & PUIFLAG_EXTEND) &&
233 1.9 martin pset->cur_free_space > 0) {
234 1.9 martin size = pset->infos[opt].size + pset->cur_free_space;
235 1.9 martin snprintf(inc_free, sizeof inc_free, " (%" PRIu64 ")",
236 1.9 martin size / sizemult);
237 1.9 martin }
238 1.9 martin size = pset->infos[opt].size;
239 1.9 martin snprintf(psize, sizeof psize, "%" PRIu64 "%s",
240 1.9 martin size / sizemult, inc_free);
241 1.9 martin
242 1.9 martin if (pset->infos[opt].type == PT_swap) {
243 1.9 martin snprintf(swap, sizeof swap, "<%s>",
244 1.9 martin msg_string(MSG_swap_display));
245 1.9 martin mount = swap;
246 1.9 martin } else if (pset->infos[opt].flags & PUIFLG_JUST_MOUNTPOINT) {
247 1.9 martin snprintf(swap, sizeof swap, "%s (%s)",
248 1.9 martin pset->infos[opt].mount,
249 1.9 martin getfslabelname(pset->infos[opt].fs_type,
250 1.9 martin pset->infos[opt].fs_version));
251 1.9 martin mount = swap;
252 1.14 christos } else if (pset->infos[opt].mount[0]) {
253 1.9 martin mount = pset->infos[opt].mount;
254 1.9 martin } else {
255 1.23 martin mount = NULL;
256 1.23 martin if (pset->infos[opt].parts->pscheme->other_partition_identifier
257 1.23 martin && pset->infos[opt].cur_part_id != NO_PART)
258 1.23 martin mount = pset->infos[opt].parts->pscheme->
259 1.23 martin other_partition_identifier(pset->infos[opt].parts,
260 1.23 martin pset->infos[opt].cur_part_id);
261 1.23 martin if (mount == NULL)
262 1.23 martin mount = getfslabelname(pset->infos[opt].fs_type,
263 1.23 martin pset->infos[opt].fs_version);
264 1.9 martin mount = str_arg_subst(msg_string(MSG_size_ptn_not_mounted),
265 1.9 martin 1, &mount);
266 1.9 martin free_mount = true;
267 1.9 martin }
268 1.9 martin flag = ' ';
269 1.9 martin if (pset->infos[opt].flags & PUIFLAG_EXTEND)
270 1.9 martin flag = '+';
271 1.9 martin else if (pset->infos[opt].flags & PUIFLG_IS_OUTER)
272 1.9 martin flag = '@';
273 1.9 martin else if (pset->infos[opt].cur_part_id != NO_PART)
274 1.9 martin flag = '=';
275 1.9 martin wprintw(m->mw, "%-35.35s %c %s", psize, flag, mount);
276 1.9 martin if (free_mount)
277 1.9 martin free(__UNCONST(mount));
278 1.9 martin
279 1.9 martin if (opt == 0)
280 1.9 martin set_pset_exit_str(pset);
281 1.9 martin }
282 1.9 martin
283 1.9 martin static int
284 1.9 martin add_other_ptn_size(menudesc *menu, void *arg)
285 1.9 martin {
286 1.9 martin struct partition_usage_set *pset = arg;
287 1.9 martin struct part_usage_info *p;
288 1.9 martin struct menu_ent *m;
289 1.9 martin char new_mp[MOUNTLEN], *err;
290 1.9 martin const char *args;
291 1.9 martin
292 1.9 martin for (;;) {
293 1.9 martin msg_prompt_win(partman_go?MSG_askfsmountadv:MSG_askfsmount,
294 1.9 martin -1, 18, 0, 0, NULL, new_mp, sizeof(new_mp));
295 1.9 martin if (new_mp[0] == 0)
296 1.9 martin return 0;
297 1.9 martin if (new_mp[0] != '/') {
298 1.9 martin /* we need absolute mount paths */
299 1.9 martin memmove(new_mp+1, new_mp, sizeof(new_mp)-1);
300 1.9 martin new_mp[0] = '/';
301 1.24 martin }
302 1.24 martin
303 1.24 martin /* duplicates? */
304 1.24 martin bool duplicate = false;
305 1.24 martin for (size_t i = 0; i < pset->num; i++) {
306 1.24 martin if (strcmp(pset->infos[i].mount,
307 1.24 martin new_mp) == 0) {
308 1.24 martin args = new_mp;
309 1.24 martin err = str_arg_subst(
310 1.24 martin msg_string(MSG_mp_already_exists),
311 1.24 martin 1, &args);
312 1.24 martin err_msg_win(err);
313 1.24 martin free(err);
314 1.24 martin duplicate = true;
315 1.24 martin break;
316 1.9 martin }
317 1.9 martin }
318 1.24 martin if (!duplicate)
319 1.24 martin break;
320 1.1 dholland }
321 1.9 martin
322 1.9 martin m = realloc(pset->menu_opts, (pset->num+4)*sizeof(*pset->menu_opts));
323 1.9 martin if (m == NULL)
324 1.9 martin return 0;
325 1.9 martin p = realloc(pset->infos, (pset->num+1)*sizeof(*pset->infos));
326 1.9 martin if (p == NULL)
327 1.9 martin return 0;
328 1.9 martin
329 1.9 martin pset->infos = p;
330 1.9 martin pset->menu_opts = m;
331 1.9 martin menu->opts = m;
332 1.9 martin menu->numopts = pset->num+4;
333 1.9 martin m += pset->num;
334 1.9 martin p += pset->num;
335 1.9 martin memset(m, 0, sizeof(*m));
336 1.9 martin memset(p, 0, sizeof(*p));
337 1.25 martin p->parts = pset->parts;
338 1.24 martin p->cur_part_id = NO_PART;
339 1.24 martin p->type = PT_root;
340 1.25 martin p->fs_type = FS_BSDFFS;
341 1.25 martin p->fs_version = 2;
342 1.9 martin strncpy(p->mount, new_mp, sizeof(p->mount));
343 1.9 martin
344 1.9 martin menu->cursel = pset->num;
345 1.9 martin pset->num++;
346 1.9 martin fill_ptn_menu(pset);
347 1.9 martin
348 1.9 martin return -1;
349 1.1 dholland }
350 1.1 dholland
351 1.9 martin static size_t
352 1.9 martin fill_ptn_menu(struct partition_usage_set *pset)
353 1.1 dholland {
354 1.9 martin struct part_usage_info *p;
355 1.9 martin struct disk_part_info info;
356 1.1 dholland menu_ent *m;
357 1.9 martin size_t i;
358 1.9 martin daddr_t free_space;
359 1.1 dholland
360 1.9 martin memset(pset->menu_opts, 0, (pset->num+3)*sizeof(*pset->menu_opts));
361 1.9 martin for (m = pset->menu_opts, p = pset->infos, i = 0; i < pset->num;
362 1.9 martin m++, p++, i++) {
363 1.1 dholland m->opt_action = set_ptn_size;
364 1.1 dholland }
365 1.9 martin
366 1.9 martin m->opt_name = size_separator;
367 1.9 martin m->opt_flags = OPT_IGNORE|OPT_NOSHORT;
368 1.9 martin m++;
369 1.9 martin
370 1.9 martin m->opt_name = MSG_add_another_ptn;
371 1.9 martin m->opt_action = add_other_ptn_size;
372 1.9 martin m++;
373 1.9 martin
374 1.1 dholland m->opt_name = MSG_askunits;
375 1.1 dholland m->opt_menu = MENU_sizechoice;
376 1.1 dholland m->opt_flags = OPT_SUB;
377 1.1 dholland m++;
378 1.1 dholland
379 1.9 martin /* calculate free space */
380 1.9 martin free_space = pset->parts->free_space;
381 1.9 martin for (i = 0; i < pset->parts->num_part; i++) {
382 1.9 martin if (!pset->parts->pscheme->get_part_info(pset->parts, i,
383 1.9 martin &info))
384 1.9 martin continue;
385 1.9 martin if (info.flags & (PTI_SEC_CONTAINER|PTI_WHOLE_DISK|
386 1.9 martin PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
387 1.9 martin continue;
388 1.9 martin free_space += info.size;
389 1.9 martin }
390 1.9 martin for (i = 0; i < pset->num; i++) {
391 1.9 martin if (pset->infos[i].flags &
392 1.9 martin (PUIFLG_IS_OUTER|PUIFLG_JUST_MOUNTPOINT))
393 1.9 martin continue;
394 1.9 martin free_space -= pset->infos[i].size;
395 1.9 martin }
396 1.9 martin pset->cur_free_space = free_space;
397 1.9 martin set_pset_exit_str(pset);
398 1.9 martin
399 1.9 martin if (pset->menu >= 0)
400 1.9 martin set_menu_numopts(pset->menu, m - pset->menu_opts);
401 1.9 martin
402 1.9 martin return m - pset->menu_opts;
403 1.9 martin }
404 1.9 martin
405 1.9 martin static part_id
406 1.9 martin find_part_at(struct disk_partitions *parts, daddr_t start)
407 1.9 martin {
408 1.9 martin size_t i;
409 1.9 martin struct disk_part_info info;
410 1.9 martin
411 1.9 martin for (i = 0; i < parts->num_part; i++) {
412 1.9 martin if (!parts->pscheme->get_part_info(parts, i, &info))
413 1.9 martin continue;
414 1.9 martin if (info.start == start)
415 1.9 martin return i;
416 1.9 martin }
417 1.1 dholland
418 1.9 martin return NO_PART;
419 1.1 dholland }
420 1.1 dholland
421 1.1 dholland int
422 1.1 dholland set_ptn_size(menudesc *m, void *arg)
423 1.1 dholland {
424 1.9 martin struct partition_usage_set *pset = arg;
425 1.9 martin struct part_usage_info *p = &pset->infos[m->cursel];
426 1.9 martin char answer[16], dflt[16];
427 1.9 martin const char *err_msg;
428 1.9 martin size_t i, root = ~0U;
429 1.9 martin daddr_t size, old_size, new_size_val, mult;
430 1.9 martin int rv;
431 1.9 martin bool non_zero, extend;
432 1.1 dholland
433 1.9 martin if (pset->cur_free_space == 0 && p->size == 0 &&
434 1.9 martin !(p->flags & PUIFLG_JUST_MOUNTPOINT))
435 1.1 dholland /* Don't allow 'free_parts' to go negative */
436 1.1 dholland return 0;
437 1.1 dholland
438 1.9 martin if (p->cur_part_id != NO_PART) {
439 1.9 martin rv = 0;
440 1.9 martin process_menu(MENU_ptnsize_replace_existing_partition, &rv);
441 1.9 martin if (rv == 0)
442 1.9 martin return 0;
443 1.9 martin if (!pset->parts->pscheme->delete_partition(pset->parts,
444 1.9 martin p->cur_part_id, &err_msg)) {
445 1.9 martin if (err_msg)
446 1.9 martin err_msg_win(err_msg);
447 1.1 dholland return 0;
448 1.9 martin }
449 1.9 martin p->cur_part_id = NO_PART;
450 1.9 martin /*
451 1.9 martin * All other part ids are invalid now too - update them!
452 1.9 martin */
453 1.9 martin for (i = 0; i < pset->num; i++) {
454 1.9 martin if (pset->infos[i].cur_part_id == NO_PART)
455 1.9 martin continue;
456 1.9 martin pset->infos[i].cur_part_id =
457 1.9 martin find_part_at(pset->parts, pset->infos[i].cur_start);
458 1.9 martin }
459 1.1 dholland }
460 1.1 dholland
461 1.1 dholland size = p->size;
462 1.1 dholland old_size = size;
463 1.1 dholland if (size == 0)
464 1.9 martin size = p->def_size;
465 1.1 dholland size /= sizemult;
466 1.9 martin snprintf(dflt, sizeof dflt, "%" PRIu64 "%s",
467 1.9 martin size, p->flags & PUIFLAG_EXTEND ? "+" : "");
468 1.1 dholland
469 1.1 dholland for (;;) {
470 1.13 christos msg_fmt_prompt_win(MSG_askfssize, -1, 18, 0, 0,
471 1.13 christos dflt, answer, sizeof answer, "%s%s", p->mount, multname);
472 1.9 martin
473 1.9 martin /* cp will be checked below */
474 1.9 martin mult = sizemult;
475 1.9 martin new_size_val = parse_disk_pos(answer, &mult, pm->dlcylsize,
476 1.9 martin &extend);
477 1.9 martin
478 1.9 martin if (strcmp(answer, dflt) == 0)
479 1.9 martin non_zero = p->def_size > 0;
480 1.9 martin else
481 1.9 martin non_zero = new_size_val > 0;
482 1.9 martin
483 1.1 dholland /* Some special cases when /usr is first given a size */
484 1.9 martin if (old_size == 0 && non_zero &&
485 1.9 martin strcmp(p->mount, "/usr") == 0) {
486 1.9 martin for (i = 0; i < pset->num; i++) {
487 1.9 martin if (strcmp(pset->infos[i].mount, "/") == 0) {
488 1.9 martin root = i;
489 1.9 martin break;
490 1.9 martin }
491 1.9 martin }
492 1.1 dholland /* Remove space for /usr from / */
493 1.9 martin if (root < pset->num && pset->infos[i].cur_part_id ==
494 1.9 martin NO_PART) {
495 1.9 martin pset->infos[root].size -= p->def_size;
496 1.9 martin pset->cur_free_space += p->def_size;
497 1.1 dholland }
498 1.1 dholland /* hack to add free space to default sized /usr */
499 1.9 martin if (strcmp(answer, dflt) == 0) {
500 1.9 martin size = p->def_size;
501 1.9 martin pset->infos[root].flags &= ~PUIFLAG_EXTEND;
502 1.9 martin p->flags |= PUIFLAG_EXTEND;
503 1.1 dholland goto adjust_free;
504 1.1 dholland }
505 1.1 dholland }
506 1.9 martin if (new_size_val < 0)
507 1.1 dholland continue;
508 1.9 martin size = new_size_val;
509 1.9 martin break;
510 1.1 dholland }
511 1.1 dholland
512 1.9 martin daddr_t align = pset->parts->pscheme->get_part_alignment(pset->parts);
513 1.9 martin size = NUMSEC(size, mult, align);
514 1.9 martin if (p->flags & PUIFLAG_EXTEND)
515 1.9 martin p->flags &= ~PUIFLAG_EXTEND;
516 1.9 martin if (extend && (p->limit == 0 || p->limit > p->size)) {
517 1.9 martin p->flags |= PUIFLAG_EXTEND;
518 1.1 dholland if (size == 0)
519 1.9 martin size = align;
520 1.1 dholland }
521 1.1 dholland if (p->limit != 0 && size > p->limit)
522 1.1 dholland size = p->limit;
523 1.1 dholland adjust_free:
524 1.9 martin if ((p->flags & (PUIFLG_IS_OUTER|PUIFLG_JUST_MOUNTPOINT)) == 0)
525 1.9 martin pset->cur_free_space += p->size - size;
526 1.1 dholland p->size = size;
527 1.9 martin set_pset_exit_str(pset);
528 1.1 dholland
529 1.1 dholland return 0;
530 1.1 dholland }
531 1.1 dholland
532 1.9 martin /*
533 1.9 martin * User interface to edit a "wanted" partition layout "pset" as first
534 1.9 martin * abstract phase (not concrete partitions).
535 1.9 martin * Make sure to have everything (at least theoretically) fit the
536 1.9 martin * available space.
537 1.9 martin * During editing we keep the part_usage_info and the menu_opts
538 1.9 martin * in pset in sync, that is: we always allocate just enough entries
539 1.9 martin * in pset->infos as we have usage infos in the list (pset->num),
540 1.9 martin * and two additional menu entries ("add a partition" and "select units").
541 1.9 martin * The menu exit string changes depending on content, and implies
542 1.9 martin * abort while the partition set is not valid (does not fit).
543 1.9 martin * Return true when the user wants to continue (by editing the concrete
544 1.9 martin * partitions), return false to abort.
545 1.9 martin */
546 1.9 martin bool
547 1.9 martin get_ptn_sizes(struct partition_usage_set *pset)
548 1.9 martin {
549 1.9 martin size_t num;
550 1.9 martin
551 1.9 martin wclear(stdscr);
552 1.9 martin wrefresh(stdscr);
553 1.9 martin
554 1.9 martin if (pset->menu_opts == NULL)
555 1.9 martin pset->menu_opts = calloc(pset->num+3, sizeof(*pset->menu_opts));
556 1.9 martin
557 1.9 martin pset->menu = -1;
558 1.9 martin num = fill_ptn_menu(pset);
559 1.9 martin
560 1.9 martin pset->menu = new_menu(size_menu_title, pset->menu_opts, num,
561 1.9 martin 3, -1, 12, 70,
562 1.9 martin MC_ALWAYS_SCROLL|MC_NOBOX|MC_NOCLEAR|MC_CONTINUOUS,
563 1.9 martin draw_size_menu_header, draw_size_menu_line, NULL,
564 1.9 martin NULL, size_menu_exit);
565 1.9 martin
566 1.9 martin if (pset->menu < 0) {
567 1.9 martin free(pset->menu_opts);
568 1.9 martin pset->menu_opts = NULL;
569 1.9 martin return false;
570 1.9 martin }
571 1.1 dholland
572 1.9 martin pset->ok = true;
573 1.9 martin process_menu(pset->menu, pset);
574 1.1 dholland
575 1.9 martin free_menu(pset->menu);
576 1.9 martin free(pset->menu_opts);
577 1.9 martin pset->menu = -1;
578 1.9 martin pset->menu_opts = NULL;
579 1.1 dholland
580 1.27 martin return pset->ok;
581 1.9 martin }
582 1.9 martin
583 1.9 martin static int
584 1.9 martin set_keep_existing(menudesc *m, void *arg)
585 1.9 martin {
586 1.9 martin ((arg_rep_int*)arg)->rv = LY_KEEPEXISTING;
587 1.9 martin return 0;
588 1.9 martin }
589 1.9 martin
590 1.9 martin static int
591 1.9 martin set_edit_part_sizes(menudesc *m, void *arg)
592 1.9 martin {
593 1.9 martin ((arg_rep_int*)arg)->rv = LY_SETSIZES;
594 1.9 martin return 0;
595 1.9 martin }
596 1.9 martin
597 1.9 martin static int
598 1.9 martin set_use_default_sizes(menudesc *m, void *arg)
599 1.9 martin {
600 1.9 martin ((arg_rep_int*)arg)->rv = LY_USEDEFAULT;
601 1.9 martin return 0;
602 1.1 dholland }
603 1.1 dholland
604 1.1 dholland /*
605 1.9 martin * Check if there is a reasonable pre-existing partition for
606 1.9 martin * NetBSD.
607 1.9 martin */
608 1.9 martin static bool
609 1.9 martin check_existing_netbsd(struct disk_partitions *parts)
610 1.9 martin {
611 1.9 martin size_t nbsd_parts;
612 1.9 martin struct disk_part_info info;
613 1.9 martin
614 1.9 martin nbsd_parts = 0;
615 1.9 martin for (part_id p = 0; p < parts->num_part; p++) {
616 1.17 martin if (!parts->pscheme->get_part_info(parts, p, &info))
617 1.17 martin continue;
618 1.17 martin if (info.flags & (PTI_PSCHEME_INTERNAL|PTI_RAW_PART))
619 1.17 martin continue;
620 1.17 martin if (info.nat_type && info.nat_type->generic_ptype == PT_root)
621 1.9 martin nbsd_parts++;
622 1.9 martin }
623 1.9 martin
624 1.9 martin return nbsd_parts > 0;
625 1.9 martin }
626 1.9 martin
627 1.9 martin /*
628 1.9 martin * Query a partition layout type (with available options depending on
629 1.9 martin * pre-existing partitions).
630 1.1 dholland */
631 1.9 martin static enum layout_type
632 1.9 martin ask_layout(struct disk_partitions *parts, bool have_existing)
633 1.9 martin {
634 1.9 martin arg_rep_int ai;
635 1.9 martin const char *args[2];
636 1.9 martin int menu;
637 1.9 martin size_t num_opts;
638 1.9 martin menu_ent options[3], *opt;
639 1.9 martin
640 1.9 martin args[0] = msg_string(parts->pscheme->name);
641 1.9 martin args[1] = msg_string(parts->pscheme->short_name);
642 1.9 martin ai.args.argv = args;
643 1.9 martin ai.args.argc = 2;
644 1.9 martin ai.rv = LY_SETSIZES;
645 1.9 martin
646 1.9 martin memset(options, 0, sizeof(options));
647 1.9 martin num_opts = 0;
648 1.9 martin opt = &options[0];
649 1.9 martin
650 1.9 martin if (have_existing) {
651 1.9 martin opt->opt_name = MSG_Keep_existing_partitions;
652 1.9 martin opt->opt_flags = OPT_EXIT;
653 1.9 martin opt->opt_action = set_keep_existing;
654 1.9 martin opt++;
655 1.9 martin num_opts++;
656 1.9 martin }
657 1.9 martin opt->opt_name = MSG_Set_Sizes;
658 1.9 martin opt->opt_flags = OPT_EXIT;
659 1.9 martin opt->opt_action = set_edit_part_sizes;
660 1.9 martin opt++;
661 1.9 martin num_opts++;
662 1.9 martin
663 1.9 martin opt->opt_name = MSG_Use_Default_Parts;
664 1.9 martin opt->opt_flags = OPT_EXIT;
665 1.9 martin opt->opt_action = set_use_default_sizes;
666 1.9 martin opt++;
667 1.9 martin num_opts++;
668 1.9 martin
669 1.9 martin menu = new_menu(MSG_Select_your_choice, options, num_opts,
670 1.9 martin -1, -10, 0, 0, MC_NOEXITOPT, NULL, NULL, NULL, NULL, NULL);
671 1.9 martin if (menu != -1) {
672 1.9 martin get_menudesc(menu)->expand_act = expand_all_option_texts;
673 1.9 martin process_menu(menu, &ai);
674 1.9 martin free_menu(menu);
675 1.9 martin }
676 1.9 martin
677 1.9 martin return ai.rv;
678 1.9 martin }
679 1.9 martin
680 1.9 martin static void
681 1.9 martin merge_part_with_wanted(struct disk_partitions *parts, part_id pno,
682 1.9 martin const struct disk_part_info *info, struct partition_usage_set *wanted,
683 1.22 martin size_t wanted_num, bool is_outer)
684 1.1 dholland {
685 1.9 martin struct part_usage_info *infos;
686 1.2 martin
687 1.9 martin /*
688 1.9 martin * does this partition match something in the wanted set?
689 1.9 martin */
690 1.22 martin for (size_t i = 0; i < wanted_num; i++) {
691 1.9 martin if (wanted->infos[i].type != info->nat_type->generic_ptype)
692 1.9 martin continue;
693 1.22 martin if (wanted->infos[i].type == PT_root &&
694 1.22 martin info->last_mounted != NULL && info->last_mounted[0] != 0 &&
695 1.9 martin strcmp(info->last_mounted, wanted->infos[i].mount) != 0)
696 1.9 martin continue;
697 1.9 martin if (wanted->infos[i].cur_part_id != NO_PART)
698 1.9 martin continue;
699 1.9 martin wanted->infos[i].cur_part_id = pno;
700 1.9 martin wanted->infos[i].parts = parts;
701 1.9 martin wanted->infos[i].size = info->size;
702 1.9 martin wanted->infos[i].cur_start = info->start;
703 1.9 martin wanted->infos[i].flags &= ~PUIFLAG_EXTEND;
704 1.11 martin if (wanted->infos[i].fs_type != FS_UNUSED &&
705 1.11 martin wanted->infos[i].type != PT_swap)
706 1.11 martin wanted->infos[i].instflags |= PUIINST_MOUNT;
707 1.9 martin if (is_outer)
708 1.9 martin wanted->infos[i].flags |= PUIFLG_IS_OUTER;
709 1.22 martin else
710 1.22 martin wanted->infos[i].flags &= ~PUIFLG_IS_OUTER;
711 1.9 martin return;
712 1.9 martin }
713 1.4 martin
714 1.9 martin /*
715 1.9 martin * no match - if this is fromt the outer scheme, we are done.
716 1.9 martin * otherwise it must be inserted into the wanted set.
717 1.9 martin */
718 1.9 martin if (is_outer)
719 1.9 martin return;
720 1.1 dholland
721 1.1 dholland /*
722 1.9 martin * create a new entry for this
723 1.1 dholland */
724 1.9 martin infos = realloc(wanted->infos, sizeof(*infos)*(wanted->num+1));
725 1.9 martin if (infos == NULL)
726 1.9 martin return;
727 1.9 martin wanted->infos = infos;
728 1.9 martin infos += wanted->num;
729 1.9 martin wanted->num++;
730 1.9 martin memset(infos, 0, sizeof(*infos));
731 1.9 martin if (info->last_mounted != NULL && info->last_mounted[0] != 0)
732 1.9 martin strlcpy(infos->mount, info->last_mounted,
733 1.9 martin sizeof(infos->mount));
734 1.9 martin infos->type = info->nat_type->generic_ptype;
735 1.9 martin infos->cur_part_id = pno;
736 1.9 martin infos->parts = parts;
737 1.9 martin infos->size = info->size;
738 1.9 martin infos->cur_start = info->start;
739 1.9 martin infos->fs_type = info->fs_type;
740 1.9 martin infos->fs_version = info->fs_sub_type;
741 1.9 martin if (is_outer)
742 1.9 martin infos->flags |= PUIFLG_IS_OUTER;
743 1.9 martin }
744 1.9 martin
745 1.9 martin static bool
746 1.9 martin have_x11_by_default(void)
747 1.9 martin {
748 1.9 martin static const uint8_t def_sets[] = { MD_SETS_SELECTED };
749 1.9 martin
750 1.9 martin for (size_t i = 0; i < __arraycount(def_sets); i++)
751 1.9 martin if (def_sets[i] >= SET_X11_FIRST &&
752 1.9 martin def_sets[i] <= SET_X11_LAST)
753 1.9 martin return true;
754 1.9 martin
755 1.9 martin return false;
756 1.9 martin }
757 1.1 dholland
758 1.9 martin static void
759 1.9 martin fill_defaults(struct partition_usage_set *wanted, struct disk_partitions *parts,
760 1.9 martin daddr_t ptstart, daddr_t ptsize)
761 1.9 martin {
762 1.15 martin size_t i, root = ~0U, usr = ~0U, swap = ~0U, def_usr = ~0U;
763 1.15 martin daddr_t free_space, dump_space, required;
764 1.9 martin #if defined(DEFAULT_UFS2) && !defined(HAVE_UFS2_BOOT)
765 1.9 martin size_t boot = ~0U;
766 1.9 martin #endif
767 1.1 dholland
768 1.9 martin memset(wanted, 0, sizeof(*wanted));
769 1.9 martin wanted->parts = parts;
770 1.9 martin wanted->num = __arraycount(default_parts_init);
771 1.9 martin wanted->infos = calloc(wanted->num, sizeof(*wanted->infos));
772 1.9 martin if (wanted->infos == NULL) {
773 1.9 martin err_msg_win(err_outofmem);
774 1.9 martin return;
775 1.1 dholland }
776 1.1 dholland
777 1.9 martin memcpy(wanted->infos, default_parts_init, sizeof(default_parts_init));
778 1.18 martin
779 1.18 martin #ifdef MD_PART_DEFAULTS
780 1.18 martin MD_PART_DEFAULTS(pm, wanted->infos, wanted->num);
781 1.18 martin #endif
782 1.18 martin
783 1.9 martin for (i = 0; i < wanted->num; i++) {
784 1.9 martin wanted->infos[i].parts = parts;
785 1.9 martin wanted->infos[i].cur_part_id = NO_PART;
786 1.9 martin
787 1.9 martin #if DEFSWAPSIZE == -1
788 1.9 martin if (wanted->infos[i].type == PT_swap)
789 1.9 martin wanted->infos[i].size = get_ramsize() * (MEG / 512);
790 1.1 dholland #endif
791 1.9 martin if (wanted->infos[i].type == PT_swap && swap > wanted->num)
792 1.9 martin swap = i;
793 1.9 martin #if defined(DEFAULT_UFS2) && !defined(HAVE_UFS2_BOOT)
794 1.9 martin if (wanted->infos[i].instflags & PUIINST_BOOT)
795 1.9 martin boot = i;
796 1.1 dholland #endif
797 1.15 martin if (wanted->infos[i].type == PT_root) {
798 1.15 martin if (strcmp(wanted->infos[i].mount, "/") == 0) {
799 1.9 martin root = i;
800 1.15 martin } else if (
801 1.15 martin strcmp(wanted->infos[i].mount, "/usr") == 0) {
802 1.15 martin if (wanted->infos[i].size > 0)
803 1.15 martin usr = i;
804 1.15 martin else
805 1.15 martin def_usr = i;
806 1.15 martin }
807 1.21 martin if (wanted->infos[i].fs_type == FS_UNUSED)
808 1.21 martin wanted->infos[i].fs_type = FS_BSDFFS;
809 1.21 martin if (wanted->infos[i].fs_type == FS_BSDFFS) {
810 1.9 martin #ifdef DEFAULT_UFS2
811 1.9 martin #ifndef HAVE_UFS2_BOOT
812 1.21 martin if (boot < wanted->num || i != root)
813 1.1 dholland #endif
814 1.21 martin wanted->infos[i].fs_version = 2;
815 1.1 dholland #endif
816 1.21 martin }
817 1.9 martin }
818 1.9 martin if ((wanted->infos[i].flags & PUIFLG_JUST_MOUNTPOINT) &&
819 1.9 martin wanted->infos[i].size == 0)
820 1.9 martin /* default tmpfs to 1/4 RAM */
821 1.9 martin wanted->infos[i].def_size =
822 1.9 martin get_ramsize() * (MEG/512/4);
823 1.1 dholland }
824 1.1 dholland
825 1.9 martin /*
826 1.9 martin * Now we have the defaults as if we were installing to an
827 1.9 martin * empty disk. Merge the partitions in target range that are already
828 1.9 martin * there (match with wanted) or are there additionaly.
829 1.9 martin * The only thing outside of target range that we care for
830 1.9 martin * is a potential swap partition - we assume one is enough.
831 1.9 martin */
832 1.22 martin size_t num = wanted->num;
833 1.9 martin if (parts->parent) {
834 1.9 martin for (part_id pno = 0; pno < parts->parent->num_part; pno++) {
835 1.9 martin struct disk_part_info info;
836 1.1 dholland
837 1.9 martin if (!parts->parent->pscheme->get_part_info(
838 1.9 martin parts->parent, pno, &info))
839 1.9 martin continue;
840 1.9 martin if (info.nat_type->generic_ptype != PT_swap)
841 1.1 dholland continue;
842 1.9 martin merge_part_with_wanted(parts->parent, pno, &info,
843 1.22 martin wanted, num, true);
844 1.9 martin break;
845 1.9 martin }
846 1.9 martin }
847 1.9 martin for (part_id pno = 0; pno < parts->num_part; pno++) {
848 1.9 martin struct disk_part_info info;
849 1.9 martin
850 1.9 martin if (!parts->pscheme->get_part_info(parts, pno, &info))
851 1.9 martin continue;
852 1.9 martin
853 1.9 martin if (info.flags & PTI_PSCHEME_INTERNAL)
854 1.9 martin continue;
855 1.9 martin
856 1.9 martin if (info.nat_type->generic_ptype != PT_swap &&
857 1.9 martin (info.start < ptstart ||
858 1.9 martin (info.start + info.size) > (ptstart+ptsize)))
859 1.9 martin continue;
860 1.9 martin
861 1.9 martin merge_part_with_wanted(parts, pno, &info,
862 1.22 martin wanted, num, false);
863 1.9 martin }
864 1.9 martin
865 1.9 martin daddr_t align = parts->pscheme->get_part_alignment(parts);
866 1.9 martin
867 1.10 martin if (root < wanted->num && wanted->infos[root].cur_part_id == NO_PART) {
868 1.9 martin daddr_t max_root_size = parts->disk_start + parts->disk_size;
869 1.9 martin if (root_limit > 0) {
870 1.9 martin /* Bah - bios can not read all the disk, limit root */
871 1.9 martin max_root_size = root_limit - parts->disk_start;
872 1.9 martin }
873 1.9 martin wanted->infos[root].limit = max_root_size;
874 1.9 martin }
875 1.9 martin
876 1.9 martin if (have_x11_by_default()) {
877 1.9 martin daddr_t xsize = XNEEDMB * (MEG / 512);
878 1.9 martin if (usr < wanted->num) {
879 1.10 martin if (wanted->infos[usr].cur_part_id == NO_PART) {
880 1.10 martin wanted->infos[usr].size += xsize;
881 1.10 martin wanted->infos[usr].def_size += xsize;
882 1.10 martin }
883 1.9 martin } else if (root < wanted->num &&
884 1.10 martin wanted->infos[root].cur_part_id == NO_PART &&
885 1.10 martin (wanted->infos[root].limit == 0 ||
886 1.9 martin (wanted->infos[root].size + xsize) <=
887 1.10 martin wanted->infos[root].limit)) {
888 1.9 martin wanted->infos[root].size += xsize;
889 1.9 martin }
890 1.9 martin }
891 1.10 martin if (wanted->infos[root].limit > 0 &&
892 1.10 martin wanted->infos[root].size > wanted->infos[root].limit) {
893 1.9 martin if (usr < wanted->num) {
894 1.9 martin /* move space from root to usr */
895 1.9 martin daddr_t spill = wanted->infos[root].size -
896 1.9 martin wanted->infos[root].limit;
897 1.9 martin spill = roundup(spill, align);
898 1.9 martin wanted->infos[root].size =
899 1.9 martin wanted->infos[root].limit;
900 1.9 martin wanted->infos[usr].size = spill;
901 1.9 martin } else {
902 1.9 martin wanted->infos[root].size =
903 1.9 martin wanted->infos[root].limit;
904 1.1 dholland }
905 1.9 martin }
906 1.9 martin
907 1.9 martin /*
908 1.9 martin * Preliminary calc additional space to allocate and how much
909 1.9 martin * we likely will have left over. Use that to do further
910 1.9 martin * adjustments, so we don't present the user inherently
911 1.9 martin * impossible defaults.
912 1.9 martin */
913 1.9 martin free_space = parts->free_space;
914 1.15 martin required = 0;
915 1.15 martin if (root < wanted->num)
916 1.15 martin required += wanted->infos[root].size;
917 1.15 martin if (usr < wanted->num)
918 1.15 martin required += wanted->infos[usr].size;
919 1.15 martin else if (def_usr < wanted->num)
920 1.15 martin required += wanted->infos[def_usr].def_size;
921 1.15 martin free_space -= required;
922 1.9 martin for (i = 0; i < wanted->num; i++) {
923 1.15 martin if (i == root || i == usr)
924 1.15 martin continue; /* already accounted above */
925 1.9 martin if (wanted->infos[i].cur_part_id != NO_PART)
926 1.9 martin continue;
927 1.15 martin if (wanted->infos[i].size == 0)
928 1.15 martin continue;
929 1.9 martin if (wanted->infos[i].flags
930 1.9 martin & (PUIFLG_IS_OUTER|PUIFLG_JUST_MOUNTPOINT))
931 1.9 martin continue;
932 1.9 martin free_space -= wanted->infos[i].size;
933 1.9 martin }
934 1.9 martin if (free_space < 0 && swap < wanted->num) {
935 1.9 martin /* steel from swap partition */
936 1.9 martin daddr_t d = wanted->infos[swap].size;
937 1.9 martin daddr_t inc = roundup(-free_space, align);
938 1.9 martin if (inc > d)
939 1.9 martin inc = d;
940 1.9 martin free_space += inc;
941 1.9 martin wanted->infos[swap].size -= inc;
942 1.9 martin }
943 1.9 martin if (root < wanted->num) {
944 1.9 martin /* Add space for 2 system dumps to / (traditional) */
945 1.9 martin dump_space = get_ramsize() * (MEG/512);
946 1.9 martin dump_space = roundup(dump_space, align);
947 1.9 martin if (free_space > dump_space*2)
948 1.9 martin dump_space *= 2;
949 1.9 martin if (free_space > dump_space)
950 1.9 martin wanted->infos[root].size += dump_space;
951 1.9 martin }
952 1.9 martin }
953 1.9 martin
954 1.9 martin /*
955 1.9 martin * We sort pset->infos to sync with pset->parts and
956 1.9 martin * the cur_part_id, to allow using the same index into both
957 1.9 martin * "array" in later phases. This may include inserting
958 1.9 martin * dummy entries (when we do not actually want the
959 1.9 martin * partition, but it is forced upon us, like RAW_PART in
960 1.9 martin * disklabel).
961 1.9 martin */
962 1.9 martin static void
963 1.9 martin sort_and_sync_parts(struct partition_usage_set *pset)
964 1.9 martin {
965 1.9 martin struct part_usage_info *infos;
966 1.9 martin size_t i, j, no;
967 1.9 martin part_id pno;
968 1.9 martin
969 1.9 martin pset->cur_free_space = pset->parts->free_space;
970 1.9 martin
971 1.9 martin /* count non-empty entries that are not in pset->parts */
972 1.9 martin no = pset->parts->num_part;
973 1.9 martin for (i = 0; i < pset->num; i++) {
974 1.9 martin if (pset->infos[i].size == 0)
975 1.9 martin continue;
976 1.9 martin if (pset->infos[i].cur_part_id != NO_PART)
977 1.9 martin continue;
978 1.9 martin no++;
979 1.9 martin }
980 1.9 martin
981 1.9 martin /* allocate new infos */
982 1.9 martin infos = calloc(no, sizeof *infos);
983 1.9 martin if (infos == NULL)
984 1.9 martin return;
985 1.9 martin
986 1.9 martin /* pre-initialize the first entires as dummy entries */
987 1.9 martin for (i = 0; i < pset->parts->num_part; i++) {
988 1.9 martin infos[i].cur_part_id = NO_PART;
989 1.9 martin infos[i].cur_flags = PTI_PSCHEME_INTERNAL;
990 1.9 martin }
991 1.9 martin /*
992 1.9 martin * Now copy over eveything from our old entries that points to
993 1.9 martin * a real partition.
994 1.9 martin */
995 1.9 martin for (i = 0; i < pset->num; i++) {
996 1.9 martin pno = pset->infos[i].cur_part_id;
997 1.9 martin if (pno == NO_PART)
998 1.9 martin continue;
999 1.9 martin if (pset->parts != pset->infos[i].parts)
1000 1.9 martin continue;
1001 1.19 martin if (pset->infos[i].flags & PUIFLG_JUST_MOUNTPOINT)
1002 1.19 martin continue;
1003 1.19 martin if ((pset->infos[i].flags & (PUIFLG_IS_OUTER|PUIFLAG_ADD_INNER))
1004 1.19 martin == PUIFLG_IS_OUTER)
1005 1.9 martin continue;
1006 1.9 martin if (pno >= pset->parts->num_part)
1007 1.9 martin continue;
1008 1.9 martin memcpy(infos+pno, pset->infos+i, sizeof(*infos));
1009 1.9 martin }
1010 1.9 martin /* Fill in the infos for real partitions where we had no data */
1011 1.9 martin for (pno = 0; pno < pset->parts->num_part; pno++) {
1012 1.9 martin struct disk_part_info info;
1013 1.9 martin
1014 1.9 martin if (infos[pno].cur_part_id != NO_PART)
1015 1.9 martin continue;
1016 1.9 martin
1017 1.9 martin if (!pset->parts->pscheme->get_part_info(pset->parts, pno,
1018 1.9 martin &info))
1019 1.9 martin continue;
1020 1.9 martin
1021 1.9 martin infos[pno].parts = pset->parts;
1022 1.9 martin infos[pno].cur_part_id = pno;
1023 1.9 martin infos[pno].cur_flags = info.flags;
1024 1.9 martin infos[pno].size = info.size;
1025 1.9 martin infos[pno].type = info.nat_type->generic_ptype;
1026 1.9 martin infos[pno].cur_start = info.start;
1027 1.9 martin infos[pno].fs_type = info.fs_type;
1028 1.9 martin infos[pno].fs_version = info.fs_sub_type;
1029 1.9 martin }
1030 1.9 martin /* Add the non-partition entires after that */
1031 1.9 martin j = pset->num;
1032 1.9 martin for (i = 0; i < pset->num; i++) {
1033 1.9 martin if (j >= no)
1034 1.9 martin break;
1035 1.9 martin if (pset->infos[i].size == 0)
1036 1.9 martin continue;
1037 1.9 martin if (pset->infos[i].cur_part_id != NO_PART)
1038 1.9 martin continue;
1039 1.9 martin memcpy(infos+j, pset->infos+i, sizeof(*infos));
1040 1.9 martin j++;
1041 1.9 martin }
1042 1.9 martin
1043 1.9 martin /* done, replace infos */
1044 1.9 martin free(pset->infos);
1045 1.9 martin pset->num = no;
1046 1.9 martin pset->infos = infos;
1047 1.9 martin }
1048 1.9 martin
1049 1.9 martin static void
1050 1.9 martin apply_settings_to_partitions(struct pm_devs *p, struct disk_partitions *parts,
1051 1.9 martin struct partition_usage_set *wanted, daddr_t start, daddr_t size)
1052 1.9 martin {
1053 1.9 martin size_t i, exp_ndx = ~0U;
1054 1.9 martin daddr_t planned_space = 0, nsp, from, align;
1055 1.9 martin struct disk_part_info *infos;
1056 1.9 martin struct disk_part_free_space space;
1057 1.9 martin struct disk_partitions *ps = NULL;
1058 1.9 martin part_id pno, new_part_id;
1059 1.9 martin
1060 1.9 martin infos = calloc(wanted->num, sizeof(*infos));
1061 1.9 martin if (infos == NULL) {
1062 1.9 martin err_msg_win(err_outofmem);
1063 1.9 martin return;
1064 1.9 martin }
1065 1.9 martin
1066 1.9 martin align = wanted->parts->pscheme->get_part_alignment(wanted->parts);
1067 1.9 martin
1068 1.9 martin /*
1069 1.9 martin * Pass one: calculate space available for expanding
1070 1.9 martin * the marked partition.
1071 1.9 martin */
1072 1.9 martin for (i = 0; i < wanted->num; i++) {
1073 1.9 martin if ((wanted->infos[i].flags & PUIFLAG_EXTEND) &&
1074 1.9 martin exp_ndx == ~0U)
1075 1.9 martin exp_ndx = i;
1076 1.19 martin if (wanted->infos[i].flags &
1077 1.19 martin (PUIFLG_JUST_MOUNTPOINT|PUIFLG_IS_OUTER))
1078 1.9 martin continue;
1079 1.9 martin nsp = wanted->infos[i].size;
1080 1.9 martin if (wanted->infos[i].cur_part_id != NO_PART) {
1081 1.9 martin ps = wanted->infos[i].flags & PUIFLG_IS_OUTER ?
1082 1.9 martin parts->parent : parts;
1083 1.9 martin
1084 1.9 martin if (ps->pscheme->get_part_info(ps,
1085 1.9 martin wanted->infos[i].cur_part_id, &infos[i]))
1086 1.9 martin nsp -= infos[i].size;
1087 1.1 dholland }
1088 1.9 martin if (nsp > 0)
1089 1.9 martin planned_space += roundup(nsp, align);
1090 1.1 dholland }
1091 1.1 dholland
1092 1.1 dholland /*
1093 1.9 martin * Expand the pool partition (or shrink, if we overran),
1094 1.1 dholland */
1095 1.9 martin if (exp_ndx < wanted->num)
1096 1.9 martin wanted->infos[exp_ndx].size +=
1097 1.9 martin parts->free_space - planned_space;
1098 1.9 martin
1099 1.9 martin /*
1100 1.9 martin * Now it gets tricky: we want the wanted partitions in order
1101 1.9 martin * as defined, but any already existing partitions should not
1102 1.9 martin * be moved. We allow them to change size though.
1103 1.9 martin * To keep it simple, we just assign in order and skip blocked
1104 1.9 martin * spaces. This may shuffle the order of the resulting partitions
1105 1.9 martin * compared to the wanted list.
1106 1.9 martin */
1107 1.9 martin
1108 1.9 martin /* Adjust sizes of existing partitions */
1109 1.9 martin for (i = 0; i < wanted->num; i++) {
1110 1.9 martin ps = wanted->infos[i].flags & PUIFLG_IS_OUTER ?
1111 1.9 martin parts->parent : parts;
1112 1.9 martin const struct part_usage_info *want = &wanted->infos[i];
1113 1.9 martin
1114 1.9 martin if (want->cur_part_id == NO_PART)
1115 1.9 martin continue;
1116 1.9 martin if (i == exp_ndx) /* the exp. part. can not exist yet */
1117 1.1 dholland continue;
1118 1.9 martin daddr_t free_size = ps->pscheme->max_free_space_at(ps,
1119 1.9 martin infos[i].start);
1120 1.9 martin if (free_size < wanted->infos[i].size)
1121 1.1 dholland continue;
1122 1.9 martin infos[i].size = wanted->infos[i].size;
1123 1.9 martin ps->pscheme->set_part_info(ps, want->cur_part_id,
1124 1.9 martin &infos[i], NULL);
1125 1.9 martin }
1126 1.19 martin
1127 1.9 martin from = -1;
1128 1.19 martin /*
1129 1.19 martin * First add all outer partitions - we need to align those exactly
1130 1.19 martin * with the inner counterpart later.
1131 1.19 martin */
1132 1.19 martin if (parts->parent) {
1133 1.19 martin ps = parts->parent;
1134 1.19 martin daddr_t outer_align = ps->pscheme->get_part_alignment(ps);
1135 1.19 martin
1136 1.19 martin for (i = 0; i < wanted->num; i++) {
1137 1.19 martin struct part_usage_info *want = &wanted->infos[i];
1138 1.19 martin
1139 1.19 martin if (want->cur_part_id != NO_PART)
1140 1.19 martin continue;
1141 1.19 martin if (!(want->flags & PUIFLAG_ADD_OUTER))
1142 1.19 martin continue;
1143 1.19 martin if (want->size <= 0)
1144 1.19 martin continue;
1145 1.19 martin
1146 1.19 martin size_t cnt = ps->pscheme->get_free_spaces(ps,
1147 1.19 martin &space, 1, want->size-2*outer_align,
1148 1.19 martin outer_align, from, -1);
1149 1.19 martin
1150 1.19 martin if (cnt == 0) /* no free space for this partition */
1151 1.19 martin continue;
1152 1.19 martin
1153 1.19 martin infos[i].start = space.start;
1154 1.19 martin infos[i].size = min(want->size, space.size);
1155 1.19 martin infos[i].nat_type =
1156 1.19 martin ps->pscheme->get_fs_part_type(
1157 1.19 martin want->fs_type, want->fs_version);
1158 1.19 martin infos[i].last_mounted = want->mount;
1159 1.19 martin infos[i].fs_type = want->fs_type;
1160 1.19 martin infos[i].fs_sub_type = want->fs_version;
1161 1.19 martin new_part_id = ps->pscheme->add_partition(ps,
1162 1.19 martin &infos[i], NULL);
1163 1.19 martin if (new_part_id == NO_PART)
1164 1.19 martin continue; /* failed to add, skip */
1165 1.19 martin
1166 1.19 martin ps->pscheme->get_part_info(ps,
1167 1.19 martin new_part_id, &infos[i]);
1168 1.19 martin want->cur_part_id = new_part_id;
1169 1.19 martin
1170 1.19 martin want->flags |= PUIFLAG_ADD_INNER|PUIFLG_IS_OUTER;
1171 1.19 martin from = rounddown(infos[i].start +
1172 1.19 martin infos[i].size+outer_align, outer_align);
1173 1.19 martin }
1174 1.19 martin }
1175 1.19 martin
1176 1.19 martin /*
1177 1.19 martin * Now add new inner partitions
1178 1.19 martin */
1179 1.9 martin for (i = 0; i < wanted->num && from < wanted->parts->disk_size; i++) {
1180 1.9 martin struct part_usage_info *want = &wanted->infos[i];
1181 1.9 martin
1182 1.9 martin if (want->cur_part_id != NO_PART)
1183 1.9 martin continue;
1184 1.19 martin if (want->flags & (PUIFLG_JUST_MOUNTPOINT|PUIFLG_IS_OUTER))
1185 1.9 martin continue;
1186 1.9 martin if (want->size <= 0)
1187 1.9 martin continue;
1188 1.9 martin
1189 1.9 martin size_t cnt = wanted->parts->pscheme->get_free_spaces(
1190 1.26 martin wanted->parts, &space, 1, want->size-align, align, from,
1191 1.9 martin -1);
1192 1.26 martin if (cnt == 0)
1193 1.26 martin cnt = wanted->parts->pscheme->get_free_spaces(
1194 1.26 martin wanted->parts, &space, 1,
1195 1.26 martin want->size-5*align, align, from, -1);
1196 1.9 martin
1197 1.9 martin if (cnt == 0)
1198 1.9 martin continue; /* no free space for this partition */
1199 1.9 martin
1200 1.9 martin infos[i].start = space.start;
1201 1.9 martin infos[i].size = min(want->size, space.size);
1202 1.9 martin infos[i].nat_type =
1203 1.9 martin wanted->parts->pscheme->get_fs_part_type(want->fs_type,
1204 1.9 martin want->fs_version);
1205 1.9 martin infos[i].last_mounted = want->mount;
1206 1.9 martin infos[i].fs_type = want->fs_type;
1207 1.9 martin infos[i].fs_sub_type = want->fs_version;
1208 1.20 martin if (want->fs_type != FS_UNUSED && want->type != PT_swap) {
1209 1.20 martin want->instflags |= PUIINST_NEWFS;
1210 1.20 martin if (want->mount[0] != 0)
1211 1.20 martin want->instflags |= PUIINST_MOUNT;
1212 1.20 martin }
1213 1.9 martin new_part_id = wanted->parts->pscheme->add_partition(
1214 1.9 martin wanted->parts, &infos[i], NULL);
1215 1.9 martin if (new_part_id == NO_PART)
1216 1.9 martin continue; /* failed to add, skip */
1217 1.9 martin
1218 1.9 martin wanted->parts->pscheme->get_part_info(
1219 1.9 martin wanted->parts, new_part_id, &infos[i]);
1220 1.9 martin from = rounddown(infos[i].start+infos[i].size+align, align);
1221 1.9 martin }
1222 1.9 martin
1223 1.19 martin
1224 1.19 martin /*
1225 1.19 martin * If there are any outer partitions that we need as inner ones
1226 1.19 martin * too, add them to the inner partitioning scheme.
1227 1.19 martin */
1228 1.19 martin for (i = 0; i < wanted->num; i++) {
1229 1.19 martin struct part_usage_info *want = &wanted->infos[i];
1230 1.19 martin
1231 1.19 martin if (want->cur_part_id != NO_PART)
1232 1.19 martin continue;
1233 1.19 martin if (want->flags & PUIFLG_JUST_MOUNTPOINT)
1234 1.19 martin continue;
1235 1.19 martin if (want->size <= 0)
1236 1.19 martin continue;
1237 1.19 martin
1238 1.19 martin if ((want->flags & (PUIFLAG_ADD_INNER|PUIFLG_IS_OUTER)) !=
1239 1.19 martin (PUIFLAG_ADD_INNER|PUIFLG_IS_OUTER))
1240 1.19 martin continue;
1241 1.19 martin
1242 1.19 martin infos[i].start = want->cur_start;
1243 1.19 martin infos[i].size = want->size;
1244 1.19 martin infos[i].nat_type = wanted->parts->pscheme->get_fs_part_type(
1245 1.19 martin want->fs_type, want->fs_version);
1246 1.19 martin infos[i].last_mounted = want->mount;
1247 1.19 martin infos[i].fs_type = want->fs_type;
1248 1.19 martin infos[i].fs_sub_type = want->fs_version;
1249 1.19 martin
1250 1.19 martin if (wanted->parts->pscheme->add_outer_partition
1251 1.19 martin != NULL)
1252 1.19 martin new_part_id = wanted->parts->pscheme->
1253 1.19 martin add_outer_partition(
1254 1.19 martin wanted->parts, &infos[i], NULL);
1255 1.19 martin else
1256 1.19 martin new_part_id = wanted->parts->pscheme->
1257 1.19 martin add_partition(
1258 1.19 martin wanted->parts, &infos[i], NULL);
1259 1.19 martin
1260 1.19 martin if (new_part_id == NO_PART)
1261 1.19 martin continue; /* failed to add, skip */
1262 1.19 martin
1263 1.19 martin wanted->parts->pscheme->get_part_info(
1264 1.19 martin wanted->parts, new_part_id, &infos[i]);
1265 1.19 martin }
1266 1.19 martin
1267 1.9 martin /*
1268 1.9 martin * Note: all part_ids are invalid now, as we have added things!
1269 1.9 martin */
1270 1.9 martin for (i = 0; i < wanted->num; i++)
1271 1.9 martin wanted->infos[i].cur_part_id = NO_PART;
1272 1.19 martin for (pno = 0; pno < parts->num_part; pno++) {
1273 1.9 martin struct disk_part_info t;
1274 1.9 martin
1275 1.19 martin if (!parts->pscheme->get_part_info(parts, pno, &t))
1276 1.9 martin continue;
1277 1.9 martin
1278 1.9 martin for (i = 0; i < wanted->num; i++) {
1279 1.9 martin if (wanted->infos[i].cur_part_id != NO_PART)
1280 1.9 martin continue;
1281 1.28 martin if (wanted->infos[i].size <= 0)
1282 1.28 martin continue;
1283 1.9 martin if (t.start == infos[i].start) {
1284 1.9 martin wanted->infos[i].cur_part_id = pno;
1285 1.9 martin wanted->infos[i].cur_start = infos[i].start;
1286 1.9 martin wanted->infos[i].cur_flags = infos[i].flags;
1287 1.9 martin break;
1288 1.1 dholland }
1289 1.1 dholland }
1290 1.9 martin }
1291 1.9 martin free(infos);
1292 1.9 martin
1293 1.9 martin /* sort, and sync part ids and wanted->infos[] indices */
1294 1.9 martin sort_and_sync_parts(wanted);
1295 1.9 martin }
1296 1.9 martin
1297 1.9 martin static void
1298 1.9 martin replace_by_default(struct pm_devs *p, struct disk_partitions *parts,
1299 1.9 martin daddr_t start, daddr_t size, struct partition_usage_set *wanted)
1300 1.9 martin {
1301 1.9 martin
1302 1.9 martin if (start == 0 && size == parts->disk_size)
1303 1.9 martin parts->pscheme->delete_all_partitions(parts);
1304 1.9 martin else if (parts->pscheme->delete_partitions_in_range != NULL)
1305 1.9 martin parts->pscheme->delete_partitions_in_range(parts, start, size);
1306 1.9 martin else
1307 1.9 martin assert(parts->num_part == 0);
1308 1.9 martin
1309 1.9 martin fill_defaults(wanted, parts, start, size);
1310 1.9 martin apply_settings_to_partitions(p, parts, wanted, start, size);
1311 1.9 martin }
1312 1.9 martin
1313 1.9 martin static bool
1314 1.9 martin edit_with_defaults(struct pm_devs *p, struct disk_partitions *parts,
1315 1.9 martin daddr_t start, daddr_t size, struct partition_usage_set *wanted)
1316 1.9 martin {
1317 1.9 martin bool ok;
1318 1.9 martin
1319 1.9 martin fill_defaults(wanted, parts, start, size);
1320 1.9 martin ok = get_ptn_sizes(wanted);
1321 1.9 martin if (ok)
1322 1.9 martin apply_settings_to_partitions(p, parts, wanted, start, size);
1323 1.9 martin return ok;
1324 1.9 martin }
1325 1.9 martin
1326 1.9 martin /*
1327 1.9 martin * md back-end code for menu-driven BSD disklabel editor.
1328 1.9 martin * returns 0 on failure, 1 on success.
1329 1.9 martin * fills the install target with a list for newfs/fstab.
1330 1.9 martin */
1331 1.9 martin bool
1332 1.9 martin make_bsd_partitions(struct install_partition_desc *install)
1333 1.9 martin {
1334 1.9 martin struct disk_partitions *parts = pm->parts;
1335 1.9 martin const struct disk_partitioning_scheme *pscheme;
1336 1.9 martin struct partition_usage_set wanted;
1337 1.9 martin enum layout_type layoutkind = LY_SETSIZES;
1338 1.9 martin bool have_existing;
1339 1.9 martin
1340 1.9 martin if (pm && pm->no_part && parts == NULL)
1341 1.9 martin return true;
1342 1.9 martin
1343 1.9 martin if (parts == NULL) {
1344 1.29 martin pscheme = select_part_scheme(pm, NULL, !pm->no_mbr, NULL);
1345 1.9 martin if (pscheme == NULL)
1346 1.9 martin return false;
1347 1.9 martin parts = pscheme->create_new_for_disk(pm->diskdev,
1348 1.9 martin 0, pm->dlsize, pm->dlsize, true);
1349 1.9 martin if (parts == NULL)
1350 1.9 martin return false;
1351 1.9 martin pm->parts = parts;
1352 1.9 martin } else {
1353 1.9 martin pscheme = parts->pscheme;
1354 1.9 martin }
1355 1.9 martin
1356 1.9 martin if (pscheme->secondary_partitions) {
1357 1.9 martin struct disk_partitions *p;
1358 1.9 martin
1359 1.12 martin p = pscheme->secondary_partitions(parts, pm->ptstart, false);
1360 1.9 martin if (p) {
1361 1.9 martin parts = p;
1362 1.9 martin pscheme = parts->pscheme;
1363 1.9 martin }
1364 1.9 martin }
1365 1.9 martin
1366 1.9 martin have_existing = check_existing_netbsd(parts);
1367 1.9 martin
1368 1.9 martin /*
1369 1.9 martin * Initialize global variables that track space used on this disk.
1370 1.9 martin */
1371 1.9 martin if (pm->ptsize == 0)
1372 1.9 martin pm->ptsize = pm->dlsize - pm->ptstart;
1373 1.9 martin if (pm->dlsize == 0)
1374 1.9 martin pm->dlsize = pm->ptstart + pm->ptsize;
1375 1.1 dholland
1376 1.9 martin if (logfp) fprintf(logfp, "dlsize=%" PRId64 " ptsize=%" PRId64
1377 1.9 martin " ptstart=%" PRId64 "\n",
1378 1.9 martin pm->dlsize, pm->ptsize, pm->ptstart);
1379 1.9 martin
1380 1.9 martin if (pm->current_cylsize == 0)
1381 1.9 martin pm->current_cylsize = pm->dlcylsize;
1382 1.9 martin
1383 1.9 martin /* Ask for layout type -- standard or special */
1384 1.9 martin if (partman_go == 0) {
1385 1.9 martin char bsd_size[6], min_size[6], x_size[6];
1386 1.9 martin
1387 1.9 martin humanize_number(bsd_size, sizeof(bsd_size),
1388 1.9 martin (uint64_t)pm->ptsize*pm->sectorsize,
1389 1.9 martin "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
1390 1.9 martin humanize_number(min_size, sizeof(min_size),
1391 1.9 martin (uint64_t)(DEFROOTSIZE + DEFSWAPSIZE + DEFUSRSIZE)*MEG,
1392 1.9 martin "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
1393 1.9 martin humanize_number(x_size, sizeof(x_size),
1394 1.9 martin (uint64_t)(DEFROOTSIZE + DEFSWAPSIZE + DEFUSRSIZE
1395 1.9 martin + XNEEDMB)*MEG,
1396 1.9 martin "", HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
1397 1.9 martin
1398 1.9 martin msg_display_subst(
1399 1.9 martin have_existing ? MSG_layout_prologue_existing
1400 1.9 martin : MSG_layout_prologue_none, 6, pm->diskdev,
1401 1.9 martin msg_string(parts->pscheme->name),
1402 1.9 martin msg_string(parts->pscheme->short_name),
1403 1.9 martin bsd_size, min_size, x_size);
1404 1.9 martin msg_display_add_subst(MSG_layout_main, 6,
1405 1.9 martin pm->diskdev,
1406 1.9 martin msg_string(parts->pscheme->name),
1407 1.9 martin msg_string(parts->pscheme->short_name),
1408 1.9 martin bsd_size, min_size, x_size);
1409 1.9 martin msg_display_add("\n\n");
1410 1.9 martin layoutkind = ask_layout(parts, have_existing);
1411 1.9 martin }
1412 1.9 martin
1413 1.9 martin if (layoutkind == LY_USEDEFAULT) {
1414 1.9 martin replace_by_default(pm, parts, pm->ptstart, pm->ptsize,
1415 1.9 martin &wanted);
1416 1.9 martin } else if (layoutkind == LY_SETSIZES) {
1417 1.9 martin if (!edit_with_defaults(pm, parts, pm->ptstart, pm->ptsize,
1418 1.9 martin &wanted)) {
1419 1.9 martin free_usage_set(&wanted);
1420 1.9 martin return false;
1421 1.9 martin }
1422 1.9 martin } else {
1423 1.9 martin usage_set_from_parts(&wanted, parts);
1424 1.2 martin }
1425 1.1 dholland
1426 1.1 dholland /*
1427 1.1 dholland * OK, we have a partition table. Give the user the chance to
1428 1.1 dholland * edit it and verify it's OK, or abort altogether.
1429 1.1 dholland */
1430 1.9 martin for (;;) {
1431 1.9 martin int rv = edit_and_check_label(pm, &wanted);
1432 1.9 martin if (rv == 0) {
1433 1.9 martin msg_display(MSG_abort_part);
1434 1.9 martin free_usage_set(&wanted);
1435 1.9 martin return false;
1436 1.9 martin }
1437 1.9 martin /* update install infos */
1438 1.9 martin install->num = wanted.num;
1439 1.9 martin install->infos = wanted.infos;
1440 1.9 martin /* and check them */
1441 1.9 martin if (check_partitions(install))
1442 1.9 martin break;
1443 1.9 martin }
1444 1.1 dholland
1445 1.9 martin /* we moved infos from wanted to install target */
1446 1.9 martin wanted.infos = NULL;
1447 1.9 martin free_usage_set(&wanted);
1448 1.1 dholland
1449 1.1 dholland /* Everything looks OK. */
1450 1.9 martin return true;
1451 1.1 dholland }
1452 1.1 dholland
1453 1.20 martin #ifndef MD_NEED_BOOTBLOCK
1454 1.20 martin #define MD_NEED_BOOTBLOCK(A) true
1455 1.20 martin #endif
1456 1.20 martin
1457 1.1 dholland /*
1458 1.1 dholland * check that there is at least a / somewhere.
1459 1.1 dholland */
1460 1.9 martin bool
1461 1.9 martin check_partitions(struct install_partition_desc *install)
1462 1.1 dholland {
1463 1.1 dholland #ifdef HAVE_BOOTXX_xFS
1464 1.2 martin int rv = 1;
1465 1.1 dholland char *bootxx;
1466 1.1 dholland #endif
1467 1.1 dholland #ifndef HAVE_UFS2_BOOT
1468 1.9 martin size_t i;
1469 1.1 dholland #endif
1470 1.1 dholland
1471 1.1 dholland #ifdef HAVE_BOOTXX_xFS
1472 1.20 martin if (MD_NEED_BOOTBLOCK(install)) {
1473 1.20 martin /* check if we have boot code for the root partition type */
1474 1.20 martin bootxx = bootxx_name(install);
1475 1.20 martin if (bootxx != NULL) {
1476 1.20 martin rv = access(bootxx, R_OK);
1477 1.20 martin free(bootxx);
1478 1.20 martin } else
1479 1.20 martin rv = -1;
1480 1.20 martin if (rv != 0) {
1481 1.20 martin hit_enter_to_continue(NULL, MSG_No_Bootcode);
1482 1.20 martin return false;
1483 1.20 martin }
1484 1.1 dholland }
1485 1.1 dholland #endif
1486 1.1 dholland #ifndef HAVE_UFS2_BOOT
1487 1.20 martin if (MD_NEED_BOOTBLOCK(install)) {
1488 1.20 martin for (i = 0; i < install->num; i++) {
1489 1.20 martin if (install->infos[i].type != PT_root)
1490 1.20 martin continue;
1491 1.20 martin if (strcmp(install->infos[i].mount, "/") != 0)
1492 1.20 martin continue;
1493 1.20 martin if (install->infos[i].fs_type != FS_BSDFFS)
1494 1.20 martin continue;
1495 1.20 martin if (install->infos[i].fs_version != 2)
1496 1.20 martin continue;
1497 1.20 martin hit_enter_to_continue(NULL, MSG_cannot_ufs2_root);
1498 1.20 martin return false;
1499 1.20 martin }
1500 1.1 dholland }
1501 1.1 dholland #endif
1502 1.1 dholland
1503 1.9 martin return md_check_partitions(install);
1504 1.1 dholland }
1505