Home | History | Annotate | Line # | Download | only in sysinst
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