Home | History | Annotate | Line # | Download | only in dev
cgd.c revision 1.23
      1  1.23      yamt /* $NetBSD: cgd.c,v 1.23 2005/03/31 11:28:53 yamt Exp $ */
      2   1.1     elric 
      3   1.1     elric /*-
      4   1.1     elric  * Copyright (c) 2002 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 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  * 3. All advertising materials mentioning features or use of this software
     19   1.1     elric  *    must display the following acknowledgement:
     20   1.1     elric  *        This product includes software developed by the NetBSD
     21   1.1     elric  *        Foundation, Inc. and its contributors.
     22   1.1     elric  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23   1.1     elric  *    contributors may be used to endorse or promote products derived
     24   1.1     elric  *    from this software without specific prior written permission.
     25   1.1     elric  *
     26   1.1     elric  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27   1.1     elric  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28   1.1     elric  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29   1.1     elric  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30   1.1     elric  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31   1.1     elric  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32   1.1     elric  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33   1.1     elric  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34   1.1     elric  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35   1.1     elric  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36   1.1     elric  * POSSIBILITY OF SUCH DAMAGE.
     37   1.1     elric  */
     38   1.1     elric 
     39   1.1     elric #include <sys/cdefs.h>
     40  1.23      yamt __KERNEL_RCSID(0, "$NetBSD: cgd.c,v 1.23 2005/03/31 11:28:53 yamt Exp $");
     41   1.1     elric 
     42   1.1     elric #include <sys/types.h>
     43   1.1     elric #include <sys/param.h>
     44   1.1     elric #include <sys/systm.h>
     45   1.1     elric #include <sys/proc.h>
     46   1.1     elric #include <sys/errno.h>
     47   1.1     elric #include <sys/buf.h>
     48  1.21      yamt #include <sys/bufq.h>
     49   1.1     elric #include <sys/malloc.h>
     50   1.1     elric #include <sys/pool.h>
     51   1.1     elric #include <sys/ioctl.h>
     52   1.1     elric #include <sys/device.h>
     53   1.1     elric #include <sys/disk.h>
     54   1.1     elric #include <sys/disklabel.h>
     55   1.1     elric #include <sys/fcntl.h>
     56   1.1     elric #include <sys/vnode.h>
     57   1.1     elric #include <sys/lock.h>
     58   1.1     elric #include <sys/conf.h>
     59   1.1     elric 
     60   1.1     elric #include <dev/dkvar.h>
     61   1.1     elric #include <dev/cgdvar.h>
     62   1.1     elric 
     63   1.1     elric /* Entry Point Functions */
     64   1.1     elric 
     65   1.1     elric void	cgdattach(int);
     66   1.1     elric 
     67  1.18   thorpej static dev_type_open(cgdopen);
     68  1.18   thorpej static dev_type_close(cgdclose);
     69  1.18   thorpej static dev_type_read(cgdread);
     70  1.18   thorpej static dev_type_write(cgdwrite);
     71  1.18   thorpej static dev_type_ioctl(cgdioctl);
     72  1.18   thorpej static dev_type_strategy(cgdstrategy);
     73  1.18   thorpej static dev_type_dump(cgddump);
     74  1.18   thorpej static dev_type_size(cgdsize);
     75   1.1     elric 
     76   1.1     elric const struct bdevsw cgd_bdevsw = {
     77   1.1     elric 	cgdopen, cgdclose, cgdstrategy, cgdioctl,
     78   1.1     elric 	cgddump, cgdsize, D_DISK
     79   1.1     elric };
     80   1.1     elric 
     81   1.1     elric const struct cdevsw cgd_cdevsw = {
     82   1.1     elric 	cgdopen, cgdclose, cgdread, cgdwrite, cgdioctl,
     83   1.4  jdolecek 	nostop, notty, nopoll, nommap, nokqfilter, D_DISK
     84   1.1     elric };
     85   1.1     elric 
     86   1.1     elric /* Internal Functions */
     87   1.1     elric 
     88  1.16     elric static int	cgdstart(struct dk_softc *, struct buf *);
     89   1.1     elric static void	cgdiodone(struct buf *);
     90   1.1     elric 
     91  1.12      fvdl static int	cgd_ioctl_set(struct cgd_softc *, void *, struct proc *);
     92  1.12      fvdl static int	cgd_ioctl_clr(struct cgd_softc *, void *, struct proc *);
     93   1.1     elric static int	cgdinit(struct cgd_softc *, char *, struct vnode *,
     94  1.12      fvdl 			struct proc *);
     95   1.1     elric static void	cgd_cipher(struct cgd_softc *, caddr_t, caddr_t,
     96   1.1     elric 			   size_t, daddr_t, size_t, int);
     97   1.1     elric 
     98   1.1     elric /* Pseudo-disk Interface */
     99   1.1     elric 
    100   1.1     elric static struct dk_intf the_dkintf = {
    101   1.1     elric 	DTYPE_CGD,
    102   1.1     elric 	"cgd",
    103   1.1     elric 	cgdopen,
    104   1.1     elric 	cgdclose,
    105   1.1     elric 	cgdstrategy,
    106   1.1     elric 	cgdstart,
    107   1.1     elric };
    108   1.1     elric static struct dk_intf *di = &the_dkintf;
    109   1.1     elric 
    110   1.1     elric /* DIAGNOSTIC and DEBUG definitions */
    111   1.1     elric 
    112   1.1     elric #if defined(CGDDEBUG) && !defined(DEBUG)
    113   1.1     elric #define DEBUG
    114   1.1     elric #endif
    115   1.1     elric 
    116   1.1     elric #ifdef DEBUG
    117   1.1     elric int cgddebug = 0;
    118   1.1     elric 
    119   1.1     elric #define CGDB_FOLLOW	0x1
    120   1.1     elric #define CGDB_IO	0x2
    121   1.1     elric #define CGDB_CRYPTO	0x4
    122   1.1     elric 
    123   1.1     elric #define IFDEBUG(x,y)		if (cgddebug & (x)) y
    124   1.1     elric #define DPRINTF(x,y)		IFDEBUG(x, printf y)
    125   1.1     elric #define DPRINTF_FOLLOW(y)	DPRINTF(CGDB_FOLLOW, y)
    126   1.1     elric 
    127   1.1     elric static void	hexprint(char *, void *, int);
    128   1.1     elric 
    129   1.1     elric #else
    130   1.1     elric #define IFDEBUG(x,y)
    131   1.1     elric #define DPRINTF(x,y)
    132   1.1     elric #define DPRINTF_FOLLOW(y)
    133   1.1     elric #endif
    134   1.1     elric 
    135   1.1     elric #ifdef DIAGNOSTIC
    136  1.22     perry #define DIAGPANIC(x)		panic x
    137   1.1     elric #define DIAGCONDPANIC(x,y)	if (x) panic y
    138   1.1     elric #else
    139   1.1     elric #define DIAGPANIC(x)
    140   1.1     elric #define DIAGCONDPANIC(x,y)
    141   1.1     elric #endif
    142   1.1     elric 
    143   1.1     elric /* Global variables */
    144   1.1     elric 
    145   1.1     elric struct	cgd_softc *cgd_softc;
    146   1.1     elric int	numcgd = 0;
    147   1.1     elric 
    148   1.1     elric /* Utility Functions */
    149   1.1     elric 
    150   1.1     elric #define CGDUNIT(x)		DISKUNIT(x)
    151   1.1     elric #define GETCGD_SOFTC(_cs, x)	if (!((_cs) = getcgd_softc(x))) return ENXIO
    152   1.1     elric 
    153   1.1     elric static struct cgd_softc *
    154   1.1     elric getcgd_softc(dev_t dev)
    155   1.1     elric {
    156   1.1     elric 	int	unit = CGDUNIT(dev);
    157   1.1     elric 
    158   1.1     elric 	DPRINTF_FOLLOW(("getcgd_softc(0x%x): unit = %d\n", dev, unit));
    159   1.1     elric 	if (unit >= numcgd)
    160   1.1     elric 		return NULL;
    161   1.1     elric 	return &cgd_softc[unit];
    162   1.1     elric }
    163   1.1     elric 
    164   1.1     elric /* The code */
    165   1.1     elric 
    166   1.1     elric static void
    167   1.1     elric cgdsoftc_init(struct cgd_softc *cs, int num)
    168   1.1     elric {
    169   1.1     elric 	char	buf[DK_XNAME_SIZE];
    170   1.1     elric 
    171   1.1     elric 	memset(cs, 0x0, sizeof(*cs));
    172   1.1     elric 	snprintf(buf, DK_XNAME_SIZE, "cgd%d", num);
    173  1.16     elric 	simple_lock_init(&cs->sc_slock);
    174   1.1     elric 	dk_sc_init(&cs->sc_dksc, cs, buf);
    175   1.1     elric }
    176   1.1     elric 
    177   1.1     elric void
    178   1.1     elric cgdattach(int num)
    179   1.1     elric {
    180   1.1     elric 	int	i;
    181   1.1     elric 
    182   1.1     elric 	DPRINTF_FOLLOW(("cgdattach(%d)\n", num));
    183   1.1     elric 	if (num <= 0) {
    184   1.1     elric 		DIAGPANIC(("cgdattach: count <= 0"));
    185   1.1     elric 		return;
    186   1.1     elric 	}
    187   1.1     elric 
    188  1.10       agc 	cgd_softc = (void *)malloc(num * sizeof(*cgd_softc), M_DEVBUF, M_NOWAIT);
    189  1.10       agc 	if (!cgd_softc) {
    190   1.1     elric 		printf("WARNING: unable to malloc(9) memory for crypt disks\n");
    191   1.1     elric 		DIAGPANIC(("cgdattach: cannot malloc(9) enough memory"));
    192   1.1     elric 		return;
    193   1.1     elric 	}
    194   1.1     elric 
    195   1.1     elric 	numcgd = num;
    196   1.1     elric 	for (i=0; i<num; i++)
    197   1.1     elric 		cgdsoftc_init(&cgd_softc[i], i);
    198   1.1     elric }
    199   1.1     elric 
    200  1.18   thorpej static int
    201  1.12      fvdl cgdopen(dev_t dev, int flags, int fmt, struct proc *p)
    202   1.1     elric {
    203   1.1     elric 	struct	cgd_softc *cs;
    204   1.1     elric 
    205   1.1     elric 	DPRINTF_FOLLOW(("cgdopen(%d, %d)\n", dev, flags));
    206   1.1     elric 	GETCGD_SOFTC(cs, dev);
    207  1.12      fvdl 	return dk_open(di, &cs->sc_dksc, dev, flags, fmt, p);
    208   1.1     elric }
    209   1.1     elric 
    210  1.18   thorpej static int
    211  1.12      fvdl cgdclose(dev_t dev, int flags, int fmt, struct proc *p)
    212   1.1     elric {
    213   1.1     elric 	struct	cgd_softc *cs;
    214   1.1     elric 
    215   1.1     elric 	DPRINTF_FOLLOW(("cgdclose(%d, %d)\n", dev, flags));
    216   1.1     elric 	GETCGD_SOFTC(cs, dev);
    217  1.12      fvdl 	return dk_close(di, &cs->sc_dksc, dev, flags, fmt, p);
    218   1.1     elric }
    219   1.1     elric 
    220  1.18   thorpej static void
    221   1.1     elric cgdstrategy(struct buf *bp)
    222   1.1     elric {
    223   1.1     elric 	struct	cgd_softc *cs = getcgd_softc(bp->b_dev);
    224   1.1     elric 
    225   1.1     elric 	DPRINTF_FOLLOW(("cgdstrategy(%p): b_bcount = %ld\n", bp,
    226   1.1     elric 	    (long)bp->b_bcount));
    227   1.1     elric 	/* XXXrcd: Should we test for (cs != NULL)? */
    228   1.1     elric 	dk_strategy(di, &cs->sc_dksc, bp);
    229   1.1     elric 	return;
    230   1.1     elric }
    231   1.1     elric 
    232  1.18   thorpej static int
    233   1.1     elric cgdsize(dev_t dev)
    234   1.1     elric {
    235   1.1     elric 	struct cgd_softc *cs = getcgd_softc(dev);
    236   1.1     elric 
    237   1.1     elric 	DPRINTF_FOLLOW(("cgdsize(%d)\n", dev));
    238   1.1     elric 	if (!cs)
    239   1.1     elric 		return -1;
    240   1.1     elric 	return dk_size(di, &cs->sc_dksc, dev);
    241   1.1     elric }
    242   1.1     elric 
    243  1.16     elric /*
    244  1.16     elric  * cgd_{get,put}data are functions that deal with getting a buffer
    245  1.16     elric  * for the new encrypted data.  We have a buffer per device so that
    246  1.16     elric  * we can ensure that we can always have a transaction in flight.
    247  1.16     elric  * We use this buffer first so that we have one less piece of
    248  1.16     elric  * malloc'ed data at any given point.
    249  1.16     elric  */
    250  1.16     elric 
    251  1.16     elric static void *
    252  1.16     elric cgd_getdata(struct dk_softc *dksc, unsigned long size)
    253  1.16     elric {
    254  1.16     elric 	struct	cgd_softc *cs =dksc->sc_osc;
    255  1.16     elric 	caddr_t	data = NULL;
    256  1.16     elric 
    257  1.16     elric 	simple_lock(&cs->sc_slock);
    258  1.16     elric 	if (cs->sc_data_used == 0) {
    259  1.16     elric 		cs->sc_data_used = 1;
    260  1.16     elric 		data = cs->sc_data;
    261  1.16     elric 	}
    262  1.16     elric 	simple_unlock(&cs->sc_slock);
    263  1.16     elric 
    264  1.16     elric 	if (data)
    265  1.16     elric 		return data;
    266  1.16     elric 
    267  1.16     elric 	return malloc(size, M_DEVBUF, M_NOWAIT);
    268  1.16     elric }
    269  1.16     elric 
    270   1.1     elric static void
    271  1.16     elric cgd_putdata(struct dk_softc *dksc, caddr_t data)
    272  1.16     elric {
    273  1.16     elric 	struct	cgd_softc *cs =dksc->sc_osc;
    274  1.16     elric 
    275  1.16     elric 	if (data == cs->sc_data) {
    276  1.16     elric 		simple_lock(&cs->sc_slock);
    277  1.16     elric 		cs->sc_data_used = 0;
    278  1.16     elric 		simple_unlock(&cs->sc_slock);
    279  1.16     elric 	} else {
    280  1.16     elric 		free(data, M_DEVBUF);
    281  1.16     elric 	}
    282  1.16     elric }
    283  1.16     elric 
    284  1.16     elric static int
    285   1.1     elric cgdstart(struct dk_softc *dksc, struct buf *bp)
    286   1.1     elric {
    287   1.1     elric 	struct	cgd_softc *cs = dksc->sc_osc;
    288  1.17       dbj 	struct	buf *nbp;
    289   1.1     elric 	struct	partition *pp;
    290   1.1     elric 	caddr_t	addr;
    291   1.1     elric 	caddr_t	newaddr;
    292   1.1     elric 	daddr_t	bn;
    293  1.17       dbj 	int s;
    294   1.1     elric 
    295   1.1     elric 	DPRINTF_FOLLOW(("cgdstart(%p, %p)\n", dksc, bp));
    296   1.1     elric 	disk_busy(&dksc->sc_dkdev); /* XXX: put in dksubr.c */
    297   1.1     elric 
    298   1.1     elric 	/* XXXrcd:
    299   1.1     elric 	 * Translate partition relative blocks to absolute blocks,
    300   1.1     elric 	 * this probably belongs (somehow) in dksubr.c, since it
    301   1.1     elric 	 * is independant of the underlying code...  This will require
    302   1.1     elric 	 * that the interface be expanded slightly, though.
    303   1.1     elric 	 */
    304   1.1     elric 	bn = bp->b_blkno;
    305   1.1     elric 	if (DISKPART(bp->b_dev) != RAW_PART) {
    306   1.1     elric 		pp = &cs->sc_dksc.sc_dkdev.dk_label->d_partitions[DISKPART(bp->b_dev)];
    307   1.1     elric 		bn += pp->p_offset;
    308   1.1     elric 	}
    309   1.1     elric 
    310   1.1     elric 	/*
    311  1.16     elric 	 * We attempt to allocate all of our resources up front, so that
    312  1.16     elric 	 * we can fail quickly if they are unavailable.
    313  1.16     elric 	 */
    314  1.22     perry 
    315  1.17       dbj 	s = splbio();
    316  1.17       dbj 	nbp = pool_get(&bufpool, PR_NOWAIT);
    317  1.17       dbj 	splx(s);
    318  1.17       dbj 	if (nbp == NULL) {
    319  1.16     elric 		disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    320  1.16     elric 		return -1;
    321  1.16     elric 	}
    322  1.16     elric 
    323  1.16     elric 	/*
    324   1.1     elric 	 * If we are writing, then we need to encrypt the outgoing
    325  1.16     elric 	 * block into a new block of memory.  If we fail, then we
    326  1.16     elric 	 * return an error and let the dksubr framework deal with it.
    327   1.1     elric 	 */
    328   1.1     elric 	newaddr = addr = bp->b_data;
    329   1.1     elric 	if ((bp->b_flags & B_READ) == 0) {
    330  1.16     elric 		newaddr = cgd_getdata(dksc, bp->b_bcount);
    331  1.16     elric 		if (!newaddr) {
    332  1.17       dbj 			s = splbio();
    333  1.17       dbj 			pool_put(&bufpool, nbp);
    334  1.17       dbj 			splx(s);
    335  1.16     elric 			disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    336  1.16     elric 			return -1;
    337  1.16     elric 		}
    338   1.1     elric 		cgd_cipher(cs, newaddr, addr, bp->b_bcount, bn,
    339   1.1     elric 		    DEV_BSIZE, CGD_CIPHER_ENCRYPT);
    340   1.1     elric 	}
    341   1.1     elric 
    342  1.17       dbj 	BUF_INIT(nbp);
    343  1.17       dbj 	nbp->b_data = newaddr;
    344  1.17       dbj 	nbp->b_flags = bp->b_flags | B_CALL;
    345  1.17       dbj 	nbp->b_iodone = cgdiodone;
    346  1.17       dbj 	nbp->b_proc = bp->b_proc;
    347  1.17       dbj 	nbp->b_blkno = bn;
    348  1.17       dbj 	nbp->b_vp = cs->sc_tvn;
    349  1.17       dbj 	nbp->b_bcount = bp->b_bcount;
    350  1.17       dbj 	nbp->b_private = bp;
    351  1.17       dbj 
    352  1.17       dbj 	BIO_COPYPRIO(nbp, bp);
    353  1.17       dbj 
    354  1.17       dbj 	if ((nbp->b_flags & B_READ) == 0) {
    355  1.17       dbj 		V_INCR_NUMOUTPUT(nbp->b_vp);
    356  1.17       dbj 	}
    357  1.17       dbj 	VOP_STRATEGY(cs->sc_tvn, nbp);
    358  1.16     elric 	return 0;
    359   1.1     elric }
    360   1.1     elric 
    361  1.17       dbj /* expected to be called at splbio() */
    362  1.18   thorpej static void
    363  1.17       dbj cgdiodone(struct buf *nbp)
    364   1.1     elric {
    365  1.17       dbj 	struct	buf *obp = nbp->b_private;
    366  1.17       dbj 	struct	cgd_softc *cs = getcgd_softc(obp->b_dev);
    367   1.1     elric 	struct	dk_softc *dksc = &cs->sc_dksc;
    368  1.22     perry 
    369  1.17       dbj 	KDASSERT(cs);
    370   1.1     elric 
    371  1.17       dbj 	DPRINTF_FOLLOW(("cgdiodone(%p)\n", nbp));
    372  1.20      yamt 	DPRINTF(CGDB_IO, ("cgdiodone: bp %p bcount %d resid %d\n",
    373   1.1     elric 	    obp, obp->b_bcount, obp->b_resid));
    374  1.20      yamt 	DPRINTF(CGDB_IO, (" dev 0x%x, nbp %p bn %" PRId64 " addr %p bcnt %d\n",
    375  1.17       dbj 	    nbp->b_dev, nbp, nbp->b_blkno, nbp->b_data,
    376  1.17       dbj 	    nbp->b_bcount));
    377   1.1     elric 	if (nbp->b_flags & B_ERROR) {
    378   1.1     elric 		obp->b_flags |= B_ERROR;
    379   1.1     elric 		obp->b_error  = nbp->b_error ? nbp->b_error : EIO;
    380   1.1     elric 
    381   1.1     elric 		printf("%s: error %d\n", dksc->sc_xname, obp->b_error);
    382   1.1     elric 	}
    383   1.1     elric 
    384  1.16     elric 	/* Perform the decryption if we are reading.
    385   1.1     elric 	 *
    386   1.1     elric 	 * Note: use the blocknumber from nbp, since it is what
    387   1.1     elric 	 *       we used to encrypt the blocks.
    388   1.1     elric 	 */
    389   1.1     elric 
    390  1.16     elric 	if (nbp->b_flags & B_READ)
    391   1.1     elric 		cgd_cipher(cs, obp->b_data, obp->b_data, obp->b_bcount,
    392   1.1     elric 		    nbp->b_blkno, DEV_BSIZE, CGD_CIPHER_DECRYPT);
    393   1.1     elric 
    394  1.16     elric 	/* If we allocated memory, free it now... */
    395   1.1     elric 	if (nbp->b_data != obp->b_data)
    396  1.16     elric 		cgd_putdata(dksc, nbp->b_data);
    397   1.1     elric 
    398  1.17       dbj 	pool_put(&bufpool, nbp);
    399   1.1     elric 
    400   1.1     elric 	/* Request is complete for whatever reason */
    401   1.1     elric 	obp->b_resid = 0;
    402   1.1     elric 	if (obp->b_flags & B_ERROR)
    403   1.1     elric 		obp->b_resid = obp->b_bcount;
    404   1.5       mrg 	disk_unbusy(&dksc->sc_dkdev, obp->b_bcount - obp->b_resid,
    405   1.5       mrg 	    (obp->b_flags & B_READ));
    406   1.1     elric 	biodone(obp);
    407  1.16     elric 	dk_iodone(di, dksc);
    408   1.1     elric }
    409   1.1     elric 
    410   1.1     elric /* XXX: we should probably put these into dksubr.c, mostly */
    411  1.18   thorpej static int
    412   1.1     elric cgdread(dev_t dev, struct uio *uio, int flags)
    413   1.1     elric {
    414   1.1     elric 	struct	cgd_softc *cs;
    415   1.1     elric 	struct	dk_softc *dksc;
    416   1.1     elric 
    417   1.1     elric 	DPRINTF_FOLLOW(("cgdread(%d, %p, %d)\n", dev, uio, flags));
    418   1.1     elric 	GETCGD_SOFTC(cs, dev);
    419   1.1     elric 	dksc = &cs->sc_dksc;
    420   1.1     elric 	if ((dksc->sc_flags & DKF_INITED) == 0)
    421   1.1     elric 		return ENXIO;
    422   1.1     elric 	return physio(cgdstrategy, NULL, dev, B_READ, minphys, uio);
    423   1.1     elric }
    424   1.1     elric 
    425   1.1     elric /* XXX: we should probably put these into dksubr.c, mostly */
    426  1.18   thorpej static int
    427   1.1     elric cgdwrite(dev_t dev, struct uio *uio, int flags)
    428   1.1     elric {
    429   1.1     elric 	struct	cgd_softc *cs;
    430   1.1     elric 	struct	dk_softc *dksc;
    431   1.1     elric 
    432   1.1     elric 	DPRINTF_FOLLOW(("cgdwrite(%d, %p, %d)\n", dev, uio, flags));
    433   1.1     elric 	GETCGD_SOFTC(cs, dev);
    434   1.1     elric 	dksc = &cs->sc_dksc;
    435   1.1     elric 	if ((dksc->sc_flags & DKF_INITED) == 0)
    436   1.1     elric 		return ENXIO;
    437   1.1     elric 	return physio(cgdstrategy, NULL, dev, B_WRITE, minphys, uio);
    438   1.1     elric }
    439   1.1     elric 
    440  1.18   thorpej static int
    441  1.12      fvdl cgdioctl(dev_t dev, u_long cmd, caddr_t data, int flag, struct proc *p)
    442   1.1     elric {
    443   1.1     elric 	struct	cgd_softc *cs;
    444   1.1     elric 	struct	dk_softc *dksc;
    445   1.1     elric 	int	ret;
    446   1.1     elric 	int	part = DISKPART(dev);
    447   1.1     elric 	int	pmask = 1 << part;
    448   1.1     elric 
    449   1.1     elric 	DPRINTF_FOLLOW(("cgdioctl(%d, %ld, %p, %d, %p)\n",
    450  1.12      fvdl 	    dev, cmd, data, flag, p));
    451   1.1     elric 	GETCGD_SOFTC(cs, dev);
    452   1.1     elric 	dksc = &cs->sc_dksc;
    453   1.1     elric 	switch (cmd) {
    454   1.1     elric 	case CGDIOCSET:
    455   1.1     elric 	case CGDIOCCLR:
    456   1.1     elric 		if ((flag & FWRITE) == 0)
    457   1.1     elric 			return EBADF;
    458   1.1     elric 	}
    459   1.1     elric 
    460   1.1     elric 	if ((ret = lockmgr(&dksc->sc_lock, LK_EXCLUSIVE, NULL)) != 0)
    461   1.1     elric 		return ret;
    462   1.1     elric 
    463   1.1     elric 	switch (cmd) {
    464   1.1     elric 	case CGDIOCSET:
    465   1.1     elric 		if (dksc->sc_flags & DKF_INITED)
    466   1.1     elric 			ret = EBUSY;
    467   1.1     elric 		else
    468  1.12      fvdl 			ret = cgd_ioctl_set(cs, data, p);
    469   1.1     elric 		break;
    470   1.1     elric 	case CGDIOCCLR:
    471   1.1     elric 		if (!(dksc->sc_flags & DKF_INITED)) {
    472   1.1     elric 			ret = ENXIO;
    473   1.1     elric 			break;
    474   1.1     elric 		}
    475   1.1     elric 		if (DK_BUSY(&cs->sc_dksc, pmask)) {
    476   1.1     elric 			ret = EBUSY;
    477   1.1     elric 			break;
    478   1.1     elric 		}
    479  1.12      fvdl 		ret = cgd_ioctl_clr(cs, data, p);
    480   1.1     elric 		break;
    481   1.1     elric 	default:
    482  1.12      fvdl 		ret = dk_ioctl(di, dksc, dev, cmd, data, flag, p);
    483   1.1     elric 		break;
    484   1.1     elric 	}
    485   1.1     elric 
    486   1.1     elric 	lockmgr(&dksc->sc_lock, LK_RELEASE, NULL);
    487   1.1     elric 	return ret;
    488   1.1     elric }
    489   1.1     elric 
    490  1.18   thorpej static int
    491   1.1     elric cgddump(dev_t dev, daddr_t blkno, caddr_t va, size_t size)
    492   1.1     elric {
    493   1.1     elric 	struct	cgd_softc *cs;
    494   1.1     elric 
    495   1.6    bouyer 	DPRINTF_FOLLOW(("cgddump(%d, %" PRId64 ", %p, %lu)\n", dev, blkno, va,
    496   1.2     elric 	    (unsigned long)size));
    497   1.1     elric 	GETCGD_SOFTC(cs, dev);
    498   1.1     elric 	return dk_dump(di, &cs->sc_dksc, dev, blkno, va, size);
    499   1.1     elric }
    500   1.1     elric 
    501   1.1     elric /*
    502   1.1     elric  * XXXrcd:
    503   1.1     elric  *  for now we hardcode the maximum key length.
    504   1.1     elric  */
    505   1.1     elric #define MAX_KEYSIZE	1024
    506   1.1     elric 
    507   1.1     elric /* ARGSUSED */
    508   1.1     elric static int
    509  1.12      fvdl cgd_ioctl_set(struct cgd_softc *cs, void *data, struct proc *p)
    510   1.1     elric {
    511   1.1     elric 	struct	 cgd_ioctl *ci = data;
    512   1.1     elric 	struct	 vnode *vp;
    513   1.1     elric 	int	 ret;
    514  1.15       dan 	int	 keybytes;			/* key length in bytes */
    515   1.1     elric 	char	*cp;
    516   1.1     elric 	char	 inbuf[MAX_KEYSIZE];
    517   1.1     elric 
    518   1.1     elric 	cp = ci->ci_disk;
    519  1.12      fvdl 	if ((ret = dk_lookup(cp, p, &vp)) != 0)
    520   1.1     elric 		return ret;
    521   1.1     elric 
    522  1.12      fvdl 	if ((ret = cgdinit(cs, cp, vp, p)) != 0)
    523   1.1     elric 		goto bail;
    524   1.1     elric 
    525   1.1     elric 	memset(inbuf, 0x0, sizeof(inbuf));
    526   1.1     elric 	ret = copyinstr(ci->ci_alg, inbuf, 256, NULL);
    527   1.1     elric 	if (ret)
    528   1.1     elric 		goto bail;
    529   1.1     elric 	cs->sc_cfuncs = cryptfuncs_find(inbuf);
    530   1.1     elric 	if (!cs->sc_cfuncs) {
    531   1.1     elric 		ret = EINVAL;
    532   1.1     elric 		goto bail;
    533   1.1     elric 	}
    534   1.1     elric 
    535   1.1     elric 	/* right now we only support encblkno, so hard-code it */
    536   1.1     elric 	memset(inbuf, 0x0, sizeof(inbuf));
    537   1.1     elric 	ret = copyinstr(ci->ci_ivmethod, inbuf, sizeof(inbuf), NULL);
    538   1.1     elric 	if (ret)
    539   1.1     elric 		goto bail;
    540   1.1     elric 	if (strcmp("encblkno", inbuf)) {
    541   1.1     elric 		ret = EINVAL;
    542   1.1     elric 		goto bail;
    543   1.1     elric 	}
    544   1.1     elric 
    545  1.15       dan 	keybytes = ci->ci_keylen / 8 + 1;
    546  1.15       dan 	if (keybytes > MAX_KEYSIZE) {
    547   1.1     elric 		ret = EINVAL;
    548   1.1     elric 		goto bail;
    549   1.1     elric 	}
    550   1.1     elric 	memset(inbuf, 0x0, sizeof(inbuf));
    551  1.15       dan 	ret = copyin(ci->ci_key, inbuf, keybytes);
    552   1.1     elric 	if (ret)
    553   1.1     elric 		goto bail;
    554   1.1     elric 
    555   1.1     elric 	cs->sc_cdata.cf_blocksize = ci->ci_blocksize;
    556   1.1     elric 	cs->sc_cdata.cf_mode = CGD_CIPHER_CBC_ENCBLKNO;
    557   1.1     elric 	cs->sc_cdata.cf_priv = cs->sc_cfuncs->cf_init(ci->ci_keylen, inbuf,
    558   1.1     elric 	    &cs->sc_cdata.cf_blocksize);
    559   1.1     elric 	memset(inbuf, 0x0, sizeof(inbuf));
    560   1.1     elric 	if (!cs->sc_cdata.cf_priv) {
    561   1.1     elric 		printf("cgd: unable to initialize cipher\n");
    562   1.1     elric 		ret = EINVAL;		/* XXX is this the right error? */
    563   1.1     elric 		goto bail;
    564   1.1     elric 	}
    565   1.1     elric 
    566  1.16     elric 	bufq_alloc(&cs->sc_dksc.sc_bufq, BUFQ_FCFS);
    567  1.16     elric 
    568  1.16     elric 	cs->sc_data = malloc(MAXPHYS, M_DEVBUF, M_WAITOK);
    569  1.16     elric 	cs->sc_data_used = 0;
    570  1.16     elric 
    571   1.1     elric 	cs->sc_dksc.sc_flags |= DKF_INITED;
    572   1.1     elric 
    573   1.1     elric 	/* Attach the disk. */
    574   1.1     elric 	disk_attach(&cs->sc_dksc.sc_dkdev);
    575   1.1     elric 
    576   1.1     elric 	/* Try and read the disklabel. */
    577   1.1     elric 	dk_getdisklabel(di, &cs->sc_dksc, 0 /* XXX ? */);
    578   1.1     elric 
    579   1.1     elric 	return 0;
    580   1.1     elric 
    581   1.1     elric bail:
    582  1.12      fvdl 	(void)vn_close(vp, FREAD|FWRITE, p->p_ucred, p);
    583   1.1     elric 	return ret;
    584   1.1     elric }
    585   1.1     elric 
    586   1.1     elric /* ARGSUSED */
    587   1.1     elric static int
    588  1.12      fvdl cgd_ioctl_clr(struct cgd_softc *cs, void *data, struct proc *p)
    589   1.1     elric {
    590  1.16     elric 	struct	buf *bp;
    591  1.16     elric 	int	s;
    592  1.16     elric 
    593  1.16     elric 	/* Kill off any queued buffers. */
    594  1.16     elric 	s = splbio();
    595  1.23      yamt 	bufq_drain(&cs->sc_dksc.sc_bufq);
    596  1.16     elric 	splx(s);
    597  1.16     elric 	bufq_free(&cs->sc_dksc.sc_bufq);
    598   1.1     elric 
    599  1.12      fvdl 	(void)vn_close(cs->sc_tvn, FREAD|FWRITE, p->p_ucred, p);
    600   1.1     elric 	cs->sc_cfuncs->cf_destroy(cs->sc_cdata.cf_priv);
    601   1.1     elric 	free(cs->sc_tpath, M_DEVBUF);
    602  1.16     elric 	free(cs->sc_data, M_DEVBUF);
    603  1.16     elric 	cs->sc_data_used = 0;
    604   1.1     elric 	cs->sc_dksc.sc_flags &= ~DKF_INITED;
    605   1.1     elric 	disk_detach(&cs->sc_dksc.sc_dkdev);
    606   1.1     elric 
    607   1.1     elric 	return 0;
    608   1.1     elric }
    609   1.1     elric 
    610   1.1     elric static int
    611   1.1     elric cgdinit(struct cgd_softc *cs, char *cpath, struct vnode *vp,
    612  1.12      fvdl 	struct proc *p)
    613   1.1     elric {
    614   1.1     elric 	struct	dk_geom *pdg;
    615   1.1     elric 	struct	partinfo dpart;
    616   1.1     elric 	struct	vattr va;
    617   1.1     elric 	size_t	size;
    618   1.1     elric 	int	maxsecsize = 0;
    619   1.1     elric 	int	ret;
    620   1.1     elric 	char	tmppath[MAXPATHLEN];
    621   1.1     elric 
    622   1.1     elric 	cs->sc_dksc.sc_size = 0;
    623   1.1     elric 	cs->sc_tvn = vp;
    624   1.1     elric 
    625   1.1     elric 	memset(tmppath, 0x0, sizeof(tmppath));
    626   1.1     elric 	ret = copyinstr(cpath, tmppath, MAXPATHLEN, &cs->sc_tpathlen);
    627   1.1     elric 	if (ret)
    628   1.1     elric 		goto bail;
    629   1.1     elric 	cs->sc_tpath = malloc(cs->sc_tpathlen, M_DEVBUF, M_WAITOK);
    630   1.1     elric 	memcpy(cs->sc_tpath, tmppath, cs->sc_tpathlen);
    631   1.1     elric 
    632  1.22     perry 	if ((ret = VOP_GETATTR(vp, &va, p->p_ucred, p)) != 0)
    633   1.1     elric 		goto bail;
    634   1.1     elric 
    635   1.1     elric 	cs->sc_tdev = va.va_rdev;
    636   1.1     elric 
    637  1.12      fvdl 	ret = VOP_IOCTL(vp, DIOCGPART, &dpart, FREAD, p->p_ucred, p);
    638   1.1     elric 	if (ret)
    639   1.1     elric 		goto bail;
    640   1.1     elric 
    641   1.1     elric 	maxsecsize =
    642   1.1     elric 	    ((dpart.disklab->d_secsize > maxsecsize) ?
    643   1.1     elric 	    dpart.disklab->d_secsize : maxsecsize);
    644   1.1     elric 	size = dpart.part->p_size;
    645   1.1     elric 
    646   1.1     elric 	if (!size) {
    647   1.1     elric 		ret = ENODEV;
    648   1.1     elric 		goto bail;
    649   1.1     elric 	}
    650   1.1     elric 
    651   1.1     elric 	cs->sc_dksc.sc_size = size;
    652   1.1     elric 
    653   1.1     elric 	/*
    654   1.1     elric 	 * XXX here we should probe the underlying device.  If we
    655   1.1     elric 	 *     are accessing a partition of type RAW_PART, then
    656   1.1     elric 	 *     we should populate our initial geometry with the
    657   1.1     elric 	 *     geometry that we discover from the device.
    658   1.1     elric 	 */
    659   1.1     elric 	pdg = &cs->sc_dksc.sc_geom;
    660   1.1     elric 	pdg->pdg_secsize = DEV_BSIZE;
    661   1.1     elric 	pdg->pdg_ntracks = 1;
    662   1.1     elric 	pdg->pdg_nsectors = 1024 * (1024 / pdg->pdg_secsize);
    663   1.1     elric 	pdg->pdg_ncylinders = cs->sc_dksc.sc_size / pdg->pdg_nsectors;
    664   1.1     elric 
    665   1.1     elric bail:
    666   1.1     elric 	if (ret && cs->sc_tpath)
    667   1.1     elric 		free(cs->sc_tpath, M_DEVBUF);
    668   1.1     elric 	return ret;
    669   1.1     elric }
    670   1.1     elric 
    671   1.1     elric /*
    672   1.1     elric  * Our generic cipher entry point.  This takes care of the
    673   1.1     elric  * IV mode and passes off the work to the specific cipher.
    674   1.1     elric  * We implement here the IV method ``encrypted block
    675   1.1     elric  * number''.
    676  1.22     perry  *
    677   1.1     elric  * For the encryption case, we accomplish this by setting
    678   1.1     elric  * up a struct uio where the first iovec of the source is
    679   1.1     elric  * the blocknumber and the first iovec of the dest is a
    680   1.1     elric  * sink.  We then call the cipher with an IV of zero, and
    681   1.1     elric  * the right thing happens.
    682  1.22     perry  *
    683   1.1     elric  * For the decryption case, we use the same basic mechanism
    684   1.1     elric  * for symmetry, but we encrypt the block number in the
    685   1.1     elric  * first iovec.
    686   1.1     elric  *
    687   1.1     elric  * We mainly do this to avoid requiring the definition of
    688   1.1     elric  * an ECB mode.
    689   1.1     elric  *
    690   1.1     elric  * XXXrcd: for now we rely on our own crypto framework defined
    691   1.1     elric  *         in dev/cgd_crypto.c.  This will change when we
    692   1.1     elric  *         get a generic kernel crypto framework.
    693   1.1     elric  */
    694   1.1     elric 
    695   1.1     elric static void
    696   1.1     elric blkno2blkno_buf(char *buf, daddr_t blkno)
    697   1.1     elric {
    698   1.1     elric 	int	i;
    699   1.1     elric 
    700   1.1     elric 	/* Set up the blkno in blkno_buf, here we do not care much
    701   1.1     elric 	 * about the final layout of the information as long as we
    702   1.1     elric 	 * can guarantee that each sector will have a different IV
    703   1.1     elric 	 * and that the endianness of the machine will not affect
    704   1.1     elric 	 * the representation that we have chosen.
    705   1.1     elric 	 *
    706   1.1     elric 	 * We choose this representation, because it does not rely
    707   1.1     elric 	 * on the size of buf (which is the blocksize of the cipher),
    708   1.1     elric 	 * but allows daddr_t to grow without breaking existing
    709   1.1     elric 	 * disks.
    710   1.1     elric 	 *
    711   1.1     elric 	 * Note that blkno2blkno_buf does not take a size as input,
    712   1.1     elric 	 * and hence must be called on a pre-zeroed buffer of length
    713   1.1     elric 	 * greater than or equal to sizeof(daddr_t).
    714   1.1     elric 	 */
    715   1.1     elric 	for (i=0; i < sizeof(daddr_t); i++) {
    716   1.1     elric 		*buf++ = blkno & 0xff;
    717   1.1     elric 		blkno >>= 8;
    718   1.1     elric 	}
    719   1.1     elric }
    720   1.1     elric 
    721   1.1     elric static void
    722   1.1     elric cgd_cipher(struct cgd_softc *cs, caddr_t dst, caddr_t src,
    723   1.1     elric 	   size_t len, daddr_t blkno, size_t secsize, int dir)
    724   1.1     elric {
    725   1.1     elric 	cfunc_cipher	*cipher = cs->sc_cfuncs->cf_cipher;
    726   1.1     elric 	struct uio	dstuio;
    727   1.1     elric 	struct uio	srcuio;
    728   1.1     elric 	struct iovec	dstiov[2];
    729   1.1     elric 	struct iovec	srciov[2];
    730   1.1     elric 	int		blocksize = cs->sc_cdata.cf_blocksize;
    731   1.1     elric 	char		sink[blocksize];
    732   1.1     elric 	char		zero_iv[blocksize];
    733   1.1     elric 	char		blkno_buf[blocksize];
    734   1.1     elric 
    735   1.1     elric 	DPRINTF_FOLLOW(("cgd_cipher() dir=%d\n", dir));
    736   1.1     elric 
    737  1.22     perry 	DIAGCONDPANIC(len % blocksize != 0,
    738   1.1     elric 	    ("cgd_cipher: len %% blocksize != 0"));
    739   1.1     elric 
    740   1.1     elric 	/* ensure that sizeof(daddr_t) <= blocksize (for encblkno IVing) */
    741   1.1     elric 	DIAGCONDPANIC(sizeof(daddr_t) > blocksize,
    742   1.1     elric 	    ("cgd_cipher: sizeof(daddr_t) > blocksize"));
    743   1.1     elric 
    744   1.1     elric 	memset(zero_iv, 0x0, sizeof(zero_iv));
    745   1.1     elric 
    746   1.1     elric 	dstuio.uio_iov = dstiov;
    747   1.1     elric 	dstuio.uio_iovcnt = 2;
    748   1.1     elric 
    749   1.1     elric 	srcuio.uio_iov = srciov;
    750   1.1     elric 	srcuio.uio_iovcnt = 2;
    751   1.1     elric 
    752   1.1     elric 	dstiov[0].iov_base = sink;
    753   1.1     elric 	dstiov[0].iov_len  = blocksize;
    754   1.1     elric 	srciov[0].iov_base = blkno_buf;
    755   1.1     elric 	srciov[0].iov_len  = blocksize;
    756   1.1     elric 	dstiov[1].iov_len  = secsize;
    757   1.1     elric 	srciov[1].iov_len  = secsize;
    758   1.1     elric 
    759   1.1     elric 	for (; len > 0; len -= secsize) {
    760   1.1     elric 		dstiov[1].iov_base = dst;
    761   1.1     elric 		srciov[1].iov_base = src;
    762   1.1     elric 
    763   1.1     elric 		memset(blkno_buf, 0x0, sizeof(blkno_buf));
    764   1.1     elric 		blkno2blkno_buf(blkno_buf, blkno);
    765   1.1     elric 		if (dir == CGD_CIPHER_DECRYPT) {
    766   1.1     elric 			dstuio.uio_iovcnt = 1;
    767   1.1     elric 			srcuio.uio_iovcnt = 1;
    768   1.1     elric 			IFDEBUG(CGDB_CRYPTO, hexprint("step 0: blkno_buf",
    769   1.1     elric 			    blkno_buf, sizeof(blkno_buf)));
    770   1.1     elric 			cipher(cs->sc_cdata.cf_priv, &dstuio, &srcuio,
    771   1.1     elric 			    zero_iv, CGD_CIPHER_ENCRYPT);
    772   1.1     elric 			memcpy(blkno_buf, sink, blocksize);
    773   1.1     elric 			dstuio.uio_iovcnt = 2;
    774   1.1     elric 			srcuio.uio_iovcnt = 2;
    775   1.1     elric 		}
    776   1.1     elric 
    777   1.1     elric 		IFDEBUG(CGDB_CRYPTO, hexprint("step 1: blkno_buf",
    778   1.1     elric 		    blkno_buf, sizeof(blkno_buf)));
    779   1.1     elric 		cipher(cs->sc_cdata.cf_priv, &dstuio, &srcuio, zero_iv, dir);
    780   1.1     elric 		IFDEBUG(CGDB_CRYPTO, hexprint("step 2: sink",
    781   1.1     elric 		    sink, sizeof(sink)));
    782   1.1     elric 
    783   1.1     elric 		dst += secsize;
    784   1.1     elric 		src += secsize;
    785   1.1     elric 		blkno++;
    786   1.1     elric 	}
    787   1.1     elric }
    788   1.1     elric 
    789   1.1     elric #ifdef DEBUG
    790   1.1     elric static void
    791   1.1     elric hexprint(char *start, void *buf, int len)
    792   1.1     elric {
    793   1.1     elric 	char	*c = buf;
    794   1.1     elric 
    795   1.1     elric 	DIAGCONDPANIC(len < 0, ("hexprint: called with len < 0"));
    796   1.1     elric 	printf("%s: len=%06d 0x", start, len);
    797   1.1     elric 	while (len--)
    798   1.1     elric 		printf("%02x", (unsigned) *c++);
    799   1.1     elric }
    800   1.1     elric #endif
    801