Home | History | Annotate | Line # | Download | only in dev
cgd.c revision 1.68
      1 /* $NetBSD: cgd.c,v 1.68 2010/01/20 19:00:47 dyoung Exp $ */
      2 
      3 /*-
      4  * Copyright (c) 2002 The NetBSD Foundation, Inc.
      5  * All rights reserved.
      6  *
      7  * This code is derived from software contributed to The NetBSD Foundation
      8  * by Roland C. Dowdeswell.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted provided that the following conditions
     12  * are met:
     13  * 1. Redistributions of source code must retain the above copyright
     14  *    notice, this list of conditions and the following disclaimer.
     15  * 2. Redistributions in binary form must reproduce the above copyright
     16  *    notice, this list of conditions and the following disclaimer in the
     17  *    documentation and/or other materials provided with the distribution.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29  * POSSIBILITY OF SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 __KERNEL_RCSID(0, "$NetBSD: cgd.c,v 1.68 2010/01/20 19:00:47 dyoung Exp $");
     34 
     35 #include <sys/types.h>
     36 #include <sys/param.h>
     37 #include <sys/systm.h>
     38 #include <sys/proc.h>
     39 #include <sys/errno.h>
     40 #include <sys/buf.h>
     41 #include <sys/bufq.h>
     42 #include <sys/malloc.h>
     43 #include <sys/pool.h>
     44 #include <sys/ioctl.h>
     45 #include <sys/device.h>
     46 #include <sys/disk.h>
     47 #include <sys/disklabel.h>
     48 #include <sys/fcntl.h>
     49 #include <sys/vnode.h>
     50 #include <sys/conf.h>
     51 #include <sys/syslog.h>
     52 
     53 #include <dev/dkvar.h>
     54 #include <dev/cgdvar.h>
     55 
     56 /* Entry Point Functions */
     57 
     58 void	cgdattach(int);
     59 
     60 static dev_type_open(cgdopen);
     61 static dev_type_close(cgdclose);
     62 static dev_type_read(cgdread);
     63 static dev_type_write(cgdwrite);
     64 static dev_type_ioctl(cgdioctl);
     65 static dev_type_strategy(cgdstrategy);
     66 static dev_type_dump(cgddump);
     67 static dev_type_size(cgdsize);
     68 
     69 const struct bdevsw cgd_bdevsw = {
     70 	cgdopen, cgdclose, cgdstrategy, cgdioctl,
     71 	cgddump, cgdsize, D_DISK
     72 };
     73 
     74 const struct cdevsw cgd_cdevsw = {
     75 	cgdopen, cgdclose, cgdread, cgdwrite, cgdioctl,
     76 	nostop, notty, nopoll, nommap, nokqfilter, D_DISK
     77 };
     78 
     79 static int cgd_match(device_t, cfdata_t, void *);
     80 static void cgd_attach(device_t, device_t, void *);
     81 static int cgd_detach(device_t, int);
     82 static struct cgd_softc	*cgd_spawn(int);
     83 static int cgd_destroy(device_t);
     84 
     85 /* Internal Functions */
     86 
     87 static int	cgdstart(struct dk_softc *, struct buf *);
     88 static void	cgdiodone(struct buf *);
     89 
     90 static int	cgd_ioctl_set(struct cgd_softc *, void *, struct lwp *);
     91 static int	cgd_ioctl_clr(struct cgd_softc *, struct lwp *);
     92 static int	cgdinit(struct cgd_softc *, const char *, struct vnode *,
     93 			struct lwp *);
     94 static void	cgd_cipher(struct cgd_softc *, void *, void *,
     95 			   size_t, daddr_t, size_t, int);
     96 
     97 /* Pseudo-disk Interface */
     98 
     99 static struct dk_intf the_dkintf = {
    100 	DTYPE_CGD,
    101 	"cgd",
    102 	cgdopen,
    103 	cgdclose,
    104 	cgdstrategy,
    105 	cgdstart,
    106 };
    107 static struct dk_intf *di = &the_dkintf;
    108 
    109 static struct dkdriver cgddkdriver = {
    110 	.d_strategy = cgdstrategy,
    111 	.d_minphys = minphys,
    112 };
    113 
    114 CFATTACH_DECL3_NEW(cgd, sizeof(struct cgd_softc),
    115     cgd_match, cgd_attach, cgd_detach, NULL, NULL, NULL, DVF_DETACH_SHUTDOWN);
    116 extern struct cfdriver cgd_cd;
    117 
    118 /* DIAGNOSTIC and DEBUG definitions */
    119 
    120 #if defined(CGDDEBUG) && !defined(DEBUG)
    121 #define DEBUG
    122 #endif
    123 
    124 #ifdef DEBUG
    125 int cgddebug = 0;
    126 
    127 #define CGDB_FOLLOW	0x1
    128 #define CGDB_IO	0x2
    129 #define CGDB_CRYPTO	0x4
    130 
    131 #define IFDEBUG(x,y)		if (cgddebug & (x)) y
    132 #define DPRINTF(x,y)		IFDEBUG(x, printf y)
    133 #define DPRINTF_FOLLOW(y)	DPRINTF(CGDB_FOLLOW, y)
    134 
    135 static void	hexprint(const char *, void *, int);
    136 
    137 #else
    138 #define IFDEBUG(x,y)
    139 #define DPRINTF(x,y)
    140 #define DPRINTF_FOLLOW(y)
    141 #endif
    142 
    143 #ifdef DIAGNOSTIC
    144 #define DIAGPANIC(x)		panic x
    145 #define DIAGCONDPANIC(x,y)	if (x) panic y
    146 #else
    147 #define DIAGPANIC(x)
    148 #define DIAGCONDPANIC(x,y)
    149 #endif
    150 
    151 /* Global variables */
    152 
    153 /* Utility Functions */
    154 
    155 #define CGDUNIT(x)		DISKUNIT(x)
    156 #define GETCGD_SOFTC(_cs, x)	if (!((_cs) = getcgd_softc(x))) return ENXIO
    157 
    158 /* The code */
    159 
    160 static struct cgd_softc *
    161 getcgd_softc(dev_t dev)
    162 {
    163 	int	unit = CGDUNIT(dev);
    164 	struct cgd_softc *sc;
    165 
    166 	DPRINTF_FOLLOW(("getcgd_softc(0x%"PRIx64"): unit = %d\n", dev, unit));
    167 
    168 	sc = device_lookup_private(&cgd_cd, unit);
    169 	if (sc == NULL)
    170 		sc = cgd_spawn(unit);
    171 	return sc;
    172 }
    173 
    174 static int
    175 cgd_match(device_t self, cfdata_t cfdata, void *aux)
    176 {
    177 
    178 	return 1;
    179 }
    180 
    181 static void
    182 cgd_attach(device_t parent, device_t self, void *aux)
    183 {
    184 	struct cgd_softc *sc = device_private(self);
    185 
    186 	sc->sc_dev = self;
    187 	simple_lock_init(&sc->sc_slock);
    188 	dk_sc_init(&sc->sc_dksc, sc, device_xname(sc->sc_dev));
    189 	disk_init(&sc->sc_dksc.sc_dkdev, sc->sc_dksc.sc_xname, &cgddkdriver);
    190 }
    191 
    192 
    193 static int
    194 cgd_detach(device_t self, int flags)
    195 {
    196 	int ret;
    197 	const int pmask = 1 << RAW_PART;
    198 	struct cgd_softc *sc = device_private(self);
    199 	struct dk_softc *dksc = &sc->sc_dksc;
    200 
    201 	if (DK_BUSY(dksc, pmask))
    202 		return EBUSY;
    203 
    204 	if ((dksc->sc_flags & DKF_INITED) != 0 &&
    205 	    (ret = cgd_ioctl_clr(sc, curlwp)) != 0)
    206 		return ret;
    207 
    208 	disk_destroy(&dksc->sc_dkdev);
    209 
    210 	return 0;
    211 }
    212 
    213 void
    214 cgdattach(int num)
    215 {
    216 	int error;
    217 
    218 	error = config_cfattach_attach(cgd_cd.cd_name, &cgd_ca);
    219 	if (error != 0)
    220 		aprint_error("%s: unable to register cfattach\n",
    221 		    cgd_cd.cd_name);
    222 }
    223 
    224 static struct cgd_softc *
    225 cgd_spawn(int unit)
    226 {
    227 	cfdata_t cf;
    228 
    229 	cf = malloc(sizeof(*cf), M_DEVBUF, M_WAITOK);
    230 	cf->cf_name = cgd_cd.cd_name;
    231 	cf->cf_atname = cgd_cd.cd_name;
    232 	cf->cf_unit = unit;
    233 	cf->cf_fstate = FSTATE_STAR;
    234 
    235 	return device_private(config_attach_pseudo(cf));
    236 }
    237 
    238 static int
    239 cgd_destroy(device_t dev)
    240 {
    241 	int error;
    242 	cfdata_t cf;
    243 
    244 	cf = device_cfdata(dev);
    245 	error = config_detach(dev, DETACH_QUIET);
    246 	if (error)
    247 		return error;
    248 	free(cf, M_DEVBUF);
    249 	return 0;
    250 }
    251 
    252 static int
    253 cgdopen(dev_t dev, int flags, int fmt, struct lwp *l)
    254 {
    255 	struct	cgd_softc *cs;
    256 
    257 	DPRINTF_FOLLOW(("cgdopen(0x%"PRIx64", %d)\n", dev, flags));
    258 	GETCGD_SOFTC(cs, dev);
    259 	return dk_open(di, &cs->sc_dksc, dev, flags, fmt, l);
    260 }
    261 
    262 static int
    263 cgdclose(dev_t dev, int flags, int fmt, struct lwp *l)
    264 {
    265 	int error;
    266 	struct	cgd_softc *cs;
    267 	struct	dk_softc *dksc;
    268 
    269 	DPRINTF_FOLLOW(("cgdclose(0x%"PRIx64", %d)\n", dev, flags));
    270 	GETCGD_SOFTC(cs, dev);
    271 	dksc = &cs->sc_dksc;
    272 	if ((error =  dk_close(di, dksc, dev, flags, fmt, l)) != 0)
    273 		return error;
    274 
    275 	if ((dksc->sc_flags & DKF_INITED) == 0) {
    276 		if ((error = cgd_destroy(cs->sc_dev)) != 0) {
    277 			aprint_error_dev(cs->sc_dev,
    278 			    "unable to detach instance\n");
    279 			return error;
    280 		}
    281 	}
    282 	return 0;
    283 }
    284 
    285 static void
    286 cgdstrategy(struct buf *bp)
    287 {
    288 	struct	cgd_softc *cs = getcgd_softc(bp->b_dev);
    289 
    290 	DPRINTF_FOLLOW(("cgdstrategy(%p): b_bcount = %ld\n", bp,
    291 	    (long)bp->b_bcount));
    292 	/* XXXrcd: Should we test for (cs != NULL)? */
    293 	dk_strategy(di, &cs->sc_dksc, bp);
    294 	return;
    295 }
    296 
    297 static int
    298 cgdsize(dev_t dev)
    299 {
    300 	struct cgd_softc *cs = getcgd_softc(dev);
    301 
    302 	DPRINTF_FOLLOW(("cgdsize(0x%"PRIx64")\n", dev));
    303 	if (!cs)
    304 		return -1;
    305 	return dk_size(di, &cs->sc_dksc, dev);
    306 }
    307 
    308 /*
    309  * cgd_{get,put}data are functions that deal with getting a buffer
    310  * for the new encrypted data.  We have a buffer per device so that
    311  * we can ensure that we can always have a transaction in flight.
    312  * We use this buffer first so that we have one less piece of
    313  * malloc'ed data at any given point.
    314  */
    315 
    316 static void *
    317 cgd_getdata(struct dk_softc *dksc, unsigned long size)
    318 {
    319 	struct	cgd_softc *cs =dksc->sc_osc;
    320 	void *	data = NULL;
    321 
    322 	simple_lock(&cs->sc_slock);
    323 	if (cs->sc_data_used == 0) {
    324 		cs->sc_data_used = 1;
    325 		data = cs->sc_data;
    326 	}
    327 	simple_unlock(&cs->sc_slock);
    328 
    329 	if (data)
    330 		return data;
    331 
    332 	return malloc(size, M_DEVBUF, M_NOWAIT);
    333 }
    334 
    335 static void
    336 cgd_putdata(struct dk_softc *dksc, void *data)
    337 {
    338 	struct	cgd_softc *cs =dksc->sc_osc;
    339 
    340 	if (data == cs->sc_data) {
    341 		simple_lock(&cs->sc_slock);
    342 		cs->sc_data_used = 0;
    343 		simple_unlock(&cs->sc_slock);
    344 	} else {
    345 		free(data, M_DEVBUF);
    346 	}
    347 }
    348 
    349 static int
    350 cgdstart(struct dk_softc *dksc, struct buf *bp)
    351 {
    352 	struct	cgd_softc *cs = dksc->sc_osc;
    353 	struct	buf *nbp;
    354 	void *	addr;
    355 	void *	newaddr;
    356 	daddr_t	bn;
    357 	struct	vnode *vp;
    358 
    359 	DPRINTF_FOLLOW(("cgdstart(%p, %p)\n", dksc, bp));
    360 	disk_busy(&dksc->sc_dkdev); /* XXX: put in dksubr.c */
    361 
    362 	bn = bp->b_rawblkno;
    363 
    364 	/*
    365 	 * We attempt to allocate all of our resources up front, so that
    366 	 * we can fail quickly if they are unavailable.
    367 	 */
    368 
    369 	nbp = getiobuf(cs->sc_tvn, false);
    370 	if (nbp == NULL) {
    371 		disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    372 		return -1;
    373 	}
    374 
    375 	/*
    376 	 * If we are writing, then we need to encrypt the outgoing
    377 	 * block into a new block of memory.  If we fail, then we
    378 	 * return an error and let the dksubr framework deal with it.
    379 	 */
    380 	newaddr = addr = bp->b_data;
    381 	if ((bp->b_flags & B_READ) == 0) {
    382 		newaddr = cgd_getdata(dksc, bp->b_bcount);
    383 		if (!newaddr) {
    384 			putiobuf(nbp);
    385 			disk_unbusy(&dksc->sc_dkdev, 0, (bp->b_flags & B_READ));
    386 			return -1;
    387 		}
    388 		cgd_cipher(cs, newaddr, addr, bp->b_bcount, bn,
    389 		    DEV_BSIZE, CGD_CIPHER_ENCRYPT);
    390 	}
    391 
    392 	nbp->b_data = newaddr;
    393 	nbp->b_flags = bp->b_flags;
    394 	nbp->b_oflags = bp->b_oflags;
    395 	nbp->b_cflags = bp->b_cflags;
    396 	nbp->b_iodone = cgdiodone;
    397 	nbp->b_proc = bp->b_proc;
    398 	nbp->b_blkno = bn;
    399 	nbp->b_bcount = bp->b_bcount;
    400 	nbp->b_private = bp;
    401 
    402 	BIO_COPYPRIO(nbp, bp);
    403 
    404 	if ((nbp->b_flags & B_READ) == 0) {
    405 		vp = nbp->b_vp;
    406 		mutex_enter(&vp->v_interlock);
    407 		vp->v_numoutput++;
    408 		mutex_exit(&vp->v_interlock);
    409 	}
    410 	VOP_STRATEGY(cs->sc_tvn, nbp);
    411 	return 0;
    412 }
    413 
    414 /* expected to be called at splbio() */
    415 static void
    416 cgdiodone(struct buf *nbp)
    417 {
    418 	struct	buf *obp = nbp->b_private;
    419 	struct	cgd_softc *cs = getcgd_softc(obp->b_dev);
    420 	struct	dk_softc *dksc = &cs->sc_dksc;
    421 
    422 	KDASSERT(cs);
    423 
    424 	DPRINTF_FOLLOW(("cgdiodone(%p)\n", nbp));
    425 	DPRINTF(CGDB_IO, ("cgdiodone: bp %p bcount %d resid %d\n",
    426 	    obp, obp->b_bcount, obp->b_resid));
    427 	DPRINTF(CGDB_IO, (" dev 0x%"PRIx64", nbp %p bn %" PRId64 " addr %p bcnt %d\n",
    428 	    nbp->b_dev, nbp, nbp->b_blkno, nbp->b_data,
    429 	    nbp->b_bcount));
    430 	if (nbp->b_error != 0) {
    431 		obp->b_error = nbp->b_error;
    432 		DPRINTF(CGDB_IO, ("%s: error %d\n", dksc->sc_xname,
    433 		    obp->b_error));
    434 	}
    435 
    436 	/* Perform the decryption if we are reading.
    437 	 *
    438 	 * Note: use the blocknumber from nbp, since it is what
    439 	 *       we used to encrypt the blocks.
    440 	 */
    441 
    442 	if (nbp->b_flags & B_READ)
    443 		cgd_cipher(cs, obp->b_data, obp->b_data, obp->b_bcount,
    444 		    nbp->b_blkno, DEV_BSIZE, CGD_CIPHER_DECRYPT);
    445 
    446 	/* If we allocated memory, free it now... */
    447 	if (nbp->b_data != obp->b_data)
    448 		cgd_putdata(dksc, nbp->b_data);
    449 
    450 	putiobuf(nbp);
    451 
    452 	/* Request is complete for whatever reason */
    453 	obp->b_resid = 0;
    454 	if (obp->b_error != 0)
    455 		obp->b_resid = obp->b_bcount;
    456 	disk_unbusy(&dksc->sc_dkdev, obp->b_bcount - obp->b_resid,
    457 	    (obp->b_flags & B_READ));
    458 	biodone(obp);
    459 	dk_iodone(di, dksc);
    460 }
    461 
    462 /* XXX: we should probably put these into dksubr.c, mostly */
    463 static int
    464 cgdread(dev_t dev, struct uio *uio, int flags)
    465 {
    466 	struct	cgd_softc *cs;
    467 	struct	dk_softc *dksc;
    468 
    469 	DPRINTF_FOLLOW(("cgdread(0x%llx, %p, %d)\n",
    470 	    (unsigned long long)dev, uio, flags));
    471 	GETCGD_SOFTC(cs, dev);
    472 	dksc = &cs->sc_dksc;
    473 	if ((dksc->sc_flags & DKF_INITED) == 0)
    474 		return ENXIO;
    475 	return physio(cgdstrategy, NULL, dev, B_READ, minphys, uio);
    476 }
    477 
    478 /* XXX: we should probably put these into dksubr.c, mostly */
    479 static int
    480 cgdwrite(dev_t dev, struct uio *uio, int flags)
    481 {
    482 	struct	cgd_softc *cs;
    483 	struct	dk_softc *dksc;
    484 
    485 	DPRINTF_FOLLOW(("cgdwrite(0x%"PRIx64", %p, %d)\n", dev, uio, flags));
    486 	GETCGD_SOFTC(cs, dev);
    487 	dksc = &cs->sc_dksc;
    488 	if ((dksc->sc_flags & DKF_INITED) == 0)
    489 		return ENXIO;
    490 	return physio(cgdstrategy, NULL, dev, B_WRITE, minphys, uio);
    491 }
    492 
    493 static int
    494 cgdioctl(dev_t dev, u_long cmd, void *data, int flag, struct lwp *l)
    495 {
    496 	struct	cgd_softc *cs;
    497 	struct	dk_softc *dksc;
    498 	struct	disk *dk;
    499 	int	part = DISKPART(dev);
    500 	int	pmask = 1 << part;
    501 
    502 	DPRINTF_FOLLOW(("cgdioctl(0x%"PRIx64", %ld, %p, %d, %p)\n",
    503 	    dev, cmd, data, flag, l));
    504 	GETCGD_SOFTC(cs, dev);
    505 	dksc = &cs->sc_dksc;
    506 	dk = &dksc->sc_dkdev;
    507 	switch (cmd) {
    508 	case CGDIOCSET:
    509 	case CGDIOCCLR:
    510 		if ((flag & FWRITE) == 0)
    511 			return EBADF;
    512 	}
    513 
    514 	switch (cmd) {
    515 	case CGDIOCSET:
    516 		if (dksc->sc_flags & DKF_INITED)
    517 			return EBUSY;
    518 		return cgd_ioctl_set(cs, data, l);
    519 	case CGDIOCCLR:
    520 		if (DK_BUSY(&cs->sc_dksc, pmask))
    521 			return EBUSY;
    522 		return cgd_ioctl_clr(cs, l);
    523 	case DIOCCACHESYNC:
    524 		/*
    525 		 * XXX Do we really need to care about having a writable
    526 		 * file descriptor here?
    527 		 */
    528 		if ((flag & FWRITE) == 0)
    529 			return (EBADF);
    530 
    531 		/*
    532 		 * We pass this call down to the underlying disk.
    533 		 */
    534 		return VOP_IOCTL(cs->sc_tvn, cmd, data, flag, l->l_cred);
    535 	default:
    536 		return dk_ioctl(di, dksc, dev, cmd, data, flag, l);
    537 	}
    538 }
    539 
    540 static int
    541 cgddump(dev_t dev, daddr_t blkno, void *va, size_t size)
    542 {
    543 	struct	cgd_softc *cs;
    544 
    545 	DPRINTF_FOLLOW(("cgddump(0x%"PRIx64", %" PRId64 ", %p, %lu)\n",
    546 	    dev, blkno, va, (unsigned long)size));
    547 	GETCGD_SOFTC(cs, dev);
    548 	return dk_dump(di, &cs->sc_dksc, dev, blkno, va, size);
    549 }
    550 
    551 /*
    552  * XXXrcd:
    553  *  for now we hardcode the maximum key length.
    554  */
    555 #define MAX_KEYSIZE	1024
    556 
    557 static const struct {
    558 	const char *n;
    559 	int v;
    560 	int d;
    561 } encblkno[] = {
    562 	{ "encblkno",  CGD_CIPHER_CBC_ENCBLKNO8, 1 },
    563 	{ "encblkno8", CGD_CIPHER_CBC_ENCBLKNO8, 1 },
    564 	{ "encblkno1", CGD_CIPHER_CBC_ENCBLKNO1, 8 },
    565 };
    566 
    567 /* ARGSUSED */
    568 static int
    569 cgd_ioctl_set(struct cgd_softc *cs, void *data, struct lwp *l)
    570 {
    571 	struct	 cgd_ioctl *ci = data;
    572 	struct	 vnode *vp;
    573 	int	 ret;
    574 	size_t	 i;
    575 	size_t	 keybytes;			/* key length in bytes */
    576 	const char *cp;
    577 	char	 *inbuf;
    578 
    579 	cp = ci->ci_disk;
    580 	if ((ret = dk_lookup(cp, l, &vp, UIO_USERSPACE)) != 0)
    581 		return ret;
    582 
    583 	inbuf = malloc(MAX_KEYSIZE, M_TEMP, M_WAITOK);
    584 
    585 	if ((ret = cgdinit(cs, cp, vp, l)) != 0)
    586 		goto bail;
    587 
    588 	(void)memset(inbuf, 0, MAX_KEYSIZE);
    589 	ret = copyinstr(ci->ci_alg, inbuf, 256, NULL);
    590 	if (ret)
    591 		goto bail;
    592 	cs->sc_cfuncs = cryptfuncs_find(inbuf);
    593 	if (!cs->sc_cfuncs) {
    594 		ret = EINVAL;
    595 		goto bail;
    596 	}
    597 
    598 	(void)memset(inbuf, 0, MAX_KEYSIZE);
    599 	ret = copyinstr(ci->ci_ivmethod, inbuf, MAX_KEYSIZE, NULL);
    600 	if (ret)
    601 		goto bail;
    602 
    603 	for (i = 0; i < __arraycount(encblkno); i++)
    604 		if (strcmp(encblkno[i].n, inbuf) == 0)
    605 			break;
    606 
    607 	if (i == __arraycount(encblkno)) {
    608 		ret = EINVAL;
    609 		goto bail;
    610 	}
    611 
    612 	keybytes = ci->ci_keylen / 8 + 1;
    613 	if (keybytes > MAX_KEYSIZE) {
    614 		ret = EINVAL;
    615 		goto bail;
    616 	}
    617 
    618 	(void)memset(inbuf, 0, MAX_KEYSIZE);
    619 	ret = copyin(ci->ci_key, inbuf, keybytes);
    620 	if (ret)
    621 		goto bail;
    622 
    623 	cs->sc_cdata.cf_blocksize = ci->ci_blocksize;
    624 	cs->sc_cdata.cf_mode = encblkno[i].v;
    625 	cs->sc_cdata.cf_priv = cs->sc_cfuncs->cf_init(ci->ci_keylen, inbuf,
    626 	    &cs->sc_cdata.cf_blocksize);
    627 	if (cs->sc_cdata.cf_blocksize > CGD_MAXBLOCKSIZE) {
    628 	    log(LOG_WARNING, "cgd: Disallowed cipher with blocksize %zu > %u\n",
    629 		cs->sc_cdata.cf_blocksize, CGD_MAXBLOCKSIZE);
    630 	    cs->sc_cdata.cf_priv = NULL;
    631 	}
    632 
    633 	/*
    634 	 * The blocksize is supposed to be in bytes. Unfortunately originally
    635 	 * it was expressed in bits. For compatibility we maintain encblkno
    636 	 * and encblkno8.
    637 	 */
    638 	cs->sc_cdata.cf_blocksize /= encblkno[i].d;
    639 	(void)memset(inbuf, 0, MAX_KEYSIZE);
    640 	if (!cs->sc_cdata.cf_priv) {
    641 		ret = EINVAL;		/* XXX is this the right error? */
    642 		goto bail;
    643 	}
    644 	free(inbuf, M_TEMP);
    645 
    646 	bufq_alloc(&cs->sc_dksc.sc_bufq, "fcfs", 0);
    647 
    648 	cs->sc_data = malloc(MAXPHYS, M_DEVBUF, M_WAITOK);
    649 	cs->sc_data_used = 0;
    650 
    651 	cs->sc_dksc.sc_flags |= DKF_INITED;
    652 
    653 	/* Attach the disk. */
    654 	disk_attach(&cs->sc_dksc.sc_dkdev);
    655 
    656 	/* Try and read the disklabel. */
    657 	dk_getdisklabel(di, &cs->sc_dksc, 0 /* XXX ? (cause of PR 41704) */);
    658 
    659 	/* Discover wedges on this disk. */
    660 	dkwedge_discover(&cs->sc_dksc.sc_dkdev);
    661 
    662 	return 0;
    663 
    664 bail:
    665 	free(inbuf, M_TEMP);
    666 	(void)vn_close(vp, FREAD|FWRITE, l->l_cred);
    667 	return ret;
    668 }
    669 
    670 /* ARGSUSED */
    671 static int
    672 cgd_ioctl_clr(struct cgd_softc *cs, struct lwp *l)
    673 {
    674 	int	s;
    675 	struct	dk_softc *dksc;
    676 
    677 	dksc = &cs->sc_dksc;
    678 
    679 	if ((dksc->sc_flags & DKF_INITED) == 0)
    680 		return ENXIO;
    681 
    682 	/* Delete all of our wedges. */
    683 	dkwedge_delall(&cs->sc_dksc.sc_dkdev);
    684 
    685 	/* Kill off any queued buffers. */
    686 	s = splbio();
    687 	bufq_drain(cs->sc_dksc.sc_bufq);
    688 	splx(s);
    689 	bufq_free(cs->sc_dksc.sc_bufq);
    690 
    691 	(void)vn_close(cs->sc_tvn, FREAD|FWRITE, l->l_cred);
    692 	cs->sc_cfuncs->cf_destroy(cs->sc_cdata.cf_priv);
    693 	free(cs->sc_tpath, M_DEVBUF);
    694 	free(cs->sc_data, M_DEVBUF);
    695 	cs->sc_data_used = 0;
    696 	cs->sc_dksc.sc_flags &= ~DKF_INITED;
    697 	disk_detach(&cs->sc_dksc.sc_dkdev);
    698 
    699 	return 0;
    700 }
    701 
    702 static int
    703 getsize(struct lwp *l, struct vnode *vp, size_t *size)
    704 {
    705 	struct partinfo dpart;
    706 	struct dkwedge_info dkw;
    707 	int ret;
    708 
    709 	if ((ret = VOP_IOCTL(vp, DIOCGWEDGEINFO, &dkw, FREAD,
    710 	    l->l_cred)) == 0) {
    711 		*size = dkw.dkw_size;
    712 		return 0;
    713 	}
    714 
    715 	if ((ret = VOP_IOCTL(vp, DIOCGPART, &dpart, FREAD, l->l_cred)) == 0) {
    716 		*size = dpart.part->p_size;
    717 		return 0;
    718 	}
    719 
    720 	return ret;
    721 }
    722 
    723 
    724 static int
    725 cgdinit(struct cgd_softc *cs, const char *cpath, struct vnode *vp,
    726 	struct lwp *l)
    727 {
    728 	struct	dk_geom *pdg;
    729 	struct	vattr va;
    730 	size_t	size;
    731 	int	ret;
    732 	char	*tmppath;
    733 
    734 	cs->sc_dksc.sc_size = 0;
    735 	cs->sc_tvn = vp;
    736 	cs->sc_tpath = NULL;
    737 
    738 	tmppath = malloc(MAXPATHLEN, M_TEMP, M_WAITOK);
    739 	ret = copyinstr(cpath, tmppath, MAXPATHLEN, &cs->sc_tpathlen);
    740 	if (ret)
    741 		goto bail;
    742 	cs->sc_tpath = malloc(cs->sc_tpathlen, M_DEVBUF, M_WAITOK);
    743 	memcpy(cs->sc_tpath, tmppath, cs->sc_tpathlen);
    744 
    745 	if ((ret = VOP_GETATTR(vp, &va, l->l_cred)) != 0)
    746 		goto bail;
    747 
    748 	cs->sc_tdev = va.va_rdev;
    749 
    750 	if ((ret = getsize(l, vp, &size)) != 0)
    751 		goto bail;
    752 
    753 	if (!size) {
    754 		ret = ENODEV;
    755 		goto bail;
    756 	}
    757 
    758 	cs->sc_dksc.sc_size = size;
    759 
    760 	/*
    761 	 * XXX here we should probe the underlying device.  If we
    762 	 *     are accessing a partition of type RAW_PART, then
    763 	 *     we should populate our initial geometry with the
    764 	 *     geometry that we discover from the device.
    765 	 */
    766 	pdg = &cs->sc_dksc.sc_geom;
    767 	pdg->pdg_secsize = DEV_BSIZE;
    768 	pdg->pdg_ntracks = 1;
    769 	pdg->pdg_nsectors = 1024 * (1024 / pdg->pdg_secsize);
    770 	pdg->pdg_ncylinders = cs->sc_dksc.sc_size / pdg->pdg_nsectors;
    771 
    772 bail:
    773 	free(tmppath, M_TEMP);
    774 	if (ret && cs->sc_tpath)
    775 		free(cs->sc_tpath, M_DEVBUF);
    776 	return ret;
    777 }
    778 
    779 /*
    780  * Our generic cipher entry point.  This takes care of the
    781  * IV mode and passes off the work to the specific cipher.
    782  * We implement here the IV method ``encrypted block
    783  * number''.
    784  *
    785  * For the encryption case, we accomplish this by setting
    786  * up a struct uio where the first iovec of the source is
    787  * the blocknumber and the first iovec of the dest is a
    788  * sink.  We then call the cipher with an IV of zero, and
    789  * the right thing happens.
    790  *
    791  * For the decryption case, we use the same basic mechanism
    792  * for symmetry, but we encrypt the block number in the
    793  * first iovec.
    794  *
    795  * We mainly do this to avoid requiring the definition of
    796  * an ECB mode.
    797  *
    798  * XXXrcd: for now we rely on our own crypto framework defined
    799  *         in dev/cgd_crypto.c.  This will change when we
    800  *         get a generic kernel crypto framework.
    801  */
    802 
    803 static void
    804 blkno2blkno_buf(char *sbuf, daddr_t blkno)
    805 {
    806 	int	i;
    807 
    808 	/* Set up the blkno in blkno_buf, here we do not care much
    809 	 * about the final layout of the information as long as we
    810 	 * can guarantee that each sector will have a different IV
    811 	 * and that the endianness of the machine will not affect
    812 	 * the representation that we have chosen.
    813 	 *
    814 	 * We choose this representation, because it does not rely
    815 	 * on the size of buf (which is the blocksize of the cipher),
    816 	 * but allows daddr_t to grow without breaking existing
    817 	 * disks.
    818 	 *
    819 	 * Note that blkno2blkno_buf does not take a size as input,
    820 	 * and hence must be called on a pre-zeroed buffer of length
    821 	 * greater than or equal to sizeof(daddr_t).
    822 	 */
    823 	for (i=0; i < sizeof(daddr_t); i++) {
    824 		*sbuf++ = blkno & 0xff;
    825 		blkno >>= 8;
    826 	}
    827 }
    828 
    829 static void
    830 cgd_cipher(struct cgd_softc *cs, void *dstv, void *srcv,
    831     size_t len, daddr_t blkno, size_t secsize, int dir)
    832 {
    833 	char		*dst = dstv;
    834 	char 		*src = srcv;
    835 	cfunc_cipher	*cipher = cs->sc_cfuncs->cf_cipher;
    836 	struct uio	dstuio;
    837 	struct uio	srcuio;
    838 	struct iovec	dstiov[2];
    839 	struct iovec	srciov[2];
    840 	size_t		blocksize = cs->sc_cdata.cf_blocksize;
    841 	char		sink[CGD_MAXBLOCKSIZE];
    842 	char		zero_iv[CGD_MAXBLOCKSIZE];
    843 	char		blkno_buf[CGD_MAXBLOCKSIZE];
    844 
    845 	DPRINTF_FOLLOW(("cgd_cipher() dir=%d\n", dir));
    846 
    847 	DIAGCONDPANIC(len % blocksize != 0,
    848 	    ("cgd_cipher: len %% blocksize != 0"));
    849 
    850 	/* ensure that sizeof(daddr_t) <= blocksize (for encblkno IVing) */
    851 	DIAGCONDPANIC(sizeof(daddr_t) > blocksize,
    852 	    ("cgd_cipher: sizeof(daddr_t) > blocksize"));
    853 
    854 	memset(zero_iv, 0x0, blocksize);
    855 
    856 	dstuio.uio_iov = dstiov;
    857 	dstuio.uio_iovcnt = 2;
    858 
    859 	srcuio.uio_iov = srciov;
    860 	srcuio.uio_iovcnt = 2;
    861 
    862 	dstiov[0].iov_base = sink;
    863 	dstiov[0].iov_len  = blocksize;
    864 	srciov[0].iov_base = blkno_buf;
    865 	srciov[0].iov_len  = blocksize;
    866 	dstiov[1].iov_len  = secsize;
    867 	srciov[1].iov_len  = secsize;
    868 
    869 	for (; len > 0; len -= secsize) {
    870 		dstiov[1].iov_base = dst;
    871 		srciov[1].iov_base = src;
    872 
    873 		memset(blkno_buf, 0x0, blocksize);
    874 		blkno2blkno_buf(blkno_buf, blkno);
    875 		if (dir == CGD_CIPHER_DECRYPT) {
    876 			dstuio.uio_iovcnt = 1;
    877 			srcuio.uio_iovcnt = 1;
    878 			IFDEBUG(CGDB_CRYPTO, hexprint("step 0: blkno_buf",
    879 			    blkno_buf, blocksize));
    880 			cipher(cs->sc_cdata.cf_priv, &dstuio, &srcuio,
    881 			    zero_iv, CGD_CIPHER_ENCRYPT);
    882 			memcpy(blkno_buf, sink, blocksize);
    883 			dstuio.uio_iovcnt = 2;
    884 			srcuio.uio_iovcnt = 2;
    885 		}
    886 
    887 		IFDEBUG(CGDB_CRYPTO, hexprint("step 1: blkno_buf",
    888 		    blkno_buf, blocksize));
    889 		cipher(cs->sc_cdata.cf_priv, &dstuio, &srcuio, zero_iv, dir);
    890 		IFDEBUG(CGDB_CRYPTO, hexprint("step 2: sink",
    891 		    sink, blocksize));
    892 
    893 		dst += secsize;
    894 		src += secsize;
    895 		blkno++;
    896 	}
    897 }
    898 
    899 #ifdef DEBUG
    900 static void
    901 hexprint(const char *start, void *buf, int len)
    902 {
    903 	char	*c = buf;
    904 
    905 	DIAGCONDPANIC(len < 0, ("hexprint: called with len < 0"));
    906 	printf("%s: len=%06d 0x", start, len);
    907 	while (len--)
    908 		printf("%02x", (unsigned char) *c++);
    909 }
    910 #endif
    911 
    912 #ifdef _MODULE
    913 
    914 #include <sys/module.h>
    915 
    916 MODULE(MODULE_CLASS_DRIVER, cgd, NULL);
    917 CFDRIVER_DECL(cgd, DV_DISK, NULL);
    918 
    919 static int
    920 cgd_modcmd(modcmd_t cmd, void *arg)
    921 {
    922 	int bmajor = -1, cmajor = -1,  error = 0;
    923 
    924 	switch (cmd) {
    925 	case MODULE_CMD_INIT:
    926 		error = config_cfdriver_attach(&cgd_cd);
    927 		if (error)
    928 			break;
    929 
    930 		error = config_cfattach_attach(cgd_cd.cd_name, &cgd_ca);
    931 	        if (error) {
    932 			config_cfdriver_detach(&cgd_cd);
    933 			aprint_error("%s: unable to register cfattach\n",
    934 			    cgd_cd.cd_name);
    935 			break;
    936 		}
    937 
    938 		error = devsw_attach("cgd", &cgd_bdevsw, &bmajor,
    939 		    &cgd_cdevsw, &cmajor);
    940 		if (error) {
    941 			config_cfattach_detach(cgd_cd.cd_name, &cgd_ca);
    942 			config_cfdriver_detach(&cgd_cd);
    943 			break;
    944 		}
    945 
    946 		break;
    947 
    948 	case MODULE_CMD_FINI:
    949 		error = config_cfattach_detach(cgd_cd.cd_name, &cgd_ca);
    950 		if (error)
    951 			break;
    952 		config_cfdriver_detach(&cgd_cd);
    953 		devsw_detach(&cgd_bdevsw, &cgd_cdevsw);
    954 		break;
    955 
    956 	case MODULE_CMD_STAT:
    957 		return ENOTTY;
    958 
    959 	default:
    960 		return ENOTTY;
    961 	}
    962 
    963 	return error;
    964 }
    965 
    966 #endif
    967