kernel/drivers/crypto/nx/nx.c

   1 /**
   2  * Routines supporting the Power 7+ Nest Accelerators driver
   3  *
   4  * Copyright (C) 2011-2012 International Business Machines Inc.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; version 2 only.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  *
  19  * Author: Kent Yoder <yoder1@us.ibm.com>
  20  */
  21
  22 #include <crypto/internal/hash.h>
  23 #include <crypto/hash.h>
  24 #include <crypto/aes.h>
  25 #include <crypto/sha.h>
  26 #include <crypto/algapi.h>
  27 #include <crypto/scatterwalk.h>
  28 #include <linux/module.h>
  29 #include <linux/moduleparam.h>
  30 #include <linux/types.h>
  31 #include <linux/mm.h>
  32 #include <linux/crypto.h>
  33 #include <linux/scatterlist.h>
  34 #include <linux/device.h>
  35 #include <linux/of.h>
  36 #include <asm/hvcall.h>
  37 #include <asm/vio.h>
  38
  39 #include "nx_csbcpb.h"
  40 #include "nx.h"
  41
  42
  43 /**
  44  * nx_hcall_sync - make an H_COP_OP hcall for the passed in op structure
  45  *
  46  * @nx_ctx: the crypto context handle
  47  * @op: PFO operation struct to pass in
  48  * @may_sleep: flag indicating the request can sleep
  49  *
  50  * Make the hcall, retrying while the hardware is busy. If we cannot yield
  51  * the thread, limit the number of retries to 10 here.
  52  */
  53 int nx_hcall_sync(struct nx_crypto_ctx *nx_ctx,
  54                   struct vio_pfo_op    *op,
  55                   u32                   may_sleep)
  56 {
  57         int rc, retries = 10;
  58         struct vio_dev *viodev = nx_driver.viodev;
  59
  60         atomic_inc(&(nx_ctx->stats->sync_ops));
  61
  62         do {
  63                 rc = vio_h_cop_sync(viodev, op);
  64         } while (rc == -EBUSY && !may_sleep && retries--);
  65
  66         if (rc) {
  67                 dev_dbg(&viodev->dev, "vio_h_cop_sync failed: rc: %d "
  68                         "hcall rc: %ld\n", rc, op->hcall_err);
  69                 atomic_inc(&(nx_ctx->stats->errors));
  70                 atomic_set(&(nx_ctx->stats->last_error), op->hcall_err);
  71                 atomic_set(&(nx_ctx->stats->last_error_pid), current->pid);
  72         }
  73
  74         return rc;
  75 }
  76
  77 /**
  78  * nx_build_sg_list - build an NX scatter list describing a single  buffer
  79  *
  80  * @sg_head: pointer to the first scatter list element to build
  81  * @start_addr: pointer to the linear buffer
  82  * @len: length of the data at @start_addr
  83  * @sgmax: the largest number of scatter list elements we're allowed to create
  84  *
  85  * This function will start writing nx_sg elements at @sg_head and keep
  86  * writing them until all of the data from @start_addr is described or
  87  * until sgmax elements have been written. Scatter list elements will be
  88  * created such that none of the elements describes a buffer that crosses a 4K
  89  * boundary.
  90  */
  91 struct nx_sg *nx_build_sg_list(struct nx_sg *sg_head,
  92                                u8           *start_addr,
  93                                unsigned int *len,
  94                                u32           sgmax)
  95 {
  96         unsigned int sg_len = 0;
  97         struct nx_sg *sg;
  98         u64 sg_addr = (u64)start_addr;
  99         u64 end_addr;
 100
 101         /* determine the start and end for this address range - slightly
 102          * different if this is in VMALLOC_REGION */
 103         if (is_vmalloc_addr(start_addr))
 104                 sg_addr = page_to_phys(vmalloc_to_page(start_addr))
 105                           + offset_in_page(sg_addr);
 106         else
 107                 sg_addr = __pa(sg_addr);
 108
 109         end_addr = sg_addr + *len;
 110
 111         /* each iteration will write one struct nx_sg element and add the
 112          * length of data described by that element to sg_len. Once @len bytes
 113          * have been described (or @sgmax elements have been written), the
 114          * loop ends. min_t is used to ensure @end_addr falls on the same page
 115          * as sg_addr, if not, we need to create another nx_sg element for the
 116          * data on the next page.
 117          *
 118          * Also when using vmalloc'ed data, every time that a system page
 119          * boundary is crossed the physical address needs to be re-calculated.
 120          */
 121         for (sg = sg_head; sg_len < *len; sg++) {
 122                 u64 next_page;
 123
 124                 sg->addr = sg_addr;
 125                 sg_addr = min_t(u64, NX_PAGE_NUM(sg_addr + NX_PAGE_SIZE),
 126                                 end_addr);
 127
 128                 next_page = (sg->addr & PAGE_MASK) + PAGE_SIZE;
 129                 sg->len = min_t(u64, sg_addr, next_page) - sg->addr;
 130                 sg_len += sg->len;
 131
 132                 if (sg_addr >= next_page &&
 133                                 is_vmalloc_addr(start_addr + sg_len)) {
 134                         sg_addr = page_to_phys(vmalloc_to_page(
 135                                                 start_addr + sg_len));
 136                         end_addr = sg_addr + *len - sg_len;
 137                 }
 138
 139                 if ((sg - sg_head) == sgmax) {
 140                         pr_err("nx: scatter/gather list overflow, pid: %d\n",
 141                                current->pid);
 142                         sg++;
 143                         break;
 144                 }
 145         }
 146         *len = sg_len;
 147
 148         /* return the moved sg_head pointer */
 149         return sg;
 150 }
 151
 152 /**
 153  * nx_walk_and_build - walk a linux scatterlist and build an nx scatterlist
 154  *
 155  * @nx_dst: pointer to the first nx_sg element to write
 156  * @sglen: max number of nx_sg entries we're allowed to write
 157  * @sg_src: pointer to the source linux scatterlist to walk
 158  * @start: number of bytes to fast-forward past at the beginning of @sg_src
 159  * @src_len: number of bytes to walk in @sg_src
 160  */
 161 struct nx_sg *nx_walk_and_build(struct nx_sg       *nx_dst,
 162                                 unsigned int        sglen,
 163                                 struct scatterlist *sg_src,
 164                                 unsigned int        start,
 165                                 unsigned int       *src_len)
 166 {
 167         struct scatter_walk walk;
 168         struct nx_sg *nx_sg = nx_dst;
 169         unsigned int n, offset = 0, len = *src_len;
 170         char *dst;
 171
 172         /* we need to fast forward through @start bytes first */
 173         for (;;) {
 174                 scatterwalk_start(&walk, sg_src);
 175
 176                 if (start < offset + sg_src->length)
 177                         break;
 178
 179                 offset += sg_src->length;
 180                 sg_src = sg_next(sg_src);
 181         }
 182
 183         /* start - offset is the number of bytes to advance in the scatterlist
 184          * element we're currently looking at */
 185         scatterwalk_advance(&walk, start - offset);
 186
 187         while (len && (nx_sg - nx_dst) < sglen) {
 188                 n = scatterwalk_clamp(&walk, len);
 189                 if (!n) {
 190                         /* In cases where we have scatterlist chain sg_next
 191                          * handles with it properly */
 192                         scatterwalk_start(&walk, sg_next(walk.sg));
 193                         n = scatterwalk_clamp(&walk, len);
 194                 }
 195                 dst = scatterwalk_map(&walk);
 196
 197                 nx_sg = nx_build_sg_list(nx_sg, dst, &n, sglen - (nx_sg - nx_dst));
 198                 len -= n;
 199
 200                 scatterwalk_unmap(dst);
 201                 scatterwalk_advance(&walk, n);
 202                 scatterwalk_done(&walk, SCATTERWALK_FROM_SG, len);
 203         }
 204         /* update to_process */
 205         *src_len -= len;
 206
 207         /* return the moved destination pointer */
 208         return nx_sg;
 209 }
 210
 211 /**
 212  * trim_sg_list - ensures the bound in sg list.
 213  * @sg: sg list head
 214  * @end: sg lisg end
 215  * @delta:  is the amount we need to crop in order to bound the list.
 216  *
 217  */
 218 static long int trim_sg_list(struct nx_sg *sg,
 219                              struct nx_sg *end,
 220                              unsigned int delta,
 221                              unsigned int *nbytes)
 222 {
 223         long int oplen;
 224         long int data_back;
 225         unsigned int is_delta = delta;
 226
 227         while (delta && end > sg) {
 228                 struct nx_sg *last = end - 1;
 229
 230                 if (last->len > delta) {
 231                         last->len -= delta;
 232                         delta = 0;
 233                 } else {
 234                         end--;
 235                         delta -= last->len;
 236                 }
 237         }
 238
 239         /* There are cases where we need to crop list in order to make it
 240          * a block size multiple, but we also need to align data. In order to
 241          * that we need to calculate how much we need to put back to be
 242          * processed
 243          */
 244         oplen = (sg - end) * sizeof(struct nx_sg);
 245         if (is_delta) {
 246                 data_back = (abs(oplen) / AES_BLOCK_SIZE) *  sg->len;
 247                 data_back = *nbytes - (data_back & ~(AES_BLOCK_SIZE - 1));
 248                 *nbytes -= data_back;
 249         }
 250
 251         return oplen;
 252 }
 253
 254 /**
 255  * nx_build_sg_lists - walk the input scatterlists and build arrays of NX
 256  *                     scatterlists based on them.
 257  *
 258  * @nx_ctx: NX crypto context for the lists we're building
 259  * @desc: the block cipher descriptor for the operation
 260  * @dst: destination scatterlist
 261  * @src: source scatterlist
 262  * @nbytes: length of data described in the scatterlists
 263  * @offset: number of bytes to fast-forward past at the beginning of
 264  *          scatterlists.
 265  * @iv: destination for the iv data, if the algorithm requires it
 266  *
 267  * This is common code shared by all the AES algorithms. It uses the block
 268  * cipher walk routines to traverse input and output scatterlists, building
 269  * corresponding NX scatterlists
 270  */
 271 int nx_build_sg_lists(struct nx_crypto_ctx  *nx_ctx,
 272                       struct blkcipher_desc *desc,
 273                       struct scatterlist    *dst,
 274                       struct scatterlist    *src,
 275                       unsigned int          *nbytes,
 276                       unsigned int           offset,
 277                       u8                    *iv)
 278 {
 279         unsigned int delta = 0;
 280         unsigned int total = *nbytes;
 281         struct nx_sg *nx_insg = nx_ctx->in_sg;
 282         struct nx_sg *nx_outsg = nx_ctx->out_sg;
 283         unsigned int max_sg_len;
 284
 285         max_sg_len = min_t(u64, nx_ctx->ap->sglen,
 286                         nx_driver.of.max_sg_len/sizeof(struct nx_sg));
 287         max_sg_len = min_t(u64, max_sg_len,
 288                         nx_ctx->ap->databytelen/NX_PAGE_SIZE);
 289
 290         if (iv)
 291                 memcpy(iv, desc->info, AES_BLOCK_SIZE);
 292
 293         *nbytes = min_t(u64, *nbytes, nx_ctx->ap->databytelen);
 294
 295         nx_outsg = nx_walk_and_build(nx_outsg, max_sg_len, dst,
 296                                         offset, nbytes);
 297         nx_insg = nx_walk_and_build(nx_insg, max_sg_len, src,
 298                                         offset, nbytes);
 299
 300         if (*nbytes < total)
 301                 delta = *nbytes - (*nbytes & ~(AES_BLOCK_SIZE - 1));
 302
 303         /* these lengths should be negative, which will indicate to phyp that
 304          * the input and output parameters are scatterlists, not linear
 305          * buffers */
 306         nx_ctx->op.inlen = trim_sg_list(nx_ctx->in_sg, nx_insg, delta, nbytes);
 307         nx_ctx->op.outlen = trim_sg_list(nx_ctx->out_sg, nx_outsg, delta, nbytes);
 308
 309         return 0;
 310 }
 311
 312 /**
 313  * nx_ctx_init - initialize an nx_ctx's vio_pfo_op struct
 314  *
 315  * @nx_ctx: the nx context to initialize
 316  * @function: the function code for the op
 317  */
 318 void nx_ctx_init(struct nx_crypto_ctx *nx_ctx, unsigned int function)
 319 {
 320         spin_lock_init(&nx_ctx->lock);
 321         memset(nx_ctx->kmem, 0, nx_ctx->kmem_len);
 322         nx_ctx->csbcpb->csb.valid |= NX_CSB_VALID_BIT;
 323
 324         nx_ctx->op.flags = function;
 325         nx_ctx->op.csbcpb = __pa(nx_ctx->csbcpb);
 326         nx_ctx->op.in = __pa(nx_ctx->in_sg);
 327         nx_ctx->op.out = __pa(nx_ctx->out_sg);
 328
 329         if (nx_ctx->csbcpb_aead) {
 330                 nx_ctx->csbcpb_aead->csb.valid |= NX_CSB_VALID_BIT;
 331
 332                 nx_ctx->op_aead.flags = function;
 333                 nx_ctx->op_aead.csbcpb = __pa(nx_ctx->csbcpb_aead);
 334                 nx_ctx->op_aead.in = __pa(nx_ctx->in_sg);
 335                 nx_ctx->op_aead.out = __pa(nx_ctx->out_sg);
 336         }
 337 }
 338
 339 static void nx_of_update_status(struct device   *dev,
 340                                struct property *p,
 341                                struct nx_of    *props)
 342 {
 343         if (!strncmp(p->value, "okay", p->length)) {
 344                 props->status = NX_WAITING;
 345                 props->flags |= NX_OF_FLAG_STATUS_SET;
 346         } else {
 347                 dev_info(dev, "%s: status '%s' is not 'okay'\n", __func__,
 348                          (char *)p->value);
 349         }
 350 }
 351
 352 static void nx_of_update_sglen(struct device   *dev,
 353                                struct property *p,
 354                                struct nx_of    *props)
 355 {
 356         if (p->length != sizeof(props->max_sg_len)) {
 357                 dev_err(dev, "%s: unexpected format for "
 358                         "ibm,max-sg-len property\n", __func__);
 359                 dev_dbg(dev, "%s: ibm,max-sg-len is %d bytes "
 360                         "long, expected %zd bytes\n", __func__,
 361                         p->length, sizeof(props->max_sg_len));
 362                 return;
 363         }
 364
 365         props->max_sg_len = *(u32 *)p->value;
 366         props->flags |= NX_OF_FLAG_MAXSGLEN_SET;
 367 }
 368
 369 static void nx_of_update_msc(struct device   *dev,
 370                              struct property *p,
 371                              struct nx_of    *props)
 372 {
 373         struct msc_triplet *trip;
 374         struct max_sync_cop *msc;
 375         unsigned int bytes_so_far, i, lenp;
 376
 377         msc = (struct max_sync_cop *)p->value;
 378         lenp = p->length;
 379
 380         /* You can't tell if the data read in for this property is sane by its
 381          * size alone. This is because there are sizes embedded in the data
 382          * structure. The best we can do is check lengths as we parse and bail
 383          * as soon as a length error is detected. */
 384         bytes_so_far = 0;
 385
 386         while ((bytes_so_far + sizeof(struct max_sync_cop)) <= lenp) {
 387                 bytes_so_far += sizeof(struct max_sync_cop);
 388
 389                 trip = msc->trip;
 390
 391                 for (i = 0;
 392                      ((bytes_so_far + sizeof(struct msc_triplet)) <= lenp) &&
 393                      i < msc->triplets;
 394                      i++) {
 395                         if (msc->fc > NX_MAX_FC || msc->mode > NX_MAX_MODE) {
 396                                 dev_err(dev, "unknown function code/mode "
 397                                         "combo: %d/%d (ignored)\n", msc->fc,
 398                                         msc->mode);
 399                                 goto next_loop;
 400                         }
 401
 402                         switch (trip->keybitlen) {
 403                         case 128:
 404                         case 160:
 405                                 props->ap[msc->fc][msc->mode][0].databytelen =
 406                                         trip->databytelen;
 407                                 props->ap[msc->fc][msc->mode][0].sglen =
 408                                         trip->sglen;
 409                                 break;
 410                         case 192:
 411                                 props->ap[msc->fc][msc->mode][1].databytelen =
 412                                         trip->databytelen;
 413                                 props->ap[msc->fc][msc->mode][1].sglen =
 414                                         trip->sglen;
 415                                 break;
 416                         case 256:
 417                                 if (msc->fc == NX_FC_AES) {
 418                                         props->ap[msc->fc][msc->mode][2].
 419                                                 databytelen = trip->databytelen;
 420                                         props->ap[msc->fc][msc->mode][2].sglen =
 421                                                 trip->sglen;
 422                                 } else if (msc->fc == NX_FC_AES_HMAC ||
 423                                            msc->fc == NX_FC_SHA) {
 424                                         props->ap[msc->fc][msc->mode][1].
 425                                                 databytelen = trip->databytelen;
 426                                         props->ap[msc->fc][msc->mode][1].sglen =
 427                                                 trip->sglen;
 428                                 } else {
 429                                         dev_warn(dev, "unknown function "
 430                                                 "code/key bit len combo"
 431                                                 ": (%u/256)\n", msc->fc);
 432                                 }
 433                                 break;
 434                         case 512:
 435                                 props->ap[msc->fc][msc->mode][2].databytelen =
 436                                         trip->databytelen;
 437                                 props->ap[msc->fc][msc->mode][2].sglen =
 438                                         trip->sglen;
 439                                 break;
 440                         default:
 441                                 dev_warn(dev, "unknown function code/key bit "
 442                                          "len combo: (%u/%u)\n", msc->fc,
 443                                          trip->keybitlen);
 444                                 break;
 445                         }
 446 next_loop:
 447                         bytes_so_far += sizeof(struct msc_triplet);
 448                         trip++;
 449                 }
 450
 451                 msc = (struct max_sync_cop *)trip;
 452         }
 453
 454         props->flags |= NX_OF_FLAG_MAXSYNCCOP_SET;
 455 }
 456
 457 /**
 458  * nx_of_init - read openFirmware values from the device tree
 459  *
 460  * @dev: device handle
 461  * @props: pointer to struct to hold the properties values
 462  *
 463  * Called once at driver probe time, this function will read out the
 464  * openFirmware properties we use at runtime. If all the OF properties are
 465  * acceptable, when we exit this function props->flags will indicate that
 466  * we're ready to register our crypto algorithms.
 467  */
 468 static void nx_of_init(struct device *dev, struct nx_of *props)
 469 {
 470         struct device_node *base_node = dev->of_node;
 471         struct property *p;
 472
 473         p = of_find_property(base_node, "status", NULL);
 474         if (!p)
 475                 dev_info(dev, "%s: property 'status' not found\n", __func__);
 476         else
 477                 nx_of_update_status(dev, p, props);
 478
 479         p = of_find_property(base_node, "ibm,max-sg-len", NULL);
 480         if (!p)
 481                 dev_info(dev, "%s: property 'ibm,max-sg-len' not found\n",
 482                          __func__);
 483         else
 484                 nx_of_update_sglen(dev, p, props);
 485
 486         p = of_find_property(base_node, "ibm,max-sync-cop", NULL);
 487         if (!p)
 488                 dev_info(dev, "%s: property 'ibm,max-sync-cop' not found\n",
 489                          __func__);
 490         else
 491                 nx_of_update_msc(dev, p, props);
 492 }
 493
 494 /**
 495  * nx_register_algs - register algorithms with the crypto API
 496  *
 497  * Called from nx_probe()
 498  *
 499  * If all OF properties are in an acceptable state, the driver flags will
 500  * indicate that we're ready and we'll create our debugfs files and register
 501  * out crypto algorithms.
 502  */
 503 static int nx_register_algs(void)
 504 {
 505         int rc = -1;
 506
 507         if (nx_driver.of.flags != NX_OF_FLAG_MASK_READY)
 508                 goto out;
 509
 510         memset(&nx_driver.stats, 0, sizeof(struct nx_stats));
 511
 512         rc = NX_DEBUGFS_INIT(&nx_driver);
 513         if (rc)
 514                 goto out;
 515
 516         nx_driver.of.status = NX_OKAY;
 517
 518         rc = crypto_register_alg(&nx_ecb_aes_alg);
 519         if (rc)
 520                 goto out;
 521
 522         rc = crypto_register_alg(&nx_cbc_aes_alg);
 523         if (rc)
 524                 goto out_unreg_ecb;
 525
 526         rc = crypto_register_alg(&nx_ctr_aes_alg);
 527         if (rc)
 528                 goto out_unreg_cbc;
 529
 530         rc = crypto_register_alg(&nx_ctr3686_aes_alg);
 531         if (rc)
 532                 goto out_unreg_ctr;
 533
 534         rc = crypto_register_alg(&nx_gcm_aes_alg);
 535         if (rc)
 536                 goto out_unreg_ctr3686;
 537
 538         rc = crypto_register_alg(&nx_gcm4106_aes_alg);
 539         if (rc)
 540                 goto out_unreg_gcm;
 541
 542         rc = crypto_register_alg(&nx_ccm_aes_alg);
 543         if (rc)
 544                 goto out_unreg_gcm4106;
 545
 546         rc = crypto_register_alg(&nx_ccm4309_aes_alg);
 547         if (rc)
 548                 goto out_unreg_ccm;
 549
 550         rc = crypto_register_shash(&nx_shash_sha256_alg);
 551         if (rc)
 552                 goto out_unreg_ccm4309;
 553
 554         rc = crypto_register_shash(&nx_shash_sha512_alg);
 555         if (rc)
 556                 goto out_unreg_s256;
 557
 558         rc = crypto_register_shash(&nx_shash_aes_xcbc_alg);
 559         if (rc)
 560                 goto out_unreg_s512;
 561
 562         goto out;
 563
 564 out_unreg_s512:
 565         crypto_unregister_shash(&nx_shash_sha512_alg);
 566 out_unreg_s256:
 567         crypto_unregister_shash(&nx_shash_sha256_alg);
 568 out_unreg_ccm4309:
 569         crypto_unregister_alg(&nx_ccm4309_aes_alg);
 570 out_unreg_ccm:
 571         crypto_unregister_alg(&nx_ccm_aes_alg);
 572 out_unreg_gcm4106:
 573         crypto_unregister_alg(&nx_gcm4106_aes_alg);
 574 out_unreg_gcm:
 575         crypto_unregister_alg(&nx_gcm_aes_alg);
 576 out_unreg_ctr3686:
 577         crypto_unregister_alg(&nx_ctr3686_aes_alg);
 578 out_unreg_ctr:
 579         crypto_unregister_alg(&nx_ctr_aes_alg);
 580 out_unreg_cbc:
 581         crypto_unregister_alg(&nx_cbc_aes_alg);
 582 out_unreg_ecb:
 583         crypto_unregister_alg(&nx_ecb_aes_alg);
 584 out:
 585         return rc;
 586 }
 587
 588 /**
 589  * nx_crypto_ctx_init - create and initialize a crypto api context
 590  *
 591  * @nx_ctx: the crypto api context
 592  * @fc: function code for the context
 593  * @mode: the function code specific mode for this context
 594  */
 595 static int nx_crypto_ctx_init(struct nx_crypto_ctx *nx_ctx, u32 fc, u32 mode)
 596 {
 597         if (nx_driver.of.status != NX_OKAY) {
 598                 pr_err("Attempt to initialize NX crypto context while device "
 599                        "is not available!\n");
 600                 return -ENODEV;
 601         }
 602
 603         /* we need an extra page for csbcpb_aead for these modes */
 604         if (mode == NX_MODE_AES_GCM || mode == NX_MODE_AES_CCM)
 605                 nx_ctx->kmem_len = (5 * NX_PAGE_SIZE) +
 606                                    sizeof(struct nx_csbcpb);
 607         else
 608                 nx_ctx->kmem_len = (4 * NX_PAGE_SIZE) +
 609                                    sizeof(struct nx_csbcpb);
 610
 611         nx_ctx->kmem = kmalloc(nx_ctx->kmem_len, GFP_KERNEL);
 612         if (!nx_ctx->kmem)
 613                 return -ENOMEM;
 614
 615         /* the csbcpb and scatterlists must be 4K aligned pages */
 616         nx_ctx->csbcpb = (struct nx_csbcpb *)(round_up((u64)nx_ctx->kmem,
 617                                                        (u64)NX_PAGE_SIZE));
 618         nx_ctx->in_sg = (struct nx_sg *)((u8 *)nx_ctx->csbcpb + NX_PAGE_SIZE);
 619         nx_ctx->out_sg = (struct nx_sg *)((u8 *)nx_ctx->in_sg + NX_PAGE_SIZE);
 620
 621         if (mode == NX_MODE_AES_GCM || mode == NX_MODE_AES_CCM)
 622                 nx_ctx->csbcpb_aead =
 623                         (struct nx_csbcpb *)((u8 *)nx_ctx->out_sg +
 624                                              NX_PAGE_SIZE);
 625
 626         /* give each context a pointer to global stats and their OF
 627          * properties */
 628         nx_ctx->stats = &nx_driver.stats;
 629         memcpy(nx_ctx->props, nx_driver.of.ap[fc][mode],
 630                sizeof(struct alg_props) * 3);
 631
 632         return 0;
 633 }
 634
 635 /* entry points from the crypto tfm initializers */
 636 int nx_crypto_ctx_aes_ccm_init(struct crypto_tfm *tfm)
 637 {
 638         tfm->crt_aead.reqsize = sizeof(struct nx_ccm_rctx);
 639         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 640                                   NX_MODE_AES_CCM);
 641 }
 642
 643 int nx_crypto_ctx_aes_gcm_init(struct crypto_tfm *tfm)
 644 {
 645         tfm->crt_aead.reqsize = sizeof(struct nx_gcm_rctx);
 646         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 647                                   NX_MODE_AES_GCM);
 648 }
 649
 650 int nx_crypto_ctx_aes_ctr_init(struct crypto_tfm *tfm)
 651 {
 652         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 653                                   NX_MODE_AES_CTR);
 654 }
 655
 656 int nx_crypto_ctx_aes_cbc_init(struct crypto_tfm *tfm)
 657 {
 658         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 659                                   NX_MODE_AES_CBC);
 660 }
 661
 662 int nx_crypto_ctx_aes_ecb_init(struct crypto_tfm *tfm)
 663 {
 664         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 665                                   NX_MODE_AES_ECB);
 666 }
 667
 668 int nx_crypto_ctx_sha_init(struct crypto_tfm *tfm)
 669 {
 670         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_SHA, NX_MODE_SHA);
 671 }
 672
 673 int nx_crypto_ctx_aes_xcbc_init(struct crypto_tfm *tfm)
 674 {
 675         return nx_crypto_ctx_init(crypto_tfm_ctx(tfm), NX_FC_AES,
 676                                   NX_MODE_AES_XCBC_MAC);
 677 }
 678
 679 /**
 680  * nx_crypto_ctx_exit - destroy a crypto api context
 681  *
 682  * @tfm: the crypto transform pointer for the context
 683  *
 684  * As crypto API contexts are destroyed, this exit hook is called to free the
 685  * memory associated with it.
 686  */
 687 void nx_crypto_ctx_exit(struct crypto_tfm *tfm)
 688 {
 689         struct nx_crypto_ctx *nx_ctx = crypto_tfm_ctx(tfm);
 690
 691         kzfree(nx_ctx->kmem);
 692         nx_ctx->csbcpb = NULL;
 693         nx_ctx->csbcpb_aead = NULL;
 694         nx_ctx->in_sg = NULL;
 695         nx_ctx->out_sg = NULL;
 696 }
 697
 698 static int nx_probe(struct vio_dev *viodev, const struct vio_device_id *id)
 699 {
 700         dev_dbg(&viodev->dev, "driver probed: %s resource id: 0x%x\n",
 701                 viodev->name, viodev->resource_id);
 702
 703         if (nx_driver.viodev) {
 704                 dev_err(&viodev->dev, "%s: Attempt to register more than one "
 705                         "instance of the hardware\n", __func__);
 706                 return -EINVAL;
 707         }
 708
 709         nx_driver.viodev = viodev;
 710
 711         nx_of_init(&viodev->dev, &nx_driver.of);
 712
 713         return nx_register_algs();
 714 }
 715
 716 static int nx_remove(struct vio_dev *viodev)
 717 {
 718         dev_dbg(&viodev->dev, "entering nx_remove for UA 0x%x\n",
 719                 viodev->unit_address);
 720
 721         if (nx_driver.of.status == NX_OKAY) {
 722                 NX_DEBUGFS_FINI(&nx_driver);
 723
 724                 crypto_unregister_alg(&nx_ccm_aes_alg);
 725                 crypto_unregister_alg(&nx_ccm4309_aes_alg);
 726                 crypto_unregister_alg(&nx_gcm_aes_alg);
 727                 crypto_unregister_alg(&nx_gcm4106_aes_alg);
 728                 crypto_unregister_alg(&nx_ctr_aes_alg);
 729                 crypto_unregister_alg(&nx_ctr3686_aes_alg);
 730                 crypto_unregister_alg(&nx_cbc_aes_alg);
 731                 crypto_unregister_alg(&nx_ecb_aes_alg);
 732                 crypto_unregister_shash(&nx_shash_sha256_alg);
 733                 crypto_unregister_shash(&nx_shash_sha512_alg);
 734                 crypto_unregister_shash(&nx_shash_aes_xcbc_alg);
 735         }
 736
 737         return 0;
 738 }
 739
 740
 741 /* module wide initialization/cleanup */
 742 static int __init nx_init(void)
 743 {
 744         return vio_register_driver(&nx_driver.viodriver);
 745 }
 746
 747 static void __exit nx_fini(void)
 748 {
 749         vio_unregister_driver(&nx_driver.viodriver);
 750 }
 751
 752 static struct vio_device_id nx_crypto_driver_ids[] = {
 753         { "ibm,sym-encryption-v1", "ibm,sym-encryption" },
 754         { "", "" }
 755 };
 756 MODULE_DEVICE_TABLE(vio, nx_crypto_driver_ids);
 757
 758 /* driver state structure */
 759 struct nx_crypto_driver nx_driver = {
 760         .viodriver = {
 761                 .id_table = nx_crypto_driver_ids,
 762                 .probe = nx_probe,
 763                 .remove = nx_remove,
 764                 .name  = NX_NAME,
 765         },
 766 };
 767
 768 module_init(nx_init);
 769 module_exit(nx_fini);
 770
 771 MODULE_AUTHOR("Kent Yoder <yoder1@us.ibm.com>");
 772 MODULE_DESCRIPTION(NX_STRING);
 773 MODULE_LICENSE("GPL");
 774 MODULE_VERSION(NX_VERSION);