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