Home | History | Annotate | Line # | Download | only in prep
md.c revision 1.2.2.2
      1  1.2.2.2  tls /*	$NetBSD: md.c,v 1.2.2.2 2014/08/10 07:00:28 tls Exp $	*/
      2  1.2.2.2  tls 
      3  1.2.2.2  tls /*
      4  1.2.2.2  tls  * Copyright 1997 Piermont Information Systems Inc.
      5  1.2.2.2  tls  * All rights reserved.
      6  1.2.2.2  tls  *
      7  1.2.2.2  tls  * Based on code written by Philip A. Nelson for Piermont Information
      8  1.2.2.2  tls  * Systems Inc.
      9  1.2.2.2  tls  *
     10  1.2.2.2  tls  * Redistribution and use in source and binary forms, with or without
     11  1.2.2.2  tls  * modification, are permitted provided that the following conditions
     12  1.2.2.2  tls  * are met:
     13  1.2.2.2  tls  * 1. Redistributions of source code must retain the above copyright
     14  1.2.2.2  tls  *    notice, this list of conditions and the following disclaimer.
     15  1.2.2.2  tls  * 2. Redistributions in binary form must reproduce the above copyright
     16  1.2.2.2  tls  *    notice, this list of conditions and the following disclaimer in the
     17  1.2.2.2  tls  *    documentation and/or other materials provided with the distribution.
     18  1.2.2.2  tls  * 3. The name of Piermont Information Systems Inc. may not be used to endorse
     19  1.2.2.2  tls  *    or promote products derived from this software without specific prior
     20  1.2.2.2  tls  *    written permission.
     21  1.2.2.2  tls  *
     22  1.2.2.2  tls  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
     23  1.2.2.2  tls  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     24  1.2.2.2  tls  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     25  1.2.2.2  tls  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
     26  1.2.2.2  tls  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     27  1.2.2.2  tls  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     28  1.2.2.2  tls  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     29  1.2.2.2  tls  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     30  1.2.2.2  tls  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     31  1.2.2.2  tls  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
     32  1.2.2.2  tls  * THE POSSIBILITY OF SUCH DAMAGE.
     33  1.2.2.2  tls  */
     34  1.2.2.2  tls 
     35  1.2.2.2  tls /* md.c -- prep machine specific routines */
     36  1.2.2.2  tls 
     37  1.2.2.2  tls #include <sys/param.h>
     38  1.2.2.2  tls #include <sys/sysctl.h>
     39  1.2.2.2  tls #include <stdio.h>
     40  1.2.2.2  tls #include <util.h>
     41  1.2.2.2  tls #include <machine/cpu.h>
     42  1.2.2.2  tls 
     43  1.2.2.2  tls #include "defs.h"
     44  1.2.2.2  tls #include "md.h"
     45  1.2.2.2  tls #include "msg_defs.h"
     46  1.2.2.2  tls #include "menu_defs.h"
     47  1.2.2.2  tls #include "endian.h"
     48  1.2.2.2  tls 
     49  1.2.2.2  tls int prep_nobootfix = 0, prep_rawdevfix = 0, prep_bootpart = PART_BOOT;
     50  1.2.2.2  tls 
     51  1.2.2.2  tls void
     52  1.2.2.2  tls md_init(void)
     53  1.2.2.2  tls {
     54  1.2.2.2  tls }
     55  1.2.2.2  tls 
     56  1.2.2.2  tls void
     57  1.2.2.2  tls md_init_set_status(int flags)
     58  1.2.2.2  tls {
     59  1.2.2.2  tls 	(void)flags;
     60  1.2.2.2  tls }
     61  1.2.2.2  tls 
     62  1.2.2.2  tls int
     63  1.2.2.2  tls md_get_info(void)
     64  1.2.2.2  tls {
     65  1.2.2.2  tls 	return set_bios_geom_with_mbr_guess();
     66  1.2.2.2  tls }
     67  1.2.2.2  tls 
     68  1.2.2.2  tls /*
     69  1.2.2.2  tls  * md back-end code for menu-driven BSD disklabel editor.
     70  1.2.2.2  tls  */
     71  1.2.2.2  tls int
     72  1.2.2.2  tls md_make_bsd_partitions(void)
     73  1.2.2.2  tls {
     74  1.2.2.2  tls 	return make_bsd_partitions();
     75  1.2.2.2  tls }
     76  1.2.2.2  tls 
     77  1.2.2.2  tls /*
     78  1.2.2.2  tls  * any additional partition validation
     79  1.2.2.2  tls  */
     80  1.2.2.2  tls int
     81  1.2.2.2  tls md_check_partitions(void)
     82  1.2.2.2  tls {
     83  1.2.2.2  tls 	int part;
     84  1.2.2.2  tls 
     85  1.2.2.2  tls 	/* we need to find a boot partition, otherwise we can't write our
     86  1.2.2.2  tls 	 * "bootblock".  We make the assumption that the user hasn't done
     87  1.2.2.2  tls 	 * something stupid, like move it away from the MBR partition.
     88  1.2.2.2  tls 	 */
     89  1.2.2.2  tls 	for (part = PART_A; part < MAXPARTITIONS; part++)
     90  1.2.2.2  tls 		if (pm->bsdlabel[part].pi_fstype == FS_BOOT) {
     91  1.2.2.2  tls 			prep_bootpart = part;
     92  1.2.2.2  tls 			return 1;
     93  1.2.2.2  tls 		}
     94  1.2.2.2  tls 
     95  1.2.2.2  tls 	msg_display(MSG_prepnobootpart);
     96  1.2.2.2  tls 	process_menu(MENU_ok, NULL);
     97  1.2.2.2  tls 	return 0;
     98  1.2.2.2  tls }
     99  1.2.2.2  tls 
    100  1.2.2.2  tls /*
    101  1.2.2.2  tls  * hook called before writing new disklabel.
    102  1.2.2.2  tls  */
    103  1.2.2.2  tls int
    104  1.2.2.2  tls md_pre_disklabel(void)
    105  1.2.2.2  tls {
    106  1.2.2.2  tls 	msg_display(MSG_dofdisk);
    107  1.2.2.2  tls 
    108  1.2.2.2  tls 	/* write edited MBR onto disk. */
    109  1.2.2.2  tls 	if (write_mbr(pm->diskdev, &mbr, 1) != 0) {
    110  1.2.2.2  tls 		msg_display(MSG_wmbrfail);
    111  1.2.2.2  tls 		process_menu(MENU_ok, NULL);
    112  1.2.2.2  tls 		return 1;
    113  1.2.2.2  tls 	}
    114  1.2.2.2  tls 	return 0;
    115  1.2.2.2  tls }
    116  1.2.2.2  tls 
    117  1.2.2.2  tls /*
    118  1.2.2.2  tls  * hook called after writing disklabel to new target disk.
    119  1.2.2.2  tls  */
    120  1.2.2.2  tls int
    121  1.2.2.2  tls md_post_disklabel(void)
    122  1.2.2.2  tls {
    123  1.2.2.2  tls 	return 0;
    124  1.2.2.2  tls }
    125  1.2.2.2  tls 
    126  1.2.2.2  tls /*
    127  1.2.2.2  tls  * hook called after upgrade() or install() has finished setting
    128  1.2.2.2  tls  * up the target disk but immediately before the user is given the
    129  1.2.2.2  tls  * ``disks are now set up'' message.
    130  1.2.2.2  tls  */
    131  1.2.2.2  tls int
    132  1.2.2.2  tls md_post_newfs(void)
    133  1.2.2.2  tls {
    134  1.2.2.2  tls 	return 0;
    135  1.2.2.2  tls }
    136  1.2.2.2  tls 
    137  1.2.2.2  tls int
    138  1.2.2.2  tls md_post_extract(void)
    139  1.2.2.2  tls {
    140  1.2.2.2  tls 	char rawdev[100], bootpart[100], bootloader[100];
    141  1.2.2.2  tls 
    142  1.2.2.2  tls 	/* if we can't make it bootable, just punt */
    143  1.2.2.2  tls 	if (prep_nobootfix)
    144  1.2.2.2  tls 		return 0;
    145  1.2.2.2  tls 
    146  1.2.2.2  tls 	process_menu(MENU_prepconsole, NULL);
    147  1.2.2.2  tls 	if (yesno == 1)
    148  1.2.2.2  tls 		snprintf(bootloader, 100, "/usr/mdec/boot_com0");
    149  1.2.2.2  tls 	else
    150  1.2.2.2  tls 		snprintf(bootloader, 100, "/usr/mdec/boot");
    151  1.2.2.2  tls 
    152  1.2.2.2  tls 	snprintf(rawdev, 100, "/dev/r%s%c", pm->diskdev, 'a' + getrawpartition());
    153  1.2.2.2  tls 	snprintf(bootpart, 100, "/dev/r%s%c", pm->diskdev, 'a' + prep_bootpart);
    154  1.2.2.2  tls 	if (prep_rawdevfix)
    155  1.2.2.2  tls 		run_program(RUN_DISPLAY|RUN_CHROOT,
    156  1.2.2.2  tls 		    "/usr/mdec/mkbootimage -b %s -k /netbsd "
    157  1.2.2.2  tls 		    "-r %s /.bootimage", bootloader, rawdev);
    158  1.2.2.2  tls 	else
    159  1.2.2.2  tls 		run_program(RUN_DISPLAY|RUN_CHROOT,
    160  1.2.2.2  tls 		    "/usr/mdec/mkbootimage -s -b %s -k /netbsd /.bootimage",
    161  1.2.2.2  tls 		    bootloader);
    162  1.2.2.2  tls 	run_program(RUN_DISPLAY|RUN_CHROOT, "/bin/dd if=/.bootimage of=%s "
    163  1.2.2.2  tls 	    "bs=512 conv=sync", bootpart);
    164  1.2.2.2  tls 
    165  1.2.2.2  tls 	return 0;
    166  1.2.2.2  tls }
    167  1.2.2.2  tls 
    168  1.2.2.2  tls void
    169  1.2.2.2  tls md_cleanup_install(void)
    170  1.2.2.2  tls {
    171  1.2.2.2  tls #ifndef DEBUG
    172  1.2.2.2  tls 	enable_rc_conf();
    173  1.2.2.2  tls #endif
    174  1.2.2.2  tls 	run_program(0, "rm -f %s", target_expand("/.bootimage"));
    175  1.2.2.2  tls }
    176  1.2.2.2  tls 
    177  1.2.2.2  tls int
    178  1.2.2.2  tls md_pre_update(void)
    179  1.2.2.2  tls {
    180  1.2.2.2  tls 	struct mbr_partition *part;
    181  1.2.2.2  tls 	mbr_info_t *ext;
    182  1.2.2.2  tls 	int i;
    183  1.2.2.2  tls 
    184  1.2.2.2  tls 	read_mbr(pm->diskdev, &mbr);
    185  1.2.2.2  tls 	/* do a sanity check of the partition table */
    186  1.2.2.2  tls 	for (ext = &mbr; ext; ext = ext->extended) {
    187  1.2.2.2  tls 		part = ext->mbr.mbr_parts;
    188  1.2.2.2  tls 		for (i = 0; i < MBR_PART_COUNT; part++, i++) {
    189  1.2.2.2  tls 			if (part->mbrp_type != MBR_PTYPE_PREP)
    190  1.2.2.2  tls 				continue;
    191  1.2.2.2  tls 			if (part->mbrp_size < (MIN_PREP_BOOT/512)) {
    192  1.2.2.2  tls 				msg_display(MSG_preptoosmall);
    193  1.2.2.2  tls 				msg_display_add(MSG_prepnobootpart, 0);
    194  1.2.2.2  tls 				process_menu(MENU_yesno, NULL);
    195  1.2.2.2  tls 				if (!yesno)
    196  1.2.2.2  tls 					return 0;
    197  1.2.2.2  tls 				prep_nobootfix = 1;
    198  1.2.2.2  tls 			}
    199  1.2.2.2  tls 			if (part->mbrp_start == 0)
    200  1.2.2.2  tls 				prep_rawdevfix = 1;
    201  1.2.2.2  tls 		}
    202  1.2.2.2  tls 	}
    203  1.2.2.2  tls 	if (md_check_partitions() == 0)
    204  1.2.2.2  tls 		prep_nobootfix = 1;
    205  1.2.2.2  tls 	return 1;
    206  1.2.2.2  tls }
    207  1.2.2.2  tls 
    208  1.2.2.2  tls /* Upgrade support */
    209  1.2.2.2  tls int
    210  1.2.2.2  tls md_update(void)
    211  1.2.2.2  tls {
    212  1.2.2.2  tls 	md_post_newfs();
    213  1.2.2.2  tls 	return 1;
    214  1.2.2.2  tls }
    215  1.2.2.2  tls 
    216  1.2.2.2  tls int
    217  1.2.2.2  tls md_check_mbr(mbr_info_t *mbri)
    218  1.2.2.2  tls {
    219  1.2.2.2  tls 	mbr_info_t *ext;
    220  1.2.2.2  tls 	struct mbr_partition *part;
    221  1.2.2.2  tls 	int i;
    222  1.2.2.2  tls 
    223  1.2.2.2  tls 	for (ext = mbri; ext; ext = ext->extended) {
    224  1.2.2.2  tls 		part = ext->mbr.mbr_parts;
    225  1.2.2.2  tls 		for (i = 0; i < MBR_PART_COUNT; part++, i++) {
    226  1.2.2.2  tls 			if (part->mbrp_type != MBR_PTYPE_PREP)
    227  1.2.2.2  tls 				continue;
    228  1.2.2.2  tls 			pm->bootstart = part->mbrp_start;
    229  1.2.2.2  tls 			pm->bootsize = part->mbrp_size;
    230  1.2.2.2  tls 			break;
    231  1.2.2.2  tls 		}
    232  1.2.2.2  tls 	}
    233  1.2.2.2  tls 	if (pm->bootsize < (MIN_PREP_BOOT/512)) {
    234  1.2.2.2  tls 		msg_display(MSG_preptoosmall);
    235  1.2.2.2  tls 		msg_display_add(MSG_reeditpart, 0);
    236  1.2.2.2  tls 		process_menu(MENU_yesno, NULL);
    237  1.2.2.2  tls 		if (!yesno)
    238  1.2.2.2  tls 			return 0;
    239  1.2.2.2  tls 		return 1;
    240  1.2.2.2  tls 	}
    241  1.2.2.2  tls 	if (pm->bootstart == 0 || pm->bootsize == 0) {
    242  1.2.2.2  tls 		msg_display(MSG_nopreppart);
    243  1.2.2.2  tls 		msg_display_add(MSG_reeditpart, 0);
    244  1.2.2.2  tls 		process_menu(MENU_yesno, NULL);
    245  1.2.2.2  tls 		if (!yesno)
    246  1.2.2.2  tls 			return 0;
    247  1.2.2.2  tls 		return 1;
    248  1.2.2.2  tls 	}
    249  1.2.2.2  tls 	return 2;
    250  1.2.2.2  tls }
    251  1.2.2.2  tls 
    252  1.2.2.2  tls int
    253  1.2.2.2  tls md_mbr_use_wholedisk(mbr_info_t *mbri)
    254  1.2.2.2  tls {
    255  1.2.2.2  tls 	struct mbr_sector *mbrs = &mbri->mbr;
    256  1.2.2.2  tls 	mbr_info_t *ext;
    257  1.2.2.2  tls 	struct mbr_partition *part;
    258  1.2.2.2  tls 
    259  1.2.2.2  tls 	part = &mbrs->mbr_parts[0];
    260  1.2.2.2  tls 	/* Set the partition information for full disk usage. */
    261  1.2.2.2  tls 	while ((ext = mbri->extended)) {
    262  1.2.2.2  tls 		mbri->extended = ext->extended;
    263  1.2.2.2  tls 		free(ext);
    264  1.2.2.2  tls 	}
    265  1.2.2.2  tls 	memset(part, 0, MBR_PART_COUNT * sizeof *part);
    266  1.2.2.2  tls #ifdef BOOTSEL
    267  1.2.2.2  tls 	memset(&mbri->mbrb, 0, sizeof mbri->mbrb);
    268  1.2.2.2  tls #endif
    269  1.2.2.2  tls 	part[0].mbrp_type = MBR_PTYPE_PREP;
    270  1.2.2.2  tls 	part[0].mbrp_size = PREP_BOOT_SIZE/512;
    271  1.2.2.2  tls 	part[0].mbrp_start = bsec;
    272  1.2.2.2  tls 	part[0].mbrp_flag = MBR_PFLAG_ACTIVE;
    273  1.2.2.2  tls 
    274  1.2.2.2  tls 	part[1].mbrp_type = MBR_PTYPE_NETBSD;
    275  1.2.2.2  tls 	part[1].mbrp_size = pm->dlsize - (bsec + PREP_BOOT_SIZE/512);
    276  1.2.2.2  tls 	part[1].mbrp_start = bsec + PREP_BOOT_SIZE/512;
    277  1.2.2.2  tls 	part[1].mbrp_flag = 0;
    278  1.2.2.2  tls 
    279  1.2.2.2  tls 	pm->ptstart = part[1].mbrp_start;
    280  1.2.2.2  tls 	pm->ptsize = part[1].mbrp_size;
    281  1.2.2.2  tls 	pm->bootstart = part[0].mbrp_start;
    282  1.2.2.2  tls 	pm->bootsize = part[0].mbrp_size;
    283  1.2.2.2  tls 	return 1;
    284  1.2.2.2  tls }
    285  1.2.2.2  tls 
    286  1.2.2.2  tls 
    287  1.2.2.2  tls int
    288  1.2.2.2  tls md_pre_mount()
    289  1.2.2.2  tls {
    290  1.2.2.2  tls 	return 0;
    291  1.2.2.2  tls }
    292