Home | History | Annotate | Line # | Download | only in dev
dksubr.c revision 1.93
      1 /* $NetBSD: dksubr.c,v 1.93 2016/12/08 12:22:56 mlelstv 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.93 2016/12/08 12:22:56 mlelstv 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 	if (dksc->sc_busy)
    382 		goto done;
    383 	dksc->sc_busy = true;
    384 
    385 	/*
    386 	 * Peeking at the buffer queue and committing the operation
    387 	 * only after success isn't atomic.
    388 	 *
    389 	 * So when a diskstart fails, the buffer is saved
    390 	 * and tried again before the next buffer is fetched.
    391 	 * dk_drain() handles flushing of a saved buffer.
    392 	 *
    393 	 * This keeps order of I/O operations, unlike bufq_put.
    394 	 */
    395 
    396 	bp = dksc->sc_deferred;
    397 	dksc->sc_deferred = NULL;
    398 
    399 	if (bp == NULL)
    400 		bp = bufq_get(dksc->sc_bufq);
    401 
    402 	while (bp != NULL) {
    403 
    404 		disk_busy(&dksc->sc_dkdev);
    405 		mutex_exit(&dksc->sc_iolock);
    406 		error = dkd->d_diskstart(dksc->sc_dev, bp);
    407 		mutex_enter(&dksc->sc_iolock);
    408 		if (error == EAGAIN) {
    409 			dksc->sc_deferred = bp;
    410 			disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    411 			break;
    412 		}
    413 
    414 		if (error != 0) {
    415 			bp->b_error = error;
    416 			bp->b_resid = bp->b_bcount;
    417 			dk_done1(dksc, bp, false);
    418 		}
    419 
    420 		bp = bufq_get(dksc->sc_bufq);
    421 	}
    422 
    423 	dksc->sc_busy = false;
    424 done:
    425 	mutex_exit(&dksc->sc_iolock);
    426 }
    427 
    428 static void
    429 dk_done1(struct dk_softc *dksc, struct buf *bp, bool lock)
    430 {
    431 	struct disk *dk = &dksc->sc_dkdev;
    432 
    433 	if (bp->b_error != 0) {
    434 		struct cfdriver *cd = device_cfdriver(dksc->sc_dev);
    435 
    436 		diskerr(bp, cd->cd_name, "error", LOG_PRINTF, 0,
    437 			dk->dk_label);
    438 		printf("\n");
    439 	}
    440 
    441 	if (lock)
    442 		mutex_enter(&dksc->sc_iolock);
    443 	disk_unbusy(dk, bp->b_bcount - bp->b_resid, (bp->b_flags & B_READ));
    444 	if (lock)
    445 		mutex_exit(&dksc->sc_iolock);
    446 
    447 	rnd_add_uint32(&dksc->sc_rnd_source, bp->b_rawblkno);
    448 
    449 	biodone(bp);
    450 }
    451 
    452 void
    453 dk_done(struct dk_softc *dksc, struct buf *bp)
    454 {
    455 	dk_done1(dksc, bp, true);
    456 }
    457 
    458 void
    459 dk_drain(struct dk_softc *dksc)
    460 {
    461 	struct buf *bp;
    462 
    463 	mutex_enter(&dksc->sc_iolock);
    464 	bp = dksc->sc_deferred;
    465 	dksc->sc_deferred = NULL;
    466 	if (bp != NULL) {
    467 		bp->b_error = EIO;
    468 		bp->b_resid = bp->b_bcount;
    469 		biodone(bp);
    470 	}
    471 	bufq_drain(dksc->sc_bufq);
    472 	mutex_exit(&dksc->sc_iolock);
    473 }
    474 
    475 int
    476 dk_discard(struct dk_softc *dksc, dev_t dev, off_t pos, off_t len)
    477 {
    478 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    479 	unsigned secsize = dksc->sc_dkdev.dk_geom.dg_secsize;
    480 	struct buf tmp, *bp = &tmp;
    481 	int error;
    482 
    483 	DPRINTF_FOLLOW(("%s(%s, %p, 0x"PRIx64", %jd, %jd)\n", __func__,
    484 	    dksc->sc_xname, dksc, (intmax_t)pos, (intmax_t)len));
    485 
    486 	if (!(dksc->sc_flags & DKF_INITED)) {
    487 		DPRINTF_FOLLOW(("%s: not inited\n", __func__));
    488 		return ENXIO;
    489 	}
    490 
    491 	if (secsize == 0 || (pos % secsize) != 0)
    492 		return EINVAL;
    493 
    494 	/* enough data to please the bounds checking code */
    495 	bp->b_dev = dev;
    496 	bp->b_blkno = (daddr_t)(pos / secsize);
    497 	bp->b_bcount = len;
    498 	bp->b_flags = B_WRITE;
    499 
    500 	error = dk_translate(dksc, bp);
    501 	if (error >= 0)
    502 		return error;
    503 
    504 	error = dkd->d_discard(dksc->sc_dev,
    505 		(off_t)bp->b_rawblkno * secsize,
    506 		(off_t)bp->b_bcount);
    507 
    508 	return error;
    509 }
    510 
    511 int
    512 dk_size(struct dk_softc *dksc, dev_t dev)
    513 {
    514 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    515 	struct	disklabel *lp;
    516 	int	is_open;
    517 	int	part;
    518 	int	size;
    519 
    520 	if ((dksc->sc_flags & DKF_INITED) == 0)
    521 		return -1;
    522 
    523 	part = DISKPART(dev);
    524 	is_open = dksc->sc_dkdev.dk_openmask & (1 << part);
    525 
    526 	if (!is_open && dkd->d_open(dev, 0, S_IFBLK, curlwp))
    527 		return -1;
    528 
    529 	lp = dksc->sc_dkdev.dk_label;
    530 	if (lp->d_partitions[part].p_fstype != FS_SWAP)
    531 		size = -1;
    532 	else
    533 		size = lp->d_partitions[part].p_size *
    534 		    (lp->d_secsize / DEV_BSIZE);
    535 
    536 	if (!is_open && dkd->d_close(dev, 0, S_IFBLK, curlwp))
    537 		return -1;
    538 
    539 	return size;
    540 }
    541 
    542 int
    543 dk_ioctl(struct dk_softc *dksc, dev_t dev,
    544 	    u_long cmd, void *data, int flag, struct lwp *l)
    545 {
    546 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    547 	struct	disklabel *lp;
    548 	struct	disk *dk = &dksc->sc_dkdev;
    549 #ifdef __HAVE_OLD_DISKLABEL
    550 	struct	disklabel newlabel;
    551 #endif
    552 	int	error;
    553 
    554 	DPRINTF_FOLLOW(("%s(%s, %p, 0x%"PRIx64", 0x%lx)\n", __func__,
    555 	    dksc->sc_xname, dksc, dev, cmd));
    556 
    557 	/* ensure that the pseudo disk is open for writes for these commands */
    558 	switch (cmd) {
    559 	case DIOCSDINFO:
    560 	case DIOCWDINFO:
    561 #ifdef __HAVE_OLD_DISKLABEL
    562 	case ODIOCSDINFO:
    563 	case ODIOCWDINFO:
    564 #endif
    565 	case DIOCKLABEL:
    566 	case DIOCWLABEL:
    567 	case DIOCAWEDGE:
    568 	case DIOCDWEDGE:
    569 	case DIOCSSTRATEGY:
    570 		if ((flag & FWRITE) == 0)
    571 			return EBADF;
    572 	}
    573 
    574 	/* ensure that the pseudo-disk is initialized for these */
    575 	switch (cmd) {
    576 	case DIOCGDINFO:
    577 	case DIOCSDINFO:
    578 	case DIOCWDINFO:
    579 	case DIOCGPARTINFO:
    580 	case DIOCKLABEL:
    581 	case DIOCWLABEL:
    582 	case DIOCGDEFLABEL:
    583 	case DIOCAWEDGE:
    584 	case DIOCDWEDGE:
    585 	case DIOCLWEDGES:
    586 	case DIOCMWEDGES:
    587 	case DIOCCACHESYNC:
    588 #ifdef __HAVE_OLD_DISKLABEL
    589 	case ODIOCGDINFO:
    590 	case ODIOCSDINFO:
    591 	case ODIOCWDINFO:
    592 	case ODIOCGDEFLABEL:
    593 #endif
    594 		if ((dksc->sc_flags & DKF_INITED) == 0)
    595 			return ENXIO;
    596 	}
    597 
    598 	error = disk_ioctl(dk, dev, cmd, data, flag, l);
    599 	if (error != EPASSTHROUGH)
    600 		return error;
    601 	else
    602 		error = 0;
    603 
    604 	switch (cmd) {
    605 	case DIOCWDINFO:
    606 	case DIOCSDINFO:
    607 #ifdef __HAVE_OLD_DISKLABEL
    608 	case ODIOCWDINFO:
    609 	case ODIOCSDINFO:
    610 #endif
    611 #ifdef __HAVE_OLD_DISKLABEL
    612 		if (cmd == ODIOCSDINFO || cmd == ODIOCWDINFO) {
    613 			memset(&newlabel, 0, sizeof newlabel);
    614 			memcpy(&newlabel, data, sizeof (struct olddisklabel));
    615 			lp = &newlabel;
    616 		} else
    617 #endif
    618 		lp = (struct disklabel *)data;
    619 
    620 		mutex_enter(&dk->dk_openlock);
    621 		dksc->sc_flags |= DKF_LABELLING;
    622 
    623 		error = setdisklabel(dksc->sc_dkdev.dk_label,
    624 		    lp, 0, dksc->sc_dkdev.dk_cpulabel);
    625 		if (error == 0) {
    626 			if (cmd == DIOCWDINFO
    627 #ifdef __HAVE_OLD_DISKLABEL
    628 			    || cmd == ODIOCWDINFO
    629 #endif
    630 			   )
    631 				error = writedisklabel(DKLABELDEV(dev),
    632 				    dkd->d_strategy, dksc->sc_dkdev.dk_label,
    633 				    dksc->sc_dkdev.dk_cpulabel);
    634 		}
    635 
    636 		dksc->sc_flags &= ~DKF_LABELLING;
    637 		mutex_exit(&dk->dk_openlock);
    638 		break;
    639 
    640 	case DIOCKLABEL:
    641 		if (*(int *)data != 0)
    642 			dksc->sc_flags |= DKF_KLABEL;
    643 		else
    644 			dksc->sc_flags &= ~DKF_KLABEL;
    645 		break;
    646 
    647 	case DIOCWLABEL:
    648 		if (*(int *)data != 0)
    649 			dksc->sc_flags |= DKF_WLABEL;
    650 		else
    651 			dksc->sc_flags &= ~DKF_WLABEL;
    652 		break;
    653 
    654 	case DIOCGDEFLABEL:
    655 		dk_getdefaultlabel(dksc, (struct disklabel *)data);
    656 		break;
    657 
    658 #ifdef __HAVE_OLD_DISKLABEL
    659 	case ODIOCGDEFLABEL:
    660 		dk_getdefaultlabel(dksc, &newlabel);
    661 		if (newlabel.d_npartitions > OLDMAXPARTITIONS)
    662 			return ENOTTY;
    663 		memcpy(data, &newlabel, sizeof (struct olddisklabel));
    664 		break;
    665 #endif
    666 
    667 	case DIOCGSTRATEGY:
    668 	    {
    669 		struct disk_strategy *dks = (void *)data;
    670 
    671 		mutex_enter(&dksc->sc_iolock);
    672 		if (dksc->sc_bufq != NULL)
    673 			strlcpy(dks->dks_name,
    674 			    bufq_getstrategyname(dksc->sc_bufq),
    675 			    sizeof(dks->dks_name));
    676 		else
    677 			error = EINVAL;
    678 		mutex_exit(&dksc->sc_iolock);
    679 		dks->dks_paramlen = 0;
    680 		break;
    681 	    }
    682 
    683 	case DIOCSSTRATEGY:
    684 	    {
    685 		struct disk_strategy *dks = (void *)data;
    686 		struct bufq_state *new;
    687 		struct bufq_state *old;
    688 
    689 		if (dks->dks_param != NULL) {
    690 			return EINVAL;
    691 		}
    692 		dks->dks_name[sizeof(dks->dks_name) - 1] = 0; /* ensure term */
    693 		error = bufq_alloc(&new, dks->dks_name,
    694 		    BUFQ_EXACT|BUFQ_SORT_RAWBLOCK);
    695 		if (error) {
    696 			return error;
    697 		}
    698 		mutex_enter(&dksc->sc_iolock);
    699 		old = dksc->sc_bufq;
    700 		if (old)
    701 			bufq_move(new, old);
    702 		dksc->sc_bufq = new;
    703 		mutex_exit(&dksc->sc_iolock);
    704 		if (old)
    705 			bufq_free(old);
    706 		break;
    707 	    }
    708 
    709 	default:
    710 		error = ENOTTY;
    711 	}
    712 
    713 	return error;
    714 }
    715 
    716 /*
    717  * dk_dump dumps all of physical memory into the partition specified.
    718  * This requires substantially more framework than {s,w}ddump, and hence
    719  * is probably much more fragile.
    720  *
    721  */
    722 
    723 #define DKFF_READYFORDUMP(x)	(((x) & DKF_READYFORDUMP) == DKF_READYFORDUMP)
    724 static volatile int	dk_dumping = 0;
    725 
    726 /* ARGSUSED */
    727 int
    728 dk_dump(struct dk_softc *dksc, dev_t dev,
    729     daddr_t blkno, void *vav, size_t size)
    730 {
    731 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    732 	char *va = vav;
    733 	struct disklabel *lp;
    734 	struct partition *p;
    735 	int part, towrt, nsects, sectoff, maxblkcnt, nblk;
    736 	int maxxfer, rv = 0;
    737 
    738 	/*
    739 	 * ensure that we consider this device to be safe for dumping,
    740 	 * and that the device is configured.
    741 	 */
    742 	if (!DKFF_READYFORDUMP(dksc->sc_flags)) {
    743 		DPRINTF(DKDB_DUMP, ("%s: bad dump flags 0x%x\n", __func__,
    744 		    dksc->sc_flags));
    745 		return ENXIO;
    746 	}
    747 
    748 	/* ensure that we are not already dumping */
    749 	if (dk_dumping)
    750 		return EFAULT;
    751 	dk_dumping = 1;
    752 
    753 	if (dkd->d_dumpblocks == NULL) {
    754 		DPRINTF(DKDB_DUMP, ("%s: no dumpblocks\n", __func__));
    755 		return ENXIO;
    756 	}
    757 
    758 	/* device specific max transfer size */
    759 	maxxfer = MAXPHYS;
    760 	if (dkd->d_iosize != NULL)
    761 		(*dkd->d_iosize)(dksc->sc_dev, &maxxfer);
    762 
    763 	/* Convert to disk sectors.  Request must be a multiple of size. */
    764 	part = DISKPART(dev);
    765 	lp = dksc->sc_dkdev.dk_label;
    766 	if ((size % lp->d_secsize) != 0) {
    767 		DPRINTF(DKDB_DUMP, ("%s: odd size %zu\n", __func__, size));
    768 		return EFAULT;
    769 	}
    770 	towrt = size / lp->d_secsize;
    771 	blkno = dbtob(blkno) / lp->d_secsize;   /* blkno in secsize units */
    772 
    773 	p = &lp->d_partitions[part];
    774 	if (p->p_fstype != FS_SWAP) {
    775 		DPRINTF(DKDB_DUMP, ("%s: bad fstype %d\n", __func__,
    776 		    p->p_fstype));
    777 		return ENXIO;
    778 	}
    779 	nsects = p->p_size;
    780 	sectoff = p->p_offset;
    781 
    782 	/* Check transfer bounds against partition size. */
    783 	if ((blkno < 0) || ((blkno + towrt) > nsects)) {
    784 		DPRINTF(DKDB_DUMP, ("%s: out of bounds blkno=%jd, towrt=%d, "
    785 		    "nsects=%d\n", __func__, (intmax_t)blkno, towrt, nsects));
    786 		return EINVAL;
    787 	}
    788 
    789 	/* Offset block number to start of partition. */
    790 	blkno += sectoff;
    791 
    792 	/* Start dumping and return when done. */
    793 	maxblkcnt = howmany(maxxfer, lp->d_secsize);
    794 	while (towrt > 0) {
    795 		nblk = min(maxblkcnt, towrt);
    796 
    797 		if ((rv = (*dkd->d_dumpblocks)(dksc->sc_dev, va, blkno, nblk))
    798 		    != 0) {
    799 			DPRINTF(DKDB_DUMP, ("%s: dumpblocks %d\n", __func__,
    800 			    rv));
    801 			return rv;
    802 		}
    803 
    804 		towrt -= nblk;
    805 		blkno += nblk;
    806 		va += nblk * lp->d_secsize;
    807 	}
    808 
    809 	dk_dumping = 0;
    810 
    811 	return 0;
    812 }
    813 
    814 /* ARGSUSED */
    815 void
    816 dk_getdefaultlabel(struct dk_softc *dksc, struct disklabel *lp)
    817 {
    818 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    819 	struct disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    820 
    821 	memset(lp, 0, sizeof(*lp));
    822 
    823 	if (dg->dg_secperunit > UINT32_MAX)
    824 		lp->d_secperunit = UINT32_MAX;
    825 	else
    826 		lp->d_secperunit = dg->dg_secperunit;
    827 	lp->d_secsize = dg->dg_secsize;
    828 	lp->d_nsectors = dg->dg_nsectors;
    829 	lp->d_ntracks = dg->dg_ntracks;
    830 	lp->d_ncylinders = dg->dg_ncylinders;
    831 	lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
    832 
    833 	strlcpy(lp->d_typename, dksc->sc_xname, sizeof(lp->d_typename));
    834 	lp->d_type = dksc->sc_dtype;
    835 	strlcpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
    836 	lp->d_rpm = 3600;
    837 	lp->d_interleave = 1;
    838 	lp->d_flags = 0;
    839 
    840 	lp->d_partitions[RAW_PART].p_offset = 0;
    841 	lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
    842 	lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
    843 	lp->d_npartitions = RAW_PART + 1;
    844 
    845 	lp->d_magic = DISKMAGIC;
    846 	lp->d_magic2 = DISKMAGIC;
    847 
    848 	if (dkd->d_label)
    849 		dkd->d_label(dksc->sc_dev, lp);
    850 
    851 	lp->d_checksum = dkcksum(lp);
    852 }
    853 
    854 /* ARGSUSED */
    855 void
    856 dk_getdisklabel(struct dk_softc *dksc, dev_t dev)
    857 {
    858 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    859 	struct	 disklabel *lp = dksc->sc_dkdev.dk_label;
    860 	struct	 cpu_disklabel *clp = dksc->sc_dkdev.dk_cpulabel;
    861 	struct   disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    862 	struct	 partition *pp;
    863 	int	 i;
    864 	const char	*errstring;
    865 
    866 	memset(clp, 0x0, sizeof(*clp));
    867 	dk_getdefaultlabel(dksc, lp);
    868 	errstring = readdisklabel(DKLABELDEV(dev), dkd->d_strategy,
    869 	    dksc->sc_dkdev.dk_label, dksc->sc_dkdev.dk_cpulabel);
    870 	if (errstring) {
    871 		dk_makedisklabel(dksc);
    872 		if (dksc->sc_flags & DKF_WARNLABEL)
    873 			printf("%s: %s\n", dksc->sc_xname, errstring);
    874 		return;
    875 	}
    876 
    877 	if ((dksc->sc_flags & DKF_LABELSANITY) == 0)
    878 		return;
    879 
    880 	/* Sanity check */
    881 	if (lp->d_secperunit < UINT32_MAX ?
    882 		lp->d_secperunit != dg->dg_secperunit :
    883 		lp->d_secperunit > dg->dg_secperunit)
    884 		printf("WARNING: %s: total sector size in disklabel (%ju) "
    885 		    "!= the size of %s (%ju)\n", dksc->sc_xname,
    886 		    (uintmax_t)lp->d_secperunit, dksc->sc_xname,
    887 		    (uintmax_t)dg->dg_secperunit);
    888 
    889 	for (i=0; i < lp->d_npartitions; i++) {
    890 		pp = &lp->d_partitions[i];
    891 		if (pp->p_offset + pp->p_size > dg->dg_secperunit)
    892 			printf("WARNING: %s: end of partition `%c' exceeds "
    893 			    "the size of %s (%ju)\n", dksc->sc_xname,
    894 			    'a' + i, dksc->sc_xname,
    895 			    (uintmax_t)dg->dg_secperunit);
    896 	}
    897 }
    898 
    899 /* ARGSUSED */
    900 static void
    901 dk_makedisklabel(struct dk_softc *dksc)
    902 {
    903 	struct	disklabel *lp = dksc->sc_dkdev.dk_label;
    904 
    905 	lp->d_partitions[RAW_PART].p_fstype = FS_BSDFFS;
    906 	strlcpy(lp->d_packname, "default label", sizeof(lp->d_packname));
    907 	lp->d_checksum = dkcksum(lp);
    908 }
    909 
    910 /* This function is taken from ccd.c:1.76  --rcd */
    911 
    912 /*
    913  * XXX this function looks too generic for dksubr.c, shouldn't we
    914  *     put it somewhere better?
    915  */
    916 
    917 /*
    918  * Lookup the provided name in the filesystem.  If the file exists,
    919  * is a valid block device, and isn't being used by anyone else,
    920  * set *vpp to the file's vnode.
    921  */
    922 int
    923 dk_lookup(struct pathbuf *pb, struct lwp *l, struct vnode **vpp)
    924 {
    925 	struct nameidata nd;
    926 	struct vnode *vp;
    927 	int     error;
    928 
    929 	if (l == NULL)
    930 		return ESRCH;	/* Is ESRCH the best choice? */
    931 
    932 	NDINIT(&nd, LOOKUP, FOLLOW, pb);
    933 	if ((error = vn_open(&nd, FREAD | FWRITE, 0)) != 0) {
    934 		DPRINTF((DKDB_FOLLOW|DKDB_INIT),
    935 		    ("%s: vn_open error = %d\n", __func__, error));
    936 		return error;
    937 	}
    938 
    939 	vp = nd.ni_vp;
    940 	if (vp->v_type != VBLK) {
    941 		error = ENOTBLK;
    942 		goto out;
    943 	}
    944 
    945 	/* Reopen as anonymous vnode to protect against forced unmount. */
    946 	if ((error = bdevvp(vp->v_rdev, vpp)) != 0)
    947 		goto out;
    948 	VOP_UNLOCK(vp);
    949 	if ((error = vn_close(vp, FREAD | FWRITE, l->l_cred)) != 0) {
    950 		vrele(*vpp);
    951 		return error;
    952 	}
    953 	if ((error = VOP_OPEN(*vpp, FREAD | FWRITE, l->l_cred)) != 0) {
    954 		vrele(*vpp);
    955 		return error;
    956 	}
    957 	mutex_enter((*vpp)->v_interlock);
    958 	(*vpp)->v_writecount++;
    959 	mutex_exit((*vpp)->v_interlock);
    960 
    961 	IFDEBUG(DKDB_VNODE, vprint("dk_lookup: vnode info", *vpp));
    962 
    963 	return 0;
    964 out:
    965 	VOP_UNLOCK(vp);
    966 	(void) vn_close(vp, FREAD | FWRITE, l->l_cred);
    967 	return error;
    968 }
    969 
    970 MODULE(MODULE_CLASS_MISC, dk_subr, NULL);
    971 
    972 static int
    973 dk_subr_modcmd(modcmd_t cmd, void *arg)
    974 {
    975 	switch (cmd) {
    976 	case MODULE_CMD_INIT:
    977 	case MODULE_CMD_FINI:
    978 		return 0;
    979 	case MODULE_CMD_STAT:
    980 	case MODULE_CMD_AUTOUNLOAD:
    981 	default:
    982 		return ENOTTY;
    983 	}
    984 }
    985