lustre/ptlrpc/gss/gss_svc_upcall.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  *
   4  * Modifications for Lustre
   5  *
   6  * Copyright  2008 Sun Microsystems, Inc. All rights reserved
   7  *
   8  * Author: Eric Mei <ericm@clusterfs.com>
   9  */
  10
  11 /*
  12  * Neil Brown <neilb@cse.unsw.edu.au>
  13  * J. Bruce Fields <bfields@umich.edu>
  14  * Andy Adamson <andros@umich.edu>
  15  * Dug Song <dugsong@monkey.org>
  16  *
  17  * RPCSEC_GSS server authentication.
  18  * This implements RPCSEC_GSS as defined in rfc2203 (rpcsec_gss) and rfc2078
  19  * (gssapi)
  20  *
  21  * The RPCSEC_GSS involves three stages:
  22  *  1/ context creation
  23  *  2/ data exchange
  24  *  3/ context destruction
  25  *
  26  * Context creation is handled largely by upcalls to user-space.
  27  *  In particular, GSS_Accept_sec_context is handled by an upcall
  28  * Data exchange is handled entirely within the kernel
  29  *  In particular, GSS_GetMIC, GSS_VerifyMIC, GSS_Seal, GSS_Unseal are in-kernel.
  30  * Context destruction is handled in-kernel
  31  *  GSS_Delete_sec_context is in-kernel
  32  *
  33  * Context creation is initiated by a RPCSEC_GSS_INIT request arriving.
  34  * The context handle and gss_token are used as a key into the rpcsec_init cache.
  35  * The content of this cache includes some of the outputs of GSS_Accept_sec_context,
  36  * being major_status, minor_status, context_handle, reply_token.
  37  * These are sent back to the client.
  38  * Sequence window management is handled by the kernel.  The window size if currently
  39  * a compile time constant.
  40  *
  41  * When user-space is happy that a context is established, it places an entry
  42  * in the rpcsec_context cache. The key for this cache is the context_handle.
  43  * The content includes:
  44  *   uid/gidlist - for determining access rights
  45  *   mechanism type
  46  *   mechanism specific information, such as a key
  47  *
  48  */
  49
  50 #define DEBUG_SUBSYSTEM S_SEC
  51 #ifdef __KERNEL__
  52 #include <linux/types.h>
  53 #include <linux/init.h>
  54 #include <linux/module.h>
  55 #include <linux/slab.h>
  56 #include <linux/hash.h>
  57 #include <linux/mutex.h>
  58 #include <linux/sunrpc/cache.h>
  59 #else
  60 #include <liblustre.h>
  61 #endif
  62
  63 #include <obd.h>
  64 #include <obd_class.h>
  65 #include <obd_support.h>
  66 #include <lustre/lustre_idl.h>
  67 #include <lustre_net.h>
  68 #include <lustre_import.h>
  69 #include <lustre_sec.h>
  70
  71 #include "gss_err.h"
  72 #include "gss_internal.h"
  73 #include "gss_api.h"
  74
  75 #define GSS_SVC_UPCALL_TIMEOUT  (20)
  76
  77 static spinlock_t __ctx_index_lock = SPIN_LOCK_UNLOCKED;
  78 static __u64 __ctx_index;
  79
  80 __u64 gss_get_next_ctx_index(void)
  81 {
  82         __u64 idx;
  83
  84         spin_lock(&__ctx_index_lock);
  85         idx = __ctx_index++;
  86         spin_unlock(&__ctx_index_lock);
  87
  88         return idx;
  89 }
  90
  91 static inline unsigned long hash_mem(char *buf, int length, int bits)
  92 {
  93         unsigned long hash = 0;
  94         unsigned long l = 0;
  95         int len = 0;
  96         unsigned char c;
  97
  98         do {
  99                 if (len == length) {
 100                         c = (char) len;
 101                         len = -1;
 102                 } else
 103                         c = *buf++;
 104
 105                 l = (l << 8) | c;
 106                 len++;
 107
 108                 if ((len & (BITS_PER_LONG/8-1)) == 0)
 109                         hash = hash_long(hash^l, BITS_PER_LONG);
 110         } while (len);
 111
 112         return hash >> (BITS_PER_LONG - bits);
 113 }
 114
 115 /****************************************
 116  * rsi cache                            *
 117  ****************************************/
 118
 119 #define RSI_HASHBITS    (6)
 120 #define RSI_HASHMAX     (1 << RSI_HASHBITS)
 121 #define RSI_HASHMASK    (RSI_HASHMAX - 1)
 122
 123 struct rsi {
 124         struct cache_head       h;
 125         __u32                   lustre_svc;
 126         __u64                   nid;
 127         wait_queue_head_t       waitq;
 128         rawobj_t                in_handle, in_token;
 129         rawobj_t                out_handle, out_token;
 130         int                     major_status, minor_status;
 131 };
 132
 133 static struct cache_head *rsi_table[RSI_HASHMAX];
 134 static struct cache_detail rsi_cache;
 135 #ifdef HAVE_SUNRPC_CACHE_V2
 136 static struct rsi *rsi_update(struct rsi *new, struct rsi *old);
 137 static struct rsi *rsi_lookup(struct rsi *item);
 138 #else
 139 static struct rsi *rsi_lookup(struct rsi *item, int set);
 140 #endif
 141
 142 static inline int rsi_hash(struct rsi *item)
 143 {
 144         return hash_mem((char *)item->in_handle.data, item->in_handle.len,
 145                         RSI_HASHBITS) ^
 146                hash_mem((char *)item->in_token.data, item->in_token.len,
 147                         RSI_HASHBITS);
 148 }
 149
 150 static inline int __rsi_match(struct rsi *item, struct rsi *tmp)
 151 {
 152         return (rawobj_equal(&item->in_handle, &tmp->in_handle) &&
 153                 rawobj_equal(&item->in_token, &tmp->in_token));
 154 }
 155
 156 static void rsi_free(struct rsi *rsi)
 157 {
 158         rawobj_free(&rsi->in_handle);
 159         rawobj_free(&rsi->in_token);
 160         rawobj_free(&rsi->out_handle);
 161         rawobj_free(&rsi->out_token);
 162 }
 163
 164 static void rsi_request(struct cache_detail *cd,
 165                         struct cache_head *h,
 166                         char **bpp, int *blen)
 167 {
 168         struct rsi *rsi = container_of(h, struct rsi, h);
 169         __u64 index = 0;
 170
 171         /* if in_handle is null, provide kernel suggestion */
 172         if (rsi->in_handle.len == 0)
 173                 index = gss_get_next_ctx_index();
 174
 175         qword_addhex(bpp, blen, (char *) &rsi->lustre_svc,
 176                      sizeof(rsi->lustre_svc));
 177         qword_addhex(bpp, blen, (char *) &rsi->nid, sizeof(rsi->nid));
 178         qword_addhex(bpp, blen, (char *) &index, sizeof(index));
 179         qword_addhex(bpp, blen, rsi->in_handle.data, rsi->in_handle.len);
 180         qword_addhex(bpp, blen, rsi->in_token.data, rsi->in_token.len);
 181         (*bpp)[-1] = '\n';
 182 }
 183
 184 static inline void __rsi_init(struct rsi *new, struct rsi *item)
 185 {
 186         new->out_handle = RAWOBJ_EMPTY;
 187         new->out_token = RAWOBJ_EMPTY;
 188
 189         new->in_handle = item->in_handle;
 190         item->in_handle = RAWOBJ_EMPTY;
 191         new->in_token = item->in_token;
 192         item->in_token = RAWOBJ_EMPTY;
 193
 194         new->lustre_svc = item->lustre_svc;
 195         new->nid = item->nid;
 196         init_waitqueue_head(&new->waitq);
 197 }
 198
 199 static inline void __rsi_update(struct rsi *new, struct rsi *item)
 200 {
 201         LASSERT(new->out_handle.len == 0);
 202         LASSERT(new->out_token.len == 0);
 203
 204         new->out_handle = item->out_handle;
 205         item->out_handle = RAWOBJ_EMPTY;
 206         new->out_token = item->out_token;
 207         item->out_token = RAWOBJ_EMPTY;
 208
 209         new->major_status = item->major_status;
 210         new->minor_status = item->minor_status;
 211 }
 212
 213 #ifdef HAVE_SUNRPC_CACHE_V2
 214
 215 static void rsi_put(struct kref *ref)
 216 {
 217         struct rsi *rsi = container_of(ref, struct rsi, h.ref);
 218
 219         LASSERT(rsi->h.next == NULL);
 220         rsi_free(rsi);
 221         OBD_FREE_PTR(rsi);
 222 }
 223
 224 static int rsi_match(struct cache_head *a, struct cache_head *b)
 225 {
 226         struct rsi *item = container_of(a, struct rsi, h);
 227         struct rsi *tmp = container_of(b, struct rsi, h);
 228
 229         return __rsi_match(item, tmp);
 230 }
 231
 232 static void rsi_init(struct cache_head *cnew, struct cache_head *citem)
 233 {
 234         struct rsi *new = container_of(cnew, struct rsi, h);
 235         struct rsi *item = container_of(citem, struct rsi, h);
 236
 237         __rsi_init(new, item);
 238 }
 239
 240 static void update_rsi(struct cache_head *cnew, struct cache_head *citem)
 241 {
 242         struct rsi *new = container_of(cnew, struct rsi, h);
 243         struct rsi *item = container_of(citem, struct rsi, h);
 244
 245         __rsi_update(new, item);
 246 }
 247
 248 static struct cache_head *rsi_alloc(void)
 249 {
 250         struct rsi *rsi;
 251
 252         OBD_ALLOC_PTR(rsi);
 253         if (rsi)
 254                 return &rsi->h;
 255         else
 256                 return NULL;
 257 }
 258
 259 static int rsi_parse(struct cache_detail *cd, char *mesg, int mlen)
 260 {
 261         char           *buf = mesg;
 262         char           *ep;
 263         int             len;
 264         struct rsi      rsii, *rsip = NULL;
 265         time_t          expiry;
 266         int             status = -EINVAL;
 267         ENTRY;
 268
 269
 270         memset(&rsii, 0, sizeof(rsii));
 271
 272         /* handle */
 273         len = qword_get(&mesg, buf, mlen);
 274         if (len < 0)
 275                 goto out;
 276         if (rawobj_alloc(&rsii.in_handle, buf, len)) {
 277                 status = -ENOMEM;
 278                 goto out;
 279         }
 280
 281         /* token */
 282         len = qword_get(&mesg, buf, mlen);
 283         if (len < 0)
 284                 goto out;
 285         if (rawobj_alloc(&rsii.in_token, buf, len)) {
 286                 status = -ENOMEM;
 287                 goto out;
 288         }
 289
 290         rsip = rsi_lookup(&rsii);
 291         if (!rsip)
 292                 goto out;
 293
 294         rsii.h.flags = 0;
 295         /* expiry */
 296         expiry = get_expiry(&mesg);
 297         if (expiry == 0)
 298                 goto out;
 299
 300         len = qword_get(&mesg, buf, mlen);
 301         if (len <= 0)
 302                 goto out;
 303
 304         /* major */
 305         rsii.major_status = simple_strtol(buf, &ep, 10);
 306         if (*ep)
 307                 goto out;
 308
 309         /* minor */
 310         len = qword_get(&mesg, buf, mlen);
 311         if (len <= 0)
 312                 goto out;
 313         rsii.minor_status = simple_strtol(buf, &ep, 10);
 314         if (*ep)
 315                 goto out;
 316
 317         /* out_handle */
 318         len = qword_get(&mesg, buf, mlen);
 319         if (len < 0)
 320                 goto out;
 321         if (rawobj_alloc(&rsii.out_handle, buf, len)) {
 322                 status = -ENOMEM;
 323                 goto out;
 324         }
 325
 326         /* out_token */
 327         len = qword_get(&mesg, buf, mlen);
 328         if (len < 0)
 329                 goto out;
 330         if (rawobj_alloc(&rsii.out_token, buf, len)) {
 331                 status = -ENOMEM;
 332                 goto out;
 333         }
 334
 335         rsii.h.expiry_time = expiry;
 336         rsip = rsi_update(&rsii, rsip);
 337         status = 0;
 338 out:
 339         rsi_free(&rsii);
 340         if (rsip) {
 341                 wake_up_all(&rsip->waitq);
 342                 cache_put(&rsip->h, &rsi_cache);
 343         } else {
 344                 status = -ENOMEM;
 345         }
 346
 347         if (status)
 348                 CERROR("rsi parse error %d\n", status);
 349         RETURN(status);
 350 }
 351
 352 #else /* !HAVE_SUNRPC_CACHE_V2 */
 353
 354 static void rsi_put(struct cache_head *item, struct cache_detail *cd)
 355 {
 356         struct rsi *rsi = container_of(item, struct rsi, h);
 357
 358         LASSERT(atomic_read(&item->refcnt) > 0);
 359
 360         if (cache_put(item, cd)) {
 361                 LASSERT(item->next == NULL);
 362                 rsi_free(rsi);
 363                 kfree(rsi); /* created by cache mgmt using kmalloc */
 364         }
 365 }
 366
 367 static inline int rsi_match(struct rsi *item, struct rsi *tmp)
 368 {
 369         return __rsi_match(item, tmp);
 370 }
 371
 372 static inline void rsi_init(struct rsi *new, struct rsi *item)
 373 {
 374         __rsi_init(new, item);
 375 }
 376
 377 static inline void rsi_update(struct rsi *new, struct rsi *item)
 378 {
 379         __rsi_update(new, item);
 380 }
 381
 382 static int rsi_parse(struct cache_detail *cd, char *mesg, int mlen)
 383 {
 384         char           *buf = mesg;
 385         char           *ep;
 386         int             len;
 387         struct rsi      rsii, *rsip = NULL;
 388         time_t          expiry;
 389         int             status = -EINVAL;
 390         ENTRY;
 391
 392
 393         memset(&rsii, 0, sizeof(rsii));
 394
 395         /* handle */
 396         len = qword_get(&mesg, buf, mlen);
 397         if (len < 0)
 398                 goto out;
 399         if (rawobj_alloc(&rsii.in_handle, buf, len)) {
 400                 status = -ENOMEM;
 401                 goto out;
 402         }
 403
 404         /* token */
 405         len = qword_get(&mesg, buf, mlen);
 406         if (len < 0)
 407                 goto out;
 408         if (rawobj_alloc(&rsii.in_token, buf, len)) {
 409                 status = -ENOMEM;
 410                 goto out;
 411         }
 412
 413         /* expiry */
 414         expiry = get_expiry(&mesg);
 415         if (expiry == 0)
 416                 goto out;
 417
 418         len = qword_get(&mesg, buf, mlen);
 419         if (len <= 0)
 420                 goto out;
 421
 422         /* major */
 423         rsii.major_status = simple_strtol(buf, &ep, 10);
 424         if (*ep)
 425                 goto out;
 426
 427         /* minor */
 428         len = qword_get(&mesg, buf, mlen);
 429         if (len <= 0)
 430                 goto out;
 431         rsii.minor_status = simple_strtol(buf, &ep, 10);
 432         if (*ep)
 433                 goto out;
 434
 435         /* out_handle */
 436         len = qword_get(&mesg, buf, mlen);
 437         if (len < 0)
 438                 goto out;
 439         if (rawobj_alloc(&rsii.out_handle, buf, len)) {
 440                 status = -ENOMEM;
 441                 goto out;
 442         }
 443
 444         /* out_token */
 445         len = qword_get(&mesg, buf, mlen);
 446         if (len < 0)
 447                 goto out;
 448         if (rawobj_alloc(&rsii.out_token, buf, len)) {
 449                 status = -ENOMEM;
 450                 goto out;
 451         }
 452
 453         rsii.h.expiry_time = expiry;
 454         rsip = rsi_lookup(&rsii, 1);
 455         status = 0;
 456 out:
 457         rsi_free(&rsii);
 458         if (rsip) {
 459                 wake_up_all(&rsip->waitq);
 460                 rsi_put(&rsip->h, &rsi_cache);
 461         }
 462
 463         if (status)
 464                 CERROR("rsi parse error %d\n", status);
 465         RETURN(status);
 466 }
 467
 468 #endif /* HAVE_SUNRPC_CACHE_V2 */
 469
 470 static struct cache_detail rsi_cache = {
 471         .hash_size      = RSI_HASHMAX,
 472         .hash_table     = rsi_table,
 473         .name           = "auth.sptlrpc.init",
 474         .cache_put      = rsi_put,
 475         .cache_request  = rsi_request,
 476         .cache_parse    = rsi_parse,
 477 #ifdef HAVE_SUNRPC_CACHE_V2
 478         .match          = rsi_match,
 479         .init           = rsi_init,
 480         .update         = update_rsi,
 481         .alloc          = rsi_alloc,
 482 #endif
 483 };
 484
 485 #ifdef HAVE_SUNRPC_CACHE_V2
 486
 487 static struct rsi *rsi_lookup(struct rsi *item)
 488 {
 489         struct cache_head *ch;
 490         int hash = rsi_hash(item);
 491
 492         ch = sunrpc_cache_lookup(&rsi_cache, &item->h, hash);
 493         if (ch)
 494                 return container_of(ch, struct rsi, h);
 495         else
 496                 return NULL;
 497 }
 498
 499 static struct rsi *rsi_update(struct rsi *new, struct rsi *old)
 500 {
 501         struct cache_head *ch;
 502         int hash = rsi_hash(new);
 503
 504         ch = sunrpc_cache_update(&rsi_cache, &new->h, &old->h, hash);
 505         if (ch)
 506                 return container_of(ch, struct rsi, h);
 507         else
 508                 return NULL;
 509 }
 510
 511 #else
 512
 513 static DefineSimpleCacheLookup(rsi, 0)
 514
 515 #endif
 516
 517 /****************************************
 518  * rsc cache                            *
 519  ****************************************/
 520
 521 #define RSC_HASHBITS    (10)
 522 #define RSC_HASHMAX     (1 << RSC_HASHBITS)
 523 #define RSC_HASHMASK    (RSC_HASHMAX - 1)
 524
 525 struct rsc {
 526         struct cache_head       h;
 527         struct obd_device      *target;
 528         rawobj_t                handle;
 529         struct gss_svc_ctx      ctx;
 530 };
 531
 532 static struct cache_head *rsc_table[RSC_HASHMAX];
 533 static struct cache_detail rsc_cache;
 534 #ifdef HAVE_SUNRPC_CACHE_V2
 535 static struct rsc *rsc_update(struct rsc *new, struct rsc *old);
 536 static struct rsc *rsc_lookup(struct rsc *item);
 537 #else
 538 static struct rsc *rsc_lookup(struct rsc *item, int set);
 539 #endif
 540
 541 static void rsc_free(struct rsc *rsci)
 542 {
 543         rawobj_free(&rsci->handle);
 544         rawobj_free(&rsci->ctx.gsc_rvs_hdl);
 545         lgss_delete_sec_context(&rsci->ctx.gsc_mechctx);
 546 }
 547
 548 static inline int rsc_hash(struct rsc *rsci)
 549 {
 550         return hash_mem((char *)rsci->handle.data,
 551                         rsci->handle.len, RSC_HASHBITS);
 552 }
 553
 554 static inline int __rsc_match(struct rsc *new, struct rsc *tmp)
 555 {
 556         return rawobj_equal(&new->handle, &tmp->handle);
 557 }
 558
 559 static inline void __rsc_init(struct rsc *new, struct rsc *tmp)
 560 {
 561         new->handle = tmp->handle;
 562         tmp->handle = RAWOBJ_EMPTY;
 563
 564         new->target = NULL;
 565         memset(&new->ctx, 0, sizeof(new->ctx));
 566         new->ctx.gsc_rvs_hdl = RAWOBJ_EMPTY;
 567 }
 568
 569 static inline void __rsc_update(struct rsc *new, struct rsc *tmp)
 570 {
 571         new->ctx = tmp->ctx;
 572         tmp->ctx.gsc_rvs_hdl = RAWOBJ_EMPTY;
 573         tmp->ctx.gsc_mechctx = NULL;
 574
 575         memset(&new->ctx.gsc_seqdata, 0, sizeof(new->ctx.gsc_seqdata));
 576         spin_lock_init(&new->ctx.gsc_seqdata.ssd_lock);
 577 }
 578
 579 #ifdef HAVE_SUNRPC_CACHE_V2
 580
 581 static void rsc_put(struct kref *ref)
 582 {
 583         struct rsc *rsci = container_of(ref, struct rsc, h.ref);
 584
 585         LASSERT(rsci->h.next == NULL);
 586         rsc_free(rsci);
 587         OBD_FREE_PTR(rsci);
 588 }
 589
 590 static int rsc_match(struct cache_head *a, struct cache_head *b)
 591 {
 592         struct rsc *new = container_of(a, struct rsc, h);
 593         struct rsc *tmp = container_of(b, struct rsc, h);
 594
 595         return __rsc_match(new, tmp);
 596 }
 597
 598 static void rsc_init(struct cache_head *cnew, struct cache_head *ctmp)
 599 {
 600         struct rsc *new = container_of(cnew, struct rsc, h);
 601         struct rsc *tmp = container_of(ctmp, struct rsc, h);
 602
 603         __rsc_init(new, tmp);
 604 }
 605
 606 static void update_rsc(struct cache_head *cnew, struct cache_head *ctmp)
 607 {
 608         struct rsc *new = container_of(cnew, struct rsc, h);
 609         struct rsc *tmp = container_of(ctmp, struct rsc, h);
 610
 611         __rsc_update(new, tmp);
 612 }
 613
 614 static struct cache_head * rsc_alloc(void)
 615 {
 616         struct rsc *rsc;
 617
 618         OBD_ALLOC_PTR(rsc);
 619         if (rsc)
 620                 return &rsc->h;
 621         else
 622                 return NULL;
 623 }
 624
 625 static int rsc_parse(struct cache_detail *cd, char *mesg, int mlen)
 626 {
 627         char                *buf = mesg;
 628         int                  len, rv, tmp_int;
 629         struct rsc           rsci, *rscp = NULL;
 630         time_t               expiry;
 631         int                  status = -EINVAL;
 632         struct gss_api_mech *gm = NULL;
 633
 634         memset(&rsci, 0, sizeof(rsci));
 635
 636         /* context handle */
 637         len = qword_get(&mesg, buf, mlen);
 638         if (len < 0) goto out;
 639         status = -ENOMEM;
 640         if (rawobj_alloc(&rsci.handle, buf, len))
 641                 goto out;
 642
 643         rsci.h.flags = 0;
 644         /* expiry */
 645         expiry = get_expiry(&mesg);
 646         status = -EINVAL;
 647         if (expiry == 0)
 648                 goto out;
 649
 650         /* remote flag */
 651         rv = get_int(&mesg, &tmp_int);
 652         if (rv) {
 653                 CERROR("fail to get remote flag\n");
 654                 goto out;
 655         }
 656         rsci.ctx.gsc_remote = (tmp_int != 0);
 657
 658         /* root user flag */
 659         rv = get_int(&mesg, &tmp_int);
 660         if (rv) {
 661                 CERROR("fail to get oss user flag\n");
 662                 goto out;
 663         }
 664         rsci.ctx.gsc_usr_root = (tmp_int != 0);
 665
 666         /* mds user flag */
 667         rv = get_int(&mesg, &tmp_int);
 668         if (rv) {
 669                 CERROR("fail to get mds user flag\n");
 670                 goto out;
 671         }
 672         rsci.ctx.gsc_usr_mds = (tmp_int != 0);
 673
 674         /* mapped uid */
 675         rv = get_int(&mesg, (int *) &rsci.ctx.gsc_mapped_uid);
 676         if (rv) {
 677                 CERROR("fail to get mapped uid\n");
 678                 goto out;
 679         }
 680
 681         rscp = rsc_lookup(&rsci);
 682         if (!rscp)
 683                 goto out;
 684
 685         /* uid, or NEGATIVE */
 686         rv = get_int(&mesg, (int *) &rsci.ctx.gsc_uid);
 687         if (rv == -EINVAL)
 688                 goto out;
 689         if (rv == -ENOENT) {
 690                 CERROR("NOENT? set rsc entry negative\n");
 691                 set_bit(CACHE_NEGATIVE, &rsci.h.flags);
 692         } else {
 693                 rawobj_t tmp_buf;
 694                 unsigned long ctx_expiry;
 695
 696                 /* gid */
 697                 if (get_int(&mesg, (int *) &rsci.ctx.gsc_gid))
 698                         goto out;
 699
 700                 /* mech name */
 701                 len = qword_get(&mesg, buf, mlen);
 702                 if (len < 0)
 703                         goto out;
 704                 gm = lgss_name_to_mech(buf);
 705                 status = -EOPNOTSUPP;
 706                 if (!gm)
 707                         goto out;
 708
 709                 status = -EINVAL;
 710                 /* mech-specific data: */
 711                 len = qword_get(&mesg, buf, mlen);
 712                 if (len < 0)
 713                         goto out;
 714
 715                 tmp_buf.len = len;
 716                 tmp_buf.data = (unsigned char *)buf;
 717                 if (lgss_import_sec_context(&tmp_buf, gm,
 718                                             &rsci.ctx.gsc_mechctx))
 719                         goto out;
 720
 721                 /* currently the expiry time passed down from user-space
 722                  * is invalid, here we retrive it from mech. */
 723                 if (lgss_inquire_context(rsci.ctx.gsc_mechctx, &ctx_expiry)) {
 724                         CERROR("unable to get expire time, drop it\n");
 725                         goto out;
 726                 }
 727                 expiry = (time_t) ctx_expiry;
 728         }
 729
 730         rsci.h.expiry_time = expiry;
 731         rscp = rsc_update(&rsci, rscp);
 732         status = 0;
 733 out:
 734         if (gm)
 735                 lgss_mech_put(gm);
 736         rsc_free(&rsci);
 737         if (rscp)
 738                 cache_put(&rscp->h, &rsc_cache);
 739         else
 740                 status = -ENOMEM;
 741
 742         if (status)
 743                 CERROR("parse rsc error %d\n", status);
 744         return status;
 745 }
 746
 747 #else /* !HAVE_SUNRPC_CACHE_V2 */
 748
 749 static void rsc_put(struct cache_head *item, struct cache_detail *cd)
 750 {
 751         struct rsc *rsci = container_of(item, struct rsc, h);
 752
 753         LASSERT(atomic_read(&item->refcnt) > 0);
 754
 755         if (cache_put(item, cd)) {
 756                 LASSERT(item->next == NULL);
 757                 rsc_free(rsci);
 758                 kfree(rsci); /* created by cache mgmt using kmalloc */
 759         }
 760 }
 761
 762 static inline int rsc_match(struct rsc *new, struct rsc *tmp)
 763 {
 764         return __rsc_match(new, tmp);
 765 }
 766
 767 static inline void rsc_init(struct rsc *new, struct rsc *tmp)
 768 {
 769         __rsc_init(new, tmp);
 770 }
 771
 772 static inline void rsc_update(struct rsc *new, struct rsc *tmp)
 773 {
 774         __rsc_update(new, tmp);
 775 }
 776
 777 static int rsc_parse(struct cache_detail *cd, char *mesg, int mlen)
 778 {
 779         char       *buf = mesg;
 780         int         len, rv, tmp_int;
 781         struct rsc  rsci, *rscp = NULL;
 782         time_t      expiry;
 783         int         status = -EINVAL;
 784
 785         memset(&rsci, 0, sizeof(rsci));
 786
 787         /* context handle */
 788         len = qword_get(&mesg, buf, mlen);
 789         if (len < 0) goto out;
 790         status = -ENOMEM;
 791         if (rawobj_alloc(&rsci.handle, buf, len))
 792                 goto out;
 793
 794         rsci.h.flags = 0;
 795         /* expiry */
 796         expiry = get_expiry(&mesg);
 797         status = -EINVAL;
 798         if (expiry == 0)
 799                 goto out;
 800
 801         /* remote flag */
 802         rv = get_int(&mesg, &tmp_int);
 803         if (rv) {
 804                 CERROR("fail to get remote flag\n");
 805                 goto out;
 806         }
 807         rsci.ctx.gsc_remote = (tmp_int != 0);
 808
 809         /* root user flag */
 810         rv = get_int(&mesg, &tmp_int);
 811         if (rv) {
 812                 CERROR("fail to get oss user flag\n");
 813                 goto out;
 814         }
 815         rsci.ctx.gsc_usr_root = (tmp_int != 0);
 816
 817         /* mds user flag */
 818         rv = get_int(&mesg, &tmp_int);
 819         if (rv) {
 820                 CERROR("fail to get mds user flag\n");
 821                 goto out;
 822         }
 823         rsci.ctx.gsc_usr_mds = (tmp_int != 0);
 824
 825         /* mapped uid */
 826         rv = get_int(&mesg, (int *) &rsci.ctx.gsc_mapped_uid);
 827         if (rv) {
 828                 CERROR("fail to get mapped uid\n");
 829                 goto out;
 830         }
 831
 832         /* uid, or NEGATIVE */
 833         rv = get_int(&mesg, (int *) &rsci.ctx.gsc_uid);
 834         if (rv == -EINVAL)
 835                 goto out;
 836         if (rv == -ENOENT) {
 837                 CERROR("NOENT? set rsc entry negative\n");
 838                 set_bit(CACHE_NEGATIVE, &rsci.h.flags);
 839         } else {
 840                 struct gss_api_mech *gm;
 841                 rawobj_t tmp_buf;
 842                 unsigned long ctx_expiry;
 843
 844                 /* gid */
 845                 if (get_int(&mesg, (int *) &rsci.ctx.gsc_gid))
 846                         goto out;
 847
 848                 /* mech name */
 849                 len = qword_get(&mesg, buf, mlen);
 850                 if (len < 0)
 851                         goto out;
 852                 gm = lgss_name_to_mech(buf);
 853                 status = -EOPNOTSUPP;
 854                 if (!gm)
 855                         goto out;
 856
 857                 status = -EINVAL;
 858                 /* mech-specific data: */
 859                 len = qword_get(&mesg, buf, mlen);
 860                 if (len < 0) {
 861                         lgss_mech_put(gm);
 862                         goto out;
 863                 }
 864                 tmp_buf.len = len;
 865                 tmp_buf.data = (unsigned char *)buf;
 866                 if (lgss_import_sec_context(&tmp_buf, gm,
 867                                             &rsci.ctx.gsc_mechctx)) {
 868                         lgss_mech_put(gm);
 869                         goto out;
 870                 }
 871
 872                 /* currently the expiry time passed down from user-space
 873                  * is invalid, here we retrive it from mech. */
 874                 if (lgss_inquire_context(rsci.ctx.gsc_mechctx, &ctx_expiry)) {
 875                         CERROR("unable to get expire time, drop it\n");
 876                         lgss_mech_put(gm);
 877                         goto out;
 878                 }
 879                 expiry = (time_t) ctx_expiry;
 880
 881                 lgss_mech_put(gm);
 882         }
 883
 884         rsci.h.expiry_time = expiry;
 885         rscp = rsc_lookup(&rsci, 1);
 886         status = 0;
 887 out:
 888         rsc_free(&rsci);
 889         if (rscp)
 890                 rsc_put(&rscp->h, &rsc_cache);
 891
 892         if (status)
 893                 CERROR("parse rsc error %d\n", status);
 894         return status;
 895 }
 896
 897 #endif /* HAVE_SUNRPC_CACHE_V2 */
 898
 899
 900 static struct cache_detail rsc_cache = {
 901         .hash_size      = RSC_HASHMAX,
 902         .hash_table     = rsc_table,
 903         .name           = "auth.sptlrpc.context",
 904         .cache_put      = rsc_put,
 905         .cache_parse    = rsc_parse,
 906 #ifdef HAVE_SUNRPC_CACHE_V2
 907         .match          = rsc_match,
 908         .init           = rsc_init,
 909         .update         = update_rsc,
 910         .alloc          = rsc_alloc,
 911 #endif
 912 };
 913
 914 #ifdef HAVE_SUNRPC_CACHE_V2
 915
 916 static struct rsc *rsc_lookup(struct rsc *item)
 917 {
 918         struct cache_head *ch;
 919         int                hash = rsc_hash(item);
 920
 921         ch = sunrpc_cache_lookup(&rsc_cache, &item->h, hash);
 922         if (ch)
 923                 return container_of(ch, struct rsc, h);
 924         else
 925                 return NULL;
 926 }
 927
 928 static struct rsc *rsc_update(struct rsc *new, struct rsc *old)
 929 {
 930         struct cache_head *ch;
 931         int                hash = rsc_hash(new);
 932
 933         ch = sunrpc_cache_update(&rsc_cache, &new->h, &old->h, hash);
 934         if (ch)
 935                 return container_of(ch, struct rsc, h);
 936         else
 937                 return NULL;
 938 }
 939
 940 #define COMPAT_RSC_PUT(item, cd)        cache_put((item), (cd))
 941
 942 #else
 943
 944 static DefineSimpleCacheLookup(rsc, 0);
 945
 946 #define COMPAT_RSC_PUT(item, cd)        rsc_put((item), (cd))
 947
 948 #endif
 949
 950 /****************************************
 951  * rsc cache flush                      *
 952  ****************************************/
 953
 954 typedef int rsc_entry_match(struct rsc *rscp, long data);
 955
 956 static void rsc_flush(rsc_entry_match *match, long data)
 957 {
 958         struct cache_head **ch;
 959         struct rsc *rscp;
 960         int n;
 961         ENTRY;
 962
 963         write_lock(&rsc_cache.hash_lock);
 964         for (n = 0; n < RSC_HASHMAX; n++) {
 965                 for (ch = &rsc_cache.hash_table[n]; *ch;) {
 966                         rscp = container_of(*ch, struct rsc, h);
 967
 968                         if (!match(rscp, data)) {
 969                                 ch = &((*ch)->next);
 970                                 continue;
 971                         }
 972
 973                         /* it seems simply set NEGATIVE doesn't work */
 974                         *ch = (*ch)->next;
 975                         rscp->h.next = NULL;
 976                         cache_get(&rscp->h);
 977                         set_bit(CACHE_NEGATIVE, &rscp->h.flags);
 978                         COMPAT_RSC_PUT(&rscp->h, &rsc_cache);
 979                         rsc_cache.entries--;
 980                 }
 981         }
 982         write_unlock(&rsc_cache.hash_lock);
 983         EXIT;
 984 }
 985
 986 static int match_uid(struct rsc *rscp, long uid)
 987 {
 988         if ((int) uid == -1)
 989                 return 1;
 990         return ((int) rscp->ctx.gsc_uid == (int) uid);
 991 }
 992
 993 static int match_target(struct rsc *rscp, long target)
 994 {
 995         return (rscp->target == (struct obd_device *) target);
 996 }
 997
 998 static inline void rsc_flush_uid(int uid)
 999 {
1000         if (uid == -1)
1001                 CWARN("flush all gss contexts...\n");
1002
1003         rsc_flush(match_uid, (long) uid);
1004 }
1005
1006 static inline void rsc_flush_target(struct obd_device *target)
1007 {
1008         rsc_flush(match_target, (long) target);
1009 }
1010
1011 void gss_secsvc_flush(struct obd_device *target)
1012 {
1013         rsc_flush_target(target);
1014 }
1015 EXPORT_SYMBOL(gss_secsvc_flush);
1016
1017 static struct rsc *gss_svc_searchbyctx(rawobj_t *handle)
1018 {
1019         struct rsc  rsci;
1020         struct rsc *found;
1021
1022         memset(&rsci, 0, sizeof(rsci));
1023         if (rawobj_dup(&rsci.handle, handle))
1024                 return NULL;
1025
1026 #ifdef HAVE_SUNRPC_CACHE_V2
1027         found = rsc_lookup(&rsci);
1028 #else
1029         found = rsc_lookup(&rsci, 0);
1030 #endif
1031         rsc_free(&rsci);
1032         if (!found)
1033                 return NULL;
1034         if (cache_check(&rsc_cache, &found->h, NULL))
1035                 return NULL;
1036         return found;
1037 }
1038
1039 #ifdef HAVE_SUNRPC_CACHE_V2
1040
1041 int gss_svc_upcall_install_rvs_ctx(struct obd_import *imp,
1042                                    struct gss_sec *gsec,
1043                                    struct gss_cli_ctx *gctx)
1044 {
1045         struct rsc      rsci, *rscp = NULL;
1046         unsigned long   ctx_expiry;
1047         __u32           major;
1048         int             rc;
1049         ENTRY;
1050
1051         memset(&rsci, 0, sizeof(rsci));
1052
1053         if (rawobj_alloc(&rsci.handle, (char *) &gsec->gs_rvs_hdl,
1054                          sizeof(gsec->gs_rvs_hdl)))
1055                 GOTO(out, rc = -ENOMEM);
1056
1057         rscp = rsc_lookup(&rsci);
1058         if (rscp == NULL)
1059                 GOTO(out, rc = -ENOMEM);
1060
1061         major = lgss_copy_reverse_context(gctx->gc_mechctx,
1062                                           &rsci.ctx.gsc_mechctx);
1063         if (major != GSS_S_COMPLETE)
1064                 GOTO(out, rc = -ENOMEM);
1065
1066         if (lgss_inquire_context(rsci.ctx.gsc_mechctx, &ctx_expiry)) {
1067                 CERROR("unable to get expire time, drop it\n");
1068                 GOTO(out, rc = -EINVAL);
1069         }
1070         rsci.h.expiry_time = (time_t) ctx_expiry;
1071
1072         /* FIXME */
1073         rsci.ctx.gsc_usr_root = 1;
1074         rsci.ctx.gsc_usr_mds= 1;
1075         rsci.ctx.gsc_reverse = 1;
1076
1077         rscp = rsc_update(&rsci, rscp);
1078         if (rscp == NULL)
1079                 GOTO(out, rc = -ENOMEM);
1080
1081         rscp->target = imp->imp_obd;
1082         rawobj_dup(&gctx->gc_svc_handle, &rscp->handle);
1083
1084         CWARN("create reverse svc ctx %p to %s: idx "LPX64"\n",
1085               &rscp->ctx, obd2cli_tgt(imp->imp_obd), gsec->gs_rvs_hdl);
1086         rc = 0;
1087 out:
1088         if (rscp)
1089                 cache_put(&rscp->h, &rsc_cache);
1090         rsc_free(&rsci);
1091
1092         if (rc)
1093                 CERROR("create reverse svc ctx: idx "LPX64", rc %d\n",
1094                        gsec->gs_rvs_hdl, rc);
1095         RETURN(rc);
1096 }
1097
1098 #else /* !HAVE_SUNRPC_CACHE_V2 */
1099
1100 int gss_svc_upcall_install_rvs_ctx(struct obd_import *imp,
1101                                    struct gss_sec *gsec,
1102                                    struct gss_cli_ctx *gctx)
1103 {
1104         struct rsc      rsci, *rscp;
1105         unsigned long   ctx_expiry;
1106         __u32           major;
1107         int             rc;
1108         ENTRY;
1109
1110         memset(&rsci, 0, sizeof(rsci));
1111
1112         if (rawobj_alloc(&rsci.handle, (char *) &gsec->gs_rvs_hdl,
1113                          sizeof(gsec->gs_rvs_hdl)))
1114                 GOTO(out, rc = -ENOMEM);
1115
1116         major = lgss_copy_reverse_context(gctx->gc_mechctx,
1117                                           &rsci.ctx.gsc_mechctx);
1118         if (major != GSS_S_COMPLETE)
1119                 GOTO(out, rc = -ENOMEM);
1120
1121         if (lgss_inquire_context(rsci.ctx.gsc_mechctx, &ctx_expiry)) {
1122                 CERROR("unable to get expire time, drop it\n");
1123                 GOTO(out, rc = -ENOMEM);
1124         }
1125         rsci.h.expiry_time = (time_t) ctx_expiry;
1126
1127         /* FIXME */
1128         rsci.ctx.gsc_usr_root = 1;
1129         rsci.ctx.gsc_usr_mds= 1;
1130         rsci.ctx.gsc_reverse = 1;
1131
1132         rscp = rsc_lookup(&rsci, 1);
1133         if (rscp == NULL) {
1134                 CERROR("rsc lookup failed\n");
1135                 GOTO(out, rc = -ENOMEM);
1136         }
1137
1138         rscp->target = imp->imp_obd;
1139         rawobj_dup(&gctx->gc_svc_handle, &rscp->handle);
1140
1141         CWARN("create reverse svc ctx %p to %s: idx "LPX64"\n",
1142               &rscp->ctx, obd2cli_tgt(imp->imp_obd), gsec->gs_rvs_hdl);
1143         rsc_put(&rscp->h, &rsc_cache);
1144         rc = 0;
1145 out:
1146         rsc_free(&rsci);
1147         if (rc)
1148                 CERROR("create reverse svc ctx: idx "LPX64", rc %d\n",
1149                        gsec->gs_rvs_hdl, rc);
1150         RETURN(rc);
1151 }
1152
1153 #endif /* HAVE_SUNRPC_CACHE_V2 */
1154
1155 int gss_svc_upcall_expire_rvs_ctx(rawobj_t *handle)
1156 {
1157         const cfs_time_t        expire = 20;
1158         struct rsc             *rscp;
1159
1160         rscp = gss_svc_searchbyctx(handle);
1161         if (rscp) {
1162                 CDEBUG(D_SEC, "reverse svcctx %p (rsc %p) expire soon\n",
1163                        &rscp->ctx, rscp);
1164
1165                 rscp->h.expiry_time = cfs_time_current_sec() + expire;
1166                 COMPAT_RSC_PUT(&rscp->h, &rsc_cache);
1167         }
1168         return 0;
1169 }
1170
1171 int gss_svc_upcall_dup_handle(rawobj_t *handle, struct gss_svc_ctx *ctx)
1172 {
1173         struct rsc *rscp = container_of(ctx, struct rsc, ctx);
1174
1175         return rawobj_dup(handle, &rscp->handle);
1176 }
1177
1178 int gss_svc_upcall_update_sequence(rawobj_t *handle, __u32 seq)
1179 {
1180         struct rsc             *rscp;
1181
1182         rscp = gss_svc_searchbyctx(handle);
1183         if (rscp) {
1184                 CDEBUG(D_SEC, "reverse svcctx %p (rsc %p) update seq to %u\n",
1185                        &rscp->ctx, rscp, seq + 1);
1186
1187                 rscp->ctx.gsc_rvs_seq = seq + 1;
1188                 COMPAT_RSC_PUT(&rscp->h, &rsc_cache);
1189         }
1190         return 0;
1191 }
1192
1193 static struct cache_deferred_req* cache_upcall_defer(struct cache_req *req)
1194 {
1195         return NULL;
1196 }
1197 static struct cache_req cache_upcall_chandle = { cache_upcall_defer };
1198
1199 int gss_svc_upcall_handle_init(struct ptlrpc_request *req,
1200                                struct gss_svc_reqctx *grctx,
1201                                struct gss_wire_ctx *gw,
1202                                struct obd_device *target,
1203                                __u32 lustre_svc,
1204                                rawobj_t *rvs_hdl,
1205                                rawobj_t *in_token)
1206 {
1207         struct ptlrpc_reply_state *rs;
1208         struct rsc                *rsci = NULL;
1209         struct rsi                *rsip = NULL, rsikey;
1210         wait_queue_t               wait;
1211         int                        replen = sizeof(struct ptlrpc_body);
1212         struct gss_rep_header     *rephdr;
1213         int                        first_check = 1;
1214         int                        rc = SECSVC_DROP;
1215         ENTRY;
1216
1217         memset(&rsikey, 0, sizeof(rsikey));
1218         rsikey.lustre_svc = lustre_svc;
1219         rsikey.nid = (__u64) req->rq_peer.nid;
1220
1221         /* duplicate context handle. for INIT it always 0 */
1222         if (rawobj_dup(&rsikey.in_handle, &gw->gw_handle)) {
1223                 CERROR("fail to dup context handle\n");
1224                 GOTO(out, rc);
1225         }
1226
1227         if (rawobj_dup(&rsikey.in_token, in_token)) {
1228                 CERROR("can't duplicate token\n");
1229                 rawobj_free(&rsikey.in_handle);
1230                 GOTO(out, rc);
1231         }
1232
1233 #ifdef HAVE_SUNRPC_CACHE_V2
1234         rsip = rsi_lookup(&rsikey);
1235 #else
1236         rsip = rsi_lookup(&rsikey, 0);
1237 #endif
1238         rsi_free(&rsikey);
1239         if (!rsip) {
1240                 CERROR("error in rsi_lookup.\n");
1241
1242                 if (!gss_pack_err_notify(req, GSS_S_FAILURE, 0))
1243                         rc = SECSVC_COMPLETE;
1244
1245                 GOTO(out, rc);
1246         }
1247
1248         cache_get(&rsip->h); /* take an extra ref */
1249         init_waitqueue_head(&rsip->waitq);
1250         init_waitqueue_entry(&wait, current);
1251         add_wait_queue(&rsip->waitq, &wait);
1252
1253 cache_check:
1254         /* Note each time cache_check() will drop a reference if return
1255          * non-zero. We hold an extra reference on initial rsip, but must
1256          * take care of following calls. */
1257         rc = cache_check(&rsi_cache, &rsip->h, &cache_upcall_chandle);
1258         switch (rc) {
1259         case -EAGAIN: {
1260                 int valid;
1261
1262                 if (first_check) {
1263                         first_check = 0;
1264
1265                         read_lock(&rsi_cache.hash_lock);
1266                         valid = test_bit(CACHE_VALID, &rsip->h.flags);
1267                         if (valid == 0)
1268                                 set_current_state(TASK_INTERRUPTIBLE);
1269                         read_unlock(&rsi_cache.hash_lock);
1270
1271                         if (valid == 0)
1272                                 schedule_timeout(GSS_SVC_UPCALL_TIMEOUT * HZ);
1273
1274                         cache_get(&rsip->h);
1275                         goto cache_check;
1276                 }
1277                 CWARN("waited %ds timeout, drop\n", GSS_SVC_UPCALL_TIMEOUT);
1278                 break;
1279         }
1280         case -ENOENT:
1281                 CWARN("cache_check return ENOENT, drop\n");
1282                 break;
1283         case 0:
1284                 /* if not the first check, we have to release the extra
1285                  * reference we just added on it. */
1286                 if (!first_check)
1287                         cache_put(&rsip->h, &rsi_cache);
1288                 CDEBUG(D_SEC, "cache_check is good\n");
1289                 break;
1290         }
1291
1292         remove_wait_queue(&rsip->waitq, &wait);
1293         cache_put(&rsip->h, &rsi_cache);
1294
1295         if (rc)
1296                 GOTO(out, rc = SECSVC_DROP);
1297
1298         rc = SECSVC_DROP;
1299         rsci = gss_svc_searchbyctx(&rsip->out_handle);
1300         if (!rsci) {
1301                 CERROR("authentication failed\n");
1302
1303                 if (!gss_pack_err_notify(req, GSS_S_FAILURE, 0))
1304                         rc = SECSVC_COMPLETE;
1305
1306                 GOTO(out, rc);
1307         } else {
1308                 cache_get(&rsci->h);
1309                 grctx->src_ctx = &rsci->ctx;
1310         }
1311
1312         if (rawobj_dup(&rsci->ctx.gsc_rvs_hdl, rvs_hdl)) {
1313                 CERROR("failed duplicate reverse handle\n");
1314                 GOTO(out, rc);
1315         }
1316
1317         rsci->target = target;
1318
1319         CDEBUG(D_SEC, "server create rsc %p(%u->%s)\n",
1320                rsci, rsci->ctx.gsc_uid, libcfs_nid2str(req->rq_peer.nid));
1321
1322         if (rsip->out_handle.len > PTLRPC_GSS_MAX_HANDLE_SIZE) {
1323                 CERROR("handle size %u too large\n", rsip->out_handle.len);
1324                 GOTO(out, rc = SECSVC_DROP);
1325         }
1326
1327         grctx->src_init = 1;
1328         grctx->src_reserve_len = size_round4(rsip->out_token.len);
1329
1330         rc = lustre_pack_reply_v2(req, 1, &replen, NULL, 0);
1331         if (rc) {
1332                 CERROR("failed to pack reply: %d\n", rc);
1333                 GOTO(out, rc = SECSVC_DROP);
1334         }
1335
1336         rs = req->rq_reply_state;
1337         LASSERT(rs->rs_repbuf->lm_bufcount == 3);
1338         LASSERT(rs->rs_repbuf->lm_buflens[0] >=
1339                 sizeof(*rephdr) + rsip->out_handle.len);
1340         LASSERT(rs->rs_repbuf->lm_buflens[2] >= rsip->out_token.len);
1341
1342         rephdr = lustre_msg_buf(rs->rs_repbuf, 0, 0);
1343         rephdr->gh_version = PTLRPC_GSS_VERSION;
1344         rephdr->gh_flags = 0;
1345         rephdr->gh_proc = PTLRPC_GSS_PROC_ERR;
1346         rephdr->gh_major = rsip->major_status;
1347         rephdr->gh_minor = rsip->minor_status;
1348         rephdr->gh_seqwin = GSS_SEQ_WIN;
1349         rephdr->gh_handle.len = rsip->out_handle.len;
1350         memcpy(rephdr->gh_handle.data, rsip->out_handle.data,
1351                rsip->out_handle.len);
1352
1353         memcpy(lustre_msg_buf(rs->rs_repbuf, 2, 0), rsip->out_token.data,
1354                rsip->out_token.len);
1355
1356         rs->rs_repdata_len = lustre_shrink_msg(rs->rs_repbuf, 2,
1357                                                rsip->out_token.len, 0);
1358
1359         rc = SECSVC_OK;
1360
1361 out:
1362         /* it looks like here we should put rsip also, but this mess up
1363          * with NFS cache mgmt code... FIXME */
1364 #if 0
1365         if (rsip)
1366                 rsi_put(&rsip->h, &rsi_cache);
1367 #endif
1368
1369         if (rsci) {
1370                 /* if anything went wrong, we don't keep the context too */
1371                 if (rc != SECSVC_OK)
1372                         set_bit(CACHE_NEGATIVE, &rsci->h.flags);
1373                 else
1374                         CDEBUG(D_SEC, "create rsc with idx "LPX64"\n",
1375                                gss_handle_to_u64(&rsci->handle));
1376
1377                 COMPAT_RSC_PUT(&rsci->h, &rsc_cache);
1378         }
1379         RETURN(rc);
1380 }
1381
1382 struct gss_svc_ctx *gss_svc_upcall_get_ctx(struct ptlrpc_request *req,
1383                                            struct gss_wire_ctx *gw)
1384 {
1385         struct rsc *rsc;
1386
1387         rsc = gss_svc_searchbyctx(&gw->gw_handle);
1388         if (!rsc) {
1389                 CWARN("Invalid gss ctx idx "LPX64" from %s\n",
1390                       gss_handle_to_u64(&gw->gw_handle),
1391                       libcfs_nid2str(req->rq_peer.nid));
1392                 return NULL;
1393         }
1394
1395         return &rsc->ctx;
1396 }
1397
1398 void gss_svc_upcall_put_ctx(struct gss_svc_ctx *ctx)
1399 {
1400         struct rsc *rsc = container_of(ctx, struct rsc, ctx);
1401
1402         COMPAT_RSC_PUT(&rsc->h, &rsc_cache);
1403 }
1404
1405 void gss_svc_upcall_destroy_ctx(struct gss_svc_ctx *ctx)
1406 {
1407         struct rsc *rsc = container_of(ctx, struct rsc, ctx);
1408
1409         /* can't be found */
1410         set_bit(CACHE_NEGATIVE, &rsc->h.flags);
1411         /* to be removed at next scan */
1412         rsc->h.expiry_time = 1;
1413 }
1414
1415 int __init gss_init_svc_upcall(void)
1416 {
1417         int     i;
1418
1419         cache_register(&rsi_cache);
1420         cache_register(&rsc_cache);
1421
1422         /* FIXME this looks stupid. we intend to give lsvcgssd a chance to open
1423          * the init upcall channel, otherwise there's big chance that the first
1424          * upcall issued before the channel be opened thus nfsv4 cache code will
1425          * drop the request direclty, thus lead to unnecessary recovery time.
1426          * here we wait at miximum 1.5 seconds. */
1427         for (i = 0; i < 6; i++) {
1428                 if (atomic_read(&rsi_cache.readers) > 0)
1429                         break;
1430                 set_current_state(TASK_UNINTERRUPTIBLE);
1431                 LASSERT(HZ >= 4);
1432                 schedule_timeout(HZ / 4);
1433         }
1434
1435         if (atomic_read(&rsi_cache.readers) == 0)
1436                 CWARN("Init channel is not opened by lsvcgssd, following "
1437                       "request might be dropped until lsvcgssd is active\n");
1438
1439         /* this helps reducing context index confliction. after server reboot,
1440          * conflicting request from clients might be filtered out by initial
1441          * sequence number checking, thus no chance to sent error notification
1442          * back to clients. */
1443         get_random_bytes(&__ctx_index, sizeof(__ctx_index));
1444
1445         return 0;
1446 }
1447
1448 void __exit gss_exit_svc_upcall(void)
1449 {
1450         int rc;
1451
1452         cache_purge(&rsi_cache);
1453         if ((rc = cache_unregister(&rsi_cache)))
1454                 CERROR("unregister rsi cache: %d\n", rc);
1455
1456         cache_purge(&rsc_cache);
1457         if ((rc = cache_unregister(&rsc_cache)))
1458                 CERROR("unregister rsc cache: %d\n", rc);
1459 }