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