lustre/ptlrpc/gss/gss_krb5_mech.c

   1 /*
   2  * Modifications for Lustre
   3  *
   4  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
   5  *
   6  * Copyright (c) 2011, 2013, Intel Corporation.
   7  *
   8  * Author: Eric Mei <ericm@clusterfs.com>
   9  */
  10
  11 /*
  12  *  linux/net/sunrpc/gss_krb5_mech.c
  13  *  linux/net/sunrpc/gss_krb5_crypto.c
  14  *  linux/net/sunrpc/gss_krb5_seal.c
  15  *  linux/net/sunrpc/gss_krb5_seqnum.c
  16  *  linux/net/sunrpc/gss_krb5_unseal.c
  17  *
  18  *  Copyright (c) 2001 The Regents of the University of Michigan.
  19  *  All rights reserved.
  20  *
  21  *  Andy Adamson <andros@umich.edu>
  22  *  J. Bruce Fields <bfields@umich.edu>
  23  *
  24  *  Redistribution and use in source and binary forms, with or without
  25  *  modification, are permitted provided that the following conditions
  26  *  are met:
  27  *
  28  *  1. Redistributions of source code must retain the above copyright
  29  *     notice, this list of conditions and the following disclaimer.
  30  *  2. Redistributions in binary form must reproduce the above copyright
  31  *     notice, this list of conditions and the following disclaimer in the
  32  *     documentation and/or other materials provided with the distribution.
  33  *  3. Neither the name of the University nor the names of its
  34  *     contributors may be used to endorse or promote products derived
  35  *     from this software without specific prior written permission.
  36  *
  37  *  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
  38  *  WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  39  *  MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  40  *  DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  41  *  FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  42  *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  43  *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
  44  *  BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  45  *  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  46  *  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  47  *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  48  *
  49  */
  50
  51 #define DEBUG_SUBSYSTEM S_SEC
  52 #ifdef __KERNEL__
  53 #include <linux/init.h>
  54 #include <linux/module.h>
  55 #include <linux/slab.h>
  56 #include <linux/crypto.h>
  57 #include <linux/mutex.h>
  58 #else
  59 #include <liblustre.h>
  60 #endif
  61
  62 #include <obd.h>
  63 #include <obd_class.h>
  64 #include <obd_support.h>
  65 #include <lustre/lustre_idl.h>
  66 #include <lustre_net.h>
  67 #include <lustre_import.h>
  68 #include <lustre_sec.h>
  69
  70 #include "gss_err.h"
  71 #include "gss_internal.h"
  72 #include "gss_api.h"
  73 #include "gss_asn1.h"
  74 #include "gss_krb5.h"
  75
  76 static spinlock_t krb5_seq_lock;
  77
  78 struct krb5_enctype {
  79         char           *ke_dispname;
  80         char           *ke_enc_name;            /* linux tfm name */
  81         char           *ke_hash_name;           /* linux tfm name */
  82         int             ke_enc_mode;            /* linux tfm mode */
  83         int             ke_hash_size;           /* checksum size */
  84         int             ke_conf_size;           /* confounder size */
  85         unsigned int    ke_hash_hmac:1;         /* is hmac? */
  86 };
  87
  88 /*
  89  * NOTE: for aes128-cts and aes256-cts, MIT implementation use CTS encryption.
  90  * but currently we simply CBC with padding, because linux doesn't support CTS
  91  * yet. this need to be fixed in the future.
  92  */
  93 static struct krb5_enctype enctypes[] = {
  94         [ENCTYPE_DES_CBC_RAW] = {               /* des-cbc-md5 */
  95                 "des-cbc-md5",
  96                 "cbc(des)",
  97                 "md5",
  98                 0,
  99                 16,
 100                 8,
 101                 0,
 102         },
 103         [ENCTYPE_DES3_CBC_RAW] = {              /* des3-hmac-sha1 */
 104                 "des3-hmac-sha1",
 105                 "cbc(des3_ede)",
 106                 "hmac(sha1)",
 107                 0,
 108                 20,
 109                 8,
 110                 1,
 111         },
 112         [ENCTYPE_AES128_CTS_HMAC_SHA1_96] = {   /* aes128-cts */
 113                 "aes128-cts-hmac-sha1-96",
 114                 "cbc(aes)",
 115                 "hmac(sha1)",
 116                 0,
 117                 12,
 118                 16,
 119                 1,
 120         },
 121         [ENCTYPE_AES256_CTS_HMAC_SHA1_96] = {   /* aes256-cts */
 122                 "aes256-cts-hmac-sha1-96",
 123                 "cbc(aes)",
 124                 "hmac(sha1)",
 125                 0,
 126                 12,
 127                 16,
 128                 1,
 129         },
 130         [ENCTYPE_ARCFOUR_HMAC] = {              /* arcfour-hmac-md5 */
 131                 "arcfour-hmac-md5",
 132                 "ecb(arc4)",
 133                 "hmac(md5)",
 134                 0,
 135                 16,
 136                 8,
 137                 1,
 138         },
 139 };
 140
 141 #define MAX_ENCTYPES    sizeof(enctypes)/sizeof(struct krb5_enctype)
 142
 143 static const char * enctype2str(__u32 enctype)
 144 {
 145         if (enctype < MAX_ENCTYPES && enctypes[enctype].ke_dispname)
 146                 return enctypes[enctype].ke_dispname;
 147
 148         return "unknown";
 149 }
 150
 151 static
 152 int keyblock_init(struct krb5_keyblock *kb, char *alg_name, int alg_mode)
 153 {
 154         kb->kb_tfm = crypto_alloc_blkcipher(alg_name, alg_mode, 0);
 155         if (IS_ERR(kb->kb_tfm)) {
 156                 CERROR("failed to alloc tfm: %s, mode %d\n",
 157                        alg_name, alg_mode);
 158                 return -1;
 159         }
 160
 161         if (crypto_blkcipher_setkey(kb->kb_tfm, kb->kb_key.data, kb->kb_key.len)) {
 162                 CERROR("failed to set %s key, len %d\n",
 163                        alg_name, kb->kb_key.len);
 164                 return -1;
 165         }
 166
 167         return 0;
 168 }
 169
 170 static
 171 int krb5_init_keys(struct krb5_ctx *kctx)
 172 {
 173         struct krb5_enctype *ke;
 174
 175         if (kctx->kc_enctype >= MAX_ENCTYPES ||
 176             enctypes[kctx->kc_enctype].ke_hash_size == 0) {
 177                 CERROR("unsupported enctype %x\n", kctx->kc_enctype);
 178                 return -1;
 179         }
 180
 181         ke = &enctypes[kctx->kc_enctype];
 182
 183         /* tfm arc4 is stateful, user should alloc-use-free by his own */
 184         if (kctx->kc_enctype != ENCTYPE_ARCFOUR_HMAC &&
 185             keyblock_init(&kctx->kc_keye, ke->ke_enc_name, ke->ke_enc_mode))
 186                 return -1;
 187
 188         /* tfm hmac is stateful, user should alloc-use-free by his own */
 189         if (ke->ke_hash_hmac == 0 &&
 190             keyblock_init(&kctx->kc_keyi, ke->ke_enc_name, ke->ke_enc_mode))
 191                 return -1;
 192         if (ke->ke_hash_hmac == 0 &&
 193             keyblock_init(&kctx->kc_keyc, ke->ke_enc_name, ke->ke_enc_mode))
 194                 return -1;
 195
 196         return 0;
 197 }
 198
 199 static
 200 void keyblock_free(struct krb5_keyblock *kb)
 201 {
 202         rawobj_free(&kb->kb_key);
 203         if (kb->kb_tfm)
 204                 crypto_free_blkcipher(kb->kb_tfm);
 205 }
 206
 207 static
 208 int keyblock_dup(struct krb5_keyblock *new, struct krb5_keyblock *kb)
 209 {
 210         return rawobj_dup(&new->kb_key, &kb->kb_key);
 211 }
 212
 213 static
 214 int get_bytes(char **ptr, const char *end, void *res, int len)
 215 {
 216         char *p, *q;
 217         p = *ptr;
 218         q = p + len;
 219         if (q > end || q < p)
 220                 return -1;
 221         memcpy(res, p, len);
 222         *ptr = q;
 223         return 0;
 224 }
 225
 226 static
 227 int get_rawobj(char **ptr, const char *end, rawobj_t *res)
 228 {
 229         char   *p, *q;
 230         __u32   len;
 231
 232         p = *ptr;
 233         if (get_bytes(&p, end, &len, sizeof(len)))
 234                 return -1;
 235
 236         q = p + len;
 237         if (q > end || q < p)
 238                 return -1;
 239
 240         OBD_ALLOC_LARGE(res->data, len);
 241         if (!res->data)
 242                 return -1;
 243
 244         res->len = len;
 245         memcpy(res->data, p, len);
 246         *ptr = q;
 247         return 0;
 248 }
 249
 250 static
 251 int get_keyblock(char **ptr, const char *end,
 252                  struct krb5_keyblock *kb, __u32 keysize)
 253 {
 254         char *buf;
 255
 256         OBD_ALLOC_LARGE(buf, keysize);
 257         if (buf == NULL)
 258                 return -1;
 259
 260         if (get_bytes(ptr, end, buf, keysize)) {
 261                 OBD_FREE_LARGE(buf, keysize);
 262                 return -1;
 263         }
 264
 265         kb->kb_key.len = keysize;
 266         kb->kb_key.data = buf;
 267         return 0;
 268 }
 269
 270 static
 271 void delete_context_kerberos(struct krb5_ctx *kctx)
 272 {
 273         rawobj_free(&kctx->kc_mech_used);
 274
 275         keyblock_free(&kctx->kc_keye);
 276         keyblock_free(&kctx->kc_keyi);
 277         keyblock_free(&kctx->kc_keyc);
 278 }
 279
 280 static
 281 __u32 import_context_rfc1964(struct krb5_ctx *kctx, char *p, char *end)
 282 {
 283         unsigned int    tmp_uint, keysize;
 284
 285         /* seed_init flag */
 286         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)))
 287                 goto out_err;
 288         kctx->kc_seed_init = (tmp_uint != 0);
 289
 290         /* seed */
 291         if (get_bytes(&p, end, kctx->kc_seed, sizeof(kctx->kc_seed)))
 292                 goto out_err;
 293
 294         /* sign/seal algorithm, not really used now */
 295         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)) ||
 296             get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)))
 297                 goto out_err;
 298
 299         /* end time */
 300         if (get_bytes(&p, end, &kctx->kc_endtime, sizeof(kctx->kc_endtime)))
 301                 goto out_err;
 302
 303         /* seq send */
 304         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)))
 305                 goto out_err;
 306         kctx->kc_seq_send = tmp_uint;
 307
 308         /* mech oid */
 309         if (get_rawobj(&p, end, &kctx->kc_mech_used))
 310                 goto out_err;
 311
 312         /* old style enc/seq keys in format:
 313          *   - enctype (u32)
 314          *   - keysize (u32)
 315          *   - keydata
 316          * we decompose them to fit into the new context
 317          */
 318
 319         /* enc key */
 320         if (get_bytes(&p, end, &kctx->kc_enctype, sizeof(kctx->kc_enctype)))
 321                 goto out_err;
 322
 323         if (get_bytes(&p, end, &keysize, sizeof(keysize)))
 324                 goto out_err;
 325
 326         if (get_keyblock(&p, end, &kctx->kc_keye, keysize))
 327                 goto out_err;
 328
 329         /* seq key */
 330         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)) ||
 331             tmp_uint != kctx->kc_enctype)
 332                 goto out_err;
 333
 334         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)) ||
 335             tmp_uint != keysize)
 336                 goto out_err;
 337
 338         if (get_keyblock(&p, end, &kctx->kc_keyc, keysize))
 339                 goto out_err;
 340
 341         /* old style fallback */
 342         if (keyblock_dup(&kctx->kc_keyi, &kctx->kc_keyc))
 343                 goto out_err;
 344
 345         if (p != end)
 346                 goto out_err;
 347
 348         CDEBUG(D_SEC, "succesfully imported rfc1964 context\n");
 349         return 0;
 350 out_err:
 351         return GSS_S_FAILURE;
 352 }
 353
 354 /* Flags for version 2 context flags */
 355 #define KRB5_CTX_FLAG_INITIATOR         0x00000001
 356 #define KRB5_CTX_FLAG_CFX               0x00000002
 357 #define KRB5_CTX_FLAG_ACCEPTOR_SUBKEY   0x00000004
 358
 359 static
 360 __u32 import_context_rfc4121(struct krb5_ctx *kctx, char *p, char *end)
 361 {
 362         unsigned int    tmp_uint, keysize;
 363
 364         /* end time */
 365         if (get_bytes(&p, end, &kctx->kc_endtime, sizeof(kctx->kc_endtime)))
 366                 goto out_err;
 367
 368         /* flags */
 369         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)))
 370                 goto out_err;
 371
 372         if (tmp_uint & KRB5_CTX_FLAG_INITIATOR)
 373                 kctx->kc_initiate = 1;
 374         if (tmp_uint & KRB5_CTX_FLAG_CFX)
 375                 kctx->kc_cfx = 1;
 376         if (tmp_uint & KRB5_CTX_FLAG_ACCEPTOR_SUBKEY)
 377                 kctx->kc_have_acceptor_subkey = 1;
 378
 379         /* seq send */
 380         if (get_bytes(&p, end, &kctx->kc_seq_send, sizeof(kctx->kc_seq_send)))
 381                 goto out_err;
 382
 383         /* enctype */
 384         if (get_bytes(&p, end, &kctx->kc_enctype, sizeof(kctx->kc_enctype)))
 385                 goto out_err;
 386
 387         /* size of each key */
 388         if (get_bytes(&p, end, &keysize, sizeof(keysize)))
 389                 goto out_err;
 390
 391         /* number of keys - should always be 3 */
 392         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint)))
 393                 goto out_err;
 394
 395         if (tmp_uint != 3) {
 396                 CERROR("Invalid number of keys: %u\n", tmp_uint);
 397                 goto out_err;
 398         }
 399
 400         /* ke */
 401         if (get_keyblock(&p, end, &kctx->kc_keye, keysize))
 402                 goto out_err;
 403         /* ki */
 404         if (get_keyblock(&p, end, &kctx->kc_keyi, keysize))
 405                 goto out_err;
 406         /* ki */
 407         if (get_keyblock(&p, end, &kctx->kc_keyc, keysize))
 408                 goto out_err;
 409
 410         CDEBUG(D_SEC, "succesfully imported v2 context\n");
 411         return 0;
 412 out_err:
 413         return GSS_S_FAILURE;
 414 }
 415
 416 /*
 417  * The whole purpose here is trying to keep user level gss context parsing
 418  * from nfs-utils unchanged as possible as we can, they are not quite mature
 419  * yet, and many stuff still not clear, like heimdal etc.
 420  */
 421 static
 422 __u32 gss_import_sec_context_kerberos(rawobj_t *inbuf,
 423                                       struct gss_ctx *gctx)
 424 {
 425         struct krb5_ctx *kctx;
 426         char            *p = (char *) inbuf->data;
 427         char            *end = (char *) (inbuf->data + inbuf->len);
 428         unsigned int     tmp_uint, rc;
 429
 430         if (get_bytes(&p, end, &tmp_uint, sizeof(tmp_uint))) {
 431                 CERROR("Fail to read version\n");
 432                 return GSS_S_FAILURE;
 433         }
 434
 435         /* only support 0, 1 for the moment */
 436         if (tmp_uint > 2) {
 437                 CERROR("Invalid version %u\n", tmp_uint);
 438                 return GSS_S_FAILURE;
 439         }
 440
 441         OBD_ALLOC_PTR(kctx);
 442         if (!kctx)
 443                 return GSS_S_FAILURE;
 444
 445         if (tmp_uint == 0 || tmp_uint == 1) {
 446                 kctx->kc_initiate = tmp_uint;
 447                 rc = import_context_rfc1964(kctx, p, end);
 448         } else {
 449                 rc = import_context_rfc4121(kctx, p, end);
 450         }
 451
 452         if (rc == 0)
 453                 rc = krb5_init_keys(kctx);
 454
 455         if (rc) {
 456                 delete_context_kerberos(kctx);
 457                 OBD_FREE_PTR(kctx);
 458
 459                 return GSS_S_FAILURE;
 460         }
 461
 462         gctx->internal_ctx_id = kctx;
 463         return GSS_S_COMPLETE;
 464 }
 465
 466 static
 467 __u32 gss_copy_reverse_context_kerberos(struct gss_ctx *gctx,
 468                                         struct gss_ctx *gctx_new)
 469 {
 470         struct krb5_ctx *kctx = gctx->internal_ctx_id;
 471         struct krb5_ctx *knew;
 472
 473         OBD_ALLOC_PTR(knew);
 474         if (!knew)
 475                 return GSS_S_FAILURE;
 476
 477         knew->kc_initiate = kctx->kc_initiate ? 0 : 1;
 478         knew->kc_cfx = kctx->kc_cfx;
 479         knew->kc_seed_init = kctx->kc_seed_init;
 480         knew->kc_have_acceptor_subkey = kctx->kc_have_acceptor_subkey;
 481         knew->kc_endtime = kctx->kc_endtime;
 482
 483         memcpy(knew->kc_seed, kctx->kc_seed, sizeof(kctx->kc_seed));
 484         knew->kc_seq_send = kctx->kc_seq_recv;
 485         knew->kc_seq_recv = kctx->kc_seq_send;
 486         knew->kc_enctype = kctx->kc_enctype;
 487
 488         if (rawobj_dup(&knew->kc_mech_used, &kctx->kc_mech_used))
 489                 goto out_err;
 490
 491         if (keyblock_dup(&knew->kc_keye, &kctx->kc_keye))
 492                 goto out_err;
 493         if (keyblock_dup(&knew->kc_keyi, &kctx->kc_keyi))
 494                 goto out_err;
 495         if (keyblock_dup(&knew->kc_keyc, &kctx->kc_keyc))
 496                 goto out_err;
 497         if (krb5_init_keys(knew))
 498                 goto out_err;
 499
 500         gctx_new->internal_ctx_id = knew;
 501         CDEBUG(D_SEC, "succesfully copied reverse context\n");
 502         return GSS_S_COMPLETE;
 503
 504 out_err:
 505         delete_context_kerberos(knew);
 506         OBD_FREE_PTR(knew);
 507         return GSS_S_FAILURE;
 508 }
 509
 510 static
 511 __u32 gss_inquire_context_kerberos(struct gss_ctx *gctx,
 512                                    unsigned long  *endtime)
 513 {
 514         struct krb5_ctx *kctx = gctx->internal_ctx_id;
 515
 516         *endtime = (unsigned long) ((__u32) kctx->kc_endtime);
 517         return GSS_S_COMPLETE;
 518 }
 519
 520 static
 521 void gss_delete_sec_context_kerberos(void *internal_ctx)
 522 {
 523         struct krb5_ctx *kctx = internal_ctx;
 524
 525         delete_context_kerberos(kctx);
 526         OBD_FREE_PTR(kctx);
 527 }
 528
 529 static
 530 void buf_to_sg(struct scatterlist *sg, void *ptr, int len)
 531 {
 532         sg_init_table(sg, 1);
 533         sg_set_buf(sg, ptr, len);
 534 }
 535
 536 static
 537 __u32 krb5_encrypt(struct crypto_blkcipher *tfm,
 538                    int decrypt,
 539                    void * iv,
 540                    void * in,
 541                    void * out,
 542                    int length)
 543 {
 544         struct blkcipher_desc desc;
 545         struct scatterlist    sg;
 546         __u8 local_iv[16] = {0};
 547         __u32 ret = -EINVAL;
 548
 549         LASSERT(tfm);
 550         desc.tfm  = tfm;
 551         desc.info = local_iv;
 552         desc.flags= 0;
 553
 554         if (length % crypto_blkcipher_blocksize(tfm) != 0) {
 555                 CERROR("output length %d mismatch blocksize %d\n",
 556                        length, crypto_blkcipher_blocksize(tfm));
 557                 goto out;
 558         }
 559
 560         if (crypto_blkcipher_ivsize(tfm) > 16) {
 561                 CERROR("iv size too large %d\n", crypto_blkcipher_ivsize(tfm));
 562                 goto out;
 563         }
 564
 565         if (iv)
 566                 memcpy(local_iv, iv, crypto_blkcipher_ivsize(tfm));
 567
 568         memcpy(out, in, length);
 569         buf_to_sg(&sg, out, length);
 570
 571         if (decrypt)
 572                 ret = crypto_blkcipher_decrypt_iv(&desc, &sg, &sg, length);
 573         else
 574                 ret = crypto_blkcipher_encrypt_iv(&desc, &sg, &sg, length);
 575
 576 out:
 577         return(ret);
 578 }
 579
 580 static inline
 581 int krb5_digest_hmac(struct crypto_hash *tfm,
 582                      rawobj_t *key,
 583                      struct krb5_header *khdr,
 584                      int msgcnt, rawobj_t *msgs,
 585                      int iovcnt, lnet_kiov_t *iovs,
 586                      rawobj_t *cksum)
 587 {
 588         struct hash_desc   desc;
 589         struct scatterlist sg[1];
 590         int                i;
 591
 592         crypto_hash_setkey(tfm, key->data, key->len);
 593         desc.tfm  = tfm;
 594         desc.flags= 0;
 595
 596         crypto_hash_init(&desc);
 597
 598         for (i = 0; i < msgcnt; i++) {
 599                 if (msgs[i].len == 0)
 600                         continue;
 601                 buf_to_sg(sg, (char *) msgs[i].data, msgs[i].len);
 602                 crypto_hash_update(&desc, sg, msgs[i].len);
 603         }
 604
 605         for (i = 0; i < iovcnt; i++) {
 606                 if (iovs[i].kiov_len == 0)
 607                         continue;
 608
 609                 sg_set_page(&sg[0], iovs[i].kiov_page, iovs[i].kiov_len,
 610                             iovs[i].kiov_offset);
 611                 crypto_hash_update(&desc, sg, iovs[i].kiov_len);
 612         }
 613
 614         if (khdr) {
 615                 buf_to_sg(sg, (char *) khdr, sizeof(*khdr));
 616                 crypto_hash_update(&desc, sg, sizeof(*khdr));
 617         }
 618
 619         return crypto_hash_final(&desc, cksum->data);
 620 }
 621
 622 static inline
 623 int krb5_digest_norm(struct crypto_hash *tfm,
 624                      struct krb5_keyblock *kb,
 625                      struct krb5_header *khdr,
 626                      int msgcnt, rawobj_t *msgs,
 627                      int iovcnt, lnet_kiov_t *iovs,
 628                      rawobj_t *cksum)
 629 {
 630         struct hash_desc   desc;
 631         struct scatterlist sg[1];
 632         int                i;
 633
 634         LASSERT(kb->kb_tfm);
 635         desc.tfm  = tfm;
 636         desc.flags= 0;
 637
 638         crypto_hash_init(&desc);
 639
 640         for (i = 0; i < msgcnt; i++) {
 641                 if (msgs[i].len == 0)
 642                         continue;
 643                 buf_to_sg(sg, (char *) msgs[i].data, msgs[i].len);
 644                 crypto_hash_update(&desc, sg, msgs[i].len);
 645         }
 646
 647         for (i = 0; i < iovcnt; i++) {
 648                 if (iovs[i].kiov_len == 0)
 649                         continue;
 650
 651                 sg_set_page(&sg[0], iovs[i].kiov_page, iovs[i].kiov_len,
 652                             iovs[i].kiov_offset);
 653                 crypto_hash_update(&desc, sg, iovs[i].kiov_len);
 654         }
 655
 656         if (khdr) {
 657                 buf_to_sg(sg, (char *) khdr, sizeof(*khdr));
 658                 crypto_hash_update(&desc, sg, sizeof(*khdr));
 659         }
 660
 661         crypto_hash_final(&desc, cksum->data);
 662
 663         return krb5_encrypt(kb->kb_tfm, 0, NULL, cksum->data,
 664                             cksum->data, cksum->len);
 665 }
 666
 667 /*
 668  * compute (keyed/keyless) checksum against the plain text which appended
 669  * with krb5 wire token header.
 670  */
 671 static
 672 __s32 krb5_make_checksum(__u32 enctype,
 673                          struct krb5_keyblock *kb,
 674                          struct krb5_header *khdr,
 675                          int msgcnt, rawobj_t *msgs,
 676                          int iovcnt, lnet_kiov_t *iovs,
 677                          rawobj_t *cksum)
 678 {
 679         struct krb5_enctype   *ke = &enctypes[enctype];
 680         struct crypto_hash    *tfm;
 681         __u32                  code = GSS_S_FAILURE;
 682         int                    rc;
 683
 684         if (!(tfm = crypto_alloc_hash(ke->ke_hash_name, 0, 0))) {
 685                 CERROR("failed to alloc TFM: %s\n", ke->ke_hash_name);
 686                 return GSS_S_FAILURE;
 687         }
 688
 689         cksum->len = crypto_hash_digestsize(tfm);
 690         OBD_ALLOC_LARGE(cksum->data, cksum->len);
 691         if (!cksum->data) {
 692                 cksum->len = 0;
 693                 goto out_tfm;
 694         }
 695
 696         if (ke->ke_hash_hmac)
 697                 rc = krb5_digest_hmac(tfm, &kb->kb_key,
 698                                       khdr, msgcnt, msgs, iovcnt, iovs, cksum);
 699         else
 700                 rc = krb5_digest_norm(tfm, kb,
 701                                       khdr, msgcnt, msgs, iovcnt, iovs, cksum);
 702
 703         if (rc == 0)
 704                 code = GSS_S_COMPLETE;
 705 out_tfm:
 706         crypto_free_hash(tfm);
 707         return code;
 708 }
 709
 710 static void fill_krb5_header(struct krb5_ctx *kctx,
 711                              struct krb5_header *khdr,
 712                              int privacy)
 713 {
 714         unsigned char acceptor_flag;
 715
 716         acceptor_flag = kctx->kc_initiate ? 0 : FLAG_SENDER_IS_ACCEPTOR;
 717
 718         if (privacy) {
 719                 khdr->kh_tok_id = cpu_to_be16(KG_TOK_WRAP_MSG);
 720                 khdr->kh_flags = acceptor_flag | FLAG_WRAP_CONFIDENTIAL;
 721                 khdr->kh_ec = cpu_to_be16(0);
 722                 khdr->kh_rrc = cpu_to_be16(0);
 723         } else {
 724                 khdr->kh_tok_id = cpu_to_be16(KG_TOK_MIC_MSG);
 725                 khdr->kh_flags = acceptor_flag;
 726                 khdr->kh_ec = cpu_to_be16(0xffff);
 727                 khdr->kh_rrc = cpu_to_be16(0xffff);
 728         }
 729
 730         khdr->kh_filler = 0xff;
 731         spin_lock(&krb5_seq_lock);
 732         khdr->kh_seq = cpu_to_be64(kctx->kc_seq_send++);
 733         spin_unlock(&krb5_seq_lock);
 734 }
 735
 736 static __u32 verify_krb5_header(struct krb5_ctx *kctx,
 737                                 struct krb5_header *khdr,
 738                                 int privacy)
 739 {
 740         unsigned char acceptor_flag;
 741         __u16         tok_id, ec_rrc;
 742
 743         acceptor_flag = kctx->kc_initiate ? FLAG_SENDER_IS_ACCEPTOR : 0;
 744
 745         if (privacy) {
 746                 tok_id = KG_TOK_WRAP_MSG;
 747                 ec_rrc = 0x0;
 748         } else {
 749                 tok_id = KG_TOK_MIC_MSG;
 750                 ec_rrc = 0xffff;
 751         }
 752
 753         /* sanity checks */
 754         if (be16_to_cpu(khdr->kh_tok_id) != tok_id) {
 755                 CERROR("bad token id\n");
 756                 return GSS_S_DEFECTIVE_TOKEN;
 757         }
 758         if ((khdr->kh_flags & FLAG_SENDER_IS_ACCEPTOR) != acceptor_flag) {
 759                 CERROR("bad direction flag\n");
 760                 return GSS_S_BAD_SIG;
 761         }
 762         if (privacy && (khdr->kh_flags & FLAG_WRAP_CONFIDENTIAL) == 0) {
 763                 CERROR("missing confidential flag\n");
 764                 return GSS_S_BAD_SIG;
 765         }
 766         if (khdr->kh_filler != 0xff) {
 767                 CERROR("bad filler\n");
 768                 return GSS_S_DEFECTIVE_TOKEN;
 769         }
 770         if (be16_to_cpu(khdr->kh_ec) != ec_rrc ||
 771             be16_to_cpu(khdr->kh_rrc) != ec_rrc) {
 772                 CERROR("bad EC or RRC\n");
 773                 return GSS_S_DEFECTIVE_TOKEN;
 774         }
 775         return GSS_S_COMPLETE;
 776 }
 777
 778 static
 779 __u32 gss_get_mic_kerberos(struct gss_ctx *gctx,
 780                            int msgcnt,
 781                            rawobj_t *msgs,
 782                            int iovcnt,
 783                            lnet_kiov_t *iovs,
 784                            rawobj_t *token)
 785 {
 786         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
 787         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
 788         struct krb5_header  *khdr;
 789         rawobj_t             cksum = RAWOBJ_EMPTY;
 790
 791         /* fill krb5 header */
 792         LASSERT(token->len >= sizeof(*khdr));
 793         khdr = (struct krb5_header *) token->data;
 794         fill_krb5_header(kctx, khdr, 0);
 795
 796         /* checksum */
 797         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyc,
 798                                khdr, msgcnt, msgs, iovcnt, iovs, &cksum))
 799                 return GSS_S_FAILURE;
 800
 801         LASSERT(cksum.len >= ke->ke_hash_size);
 802         LASSERT(token->len >= sizeof(*khdr) + ke->ke_hash_size);
 803         memcpy(khdr + 1, cksum.data + cksum.len - ke->ke_hash_size,
 804                ke->ke_hash_size);
 805
 806         token->len = sizeof(*khdr) + ke->ke_hash_size;
 807         rawobj_free(&cksum);
 808         return GSS_S_COMPLETE;
 809 }
 810
 811 static
 812 __u32 gss_verify_mic_kerberos(struct gss_ctx *gctx,
 813                               int msgcnt,
 814                               rawobj_t *msgs,
 815                               int iovcnt,
 816                               lnet_kiov_t *iovs,
 817                               rawobj_t *token)
 818 {
 819         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
 820         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
 821         struct krb5_header  *khdr;
 822         rawobj_t             cksum = RAWOBJ_EMPTY;
 823         __u32                major;
 824
 825         if (token->len < sizeof(*khdr)) {
 826                 CERROR("short signature: %u\n", token->len);
 827                 return GSS_S_DEFECTIVE_TOKEN;
 828         }
 829
 830         khdr = (struct krb5_header *) token->data;
 831
 832         major = verify_krb5_header(kctx, khdr, 0);
 833         if (major != GSS_S_COMPLETE) {
 834                 CERROR("bad krb5 header\n");
 835                 return major;
 836         }
 837
 838         if (token->len < sizeof(*khdr) + ke->ke_hash_size) {
 839                 CERROR("short signature: %u, require %d\n",
 840                        token->len, (int) sizeof(*khdr) + ke->ke_hash_size);
 841                 return GSS_S_FAILURE;
 842         }
 843
 844         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyc,
 845                                khdr, msgcnt, msgs, iovcnt, iovs, &cksum)) {
 846                 CERROR("failed to make checksum\n");
 847                 return GSS_S_FAILURE;
 848         }
 849
 850         LASSERT(cksum.len >= ke->ke_hash_size);
 851         if (memcmp(khdr + 1, cksum.data + cksum.len - ke->ke_hash_size,
 852                    ke->ke_hash_size)) {
 853                 CERROR("checksum mismatch\n");
 854                 rawobj_free(&cksum);
 855                 return GSS_S_BAD_SIG;
 856         }
 857
 858         rawobj_free(&cksum);
 859         return GSS_S_COMPLETE;
 860 }
 861
 862 static
 863 int add_padding(rawobj_t *msg, int msg_buflen, int blocksize)
 864 {
 865         int padding;
 866
 867         padding = (blocksize - (msg->len & (blocksize - 1))) &
 868                   (blocksize - 1);
 869         if (!padding)
 870                 return 0;
 871
 872         if (msg->len + padding > msg_buflen) {
 873                 CERROR("bufsize %u too small: datalen %u, padding %u\n",
 874                         msg_buflen, msg->len, padding);
 875                 return -EINVAL;
 876         }
 877
 878         memset(msg->data + msg->len, padding, padding);
 879         msg->len += padding;
 880         return 0;
 881 }
 882
 883 static
 884 int krb5_encrypt_rawobjs(struct crypto_blkcipher *tfm,
 885                          int mode_ecb,
 886                          int inobj_cnt,
 887                          rawobj_t *inobjs,
 888                          rawobj_t *outobj,
 889                          int enc)
 890 {
 891         struct blkcipher_desc desc;
 892         struct scatterlist    src, dst;
 893         __u8                  local_iv[16] = {0}, *buf;
 894         __u32                 datalen = 0;
 895         int                   i, rc;
 896         ENTRY;
 897
 898         buf = outobj->data;
 899         desc.tfm  = tfm;
 900         desc.info = local_iv;
 901         desc.flags = 0;
 902
 903         for (i = 0; i < inobj_cnt; i++) {
 904                 LASSERT(buf + inobjs[i].len <= outobj->data + outobj->len);
 905
 906                 buf_to_sg(&src, inobjs[i].data, inobjs[i].len);
 907                 buf_to_sg(&dst, buf, outobj->len - datalen);
 908
 909                 if (mode_ecb) {
 910                         if (enc)
 911                                 rc = crypto_blkcipher_encrypt(
 912                                         &desc, &dst, &src, src.length);
 913                         else
 914                                 rc = crypto_blkcipher_decrypt(
 915                                         &desc, &dst, &src, src.length);
 916                 } else {
 917                         if (enc)
 918                                 rc = crypto_blkcipher_encrypt_iv(
 919                                         &desc, &dst, &src, src.length);
 920                         else
 921                                 rc = crypto_blkcipher_decrypt_iv(
 922                                         &desc, &dst, &src, src.length);
 923                 }
 924
 925                 if (rc) {
 926                         CERROR("encrypt error %d\n", rc);
 927                         RETURN(rc);
 928                 }
 929
 930                 datalen += inobjs[i].len;
 931                 buf += inobjs[i].len;
 932         }
 933
 934         outobj->len = datalen;
 935         RETURN(0);
 936 }
 937
 938 /*
 939  * if adj_nob != 0, we adjust desc->bd_nob to the actual cipher text size.
 940  */
 941 static
 942 int krb5_encrypt_bulk(struct crypto_blkcipher *tfm,
 943                       struct krb5_header *khdr,
 944                       char *confounder,
 945                       struct ptlrpc_bulk_desc *desc,
 946                       rawobj_t *cipher,
 947                       int adj_nob)
 948 {
 949         struct blkcipher_desc   ciph_desc;
 950         __u8                    local_iv[16] = {0};
 951         struct scatterlist      src, dst;
 952         int                     blocksize, i, rc, nob = 0;
 953
 954         LASSERT(desc->bd_iov_count);
 955         LASSERT(desc->bd_enc_iov);
 956
 957         blocksize = crypto_blkcipher_blocksize(tfm);
 958         LASSERT(blocksize > 1);
 959         LASSERT(cipher->len == blocksize + sizeof(*khdr));
 960
 961         ciph_desc.tfm  = tfm;
 962         ciph_desc.info = local_iv;
 963         ciph_desc.flags = 0;
 964
 965         /* encrypt confounder */
 966         buf_to_sg(&src, confounder, blocksize);
 967         buf_to_sg(&dst, cipher->data, blocksize);
 968
 969         rc = crypto_blkcipher_encrypt_iv(&ciph_desc, &dst, &src, blocksize);
 970         if (rc) {
 971                 CERROR("error to encrypt confounder: %d\n", rc);
 972                 return rc;
 973         }
 974
 975         /* encrypt clear pages */
 976         for (i = 0; i < desc->bd_iov_count; i++) {
 977                 sg_set_page(&src, desc->bd_iov[i].kiov_page,
 978                             (desc->bd_iov[i].kiov_len + blocksize - 1) &
 979                             (~(blocksize - 1)),
 980                             desc->bd_iov[i].kiov_offset);
 981                 if (adj_nob)
 982                         nob += src.length;
 983                 sg_set_page(&dst, desc->bd_enc_iov[i].kiov_page, src.length,
 984                             src.offset);
 985
 986                 desc->bd_enc_iov[i].kiov_offset = dst.offset;
 987                 desc->bd_enc_iov[i].kiov_len = dst.length;
 988
 989                 rc = crypto_blkcipher_encrypt_iv(&ciph_desc, &dst, &src,
 990                                                     src.length);
 991                 if (rc) {
 992                         CERROR("error to encrypt page: %d\n", rc);
 993                         return rc;
 994                 }
 995         }
 996
 997         /* encrypt krb5 header */
 998         buf_to_sg(&src, khdr, sizeof(*khdr));
 999         buf_to_sg(&dst, cipher->data + blocksize, sizeof(*khdr));
1000
1001         rc = crypto_blkcipher_encrypt_iv(&ciph_desc, &dst, &src,
1002                                          sizeof(*khdr));
1003         if (rc) {
1004                 CERROR("error to encrypt krb5 header: %d\n", rc);
1005                 return rc;
1006         }
1007
1008         if (adj_nob)
1009                 desc->bd_nob = nob;
1010
1011         return 0;
1012 }
1013
1014 /*
1015  * desc->bd_nob_transferred is the size of cipher text received.
1016  * desc->bd_nob is the target size of plain text supposed to be.
1017  *
1018  * if adj_nob != 0, we adjust each page's kiov_len to the actual
1019  * plain text size.
1020  * - for client read: we don't know data size for each page, so
1021  *   bd_iov[]->kiov_len is set to PAGE_SIZE, but actual data received might
1022  *   be smaller, so we need to adjust it according to bd_enc_iov[]->kiov_len.
1023  *   this means we DO NOT support the situation that server send an odd size
1024  *   data in a page which is not the last one.
1025  * - for server write: we knows exactly data size for each page being expected,
1026  *   thus kiov_len is accurate already, so we should not adjust it at all.
1027  *   and bd_enc_iov[]->kiov_len should be round_up(bd_iov[]->kiov_len) which
1028  *   should have been done by prep_bulk().
1029  */
1030 static
1031 int krb5_decrypt_bulk(struct crypto_blkcipher *tfm,
1032                       struct krb5_header *khdr,
1033                       struct ptlrpc_bulk_desc *desc,
1034                       rawobj_t *cipher,
1035                       rawobj_t *plain,
1036                       int adj_nob)
1037 {
1038         struct blkcipher_desc   ciph_desc;
1039         __u8                    local_iv[16] = {0};
1040         struct scatterlist      src, dst;
1041         int                     ct_nob = 0, pt_nob = 0;
1042         int                     blocksize, i, rc;
1043
1044         LASSERT(desc->bd_iov_count);
1045         LASSERT(desc->bd_enc_iov);
1046         LASSERT(desc->bd_nob_transferred);
1047
1048         blocksize = crypto_blkcipher_blocksize(tfm);
1049         LASSERT(blocksize > 1);
1050         LASSERT(cipher->len == blocksize + sizeof(*khdr));
1051
1052         ciph_desc.tfm  = tfm;
1053         ciph_desc.info = local_iv;
1054         ciph_desc.flags = 0;
1055
1056         if (desc->bd_nob_transferred % blocksize) {
1057                 CERROR("odd transferred nob: %d\n", desc->bd_nob_transferred);
1058                 return -EPROTO;
1059         }
1060
1061         /* decrypt head (confounder) */
1062         buf_to_sg(&src, cipher->data, blocksize);
1063         buf_to_sg(&dst, plain->data, blocksize);
1064
1065         rc = crypto_blkcipher_decrypt_iv(&ciph_desc, &dst, &src, blocksize);
1066         if (rc) {
1067                 CERROR("error to decrypt confounder: %d\n", rc);
1068                 return rc;
1069         }
1070
1071         for (i = 0; i < desc->bd_iov_count && ct_nob < desc->bd_nob_transferred;
1072              i++) {
1073                 if (desc->bd_enc_iov[i].kiov_offset % blocksize != 0 ||
1074                     desc->bd_enc_iov[i].kiov_len % blocksize != 0) {
1075                         CERROR("page %d: odd offset %u len %u, blocksize %d\n",
1076                                i, desc->bd_enc_iov[i].kiov_offset,
1077                                desc->bd_enc_iov[i].kiov_len, blocksize);
1078                         return -EFAULT;
1079                 }
1080
1081                 if (adj_nob) {
1082                         if (ct_nob + desc->bd_enc_iov[i].kiov_len >
1083                             desc->bd_nob_transferred)
1084                                 desc->bd_enc_iov[i].kiov_len =
1085                                         desc->bd_nob_transferred - ct_nob;
1086
1087                         desc->bd_iov[i].kiov_len = desc->bd_enc_iov[i].kiov_len;
1088                         if (pt_nob + desc->bd_enc_iov[i].kiov_len >desc->bd_nob)
1089                                 desc->bd_iov[i].kiov_len = desc->bd_nob -pt_nob;
1090                 } else {
1091                         /* this should be guaranteed by LNET */
1092                         LASSERT(ct_nob + desc->bd_enc_iov[i].kiov_len <=
1093                                 desc->bd_nob_transferred);
1094                         LASSERT(desc->bd_iov[i].kiov_len <=
1095                                 desc->bd_enc_iov[i].kiov_len);
1096                 }
1097
1098                 if (desc->bd_enc_iov[i].kiov_len == 0)
1099                         continue;
1100
1101                 sg_set_page(&src, desc->bd_enc_iov[i].kiov_page,
1102                             desc->bd_enc_iov[i].kiov_len,
1103                             desc->bd_enc_iov[i].kiov_offset);
1104                 dst = src;
1105                 if (desc->bd_iov[i].kiov_len % blocksize == 0)
1106                         sg_assign_page(&dst, desc->bd_iov[i].kiov_page);
1107
1108                 rc = crypto_blkcipher_decrypt_iv(&ciph_desc, &dst, &src,
1109                                                  src.length);
1110                 if (rc) {
1111                         CERROR("error to decrypt page: %d\n", rc);
1112                         return rc;
1113                 }
1114
1115                 if (desc->bd_iov[i].kiov_len % blocksize != 0) {
1116                         memcpy(page_address(desc->bd_iov[i].kiov_page) +
1117                                desc->bd_iov[i].kiov_offset,
1118                                page_address(desc->bd_enc_iov[i].kiov_page) +
1119                                desc->bd_iov[i].kiov_offset,
1120                                desc->bd_iov[i].kiov_len);
1121                 }
1122
1123                 ct_nob += desc->bd_enc_iov[i].kiov_len;
1124                 pt_nob += desc->bd_iov[i].kiov_len;
1125         }
1126
1127         if (unlikely(ct_nob != desc->bd_nob_transferred)) {
1128                 CERROR("%d cipher text transferred but only %d decrypted\n",
1129                        desc->bd_nob_transferred, ct_nob);
1130                 return -EFAULT;
1131         }
1132
1133         if (unlikely(!adj_nob && pt_nob != desc->bd_nob)) {
1134                 CERROR("%d plain text expected but only %d received\n",
1135                        desc->bd_nob, pt_nob);
1136                 return -EFAULT;
1137         }
1138
1139         /* if needed, clear up the rest unused iovs */
1140         if (adj_nob)
1141                 while (i < desc->bd_iov_count)
1142                         desc->bd_iov[i++].kiov_len = 0;
1143
1144         /* decrypt tail (krb5 header) */
1145         buf_to_sg(&src, cipher->data + blocksize, sizeof(*khdr));
1146         buf_to_sg(&dst, cipher->data + blocksize, sizeof(*khdr));
1147
1148         rc = crypto_blkcipher_decrypt_iv(&ciph_desc, &dst, &src,
1149                                          sizeof(*khdr));
1150         if (rc) {
1151                 CERROR("error to decrypt tail: %d\n", rc);
1152                 return rc;
1153         }
1154
1155         if (memcmp(cipher->data + blocksize, khdr, sizeof(*khdr))) {
1156                 CERROR("krb5 header doesn't match\n");
1157                 return -EACCES;
1158         }
1159
1160         return 0;
1161 }
1162
1163 static
1164 __u32 gss_wrap_kerberos(struct gss_ctx *gctx,
1165                         rawobj_t *gsshdr,
1166                         rawobj_t *msg,
1167                         int msg_buflen,
1168                         rawobj_t *token)
1169 {
1170         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
1171         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
1172         struct krb5_header  *khdr;
1173         int                  blocksize;
1174         rawobj_t             cksum = RAWOBJ_EMPTY;
1175         rawobj_t             data_desc[3], cipher;
1176         __u8                 conf[GSS_MAX_CIPHER_BLOCK];
1177         int                  rc = 0;
1178
1179         LASSERT(ke);
1180         LASSERT(ke->ke_conf_size <= GSS_MAX_CIPHER_BLOCK);
1181         LASSERT(kctx->kc_keye.kb_tfm == NULL ||
1182                 ke->ke_conf_size >=
1183                 crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm));
1184
1185         /*
1186          * final token format:
1187          * ---------------------------------------------------
1188          * | krb5 header | cipher text | checksum (16 bytes) |
1189          * ---------------------------------------------------
1190          */
1191
1192         /* fill krb5 header */
1193         LASSERT(token->len >= sizeof(*khdr));
1194         khdr = (struct krb5_header *) token->data;
1195         fill_krb5_header(kctx, khdr, 1);
1196
1197         /* generate confounder */
1198         cfs_get_random_bytes(conf, ke->ke_conf_size);
1199
1200         /* get encryption blocksize. note kc_keye might not associated with
1201          * a tfm, currently only for arcfour-hmac */
1202         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1203                 LASSERT(kctx->kc_keye.kb_tfm == NULL);
1204                 blocksize = 1;
1205         } else {
1206                 LASSERT(kctx->kc_keye.kb_tfm);
1207                 blocksize = crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm);
1208         }
1209         LASSERT(blocksize <= ke->ke_conf_size);
1210
1211         /* padding the message */
1212         if (add_padding(msg, msg_buflen, blocksize))
1213                 return GSS_S_FAILURE;
1214
1215         /*
1216          * clear text layout for checksum:
1217          * ------------------------------------------------------
1218          * | confounder | gss header | clear msgs | krb5 header |
1219          * ------------------------------------------------------
1220          */
1221         data_desc[0].data = conf;
1222         data_desc[0].len = ke->ke_conf_size;
1223         data_desc[1].data = gsshdr->data;
1224         data_desc[1].len = gsshdr->len;
1225         data_desc[2].data = msg->data;
1226         data_desc[2].len = msg->len;
1227
1228         /* compute checksum */
1229         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyi,
1230                                khdr, 3, data_desc, 0, NULL, &cksum))
1231                 return GSS_S_FAILURE;
1232         LASSERT(cksum.len >= ke->ke_hash_size);
1233
1234         /*
1235          * clear text layout for encryption:
1236          * -----------------------------------------
1237          * | confounder | clear msgs | krb5 header |
1238          * -----------------------------------------
1239          */
1240         data_desc[0].data = conf;
1241         data_desc[0].len = ke->ke_conf_size;
1242         data_desc[1].data = msg->data;
1243         data_desc[1].len = msg->len;
1244         data_desc[2].data = (__u8 *) khdr;
1245         data_desc[2].len = sizeof(*khdr);
1246
1247         /* cipher text will be directly inplace */
1248         cipher.data = (__u8 *) (khdr + 1);
1249         cipher.len = token->len - sizeof(*khdr);
1250         LASSERT(cipher.len >= ke->ke_conf_size + msg->len + sizeof(*khdr));
1251
1252         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1253                 rawobj_t                 arc4_keye;
1254                 struct crypto_blkcipher *arc4_tfm;
1255
1256                 if (krb5_make_checksum(ENCTYPE_ARCFOUR_HMAC, &kctx->kc_keyi,
1257                                        NULL, 1, &cksum, 0, NULL, &arc4_keye)) {
1258                         CERROR("failed to obtain arc4 enc key\n");
1259                         GOTO(arc4_out, rc = -EACCES);
1260                 }
1261
1262                 arc4_tfm = crypto_alloc_blkcipher("ecb(arc4)", 0, 0);
1263                 if (IS_ERR(arc4_tfm)) {
1264                         CERROR("failed to alloc tfm arc4 in ECB mode\n");
1265                         GOTO(arc4_out_key, rc = -EACCES);
1266                 }
1267
1268                 if (crypto_blkcipher_setkey(arc4_tfm, arc4_keye.data,
1269                                                arc4_keye.len)) {
1270                         CERROR("failed to set arc4 key, len %d\n",
1271                                arc4_keye.len);
1272                         GOTO(arc4_out_tfm, rc = -EACCES);
1273                 }
1274
1275                 rc = krb5_encrypt_rawobjs(arc4_tfm, 1,
1276                                           3, data_desc, &cipher, 1);
1277 arc4_out_tfm:
1278                 crypto_free_blkcipher(arc4_tfm);
1279 arc4_out_key:
1280                 rawobj_free(&arc4_keye);
1281 arc4_out:
1282                 do {} while(0); /* just to avoid compile warning */
1283         } else {
1284                 rc = krb5_encrypt_rawobjs(kctx->kc_keye.kb_tfm, 0,
1285                                           3, data_desc, &cipher, 1);
1286         }
1287
1288         if (rc != 0) {
1289                 rawobj_free(&cksum);
1290                 return GSS_S_FAILURE;
1291         }
1292
1293         /* fill in checksum */
1294         LASSERT(token->len >= sizeof(*khdr) + cipher.len + ke->ke_hash_size);
1295         memcpy((char *)(khdr + 1) + cipher.len,
1296                cksum.data + cksum.len - ke->ke_hash_size,
1297                ke->ke_hash_size);
1298         rawobj_free(&cksum);
1299
1300         /* final token length */
1301         token->len = sizeof(*khdr) + cipher.len + ke->ke_hash_size;
1302         return GSS_S_COMPLETE;
1303 }
1304
1305 static
1306 __u32 gss_prep_bulk_kerberos(struct gss_ctx *gctx,
1307                              struct ptlrpc_bulk_desc *desc)
1308 {
1309         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
1310         int                  blocksize, i;
1311
1312         LASSERT(desc->bd_iov_count);
1313         LASSERT(desc->bd_enc_iov);
1314         LASSERT(kctx->kc_keye.kb_tfm);
1315
1316         blocksize = crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm);
1317
1318         for (i = 0; i < desc->bd_iov_count; i++) {
1319                 LASSERT(desc->bd_enc_iov[i].kiov_page);
1320                 /*
1321                  * offset should always start at page boundary of either
1322                  * client or server side.
1323                  */
1324                 if (desc->bd_iov[i].kiov_offset & blocksize) {
1325                         CERROR("odd offset %d in page %d\n",
1326                                desc->bd_iov[i].kiov_offset, i);
1327                         return GSS_S_FAILURE;
1328                 }
1329
1330                 desc->bd_enc_iov[i].kiov_offset = desc->bd_iov[i].kiov_offset;
1331                 desc->bd_enc_iov[i].kiov_len = (desc->bd_iov[i].kiov_len +
1332                                                 blocksize - 1) & (~(blocksize - 1));
1333         }
1334
1335         return GSS_S_COMPLETE;
1336 }
1337
1338 static
1339 __u32 gss_wrap_bulk_kerberos(struct gss_ctx *gctx,
1340                              struct ptlrpc_bulk_desc *desc,
1341                              rawobj_t *token, int adj_nob)
1342 {
1343         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
1344         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
1345         struct krb5_header  *khdr;
1346         int                  blocksize;
1347         rawobj_t             cksum = RAWOBJ_EMPTY;
1348         rawobj_t             data_desc[1], cipher;
1349         __u8                 conf[GSS_MAX_CIPHER_BLOCK];
1350         int                  rc = 0;
1351
1352         LASSERT(ke);
1353         LASSERT(ke->ke_conf_size <= GSS_MAX_CIPHER_BLOCK);
1354
1355         /*
1356          * final token format:
1357          * --------------------------------------------------
1358          * | krb5 header | head/tail cipher text | checksum |
1359          * --------------------------------------------------
1360          */
1361
1362         /* fill krb5 header */
1363         LASSERT(token->len >= sizeof(*khdr));
1364         khdr = (struct krb5_header *) token->data;
1365         fill_krb5_header(kctx, khdr, 1);
1366
1367         /* generate confounder */
1368         cfs_get_random_bytes(conf, ke->ke_conf_size);
1369
1370         /* get encryption blocksize. note kc_keye might not associated with
1371          * a tfm, currently only for arcfour-hmac */
1372         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1373                 LASSERT(kctx->kc_keye.kb_tfm == NULL);
1374                 blocksize = 1;
1375         } else {
1376                 LASSERT(kctx->kc_keye.kb_tfm);
1377                 blocksize = crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm);
1378         }
1379
1380         /*
1381          * we assume the size of krb5_header (16 bytes) must be n * blocksize.
1382          * the bulk token size would be exactly (sizeof(krb5_header) +
1383          * blocksize + sizeof(krb5_header) + hashsize)
1384          */
1385         LASSERT(blocksize <= ke->ke_conf_size);
1386         LASSERT(sizeof(*khdr) >= blocksize && sizeof(*khdr) % blocksize == 0);
1387         LASSERT(token->len >= sizeof(*khdr) + blocksize + sizeof(*khdr) + 16);
1388
1389         /*
1390          * clear text layout for checksum:
1391          * ------------------------------------------
1392          * | confounder | clear pages | krb5 header |
1393          * ------------------------------------------
1394          */
1395         data_desc[0].data = conf;
1396         data_desc[0].len = ke->ke_conf_size;
1397
1398         /* compute checksum */
1399         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyi,
1400                                khdr, 1, data_desc,
1401                                desc->bd_iov_count, desc->bd_iov,
1402                                &cksum))
1403                 return GSS_S_FAILURE;
1404         LASSERT(cksum.len >= ke->ke_hash_size);
1405
1406         /*
1407          * clear text layout for encryption:
1408          * ------------------------------------------
1409          * | confounder | clear pages | krb5 header |
1410          * ------------------------------------------
1411          *        |              |             |
1412          *        ----------  (cipher pages)   |
1413          * result token:   |                   |
1414          * -------------------------------------------
1415          * | krb5 header | cipher text | cipher text |
1416          * -------------------------------------------
1417          */
1418         data_desc[0].data = conf;
1419         data_desc[0].len = ke->ke_conf_size;
1420
1421         cipher.data = (__u8 *) (khdr + 1);
1422         cipher.len = blocksize + sizeof(*khdr);
1423
1424         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1425                 LBUG();
1426                 rc = 0;
1427         } else {
1428                 rc = krb5_encrypt_bulk(kctx->kc_keye.kb_tfm, khdr,
1429                                        conf, desc, &cipher, adj_nob);
1430         }
1431
1432         if (rc != 0) {
1433                 rawobj_free(&cksum);
1434                 return GSS_S_FAILURE;
1435         }
1436
1437         /* fill in checksum */
1438         LASSERT(token->len >= sizeof(*khdr) + cipher.len + ke->ke_hash_size);
1439         memcpy((char *)(khdr + 1) + cipher.len,
1440                cksum.data + cksum.len - ke->ke_hash_size,
1441                ke->ke_hash_size);
1442         rawobj_free(&cksum);
1443
1444         /* final token length */
1445         token->len = sizeof(*khdr) + cipher.len + ke->ke_hash_size;
1446         return GSS_S_COMPLETE;
1447 }
1448
1449 static
1450 __u32 gss_unwrap_kerberos(struct gss_ctx  *gctx,
1451                           rawobj_t        *gsshdr,
1452                           rawobj_t        *token,
1453                           rawobj_t        *msg)
1454 {
1455         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
1456         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
1457         struct krb5_header  *khdr;
1458         unsigned char       *tmpbuf;
1459         int                  blocksize, bodysize;
1460         rawobj_t             cksum = RAWOBJ_EMPTY;
1461         rawobj_t             cipher_in, plain_out;
1462         rawobj_t             hash_objs[3];
1463         int                  rc = 0;
1464         __u32                major;
1465
1466         LASSERT(ke);
1467
1468         if (token->len < sizeof(*khdr)) {
1469                 CERROR("short signature: %u\n", token->len);
1470                 return GSS_S_DEFECTIVE_TOKEN;
1471         }
1472
1473         khdr = (struct krb5_header *) token->data;
1474
1475         major = verify_krb5_header(kctx, khdr, 1);
1476         if (major != GSS_S_COMPLETE) {
1477                 CERROR("bad krb5 header\n");
1478                 return major;
1479         }
1480
1481         /* block size */
1482         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1483                 LASSERT(kctx->kc_keye.kb_tfm == NULL);
1484                 blocksize = 1;
1485         } else {
1486                 LASSERT(kctx->kc_keye.kb_tfm);
1487                 blocksize = crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm);
1488         }
1489
1490         /* expected token layout:
1491          * ----------------------------------------
1492          * | krb5 header | cipher text | checksum |
1493          * ----------------------------------------
1494          */
1495         bodysize = token->len - sizeof(*khdr) - ke->ke_hash_size;
1496
1497         if (bodysize % blocksize) {
1498                 CERROR("odd bodysize %d\n", bodysize);
1499                 return GSS_S_DEFECTIVE_TOKEN;
1500         }
1501
1502         if (bodysize <= ke->ke_conf_size + sizeof(*khdr)) {
1503                 CERROR("incomplete token: bodysize %d\n", bodysize);
1504                 return GSS_S_DEFECTIVE_TOKEN;
1505         }
1506
1507         if (msg->len < bodysize - ke->ke_conf_size - sizeof(*khdr)) {
1508                 CERROR("buffer too small: %u, require %d\n",
1509                        msg->len, bodysize - ke->ke_conf_size);
1510                 return GSS_S_FAILURE;
1511         }
1512
1513         /* decrypting */
1514         OBD_ALLOC_LARGE(tmpbuf, bodysize);
1515         if (!tmpbuf)
1516                 return GSS_S_FAILURE;
1517
1518         major = GSS_S_FAILURE;
1519
1520         cipher_in.data = (__u8 *) (khdr + 1);
1521         cipher_in.len = bodysize;
1522         plain_out.data = tmpbuf;
1523         plain_out.len = bodysize;
1524
1525         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1526                 rawobj_t                 arc4_keye;
1527                 struct crypto_blkcipher *arc4_tfm;
1528
1529                 cksum.data = token->data + token->len - ke->ke_hash_size;
1530                 cksum.len = ke->ke_hash_size;
1531
1532                 if (krb5_make_checksum(ENCTYPE_ARCFOUR_HMAC, &kctx->kc_keyi,
1533                                        NULL, 1, &cksum, 0, NULL, &arc4_keye)) {
1534                         CERROR("failed to obtain arc4 enc key\n");
1535                         GOTO(arc4_out, rc = -EACCES);
1536                 }
1537
1538                 arc4_tfm = crypto_alloc_blkcipher("ecb(arc4)", 0, 0);
1539                 if (IS_ERR(arc4_tfm)) {
1540                         CERROR("failed to alloc tfm arc4 in ECB mode\n");
1541                         GOTO(arc4_out_key, rc = -EACCES);
1542                 }
1543
1544                 if (crypto_blkcipher_setkey(arc4_tfm,
1545                                          arc4_keye.data, arc4_keye.len)) {
1546                         CERROR("failed to set arc4 key, len %d\n",
1547                                arc4_keye.len);
1548                         GOTO(arc4_out_tfm, rc = -EACCES);
1549                 }
1550
1551                 rc = krb5_encrypt_rawobjs(arc4_tfm, 1,
1552                                           1, &cipher_in, &plain_out, 0);
1553 arc4_out_tfm:
1554                 crypto_free_blkcipher(arc4_tfm);
1555 arc4_out_key:
1556                 rawobj_free(&arc4_keye);
1557 arc4_out:
1558                 cksum = RAWOBJ_EMPTY;
1559         } else {
1560                 rc = krb5_encrypt_rawobjs(kctx->kc_keye.kb_tfm, 0,
1561                                           1, &cipher_in, &plain_out, 0);
1562         }
1563
1564         if (rc != 0) {
1565                 CERROR("error decrypt\n");
1566                 goto out_free;
1567         }
1568         LASSERT(plain_out.len == bodysize);
1569
1570         /* expected clear text layout:
1571          * -----------------------------------------
1572          * | confounder | clear msgs | krb5 header |
1573          * -----------------------------------------
1574          */
1575
1576         /* verify krb5 header in token is not modified */
1577         if (memcmp(khdr, plain_out.data + plain_out.len - sizeof(*khdr),
1578                    sizeof(*khdr))) {
1579                 CERROR("decrypted krb5 header mismatch\n");
1580                 goto out_free;
1581         }
1582
1583         /* verify checksum, compose clear text as layout:
1584          * ------------------------------------------------------
1585          * | confounder | gss header | clear msgs | krb5 header |
1586          * ------------------------------------------------------
1587          */
1588         hash_objs[0].len = ke->ke_conf_size;
1589         hash_objs[0].data = plain_out.data;
1590         hash_objs[1].len = gsshdr->len;
1591         hash_objs[1].data = gsshdr->data;
1592         hash_objs[2].len = plain_out.len - ke->ke_conf_size - sizeof(*khdr);
1593         hash_objs[2].data = plain_out.data + ke->ke_conf_size;
1594         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyi,
1595                                khdr, 3, hash_objs, 0, NULL, &cksum))
1596                 goto out_free;
1597
1598         LASSERT(cksum.len >= ke->ke_hash_size);
1599         if (memcmp((char *)(khdr + 1) + bodysize,
1600                    cksum.data + cksum.len - ke->ke_hash_size,
1601                    ke->ke_hash_size)) {
1602                 CERROR("checksum mismatch\n");
1603                 goto out_free;
1604         }
1605
1606         msg->len =  bodysize - ke->ke_conf_size - sizeof(*khdr);
1607         memcpy(msg->data, tmpbuf + ke->ke_conf_size, msg->len);
1608
1609         major = GSS_S_COMPLETE;
1610 out_free:
1611         OBD_FREE_LARGE(tmpbuf, bodysize);
1612         rawobj_free(&cksum);
1613         return major;
1614 }
1615
1616 static
1617 __u32 gss_unwrap_bulk_kerberos(struct gss_ctx *gctx,
1618                                struct ptlrpc_bulk_desc *desc,
1619                                rawobj_t *token, int adj_nob)
1620 {
1621         struct krb5_ctx     *kctx = gctx->internal_ctx_id;
1622         struct krb5_enctype *ke = &enctypes[kctx->kc_enctype];
1623         struct krb5_header  *khdr;
1624         int                  blocksize;
1625         rawobj_t             cksum = RAWOBJ_EMPTY;
1626         rawobj_t             cipher, plain;
1627         rawobj_t             data_desc[1];
1628         int                  rc;
1629         __u32                major;
1630
1631         LASSERT(ke);
1632
1633         if (token->len < sizeof(*khdr)) {
1634                 CERROR("short signature: %u\n", token->len);
1635                 return GSS_S_DEFECTIVE_TOKEN;
1636         }
1637
1638         khdr = (struct krb5_header *) token->data;
1639
1640         major = verify_krb5_header(kctx, khdr, 1);
1641         if (major != GSS_S_COMPLETE) {
1642                 CERROR("bad krb5 header\n");
1643                 return major;
1644         }
1645
1646         /* block size */
1647         if (kctx->kc_enctype == ENCTYPE_ARCFOUR_HMAC) {
1648                 LASSERT(kctx->kc_keye.kb_tfm == NULL);
1649                 blocksize = 1;
1650                 LBUG();
1651         } else {
1652                 LASSERT(kctx->kc_keye.kb_tfm);
1653                 blocksize = crypto_blkcipher_blocksize(kctx->kc_keye.kb_tfm);
1654         }
1655         LASSERT(sizeof(*khdr) >= blocksize && sizeof(*khdr) % blocksize == 0);
1656
1657         /*
1658          * token format is expected as:
1659          * -----------------------------------------------
1660          * | krb5 header | head/tail cipher text | cksum |
1661          * -----------------------------------------------
1662          */
1663         if (token->len < sizeof(*khdr) + blocksize + sizeof(*khdr) +
1664                          ke->ke_hash_size) {
1665                 CERROR("short token size: %u\n", token->len);
1666                 return GSS_S_DEFECTIVE_TOKEN;
1667         }
1668
1669         cipher.data = (__u8 *) (khdr + 1);
1670         cipher.len = blocksize + sizeof(*khdr);
1671         plain.data = cipher.data;
1672         plain.len = cipher.len;
1673
1674         rc = krb5_decrypt_bulk(kctx->kc_keye.kb_tfm, khdr,
1675                                desc, &cipher, &plain, adj_nob);
1676         if (rc)
1677                 return GSS_S_DEFECTIVE_TOKEN;
1678
1679         /*
1680          * verify checksum, compose clear text as layout:
1681          * ------------------------------------------
1682          * | confounder | clear pages | krb5 header |
1683          * ------------------------------------------
1684          */
1685         data_desc[0].data = plain.data;
1686         data_desc[0].len = blocksize;
1687
1688         if (krb5_make_checksum(kctx->kc_enctype, &kctx->kc_keyi,
1689                                khdr, 1, data_desc,
1690                                desc->bd_iov_count, desc->bd_iov,
1691                                &cksum))
1692                 return GSS_S_FAILURE;
1693         LASSERT(cksum.len >= ke->ke_hash_size);
1694
1695         if (memcmp(plain.data + blocksize + sizeof(*khdr),
1696                    cksum.data + cksum.len - ke->ke_hash_size,
1697                    ke->ke_hash_size)) {
1698                 CERROR("checksum mismatch\n");
1699                 rawobj_free(&cksum);
1700                 return GSS_S_BAD_SIG;
1701         }
1702
1703         rawobj_free(&cksum);
1704         return GSS_S_COMPLETE;
1705 }
1706
1707 int gss_display_kerberos(struct gss_ctx        *ctx,
1708                          char                  *buf,
1709                          int                    bufsize)
1710 {
1711         struct krb5_ctx    *kctx = ctx->internal_ctx_id;
1712         int                 written;
1713
1714         written = snprintf(buf, bufsize, "krb5 (%s)",
1715                            enctype2str(kctx->kc_enctype));
1716         return written;
1717 }
1718
1719 static struct gss_api_ops gss_kerberos_ops = {
1720         .gss_import_sec_context     = gss_import_sec_context_kerberos,
1721         .gss_copy_reverse_context   = gss_copy_reverse_context_kerberos,
1722         .gss_inquire_context        = gss_inquire_context_kerberos,
1723         .gss_get_mic                = gss_get_mic_kerberos,
1724         .gss_verify_mic             = gss_verify_mic_kerberos,
1725         .gss_wrap                   = gss_wrap_kerberos,
1726         .gss_unwrap                 = gss_unwrap_kerberos,
1727         .gss_prep_bulk              = gss_prep_bulk_kerberos,
1728         .gss_wrap_bulk              = gss_wrap_bulk_kerberos,
1729         .gss_unwrap_bulk            = gss_unwrap_bulk_kerberos,
1730         .gss_delete_sec_context     = gss_delete_sec_context_kerberos,
1731         .gss_display                = gss_display_kerberos,
1732 };
1733
1734 static struct subflavor_desc gss_kerberos_sfs[] = {
1735         {
1736                 .sf_subflavor   = SPTLRPC_SUBFLVR_KRB5N,
1737                 .sf_qop         = 0,
1738                 .sf_service     = SPTLRPC_SVC_NULL,
1739                 .sf_name        = "krb5n"
1740         },
1741         {
1742                 .sf_subflavor   = SPTLRPC_SUBFLVR_KRB5A,
1743                 .sf_qop         = 0,
1744                 .sf_service     = SPTLRPC_SVC_AUTH,
1745                 .sf_name        = "krb5a"
1746         },
1747         {
1748                 .sf_subflavor   = SPTLRPC_SUBFLVR_KRB5I,
1749                 .sf_qop         = 0,
1750                 .sf_service     = SPTLRPC_SVC_INTG,
1751                 .sf_name        = "krb5i"
1752         },
1753         {
1754                 .sf_subflavor   = SPTLRPC_SUBFLVR_KRB5P,
1755                 .sf_qop         = 0,
1756                 .sf_service     = SPTLRPC_SVC_PRIV,
1757                 .sf_name        = "krb5p"
1758         },
1759 };
1760
1761 /*
1762  * currently we leave module owner NULL
1763  */
1764 static struct gss_api_mech gss_kerberos_mech = {
1765         .gm_owner       = NULL, /*THIS_MODULE, */
1766         .gm_name        = "krb5",
1767         .gm_oid         = (rawobj_t)
1768                                 {9, "\052\206\110\206\367\022\001\002\002"},
1769         .gm_ops         = &gss_kerberos_ops,
1770         .gm_sf_num      = 4,
1771         .gm_sfs         = gss_kerberos_sfs,
1772 };
1773
1774 int __init init_kerberos_module(void)
1775 {
1776         int status;
1777
1778         spin_lock_init(&krb5_seq_lock);
1779
1780         status = lgss_mech_register(&gss_kerberos_mech);
1781         if (status)
1782                 CERROR("Failed to register kerberos gss mechanism!\n");
1783         return status;
1784 }
1785
1786 void cleanup_kerberos_module(void)
1787 {
1788         lgss_mech_unregister(&gss_kerberos_mech);
1789 }