Home | History | Annotate | Line # | Download | only in sysinst
      1 /*	$NetBSD: install.c,v 1.25 2023/01/06 18:13:40 martin Exp $	*/
      2 
      3 /*
      4  * Copyright 1997 Piermont Information Systems Inc.
      5  * All rights reserved.
      6  *
      7  * Written by Philip A. Nelson for Piermont Information Systems Inc.
      8  *
      9  * Redistribution and use in source and binary forms, with or without
     10  * modification, are permitted provided that the following conditions
     11  * are met:
     12  * 1. Redistributions of source code must retain the above copyright
     13  *    notice, this list of conditions and the following disclaimer.
     14  * 2. Redistributions in binary form must reproduce the above copyright
     15  *    notice, this list of conditions and the following disclaimer in the
     16  *    documentation and/or other materials provided with the distribution.
     17  * 3. The name of Piermont Information Systems Inc. may not be used to endorse
     18  *    or promote products derived from this software without specific prior
     19  *    written permission.
     20  *
     21  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
     22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     24  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
     25  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     31  * THE POSSIBILITY OF SUCH DAMAGE.
     32  *
     33  */
     34 
     35 /* install.c -- system installation. */
     36 
     37 #include <sys/param.h>
     38 #include <stdio.h>
     39 #include <curses.h>
     40 #include "defs.h"
     41 #include "msg_defs.h"
     42 #include "menu_defs.h"
     43 
     44 /*
     45  * helper function: call the md pre/post disklabel callback for all involved
     46  * inner partitions and write them back.
     47  * return net result
     48  */
     49 static bool
     50 write_all_parts(struct install_partition_desc *install)
     51 {
     52 	struct disk_partitions **allparts, *parts;
     53 #ifndef NO_CLONES
     54 	struct selected_partition *src;
     55 #endif
     56 	size_t num_parts, i, j;
     57 	bool found, res;
     58 
     59 	/* pessimistic assumption: all partitions on different devices */
     60 	allparts = calloc(install->num + install->num_write_back,
     61 	    sizeof(*allparts));
     62 	if (allparts == NULL)
     63 		return false;
     64 
     65 	/* collect all different partition sets */
     66 	num_parts = 0;
     67 	for (i = 0; i < install->num_write_back; i++) {
     68 		parts = install->write_back[i];
     69 		if (parts == NULL)
     70 			continue;
     71 		found = false;
     72 		for (j = 0; j < num_parts; j++) {
     73 			if (allparts[j] == parts) {
     74 				found = true;
     75 				break;
     76 			}
     77 		}
     78 		if (found)
     79 			continue;
     80 		allparts[num_parts++] = parts;
     81 	}
     82 	for (i = 0; i < install->num; i++) {
     83 		parts = install->infos[i].parts;
     84 		if (parts == NULL)
     85 			continue;
     86 		found = false;
     87 		for (j = 0; j < num_parts; j++) {
     88 			if (allparts[j] == parts) {
     89 				found = true;
     90 				break;
     91 			}
     92 		}
     93 		if (found)
     94 			continue;
     95 		allparts[num_parts++] = parts;
     96 	}
     97 
     98 	/* do multiple phases, abort anytime and go out, returning res */
     99 
    100 	res = true;
    101 
    102 	/* phase 1: pre disklabel - used to write MBR and similar */
    103 	for (i = 0; i < num_parts; i++) {
    104 		if (!md_pre_disklabel(install, allparts[i])) {
    105 			res = false;
    106 			goto out;
    107 		}
    108 	}
    109 
    110 	/* phase 2: write our partitions (used to be: disklabel) */
    111 	for (i = 0; i < num_parts; i++) {
    112 		if (!allparts[i]->pscheme->write_to_disk(allparts[i])) {
    113 			res = false;
    114 			goto out;
    115 		}
    116 	}
    117 
    118 	/* phase 3: now we may have a first chance to enable swap space */
    119 	set_swap_if_low_ram(install);
    120 
    121 #ifndef NO_CLONES
    122 	/* phase 4: copy any cloned partitions data (if requested) */
    123 	for (i = 0; i < install->num; i++) {
    124 		if ((install->infos[i].flags & PUIFLG_CLONE_PARTS) == 0
    125 		    || install->infos[i].clone_src == NULL
    126 		    || !install->infos[i].clone_src->with_data)
    127 			continue;
    128 		src = &install->infos[i].clone_src
    129 		    ->selection[install->infos[i].clone_ndx];
    130 		clone_partition_data(install->infos[i].parts,
    131 		    install->infos[i].cur_part_id,
    132 		    src->parts, src->id);
    133 	}
    134 #endif
    135 
    136 	/* phase 5: post disklabel (used for updating boot loaders) */
    137 	for (i = 0; i < num_parts; i++) {
    138 		if (!md_post_disklabel(install, allparts[i])) {
    139 			res = false;
    140 			goto out;
    141 		}
    142 	}
    143 
    144 out:
    145 	free(allparts);
    146 
    147 	return res;
    148 }
    149 
    150 /* Do the system install. */
    151 
    152 void
    153 do_install(void)
    154 {
    155 	int find_disks_ret;
    156 	int retcode = 0, res;
    157 	struct install_partition_desc install = {};
    158 
    159 #ifndef NO_PARTMAN
    160 	partman_go = -1;
    161 #else
    162 	partman_go = 0;
    163 #endif
    164 
    165 #ifndef DEBUG
    166 	msg_display(MSG_installusure);
    167 	if (!ask_noyes(NULL))
    168 		return;
    169 #endif
    170 
    171 	memset(&install, 0, sizeof install);
    172 
    173 	/* Create and mount partitions */
    174 	find_disks_ret = find_disks(msg_string(MSG_install), false);
    175 	if (partman_go == 1) {
    176 		if (partman(&install) < 0) {
    177 			hit_enter_to_continue(MSG_abort_part, NULL);
    178 			return;
    179 		}
    180 	} else if (find_disks_ret < 0)
    181 		return;
    182 	else {
    183 	/* Classical partitioning wizard */
    184 		partman_go = 0;
    185 		clear();
    186 		refresh();
    187 
    188 		if (check_swap(pm->diskdev, 0) > 0) {
    189 			hit_enter_to_continue(MSG_swapactive, NULL);
    190 			if (check_swap(pm->diskdev, 1) < 0) {
    191 				hit_enter_to_continue(MSG_swapdelfailed, NULL);
    192 				if (!debug)
    193 					return;
    194 			}
    195 		}
    196 
    197 		for (;;) {
    198 			if (md_get_info(&install)) {
    199 				res = md_make_bsd_partitions(&install);
    200 				if (res == -1) {
    201 					pm->parts = NULL;
    202 					continue;
    203 				} else if (res == 1) {
    204 					break;
    205 				}
    206 			}
    207 			hit_enter_to_continue(MSG_abort_inst, NULL);
    208 			goto error;
    209 		}
    210 
    211 		/* Last chance ... do you really want to do this? */
    212 		clear();
    213 		refresh();
    214 		msg_fmt_display(MSG_lastchance, "%s", pm->diskdev);
    215 		if (!ask_noyes(NULL))
    216 			goto error;
    217 
    218 		if ((!pm->no_part && !write_all_parts(&install)) ||
    219 		    make_filesystems(&install) ||
    220 		    make_fstab(&install) != 0 ||
    221 		    md_post_newfs(&install) != 0)
    222 			goto error;
    223 	}
    224 
    225 	/* Unpack the distribution. */
    226 	process_menu(MENU_distset, &retcode);
    227 	if (retcode == 0)
    228 		goto error;
    229 	if (get_and_unpack_sets(0, MSG_disksetupdone,
    230 	    MSG_extractcomplete, MSG_abortinst) != 0)
    231 		goto error;
    232 
    233 	if (md_post_extract(&install, false) != 0)
    234 		goto error;
    235 
    236 	root_pw_setup();
    237 #if CHECK_ENTROPY
    238 	do_add_entropy();
    239 #endif
    240 	do_configmenu(&install);
    241 
    242 	sanity_check();
    243 
    244 	md_cleanup_install(&install);
    245 
    246 	hit_enter_to_continue(MSG_instcomplete, NULL);
    247 
    248 error:
    249 	free_install_desc(&install);
    250 }
    251