Home | History | Annotate | Line # | Download | only in dkwedge
dkwedge_bsdlabel.c revision 1.3
      1 /*	$NetBSD: dkwedge_bsdlabel.c,v 1.3 2005/02/27 00:26:59 perry Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 2004 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Jason R. Thorpe.
      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 by the NetBSD
     21  *	Foundation, Inc. and its contributors.
     22  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  *    contributors may be used to endorse or promote products derived
     24  *    from this software without specific prior written permission.
     25  *
     26  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  * BE 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 THE
     36  * POSSIBILITY OF SUCH DAMAGE.
     37  */
     38 
     39 /*
     40  * Adapted from kern/subr_disk_mbr.c:
     41  *
     42  * Copyright (c) 1982, 1986, 1988 Regents of the University of California.
     43  * All rights reserved.
     44  *
     45  * Redistribution and use in source and binary forms, with or without
     46  * modification, are permitted provided that the following conditions
     47  * are met:
     48  * 1. Redistributions of source code must retain the above copyright
     49  *    notice, this list of conditions and the following disclaimer.
     50  * 2. Redistributions in binary form must reproduce the above copyright
     51  *    notice, this list of conditions and the following disclaimer in the
     52  *    documentation and/or other materials provided with the distribution.
     53  * 3. Neither the name of the University nor the names of its contributors
     54  *    may be used to endorse or promote products derived from this software
     55  *    without specific prior written permission.
     56  *
     57  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     58  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     59  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     60  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     61  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     62  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     63  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     64  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     65  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     66  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     67  * SUCH DAMAGE.
     68  *
     69  *	@(#)ufs_disksubr.c      7.16 (Berkeley) 5/4/91
     70  */
     71 
     72 /*
     73  * 4.4BSD disklabel support for disk wedges
     74  *
     75  * Here is the basic search algorithm in use here:
     76  *
     77  * For historical reasons, we scan for x86-style MBR partitions looking
     78  * for a MBR_PTYPE_NETBSD (or MBR_PTYPE_386BSD) partition.  The first
     79  * 4.4BSD disklabel found in the 2nd sector of such a partition is used.
     80  * We assume that the 4.4BSD disklabel describes all partitions on the
     81  * disk; we do not use any partition information from the MBR partition
     82  * table.
     83  *
     84  * If that fails, then we fall back on a table of known locations for
     85  * various platforms.
     86  */
     87 
     88 #include <sys/cdefs.h>
     89 __KERNEL_RCSID(0, "$NetBSD: dkwedge_bsdlabel.c,v 1.3 2005/02/27 00:26:59 perry Exp $");
     90 
     91 #include <sys/param.h>
     92 #include <sys/systm.h>
     93 #include <sys/proc.h>
     94 #include <sys/errno.h>
     95 #include <sys/disk.h>
     96 #include <sys/vnode.h>
     97 #include <sys/malloc.h>
     98 
     99 #include <sys/bootblock.h>
    100 #include <sys/disklabel.h>
    101 
    102 #define	BSD44_MBR_LABELSECTOR	1
    103 
    104 #define	DISKLABEL_SIZE(x)						\
    105 	(offsetof(struct disklabel, d_partitions) +			\
    106 	 (sizeof(struct partition) * (x)))
    107 
    108 /*
    109  * Note the smallest MAXPARTITIONS was 8, so we allow a disklabel
    110  * that size to be locted at the end of the sector.
    111  */
    112 #define	DISKLABEL_MINSIZE	DISKLABEL_SIZE(8)
    113 
    114 /*
    115  * Table of known platform-specific disklabel locations.
    116  */
    117 static const struct disklabel_location {
    118 	daddr_t		label_sector;	/* sector containing label */
    119 	size_t		label_offset;	/* byte offset of label in sector */
    120 } disklabel_locations[] = {
    121 	{ 0,	0 },	/* mvme68k, next68k */
    122 	{ 0,	64 },	/* algor, alpha, amiga, amigappc, evbmips, evbppc,
    123 			   luna68k, mac68k, macppc, news68k, newsmips,
    124 			   pc532, pdp11, pmax, vax, x68k */
    125 	{ 0,	128 },	/* sparc, sun68k */
    126 	{ 1,	0 },	/* amd64, arc, arm, bebox, cobalt, evbppc, hp700,
    127 			   hpcarm, hpcmips, i386, ibmnws, mipsco, mvmeppc,
    128 			   ofppc, playstation2, pmppc, prep, sandpoint,
    129 			   sbmips, sgimips, sh3, sh5 */
    130 	/* XXX atari is weird */
    131 	{ 2,	0 },	/* cesfic, hp300 */
    132 
    133 	{ -1,	0 },
    134 };
    135 
    136 #define	SCAN_CONTINUE	0
    137 #define	SCAN_FOUND	1
    138 #define	SCAN_ERROR	2
    139 
    140 typedef struct mbr_args {
    141 	struct disk	*pdk;
    142 	struct vnode	*vp;
    143 	void		*buf;
    144 	int		error;
    145 } mbr_args_t;
    146 
    147 static const char *
    148 bsdlabel_fstype_to_str(uint8_t fstype)
    149 {
    150 	const char *str;
    151 
    152 	switch (fstype) {
    153 	case FS_UNUSED:		str = DKW_PTYPE_UNUSED;		break;
    154 	case FS_SWAP:		str = DKW_PTYPE_SWAP;		break;
    155 	case FS_BSDFFS:		str = DKW_PTYPE_FFS;		break;
    156 	case FS_MSDOS:		str = DKW_PTYPE_FAT;		break;
    157 	case FS_BSDLFS:		str = DKW_PTYPE_LFS;		break;
    158 	case FS_ISO9660:	str = DKW_PTYPE_ISO9660;	break;
    159 	case FS_ADOS:		str = DKW_PTYPE_AMIGADOS;	break;
    160 	case FS_HFS:		str = DKW_PTYPE_APPLEHFS;	break;
    161 	case FS_FILECORE:	str = DKW_PTYPE_FILECORE;	break;
    162 	case FS_EX2FS:		str = DKW_PTYPE_EXT2FS;		break;
    163 	case FS_NTFS:		str = DKW_PTYPE_NTFS;		break;
    164 	case FS_RAID:		str = DKW_PTYPE_RAIDFRAME;	break;
    165 	case FS_CCD:		str = DKW_PTYPE_CCD;		break;
    166 	case FS_APPLEUFS:	str = DKW_PTYPE_APPLEUFS;	break;
    167 	case FS_VINUM:		str = "vinum";			break;
    168 	default:		str = NULL;			break;
    169 	}
    170 
    171 	return (str);
    172 }
    173 
    174 static void
    175 swap_disklabel(struct disklabel *lp)
    176 {
    177 	int i;
    178 
    179 #define	SWAP16(x)	lp->x = bswap16(lp->x)
    180 #define	SWAP32(x)	lp->x = bswap32(lp->x)
    181 
    182 	SWAP32(d_magic);
    183 	SWAP16(d_type);
    184 	SWAP16(d_subtype);
    185 	SWAP32(d_secsize);
    186 	SWAP32(d_nsectors);
    187 	SWAP32(d_ntracks);
    188 	SWAP32(d_ncylinders);
    189 	SWAP32(d_secpercyl);
    190 	SWAP32(d_secperunit);
    191 	SWAP16(d_sparespertrack);
    192 	SWAP16(d_sparespercyl);
    193 	SWAP32(d_acylinders);
    194 	SWAP16(d_rpm);
    195 	SWAP16(d_interleave);
    196 	SWAP16(d_trackskew);
    197 	SWAP16(d_cylskew);
    198 	SWAP32(d_headswitch);
    199 	SWAP32(d_trkseek);
    200 	SWAP32(d_flags);
    201 
    202 	for (i = 0; i < NDDATA; i++)
    203 		SWAP32(d_drivedata[i]);
    204 	for (i = 0; i < NSPARE; i++)
    205 		SWAP32(d_spare[i]);
    206 
    207 	SWAP32(d_magic2);
    208 	SWAP16(d_checksum);
    209 	SWAP16(d_npartitions);
    210 	SWAP32(d_bbsize);
    211 	SWAP32(d_sbsize);
    212 
    213 	for (i = 0; i < lp->d_npartitions; i++) {
    214 		SWAP32(d_partitions[i].p_size);
    215 		SWAP32(d_partitions[i].p_offset);
    216 		SWAP32(d_partitions[i].p_fsize);
    217 		SWAP16(d_partitions[i].p_cpg);
    218 	}
    219 
    220 #undef SWAP16
    221 #undef SWAP32
    222 }
    223 
    224 static int
    225 validate_label(mbr_args_t *a, daddr_t label_sector, size_t label_offset)
    226 {
    227 	struct disklabel *lp;
    228 	caddr_t lp_lim;
    229 	int i, error;
    230 
    231 	error = dkwedge_read(a->pdk, a->vp, label_sector, a->buf, DEV_BSIZE);
    232 	if (error) {
    233 		aprint_error("%s: unable to read BSD disklabel @ %" PRId64
    234 		    ", error = %d\n", a->pdk->dk_name, label_sector, error);
    235 		a->error = error;
    236 		return (SCAN_ERROR);
    237 	}
    238 
    239 	/*
    240 	 * We ignore label_offset; this seems to have not been used
    241 	 * consistently in the old code, requiring us to do the search
    242 	 * in the sector.
    243 	 */
    244 	lp = a->buf;
    245 	lp_lim = (caddr_t)a->buf + DEV_BSIZE - DISKLABEL_MINSIZE;
    246 	for (;; lp = (void *)((caddr_t)lp + sizeof(uint32_t))) {
    247 		if ((caddr_t)lp > lp_lim)
    248 			return (SCAN_CONTINUE);
    249 		label_offset = (size_t)((caddr_t)lp - (caddr_t)a->buf);
    250 		if (lp->d_magic != DISKMAGIC || lp->d_magic2 != DISKMAGIC) {
    251 			if (lp->d_magic == bswap32(DISKMAGIC) &&
    252 			    lp->d_magic2 == bswap32(DISKMAGIC)) {
    253 				/*
    254 				 * Label is in the other byte order; validate
    255 				 * its length, then byte-swap it.
    256 				 */
    257 				if ((caddr_t)lp +
    258 				    DISKLABEL_SIZE(bswap16(lp->d_npartitions)) >
    259 				    (caddr_t)a->buf + DEV_BSIZE) {
    260 					aprint_error("%s: BSD disklabel @ "
    261 					    "%" PRId64
    262 					    "+%zd has bogus partition "
    263 					    "count (%u)\n", a->pdk->dk_name,
    264 					    label_sector, label_offset,
    265 					    bswap16(lp->d_npartitions));
    266 					continue;
    267 				}
    268 				swap_disklabel(lp);
    269 			} else
    270 				continue;
    271 		} else {
    272 			/*
    273 			 * Validate the disklabel length.
    274 			 */
    275 			if ((caddr_t)lp + DISKLABEL_SIZE(lp->d_npartitions) >
    276 			    (caddr_t)a->buf + DEV_BSIZE) {
    277 				aprint_error("%s: BSD disklabel @ %" PRId64
    278 				    "+%zd has bogus partition count (%u)\n",
    279 				    a->pdk->dk_name, label_sector,
    280 				    label_offset, lp->d_npartitions);
    281 				continue;
    282 			}
    283 		}
    284 
    285 		/*
    286 		 * Disklabel is now in the right order and we have validated
    287 		 * the partition count, checksum it as the final check.
    288 		 */
    289 		if (dkcksum(lp) != 0) {
    290 			aprint_error("%s: BSD disklabel @ %" PRId64
    291 			    "+%zd has bad checksum\n", a->pdk->dk_name,
    292 			    label_sector, label_offset);
    293 			continue;
    294 		}
    295 
    296 		/*
    297 		 * Ok, we have a valid NetBSD disklabel, add wedges for it.
    298 		 */
    299 		for (i = 0; i < lp->d_npartitions; i++) {
    300 			struct dkwedge_info dkw;
    301 			struct partition *p;
    302 			const char *ptype;
    303 
    304 			p = &lp->d_partitions[i];
    305 
    306 			if (p->p_fstype == FS_UNUSED)
    307 				continue;
    308 			if ((ptype =
    309 			     bsdlabel_fstype_to_str(p->p_fstype)) == NULL) {
    310 				/*
    311 				 * XXX Should probably just add these...
    312 				 * XXX maybe just have an empty ptype?
    313 				 */
    314 				aprint_verbose("%s: skipping partition %d, "
    315 				    "type %d\n", a->pdk->dk_name, i,
    316 				    p->p_fstype);
    317 				continue;
    318 			}
    319 			strcpy(dkw.dkw_ptype, ptype);
    320 
    321 			strcpy(dkw.dkw_parent, a->pdk->dk_name);
    322 			dkw.dkw_offset = p->p_offset;
    323 			dkw.dkw_size = p->p_size;
    324 
    325 			/*
    326 			 * These get historical disk naming style
    327 			 * wedge names.
    328 			 */
    329 			snprintf(dkw.dkw_wname, sizeof(dkw.dkw_wname),
    330 			    "%s%c", a->pdk->dk_name, 'a' + i);
    331 
    332 			error = dkwedge_add(&dkw);
    333 			if (error == EEXIST)
    334 				aprint_error("%s: wedge named '%s' already "
    335 				    "exists, manual intervention required\n",
    336 				    a->pdk->dk_name, dkw.dkw_wname);
    337 			else if (error)
    338 				aprint_error("%s: error %d adding partition "
    339 				    "%d type %d\n", a->pdk->dk_name, error,
    340 				    i, p->p_fstype);
    341 		}
    342 		return (SCAN_FOUND);
    343 	}
    344 }
    345 
    346 static int
    347 scan_mbr(mbr_args_t *a, int (*actn)(mbr_args_t *, struct mbr_partition *,
    348 				    int, u_int))
    349 {
    350 	struct mbr_partition ptns[MBR_PART_COUNT];
    351 	struct mbr_partition *dp;
    352 	struct mbr_sector *mbr;
    353 	u_int ext_base, this_ext, next_ext;
    354 	int i, rval;
    355 #ifdef COMPAT_386BSD_MBRPART
    356 	int dp_386bsd = -1;
    357 #endif
    358 
    359 	ext_base = 0;
    360 	this_ext = 0;
    361 	for (;;) {
    362 		a->error = dkwedge_read(a->pdk, a->vp, this_ext, a->buf,
    363 					DEV_BSIZE);
    364 		if (a->error) {
    365 			aprint_error("%s: unable to read MBR @ %u, "
    366 			    "error = %d\n", a->pdk->dk_name, this_ext,
    367 			    a->error);
    368 			return (SCAN_ERROR);
    369 		}
    370 
    371 		mbr = a->buf;
    372 		if (mbr->mbr_magic != htole16(MBR_MAGIC))
    373 			return (SCAN_CONTINUE);
    374 
    375 		/* Copy data out of buffer so action can use the buffer. */
    376 		memcpy(ptns, &mbr->mbr_parts, sizeof(ptns));
    377 
    378 		/* Looks for NetBSD partition. */
    379 		next_ext = 0;
    380 		dp = ptns;
    381 		for (i = 0; i < MBR_PART_COUNT; i++, dp++) {
    382 			if (dp->mbrp_type == 0)
    383 				continue;
    384 			if (MBR_IS_EXTENDED(dp->mbrp_type)) {
    385 				next_ext = le32toh(dp->mbrp_start);
    386 				continue;
    387 			}
    388 #ifdef COMPAT_386BSD_MBRPART
    389 			if (dp->mbrp_type == MBR_PTYPE_386BSD) {
    390 				/*
    391 				 * If more than one matches, take last,
    392 				 * as NetBSD install tool does.
    393 				 */
    394 				if (this_ext == 0)
    395 					dp_386bsd = i;
    396 				continue;
    397 			}
    398 #endif
    399 			rval = (*actn)(a, dp, i, this_ext);
    400 			if (rval != SCAN_CONTINUE)
    401 				return (rval);
    402 		}
    403 		if (next_ext == 0)
    404 			break;
    405 		if (ext_base == 0) {
    406 			ext_base = next_ext;
    407 			next_ext = 0;
    408 		}
    409 		next_ext += ext_base;
    410 		if (next_ext <= this_ext)
    411 			break;
    412 		this_ext = next_ext;
    413 	}
    414 #ifdef COMPAT_386BSD_MBRPART
    415 	if (this_ext == 0 && dp_386bsd != -1)
    416 		return ((*actn)(a, &ptns[dp_386bsd], dp_386bsd, 0));
    417 #endif
    418 	return (SCAN_CONTINUE);
    419 }
    420 
    421 static int
    422 look_netbsd_part(mbr_args_t *a, struct mbr_partition *dp, int slot,
    423 		 u_int ext_base)
    424 {
    425 	int ptn_base = ext_base + le32toh(dp->mbrp_start);
    426 	int rval;
    427 
    428 	if (
    429 #ifdef COMPAT_386BSD_MBRPART
    430 	    dp->mbrp_type == MBR_PTYPE_386BSD ||
    431 #endif
    432 	    dp->mbrp_type == MBR_PTYPE_NETBSD) {
    433 		rval = validate_label(a, ptn_base + BSD44_MBR_LABELSECTOR, 0);
    434 
    435 		/* If we got a NetBSD label, look no further. */
    436 		if (rval == SCAN_FOUND)
    437 			return (rval);
    438 	}
    439 
    440 	return (SCAN_CONTINUE);
    441 }
    442 
    443 static int
    444 dkwedge_discover_bsdlabel(struct disk *pdk, struct vnode *vp)
    445 {
    446 	mbr_args_t a;
    447 	const struct disklabel_location *dl;
    448 	int rval;
    449 
    450 	a.pdk = pdk;
    451 	a.vp = vp;
    452 	a.buf = malloc(DEV_BSIZE, M_DEVBUF, M_WAITOK);
    453 	a.error = 0;
    454 
    455 	/* MBR search. */
    456 	rval = scan_mbr(&a, look_netbsd_part);
    457 	if (rval != SCAN_CONTINUE) {
    458 		if (rval == SCAN_FOUND)
    459 			a.error = 0;	/* found it, wedges installed */
    460 		goto out;
    461 	}
    462 
    463 	/* Known location search. */
    464 	for (dl = disklabel_locations; dl->label_sector != -1; dl++) {
    465 		rval = validate_label(&a, dl->label_sector, dl->label_offset);
    466 		if (rval != SCAN_CONTINUE) {
    467 			if (rval == SCAN_FOUND)
    468 				a.error = 0;	/* found it, wedges installed */
    469 			goto out;
    470 		}
    471 	}
    472 
    473 	/* No NetBSD disklabel found. */
    474 	a.error = ESRCH;
    475  out:
    476 	free(a.buf, M_DEVBUF);
    477 	return (a.error);
    478 }
    479 
    480 DKWEDGE_DISCOVERY_METHOD_DECL(BSD44, 5, dkwedge_discover_bsdlabel);
    481