Home | History | Annotate | Line # | Download | only in dev
dksubr.c revision 1.60
      1  1.59   mlelstv /* $NetBSD: dksubr.c,v 1.60 2015/05/02 08:00:08 mlelstv 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.59   mlelstv __KERNEL_RCSID(0, "$NetBSD: dksubr.c,v 1.60 2015/05/02 08:00:08 mlelstv 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.1     elric 
     61   1.1     elric #define IFDEBUG(x,y)		if (dkdebug & (x)) y
     62   1.1     elric #define DPRINTF(x,y)		IFDEBUG(x, printf y)
     63   1.1     elric #define DPRINTF_FOLLOW(y)	DPRINTF(DKDB_FOLLOW, y)
     64   1.1     elric #else
     65   1.1     elric #define IFDEBUG(x,y)
     66   1.1     elric #define DPRINTF(x,y)
     67   1.1     elric #define DPRINTF_FOLLOW(y)
     68   1.1     elric #endif
     69   1.1     elric 
     70  1.49  pgoyette static int dk_subr_modcmd(modcmd_t, void *);
     71  1.49  pgoyette 
     72   1.1     elric #define DKLABELDEV(dev)	\
     73   1.1     elric 	(MAKEDISKDEV(major((dev)), DISKUNIT((dev)), RAW_PART))
     74   1.1     elric 
     75  1.60   mlelstv static void	dk_makedisklabel(struct dk_softc *);
     76   1.1     elric 
     77   1.1     elric void
     78  1.60   mlelstv dk_init(struct dk_softc *dksc, device_t dev, int dtype)
     79   1.1     elric {
     80   1.1     elric 
     81   1.1     elric 	memset(dksc, 0x0, sizeof(*dksc));
     82  1.60   mlelstv 	dksc->sc_dtype = dtype;
     83  1.60   mlelstv 	dksc->sc_dev = dev;
     84  1.60   mlelstv 
     85  1.60   mlelstv 	strncpy(dksc->sc_xname, device_xname(dev), DK_XNAME_SIZE);
     86   1.1     elric 	dksc->sc_dkdev.dk_name = dksc->sc_xname;
     87   1.1     elric }
     88   1.1     elric 
     89  1.60   mlelstv void
     90  1.60   mlelstv dk_attach(struct dk_softc *dksc)
     91  1.60   mlelstv {
     92  1.60   mlelstv 	dksc->sc_flags |= DKF_INITED | DKF_WARNLABEL | DKF_LABELSANITY;
     93  1.60   mlelstv }
     94  1.60   mlelstv 
     95  1.60   mlelstv void
     96  1.60   mlelstv dk_detach(struct dk_softc *dksc)
     97  1.60   mlelstv {
     98  1.60   mlelstv 	dksc->sc_flags &= ~DKF_INITED;
     99  1.60   mlelstv }
    100  1.60   mlelstv 
    101   1.1     elric /* ARGSUSED */
    102   1.1     elric int
    103  1.60   mlelstv dk_open(struct dk_softc *dksc, dev_t dev,
    104  1.27  christos     int flags, int fmt, struct lwp *l)
    105   1.1     elric {
    106   1.1     elric 	struct	disklabel *lp = dksc->sc_dkdev.dk_label;
    107   1.1     elric 	int	part = DISKPART(dev);
    108   1.1     elric 	int	pmask = 1 << part;
    109   1.1     elric 	int	ret = 0;
    110  1.16      yamt 	struct disk *dk = &dksc->sc_dkdev;
    111   1.1     elric 
    112  1.39    cegger 	DPRINTF_FOLLOW(("dk_open(%s, %p, 0x%"PRIx64", 0x%x)\n",
    113  1.60   mlelstv 	    dksc->sc_xname, dksc, dev, flags));
    114   1.1     elric 
    115  1.30        ad 	mutex_enter(&dk->dk_openlock);
    116   1.1     elric 	part = DISKPART(dev);
    117  1.16      yamt 
    118  1.16      yamt 	/*
    119  1.16      yamt 	 * If there are wedges, and this is not RAW_PART, then we
    120  1.16      yamt 	 * need to fail.
    121  1.16      yamt 	 */
    122  1.16      yamt 	if (dk->dk_nwedges != 0 && part != RAW_PART) {
    123  1.16      yamt 		ret = EBUSY;
    124  1.16      yamt 		goto done;
    125  1.16      yamt 	}
    126  1.16      yamt 
    127   1.1     elric 	pmask = 1 << part;
    128   1.1     elric 
    129   1.1     elric 	/*
    130   1.1     elric 	 * If we're init'ed and there are no other open partitions then
    131   1.1     elric 	 * update the in-core disklabel.
    132   1.1     elric 	 */
    133  1.16      yamt 	if ((dksc->sc_flags & DKF_INITED)) {
    134  1.60   mlelstv 		if ((dksc->sc_flags & DKF_VLABEL) == 0) {
    135  1.60   mlelstv 			dksc->sc_flags |= DKF_VLABEL;
    136  1.60   mlelstv 			dk_getdisklabel(dksc, dev);
    137  1.16      yamt 		}
    138  1.16      yamt 	}
    139   1.1     elric 
    140   1.1     elric 	/* Fail if we can't find the partition. */
    141  1.60   mlelstv 	if (part != RAW_PART &&
    142  1.60   mlelstv 	    ((dksc->sc_flags & DKF_VLABEL) == 0 ||
    143  1.60   mlelstv 	     part >= lp->d_npartitions ||
    144  1.60   mlelstv 	     lp->d_partitions[part].p_fstype == FS_UNUSED)) {
    145   1.1     elric 		ret = ENXIO;
    146   1.1     elric 		goto done;
    147   1.1     elric 	}
    148   1.1     elric 
    149   1.1     elric 	/* Mark our unit as open. */
    150   1.1     elric 	switch (fmt) {
    151   1.1     elric 	case S_IFCHR:
    152  1.16      yamt 		dk->dk_copenmask |= pmask;
    153   1.1     elric 		break;
    154   1.1     elric 	case S_IFBLK:
    155  1.16      yamt 		dk->dk_bopenmask |= pmask;
    156   1.1     elric 		break;
    157   1.1     elric 	}
    158   1.1     elric 
    159  1.16      yamt 	dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
    160   1.1     elric 
    161   1.1     elric done:
    162  1.30        ad 	mutex_exit(&dk->dk_openlock);
    163   1.1     elric 	return ret;
    164   1.1     elric }
    165   1.1     elric 
    166   1.1     elric /* ARGSUSED */
    167   1.1     elric int
    168  1.60   mlelstv dk_close(struct dk_softc *dksc, dev_t dev,
    169  1.27  christos     int flags, int fmt, struct lwp *l)
    170   1.1     elric {
    171  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    172   1.1     elric 	int	part = DISKPART(dev);
    173   1.1     elric 	int	pmask = 1 << part;
    174  1.16      yamt 	struct disk *dk = &dksc->sc_dkdev;
    175   1.1     elric 
    176  1.39    cegger 	DPRINTF_FOLLOW(("dk_close(%s, %p, 0x%"PRIx64", 0x%x)\n",
    177  1.60   mlelstv 	    dksc->sc_xname, dksc, dev, flags));
    178   1.1     elric 
    179  1.30        ad 	mutex_enter(&dk->dk_openlock);
    180   1.1     elric 
    181   1.1     elric 	switch (fmt) {
    182   1.1     elric 	case S_IFCHR:
    183  1.16      yamt 		dk->dk_copenmask &= ~pmask;
    184   1.1     elric 		break;
    185   1.1     elric 	case S_IFBLK:
    186  1.16      yamt 		dk->dk_bopenmask &= ~pmask;
    187   1.1     elric 		break;
    188   1.1     elric 	}
    189  1.16      yamt 	dk->dk_openmask = dk->dk_copenmask | dk->dk_bopenmask;
    190   1.1     elric 
    191  1.60   mlelstv 	if (dk->dk_openmask == 0) {
    192  1.60   mlelstv 		if (dkd->d_lastclose != NULL)
    193  1.60   mlelstv 			(*dkd->d_lastclose)(dksc->sc_dev);
    194  1.60   mlelstv 	}
    195  1.60   mlelstv 
    196  1.60   mlelstv 	if ((dksc->sc_flags & DKF_KLABEL) == 0)
    197  1.60   mlelstv 		dksc->sc_flags &= ~DKF_VLABEL;
    198  1.60   mlelstv 
    199  1.30        ad 	mutex_exit(&dk->dk_openlock);
    200   1.1     elric 	return 0;
    201   1.1     elric }
    202   1.1     elric 
    203   1.1     elric void
    204  1.60   mlelstv dk_strategy(struct dk_softc *dksc, struct buf *bp)
    205   1.1     elric {
    206  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    207  1.55   mlelstv 	int	s, part;
    208   1.1     elric 	int	wlabel;
    209  1.18      yamt 	daddr_t	blkno;
    210  1.55   mlelstv 	struct disklabel *lp;
    211  1.55   mlelstv 	struct disk *dk;
    212  1.55   mlelstv 	uint64_t numsecs;
    213  1.55   mlelstv 	unsigned secsize;
    214   1.1     elric 
    215   1.1     elric 	DPRINTF_FOLLOW(("dk_strategy(%s, %p, %p)\n",
    216  1.60   mlelstv 	    dksc->sc_xname, dksc, bp));
    217   1.1     elric 
    218   1.1     elric 	if (!(dksc->sc_flags & DKF_INITED)) {
    219  1.25       dan 		DPRINTF_FOLLOW(("dk_strategy: not inited\n"));
    220   1.1     elric 		bp->b_error  = ENXIO;
    221   1.2     elric 		biodone(bp);
    222   1.1     elric 		return;
    223   1.1     elric 	}
    224   1.1     elric 
    225  1.55   mlelstv 	lp = dksc->sc_dkdev.dk_label;
    226  1.55   mlelstv 	dk = &dksc->sc_dkdev;
    227  1.55   mlelstv 
    228  1.55   mlelstv 	part = DISKPART(bp->b_dev);
    229  1.55   mlelstv 	numsecs = dk->dk_geom.dg_secperunit;
    230  1.55   mlelstv 	secsize = dk->dk_geom.dg_secsize;
    231   1.1     elric 
    232   1.1     elric 	bp->b_resid = bp->b_bcount;
    233   1.1     elric 
    234  1.55   mlelstv 	/*
    235  1.55   mlelstv 	 * The transfer must be a whole number of blocks and the offset must
    236  1.55   mlelstv 	 * not be negative.
    237  1.55   mlelstv 	 */
    238  1.55   mlelstv 	if ((bp->b_bcount % secsize) != 0 || bp->b_blkno < 0) {
    239  1.55   mlelstv 		bp->b_error = EINVAL;
    240  1.55   mlelstv 		biodone(bp);
    241  1.55   mlelstv 		return;
    242  1.55   mlelstv 	}
    243  1.55   mlelstv 
    244   1.1     elric 	/* If there is nothing to do, then we are done */
    245   1.1     elric 	if (bp->b_bcount == 0) {
    246   1.1     elric 		biodone(bp);
    247   1.1     elric 		return;
    248   1.1     elric 	}
    249   1.1     elric 
    250   1.1     elric 	wlabel = dksc->sc_flags & (DKF_WLABEL|DKF_LABELLING);
    251  1.55   mlelstv 	if (part == RAW_PART) {
    252  1.55   mlelstv 		if (bounds_check_with_mediasize(bp, DEV_BSIZE, numsecs) <= 0) {
    253  1.55   mlelstv 			biodone(bp);
    254  1.55   mlelstv 			return;
    255  1.55   mlelstv 		}
    256  1.55   mlelstv 	} else {
    257  1.55   mlelstv 		if (bounds_check_with_label(&dksc->sc_dkdev, bp, wlabel) <= 0) {
    258  1.55   mlelstv 			biodone(bp);
    259  1.55   mlelstv 			return;
    260  1.55   mlelstv 		}
    261   1.1     elric 	}
    262   1.1     elric 
    263  1.18      yamt 	blkno = bp->b_blkno;
    264  1.55   mlelstv 	if (part != RAW_PART)
    265  1.55   mlelstv 		blkno += lp->d_partitions[DISKPART(bp->b_dev)].p_offset;
    266  1.18      yamt 	bp->b_rawblkno = blkno;
    267  1.18      yamt 
    268   1.1     elric 	/*
    269   1.1     elric 	 * Start the unit by calling the start routine
    270   1.1     elric 	 * provided by the individual driver.
    271   1.1     elric 	 */
    272   1.1     elric 	s = splbio();
    273  1.40      yamt 	bufq_put(dksc->sc_bufq, bp);
    274  1.60   mlelstv 	dkd->d_diskstart(dksc->sc_dev);
    275   1.1     elric 	splx(s);
    276   1.1     elric 	return;
    277   1.1     elric }
    278   1.1     elric 
    279  1.60   mlelstv void
    280  1.60   mlelstv dk_done(struct dk_softc *dksc, struct buf *bp)
    281  1.60   mlelstv {
    282  1.60   mlelstv 	struct disk *dk = &dksc->sc_dkdev;
    283  1.60   mlelstv 
    284  1.60   mlelstv 	if (bp->b_error != 0) {
    285  1.60   mlelstv 		diskerr(bp, dksc->sc_xname, "error", LOG_PRINTF, 0,
    286  1.60   mlelstv 			dk->dk_label);
    287  1.60   mlelstv 		printf("\n");
    288  1.60   mlelstv 	}
    289  1.60   mlelstv 
    290  1.60   mlelstv 	disk_unbusy(dk, bp->b_bcount - bp->b_resid, (bp->b_flags & B_READ));
    291  1.60   mlelstv #ifdef notyet
    292  1.60   mlelstv 	rnd_add_uint(&dksc->sc_rnd_source, bp->b_rawblkno);
    293  1.60   mlelstv #endif
    294  1.60   mlelstv 	biodone(bp);
    295  1.60   mlelstv }
    296  1.60   mlelstv 
    297   1.1     elric int
    298  1.60   mlelstv dk_size(struct dk_softc *dksc, dev_t dev)
    299   1.1     elric {
    300  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    301   1.1     elric 	struct	disklabel *lp;
    302   1.1     elric 	int	is_open;
    303   1.1     elric 	int	part;
    304   1.1     elric 	int	size;
    305   1.1     elric 
    306   1.1     elric 	if ((dksc->sc_flags & DKF_INITED) == 0)
    307   1.1     elric 		return -1;
    308   1.1     elric 
    309   1.1     elric 	part = DISKPART(dev);
    310   1.1     elric 	is_open = dksc->sc_dkdev.dk_openmask & (1 << part);
    311   1.1     elric 
    312  1.60   mlelstv 	if (!is_open && dkd->d_open(dev, 0, S_IFBLK, curlwp))
    313   1.1     elric 		return -1;
    314   1.1     elric 
    315   1.1     elric 	lp = dksc->sc_dkdev.dk_label;
    316   1.1     elric 	if (lp->d_partitions[part].p_fstype != FS_SWAP)
    317   1.1     elric 		size = -1;
    318   1.1     elric 	else
    319   1.1     elric 		size = lp->d_partitions[part].p_size *
    320   1.1     elric 		    (lp->d_secsize / DEV_BSIZE);
    321   1.1     elric 
    322  1.60   mlelstv 	if (!is_open && dkd->d_close(dev, 0, S_IFBLK, curlwp))
    323  1.59   mlelstv 		return -1;
    324   1.1     elric 
    325   1.1     elric 	return size;
    326   1.1     elric }
    327   1.1     elric 
    328   1.1     elric int
    329  1.60   mlelstv dk_ioctl(struct dk_softc *dksc, dev_t dev,
    330  1.28  christos 	    u_long cmd, void *data, int flag, struct lwp *l)
    331   1.1     elric {
    332  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    333   1.1     elric 	struct	disklabel *lp;
    334  1.57  christos 	struct	disk *dk = &dksc->sc_dkdev;
    335   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    336   1.1     elric 	struct	disklabel newlabel;
    337   1.1     elric #endif
    338  1.57  christos 	int	error;
    339   1.1     elric 
    340  1.39    cegger 	DPRINTF_FOLLOW(("dk_ioctl(%s, %p, 0x%"PRIx64", 0x%lx)\n",
    341  1.60   mlelstv 	    dksc->sc_xname, dksc, dev, cmd));
    342   1.1     elric 
    343   1.1     elric 	/* ensure that the pseudo disk is open for writes for these commands */
    344   1.1     elric 	switch (cmd) {
    345   1.1     elric 	case DIOCSDINFO:
    346   1.1     elric 	case DIOCWDINFO:
    347   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    348   1.1     elric 	case ODIOCSDINFO:
    349   1.1     elric 	case ODIOCWDINFO:
    350   1.1     elric #endif
    351   1.1     elric 	case DIOCWLABEL:
    352  1.43     elric 	case DIOCAWEDGE:
    353  1.43     elric 	case DIOCDWEDGE:
    354   1.1     elric 		if ((flag & FWRITE) == 0)
    355   1.1     elric 			return EBADF;
    356   1.1     elric 	}
    357   1.1     elric 
    358   1.1     elric 	/* ensure that the pseudo-disk is initialized for these */
    359   1.1     elric 	switch (cmd) {
    360   1.1     elric 	case DIOCGDINFO:
    361   1.1     elric 	case DIOCSDINFO:
    362   1.1     elric 	case DIOCWDINFO:
    363   1.1     elric 	case DIOCGPART:
    364  1.60   mlelstv 	case DIOCKLABEL:
    365   1.1     elric 	case DIOCWLABEL:
    366   1.1     elric 	case DIOCGDEFLABEL:
    367  1.43     elric 	case DIOCAWEDGE:
    368  1.43     elric 	case DIOCDWEDGE:
    369  1.43     elric 	case DIOCLWEDGES:
    370  1.54   mlelstv 	case DIOCMWEDGES:
    371  1.43     elric 	case DIOCCACHESYNC:
    372   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    373   1.1     elric 	case ODIOCGDINFO:
    374   1.1     elric 	case ODIOCSDINFO:
    375   1.1     elric 	case ODIOCWDINFO:
    376   1.1     elric 	case ODIOCGDEFLABEL:
    377   1.1     elric #endif
    378   1.1     elric 		if ((dksc->sc_flags & DKF_INITED) == 0)
    379   1.1     elric 			return ENXIO;
    380   1.1     elric 	}
    381   1.1     elric 
    382  1.58  christos 	error = disk_ioctl(dk, dev, cmd, data, flag, l);
    383  1.57  christos 	if (error != EPASSTHROUGH)
    384  1.57  christos 		return error;
    385  1.57  christos 	else
    386  1.57  christos 		error = 0;
    387  1.57  christos 
    388   1.1     elric 	switch (cmd) {
    389   1.1     elric 	case DIOCWDINFO:
    390   1.1     elric 	case DIOCSDINFO:
    391   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    392   1.1     elric 	case ODIOCWDINFO:
    393   1.1     elric 	case ODIOCSDINFO:
    394   1.1     elric #endif
    395   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    396   1.1     elric 		if (cmd == ODIOCSDINFO || cmd == ODIOCWDINFO) {
    397   1.1     elric 			memset(&newlabel, 0, sizeof newlabel);
    398   1.1     elric 			memcpy(&newlabel, data, sizeof (struct olddisklabel));
    399   1.1     elric 			lp = &newlabel;
    400   1.1     elric 		} else
    401   1.1     elric #endif
    402   1.1     elric 		lp = (struct disklabel *)data;
    403   1.1     elric 
    404  1.30        ad 		mutex_enter(&dk->dk_openlock);
    405   1.1     elric 		dksc->sc_flags |= DKF_LABELLING;
    406   1.1     elric 
    407   1.1     elric 		error = setdisklabel(dksc->sc_dkdev.dk_label,
    408   1.1     elric 		    lp, 0, dksc->sc_dkdev.dk_cpulabel);
    409   1.1     elric 		if (error == 0) {
    410   1.1     elric 			if (cmd == DIOCWDINFO
    411   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    412   1.1     elric 			    || cmd == ODIOCWDINFO
    413   1.1     elric #endif
    414   1.1     elric 			   )
    415   1.1     elric 				error = writedisklabel(DKLABELDEV(dev),
    416  1.60   mlelstv 				    dkd->d_strategy, dksc->sc_dkdev.dk_label,
    417   1.1     elric 				    dksc->sc_dkdev.dk_cpulabel);
    418   1.1     elric 		}
    419   1.1     elric 
    420   1.1     elric 		dksc->sc_flags &= ~DKF_LABELLING;
    421  1.30        ad 		mutex_exit(&dk->dk_openlock);
    422   1.1     elric 		break;
    423   1.1     elric 
    424  1.60   mlelstv 	case DIOCKLABEL:
    425  1.60   mlelstv 		if ((flag & FWRITE) == 0)
    426  1.60   mlelstv 			return (EBADF);
    427  1.60   mlelstv 		if (*(int *)data != 0)
    428  1.60   mlelstv 			dksc->sc_flags |= DKF_KLABEL;
    429  1.60   mlelstv 		else
    430  1.60   mlelstv 			dksc->sc_flags &= ~DKF_KLABEL;
    431  1.60   mlelstv 		break;
    432  1.60   mlelstv 
    433   1.1     elric 	case DIOCWLABEL:
    434   1.1     elric 		if (*(int *)data != 0)
    435   1.1     elric 			dksc->sc_flags |= DKF_WLABEL;
    436   1.1     elric 		else
    437   1.1     elric 			dksc->sc_flags &= ~DKF_WLABEL;
    438   1.1     elric 		break;
    439   1.1     elric 
    440   1.1     elric 	case DIOCGDEFLABEL:
    441  1.60   mlelstv 		dk_getdefaultlabel(dksc, (struct disklabel *)data);
    442   1.1     elric 		break;
    443   1.1     elric 
    444   1.1     elric #ifdef __HAVE_OLD_DISKLABEL
    445   1.1     elric 	case ODIOCGDEFLABEL:
    446  1.60   mlelstv 		dk_getdefaultlabel(dksc, &newlabel);
    447   1.1     elric 		if (newlabel.d_npartitions > OLDMAXPARTITIONS)
    448   1.1     elric 			return ENOTTY;
    449   1.1     elric 		memcpy(data, &newlabel, sizeof (struct olddisklabel));
    450   1.1     elric 		break;
    451   1.1     elric #endif
    452   1.1     elric 
    453  1.21      yamt 	case DIOCGSTRATEGY:
    454  1.21      yamt 	    {
    455  1.21      yamt 		struct disk_strategy *dks = (void *)data;
    456  1.21      yamt 		int s;
    457  1.21      yamt 
    458  1.21      yamt 		s = splbio();
    459  1.21      yamt 		strlcpy(dks->dks_name, bufq_getstrategyname(dksc->sc_bufq),
    460  1.21      yamt 		    sizeof(dks->dks_name));
    461  1.21      yamt 		splx(s);
    462  1.21      yamt 		dks->dks_paramlen = 0;
    463  1.21      yamt 
    464  1.21      yamt 		return 0;
    465  1.21      yamt 	    }
    466  1.21      yamt 
    467  1.21      yamt 	case DIOCSSTRATEGY:
    468  1.21      yamt 	    {
    469  1.21      yamt 		struct disk_strategy *dks = (void *)data;
    470  1.21      yamt 		struct bufq_state *new;
    471  1.21      yamt 		struct bufq_state *old;
    472  1.21      yamt 		int s;
    473  1.21      yamt 
    474  1.21      yamt 		if ((flag & FWRITE) == 0) {
    475  1.21      yamt 			return EBADF;
    476  1.21      yamt 		}
    477  1.21      yamt 		if (dks->dks_param != NULL) {
    478  1.21      yamt 			return EINVAL;
    479  1.21      yamt 		}
    480  1.21      yamt 		dks->dks_name[sizeof(dks->dks_name) - 1] = 0; /* ensure term */
    481  1.21      yamt 		error = bufq_alloc(&new, dks->dks_name,
    482  1.21      yamt 		    BUFQ_EXACT|BUFQ_SORT_RAWBLOCK);
    483  1.21      yamt 		if (error) {
    484  1.21      yamt 			return error;
    485  1.21      yamt 		}
    486  1.21      yamt 		s = splbio();
    487  1.21      yamt 		old = dksc->sc_bufq;
    488  1.21      yamt 		bufq_move(new, old);
    489  1.21      yamt 		dksc->sc_bufq = new;
    490  1.21      yamt 		splx(s);
    491  1.21      yamt 		bufq_free(old);
    492  1.21      yamt 
    493  1.21      yamt 		return 0;
    494  1.21      yamt 	    }
    495  1.21      yamt 
    496   1.1     elric 	default:
    497   1.1     elric 		error = ENOTTY;
    498   1.1     elric 	}
    499   1.1     elric 
    500   1.1     elric 	return error;
    501   1.1     elric }
    502   1.1     elric 
    503   1.1     elric /*
    504   1.1     elric  * dk_dump dumps all of physical memory into the partition specified.
    505   1.1     elric  * This requires substantially more framework than {s,w}ddump, and hence
    506   1.1     elric  * is probably much more fragile.
    507   1.1     elric  *
    508   1.1     elric  */
    509   1.1     elric 
    510   1.1     elric #define DKF_READYFORDUMP	(DKF_INITED|DKF_TAKEDUMP)
    511   1.1     elric #define DKFF_READYFORDUMP(x)	(((x) & DKF_READYFORDUMP) == DKF_READYFORDUMP)
    512   1.1     elric static volatile int	dk_dumping = 0;
    513   1.1     elric 
    514   1.1     elric /* ARGSUSED */
    515   1.1     elric int
    516  1.60   mlelstv dk_dump(struct dk_softc *dksc, dev_t dev,
    517  1.60   mlelstv     daddr_t blkno, void *vav, size_t size)
    518   1.1     elric {
    519  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    520  1.60   mlelstv 	char *va = vav;
    521  1.60   mlelstv 	struct disklabel *lp;
    522  1.60   mlelstv 	int part, towrt, nsects, sectoff, maxblkcnt, nblk;
    523  1.60   mlelstv 	int maxxfer, rv = 0;
    524   1.1     elric 
    525   1.1     elric 	/*
    526   1.1     elric 	 * ensure that we consider this device to be safe for dumping,
    527   1.1     elric 	 * and that the device is configured.
    528   1.1     elric 	 */
    529   1.1     elric 	if (!DKFF_READYFORDUMP(dksc->sc_flags))
    530   1.1     elric 		return ENXIO;
    531   1.1     elric 
    532   1.1     elric 	/* ensure that we are not already dumping */
    533   1.1     elric 	if (dk_dumping)
    534   1.1     elric 		return EFAULT;
    535   1.1     elric 	dk_dumping = 1;
    536   1.1     elric 
    537  1.60   mlelstv 	if (dkd->d_dumpblocks == NULL)
    538  1.60   mlelstv 		return ENXIO;
    539  1.60   mlelstv 
    540  1.60   mlelstv 	/* device specific max transfer size */
    541  1.60   mlelstv 	maxxfer = MAXPHYS;
    542  1.60   mlelstv 	if (dkd->d_iosize != NULL)
    543  1.60   mlelstv 		(*dkd->d_iosize)(dksc->sc_dev, &maxxfer);
    544  1.60   mlelstv 
    545  1.60   mlelstv 	/* Convert to disk sectors.  Request must be a multiple of size. */
    546  1.60   mlelstv 	part = DISKPART(dev);
    547  1.60   mlelstv 	lp = dksc->sc_dkdev.dk_label;
    548  1.60   mlelstv 	if ((size % lp->d_secsize) != 0)
    549  1.60   mlelstv 		return (EFAULT);
    550  1.60   mlelstv 	towrt = size / lp->d_secsize;
    551  1.60   mlelstv 	blkno = dbtob(blkno) / lp->d_secsize;   /* blkno in secsize units */
    552  1.60   mlelstv 
    553  1.60   mlelstv 	nsects = lp->d_partitions[part].p_size;
    554  1.60   mlelstv 	sectoff = lp->d_partitions[part].p_offset;
    555  1.60   mlelstv 
    556  1.60   mlelstv 	/* Check transfer bounds against partition size. */
    557  1.60   mlelstv 	if ((blkno < 0) || ((blkno + towrt) > nsects))
    558  1.60   mlelstv 		return (EINVAL);
    559  1.60   mlelstv 
    560  1.60   mlelstv 	/* Offset block number to start of partition. */
    561  1.60   mlelstv 	blkno += sectoff;
    562  1.60   mlelstv 
    563  1.60   mlelstv 	/* Start dumping and return when done. */
    564  1.60   mlelstv 	maxblkcnt = howmany(maxxfer, lp->d_secsize);
    565  1.60   mlelstv 	while (towrt > 0) {
    566  1.60   mlelstv 		nblk = min(maxblkcnt, towrt);
    567  1.60   mlelstv 
    568  1.60   mlelstv 		if ((rv = (*dkd->d_dumpblocks)(dksc->sc_dev, va, blkno, nblk)) != 0)
    569  1.60   mlelstv 			return (rv);
    570  1.60   mlelstv 
    571  1.60   mlelstv 		towrt -= nblk;
    572  1.60   mlelstv 		blkno += nblk;
    573  1.60   mlelstv 		va += nblk * lp->d_secsize;
    574  1.60   mlelstv 	}
    575   1.1     elric 
    576   1.1     elric 	dk_dumping = 0;
    577   1.1     elric 
    578  1.60   mlelstv 	return 0;
    579   1.1     elric }
    580   1.1     elric 
    581   1.1     elric /* ARGSUSED */
    582   1.1     elric void
    583  1.60   mlelstv dk_getdefaultlabel(struct dk_softc *dksc,
    584   1.1     elric 		      struct disklabel *lp)
    585   1.1     elric {
    586  1.47  christos 	struct disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    587   1.4     elric 
    588   1.4     elric 	memset(lp, 0, sizeof(*lp));
    589   1.1     elric 
    590  1.52   mlelstv 	if (dg->dg_secperunit > UINT32_MAX)
    591  1.52   mlelstv 		lp->d_secperunit = UINT32_MAX;
    592  1.52   mlelstv 	else
    593  1.52   mlelstv 		lp->d_secperunit = dg->dg_secperunit;
    594  1.47  christos 	lp->d_secsize = dg->dg_secsize;
    595  1.47  christos 	lp->d_nsectors = dg->dg_nsectors;
    596  1.47  christos 	lp->d_ntracks = dg->dg_ntracks;
    597  1.47  christos 	lp->d_ncylinders = dg->dg_ncylinders;
    598   1.1     elric 	lp->d_secpercyl = lp->d_ntracks * lp->d_nsectors;
    599   1.1     elric 
    600  1.60   mlelstv 	strncpy(lp->d_typename, dksc->sc_xname, sizeof(lp->d_typename));
    601  1.60   mlelstv 	lp->d_type = dksc->sc_dtype;
    602   1.1     elric 	strncpy(lp->d_packname, "fictitious", sizeof(lp->d_packname));
    603   1.1     elric 	lp->d_rpm = 3600;
    604   1.1     elric 	lp->d_interleave = 1;
    605   1.1     elric 	lp->d_flags = 0;
    606   1.1     elric 
    607   1.1     elric 	lp->d_partitions[RAW_PART].p_offset = 0;
    608  1.52   mlelstv 	lp->d_partitions[RAW_PART].p_size = lp->d_secperunit;
    609   1.1     elric 	lp->d_partitions[RAW_PART].p_fstype = FS_UNUSED;
    610   1.1     elric 	lp->d_npartitions = RAW_PART + 1;
    611   1.1     elric 
    612   1.1     elric 	lp->d_magic = DISKMAGIC;
    613   1.1     elric 	lp->d_magic2 = DISKMAGIC;
    614   1.1     elric 	lp->d_checksum = dkcksum(dksc->sc_dkdev.dk_label);
    615   1.1     elric }
    616   1.1     elric 
    617   1.1     elric /* ARGSUSED */
    618   1.1     elric void
    619  1.60   mlelstv dk_getdisklabel(struct dk_softc *dksc, dev_t dev)
    620   1.1     elric {
    621  1.60   mlelstv 	const struct dkdriver *dkd = dksc->sc_dkdev.dk_driver;
    622   1.1     elric 	struct	 disklabel *lp = dksc->sc_dkdev.dk_label;
    623   1.1     elric 	struct	 cpu_disklabel *clp = dksc->sc_dkdev.dk_cpulabel;
    624  1.60   mlelstv 	struct   disk_geom *dg = &dksc->sc_dkdev.dk_geom;
    625   1.1     elric 	struct	 partition *pp;
    626   1.1     elric 	int	 i;
    627   1.5       dsl 	const char	*errstring;
    628   1.1     elric 
    629   1.1     elric 	memset(clp, 0x0, sizeof(*clp));
    630  1.60   mlelstv 	dk_getdefaultlabel(dksc, lp);
    631  1.60   mlelstv 	errstring = readdisklabel(DKLABELDEV(dev), dkd->d_strategy,
    632   1.1     elric 	    dksc->sc_dkdev.dk_label, dksc->sc_dkdev.dk_cpulabel);
    633   1.1     elric 	if (errstring) {
    634  1.60   mlelstv 		dk_makedisklabel(dksc);
    635   1.1     elric 		if (dksc->sc_flags & DKF_WARNLABEL)
    636   1.1     elric 			printf("%s: %s\n", dksc->sc_xname, errstring);
    637   1.1     elric 		return;
    638   1.1     elric 	}
    639   1.1     elric 
    640   1.1     elric 	if ((dksc->sc_flags & DKF_LABELSANITY) == 0)
    641   1.1     elric 		return;
    642   1.1     elric 
    643   1.1     elric 	/* Sanity check */
    644  1.53   mlelstv 	if (lp->d_secperunit < UINT32_MAX ?
    645  1.53   mlelstv 		lp->d_secperunit != dg->dg_secperunit :
    646  1.53   mlelstv 		lp->d_secperunit > dg->dg_secperunit)
    647  1.53   mlelstv 		printf("WARNING: %s: total sector size in disklabel (%ju) "
    648  1.53   mlelstv 		    "!= the size of %s (%ju)\n", dksc->sc_xname,
    649  1.60   mlelstv 		    (uintmax_t)lp->d_secperunit, dksc->sc_xname,
    650  1.53   mlelstv 		    (uintmax_t)dg->dg_secperunit);
    651   1.1     elric 
    652   1.1     elric 	for (i=0; i < lp->d_npartitions; i++) {
    653   1.1     elric 		pp = &lp->d_partitions[i];
    654  1.48  christos 		if (pp->p_offset + pp->p_size > dg->dg_secperunit)
    655   1.1     elric 			printf("WARNING: %s: end of partition `%c' exceeds "
    656  1.53   mlelstv 			    "the size of %s (%ju)\n", dksc->sc_xname,
    657  1.60   mlelstv 			    'a' + i, dksc->sc_xname,
    658  1.53   mlelstv 			    (uintmax_t)dg->dg_secperunit);
    659   1.1     elric 	}
    660   1.1     elric }
    661   1.1     elric 
    662   1.1     elric /* ARGSUSED */
    663  1.13   thorpej static void
    664  1.60   mlelstv dk_makedisklabel(struct dk_softc *dksc)
    665   1.1     elric {
    666   1.1     elric 	struct	disklabel *lp = dksc->sc_dkdev.dk_label;
    667   1.1     elric 
    668   1.1     elric 	lp->d_partitions[RAW_PART].p_fstype = FS_BSDFFS;
    669   1.1     elric 	strncpy(lp->d_packname, "default label", sizeof(lp->d_packname));
    670   1.1     elric 	lp->d_checksum = dkcksum(lp);
    671   1.1     elric }
    672   1.1     elric 
    673   1.1     elric /* This function is taken from ccd.c:1.76  --rcd */
    674   1.1     elric 
    675   1.1     elric /*
    676   1.1     elric  * XXX this function looks too generic for dksubr.c, shouldn't we
    677   1.1     elric  *     put it somewhere better?
    678   1.1     elric  */
    679   1.1     elric 
    680   1.1     elric /*
    681   1.1     elric  * Lookup the provided name in the filesystem.  If the file exists,
    682   1.1     elric  * is a valid block device, and isn't being used by anyone else,
    683   1.1     elric  * set *vpp to the file's vnode.
    684   1.1     elric  */
    685   1.1     elric int
    686  1.42  dholland dk_lookup(struct pathbuf *pb, struct lwp *l, struct vnode **vpp)
    687   1.1     elric {
    688   1.1     elric 	struct nameidata nd;
    689   1.1     elric 	struct vnode *vp;
    690  1.24  christos 	int     error;
    691   1.1     elric 
    692  1.24  christos 	if (l == NULL)
    693  1.24  christos 		return ESRCH;	/* Is ESRCH the best choice? */
    694  1.24  christos 
    695  1.42  dholland 	NDINIT(&nd, LOOKUP, FOLLOW, pb);
    696  1.24  christos 	if ((error = vn_open(&nd, FREAD | FWRITE, 0)) != 0) {
    697   1.1     elric 		DPRINTF((DKDB_FOLLOW|DKDB_INIT),
    698   1.1     elric 		    ("dk_lookup: vn_open error = %d\n", error));
    699  1.24  christos 		return error;
    700   1.1     elric 	}
    701  1.24  christos 
    702   1.1     elric 	vp = nd.ni_vp;
    703  1.51   hannken 	if (vp->v_type != VBLK) {
    704  1.51   hannken 		error = ENOTBLK;
    705  1.24  christos 		goto out;
    706   1.1     elric 	}
    707   1.1     elric 
    708  1.51   hannken 	/* Reopen as anonymous vnode to protect against forced unmount. */
    709  1.51   hannken 	if ((error = bdevvp(vp->v_rdev, vpp)) != 0)
    710  1.24  christos 		goto out;
    711  1.51   hannken 	VOP_UNLOCK(vp);
    712  1.51   hannken 	if ((error = vn_close(vp, FREAD | FWRITE, l->l_cred)) != 0) {
    713  1.51   hannken 		vrele(*vpp);
    714  1.51   hannken 		return error;
    715  1.51   hannken 	}
    716  1.51   hannken 	if ((error = VOP_OPEN(*vpp, FREAD | FWRITE, l->l_cred)) != 0) {
    717  1.51   hannken 		vrele(*vpp);
    718  1.51   hannken 		return error;
    719  1.24  christos 	}
    720  1.51   hannken 	mutex_enter((*vpp)->v_interlock);
    721  1.51   hannken 	(*vpp)->v_writecount++;
    722  1.51   hannken 	mutex_exit((*vpp)->v_interlock);
    723  1.24  christos 
    724  1.51   hannken 	IFDEBUG(DKDB_VNODE, vprint("dk_lookup: vnode info", *vpp));
    725   1.1     elric 
    726  1.24  christos 	return 0;
    727  1.24  christos out:
    728  1.41   hannken 	VOP_UNLOCK(vp);
    729  1.35        ad 	(void) vn_close(vp, FREAD | FWRITE, l->l_cred);
    730  1.24  christos 	return error;
    731   1.1     elric }
    732  1.49  pgoyette 
    733  1.49  pgoyette MODULE(MODULE_CLASS_MISC, dk_subr, NULL);
    734  1.49  pgoyette 
    735  1.49  pgoyette static int
    736  1.49  pgoyette dk_subr_modcmd(modcmd_t cmd, void *arg)
    737  1.49  pgoyette {
    738  1.49  pgoyette 	switch (cmd) {
    739  1.49  pgoyette 	case MODULE_CMD_INIT:
    740  1.49  pgoyette 	case MODULE_CMD_FINI:
    741  1.49  pgoyette 		return 0;
    742  1.49  pgoyette 	case MODULE_CMD_STAT:
    743  1.49  pgoyette 	case MODULE_CMD_AUTOUNLOAD:
    744  1.49  pgoyette 	default:
    745  1.49  pgoyette 		return ENOTTY;
    746  1.49  pgoyette 	}
    747  1.49  pgoyette }
    748