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