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