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