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