distrib/utils/sysinst/arch/shark/md.c

   1 /*      $NetBSD: md.c,v 1.23 2008/10/07 09:58:15 abs Exp $      */
   2
   3 /*
   4  * Copyright 1997 Piermont Information Systems Inc.
   5  * All rights reserved.
   6  *
   7  * Based on code written by Philip A. Nelson for Piermont Information
   8  * Systems Inc.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. All advertising materials mentioning features or use of this software
  19  *    must display the following acknowledgement:
  20  *      This product includes software developed for the NetBSD Project by
  21  *      Piermont Information Systems Inc.
  22  * 4. The name of Piermont Information Systems Inc. may not be used to endorse
  23  *    or promote products derived from this software without specific prior
  24  *    written permission.
  25  *
  26  * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
  27  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  28  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  29  * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
  30  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  31  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  32  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  33  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  34  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  35  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  36  * THE POSSIBILITY OF SUCH DAMAGE.
  37  */
  38
  39 /* md.c -- shark machine specific routines */
  40
  41 #include <stdio.h>
  42 #include <curses.h>
  43 #include <unistd.h>
  44 #include <fcntl.h>
  45 #include <util.h>
  46 #include <sys/types.h>
  47 #include <sys/disklabel.h>
  48 #include <sys/ioctl.h>
  49 #include <sys/param.h>
  50
  51 #include "defs.h"
  52 #include "md.h"
  53 #include "msg_defs.h"
  54 #include "menu_defs.h"
  55
  56 static int clear_mbr(const char *, char *, size_t);
  57
  58 void
  59 md_init(void)
  60 {
  61 }
  62
  63 void
  64 md_init_set_status(int minimal)
  65 {
  66         (void)minimal;
  67 }
  68
  69 int
  70 md_get_info(void)
  71 {
  72         struct disklabel disklabel;
  73         int fd;
  74         char dev_name[100];
  75
  76         if (strncmp(diskdev, "wd", 2) == 0)
  77                 disktype = "ST506";
  78         else
  79                 disktype = "SCSI";
  80
  81         snprintf(dev_name, 100, "/dev/r%sc", diskdev);
  82
  83         fd = open(dev_name, O_RDONLY, 0);
  84         if (fd < 0) {
  85                 endwin();
  86                 fprintf(stderr, "Can't open %s\n", dev_name);
  87                 exit(1);
  88         }
  89         if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) {
  90                 endwin();
  91                 fprintf(stderr, "Can't read disklabel on %s.\n", dev_name);
  92                 close(fd);
  93                 exit(1);
  94         }
  95         close(fd);
  96
  97         dlcyl = disklabel.d_ncylinders;
  98         dlhead = disklabel.d_ntracks;
  99         dlsec = disklabel.d_nsectors;
 100         sectorsize = disklabel.d_secsize;
 101         dlcylsize = disklabel.d_secpercyl;
 102
 103         /*
 104          * Compute whole disk size. Take max of (dlcyl*dlhead*dlsec)
 105          * and secperunit,  just in case the disk is already labelled.
 106          * (If our new label's RAW_PART size ends up smaller than the
 107          * in-core RAW_PART size  value, updating the label will fail.)
 108          */
 109         dlsize = dlcyl*dlhead*dlsec;
 110         if (disklabel.d_secperunit > dlsize)
 111                 dlsize = disklabel.d_secperunit;
 112
 113         return 1;
 114 }
 115
 116 /*
 117  * md back-end code for menu-driven BSD disklabel editor.
 118  */
 119 int
 120 md_make_bsd_partitions(void)
 121 {
 122         return make_bsd_partitions();
 123 }
 124
 125 /*
 126  * any additional partition validation
 127  */
 128 int
 129 md_check_partitions(void)
 130 {
 131         return 1;
 132 }
 133
 134 /*
 135  * hook called before writing new disklabel.
 136  */
 137 int
 138 md_pre_disklabel(void)
 139 {
 140         char diskpath[MAXPATHLEN];
 141
 142         if (clear_mbr(diskdev, diskpath, sizeof(diskpath)) == -1) {
 143                 msg_display(MSG_badclearmbr, diskpath);
 144                 process_menu(MENU_ok, NULL);
 145         }
 146
 147         return 0;
 148 }
 149
 150 /*
 151  * hook called after writing disklabel to new target disk.
 152  */
 153 int
 154 md_post_disklabel(void)
 155 {
 156         return 0;
 157 }
 158
 159 /*
 160  * hook called after upgrade() or install() has finished setting
 161  * up the target disk but immediately before the user is given the
 162  * ``disks are now set up'' message.
 163  */
 164 int
 165 md_post_newfs(void)
 166 {
 167         return 0;
 168 }
 169
 170 int
 171 md_post_extract(void)
 172 {
 173         msg_display(MSG_setbootdevice);
 174         process_menu(MENU_ok, NULL);
 175
 176         return 0;
 177 }
 178
 179 void
 180 md_cleanup_install(void)
 181 {
 182 #ifndef DEBUG
 183         enable_rc_conf();
 184         add_rc_conf("wscons=YES\n");
 185
 186         /* Configure a single screen. */
 187         run_program(RUN_CHROOT,
 188                     "sed -an -e '/^screen/s/^/#/;/^mux/s/^/#/;"
 189                     "H;$!d;g;w /etc/wscons.conf' /etc/wscons.conf");
 190
 191 #endif
 192 }
 193
 194 int
 195 md_pre_update(void)
 196 {
 197         return 1;
 198 }
 199
 200 /* Upgrade support */
 201 int
 202 md_update(void)
 203 {
 204         md_post_newfs();
 205         return 1;
 206 }
 207
 208 /*
 209  * Clears the disk's first sector by writing all zeros over it.
 210  * Returns 0 on success, -1 on failure.  Leaves the disk's path in
 211  * the output diskpath buffer for further usage in error messages.
 212  */
 213 static int
 214 clear_mbr(const char *disk, char *diskpath, size_t diskpathlen)
 215 {
 216         int fd;
 217         char sector[512];
 218
 219         fd = opendisk(disk, O_WRONLY, diskpath, diskpathlen, 0);
 220         if (fd < 0)
 221                 return -1;
 222
 223         memset(sector, 0, sizeof(sector));
 224         if (pwrite(fd, &sector, sizeof(sector), 0) < 0) {
 225                 close(fd);
 226                 return -1;
 227         }
 228
 229         close(fd);
 230         return 0;
 231 }
 232