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:25 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 -- shark machine specific routines */ 36 1.2.2.2 tls 37 1.2.2.2 tls #include <stdio.h> 38 1.2.2.2 tls #include <curses.h> 39 1.2.2.2 tls #include <unistd.h> 40 1.2.2.2 tls #include <fcntl.h> 41 1.2.2.2 tls #include <util.h> 42 1.2.2.2 tls #include <sys/types.h> 43 1.2.2.2 tls #include <sys/disklabel.h> 44 1.2.2.2 tls #include <sys/ioctl.h> 45 1.2.2.2 tls #include <sys/param.h> 46 1.2.2.2 tls 47 1.2.2.2 tls #include "defs.h" 48 1.2.2.2 tls #include "md.h" 49 1.2.2.2 tls #include "msg_defs.h" 50 1.2.2.2 tls #include "menu_defs.h" 51 1.2.2.2 tls 52 1.2.2.2 tls int boardtype = 0, rpi_bootpart = PART_A; 53 1.2.2.2 tls 54 1.2.2.2 tls void 55 1.2.2.2 tls md_prelim_menu(void) 56 1.2.2.2 tls { 57 1.2.2.2 tls /* get the boardtype from the user */ 58 1.2.2.2 tls process_menu(MENU_prelim, NULL); 59 1.2.2.2 tls } 60 1.2.2.2 tls 61 1.2.2.2 tls void 62 1.2.2.2 tls md_init(void) 63 1.2.2.2 tls { 64 1.2.2.2 tls } 65 1.2.2.2 tls 66 1.2.2.2 tls void 67 1.2.2.2 tls md_init_set_status(int flags) 68 1.2.2.2 tls { 69 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) 70 1.2.2.2 tls set_kernel_set(SET_KERNEL_RPI); 71 1.2.2.2 tls } 72 1.2.2.2 tls 73 1.2.2.2 tls int 74 1.2.2.2 tls md_get_info(void) 75 1.2.2.2 tls { 76 1.2.2.2 tls struct disklabel disklabel; 77 1.2.2.2 tls int fd; 78 1.2.2.2 tls char dev_name[100]; 79 1.2.2.2 tls 80 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) 81 1.2.2.2 tls return set_bios_geom_with_mbr_guess(); 82 1.2.2.2 tls 83 1.2.2.2 tls if (pm->no_mbr) 84 1.2.2.2 tls return 1; 85 1.2.2.2 tls 86 1.2.2.2 tls if (read_mbr(pm->diskdev, &mbr) < 0) 87 1.2.2.2 tls memset(&mbr.mbr, 0, sizeof(mbr.mbr)-2); 88 1.2.2.2 tls 89 1.2.2.2 tls if (edit_mbr(&mbr) == 0) 90 1.2.2.2 tls return 0; 91 1.2.2.2 tls 92 1.2.2.2 tls if (strncmp(pm->diskdev, "wd", 2) == 0) 93 1.2.2.2 tls pm->disktype = "ST506"; 94 1.2.2.2 tls else 95 1.2.2.2 tls pm->disktype = "SCSI"; 96 1.2.2.2 tls 97 1.2.2.2 tls snprintf(dev_name, 100, "/dev/r%sc", pm->diskdev); 98 1.2.2.2 tls 99 1.2.2.2 tls fd = open(dev_name, O_RDONLY, 0); 100 1.2.2.2 tls if (fd < 0) { 101 1.2.2.2 tls endwin(); 102 1.2.2.2 tls fprintf(stderr, "Can't open %s\n", dev_name); 103 1.2.2.2 tls exit(1); 104 1.2.2.2 tls } 105 1.2.2.2 tls if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) { 106 1.2.2.2 tls endwin(); 107 1.2.2.2 tls fprintf(stderr, "Can't read disklabel on %s.\n", dev_name); 108 1.2.2.2 tls close(fd); 109 1.2.2.2 tls exit(1); 110 1.2.2.2 tls } 111 1.2.2.2 tls close(fd); 112 1.2.2.2 tls 113 1.2.2.2 tls pm->dlcyl = disklabel.d_ncylinders; 114 1.2.2.2 tls pm->dlhead = disklabel.d_ntracks; 115 1.2.2.2 tls pm->dlsec = disklabel.d_nsectors; 116 1.2.2.2 tls pm->sectorsize = disklabel.d_secsize; 117 1.2.2.2 tls pm->dlcylsize = disklabel.d_secpercyl; 118 1.2.2.2 tls 119 1.2.2.2 tls /* 120 1.2.2.2 tls * Compute whole disk size. Take max of (pm->dlcyl*pm->dlhead*pm->dlsec) 121 1.2.2.2 tls * and secperunit, just in case the disk is already labelled. 122 1.2.2.2 tls * (If our new label's RAW_PART size ends up smaller than the 123 1.2.2.2 tls * in-core RAW_PART size value, updating the label will fail.) 124 1.2.2.2 tls */ 125 1.2.2.2 tls pm->dlsize = pm->dlcyl*pm->dlhead*pm->dlsec; 126 1.2.2.2 tls if (disklabel.d_secperunit > pm->dlsize) 127 1.2.2.2 tls pm->dlsize = disklabel.d_secperunit; 128 1.2.2.2 tls 129 1.2.2.2 tls return 1; 130 1.2.2.2 tls } 131 1.2.2.2 tls 132 1.2.2.2 tls /* 133 1.2.2.2 tls * md back-end code for menu-driven BSD disklabel editor. 134 1.2.2.2 tls */ 135 1.2.2.2 tls int 136 1.2.2.2 tls md_make_bsd_partitions(void) 137 1.2.2.2 tls { 138 1.2.2.2 tls return make_bsd_partitions(); 139 1.2.2.2 tls } 140 1.2.2.2 tls 141 1.2.2.2 tls /* 142 1.2.2.2 tls * any additional partition validation 143 1.2.2.2 tls */ 144 1.2.2.2 tls int 145 1.2.2.2 tls md_check_partitions(void) 146 1.2.2.2 tls { 147 1.2.2.2 tls int part; 148 1.2.2.2 tls 149 1.2.2.2 tls if (boardtype == BOARD_TYPE_NORMAL) 150 1.2.2.2 tls return 1; 151 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) { 152 1.2.2.2 tls for (part = PART_A; part < MAXPARTITIONS; part++) 153 1.2.2.2 tls if (pm->bsdlabel[part].pi_fstype == FS_MSDOS) { 154 1.2.2.2 tls rpi_bootpart = part; 155 1.2.2.2 tls return 1; 156 1.2.2.2 tls } 157 1.2.2.2 tls 158 1.2.2.2 tls msg_display(MSG_nomsdospart); 159 1.2.2.2 tls process_menu(MENU_ok, NULL); 160 1.2.2.2 tls } 161 1.2.2.2 tls return 0; 162 1.2.2.2 tls } 163 1.2.2.2 tls 164 1.2.2.2 tls /* 165 1.2.2.2 tls * hook called before writing new disklabel. 166 1.2.2.2 tls */ 167 1.2.2.2 tls int 168 1.2.2.2 tls md_pre_disklabel(void) 169 1.2.2.2 tls { 170 1.2.2.2 tls if (pm->no_mbr) 171 1.2.2.2 tls return 0; 172 1.2.2.2 tls 173 1.2.2.2 tls msg_display(MSG_dofdisk); 174 1.2.2.2 tls 175 1.2.2.2 tls /* write edited MBR onto disk. */ 176 1.2.2.2 tls if (write_mbr(pm->diskdev, &mbr, 1) != 0) { 177 1.2.2.2 tls msg_display(MSG_wmbrfail); 178 1.2.2.2 tls process_menu(MENU_ok, NULL); 179 1.2.2.2 tls return 1; 180 1.2.2.2 tls } 181 1.2.2.2 tls return 0; 182 1.2.2.2 tls } 183 1.2.2.2 tls 184 1.2.2.2 tls /* 185 1.2.2.2 tls * hook called after writing disklabel to new target disk. 186 1.2.2.2 tls */ 187 1.2.2.2 tls int 188 1.2.2.2 tls md_post_disklabel(void) 189 1.2.2.2 tls { 190 1.2.2.2 tls return 0; 191 1.2.2.2 tls } 192 1.2.2.2 tls 193 1.2.2.2 tls /* 194 1.2.2.2 tls * hook called after upgrade() or install() has finished setting 195 1.2.2.2 tls * up the target disk but immediately before the user is given the 196 1.2.2.2 tls * ``disks are now set up'' message. 197 1.2.2.2 tls */ 198 1.2.2.2 tls int 199 1.2.2.2 tls md_post_newfs(void) 200 1.2.2.2 tls { 201 1.2.2.2 tls return 0; 202 1.2.2.2 tls } 203 1.2.2.2 tls 204 1.2.2.2 tls int 205 1.2.2.2 tls md_post_extract(void) 206 1.2.2.2 tls { 207 1.2.2.2 tls char kernelbin[100]; 208 1.2.2.2 tls 209 1.2.2.2 tls if (boardtype == BOARD_TYPE_NORMAL) 210 1.2.2.2 tls return 0; 211 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) { 212 1.2.2.2 tls snprintf(kernelbin, 100, "%s/netbsd.bin", targetroot_mnt); 213 1.2.2.2 tls if (file_exists_p(kernelbin)) { 214 1.2.2.2 tls run_program(RUN_DISPLAY, 215 1.2.2.2 tls "/bin/cp %s /targetroot/boot/kernel.img", kernelbin); 216 1.2.2.2 tls } else { 217 1.2.2.2 tls msg_display(MSG_rpikernelmissing); 218 1.2.2.2 tls process_menu(MENU_ok, NULL); 219 1.2.2.2 tls return 1; 220 1.2.2.2 tls } 221 1.2.2.2 tls } 222 1.2.2.2 tls return 0; 223 1.2.2.2 tls } 224 1.2.2.2 tls 225 1.2.2.2 tls void 226 1.2.2.2 tls md_cleanup_install(void) 227 1.2.2.2 tls { 228 1.2.2.2 tls #ifndef DEBUG 229 1.2.2.2 tls enable_rc_conf(); 230 1.2.2.2 tls add_rc_conf("sshd=YES\n"); 231 1.2.2.2 tls add_rc_conf("dhcpcd=YES\n"); 232 1.2.2.2 tls #endif 233 1.2.2.2 tls } 234 1.2.2.2 tls 235 1.2.2.2 tls int 236 1.2.2.2 tls md_pre_update(void) 237 1.2.2.2 tls { 238 1.2.2.2 tls return 1; 239 1.2.2.2 tls } 240 1.2.2.2 tls 241 1.2.2.2 tls /* Upgrade support */ 242 1.2.2.2 tls int 243 1.2.2.2 tls md_update(void) 244 1.2.2.2 tls { 245 1.2.2.2 tls md_post_newfs(); 246 1.2.2.2 tls return 1; 247 1.2.2.2 tls } 248 1.2.2.2 tls 249 1.2.2.2 tls int 250 1.2.2.2 tls md_pre_mount() 251 1.2.2.2 tls { 252 1.2.2.2 tls return 0; 253 1.2.2.2 tls } 254 1.2.2.2 tls 255 1.2.2.2 tls int 256 1.2.2.2 tls md_check_mbr(mbr_info_t *mbri) 257 1.2.2.2 tls { 258 1.2.2.2 tls mbr_info_t *ext; 259 1.2.2.2 tls struct mbr_partition *part; 260 1.2.2.2 tls int i, hasboot=0; 261 1.2.2.2 tls 262 1.2.2.2 tls if (boardtype == BOARD_TYPE_NORMAL) 263 1.2.2.2 tls return 2; 264 1.2.2.2 tls /* raspi code */ 265 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) { 266 1.2.2.2 tls for (ext = mbri; ext; ext = ext->extended) { 267 1.2.2.2 tls part = ext->mbr.mbr_parts; 268 1.2.2.2 tls for (i=0, hasboot=0; i < MBR_PART_COUNT; part++, i++) { 269 1.2.2.2 tls if (part->mbrp_type != MBR_PTYPE_FAT32L) 270 1.2.2.2 tls continue; 271 1.2.2.2 tls hasboot = 1; 272 1.2.2.2 tls break; 273 1.2.2.2 tls } 274 1.2.2.2 tls } 275 1.2.2.2 tls if (!hasboot) { 276 1.2.2.2 tls msg_display(MSG_nomsdospart); 277 1.2.2.2 tls msg_display_add(MSG_reeditpart, 0); 278 1.2.2.2 tls process_menu(MENU_yesno, NULL); 279 1.2.2.2 tls if (!yesno) 280 1.2.2.2 tls return 0; 281 1.2.2.2 tls return 1; 282 1.2.2.2 tls } 283 1.2.2.2 tls } 284 1.2.2.2 tls return 2; 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_mbr_use_wholedisk(mbr_info_t *mbri) 289 1.2.2.2 tls { 290 1.2.2.2 tls struct mbr_sector *mbrs = &mbri->mbr; 291 1.2.2.2 tls struct mbr_partition *part; 292 1.2.2.2 tls int offset; 293 1.2.2.2 tls 294 1.2.2.2 tls if (boardtype == BOARD_TYPE_NORMAL) { 295 1.2.2.2 tls /* this keeps it from creating /boot as msdos */ 296 1.2.2.2 tls pm->bootsize = 0; 297 1.2.2.2 tls return mbr_use_wholedisk(mbri); 298 1.2.2.2 tls } 299 1.2.2.2 tls 300 1.2.2.2 tls /* raspi code */ 301 1.2.2.2 tls if (boardtype == BOARD_TYPE_RPI) { 302 1.2.2.2 tls part = &mbrs->mbr_parts[0]; 303 1.2.2.2 tls if (part[0].mbrp_type != MBR_PTYPE_FAT32L) { 304 1.2.2.2 tls /* It's hopelessly corrupt, punt for now */ 305 1.2.2.2 tls msg_display(MSG_nomsdospart); 306 1.2.2.2 tls process_menu(MENU_ok, NULL); 307 1.2.2.2 tls return 0; 308 1.2.2.2 tls } 309 1.2.2.2 tls offset = part[0].mbrp_start + part[0].mbrp_size; 310 1.2.2.2 tls part[1].mbrp_type = MBR_PTYPE_NETBSD; 311 1.2.2.2 tls part[1].mbrp_size = pm->dlsize - offset; 312 1.2.2.2 tls part[1].mbrp_start = offset; 313 1.2.2.2 tls part[1].mbrp_flag = 0; 314 1.2.2.2 tls 315 1.2.2.2 tls pm->ptstart = part[1].mbrp_start; 316 1.2.2.2 tls pm->ptsize = part[1].mbrp_size; 317 1.2.2.2 tls pm->bootstart = part[0].mbrp_start; 318 1.2.2.2 tls pm->bootsize = part[0].mbrp_size; 319 1.2.2.2 tls return 1; 320 1.2.2.2 tls } 321 1.2.2.2 tls 322 1.2.2.2 tls return mbr_use_wholedisk(mbri); 323 1.2.2.2 tls } 324
Indexes created Thu Oct 02 14:10:14 GMT 2025