crypto/vmac.c

   1 /*
   2  * Modified to interface to the Linux kernel
   3  * Copyright (c) 2009, Intel Corporation.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms and conditions of the GNU General Public License,
   7  * version 2, as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
  16  * Place - Suite 330, Boston, MA 02111-1307 USA.
  17  */
  18
  19 /* --------------------------------------------------------------------------
  20  * VMAC and VHASH Implementation by Ted Krovetz (tdk@acm.org) and Wei Dai.
  21  * This implementation is herby placed in the public domain.
  22  * The authors offers no warranty. Use at your own risk.
  23  * Please send bug reports to the authors.
  24  * Last modified: 17 APR 08, 1700 PDT
  25  * ----------------------------------------------------------------------- */
  26
  27 #include <linux/init.h>
  28 #include <linux/types.h>
  29 #include <linux/crypto.h>
  30 #include <linux/module.h>
  31 #include <linux/scatterlist.h>
  32 #include <asm/byteorder.h>
  33 #include <crypto/scatterwalk.h>
  34 #include <crypto/vmac.h>
  35 #include <crypto/internal/hash.h>
  36
  37 /*
  38  * Constants and masks
  39  */
  40 #define UINT64_C(x) x##ULL
  41 static const u64 p64   = UINT64_C(0xfffffffffffffeff);  /* 2^64 - 257 prime  */
  42 static const u64 m62   = UINT64_C(0x3fffffffffffffff);  /* 62-bit mask       */
  43 static const u64 m63   = UINT64_C(0x7fffffffffffffff);  /* 63-bit mask       */
  44 static const u64 m64   = UINT64_C(0xffffffffffffffff);  /* 64-bit mask       */
  45 static const u64 mpoly = UINT64_C(0x1fffffff1fffffff);  /* Poly key mask     */
  46
  47 #define pe64_to_cpup le64_to_cpup               /* Prefer little endian */
  48
  49 #ifdef __LITTLE_ENDIAN
  50 #define INDEX_HIGH 1
  51 #define INDEX_LOW 0
  52 #else
  53 #define INDEX_HIGH 0
  54 #define INDEX_LOW 1
  55 #endif
  56
  57 /*
  58  * The following routines are used in this implementation. They are
  59  * written via macros to simulate zero-overhead call-by-reference.
  60  *
  61  * MUL64: 64x64->128-bit multiplication
  62  * PMUL64: assumes top bits cleared on inputs
  63  * ADD128: 128x128->128-bit addition
  64  */
  65
  66 #define ADD128(rh, rl, ih, il)                                          \
  67         do {                                                            \
  68                 u64 _il = (il);                                         \
  69                 (rl) += (_il);                                          \
  70                 if ((rl) < (_il))                                       \
  71                         (rh)++;                                         \
  72                 (rh) += (ih);                                           \
  73         } while (0)
  74
  75 #define MUL32(i1, i2)   ((u64)(u32)(i1)*(u32)(i2))
  76
  77 #define PMUL64(rh, rl, i1, i2)  /* Assumes m doesn't overflow */        \
  78         do {                                                            \
  79                 u64 _i1 = (i1), _i2 = (i2);                             \
  80                 u64 m = MUL32(_i1, _i2>>32) + MUL32(_i1>>32, _i2);      \
  81                 rh = MUL32(_i1>>32, _i2>>32);                           \
  82                 rl = MUL32(_i1, _i2);                                   \
  83                 ADD128(rh, rl, (m >> 32), (m << 32));                   \
  84         } while (0)
  85
  86 #define MUL64(rh, rl, i1, i2)                                           \
  87         do {                                                            \
  88                 u64 _i1 = (i1), _i2 = (i2);                             \
  89                 u64 m1 = MUL32(_i1, _i2>>32);                           \
  90                 u64 m2 = MUL32(_i1>>32, _i2);                           \
  91                 rh = MUL32(_i1>>32, _i2>>32);                           \
  92                 rl = MUL32(_i1, _i2);                                   \
  93                 ADD128(rh, rl, (m1 >> 32), (m1 << 32));                 \
  94                 ADD128(rh, rl, (m2 >> 32), (m2 << 32));                 \
  95         } while (0)
  96
  97 /*
  98  * For highest performance the L1 NH and L2 polynomial hashes should be
  99  * carefully implemented to take advantage of one's target architecture.
 100  * Here these two hash functions are defined multiple time; once for
 101  * 64-bit architectures, once for 32-bit SSE2 architectures, and once
 102  * for the rest (32-bit) architectures.
 103  * For each, nh_16 *must* be defined (works on multiples of 16 bytes).
 104  * Optionally, nh_vmac_nhbytes can be defined (for multiples of
 105  * VMAC_NHBYTES), and nh_16_2 and nh_vmac_nhbytes_2 (versions that do two
 106  * NH computations at once).
 107  */
 108
 109 #ifdef CONFIG_64BIT
 110
 111 #define nh_16(mp, kp, nw, rh, rl)                                       \
 112         do {                                                            \
 113                 int i; u64 th, tl;                                      \
 114                 rh = rl = 0;                                            \
 115                 for (i = 0; i < nw; i += 2) {                           \
 116                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i],     \
 117                                 pe64_to_cpup((mp)+i+1)+(kp)[i+1]);      \
 118                         ADD128(rh, rl, th, tl);                         \
 119                 }                                                       \
 120         } while (0)
 121
 122 #define nh_16_2(mp, kp, nw, rh, rl, rh1, rl1)                           \
 123         do {                                                            \
 124                 int i; u64 th, tl;                                      \
 125                 rh1 = rl1 = rh = rl = 0;                                \
 126                 for (i = 0; i < nw; i += 2) {                           \
 127                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i],     \
 128                                 pe64_to_cpup((mp)+i+1)+(kp)[i+1]);      \
 129                         ADD128(rh, rl, th, tl);                         \
 130                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i+2],   \
 131                                 pe64_to_cpup((mp)+i+1)+(kp)[i+3]);      \
 132                         ADD128(rh1, rl1, th, tl);                       \
 133                 }                                                       \
 134         } while (0)
 135
 136 #if (VMAC_NHBYTES >= 64) /* These versions do 64-bytes of message at a time */
 137 #define nh_vmac_nhbytes(mp, kp, nw, rh, rl)                             \
 138         do {                                                            \
 139                 int i; u64 th, tl;                                      \
 140                 rh = rl = 0;                                            \
 141                 for (i = 0; i < nw; i += 8) {                           \
 142                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i],     \
 143                                 pe64_to_cpup((mp)+i+1)+(kp)[i+1]);      \
 144                         ADD128(rh, rl, th, tl);                         \
 145                         MUL64(th, tl, pe64_to_cpup((mp)+i+2)+(kp)[i+2], \
 146                                 pe64_to_cpup((mp)+i+3)+(kp)[i+3]);      \
 147                         ADD128(rh, rl, th, tl);                         \
 148                         MUL64(th, tl, pe64_to_cpup((mp)+i+4)+(kp)[i+4], \
 149                                 pe64_to_cpup((mp)+i+5)+(kp)[i+5]);      \
 150                         ADD128(rh, rl, th, tl);                         \
 151                         MUL64(th, tl, pe64_to_cpup((mp)+i+6)+(kp)[i+6], \
 152                                 pe64_to_cpup((mp)+i+7)+(kp)[i+7]);      \
 153                         ADD128(rh, rl, th, tl);                         \
 154                 }                                                       \
 155         } while (0)
 156
 157 #define nh_vmac_nhbytes_2(mp, kp, nw, rh, rl, rh1, rl1)                 \
 158         do {                                                            \
 159                 int i; u64 th, tl;                                      \
 160                 rh1 = rl1 = rh = rl = 0;                                \
 161                 for (i = 0; i < nw; i += 8) {                           \
 162                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i],     \
 163                                 pe64_to_cpup((mp)+i+1)+(kp)[i+1]);      \
 164                         ADD128(rh, rl, th, tl);                         \
 165                         MUL64(th, tl, pe64_to_cpup((mp)+i)+(kp)[i+2],   \
 166                                 pe64_to_cpup((mp)+i+1)+(kp)[i+3]);      \
 167                         ADD128(rh1, rl1, th, tl);                       \
 168                         MUL64(th, tl, pe64_to_cpup((mp)+i+2)+(kp)[i+2], \
 169                                 pe64_to_cpup((mp)+i+3)+(kp)[i+3]);      \
 170                         ADD128(rh, rl, th, tl);                         \
 171                         MUL64(th, tl, pe64_to_cpup((mp)+i+2)+(kp)[i+4], \
 172                                 pe64_to_cpup((mp)+i+3)+(kp)[i+5]);      \
 173                         ADD128(rh1, rl1, th, tl);                       \
 174                         MUL64(th, tl, pe64_to_cpup((mp)+i+4)+(kp)[i+4], \
 175                                 pe64_to_cpup((mp)+i+5)+(kp)[i+5]);      \
 176                         ADD128(rh, rl, th, tl);                         \
 177                         MUL64(th, tl, pe64_to_cpup((mp)+i+4)+(kp)[i+6], \
 178                                 pe64_to_cpup((mp)+i+5)+(kp)[i+7]);      \
 179                         ADD128(rh1, rl1, th, tl);                       \
 180                         MUL64(th, tl, pe64_to_cpup((mp)+i+6)+(kp)[i+6], \
 181                                 pe64_to_cpup((mp)+i+7)+(kp)[i+7]);      \
 182                         ADD128(rh, rl, th, tl);                         \
 183                         MUL64(th, tl, pe64_to_cpup((mp)+i+6)+(kp)[i+8], \
 184                                 pe64_to_cpup((mp)+i+7)+(kp)[i+9]);      \
 185                         ADD128(rh1, rl1, th, tl);                       \
 186                 }                                                       \
 187         } while (0)
 188 #endif
 189
 190 #define poly_step(ah, al, kh, kl, mh, ml)                               \
 191         do {                                                            \
 192                 u64 t1h, t1l, t2h, t2l, t3h, t3l, z = 0;                \
 193                 /* compute ab*cd, put bd into result registers */       \
 194                 PMUL64(t3h, t3l, al, kh);                               \
 195                 PMUL64(t2h, t2l, ah, kl);                               \
 196                 PMUL64(t1h, t1l, ah, 2*kh);                             \
 197                 PMUL64(ah, al, al, kl);                                 \
 198                 /* add 2 * ac to result */                              \
 199                 ADD128(ah, al, t1h, t1l);                               \
 200                 /* add together ad + bc */                              \
 201                 ADD128(t2h, t2l, t3h, t3l);                             \
 202                 /* now (ah,al), (t2l,2*t2h) need summing */             \
 203                 /* first add the high registers, carrying into t2h */   \
 204                 ADD128(t2h, ah, z, t2l);                                \
 205                 /* double t2h and add top bit of ah */                  \
 206                 t2h = 2 * t2h + (ah >> 63);                             \
 207                 ah &= m63;                                              \
 208                 /* now add the low registers */                         \
 209                 ADD128(ah, al, mh, ml);                                 \
 210                 ADD128(ah, al, z, t2h);                                 \
 211         } while (0)
 212
 213 #else /* ! CONFIG_64BIT */
 214
 215 #ifndef nh_16
 216 #define nh_16(mp, kp, nw, rh, rl)                                       \
 217         do {                                                            \
 218                 u64 t1, t2, m1, m2, t;                                  \
 219                 int i;                                                  \
 220                 rh = rl = t = 0;                                        \
 221                 for (i = 0; i < nw; i += 2)  {                          \
 222                         t1 = pe64_to_cpup(mp+i) + kp[i];                \
 223                         t2 = pe64_to_cpup(mp+i+1) + kp[i+1];            \
 224                         m2 = MUL32(t1 >> 32, t2);                       \
 225                         m1 = MUL32(t1, t2 >> 32);                       \
 226                         ADD128(rh, rl, MUL32(t1 >> 32, t2 >> 32),       \
 227                                 MUL32(t1, t2));                         \
 228                         rh += (u64)(u32)(m1 >> 32)                      \
 229                                 + (u32)(m2 >> 32);                      \
 230                         t += (u64)(u32)m1 + (u32)m2;                    \
 231                 }                                                       \
 232                 ADD128(rh, rl, (t >> 32), (t << 32));                   \
 233         } while (0)
 234 #endif
 235
 236 static void poly_step_func(u64 *ahi, u64 *alo,
 237                         const u64 *kh, const u64 *kl,
 238                         const u64 *mh, const u64 *ml)
 239 {
 240 #define a0 (*(((u32 *)alo)+INDEX_LOW))
 241 #define a1 (*(((u32 *)alo)+INDEX_HIGH))
 242 #define a2 (*(((u32 *)ahi)+INDEX_LOW))
 243 #define a3 (*(((u32 *)ahi)+INDEX_HIGH))
 244 #define k0 (*(((u32 *)kl)+INDEX_LOW))
 245 #define k1 (*(((u32 *)kl)+INDEX_HIGH))
 246 #define k2 (*(((u32 *)kh)+INDEX_LOW))
 247 #define k3 (*(((u32 *)kh)+INDEX_HIGH))
 248
 249         u64 p, q, t;
 250         u32 t2;
 251
 252         p = MUL32(a3, k3);
 253         p += p;
 254         p += *(u64 *)mh;
 255         p += MUL32(a0, k2);
 256         p += MUL32(a1, k1);
 257         p += MUL32(a2, k0);
 258         t = (u32)(p);
 259         p >>= 32;
 260         p += MUL32(a0, k3);
 261         p += MUL32(a1, k2);
 262         p += MUL32(a2, k1);
 263         p += MUL32(a3, k0);
 264         t |= ((u64)((u32)p & 0x7fffffff)) << 32;
 265         p >>= 31;
 266         p += (u64)(((u32 *)ml)[INDEX_LOW]);
 267         p += MUL32(a0, k0);
 268         q =  MUL32(a1, k3);
 269         q += MUL32(a2, k2);
 270         q += MUL32(a3, k1);
 271         q += q;
 272         p += q;
 273         t2 = (u32)(p);
 274         p >>= 32;
 275         p += (u64)(((u32 *)ml)[INDEX_HIGH]);
 276         p += MUL32(a0, k1);
 277         p += MUL32(a1, k0);
 278         q =  MUL32(a2, k3);
 279         q += MUL32(a3, k2);
 280         q += q;
 281         p += q;
 282         *(u64 *)(alo) = (p << 32) | t2;
 283         p >>= 32;
 284         *(u64 *)(ahi) = p + t;
 285
 286 #undef a0
 287 #undef a1
 288 #undef a2
 289 #undef a3
 290 #undef k0
 291 #undef k1
 292 #undef k2
 293 #undef k3
 294 }
 295
 296 #define poly_step(ah, al, kh, kl, mh, ml)                               \
 297         poly_step_func(&(ah), &(al), &(kh), &(kl), &(mh), &(ml))
 298
 299 #endif  /* end of specialized NH and poly definitions */
 300
 301 /* At least nh_16 is defined. Defined others as needed here */
 302 #ifndef nh_16_2
 303 #define nh_16_2(mp, kp, nw, rh, rl, rh2, rl2)                           \
 304         do {                                                            \
 305                 nh_16(mp, kp, nw, rh, rl);                              \
 306                 nh_16(mp, ((kp)+2), nw, rh2, rl2);                      \
 307         } while (0)
 308 #endif
 309 #ifndef nh_vmac_nhbytes
 310 #define nh_vmac_nhbytes(mp, kp, nw, rh, rl)                             \
 311         nh_16(mp, kp, nw, rh, rl)
 312 #endif
 313 #ifndef nh_vmac_nhbytes_2
 314 #define nh_vmac_nhbytes_2(mp, kp, nw, rh, rl, rh2, rl2)                 \
 315         do {                                                            \
 316                 nh_vmac_nhbytes(mp, kp, nw, rh, rl);                    \
 317                 nh_vmac_nhbytes(mp, ((kp)+2), nw, rh2, rl2);            \
 318         } while (0)
 319 #endif
 320
 321 static void vhash_abort(struct vmac_ctx *ctx)
 322 {
 323         ctx->polytmp[0] = ctx->polykey[0] ;
 324         ctx->polytmp[1] = ctx->polykey[1] ;
 325         ctx->first_block_processed = 0;
 326 }
 327
 328 static u64 l3hash(u64 p1, u64 p2, u64 k1, u64 k2, u64 len)
 329 {
 330         u64 rh, rl, t, z = 0;
 331
 332         /* fully reduce (p1,p2)+(len,0) mod p127 */
 333         t = p1 >> 63;
 334         p1 &= m63;
 335         ADD128(p1, p2, len, t);
 336         /* At this point, (p1,p2) is at most 2^127+(len<<64) */
 337         t = (p1 > m63) + ((p1 == m63) && (p2 == m64));
 338         ADD128(p1, p2, z, t);
 339         p1 &= m63;
 340
 341         /* compute (p1,p2)/(2^64-2^32) and (p1,p2)%(2^64-2^32) */
 342         t = p1 + (p2 >> 32);
 343         t += (t >> 32);
 344         t += (u32)t > 0xfffffffeu;
 345         p1 += (t >> 32);
 346         p2 += (p1 << 32);
 347
 348         /* compute (p1+k1)%p64 and (p2+k2)%p64 */
 349         p1 += k1;
 350         p1 += (0 - (p1 < k1)) & 257;
 351         p2 += k2;
 352         p2 += (0 - (p2 < k2)) & 257;
 353
 354         /* compute (p1+k1)*(p2+k2)%p64 */
 355         MUL64(rh, rl, p1, p2);
 356         t = rh >> 56;
 357         ADD128(t, rl, z, rh);
 358         rh <<= 8;
 359         ADD128(t, rl, z, rh);
 360         t += t << 8;
 361         rl += t;
 362         rl += (0 - (rl < t)) & 257;
 363         rl += (0 - (rl > p64-1)) & 257;
 364         return rl;
 365 }
 366
 367 static void vhash_update(const unsigned char *m,
 368                         unsigned int mbytes, /* Pos multiple of VMAC_NHBYTES */
 369                         struct vmac_ctx *ctx)
 370 {
 371         u64 rh, rl, *mptr;
 372         const u64 *kptr = (u64 *)ctx->nhkey;
 373         int i;
 374         u64 ch, cl;
 375         u64 pkh = ctx->polykey[0];
 376         u64 pkl = ctx->polykey[1];
 377
 378         if (!mbytes)
 379                 return;
 380
 381         BUG_ON(mbytes % VMAC_NHBYTES);
 382
 383         mptr = (u64 *)m;
 384         i = mbytes / VMAC_NHBYTES;  /* Must be non-zero */
 385
 386         ch = ctx->polytmp[0];
 387         cl = ctx->polytmp[1];
 388
 389         if (!ctx->first_block_processed) {
 390                 ctx->first_block_processed = 1;
 391                 nh_vmac_nhbytes(mptr, kptr, VMAC_NHBYTES/8, rh, rl);
 392                 rh &= m62;
 393                 ADD128(ch, cl, rh, rl);
 394                 mptr += (VMAC_NHBYTES/sizeof(u64));
 395                 i--;
 396         }
 397
 398         while (i--) {
 399                 nh_vmac_nhbytes(mptr, kptr, VMAC_NHBYTES/8, rh, rl);
 400                 rh &= m62;
 401                 poly_step(ch, cl, pkh, pkl, rh, rl);
 402                 mptr += (VMAC_NHBYTES/sizeof(u64));
 403         }
 404
 405         ctx->polytmp[0] = ch;
 406         ctx->polytmp[1] = cl;
 407 }
 408
 409 static u64 vhash(unsigned char m[], unsigned int mbytes,
 410                         u64 *tagl, struct vmac_ctx *ctx)
 411 {
 412         u64 rh, rl, *mptr;
 413         const u64 *kptr = (u64 *)ctx->nhkey;
 414         int i, remaining;
 415         u64 ch, cl;
 416         u64 pkh = ctx->polykey[0];
 417         u64 pkl = ctx->polykey[1];
 418
 419         mptr = (u64 *)m;
 420         i = mbytes / VMAC_NHBYTES;
 421         remaining = mbytes % VMAC_NHBYTES;
 422
 423         if (ctx->first_block_processed) {
 424                 ch = ctx->polytmp[0];
 425                 cl = ctx->polytmp[1];
 426         } else if (i) {
 427                 nh_vmac_nhbytes(mptr, kptr, VMAC_NHBYTES/8, ch, cl);
 428                 ch &= m62;
 429                 ADD128(ch, cl, pkh, pkl);
 430                 mptr += (VMAC_NHBYTES/sizeof(u64));
 431                 i--;
 432         } else if (remaining) {
 433                 nh_16(mptr, kptr, 2*((remaining+15)/16), ch, cl);
 434                 ch &= m62;
 435                 ADD128(ch, cl, pkh, pkl);
 436                 mptr += (VMAC_NHBYTES/sizeof(u64));
 437                 goto do_l3;
 438         } else {/* Empty String */
 439                 ch = pkh; cl = pkl;
 440                 goto do_l3;
 441         }
 442
 443         while (i--) {
 444                 nh_vmac_nhbytes(mptr, kptr, VMAC_NHBYTES/8, rh, rl);
 445                 rh &= m62;
 446                 poly_step(ch, cl, pkh, pkl, rh, rl);
 447                 mptr += (VMAC_NHBYTES/sizeof(u64));
 448         }
 449         if (remaining) {
 450                 nh_16(mptr, kptr, 2*((remaining+15)/16), rh, rl);
 451                 rh &= m62;
 452                 poly_step(ch, cl, pkh, pkl, rh, rl);
 453         }
 454
 455 do_l3:
 456         vhash_abort(ctx);
 457         remaining *= 8;
 458         return l3hash(ch, cl, ctx->l3key[0], ctx->l3key[1], remaining);
 459 }
 460
 461 static u64 vmac(unsigned char m[], unsigned int mbytes,
 462                         const unsigned char n[16], u64 *tagl,
 463                         struct vmac_ctx_t *ctx)
 464 {
 465         u64 *in_n, *out_p;
 466         u64 p, h;
 467         int i;
 468
 469         in_n = ctx->__vmac_ctx.cached_nonce;
 470         out_p = ctx->__vmac_ctx.cached_aes;
 471
 472         i = n[15] & 1;
 473         if ((*(u64 *)(n+8) != in_n[1]) || (*(u64 *)(n) != in_n[0])) {
 474                 in_n[0] = *(u64 *)(n);
 475                 in_n[1] = *(u64 *)(n+8);
 476                 ((unsigned char *)in_n)[15] &= 0xFE;
 477                 crypto_cipher_encrypt_one(ctx->child,
 478                         (unsigned char *)out_p, (unsigned char *)in_n);
 479
 480                 ((unsigned char *)in_n)[15] |= (unsigned char)(1-i);
 481         }
 482         p = be64_to_cpup(out_p + i);
 483         h = vhash(m, mbytes, (u64 *)0, &ctx->__vmac_ctx);
 484         return le64_to_cpu(p + h);
 485 }
 486
 487 static int vmac_set_key(unsigned char user_key[], struct vmac_ctx_t *ctx)
 488 {
 489         u64 in[2] = {0}, out[2];
 490         unsigned i;
 491         int err = 0;
 492
 493         err = crypto_cipher_setkey(ctx->child, user_key, VMAC_KEY_LEN);
 494         if (err)
 495                 return err;
 496
 497         /* Fill nh key */
 498         ((unsigned char *)in)[0] = 0x80;
 499         for (i = 0; i < sizeof(ctx->__vmac_ctx.nhkey)/8; i += 2) {
 500                 crypto_cipher_encrypt_one(ctx->child,
 501                         (unsigned char *)out, (unsigned char *)in);
 502                 ctx->__vmac_ctx.nhkey[i] = be64_to_cpup(out);
 503                 ctx->__vmac_ctx.nhkey[i+1] = be64_to_cpup(out+1);
 504                 ((unsigned char *)in)[15] += 1;
 505         }
 506
 507         /* Fill poly key */
 508         ((unsigned char *)in)[0] = 0xC0;
 509         in[1] = 0;
 510         for (i = 0; i < sizeof(ctx->__vmac_ctx.polykey)/8; i += 2) {
 511                 crypto_cipher_encrypt_one(ctx->child,
 512                         (unsigned char *)out, (unsigned char *)in);
 513                 ctx->__vmac_ctx.polytmp[i] =
 514                         ctx->__vmac_ctx.polykey[i] =
 515                                 be64_to_cpup(out) & mpoly;
 516                 ctx->__vmac_ctx.polytmp[i+1] =
 517                         ctx->__vmac_ctx.polykey[i+1] =
 518                                 be64_to_cpup(out+1) & mpoly;
 519                 ((unsigned char *)in)[15] += 1;
 520         }
 521
 522         /* Fill ip key */
 523         ((unsigned char *)in)[0] = 0xE0;
 524         in[1] = 0;
 525         for (i = 0; i < sizeof(ctx->__vmac_ctx.l3key)/8; i += 2) {
 526                 do {
 527                         crypto_cipher_encrypt_one(ctx->child,
 528                                 (unsigned char *)out, (unsigned char *)in);
 529                         ctx->__vmac_ctx.l3key[i] = be64_to_cpup(out);
 530                         ctx->__vmac_ctx.l3key[i+1] = be64_to_cpup(out+1);
 531                         ((unsigned char *)in)[15] += 1;
 532                 } while (ctx->__vmac_ctx.l3key[i] >= p64
 533                         || ctx->__vmac_ctx.l3key[i+1] >= p64);
 534         }
 535
 536         /* Invalidate nonce/aes cache and reset other elements */
 537         ctx->__vmac_ctx.cached_nonce[0] = (u64)-1; /* Ensure illegal nonce */
 538         ctx->__vmac_ctx.cached_nonce[1] = (u64)0;  /* Ensure illegal nonce */
 539         ctx->__vmac_ctx.first_block_processed = 0;
 540
 541         return err;
 542 }
 543
 544 static int vmac_setkey(struct crypto_shash *parent,
 545                 const u8 *key, unsigned int keylen)
 546 {
 547         struct vmac_ctx_t *ctx = crypto_shash_ctx(parent);
 548
 549         if (keylen != VMAC_KEY_LEN) {
 550                 crypto_shash_set_flags(parent, CRYPTO_TFM_RES_BAD_KEY_LEN);
 551                 return -EINVAL;
 552         }
 553
 554         return vmac_set_key((u8 *)key, ctx);
 555 }
 556
 557 static int vmac_init(struct shash_desc *pdesc)
 558 {
 559         return 0;
 560 }
 561
 562 static int vmac_update(struct shash_desc *pdesc, const u8 *p,
 563                 unsigned int len)
 564 {
 565         struct crypto_shash *parent = pdesc->tfm;
 566         struct vmac_ctx_t *ctx = crypto_shash_ctx(parent);
 567         int expand;
 568         int min;
 569
 570         expand = VMAC_NHBYTES - ctx->partial_size > 0 ?
 571                         VMAC_NHBYTES - ctx->partial_size : 0;
 572
 573         min = len < expand ? len : expand;
 574
 575         memcpy(ctx->partial + ctx->partial_size, p, min);
 576         ctx->partial_size += min;
 577
 578         if (len < expand)
 579                 return 0;
 580
 581         vhash_update(ctx->partial, VMAC_NHBYTES, &ctx->__vmac_ctx);
 582         ctx->partial_size = 0;
 583
 584         len -= expand;
 585         p += expand;
 586
 587         if (len % VMAC_NHBYTES) {
 588                 memcpy(ctx->partial, p + len - (len % VMAC_NHBYTES),
 589                         len % VMAC_NHBYTES);
 590                 ctx->partial_size = len % VMAC_NHBYTES;
 591         }
 592
 593         vhash_update(p, len - len % VMAC_NHBYTES, &ctx->__vmac_ctx);
 594
 595         return 0;
 596 }
 597
 598 static int vmac_final(struct shash_desc *pdesc, u8 *out)
 599 {
 600         struct crypto_shash *parent = pdesc->tfm;
 601         struct vmac_ctx_t *ctx = crypto_shash_ctx(parent);
 602         vmac_t mac;
 603         u8 nonce[16] = {};
 604
 605         /* vmac() ends up accessing outside the array bounds that
 606          * we specify.  In appears to access up to the next 2-word
 607          * boundary.  We'll just be uber cautious and zero the
 608          * unwritten bytes in the buffer.
 609          */
 610         if (ctx->partial_size) {
 611                 memset(ctx->partial + ctx->partial_size, 0,
 612                         VMAC_NHBYTES - ctx->partial_size);
 613         }
 614         mac = vmac(ctx->partial, ctx->partial_size, nonce, NULL, ctx);
 615         memcpy(out, &mac, sizeof(vmac_t));
 616         memzero_explicit(&mac, sizeof(vmac_t));
 617         memset(&ctx->__vmac_ctx, 0, sizeof(struct vmac_ctx));
 618         ctx->partial_size = 0;
 619         return 0;
 620 }
 621
 622 static int vmac_init_tfm(struct crypto_tfm *tfm)
 623 {
 624         struct crypto_cipher *cipher;
 625         struct crypto_instance *inst = (void *)tfm->__crt_alg;
 626         struct crypto_spawn *spawn = crypto_instance_ctx(inst);
 627         struct vmac_ctx_t *ctx = crypto_tfm_ctx(tfm);
 628
 629         cipher = crypto_spawn_cipher(spawn);
 630         if (IS_ERR(cipher))
 631                 return PTR_ERR(cipher);
 632
 633         ctx->child = cipher;
 634         return 0;
 635 }
 636
 637 static void vmac_exit_tfm(struct crypto_tfm *tfm)
 638 {
 639         struct vmac_ctx_t *ctx = crypto_tfm_ctx(tfm);
 640         crypto_free_cipher(ctx->child);
 641 }
 642
 643 static int vmac_create(struct crypto_template *tmpl, struct rtattr **tb)
 644 {
 645         struct shash_instance *inst;
 646         struct crypto_alg *alg;
 647         int err;
 648
 649         err = crypto_check_attr_type(tb, CRYPTO_ALG_TYPE_SHASH);
 650         if (err)
 651                 return err;
 652
 653         alg = crypto_get_attr_alg(tb, CRYPTO_ALG_TYPE_CIPHER,
 654                         CRYPTO_ALG_TYPE_MASK);
 655         if (IS_ERR(alg))
 656                 return PTR_ERR(alg);
 657
 658         inst = shash_alloc_instance("vmac", alg);
 659         err = PTR_ERR(inst);
 660         if (IS_ERR(inst))
 661                 goto out_put_alg;
 662
 663         err = crypto_init_spawn(shash_instance_ctx(inst), alg,
 664                         shash_crypto_instance(inst),
 665                         CRYPTO_ALG_TYPE_MASK);
 666         if (err)
 667                 goto out_free_inst;
 668
 669         inst->alg.base.cra_priority = alg->cra_priority;
 670         inst->alg.base.cra_blocksize = alg->cra_blocksize;
 671         inst->alg.base.cra_alignmask = alg->cra_alignmask;
 672
 673         inst->alg.digestsize = sizeof(vmac_t);
 674         inst->alg.base.cra_ctxsize = sizeof(struct vmac_ctx_t);
 675         inst->alg.base.cra_init = vmac_init_tfm;
 676         inst->alg.base.cra_exit = vmac_exit_tfm;
 677
 678         inst->alg.init = vmac_init;
 679         inst->alg.update = vmac_update;
 680         inst->alg.final = vmac_final;
 681         inst->alg.setkey = vmac_setkey;
 682
 683         err = shash_register_instance(tmpl, inst);
 684         if (err) {
 685 out_free_inst:
 686                 shash_free_instance(shash_crypto_instance(inst));
 687         }
 688
 689 out_put_alg:
 690         crypto_mod_put(alg);
 691         return err;
 692 }
 693
 694 static struct crypto_template vmac_tmpl = {
 695         .name = "vmac",
 696         .create = vmac_create,
 697         .free = shash_free_instance,
 698         .module = THIS_MODULE,
 699 };
 700
 701 static int __init vmac_module_init(void)
 702 {
 703         return crypto_register_template(&vmac_tmpl);
 704 }
 705
 706 static void __exit vmac_module_exit(void)
 707 {
 708         crypto_unregister_template(&vmac_tmpl);
 709 }
 710
 711 module_init(vmac_module_init);
 712 module_exit(vmac_module_exit);
 713
 714 MODULE_LICENSE("GPL");
 715 MODULE_DESCRIPTION("VMAC hash algorithm");