Home | History | Annotate | Line # | Download | only in dev
dksubr.c revision 1.54.2.9
      1 /* $NetBSD: dksubr.c,v 1.54.2.9 2017/02/05 13:40:26 skrll Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 1996, 1997, 1998, 1999, 2002, 2008 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 and Roland C. Dowdeswell.
      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  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 __KERNEL_RCSID(0, "$NetBSD: dksubr.c,v 1.54.2.9 2017/02/05 13:40:26 skrll Exp $");
     34 
     35 #include <sys/param.h>
     36 #include <sys/systm.h>
     37 #include <sys/stat.h>
     38 #include <sys/proc.h>
     39 #include <sys/ioctl.h>
     40 #include <sys/device.h>
     41 #include <sys/disk.h>
     42 #include <sys/disklabel.h>
     43 #include <sys/buf.h>
     44 #include <sys/bufq.h>
     45 #include <sys/vnode.h>
     46 #include <sys/fcntl.h>
     47 #include <sys/namei.h>
     48 #include <sys/module.h>
     49 #include <sys/syslog.h>
     50 
     51 #include <dev/dkvar.h>
     52 #include <miscfs/specfs/specdev.h> /* for v_rdev */
     53 
     54 int	dkdebug = 0;
     55 
     56 #ifdef DEBUG
     57 #define DKDB_FOLLOW	0x1
     58 #define DKDB_INIT	0x2
     59 #define DKDB_VNODE	0x4
     60 #define DKDB_DUMP	0x8
     61 
     62 #define IFDEBUG(x,y)		if (dkdebug & (x)) y
     63 #define DPRINTF(x,y)		IFDEBUG(x, printf y)
     64 #define DPRINTF_FOLLOW(y)	DPRINTF(DKDB_FOLLOW, y)
     65 #else
     66 #define IFDEBUG(x,y)
     67 #define DPRINTF(x,y)
     68 #define DPRINTF_FOLLOW(y)
     69 #endif
     70 
     71 #define DKF_READYFORDUMP	(DKF_INITED|DKF_TAKEDUMP)
     72 
     73 static int dk_subr_modcmd(modcmd_t, void *);
     74 
     75 #define DKLABELDEV(dev)	\
     76 	(MAKEDISKDEV(major((dev)), DISKUNIT((dev)), RAW_PART))
     77 
     78 static void	dk_makedisklabel(struct dk_softc *);
     79 static int	dk_translate(struct dk_softc *, struct buf *);
     80 static void	dk_done1(struct dk_softc *, struct buf *, bool);
     81 
     82 void
     83 dk_init(struct dk_softc *dksc, device_t dev, int dtype)
     84 {
     85 
     86 	memset(dksc, 0x0, sizeof(*dksc));
     87 	dksc->sc_dtype = dtype;
     88 	dksc->sc_dev = dev;
     89 
     90 	strlcpy(dksc->sc_xname, device_xname(dev), DK_XNAME_SIZE);
     91 	dksc->sc_dkdev.dk_name = dksc->sc_xname;
     92 }
     93 
     94 void
     95 dk_attach(struct dk_softc *dksc)
     96 {
     97 	KASSERT(dksc->sc_dev != NULL);
     98 
     99 	mutex_init(&dksc->sc_iolock, MUTEX_DEFAULT, IPL_VM);
    100 	dksc->sc_flags |= DKF_READYFORDUMP;
    101 #ifdef DIAGNOSTIC
    102 	dksc->sc_flags |= DKF_WARNLABEL | DKF_LABELSANITY;
    103 #endif
    104 
    105 	/* Attach the device into the rnd source list. */
    106 	rnd_attach_source(&dksc->sc_rnd_source, dksc->sc_xname,
    107 	    RND_TYPE_DISK, RND_FLAG_DEFAULT);
    108 }
    109 
    110 void
    111 dk_detach(struct dk_softc *dksc)
    112 {
    113 	/* Unhook the entropy source. */
    114 	rnd_detach_source(&dksc->sc_rnd_source);
    115 
    116 	dksc->sc_flags &= ~DKF_READYFORDUMP;
    117 	mutex_destroy(&dksc->sc_iolock);
    118 }
    119 
    120 /* ARGSUSED */
    121 int
    122 dk_open(struct dk_softc *dksc, dev_t dev,
    123     int flags, int fmt, struct lwp *l)
    124 {
    125 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    126 	struct	disklabel *lp = dksc->sc_dkdev.dk_label;
    127 	int	part = DISKPART(dev);
    128 	int	pmask = 1 << part;
    129 	int	ret = 0;
    130 	struct disk *dk = &dksc->sc_dkdev;
    131 
    132 	DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%x)\n", __func__,
    133 	    dksc->sc_xname, dksc, dev, flags));
    134 
    135 	mutex_enter(&dk->dk_openlock);
    136 
    137 	/*
    138 	 * If there are wedges, and this is not RAW_PART, then we
    139 	 * need to fail.
    140 	 */
    141 	if (dk->dk_nwedges != 0 && part != RAW_PART) {
    142 		ret = EBUSY;
    143 		goto done;
    144 	}
    145 
    146 	/*
    147 	 * initialize driver for the first opener
    148 	 */
    149 	if (dk->dk_openmask == 0 && dkd->d_firstopen != NULL) {
    150 		ret = (*dkd->d_firstopen)(dksc->sc_dev, dev, flags, fmt);
    151 		if (ret)
    152 			goto done;
    153 	}
    154 
    155 	/*
    156 	 * If we're init'ed and there are no other open partitions then
    157 	 * update the in-core disklabel.
    158 	 */
    159 	if ((dksc->sc_flags & DKF_INITED)) {
    160 		if ((dksc->sc_flags & DKF_VLABEL) == 0) {
    161 			dksc->sc_flags |= DKF_VLABEL;
    162 			dk_getdisklabel(dksc, dev);
    163 		}
    164 	}
    165 
    166 	/* Fail if we can't find the partition. */
    167 	if (part != RAW_PART &&
    168 	    ((dksc->sc_flags & DKF_VLABEL) == 0 ||
    169 	     part >= lp->d_npartitions ||
    170 	     lp->d_partitions[part].p_fstype == FS_UNUSED)) {
    171 		ret = ENXIO;
    172 		goto done;
    173 	}
    174 
    175 	/* Mark our unit as open. */
    176 	switch (fmt) {
    177 	case S_IFCHR:
    178 		dk->dk_copenmask |= pmask;
    179 		break;
    180 	case S_IFBLK:
    181 		dk->dk_bopenmask |= pmask;
    182 		break;
    183 	}
    184 
    185 	dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
    186 
    187 done:
    188 	mutex_exit(&dk->dk_openlock);
    189 	return ret;
    190 }
    191 
    192 /* ARGSUSED */
    193 int
    194 dk_close(struct dk_softc *dksc, dev_t dev,
    195     int flags, int fmt, struct lwp *l)
    196 {
    197 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    198 	int	part = DISKPART(dev);
    199 	int	pmask = 1 << part;
    200 	struct disk *dk = &dksc->sc_dkdev;
    201 
    202 	DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%x)\n", __func__,
    203 	    dksc->sc_xname, dksc, dev, flags));
    204 
    205 	mutex_enter(&dk->dk_openlock);
    206 
    207 	switch (fmt) {
    208 	case S_IFCHR:
    209 		dk->dk_copenmask &= ~pmask;
    210 		break;
    211 	case S_IFBLK:
    212 		dk->dk_bopenmask &= ~pmask;
    213 		break;
    214 	}
    215 	dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
    216 
    217 	if (dk->dk_openmask == 0) {
    218 		if (dkd->d_lastclose != NULL)
    219 			(*dkd->d_lastclose)(dksc->sc_dev);
    220 		if ((dksc->sc_flags & DKF_KLABEL) == 0)
    221 			dksc->sc_flags &= ~DKF_VLABEL;
    222 	}
    223 
    224 	mutex_exit(&dk->dk_openlock);
    225 	return 0;
    226 }
    227 
    228 static int
    229 dk_translate(struct dk_softc *dksc, struct buf *bp)
    230 {
    231 	int	part;
    232 	int	wlabel;
    233 	daddr_t	blkno;
    234 	struct disklabel *lp;
    235 	struct disk *dk;
    236 	uint64_t numsecs;
    237 	unsigned secsize;
    238 
    239 	lp = dksc->sc_dkdev.dk_label;
    240 	dk = &dksc->sc_dkdev;
    241 
    242 	part = DISKPART(bp->b_dev);
    243 	numsecs = dk->dk_geom.dg_secperunit;
    244 	secsize = dk->dk_geom.dg_secsize;
    245 
    246 	/*
    247 	 * The transfer must be a whole number of blocks and the offset must
    248 	 * not be negative.
    249 	 */
    250 	if ((bp->b_bcount % secsize) != 0 || bp->b_blkno < 0) {
    251 		bp->b_error = EINVAL;
    252 		goto done;
    253 	}
    254 
    255 	/* If there is nothing to do, then we are done */
    256 	if (bp->b_bcount == 0)
    257 		goto done;
    258 
    259 	wlabel = dksc->sc_flags & (DKF_WLABEL|DKF_LABELLING);
    260 	if (part == RAW_PART) {
    261 		uint64_t numblocks = btodb(numsecs * secsize);
    262 		if (bounds_check_with_mediasize(bp, DEV_BSIZE, numblocks) <= 0)
    263 			goto done;
    264 	} else {
    265 		if (bounds_check_with_label(&dksc->sc_dkdev, bp, wlabel) <= 0)
    266 			goto done;
    267 	}
    268 
    269 	/*
    270 	 * Convert the block number to absolute and put it in terms
    271 	 * of the device's logical block size.
    272 	 */
    273 	if (secsize >= DEV_BSIZE)
    274 		blkno = bp->b_blkno / (secsize / DEV_BSIZE);
    275 	else
    276 		blkno = bp->b_blkno * (DEV_BSIZE / secsize);
    277 
    278 	if (part != RAW_PART)
    279 		blkno += lp->d_partitions[DISKPART(bp->b_dev)].p_offset;
    280 	bp->b_rawblkno = blkno;
    281 
    282 	return -1;
    283 
    284 done:
    285 	bp->b_resid = bp->b_bcount;
    286 	return bp->b_error;
    287 }
    288 
    289 static int
    290 dk_strategy1(struct dk_softc *dksc, struct buf *bp)
    291 {
    292 	int error;
    293 
    294 	DPRINTF_FOLLOW(("%s(%s, %p, %p)\n", __func__,
    295 	    dksc->sc_xname, dksc, bp));
    296 
    297 	if (!(dksc->sc_flags & DKF_INITED)) {
    298 		DPRINTF_FOLLOW(("%s: not inited\n", __func__));
    299 		bp->b_error = ENXIO;
    300 		bp->b_resid = bp->b_bcount;
    301 		biodone(bp);
    302 		return 1;
    303 	}
    304 
    305 	error = dk_translate(dksc, bp);
    306 	if (error >= 0) {
    307 		biodone(bp);
    308 		return 1;
    309 	}
    310 
    311 	return 0;
    312 }
    313 
    314 void
    315 dk_strategy(struct dk_softc *dksc, struct buf *bp)
    316 {
    317 	int error;
    318 
    319 	error = dk_strategy1(dksc, bp);
    320 	if (error)
    321 		return;
    322 
    323 	/*
    324 	 * Queue buffer and start unit
    325 	 */
    326 	dk_start(dksc, bp);
    327 }
    328 
    329 int
    330 dk_strategy_defer(struct dk_softc *dksc, struct buf *bp)
    331 {
    332 	int error;
    333 
    334 	error = dk_strategy1(dksc, bp);
    335 	if (error)
    336 		return error;
    337 
    338 	/*
    339 	 * Queue buffer only
    340 	 */
    341 	mutex_enter(&dksc->sc_iolock);
    342 	bufq_put(dksc->sc_bufq, bp);
    343 	mutex_exit(&dksc->sc_iolock);
    344 
    345 	return 0;
    346 }
    347 
    348 int
    349 dk_strategy_pending(struct dk_softc *dksc)
    350 {
    351 	struct buf *bp;
    352 
    353 	if (!(dksc->sc_flags & DKF_INITED)) {
    354 		DPRINTF_FOLLOW(("%s: not inited\n", __func__));
    355 		return 0;
    356 	}
    357 
    358 	mutex_enter(&dksc->sc_iolock);
    359 	bp = bufq_peek(dksc->sc_bufq);
    360 	mutex_exit(&dksc->sc_iolock);
    361 
    362 	return bp != NULL;
    363 }
    364 
    365 void
    366 dk_start(struct dk_softc *dksc, struct buf *bp)
    367 {
    368 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    369 	int error;
    370 
    371 	if (!(dksc->sc_flags & DKF_INITED)) {
    372 		DPRINTF_FOLLOW(("%s: not inited\n", __func__));
    373 		return;
    374 	}
    375 
    376 	mutex_enter(&dksc->sc_iolock);
    377 
    378 	if (bp != NULL)
    379 		bufq_put(dksc->sc_bufq, bp);
    380 
    381 	/*
    382 	 * If another thread is running the queue, increment
    383 	 * busy counter to 2 so that the queue is retried,
    384 	 * because the driver may now accept additional
    385 	 * requests.
    386 	 */
    387 	if (dksc->sc_busy < 2)
    388 		dksc->sc_busy++;
    389 	if (dksc->sc_busy > 1)
    390 		goto done;
    391 
    392 	/*
    393 	 * Peeking at the buffer queue and committing the operation
    394 	 * only after success isn't atomic.
    395 	 *
    396 	 * So when a diskstart fails, the buffer is saved
    397 	 * and tried again before the next buffer is fetched.
    398 	 * dk_drain() handles flushing of a saved buffer.
    399 	 *
    400 	 * This keeps order of I/O operations, unlike bufq_put.
    401 	 */
    402 
    403 	while (dksc->sc_busy > 0) {
    404 
    405 		bp = dksc->sc_deferred;
    406 		dksc->sc_deferred = NULL;
    407 
    408 		if (bp == NULL)
    409 			bp = bufq_get(dksc->sc_bufq);
    410 
    411 		while (bp != NULL) {
    412 
    413 			disk_busy(&dksc->sc_dkdev);
    414 			mutex_exit(&dksc->sc_iolock);
    415 			error = dkd->d_diskstart(dksc->sc_dev, bp);
    416 			mutex_enter(&dksc->sc_iolock);
    417 			if (error == EAGAIN) {
    418 				dksc->sc_deferred = bp;
    419 				disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    420 				break;
    421 			}
    422 
    423 			if (error != 0) {
    424 				bp->b_error = error;
    425 				bp->b_resid = bp->b_bcount;
    426 				dk_done1(dksc, bp, false);
    427 			}
    428 
    429 			bp = bufq_get(dksc->sc_bufq);
    430 		}
    431 
    432 		dksc->sc_busy--;
    433 	}
    434 done:
    435 	mutex_exit(&dksc->sc_iolock);
    436 }
    437 
    438 static void
    439 dk_done1(struct dk_softc *dksc, struct buf *bp, bool lock)
    440 {
    441 	struct disk *dk = &dksc->sc_dkdev;
    442 
    443 	if (bp->b_error != 0) {
    444 		struct cfdriver *cd = device_cfdriver(dksc->sc_dev);
    445 
    446 		diskerr(bp, cd->cd_name, "error", LOG_PRINTF, 0,
    447 			dk->dk_label);
    448 		printf("\n");
    449 	}
    450 
    451 	if (lock)
    452 		mutex_enter(&dksc->sc_iolock);
    453 	disk_unbusy(dk, bp->b_bcount - bp->b_resid, (bp->b_flags & B_READ));
    454 	if (lock)
    455 		mutex_exit(&dksc->sc_iolock);
    456 
    457 	rnd_add_uint32(&dksc->sc_rnd_source, bp->b_rawblkno);
    458 
    459 	biodone(bp);
    460 }
    461 
    462 void
    463 dk_done(struct dk_softc *dksc, struct buf *bp)
    464 {
    465 	dk_done1(dksc, bp, true);
    466 }
    467 
    468 void
    469 dk_drain(struct dk_softc *dksc)
    470 {
    471 	struct buf *bp;
    472 
    473 	mutex_enter(&dksc->sc_iolock);
    474 	bp = dksc->sc_deferred;
    475 	dksc->sc_deferred = NULL;
    476 	if (bp != NULL) {
    477 		bp->b_error = EIO;
    478 		bp->b_resid = bp->b_bcount;
    479 		biodone(bp);
    480 	}
    481 	bufq_drain(dksc->sc_bufq);
    482 	mutex_exit(&dksc->sc_iolock);
    483 }
    484 
    485 int
    486 dk_discard(struct dk_softc *dksc, dev_t dev, off_t pos, off_t len)
    487 {
    488 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    489 	unsigned secsize = dksc->sc_dkdev.dk_geom.dg_secsize;
    490 	struct buf tmp, *bp = &tmp;
    491 	int error;
    492 
    493 	DPRINTF_FOLLOW(("%s(%s, %p, 0x"PRIx64", %jd, %jd)\n", __func__,
    494 	    dksc->sc_xname, dksc, (intmax_t)pos, (intmax_t)len));
    495 
    496 	if (!(dksc->sc_flags & DKF_INITED)) {
    497 		DPRINTF_FOLLOW(("%s: not inited\n", __func__));
    498 		return ENXIO;
    499 	}
    500 
    501 	if (secsize == 0 || (pos % secsize) != 0)
    502 		return EINVAL;
    503 
    504 	/* enough data to please the bounds checking code */
    505 	bp->b_dev = dev;
    506 	bp->b_blkno = (daddr_t)(pos / secsize);
    507 	bp->b_bcount = len;
    508 	bp->b_flags = B_WRITE;
    509 
    510 	error = dk_translate(dksc, bp);
    511 	if (error >= 0)
    512 		return error;
    513 
    514 	error = dkd->d_discard(dksc->sc_dev,
    515 		(off_t)bp->b_rawblkno * secsize,
    516 		(off_t)bp->b_bcount);
    517 
    518 	return error;
    519 }
    520 
    521 int
    522 dk_size(struct dk_softc *dksc, dev_t dev)
    523 {
    524 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    525 	struct	disklabel *lp;
    526 	int	is_open;
    527 	int	part;
    528 	int	size;
    529 
    530 	if ((dksc->sc_flags & DKF_INITED) == 0)
    531 		return -1;
    532 
    533 	part = DISKPART(dev);
    534 	is_open = dksc->sc_dkdev.dk_openmask & (1 << part);
    535 
    536 	if (!is_open && dkd->d_open(dev, 0, S_IFBLK, curlwp))
    537 		return -1;
    538 
    539 	lp = dksc->sc_dkdev.dk_label;
    540 	if (lp->d_partitions[part].p_fstype != FS_SWAP)
    541 		size = -1;
    542 	else
    543 		size = lp->d_partitions[part].p_size *
    544 		    (lp->d_secsize / DEV_BSIZE);
    545 
    546 	if (!is_open && dkd->d_close(dev, 0, S_IFBLK, curlwp))
    547 		return -1;
    548 
    549 	return size;
    550 }
    551 
    552 int
    553 dk_ioctl(struct dk_softc *dksc, dev_t dev,
    554 	    u_long cmd, void *data, int flag, struct lwp *l)
    555 {
    556 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    557 	struct	disklabel *lp;
    558 	struct	disk *dk = &dksc->sc_dkdev;
    559 #ifdef __HAVE_OLD_DISKLABEL
    560 	struct	disklabel newlabel;
    561 #endif
    562 	int	error;
    563 
    564 	DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%lx)\n", __func__,
    565 	    dksc->sc_xname, dksc, dev, cmd));
    566 
    567 	/* ensure that the pseudo disk is open for writes for these commands */
    568 	switch (cmd) {
    569 	case DIOCSDINFO:
    570 	case DIOCWDINFO:
    571 #ifdef __HAVE_OLD_DISKLABEL
    572 	case ODIOCSDINFO:
    573 	case ODIOCWDINFO:
    574 #endif
    575 	case DIOCKLABEL:
    576 	case DIOCWLABEL:
    577 	case DIOCAWEDGE:
    578 	case DIOCDWEDGE:
    579 	case DIOCSSTRATEGY:
    580 		if ((flag & FWRITE) == 0)
    581 			return EBADF;
    582 	}
    583 
    584 	/* ensure that the pseudo-disk is initialized for these */
    585 	switch (cmd) {
    586 	case DIOCGDINFO:
    587 	case DIOCSDINFO:
    588 	case DIOCWDINFO:
    589 	case DIOCGPARTINFO:
    590 	case DIOCKLABEL:
    591 	case DIOCWLABEL:
    592 	case DIOCGDEFLABEL:
    593 	case DIOCAWEDGE:
    594 	case DIOCDWEDGE:
    595 	case DIOCLWEDGES:
    596 	case DIOCMWEDGES:
    597 	case DIOCCACHESYNC:
    598 #ifdef __HAVE_OLD_DISKLABEL
    599 	case ODIOCGDINFO:
    600 	case ODIOCSDINFO:
    601 	case ODIOCWDINFO:
    602 	case ODIOCGDEFLABEL:
    603 #endif
    604 		if ((dksc->sc_flags & DKF_INITED) == 0)
    605 			return ENXIO;
    606 	}
    607 
    608 	error = disk_ioctl(dk, dev, cmd, data, flag, l);
    609 	if (error != EPASSTHROUGH)
    610 		return error;
    611 	else
    612 		error = 0;
    613 
    614 	switch (cmd) {
    615 	case DIOCWDINFO:
    616 	case DIOCSDINFO:
    617 #ifdef __HAVE_OLD_DISKLABEL
    618 	case ODIOCWDINFO:
    619 	case ODIOCSDINFO:
    620 #endif
    621 #ifdef __HAVE_OLD_DISKLABEL
    622 		if (cmd == ODIOCSDINFO || cmd == ODIOCWDINFO) {
    623 			memset(&newlabel, 0, sizeof newlabel);
    624 			memcpy(&newlabel, data, sizeof (struct olddisklabel));
    625 			lp = &newlabel;
    626 		} else
    627 #endif
    628 		lp = (struct disklabel *)data;
    629 
    630 		mutex_enter(&dk->dk_openlock);
    631 		dksc->sc_flags |= DKF_LABELLING;
    632 
    633 		error = setdisklabel(dksc->sc_dkdev.dk_label,
    634 		    lp, 0, dksc->sc_dkdev.dk_cpulabel);
    635 		if (error == 0) {
    636 			if (cmd == DIOCWDINFO
    637 #ifdef __HAVE_OLD_DISKLABEL
    638 			    || cmd == ODIOCWDINFO
    639 #endif
    640 			   )
    641 				error = writedisklabel(DKLABELDEV(dev),
    642 				    dkd->d_strategy, dksc->sc_dkdev.dk_label,
    643 				    dksc->sc_dkdev.dk_cpulabel);
    644 		}
    645 
    646 		dksc->sc_flags &= ~DKF_LABELLING;
    647 		mutex_exit(&dk->dk_openlock);
    648 		break;
    649 
    650 	case DIOCKLABEL:
    651 		if (*(int *)data != 0)
    652 			dksc->sc_flags |= DKF_KLABEL;
    653 		else
    654 			dksc->sc_flags &= ~DKF_KLABEL;
    655 		break;
    656 
    657 	case DIOCWLABEL:
    658 		if (*(int *)data != 0)
    659 			dksc->sc_flags |= DKF_WLABEL;
    660 		else
    661 			dksc->sc_flags &= ~DKF_WLABEL;
    662 		break;
    663 
    664 	case DIOCGDEFLABEL:
    665 		dk_getdefaultlabel(dksc, (struct disklabel *)data);
    666 		break;
    667 
    668 #ifdef __HAVE_OLD_DISKLABEL
    669 	case ODIOCGDEFLABEL:
    670 		dk_getdefaultlabel(dksc, &newlabel);
    671 		if (newlabel.d_npartitions > OLDMAXPARTITIONS)
    672 			return ENOTTY;
    673 		memcpy(data, &newlabel, sizeof (struct olddisklabel));
    674 		break;
    675 #endif
    676 
    677 	case DIOCGSTRATEGY:
    678 	    {
    679 		struct disk_strategy *dks = (void *)data;
    680 
    681 		mutex_enter(&dksc->sc_iolock);
    682 		if (dksc->sc_bufq != NULL)
    683 			strlcpy(dks->dks_name,
    684 			    bufq_getstrategyname(dksc->sc_bufq),
    685 			    sizeof(dks->dks_name));
    686 		else
    687 			error = EINVAL;
    688 		mutex_exit(&dksc->sc_iolock);
    689 		dks->dks_paramlen = 0;
    690 		break;
    691 	    }
    692 
    693 	case DIOCSSTRATEGY:
    694 	    {
    695 		struct disk_strategy *dks = (void *)data;
    696 		struct bufq_state *new;
    697 		struct bufq_state *old;
    698 
    699 		if (dks->dks_param != NULL) {
    700 			return EINVAL;
    701 		}
    702 		dks->dks_name[sizeof(dks->dks_name) - 1] = 0; /* ensure term */
    703 		error = bufq_alloc(&new, dks->dks_name,
    704 		    BUFQ_EXACT|BUFQ_SORT_RAWBLOCK);
    705 		if (error) {
    706 			return error;
    707 		}
    708 		mutex_enter(&dksc->sc_iolock);
    709 		old = dksc->sc_bufq;
    710 		if (old)
    711 			bufq_move(new, old);
    712 		dksc->sc_bufq = new;
    713 		mutex_exit(&dksc->sc_iolock);
    714 		if (old)
    715 			bufq_free(old);
    716 		break;
    717 	    }
    718 
    719 	default:
    720 		error = ENOTTY;
    721 	}
    722 
    723 	return error;
    724 }
    725 
    726 /*
    727  * dk_dump dumps all of physical memory into the partition specified.
    728  * This requires substantially more framework than {s,w}ddump, and hence
    729  * is probably much more fragile.
    730  *
    731  */
    732 
    733 #define DKFF_READYFORDUMP(x)	(((x) & DKF_READYFORDUMP) == DKF_READYFORDUMP)
    734 static volatile int	dk_dumping = 0;
    735 
    736 /* ARGSUSED */
    737 int
    738 dk_dump(struct dk_softc *dksc, dev_t dev,
    739     daddr_t blkno, void *vav, size_t size)
    740 {
    741 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    742 	char *va = vav;
    743 	struct disklabel *lp;
    744 	struct partition *p;
    745 	int part, towrt, nsects, sectoff, maxblkcnt, nblk;
    746 	int maxxfer, rv = 0;
    747 
    748 	/*
    749 	 * ensure that we consider this device to be safe for dumping,
    750 	 * and that the device is configured.
    751 	 */
    752 	if (!DKFF_READYFORDUMP(dksc->sc_flags)) {
    753 		DPRINTF(DKDB_DUMP, ("%s: bad dump flags 0x%x\n", __func__,
    754 		    dksc->sc_flags));
    755 		return ENXIO;
    756 	}
    757 
    758 	/* ensure that we are not already dumping */
    759 	if (dk_dumping)
    760 		return EFAULT;
    761 	dk_dumping = 1;
    762 
    763 	if (dkd->d_dumpblocks == NULL) {
    764 		DPRINTF(DKDB_DUMP, ("%s: no dumpblocks\n", __func__));
    765 		return ENXIO;
    766 	}
    767 
    768 	/* device specific max transfer size */
    769 	maxxfer = MAXPHYS;
    770 	if (dkd->d_iosize != NULL)
    771 		(*dkd->d_iosize)(dksc->sc_dev, &maxxfer);
    772 
    773 	/* Convert to disk sectors.  Request must be a multiple of size. */
    774 	part = DISKPART(dev);
    775 	lp = dksc->sc_dkdev.dk_label;
    776 	if ((size % lp->d_secsize) != 0) {
    777 		DPRINTF(DKDB_DUMP, ("%s: odd size %zu\n", __func__, size));
    778 		return EFAULT;
    779 	}
    780 	towrt = size / lp->d_secsize;
    781 	blkno = dbtob(blkno) / lp->d_secsize;   /* blkno in secsize units */
    782 
    783 	p = &lp->d_partitions[part];
    784 	if (p->p_fstype != FS_SWAP) {
    785 		DPRINTF(DKDB_DUMP, ("%s: bad fstype %d\n", __func__,
    786 		    p->p_fstype));
    787 		return ENXIO;
    788 	}
    789 	nsects = p->p_size;
    790 	sectoff = p->p_offset;
    791 
    792 	/* Check transfer bounds against partition size. */
    793 	if ((blkno < 0) || ((blkno + towrt) > nsects)) {
    794 		DPRINTF(DKDB_DUMP, ("%s: out of bounds blkno=%jd, towrt=%d, "
    795 		    "nsects=%d\n", __func__, (intmax_t)blkno, towrt, nsects));
    796 		return EINVAL;
    797 	}
    798 
    799 	/* Offset block number to start of partition. */
    800 	blkno += sectoff;
    801 
    802 	/* Start dumping and return when done. */
    803 	maxblkcnt = howmany(maxxfer, lp->d_secsize);
    804 	while (towrt > 0) {
    805 		nblk = min(maxblkcnt, towrt);
    806 
    807 		if ((rv = (*dkd->d_dumpblocks)(dksc->sc_dev, va, blkno, nblk))
    808 		    != 0) {
    809 			DPRINTF(DKDB_DUMP, ("%s: dumpblocks %d\n", __func__,
    810 			    rv));
    811 			return rv;
    812 		}
    813 
    814 		towrt -= nblk;
    815 		blkno += nblk;
    816 		va += nblk * lp->d_secsize;
    817 	}
    818 
    819 	dk_dumping = 0;
    820 
    821 	return 0;
    822 }
    823 
    824 /* ARGSUSED */
    825 void
    826 dk_getdefaultlabel(struct dk_softc *dksc, struct disklabel *lp)
    827 {
    828 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    829 	struct disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    830 
    831 	memset(lp, 0, sizeof(*lp));
    832 
    833 	if (dg->dg_secperunit > UINT32_MAX)
    834 		lp->d_secperunit = UINT32_MAX;
    835 	else
    836 		lp->d_secperunit = dg->dg_secperunit;
    837 	lp->d_secsize = dg->dg_secsize;
    838 	lp->d_nsectors = dg->dg_nsectors;
    839 	lp->d_ntracks = dg->dg_ntracks;
    840 	lp->d_ncylinders = dg->dg_ncylinders;
    841 	lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
    842 
    843 	strlcpy(lp->d_typename, dksc->sc_xname, sizeof(lp->d_typename));
    844 	lp->d_type = dksc->sc_dtype;
    845 	strlcpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
    846 	lp->d_rpm = 3600;
    847 	lp->d_interleave = 1;
    848 	lp->d_flags = 0;
    849 
    850 	lp->d_partitions[RAW_PART].p_offset = 0;
    851 	lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
    852 	lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
    853 	lp->d_npartitions = RAW_PART + 1;
    854 
    855 	lp->d_magic = DISKMAGIC;
    856 	lp->d_magic2 = DISKMAGIC;
    857 
    858 	if (dkd->d_label)
    859 		dkd->d_label(dksc->sc_dev, lp);
    860 
    861 	lp->d_checksum = dkcksum(lp);
    862 }
    863 
    864 /* ARGSUSED */
    865 void
    866 dk_getdisklabel(struct dk_softc *dksc, dev_t dev)
    867 {
    868 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    869 	struct	 disklabel *lp = dksc->sc_dkdev.dk_label;
    870 	struct	 cpu_disklabel *clp = dksc->sc_dkdev.dk_cpulabel;
    871 	struct   disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    872 	struct	 partition *pp;
    873 	int	 i;
    874 	const char	*errstring;
    875 
    876 	memset(clp, 0x0, sizeof(*clp));
    877 	dk_getdefaultlabel(dksc, lp);
    878 	errstring = readdisklabel(DKLABELDEV(dev), dkd->d_strategy,
    879 	    dksc->sc_dkdev.dk_label, dksc->sc_dkdev.dk_cpulabel);
    880 	if (errstring) {
    881 		dk_makedisklabel(dksc);
    882 		if (dksc->sc_flags & DKF_WARNLABEL)
    883 			printf("%s: %s\n", dksc->sc_xname, errstring);
    884 		return;
    885 	}
    886 
    887 	if ((dksc->sc_flags & DKF_LABELSANITY) == 0)
    888 		return;
    889 
    890 	/* Sanity check */
    891 	if (lp->d_secperunit < UINT32_MAX ?
    892 		lp->d_secperunit != dg->dg_secperunit :
    893 		lp->d_secperunit > dg->dg_secperunit)
    894 		printf("WARNING: %s: total sector size in disklabel (%ju) "
    895 		    "!= the size of %s (%ju)\n", dksc->sc_xname,
    896 		    (uintmax_t)lp->d_secperunit, dksc->sc_xname,
    897 		    (uintmax_t)dg->dg_secperunit);
    898 
    899 	for (i=0; i < lp->d_npartitions; i++) {
    900 		pp = &lp->d_partitions[i];
    901 		if (pp->p_offset + pp->p_size > dg->dg_secperunit)
    902 			printf("WARNING: %s: end of partition `%c' exceeds "
    903 			    "the size of %s (%ju)\n", dksc->sc_xname,
    904 			    'a' + i, dksc->sc_xname,
    905 			    (uintmax_t)dg->dg_secperunit);
    906 	}
    907 }
    908 
    909 /* ARGSUSED */
    910 static void
    911 dk_makedisklabel(struct dk_softc *dksc)
    912 {
    913 	struct	disklabel *lp = dksc->sc_dkdev.dk_label;
    914 
    915 	lp->d_partitions[RAW_PART].p_fstype = FS_BSDFFS;
    916 	strlcpy(lp->d_packname, "default label", sizeof(lp->d_packname));
    917 	lp->d_checksum = dkcksum(lp);
    918 }
    919 
    920 /* This function is taken from ccd.c:1.76  --rcd */
    921 
    922 /*
    923  * XXX this function looks too generic for dksubr.c, shouldn't we
    924  *     put it somewhere better?
    925  */
    926 
    927 /*
    928  * Lookup the provided name in the filesystem.  If the file exists,
    929  * is a valid block device, and isn't being used by anyone else,
    930  * set *vpp to the file's vnode.
    931  */
    932 int
    933 dk_lookup(struct pathbuf *pb, struct lwp *l, struct vnode **vpp)
    934 {
    935 	struct nameidata nd;
    936 	struct vnode *vp;
    937 	int     error;
    938 
    939 	if (l == NULL)
    940 		return ESRCH;	/* Is ESRCH the best choice? */
    941 
    942 	NDINIT(&nd, LOOKUP, FOLLOW, pb);
    943 	if ((error = vn_open(&nd, FREAD | FWRITE, 0)) != 0) {
    944 		DPRINTF((DKDB_FOLLOW|DKDB_INIT),
    945 		    ("%s: vn_open error = %d\n", __func__, error));
    946 		return error;
    947 	}
    948 
    949 	vp = nd.ni_vp;
    950 	if (vp->v_type != VBLK) {
    951 		error = ENOTBLK;
    952 		goto out;
    953 	}
    954 
    955 	/* Reopen as anonymous vnode to protect against forced unmount. */
    956 	if ((error = bdevvp(vp->v_rdev, vpp)) != 0)
    957 		goto out;
    958 	VOP_UNLOCK(vp);
    959 	if ((error = vn_close(vp, FREAD | FWRITE, l->l_cred)) != 0) {
    960 		vrele(*vpp);
    961 		return error;
    962 	}
    963 	if ((error = VOP_OPEN(*vpp, FREAD | FWRITE, l->l_cred)) != 0) {
    964 		vrele(*vpp);
    965 		return error;
    966 	}
    967 	mutex_enter((*vpp)->v_interlock);
    968 	(*vpp)->v_writecount++;
    969 	mutex_exit((*vpp)->v_interlock);
    970 
    971 	IFDEBUG(DKDB_VNODE, vprint("dk_lookup: vnode info", *vpp));
    972 
    973 	return 0;
    974 out:
    975 	VOP_UNLOCK(vp);
    976 	(void) vn_close(vp, FREAD | FWRITE, l->l_cred);
    977 	return error;
    978 }
    979 
    980 MODULE(MODULE_CLASS_MISC, dk_subr, NULL);
    981 
    982 static int
    983 dk_subr_modcmd(modcmd_t cmd, void *arg)
    984 {
    985 	switch (cmd) {
    986 	case MODULE_CMD_INIT:
    987 	case MODULE_CMD_FINI:
    988 		return 0;
    989 	case MODULE_CMD_STAT:
    990 	case MODULE_CMD_AUTOUNLOAD:
    991 	default:
    992 		return ENOTTY;
    993 	}
    994 }
    995