1*4204f810Smartin /* $NetBSD: md.c,v 1.7 2022/01/29 16:01:20 martin Exp $ */
250dbef1aSdholland
350dbef1aSdholland /*
450dbef1aSdholland * Copyright 1997 Piermont Information Systems Inc.
550dbef1aSdholland * All rights reserved.
650dbef1aSdholland *
750dbef1aSdholland * Based on code written by Philip A. Nelson for Piermont Information
850dbef1aSdholland * Systems Inc.
950dbef1aSdholland *
1050dbef1aSdholland * Redistribution and use in source and binary forms, with or without
1150dbef1aSdholland * modification, are permitted provided that the following conditions
1250dbef1aSdholland * are met:
1350dbef1aSdholland * 1. Redistributions of source code must retain the above copyright
1450dbef1aSdholland * notice, this list of conditions and the following disclaimer.
1550dbef1aSdholland * 2. Redistributions in binary form must reproduce the above copyright
1650dbef1aSdholland * notice, this list of conditions and the following disclaimer in the
1750dbef1aSdholland * documentation and/or other materials provided with the distribution.
1850dbef1aSdholland * 3. The name of Piermont Information Systems Inc. may not be used to endorse
1950dbef1aSdholland * or promote products derived from this software without specific prior
2050dbef1aSdholland * written permission.
2150dbef1aSdholland *
2250dbef1aSdholland * THIS SOFTWARE IS PROVIDED BY PIERMONT INFORMATION SYSTEMS INC. ``AS IS''
2350dbef1aSdholland * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
2450dbef1aSdholland * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
2550dbef1aSdholland * ARE DISCLAIMED. IN NO EVENT SHALL PIERMONT INFORMATION SYSTEMS INC. BE
2650dbef1aSdholland * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
2750dbef1aSdholland * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
2850dbef1aSdholland * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
2950dbef1aSdholland * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
3050dbef1aSdholland * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
3150dbef1aSdholland * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
3250dbef1aSdholland * THE POSSIBILITY OF SUCH DAMAGE.
3350dbef1aSdholland */
3450dbef1aSdholland
3550dbef1aSdholland /* md.c -- shark machine specific routines */
3650dbef1aSdholland
3750dbef1aSdholland #include <stdio.h>
3850dbef1aSdholland #include <curses.h>
3950dbef1aSdholland #include <unistd.h>
4050dbef1aSdholland #include <fcntl.h>
4150dbef1aSdholland #include <util.h>
4250dbef1aSdholland #include <sys/types.h>
4350dbef1aSdholland #include <sys/ioctl.h>
4450dbef1aSdholland #include <sys/param.h>
4550dbef1aSdholland
4650dbef1aSdholland #include "defs.h"
4750dbef1aSdholland #include "md.h"
484103857bSmartin #include "mbr.h"
4950dbef1aSdholland #include "msg_defs.h"
5050dbef1aSdholland #include "menu_defs.h"
5150dbef1aSdholland
5250dbef1aSdholland static int clear_mbr(const char *, char *, size_t);
5350dbef1aSdholland
5450dbef1aSdholland void
md_init(void)5550dbef1aSdholland md_init(void)
5650dbef1aSdholland {
5750dbef1aSdholland }
5850dbef1aSdholland
5950dbef1aSdholland void
md_init_set_status(int flags)6050dbef1aSdholland md_init_set_status(int flags)
6150dbef1aSdholland {
6250dbef1aSdholland (void)flags;
6350dbef1aSdholland }
6450dbef1aSdholland
654103857bSmartin bool
md_get_info(struct install_partition_desc * install)664103857bSmartin md_get_info(struct install_partition_desc *install)
6750dbef1aSdholland {
6850dbef1aSdholland struct disklabel disklabel;
6950dbef1aSdholland int fd;
7050dbef1aSdholland char dev_name[100];
7150dbef1aSdholland
724b2364d9Smartin snprintf(dev_name, 100, "/dev/r%sc", pm->diskdev);
7350dbef1aSdholland
7450dbef1aSdholland fd = open(dev_name, O_RDONLY, 0);
7550dbef1aSdholland if (fd < 0) {
7650dbef1aSdholland endwin();
7750dbef1aSdholland fprintf(stderr, "Can't open %s\n", dev_name);
7850dbef1aSdholland exit(1);
7950dbef1aSdholland }
8050dbef1aSdholland if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) {
8150dbef1aSdholland endwin();
8250dbef1aSdholland fprintf(stderr, "Can't read disklabel on %s.\n", dev_name);
8350dbef1aSdholland close(fd);
8450dbef1aSdholland exit(1);
8550dbef1aSdholland }
8650dbef1aSdholland close(fd);
8750dbef1aSdholland
884b2364d9Smartin pm->dlcyl = disklabel.d_ncylinders;
894b2364d9Smartin pm->dlhead = disklabel.d_ntracks;
904b2364d9Smartin pm->dlsec = disklabel.d_nsectors;
914b2364d9Smartin pm->sectorsize = disklabel.d_secsize;
924b2364d9Smartin pm->dlcylsize = disklabel.d_secpercyl;
9350dbef1aSdholland
9450dbef1aSdholland /*
954b2364d9Smartin * Compute whole disk size. Take max of (pm->dlcyl*pm->dlhead*pm->dlsec)
9650dbef1aSdholland * and secperunit, just in case the disk is already labelled.
9750dbef1aSdholland * (If our new label's RAW_PART size ends up smaller than the
9850dbef1aSdholland * in-core RAW_PART size value, updating the label will fail.)
9950dbef1aSdholland */
1004b2364d9Smartin pm->dlsize = pm->dlcyl*pm->dlhead*pm->dlsec;
1014b2364d9Smartin if (disklabel.d_secperunit > pm->dlsize)
1024b2364d9Smartin pm->dlsize = disklabel.d_secperunit;
10350dbef1aSdholland
1044103857bSmartin return true;
10550dbef1aSdholland }
10650dbef1aSdholland
10750dbef1aSdholland /*
10850dbef1aSdholland * md back-end code for menu-driven BSD disklabel editor.
10950dbef1aSdholland */
110957b5cd6Smartin int
md_make_bsd_partitions(struct install_partition_desc * install)1114103857bSmartin md_make_bsd_partitions(struct install_partition_desc *install)
11250dbef1aSdholland {
1134103857bSmartin return make_bsd_partitions(install);
11450dbef1aSdholland }
11550dbef1aSdholland
11650dbef1aSdholland /*
11750dbef1aSdholland * any additional partition validation
11850dbef1aSdholland */
1194103857bSmartin bool
md_check_partitions(struct install_partition_desc * install)1204103857bSmartin md_check_partitions(struct install_partition_desc *install)
12150dbef1aSdholland {
1224103857bSmartin return true;
12350dbef1aSdholland }
12450dbef1aSdholland
12550dbef1aSdholland /*
12650dbef1aSdholland * hook called before writing new disklabel.
12750dbef1aSdholland */
1284103857bSmartin bool
md_pre_disklabel(struct install_partition_desc * install,struct disk_partitions * parts)1294103857bSmartin md_pre_disklabel(struct install_partition_desc *install,
1304103857bSmartin struct disk_partitions *parts)
13150dbef1aSdholland {
13250dbef1aSdholland char diskpath[MAXPATHLEN];
13350dbef1aSdholland
1344b2364d9Smartin if (clear_mbr(pm->diskdev, diskpath, sizeof(diskpath)) == -1) {
13524ecf24eSchristos msg_fmt_display(MSG_badclearmbr, "%s", diskpath);
13650dbef1aSdholland process_menu(MENU_ok, NULL);
13750dbef1aSdholland }
13850dbef1aSdholland
1394103857bSmartin return true;
14050dbef1aSdholland }
14150dbef1aSdholland
14250dbef1aSdholland /*
14350dbef1aSdholland * hook called after writing disklabel to new target disk.
14450dbef1aSdholland */
1454103857bSmartin bool
md_post_disklabel(struct install_partition_desc * install,struct disk_partitions * parts)1464103857bSmartin md_post_disklabel(struct install_partition_desc *install,
1474103857bSmartin struct disk_partitions *parts)
14850dbef1aSdholland {
1494103857bSmartin return true;
15050dbef1aSdholland }
15150dbef1aSdholland
15250dbef1aSdholland /*
15350dbef1aSdholland * hook called after upgrade() or install() has finished setting
15450dbef1aSdholland * up the target disk but immediately before the user is given the
15550dbef1aSdholland * ``disks are now set up'' message.
15650dbef1aSdholland */
15750dbef1aSdholland int
md_post_newfs(struct install_partition_desc * install)1584103857bSmartin md_post_newfs(struct install_partition_desc *install)
15950dbef1aSdholland {
16050dbef1aSdholland return 0;
16150dbef1aSdholland }
16250dbef1aSdholland
16350dbef1aSdholland int
md_post_extract(struct install_partition_desc * install,bool upgrade)164*4204f810Smartin md_post_extract(struct install_partition_desc *install, bool upgrade)
16550dbef1aSdholland {
16650dbef1aSdholland msg_display(MSG_setbootdevice);
16750dbef1aSdholland process_menu(MENU_ok, NULL);
16850dbef1aSdholland
16950dbef1aSdholland return 0;
17050dbef1aSdholland }
17150dbef1aSdholland
17250dbef1aSdholland void
md_cleanup_install(struct install_partition_desc * install)1734103857bSmartin md_cleanup_install(struct install_partition_desc *install)
17450dbef1aSdholland {
17550dbef1aSdholland #ifndef DEBUG
17650dbef1aSdholland enable_rc_conf();
17750dbef1aSdholland add_rc_conf("wscons=YES\n");
17850dbef1aSdholland
17950dbef1aSdholland /* Configure a single screen. */
18050dbef1aSdholland run_program(RUN_CHROOT,
18150dbef1aSdholland "sed -an -e '/^screen/s/^/#/;/^mux/s/^/#/;"
18250dbef1aSdholland "H;$!d;g;w /etc/wscons.conf' /etc/wscons.conf");
18350dbef1aSdholland
18450dbef1aSdholland #endif
18550dbef1aSdholland }
18650dbef1aSdholland
18750dbef1aSdholland int
md_pre_update(struct install_partition_desc * install)1884103857bSmartin md_pre_update(struct install_partition_desc *install)
18950dbef1aSdholland {
19050dbef1aSdholland return 1;
19150dbef1aSdholland }
19250dbef1aSdholland
19350dbef1aSdholland /* Upgrade support */
19450dbef1aSdholland int
md_update(struct install_partition_desc * install)1954103857bSmartin md_update(struct install_partition_desc *install)
19650dbef1aSdholland {
1974103857bSmartin md_post_newfs(install);
19850dbef1aSdholland return 1;
19950dbef1aSdholland }
20050dbef1aSdholland
20150dbef1aSdholland /*
20250dbef1aSdholland * Clears the disk's first sector by writing all zeros over it.
20350dbef1aSdholland * Returns 0 on success, -1 on failure. Leaves the disk's path in
20450dbef1aSdholland * the output diskpath buffer for further usage in error messages.
20550dbef1aSdholland */
20650dbef1aSdholland static int
clear_mbr(const char * disk,char * diskpath,size_t diskpathlen)20750dbef1aSdholland clear_mbr(const char *disk, char *diskpath, size_t diskpathlen)
20850dbef1aSdholland {
20950dbef1aSdholland int fd;
21050dbef1aSdholland char sector[512];
21150dbef1aSdholland
21250dbef1aSdholland fd = opendisk(disk, O_WRONLY, diskpath, diskpathlen, 0);
21350dbef1aSdholland if (fd < 0)
21450dbef1aSdholland return -1;
21550dbef1aSdholland
21650dbef1aSdholland memset(sector, 0, sizeof(sector));
21750dbef1aSdholland if (pwrite(fd, §or, sizeof(sector), 0) < 0) {
21850dbef1aSdholland close(fd);
21950dbef1aSdholland return -1;
22050dbef1aSdholland }
22150dbef1aSdholland
22250dbef1aSdholland close(fd);
22350dbef1aSdholland return 0;
22450dbef1aSdholland }
22550dbef1aSdholland
22650dbef1aSdholland int
md_pre_mount(struct install_partition_desc * install,size_t ndx)2274f30cbf3Smartin md_pre_mount(struct install_partition_desc *install, size_t ndx)
22850dbef1aSdholland {
22950dbef1aSdholland return 0;
23050dbef1aSdholland }
2314103857bSmartin
2324103857bSmartin /* returns false if no write-back of parts is required */
2334103857bSmartin bool
md_mbr_update_check(struct disk_partitions * parts,mbr_info_t * mbri)2344103857bSmartin md_mbr_update_check(struct disk_partitions *parts, mbr_info_t *mbri)
2354103857bSmartin {
2364103857bSmartin return false;
2374103857bSmartin }
2384103857bSmartin
2394103857bSmartin #ifdef HAVE_GPT
2404103857bSmartin /*
2414103857bSmartin * New GPT partitions have been written, update bootloader or remember
2424103857bSmartin * data untill needed in md_post_newfs
2434103857bSmartin */
2444103857bSmartin bool
md_gpt_post_write(struct disk_partitions * parts,part_id root_id,bool root_is_new,part_id efi_id,bool efi_is_new)2454103857bSmartin md_gpt_post_write(struct disk_partitions *parts, part_id root_id,
2464103857bSmartin bool root_is_new, part_id efi_id, bool efi_is_new)
2474103857bSmartin {
2484103857bSmartin return true;
2494103857bSmartin }
2504103857bSmartin #endif
2514103857bSmartin
2524103857bSmartin bool
md_parts_use_wholedisk(struct disk_partitions * parts)2534103857bSmartin md_parts_use_wholedisk(struct disk_partitions *parts)
2544103857bSmartin {
2554103857bSmartin return parts_use_wholedisk(parts, 0, NULL);
2564103857bSmartin }
257