Home | History | Annotate | Line # | Download | only in kern
subr_disk_mbr.c revision 1.56.14.1
      1  1.56.14.1      cjep /*	$NetBSD: subr_disk_mbr.c,v 1.56.14.1 2021/05/31 22:15:20 cjep Exp $	*/
      2        1.1       dsl 
      3        1.1       dsl /*
      4        1.1       dsl  * Copyright (c) 1982, 1986, 1988 Regents of the University of California.
      5        1.1       dsl  * All rights reserved.
      6        1.1       dsl  *
      7        1.1       dsl  * Redistribution and use in source and binary forms, with or without
      8        1.1       dsl  * modification, are permitted provided that the following conditions
      9        1.1       dsl  * are met:
     10        1.1       dsl  * 1. Redistributions of source code must retain the above copyright
     11        1.1       dsl  *    notice, this list of conditions and the following disclaimer.
     12        1.1       dsl  * 2. Redistributions in binary form must reproduce the above copyright
     13        1.1       dsl  *    notice, this list of conditions and the following disclaimer in the
     14        1.1       dsl  *    documentation and/or other materials provided with the distribution.
     15        1.2       agc  * 3. Neither the name of the University nor the names of its contributors
     16        1.1       dsl  *    may be used to endorse or promote products derived from this software
     17        1.1       dsl  *    without specific prior written permission.
     18        1.1       dsl  *
     19        1.1       dsl  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20        1.1       dsl  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21        1.1       dsl  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22        1.1       dsl  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23        1.1       dsl  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24        1.1       dsl  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25        1.1       dsl  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26        1.1       dsl  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27        1.1       dsl  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28        1.1       dsl  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29        1.1       dsl  * SUCH DAMAGE.
     30        1.1       dsl  *
     31        1.1       dsl  *	@(#)ufs_disksubr.c	7.16 (Berkeley) 5/4/91
     32        1.1       dsl  */
     33        1.1       dsl 
     34        1.1       dsl /*
     35        1.1       dsl  * Code to find a NetBSD label on a disk that contains an i386 style MBR.
     36        1.1       dsl  * The first NetBSD label found in the 2nd sector of a NetBSD partition
     37        1.3       dsl  * is used.
     38        1.3       dsl  * If we don't find a label searching the MBR, we look at the start of the
     39        1.3       dsl  * disk, if that fails then a label is faked up from the MBR.
     40        1.1       dsl  *
     41       1.32   reinoud  * If there isn't a disklabel or anything in the MBR then the disc is searched
     42       1.32   reinoud  * for ecma-167/iso9660/udf style partition indicators.
     43       1.32   reinoud  * Useful for media or files that contain single filesystems (etc).
     44        1.1       dsl  *
     45        1.1       dsl  * This code will read host endian netbsd labels from little endian MBR.
     46        1.1       dsl  *
     47        1.1       dsl  * Based on the i386 disksubr.c
     48        1.1       dsl  *
     49        1.3       dsl  * Since the mbr only has 32bit fields for sector addresses, we do the same.
     50        1.3       dsl  *
     51        1.1       dsl  * XXX There are potential problems writing labels to disks where there
     52        1.1       dsl  * is only space for 8 netbsd partitions but this code has been compiled
     53        1.1       dsl  * with MAXPARTITIONS=16.
     54        1.1       dsl  */
     55        1.1       dsl 
     56        1.1       dsl #include <sys/cdefs.h>
     57  1.56.14.1      cjep __KERNEL_RCSID(0, "$NetBSD: subr_disk_mbr.c,v 1.56.14.1 2021/05/31 22:15:20 cjep Exp $");
     58  1.56.14.1      cjep 
     59  1.56.14.1      cjep #ifdef _KERNEL_OPT
     60  1.56.14.1      cjep #include "opt_mbr.h"
     61  1.56.14.1      cjep #include "opt_disklabel.h"
     62  1.56.14.1      cjep #endif /* _KERNEL_OPT */
     63        1.1       dsl 
     64        1.1       dsl #include <sys/param.h>
     65        1.1       dsl #include <sys/systm.h>
     66        1.1       dsl #include <sys/buf.h>
     67       1.27    dyoung #include <sys/bootblock.h>
     68        1.1       dsl #include <sys/disklabel.h>
     69        1.1       dsl #include <sys/disk.h>
     70        1.1       dsl #include <sys/syslog.h>
     71       1.32   reinoud #include <sys/vnode.h>
     72       1.32   reinoud #include <sys/fcntl.h>
     73       1.32   reinoud #include <sys/conf.h>
     74       1.32   reinoud #include <sys/cdio.h>
     75       1.37     pooka #include <sys/dkbad.h>
     76       1.32   reinoud #include <fs/udf/ecma167-udf.h>
     77       1.32   reinoud 
     78       1.32   reinoud #include <sys/kauth.h>
     79        1.1       dsl 
     80       1.13       dsl typedef struct mbr_partition mbr_partition_t;
     81       1.13       dsl 
     82       1.17       dsl /*
     83       1.38     pooka  * We allocate a buffer 3 sectors large, and look in all....
     84       1.17       dsl  * That means we find labels written by other ports with different offsets.
     85       1.17       dsl  * LABELSECTOR and LABELOFFSET are only used if the disk doesn't have a label.
     86       1.17       dsl  */
     87       1.38     pooka #define SCANBLOCKS 3
     88       1.38     pooka #define DISKLABEL_SIZE 404
     89       1.38     pooka #if LABELSECTOR*DEV_BSIZE + LABELOFFSET > SCANBLOCKS*DEV_BSIZE - DISKLABEL_SIZE
     90       1.41     pooka #if _MACHINE != ews4800mips /* XXX: fail silently, ews4800mips LABELSECTOR */
     91       1.17       dsl #error Invalid LABELSECTOR or LABELOFFSET
     92       1.17       dsl #endif
     93       1.40   tsutsui #endif
     94       1.17       dsl 
     95        1.1       dsl #define MBR_LABELSECTOR	1
     96        1.1       dsl 
     97        1.1       dsl #define SCAN_CONTINUE	0
     98        1.1       dsl #define SCAN_FOUND	1
     99        1.1       dsl #define SCAN_ERROR	2
    100        1.1       dsl 
    101        1.1       dsl typedef struct mbr_args {
    102        1.1       dsl 	struct disklabel *lp;
    103        1.1       dsl 	void		(*strat)(struct buf *);
    104        1.1       dsl 	struct buf	*bp;
    105        1.1       dsl 	const char	*msg;
    106        1.1       dsl 	int		error;
    107        1.3       dsl 	int		written;	/* number of times we wrote label */
    108       1.17       dsl 	int		found_mbr;	/* set if disk has a valid mbr */
    109        1.3       dsl 	uint		label_sector;	/* where we found the label */
    110       1.17       dsl 	int		action;
    111       1.28    dyoung 	uint32_t	secperunit;
    112        1.1       dsl #define READ_LABEL	1
    113        1.3       dsl #define UPDATE_LABEL	2
    114        1.3       dsl #define WRITE_LABEL	3
    115       1.17       dsl } mbr_args_t;
    116       1.17       dsl 
    117       1.17       dsl static int validate_label(mbr_args_t *, uint);
    118       1.13       dsl static int look_netbsd_part(mbr_args_t *, mbr_partition_t *, int, uint);
    119       1.13       dsl static int write_netbsd_label(mbr_args_t *, mbr_partition_t *, int, uint);
    120        1.1       dsl 
    121        1.1       dsl static int
    122       1.17       dsl read_sector(mbr_args_t *a, uint sector, int count)
    123        1.1       dsl {
    124        1.1       dsl 	int error;
    125        1.1       dsl 
    126       1.28    dyoung 	error = disk_read_sectors(a->strat, a->lp, a->bp, sector, count);
    127        1.1       dsl 	if (error != 0)
    128        1.1       dsl 		a->error = error;
    129        1.1       dsl 	return error;
    130        1.1       dsl }
    131        1.1       dsl 
    132        1.6     perry /*
    133        1.1       dsl  * Scan MBR for partitions, call 'action' routine for each.
    134        1.1       dsl  */
    135        1.1       dsl 
    136        1.1       dsl static int
    137       1.13       dsl scan_mbr(mbr_args_t *a, int (*actn)(mbr_args_t *, mbr_partition_t *, int, uint))
    138        1.1       dsl {
    139       1.13       dsl 	mbr_partition_t ptns[MBR_PART_COUNT];
    140       1.13       dsl 	mbr_partition_t *dp;
    141        1.4     lukem 	struct mbr_sector *mbr;
    142        1.1       dsl 	uint ext_base, this_ext, next_ext;
    143        1.1       dsl 	int rval;
    144        1.1       dsl 	int i;
    145       1.18  christos 	int j;
    146        1.1       dsl #ifdef COMPAT_386BSD_MBRPART
    147        1.1       dsl 	int dp_386bsd = -1;
    148       1.18  christos 	int ap_386bsd = -1;
    149        1.1       dsl #endif
    150        1.1       dsl 
    151        1.1       dsl 	ext_base = 0;
    152        1.1       dsl 	this_ext = 0;
    153        1.1       dsl 	for (;;) {
    154       1.17       dsl 		if (read_sector(a, this_ext, 1)) {
    155        1.1       dsl 			a->msg = "dos partition I/O error";
    156        1.1       dsl 			return SCAN_ERROR;
    157        1.1       dsl 		}
    158        1.1       dsl 
    159        1.1       dsl 		/* Note: Magic number is little-endian. */
    160        1.1       dsl 		mbr = (void *)a->bp->b_data;
    161        1.4     lukem 		if (mbr->mbr_magic != htole16(MBR_MAGIC))
    162        1.3       dsl 			return SCAN_CONTINUE;
    163        1.1       dsl 
    164       1.46      matt 		/*
    165       1.46      matt 		 * If this is a protective MBR, bail now.
    166       1.46      matt 		 */
    167       1.46      matt 		if (mbr->mbr_parts[0].mbrp_type == MBR_PTYPE_PMBR
    168       1.46      matt 		    && mbr->mbr_parts[1].mbrp_type == MBR_PTYPE_UNUSED
    169       1.46      matt 		    && mbr->mbr_parts[2].mbrp_type == MBR_PTYPE_UNUSED
    170       1.46      matt 		    && mbr->mbr_parts[3].mbrp_type == MBR_PTYPE_UNUSED)
    171       1.46      matt 			return SCAN_CONTINUE;
    172       1.46      matt 
    173        1.1       dsl 		/* Copy data out of buffer so action can use bp */
    174        1.1       dsl 		memcpy(ptns, &mbr->mbr_parts, sizeof ptns);
    175        1.1       dsl 
    176       1.14  christos 		/* Look for drivers and skip them */
    177       1.17       dsl 		if (ext_base == 0 && ptns[0].mbrp_type == MBR_PTYPE_DM6_DDO) {
    178       1.16      jmmv 			/* We've found a DM6 DDO partition type (used by
    179       1.16      jmmv 			 * the Ontrack Disk Manager drivers).
    180       1.16      jmmv 			 *
    181       1.16      jmmv 			 * Ensure that there are no other partitions in the
    182       1.16      jmmv 			 * MBR and jump to the real partition table (stored
    183       1.16      jmmv 			 * in the first sector of the second track). */
    184       1.25   thorpej 			bool ok = true;
    185       1.14  christos 
    186       1.14  christos 			for (i = 1; i < MBR_PART_COUNT; i++)
    187       1.14  christos 				if (ptns[i].mbrp_type != MBR_PTYPE_UNUSED)
    188       1.25   thorpej 					ok = false;
    189       1.14  christos 
    190       1.14  christos 			if (ok) {
    191       1.15  christos 				this_ext = le32toh(a->lp->d_secpercyl /
    192       1.15  christos 				    a->lp->d_ntracks);
    193       1.14  christos 				continue;
    194       1.14  christos 			}
    195       1.14  christos 		}
    196       1.14  christos 
    197        1.1       dsl 		/* look for NetBSD partition */
    198        1.1       dsl 		next_ext = 0;
    199        1.1       dsl 		dp = ptns;
    200       1.18  christos 		j = 0;
    201        1.4     lukem 		for (i = 0; i < MBR_PART_COUNT; i++, dp++) {
    202       1.18  christos 			if (dp->mbrp_type == MBR_PTYPE_UNUSED)
    203        1.1       dsl 				continue;
    204       1.17       dsl 			/* Check end of partition is inside disk limits */
    205       1.17       dsl 			if ((uint64_t)ext_base + le32toh(dp->mbrp_start) +
    206       1.17       dsl 			    le32toh(dp->mbrp_size) > a->lp->d_secperunit) {
    207       1.17       dsl 				/* This mbr doesn't look good.... */
    208       1.17       dsl 				a->msg = "mbr partition exceeds disk size";
    209       1.17       dsl 				/* ...but don't report this as an error (yet) */
    210       1.17       dsl 				return SCAN_CONTINUE;
    211       1.17       dsl 			}
    212       1.17       dsl 			a->found_mbr = 1;
    213        1.4     lukem 			if (MBR_IS_EXTENDED(dp->mbrp_type)) {
    214        1.1       dsl 				next_ext = le32toh(dp->mbrp_start);
    215        1.1       dsl 				continue;
    216        1.1       dsl 			}
    217        1.1       dsl #ifdef COMPAT_386BSD_MBRPART
    218        1.4     lukem 			if (dp->mbrp_type == MBR_PTYPE_386BSD) {
    219        1.1       dsl 				/*
    220        1.1       dsl 				 * If more than one matches, take last,
    221        1.1       dsl 				 * as NetBSD install tool does.
    222        1.1       dsl 				 */
    223       1.18  christos 				if (this_ext == 0) {
    224        1.1       dsl 					dp_386bsd = i;
    225       1.18  christos 					ap_386bsd = j;
    226       1.18  christos 				}
    227        1.1       dsl 				continue;
    228        1.1       dsl 			}
    229        1.1       dsl #endif
    230       1.18  christos 			rval = (*actn)(a, dp, j, this_ext);
    231        1.1       dsl 			if (rval != SCAN_CONTINUE)
    232        1.1       dsl 				return rval;
    233       1.18  christos 			j++;
    234        1.1       dsl 		}
    235        1.1       dsl 		if (next_ext == 0)
    236        1.1       dsl 			break;
    237        1.1       dsl 		if (ext_base == 0) {
    238        1.1       dsl 			ext_base = next_ext;
    239        1.1       dsl 			next_ext = 0;
    240        1.1       dsl 		}
    241        1.1       dsl 		next_ext += ext_base;
    242        1.1       dsl 		if (next_ext <= this_ext)
    243        1.1       dsl 			break;
    244        1.1       dsl 		this_ext = next_ext;
    245        1.1       dsl 	}
    246        1.1       dsl #ifdef COMPAT_386BSD_MBRPART
    247        1.1       dsl 	if (this_ext == 0 && dp_386bsd != -1)
    248       1.18  christos 		return (*actn)(a, &ptns[dp_386bsd], ap_386bsd, 0);
    249        1.1       dsl #endif
    250        1.1       dsl 	return SCAN_CONTINUE;
    251        1.1       dsl }
    252        1.1       dsl 
    253       1.32   reinoud 
    254       1.32   reinoud static void
    255       1.32   reinoud scan_iso_vrs_session(mbr_args_t *a, uint32_t first_sector,
    256       1.32   reinoud 	int *is_iso9660, int *is_udf)
    257       1.32   reinoud {
    258       1.32   reinoud 	struct vrs_desc *vrsd;
    259       1.32   reinoud 	uint64_t vrs;
    260       1.32   reinoud 	int sector_size;
    261       1.33   reinoud 	int blks, inc;
    262       1.32   reinoud 
    263       1.32   reinoud 	sector_size = a->lp->d_secsize;
    264       1.32   reinoud 	blks = sector_size / DEV_BSIZE;
    265       1.33   reinoud 	inc  = MAX(1, 2048 / sector_size);
    266       1.32   reinoud 
    267       1.32   reinoud 	/* by definition */
    268       1.32   reinoud 	vrs = ((32*1024 + sector_size - 1) / sector_size)
    269       1.32   reinoud 	        + first_sector;
    270       1.32   reinoud 
    271       1.32   reinoud 	/* read first vrs sector */
    272       1.34   reinoud 	if (read_sector(a, vrs * blks, 1))
    273       1.32   reinoud 		return;
    274       1.32   reinoud 
    275       1.32   reinoud 	/* skip all CD001 records */
    276       1.32   reinoud 	vrsd = a->bp->b_data;
    277       1.33   reinoud 	/* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
    278       1.32   reinoud 	while (memcmp(vrsd->identifier, "CD001", 5) == 0) {
    279       1.32   reinoud 		/* for sure */
    280       1.32   reinoud 		*is_iso9660 = first_sector;
    281       1.32   reinoud 
    282       1.33   reinoud 		vrs += inc;
    283       1.34   reinoud 		if (read_sector(a, vrs * blks, 1))
    284       1.32   reinoud 			return;
    285       1.32   reinoud 	}
    286       1.32   reinoud 
    287       1.32   reinoud 	/* search for BEA01 */
    288       1.32   reinoud 	vrsd = a->bp->b_data;
    289       1.32   reinoud 	/* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
    290       1.32   reinoud 	if (memcmp(vrsd->identifier, "BEA01", 5))
    291       1.32   reinoud 		return;
    292       1.32   reinoud 
    293       1.32   reinoud 	/* read successor */
    294       1.33   reinoud 	vrs += inc;
    295       1.34   reinoud 	if (read_sector(a, vrs * blks, 1))
    296       1.32   reinoud 		return;
    297       1.32   reinoud 
    298       1.32   reinoud 	/* check for NSR[23] */
    299       1.32   reinoud 	vrsd = a->bp->b_data;
    300       1.33   reinoud 	/* printf("vrsd->identifier = `%s`\n", vrsd->identifier); */
    301       1.32   reinoud 	if (memcmp(vrsd->identifier, "NSR0", 4))
    302       1.32   reinoud 		return;
    303       1.32   reinoud 
    304       1.32   reinoud 	*is_udf = first_sector;
    305       1.32   reinoud }
    306       1.32   reinoud 
    307       1.32   reinoud 
    308       1.32   reinoud /*
    309       1.32   reinoud  * Scan for ISO Volume Recognition Sequences
    310       1.32   reinoud  */
    311       1.32   reinoud 
    312       1.32   reinoud static int
    313       1.32   reinoud scan_iso_vrs(mbr_args_t *a)
    314       1.32   reinoud {
    315       1.32   reinoud 	struct mmc_discinfo  di;
    316       1.32   reinoud 	struct mmc_trackinfo ti;
    317       1.32   reinoud 	dev_t dev;
    318       1.32   reinoud 	uint64_t sector;
    319       1.32   reinoud 	int is_iso9660, is_udf;
    320       1.32   reinoud 	int tracknr, sessionnr;
    321       1.32   reinoud 	int new_session, error;
    322       1.32   reinoud 
    323       1.32   reinoud 	is_iso9660 = is_udf = -1;
    324       1.32   reinoud 
    325       1.32   reinoud 	/* parse all sessions of disc if we're on a SCSI MMC device */
    326       1.32   reinoud 	if (a->lp->d_flags & D_SCSI_MMC) {
    327       1.32   reinoud 		/* get disc info */
    328       1.32   reinoud 		dev = a->bp->b_dev;
    329       1.32   reinoud 		error = bdev_ioctl(dev, MMCGETDISCINFO, &di, FKIOCTL, curlwp);
    330       1.32   reinoud 		if (error)
    331       1.32   reinoud 			return SCAN_CONTINUE;
    332       1.32   reinoud 
    333       1.32   reinoud 		/* go trough all (data) tracks */
    334       1.32   reinoud 		sessionnr = -1;
    335       1.32   reinoud 		for (tracknr = di.first_track;
    336       1.32   reinoud 		    tracknr <= di.first_track_last_session; tracknr++)
    337       1.32   reinoud 		{
    338       1.32   reinoud 			ti.tracknr = tracknr;
    339       1.32   reinoud 			error = bdev_ioctl(dev, MMCGETTRACKINFO, &ti,
    340       1.32   reinoud 					FKIOCTL, curlwp);
    341       1.32   reinoud 			if (error)
    342       1.32   reinoud 				return SCAN_CONTINUE;
    343       1.32   reinoud 			new_session = (ti.sessionnr != sessionnr);
    344       1.32   reinoud 			sessionnr = ti.sessionnr;
    345       1.32   reinoud 			if (new_session) {
    346       1.32   reinoud 				if (ti.flags & MMC_TRACKINFO_BLANK)
    347       1.32   reinoud 					continue;
    348       1.32   reinoud 				if (!(ti.flags & MMC_TRACKINFO_DATA))
    349       1.32   reinoud 					continue;
    350       1.32   reinoud 				sector = ti.track_start;
    351       1.32   reinoud 				scan_iso_vrs_session(a, sector,
    352       1.32   reinoud 					&is_iso9660, &is_udf);
    353       1.32   reinoud 			}
    354       1.32   reinoud 		}
    355       1.32   reinoud 	} else {
    356       1.32   reinoud 		/* try start of disc */
    357       1.32   reinoud 		sector = 0;
    358       1.32   reinoud 		scan_iso_vrs_session(a, sector, &is_iso9660, &is_udf);
    359       1.32   reinoud 	}
    360       1.32   reinoud 
    361       1.32   reinoud 	if ((is_iso9660 < 0) && (is_udf < 0))
    362       1.32   reinoud 		return SCAN_CONTINUE;
    363       1.32   reinoud 
    364       1.35   reinoud 	strncpy(a->lp->d_typename, "iso partition", 16);
    365       1.35   reinoud 
    366       1.45   mlelstv 	/* adjust session information for iso9660 partition */
    367       1.32   reinoud 	if (is_iso9660 >= 0) {
    368       1.32   reinoud 		/* set 'a' partition to iso9660 */
    369       1.32   reinoud 		a->lp->d_partitions[0].p_offset = 0;
    370       1.32   reinoud 		a->lp->d_partitions[0].p_size   = a->lp->d_secperunit;
    371       1.32   reinoud 		a->lp->d_partitions[0].p_cdsession = is_iso9660;
    372       1.32   reinoud 		a->lp->d_partitions[0].p_fstype = FS_ISO9660;
    373       1.32   reinoud 	}
    374       1.32   reinoud 
    375       1.45   mlelstv 	/* UDF doesn't care about the cd session specified here */
    376       1.32   reinoud 
    377       1.32   reinoud 	return SCAN_FOUND;
    378       1.32   reinoud }
    379       1.32   reinoud 
    380       1.32   reinoud 
    381        1.1       dsl /*
    382        1.1       dsl  * Attempt to read a disk label from a device
    383        1.1       dsl  * using the indicated strategy routine.
    384        1.1       dsl  * The label must be partly set up before this:
    385        1.1       dsl  * secpercyl, secsize and anything required for a block i/o read
    386        1.1       dsl  * operation in the driver's strategy/start routines
    387        1.1       dsl  * must be filled in before calling us.
    388        1.1       dsl  *
    389        1.1       dsl  * If dos partition table requested, attempt to load it and
    390        1.1       dsl  * find disklabel inside a DOS partition. Also, if bad block
    391        1.1       dsl  * table needed, attempt to extract it as well. Return buffer
    392        1.1       dsl  * for use in signalling errors if requested.
    393        1.1       dsl  *
    394        1.1       dsl  * Returns null on success and an error string on failure.
    395        1.1       dsl  */
    396        1.1       dsl const char *
    397       1.13       dsl readdisklabel(dev_t dev, void (*strat)(struct buf *), struct disklabel *lp,
    398       1.13       dsl     struct cpu_disklabel *osdep)
    399        1.1       dsl {
    400        1.1       dsl 	int rval;
    401        1.1       dsl 	int i;
    402        1.1       dsl 	mbr_args_t a;
    403        1.1       dsl 
    404        1.3       dsl 	memset(&a, 0, sizeof a);
    405        1.1       dsl 	a.lp = lp;
    406        1.1       dsl 	a.strat = strat;
    407       1.17       dsl 	a.action = READ_LABEL;
    408        1.1       dsl 
    409        1.1       dsl 	/* minimal requirements for architypal disk label */
    410        1.1       dsl 	if (lp->d_secsize == 0)
    411        1.1       dsl 		lp->d_secsize = DEV_BSIZE;
    412        1.1       dsl 	if (lp->d_secperunit == 0)
    413        1.1       dsl 		lp->d_secperunit = 0x1fffffff;
    414       1.28    dyoung 	a.secperunit = lp->d_secperunit;
    415        1.1       dsl 	lp->d_npartitions = RAW_PART + 1;
    416        1.1       dsl 	for (i = 0; i < RAW_PART; i++) {
    417        1.1       dsl 		lp->d_partitions[i].p_size = 0;
    418        1.1       dsl 		lp->d_partitions[i].p_offset = 0;
    419        1.1       dsl 	}
    420        1.1       dsl 	if (lp->d_partitions[RAW_PART].p_size == 0)
    421       1.17       dsl 		lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
    422        1.1       dsl 	lp->d_partitions[RAW_PART].p_offset = 0;
    423        1.1       dsl 
    424        1.1       dsl 	/*
    425        1.1       dsl 	 * Set partition 'a' to be the whole disk.
    426        1.1       dsl 	 * Cleared if we find an mbr or a netbsd label.
    427        1.1       dsl 	 */
    428        1.1       dsl 	lp->d_partitions[0].p_size = lp->d_partitions[RAW_PART].p_size;
    429        1.1       dsl 	lp->d_partitions[0].p_fstype = FS_BSDFFS;
    430        1.1       dsl 
    431       1.32   reinoud 	/*
    432       1.34   reinoud 	 * Get a buffer big enough to read a disklabel in and initialize it
    433       1.38     pooka 	 * make it three sectors long for the validate_label(); see comment at
    434       1.34   reinoud 	 * start of file.
    435       1.32   reinoud 	 */
    436       1.38     pooka 	a.bp = geteblk(SCANBLOCKS * (int)lp->d_secsize);
    437        1.1       dsl 	a.bp->b_dev = dev;
    438        1.1       dsl 
    439        1.1       dsl 	if (osdep)
    440        1.1       dsl 		/*
    441        1.1       dsl 		 * Scan mbr searching for netbsd partition and saving
    442        1.1       dsl 		 * bios partition information to use if the netbsd one
    443        1.1       dsl 		 * is absent.
    444        1.1       dsl 		 */
    445        1.1       dsl 		rval = scan_mbr(&a, look_netbsd_part);
    446        1.1       dsl 	else
    447        1.1       dsl 		rval = SCAN_CONTINUE;
    448        1.1       dsl 
    449        1.3       dsl 	if (rval == SCAN_CONTINUE) {
    450        1.1       dsl 		/* Look at start of disk */
    451       1.17       dsl 		rval = validate_label(&a, 0);
    452        1.1       dsl 	}
    453        1.1       dsl 
    454       1.32   reinoud 	if (rval == SCAN_CONTINUE) {
    455       1.32   reinoud 		rval = scan_iso_vrs(&a);
    456       1.32   reinoud 	}
    457        1.3       dsl #if 0
    458        1.3       dsl 	/*
    459        1.3       dsl 	 * Save sector where we found the label for the 'don't overwrite
    460        1.3       dsl 	 * the label' check in bounds_check_with_label.
    461        1.3       dsl 	 */
    462        1.3       dsl 	if (rval == SCAN_FOUND)
    463        1.3       dsl 		xxx->label_sector = a.label_sector;
    464        1.3       dsl #endif
    465        1.3       dsl 
    466        1.1       dsl 	/* Obtain bad sector table if requested and present */
    467       1.37     pooka #ifdef __HAVE_DISKLABEL_DKBAD
    468        1.1       dsl 	if (rval == SCAN_FOUND && osdep && (lp->d_flags & D_BADSECT)) {
    469       1.37     pooka 		struct dkbad *bdp, *db;
    470        1.1       dsl 		int blkno;
    471        1.1       dsl 
    472        1.1       dsl 		bdp = &osdep->bad;
    473        1.1       dsl 		i = 0;
    474        1.1       dsl 		rval = SCAN_ERROR;
    475        1.1       dsl 		do {
    476        1.1       dsl 			/* read a bad sector table */
    477        1.1       dsl 			blkno = lp->d_secperunit - lp->d_nsectors + i;
    478        1.1       dsl 			if (lp->d_secsize > DEV_BSIZE)
    479        1.1       dsl 				blkno *= lp->d_secsize / DEV_BSIZE;
    480        1.1       dsl 			else
    481        1.1       dsl 				blkno /= DEV_BSIZE / lp->d_secsize;
    482        1.1       dsl 			/* if successful, validate, otherwise try another */
    483       1.17       dsl 			if (read_sector(&a, blkno, 1)) {
    484        1.1       dsl 				a.msg = "bad sector table I/O error";
    485        1.1       dsl 				continue;
    486        1.1       dsl 			}
    487        1.1       dsl 			db = (struct dkbad *)(a.bp->b_data);
    488        1.1       dsl #define DKBAD_MAGIC 0x4321
    489        1.1       dsl 			if (db->bt_mbz != 0 || db->bt_flag != DKBAD_MAGIC) {
    490        1.1       dsl 				a.msg = "bad sector table corrupted";
    491        1.1       dsl 				continue;
    492        1.1       dsl 			}
    493        1.1       dsl 			rval = SCAN_FOUND;
    494        1.1       dsl 			*bdp = *db;
    495        1.1       dsl 			break;
    496       1.29        ad 		} while (a.bp->b_error && (i += 2) < 10 &&
    497        1.1       dsl 			i < lp->d_nsectors);
    498        1.1       dsl 	}
    499       1.37     pooka #endif /* __HAVE_DISKLABEL_DKBAD */
    500        1.1       dsl 
    501       1.30        ad 	brelse(a.bp, 0);
    502       1.17       dsl 	if (rval == SCAN_ERROR || rval == SCAN_CONTINUE)
    503       1.10      cube 		return a.msg;
    504        1.8   reinoud 	return NULL;
    505        1.1       dsl }
    506        1.1       dsl 
    507        1.1       dsl static int
    508       1.13       dsl look_netbsd_part(mbr_args_t *a, mbr_partition_t *dp, int slot, uint ext_base)
    509        1.1       dsl {
    510        1.1       dsl 	struct partition *pp;
    511        1.1       dsl 	int ptn_base = ext_base + le32toh(dp->mbrp_start);
    512        1.1       dsl 	int rval;
    513        1.1       dsl 
    514        1.1       dsl 	if (
    515        1.1       dsl #ifdef COMPAT_386BSD_MBRPART
    516        1.4     lukem 	    dp->mbrp_type == MBR_PTYPE_386BSD ||
    517        1.1       dsl #endif
    518        1.4     lukem 	    dp->mbrp_type == MBR_PTYPE_NETBSD) {
    519       1.17       dsl 		rval = validate_label(a, ptn_base);
    520        1.1       dsl 
    521       1.12       dsl #if RAW_PART == 3
    522        1.1       dsl 		/* Put actual location where we found the label into ptn 2 */
    523       1.12       dsl 		if (rval == SCAN_FOUND || a->lp->d_partitions[2].p_size == 0) {
    524        1.1       dsl 			a->lp->d_partitions[2].p_size = le32toh(dp->mbrp_size);
    525        1.1       dsl 			a->lp->d_partitions[2].p_offset = ptn_base;
    526        1.1       dsl 		}
    527       1.12       dsl #endif
    528        1.1       dsl 
    529        1.1       dsl 		/* If we got a netbsd label look no further */
    530        1.1       dsl 		if (rval == SCAN_FOUND)
    531        1.1       dsl 			return rval;
    532        1.1       dsl 	}
    533        1.1       dsl 
    534        1.1       dsl 	/* Install main partitions into e..h and extended into i+ */
    535        1.1       dsl 	if (ext_base == 0)
    536        1.1       dsl 		slot += 4;
    537        1.1       dsl 	else {
    538        1.4     lukem 		slot = 4 + MBR_PART_COUNT;
    539        1.1       dsl 		pp = &a->lp->d_partitions[slot];
    540        1.1       dsl 		for (; slot < MAXPARTITIONS; pp++, slot++) {
    541        1.1       dsl 			/* This gets called twice - avoid duplicates */
    542        1.1       dsl 			if (pp->p_offset == ptn_base &&
    543        1.1       dsl 			    pp->p_size == le32toh(dp->mbrp_size))
    544        1.1       dsl 				break;
    545        1.1       dsl 			if (pp->p_size == 0)
    546        1.1       dsl 				break;
    547        1.1       dsl 		}
    548        1.1       dsl 	}
    549        1.1       dsl 
    550        1.1       dsl 	if (slot < MAXPARTITIONS) {
    551        1.1       dsl 		/* Stop 'a' being the entire disk */
    552        1.1       dsl 		a->lp->d_partitions[0].p_size = 0;
    553        1.1       dsl 		a->lp->d_partitions[0].p_fstype = 0;
    554        1.1       dsl 
    555        1.1       dsl 		/* save partition info */
    556        1.1       dsl 		pp = &a->lp->d_partitions[slot];
    557        1.1       dsl 		pp->p_offset = ptn_base;
    558        1.1       dsl 		pp->p_size = le32toh(dp->mbrp_size);
    559        1.4     lukem 		pp->p_fstype = xlat_mbr_fstype(dp->mbrp_type);
    560        1.1       dsl 
    561        1.1       dsl 		if (slot >= a->lp->d_npartitions)
    562        1.1       dsl 			a->lp->d_npartitions = slot + 1;
    563        1.1       dsl 	}
    564        1.1       dsl 
    565        1.1       dsl 	return SCAN_CONTINUE;
    566        1.1       dsl }
    567        1.1       dsl 
    568       1.55     kamil __noubsan
    569       1.51  christos static bool
    570       1.51  christos check_label_magic(const struct disklabel *dlp, uint32_t diskmagic)
    571       1.51  christos {
    572       1.54     kamil 	return memcmp(&dlp->d_magic, &diskmagic, sizeof(diskmagic)) == 0 &&
    573       1.54     kamil 	    memcmp(&dlp->d_magic2, &diskmagic, sizeof(diskmagic)) == 0;
    574       1.51  christos }
    575        1.1       dsl 
    576       1.47       rin #ifdef DISKLABEL_EI
    577       1.47       rin /*
    578       1.47       rin  * - For read, convert a label to the native byte order.
    579       1.47       rin  * - For update or write, if a label already exists, keep its byte order.
    580       1.47       rin  *   Otherwise, write a new label in the native byte order.
    581       1.47       rin  */
    582       1.47       rin #endif
    583        1.1       dsl static int
    584       1.17       dsl validate_label(mbr_args_t *a, uint label_sector)
    585        1.1       dsl {
    586       1.56     kamil 	struct disklabel *dlp;
    587       1.17       dsl 	char *dlp_lim, *dlp_byte;
    588        1.1       dsl 	int error;
    589       1.47       rin #ifdef DISKLABEL_EI
    590       1.47       rin 	int swapped = 0;
    591       1.47       rin 	uint16_t npartitions;
    592       1.47       rin #endif
    593        1.1       dsl 
    594        1.1       dsl 	/* Next, dig out disk label */
    595       1.38     pooka 	if (read_sector(a, label_sector, SCANBLOCKS)) {
    596        1.3       dsl 		a->msg = "disk label read failed";
    597        1.1       dsl 		return SCAN_ERROR;
    598        1.1       dsl 	}
    599        1.1       dsl 
    600        1.1       dsl 	/* Locate disk label within block and validate */
    601        1.1       dsl 	/*
    602        1.1       dsl 	 * XXX (dsl) This search may be a waste of time, a lot of other i386
    603        1.1       dsl 	 * code assumes the label is at offset LABELOFFSET (=0) in the sector.
    604        1.1       dsl 	 *
    605        1.3       dsl 	 * If we want to support disks from other netbsd ports, then the
    606        1.1       dsl 	 * code should also allow for a shorter label nearer the end of
    607        1.1       dsl 	 * the disk sector, and (IIRC) labels within 8k of the disk start.
    608        1.1       dsl 	 */
    609        1.3       dsl 	dlp = (void *)a->bp->b_data;
    610       1.26  christos 	dlp_lim = (char *)a->bp->b_data + a->bp->b_bcount - sizeof *dlp;
    611       1.48  christos 	for (;; dlp = (void *)((char *)dlp + sizeof(uint32_t))) {
    612       1.17       dsl 		if ((char *)dlp > dlp_lim) {
    613       1.17       dsl 			if (a->action != WRITE_LABEL)
    614        1.3       dsl 				return SCAN_CONTINUE;
    615       1.42       wiz 			/* Write at arch. dependent default location */
    616       1.26  christos 			dlp_byte = (char *)a->bp->b_data + LABELOFFSET;
    617       1.17       dsl 			if (label_sector)
    618       1.17       dsl 				dlp_byte += MBR_LABELSECTOR * a->lp->d_secsize;
    619       1.17       dsl 			else
    620       1.17       dsl 				dlp_byte += LABELSECTOR * a->lp->d_secsize;
    621       1.17       dsl 			dlp = (void *)dlp_byte;
    622        1.1       dsl 			break;
    623        1.1       dsl 		}
    624       1.56     kamil 		if (!check_label_magic(dlp, DISKMAGIC))
    625       1.47       rin #ifdef DISKLABEL_EI
    626       1.47       rin 		{
    627       1.56     kamil 			if (!check_label_magic(dlp, bswap32(DISKMAGIC)))
    628       1.47       rin 				continue;
    629       1.47       rin 
    630       1.47       rin 			/*
    631       1.47       rin 			 * The label is in the other byte order. We need to
    632       1.47       rin 			 * checksum before swapping the byte order.
    633       1.47       rin 			 */
    634       1.56     kamil 			npartitions = bswap16(dlp->d_npartitions);
    635       1.47       rin 			if (npartitions > MAXPARTITIONS ||
    636       1.56     kamil 			    dkcksum_sized(dlp, npartitions) != 0)
    637       1.47       rin 				goto corrupted;
    638       1.47       rin 
    639       1.47       rin 			swapped = 1;
    640       1.47       rin 		}
    641       1.47       rin #else
    642       1.17       dsl 			continue;
    643       1.47       rin #endif
    644       1.56     kamil 		else if (dlp->d_npartitions > MAXPARTITIONS ||
    645       1.56     kamil 			 dkcksum(dlp) != 0) {
    646       1.47       rin #ifdef DISKLABEL_EI
    647       1.47       rin corrupted:
    648       1.47       rin #endif
    649       1.17       dsl 			a->msg = "disk label corrupted";
    650       1.17       dsl 			continue;
    651       1.17       dsl 		}
    652       1.17       dsl 		break;
    653        1.3       dsl 	}
    654        1.3       dsl 
    655       1.17       dsl 	switch (a->action) {
    656        1.3       dsl 	case READ_LABEL:
    657       1.47       rin #ifdef DISKLABEL_EI
    658       1.47       rin 		if (swapped)
    659  1.56.14.1      cjep 			disklabel_swap(a->lp, dlp);
    660       1.47       rin 		else
    661       1.56     kamil 			*a->lp = *dlp;
    662       1.47       rin #else
    663       1.56     kamil 		*a->lp = *dlp;
    664       1.47       rin #endif
    665       1.28    dyoung 		if ((a->msg = convertdisklabel(a->lp, a->strat, a->bp,
    666       1.28    dyoung 		                              a->secperunit)) != NULL)
    667       1.28    dyoung 			return SCAN_ERROR;
    668        1.3       dsl 		a->label_sector = label_sector;
    669        1.1       dsl 		return SCAN_FOUND;
    670        1.3       dsl 	case UPDATE_LABEL:
    671        1.3       dsl 	case WRITE_LABEL:
    672       1.47       rin #ifdef DISKLABEL_EI
    673       1.47       rin 		/* DO NOT swap a->lp itself for later references. */
    674       1.47       rin 		if (swapped)
    675  1.56.14.1      cjep 			disklabel_swap(dlp, a->lp);
    676       1.47       rin 		else
    677       1.56     kamil 			*dlp = *a->lp;
    678       1.47       rin #else
    679       1.56     kamil 		*dlp = *a->lp;
    680       1.47       rin #endif
    681       1.31        ad 		a->bp->b_oflags &= ~BO_DONE;
    682       1.31        ad 		a->bp->b_flags &= ~B_READ;
    683        1.3       dsl 		a->bp->b_flags |= B_WRITE;
    684        1.3       dsl 		(*a->strat)(a->bp);
    685        1.3       dsl 		error = biowait(a->bp);
    686        1.3       dsl 		if (error != 0) {
    687        1.3       dsl 			a->error = error;
    688        1.3       dsl 			a->msg = "disk label write failed";
    689        1.3       dsl 			return SCAN_ERROR;
    690        1.3       dsl 		}
    691        1.3       dsl 		a->written++;
    692        1.3       dsl 		/* Write label to all mbr partitions */
    693        1.3       dsl 		return SCAN_CONTINUE;
    694        1.3       dsl 	default:
    695        1.3       dsl 		return SCAN_ERROR;
    696        1.1       dsl 	}
    697        1.1       dsl }
    698        1.1       dsl 
    699        1.1       dsl /*
    700        1.1       dsl  * Write disk label back to device after modification.
    701        1.1       dsl  */
    702        1.1       dsl int
    703       1.13       dsl writedisklabel(dev_t dev, void (*strat)(struct buf *), struct disklabel *lp,
    704       1.13       dsl     struct cpu_disklabel *osdep)
    705        1.1       dsl {
    706        1.1       dsl 	mbr_args_t a;
    707        1.1       dsl 
    708        1.3       dsl 	memset(&a, 0, sizeof a);
    709        1.1       dsl 	a.lp = lp;
    710        1.1       dsl 	a.strat = strat;
    711        1.1       dsl 
    712        1.1       dsl 	/* get a buffer and initialize it */
    713       1.38     pooka 	a.bp = geteblk(SCANBLOCKS * (int)lp->d_secsize);
    714        1.1       dsl 	a.bp->b_dev = dev;
    715        1.1       dsl 
    716       1.17       dsl 	/* osdep => we expect an mbr with label in netbsd ptn */
    717       1.17       dsl 	a.action = osdep != NULL ? WRITE_LABEL : UPDATE_LABEL;
    718       1.17       dsl 
    719       1.17       dsl 	/* Write/update the label to every netbsd mbr partition */
    720       1.17       dsl 	scan_mbr(&a, write_netbsd_label);
    721        1.3       dsl 
    722       1.17       dsl 	/* Old write the label at the start of the volume on disks that
    723       1.17       dsl 	 * don't have a valid mbr (always update an existing one) */
    724       1.17       dsl 	a.action = a.found_mbr ? UPDATE_LABEL : WRITE_LABEL;
    725       1.17       dsl 	validate_label(&a, 0);
    726        1.1       dsl 
    727        1.3       dsl 	if (a.written == 0 && a.error == 0)
    728        1.1       dsl 		a.error = ESRCH;
    729        1.1       dsl 
    730       1.30        ad 	brelse(a.bp, 0);
    731        1.1       dsl 	return a.error;
    732        1.1       dsl }
    733        1.1       dsl 
    734        1.1       dsl static int
    735       1.21      yamt write_netbsd_label(mbr_args_t *a, mbr_partition_t *dp, int slot, uint ext_base)
    736        1.1       dsl {
    737        1.1       dsl 	int ptn_base = ext_base + le32toh(dp->mbrp_start);
    738        1.1       dsl 
    739        1.4     lukem 	if (dp->mbrp_type != MBR_PTYPE_NETBSD)
    740        1.1       dsl 		return SCAN_CONTINUE;
    741        1.1       dsl 
    742       1.17       dsl 	return validate_label(a, ptn_base);
    743        1.1       dsl }
    744