Whamcloud - gitweb
land b_hd_sec_client_oss onto HEAD.
[fs/lustre-release.git] / lustre / sec / gss / svcsec_gss.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  * Copyright 2004, Cluster File Systems, Inc.
6  * All rights reserved
7  * Author: Eric Mei <ericm@clusterfs.com>
8  */
9
10 /*
11  * Neil Brown <neilb@cse.unsw.edu.au>
12  * J. Bruce Fields <bfields@umich.edu>
13  * Andy Adamson <andros@umich.edu>
14  * Dug Song <dugsong@monkey.org>
15  *
16  * RPCSEC_GSS server authentication.
17  * This implements RPCSEC_GSS as defined in rfc2203 (rpcsec_gss) and rfc2078
18  * (gssapi)
19  *
20  * The RPCSEC_GSS involves three stages:
21  *  1/ context creation
22  *  2/ data exchange
23  *  3/ context destruction
24  *
25  * Context creation is handled largely by upcalls to user-space.
26  *  In particular, GSS_Accept_sec_context is handled by an upcall
27  * Data exchange is handled entirely within the kernel
28  *  In particular, GSS_GetMIC, GSS_VerifyMIC, GSS_Seal, GSS_Unseal are in-kernel.
29  * Context destruction is handled in-kernel
30  *  GSS_Delete_sec_context is in-kernel
31  *
32  * Context creation is initiated by a RPCSEC_GSS_INIT request arriving.
33  * The context handle and gss_token are used as a key into the rpcsec_init cache.
34  * The content of this cache includes some of the outputs of GSS_Accept_sec_context,
35  * being major_status, minor_status, context_handle, reply_token.
36  * These are sent back to the client.
37  * Sequence window management is handled by the kernel.  The window size if currently
38  * a compile time constant.
39  *
40  * When user-space is happy that a context is established, it places an entry
41  * in the rpcsec_context cache. The key for this cache is the context_handle.
42  * The content includes:
43  *   uid/gidlist - for determining access rights
44  *   mechanism type
45  *   mechanism specific information, such as a key
46  *
47  */
48
49 #define DEBUG_SUBSYSTEM S_SEC
50 #ifdef __KERNEL__
51 #include <linux/types.h>
52 #include <linux/init.h>
53 #include <linux/module.h>
54 #include <linux/slab.h>
55 #include <linux/hash.h>
56 #else
57 #include <liblustre.h>
58 #endif
59
60 #include <linux/sunrpc/cache.h>
61
62 #include <libcfs/kp30.h>
63 #include <linux/obd.h>
64 #include <linux/obd_class.h>
65 #include <linux/obd_support.h>
66 #include <linux/lustre_idl.h>
67 #include <linux/lustre_net.h>
68 #include <linux/lustre_import.h>
69 #include <linux/lustre_sec.h>
70                                                                                                                         
71 #include "gss_err.h"
72 #include "gss_internal.h"
73 #include "gss_api.h"
74
75 static inline unsigned long hash_mem(char *buf, int length, int bits)
76 {
77         unsigned long hash = 0;
78         unsigned long l = 0;
79         int len = 0;
80         unsigned char c;
81         do {
82                 if (len == length) {
83                         c = (char)len; len = -1;
84                 } else
85                         c = *buf++;
86                 l = (l << 8) | c;
87                 len++;
88                 if ((len & (BITS_PER_LONG/8-1))==0)
89                         hash = hash_long(hash^l, BITS_PER_LONG);
90         } while (len);
91         return hash >> (BITS_PER_LONG - bits);
92 }
93
94 /* The rpcsec_init cache is used for mapping RPCSEC_GSS_{,CONT_}INIT requests
95  * into replies.
96  *
97  * Key is context handle (\x if empty) and gss_token.
98  * Content is major_status minor_status (integers) context_handle, reply_token.
99  *
100  */
101
102 #define RSI_HASHBITS    6
103 #define RSI_HASHMAX     (1<<RSI_HASHBITS)
104 #define RSI_HASHMASK    (RSI_HASHMAX-1)
105
106 struct rsi {
107         struct cache_head       h;
108         __u32                   lustre_svc;
109         __u32                   naltype;
110         __u32                   netid;
111         __u64                   nid;
112         rawobj_t                in_handle, in_token, in_srv_type;
113         rawobj_t                out_handle, out_token;
114         int                     major_status, minor_status;
115 };
116
117 static struct cache_head *rsi_table[RSI_HASHMAX];
118 static struct cache_detail rsi_cache;
119
120 static void rsi_free(struct rsi *rsii)
121 {
122         rawobj_free(&rsii->in_handle);
123         rawobj_free(&rsii->in_token);
124         rawobj_free(&rsii->out_handle);
125         rawobj_free(&rsii->out_token);
126 }
127
128 static void rsi_put(struct cache_head *item, struct cache_detail *cd)
129 {
130         struct rsi *rsii = container_of(item, struct rsi, h);
131         LASSERT(atomic_read(&item->refcnt) > 0);
132         if (cache_put(item, cd)) {
133                 LASSERT(item->next == NULL);
134                 rsi_free(rsii);
135                 OBD_FREE(rsii, sizeof(*rsii));
136         }
137 }
138
139 static inline int rsi_hash(struct rsi *item)
140 {
141         return hash_mem((char *)item->in_handle.data, item->in_handle.len, RSI_HASHBITS)
142                 ^ hash_mem((char *)item->in_token.data, item->in_token.len, RSI_HASHBITS);
143 }
144
145 static inline int rsi_match(struct rsi *item, struct rsi *tmp)
146 {
147         return (rawobj_equal(&item->in_handle, &tmp->in_handle) &&
148                 rawobj_equal(&item->in_token, &tmp->in_token));
149 }
150
151 static void rsi_request(struct cache_detail *cd,
152                         struct cache_head *h,
153                         char **bpp, int *blen)
154 {
155         struct rsi *rsii = container_of(h, struct rsi, h);
156
157         qword_addhex(bpp, blen, (char *) &rsii->lustre_svc,
158                      sizeof(rsii->lustre_svc));
159         qword_addhex(bpp, blen, (char *) &rsii->naltype, sizeof(rsii->naltype));
160         qword_addhex(bpp, blen, (char *) &rsii->netid, sizeof(rsii->netid));
161         qword_addhex(bpp, blen, (char *) &rsii->nid, sizeof(rsii->nid));
162         qword_addhex(bpp, blen, rsii->in_handle.data, rsii->in_handle.len);
163         qword_addhex(bpp, blen, rsii->in_token.data, rsii->in_token.len);
164         (*bpp)[-1] = '\n';
165 }
166
167 static int
168 gssd_reply(struct rsi *item)
169 {
170         struct rsi *tmp;
171         struct cache_head **hp, **head;
172         ENTRY;
173
174         head = &rsi_cache.hash_table[rsi_hash(item)];
175         write_lock(&rsi_cache.hash_lock);
176         for (hp = head; *hp != NULL; hp = &tmp->h.next) {
177                 tmp = container_of(*hp, struct rsi, h);
178                 if (rsi_match(tmp, item)) {
179                         cache_get(&tmp->h);
180                         clear_bit(CACHE_HASHED, &tmp->h.flags);
181                         *hp = tmp->h.next;
182                         tmp->h.next = NULL;
183                         rsi_cache.entries--;
184                         if (test_bit(CACHE_VALID, &tmp->h.flags)) {
185                                 CERROR("rsi is valid\n");
186                                 write_unlock(&rsi_cache.hash_lock);
187                                 rsi_put(&tmp->h, &rsi_cache);
188                                 RETURN(-EINVAL);
189                         }
190                         set_bit(CACHE_HASHED, &item->h.flags);
191                         item->h.next = *hp;
192                         *hp = &item->h;
193                         rsi_cache.entries++;
194                         set_bit(CACHE_VALID, &item->h.flags);
195                         item->h.last_refresh = get_seconds();
196                         write_unlock(&rsi_cache.hash_lock);
197                         cache_fresh(&rsi_cache, &tmp->h, 0);
198                         rsi_put(&tmp->h, &rsi_cache);
199                         RETURN(0);
200                 }
201         }
202         write_unlock(&rsi_cache.hash_lock);
203         RETURN(-EINVAL);
204 }
205
206 /* XXX
207  * here we just wait here for its completion or timedout. it's a
208  * hacking but works, and we'll comeup with real fix if we decided
209  * to still stick with NFS4 cache code
210  */
211 static struct rsi *
212 gssd_upcall(struct rsi *item, struct cache_req *chandle)
213 {
214         struct rsi *tmp;
215         struct cache_head **hp, **head;
216         unsigned long starttime;
217         ENTRY;
218
219         head = &rsi_cache.hash_table[rsi_hash(item)];
220         read_lock(&rsi_cache.hash_lock);
221         for (hp = head; *hp != NULL; hp = &tmp->h.next) {
222                 tmp = container_of(*hp, struct rsi, h);
223                 if (rsi_match(tmp, item)) {
224                         LBUG();
225                         if (!test_bit(CACHE_VALID, &tmp->h.flags)) {
226                                 CERROR("found rsi without VALID\n");
227                                 read_unlock(&rsi_cache.hash_lock);
228                                 return NULL;
229                         }
230                         *hp = tmp->h.next;
231                         tmp->h.next = NULL;
232                         rsi_cache.entries--;
233                         cache_get(&tmp->h);
234                         read_unlock(&rsi_cache.hash_lock);
235                         return tmp;
236                 }
237         }
238         cache_get(&item->h);
239         //set_bit(CACHE_HASHED, &item->h.flags);
240         item->h.next = *head;
241         *head = &item->h;
242         rsi_cache.entries++;
243         read_unlock(&rsi_cache.hash_lock);
244         //cache_get(&item->h);
245
246         cache_check(&rsi_cache, &item->h, chandle);
247         starttime = get_seconds();
248         do {
249                 set_current_state(TASK_UNINTERRUPTIBLE);
250                 schedule_timeout(HZ/2);
251                 read_lock(&rsi_cache.hash_lock);
252                 for (hp = head; *hp != NULL; hp = &tmp->h.next) {
253                         tmp = container_of(*hp, struct rsi, h);
254                         if (tmp == item)
255                                 continue;
256                         if (rsi_match(tmp, item)) {
257                                 if (!test_bit(CACHE_VALID, &tmp->h.flags)) {
258                                         read_unlock(&rsi_cache.hash_lock);
259                                         return NULL;
260                                 }
261                                 cache_get(&tmp->h);
262                                 clear_bit(CACHE_HASHED, &tmp->h.flags);
263                                 *hp = tmp->h.next;
264                                 tmp->h.next = NULL;
265                                 rsi_cache.entries--;
266                                 read_unlock(&rsi_cache.hash_lock);
267                                 return tmp;
268                         }
269                 }
270                 read_unlock(&rsi_cache.hash_lock);
271         } while ((get_seconds() - starttime) <= 15);
272         CERROR("15s timeout while waiting cache refill\n");
273         return NULL;
274 }
275
276 static int rsi_parse(struct cache_detail *cd,
277                      char *mesg, int mlen)
278 {
279         /* context token expiry major minor context token */
280         char *buf = mesg;
281         char *ep;
282         int len;
283         struct rsi *rsii;
284         time_t expiry;
285         int status = -EINVAL;
286         ENTRY;
287
288         OBD_ALLOC(rsii, sizeof(*rsii));
289         if (!rsii)
290                 RETURN(-ENOMEM);
291         cache_init(&rsii->h);
292
293         /* handle */
294         len = qword_get(&mesg, buf, mlen);
295         if (len < 0)
296                 goto out;
297         if (rawobj_alloc(&rsii->in_handle, buf, len)) {
298                 status = -ENOMEM;
299                 goto out;
300         }
301
302         /* token */
303         len = qword_get(&mesg, buf, mlen);
304         if (len < 0)
305                 goto out;
306         if (rawobj_alloc(&rsii->in_token, buf, len)) {
307                 status = -ENOMEM;
308                 goto out;
309         }
310
311         /* expiry */
312         expiry = get_expiry(&mesg);
313         if (expiry == 0)
314                 goto out;
315
316         /* major */
317         len = qword_get(&mesg, buf, mlen);
318         if (len <= 0)
319                 goto out;
320         rsii->major_status = simple_strtol(buf, &ep, 10);
321         if (*ep)
322                 goto out;
323
324         /* minor */
325         len = qword_get(&mesg, buf, mlen);
326         if (len <= 0)
327                 goto out;
328         rsii->minor_status = simple_strtol(buf, &ep, 10);
329         if (*ep)
330                 goto out;
331
332         /* out_handle */
333         len = qword_get(&mesg, buf, mlen);
334         if (len < 0)
335                 goto out;
336         if (rawobj_alloc(&rsii->out_handle, buf, len)) {
337                 status = -ENOMEM;
338                 goto out;
339         }
340
341         /* out_token */
342         len = qword_get(&mesg, buf, mlen);
343         if (len < 0)
344                 goto out;
345         if (rawobj_alloc(&rsii->out_token, buf, len)) {
346                 status = -ENOMEM;
347                 goto out;
348         }
349
350         rsii->h.expiry_time = expiry;
351         status = gssd_reply(rsii);
352 out:
353         if (rsii)
354                 rsi_put(&rsii->h, &rsi_cache);
355         RETURN(status);
356 }
357
358 static struct cache_detail rsi_cache = {
359         .hash_size      = RSI_HASHMAX,
360         .hash_table     = rsi_table,
361         .name           = "auth.ptlrpcs.init",
362         .cache_put      = rsi_put,
363         .cache_request  = rsi_request,
364         .cache_parse    = rsi_parse,
365 };
366
367 /*
368  * The rpcsec_context cache is used to store a context that is
369  * used in data exchange.
370  * The key is a context handle. The content is:
371  *  uid, gidlist, mechanism, service-set, mech-specific-data
372  */
373
374 #define RSC_HASHBITS    10
375 #define RSC_HASHMAX     (1<<RSC_HASHBITS)
376 #define RSC_HASHMASK    (RSC_HASHMAX-1)
377
378 #define GSS_SEQ_WIN     512
379
380 struct gss_svc_seq_data {
381         /* highest seq number seen so far: */
382         __u32                   sd_max;
383         /* for i such that sd_max-GSS_SEQ_WIN < i <= sd_max, the i-th bit of
384          * sd_win is nonzero iff sequence number i has been seen already: */
385         unsigned long           sd_win[GSS_SEQ_WIN/BITS_PER_LONG];
386         spinlock_t              sd_lock;
387 };
388
389 struct rsc {
390         struct cache_head       h;
391         rawobj_t                handle;
392         __u32                   remote_realm;
393         struct vfs_cred         cred;
394         uid_t                   mapped_uid;
395         struct gss_svc_seq_data seqdata;
396         struct gss_ctx         *mechctx;
397 };
398
399 static struct cache_head *rsc_table[RSC_HASHMAX];
400 static struct cache_detail rsc_cache;
401
402 static void rsc_free(struct rsc *rsci)
403 {
404         rawobj_free(&rsci->handle);
405         if (rsci->mechctx)
406                 kgss_delete_sec_context(&rsci->mechctx);
407 #if 0
408         if (rsci->cred.vc_ginfo)
409                 put_group_info(rsci->cred.vc_ginfo);
410 #endif
411 }
412
413 static void rsc_put(struct cache_head *item, struct cache_detail *cd)
414 {
415         struct rsc *rsci = container_of(item, struct rsc, h);
416
417         LASSERT(atomic_read(&item->refcnt) > 0);
418         if (cache_put(item, cd)) {
419                 LASSERT(item->next == NULL);
420                 rsc_free(rsci);
421                 OBD_FREE(rsci, sizeof(*rsci));
422         }
423 }
424
425 static inline int
426 rsc_hash(struct rsc *rsci)
427 {
428         return hash_mem((char *)rsci->handle.data,
429                         rsci->handle.len, RSC_HASHBITS);
430 }
431
432 static inline int
433 rsc_match(struct rsc *new, struct rsc *tmp)
434 {
435         return rawobj_equal(&new->handle, &tmp->handle);
436 }
437
438 static struct rsc *rsc_lookup(struct rsc *item, int set)
439 {
440         struct rsc *tmp = NULL;
441         struct cache_head **hp, **head;
442         head = &rsc_cache.hash_table[rsc_hash(item)];
443         ENTRY;
444
445         if (set)
446                 write_lock(&rsc_cache.hash_lock);
447         else
448                 read_lock(&rsc_cache.hash_lock);
449         for (hp = head; *hp != NULL; hp = &tmp->h.next) {
450                 tmp = container_of(*hp, struct rsc, h);
451                 if (!rsc_match(tmp, item))
452                         continue;
453                 cache_get(&tmp->h);
454                 if (!set)
455                         goto out_noset;
456                 *hp = tmp->h.next;
457                 tmp->h.next = NULL;
458                 clear_bit(CACHE_HASHED, &tmp->h.flags);
459                 rsc_put(&tmp->h, &rsc_cache);
460                 goto out_set;
461         }
462         /* Didn't find anything */
463         if (!set)
464                 goto out_nada;
465         rsc_cache.entries++;
466 out_set:
467         set_bit(CACHE_HASHED, &item->h.flags);
468         item->h.next = *head;
469         *head = &item->h;
470         write_unlock(&rsc_cache.hash_lock);
471         cache_fresh(&rsc_cache, &item->h, item->h.expiry_time);
472         cache_get(&item->h);
473         RETURN(item);
474 out_nada:
475         tmp = NULL;
476 out_noset:
477         read_unlock(&rsc_cache.hash_lock);
478         RETURN(tmp);
479 }
480                                                                                                                         
481 static int rsc_parse(struct cache_detail *cd,
482                      char *mesg, int mlen)
483 {
484         /* contexthandle expiry [ uid gid N <n gids> mechname
485          * ...mechdata... ] */
486         char *buf = mesg;
487         int len, rv;
488         struct rsc *rsci, *res = NULL;
489         time_t expiry;
490         int status = -EINVAL;
491
492         OBD_ALLOC(rsci, sizeof(*rsci));
493         if (!rsci) {
494                 CERROR("fail to alloc rsci\n");
495                 return -ENOMEM;
496         }
497         cache_init(&rsci->h);
498
499         /* context handle */
500         len = qword_get(&mesg, buf, mlen);
501         if (len < 0) goto out;
502         status = -ENOMEM;
503         if (rawobj_alloc(&rsci->handle, buf, len))
504                 goto out;
505
506         /* expiry */
507         expiry = get_expiry(&mesg);
508         status = -EINVAL;
509         if (expiry == 0)
510                 goto out;
511
512         /* remote flag */
513         rv = get_int(&mesg, (int *)&rsci->remote_realm);
514         if (rv) {
515                 CERROR("fail to get remote flag\n");
516                 goto out;
517         }
518
519         /* mapped uid */
520         rv = get_int(&mesg, (int *)&rsci->mapped_uid);
521         if (rv) {
522                 CERROR("fail to get mapped uid\n");
523                 goto out;
524         }
525
526         /* uid, or NEGATIVE */
527         rv = get_int(&mesg, (int *)&rsci->cred.vc_uid);
528         if (rv == -EINVAL)
529                 goto out;
530         if (rv == -ENOENT) {
531                 CERROR("NOENT? set rsc entry negative\n");
532                 set_bit(CACHE_NEGATIVE, &rsci->h.flags);
533         } else {
534                 struct gss_api_mech *gm;
535                 rawobj_t tmp_buf;
536                 __u64 ctx_expiry;
537
538                 /* gid */
539                 if (get_int(&mesg, (int *)&rsci->cred.vc_gid))
540                         goto out;
541
542                 /* mech name */
543                 len = qword_get(&mesg, buf, mlen);
544                 if (len < 0)
545                         goto out;
546                 gm = kgss_name_to_mech(buf);
547                 status = -EOPNOTSUPP;
548                 if (!gm)
549                         goto out;
550
551                 status = -EINVAL;
552                 /* mech-specific data: */
553                 len = qword_get(&mesg, buf, mlen);
554                 if (len < 0) {
555                         kgss_mech_put(gm);
556                         goto out;
557                 }
558                 tmp_buf.len = len;
559                 tmp_buf.data = (unsigned char *)buf;
560                 if (kgss_import_sec_context(&tmp_buf, gm, &rsci->mechctx)) {
561                         kgss_mech_put(gm);
562                         goto out;
563                 }
564
565                 /* currently the expiry time passed down from user-space
566                  * is invalid, here we retrive it from mech.
567                  */
568                 if (kgss_inquire_context(rsci->mechctx, &ctx_expiry)) {
569                         CERROR("unable to get expire time, drop it\n");
570                         set_bit(CACHE_NEGATIVE, &rsci->h.flags);
571                         kgss_mech_put(gm);
572                         goto out;
573                 }
574                 expiry = (time_t) ctx_expiry;
575
576                 kgss_mech_put(gm);
577         }
578         rsci->h.expiry_time = expiry;
579         spin_lock_init(&rsci->seqdata.sd_lock);
580         res = rsc_lookup(rsci, 1);
581         rsc_put(&res->h, &rsc_cache);
582         status = 0;
583 out:
584         if (rsci)
585                 rsc_put(&rsci->h, &rsc_cache);
586         return status;
587 }
588
589 /*
590  * flush all entries with @uid. @uid == -1 will match all.
591  * we only know the uid, maybe netid/nid in the future, in all cases
592  * we must search the whole cache
593  */
594 static void rsc_flush(uid_t uid)
595 {
596         struct cache_head **ch;
597         struct rsc *rscp;
598         int n;
599         ENTRY;
600
601         write_lock(&rsc_cache.hash_lock);
602         for (n = 0; n < RSC_HASHMAX; n++) {
603                 for (ch = &rsc_cache.hash_table[n]; *ch;) {
604                         rscp = container_of(*ch, struct rsc, h);
605                         if (uid == -1 || rscp->cred.vc_uid == uid) {
606                                 /* it seems simply set NEGATIVE doesn't work */
607                                 *ch = (*ch)->next;
608                                 rscp->h.next = NULL;
609                                 cache_get(&rscp->h);
610                                 set_bit(CACHE_NEGATIVE, &rscp->h.flags);
611                                 clear_bit(CACHE_HASHED, &rscp->h.flags);
612                                 CDEBUG(D_SEC, "flush rsc %p for uid %u\n",
613                                        rscp, rscp->cred.vc_uid);
614                                 rsc_put(&rscp->h, &rsc_cache);
615                                 rsc_cache.entries--;
616                                 continue;
617                         }
618                         ch = &((*ch)->next);
619                 }
620         }
621         write_unlock(&rsc_cache.hash_lock);
622         EXIT;
623 }
624
625 static struct cache_detail rsc_cache = {
626         .hash_size      = RSC_HASHMAX,
627         .hash_table     = rsc_table,
628         .name           = "auth.ptlrpcs.context",
629         .cache_put      = rsc_put,
630         .cache_parse    = rsc_parse,
631 };
632
633 static struct rsc *
634 gss_svc_searchbyctx(rawobj_t *handle)
635 {
636         struct rsc rsci;
637         struct rsc *found;
638
639         rsci.handle = *handle;
640         found = rsc_lookup(&rsci, 0);
641         if (!found)
642                 return NULL;
643
644         if (cache_check(&rsc_cache, &found->h, NULL))
645                 return NULL;
646
647         return found;
648 }
649
650 /* FIXME
651  * again hacking: only try to give the svcgssd a chance to handle
652  * upcalls.
653  */
654 struct cache_deferred_req* my_defer(struct cache_req *req)
655 {
656         yield();
657         return NULL;
658 }
659 static struct cache_req my_chandle = {my_defer};
660
661 /* Implements sequence number algorithm as specified in RFC 2203. */
662 static int
663 gss_check_seq_num(struct gss_svc_seq_data *sd, __u32 seq_num)
664 {
665         int rc = 0;
666
667         spin_lock(&sd->sd_lock);
668         if (seq_num > sd->sd_max) {
669                 if (seq_num >= sd->sd_max + GSS_SEQ_WIN) {
670                         memset(sd->sd_win, 0, sizeof(sd->sd_win));
671                         sd->sd_max = seq_num;
672                 } else {
673                         while(sd->sd_max < seq_num) {
674                                 sd->sd_max++;
675                                 __clear_bit(sd->sd_max % GSS_SEQ_WIN,
676                                             sd->sd_win);
677                         }
678                 }
679                 __set_bit(seq_num % GSS_SEQ_WIN, sd->sd_win);
680                 goto exit;
681         } else if (seq_num + GSS_SEQ_WIN <= sd->sd_max) {
682                 CERROR("seq %u too low: max %u, win %d\n",
683                         seq_num, sd->sd_max, GSS_SEQ_WIN);
684                 rc = 1;
685                 goto exit;
686         }
687
688         if (__test_and_set_bit(seq_num % GSS_SEQ_WIN, sd->sd_win)) {
689                 CERROR("seq %u is replay: max %u, win %d\n",
690                         seq_num, sd->sd_max, GSS_SEQ_WIN);
691                 rc = 1;
692         }
693 exit:
694         spin_unlock(&sd->sd_lock);
695         return rc;
696 }
697
698 static int
699 gss_svc_verify_request(struct ptlrpc_request *req,
700                        struct rsc *rsci,
701                        struct rpc_gss_wire_cred *gc,
702                        __u32 *vp, __u32 vlen)
703 {
704         struct ptlrpcs_wire_hdr *sec_hdr;
705         struct gss_ctx *ctx = rsci->mechctx;
706         __u32 maj_stat;
707         rawobj_t msg;
708         rawobj_t mic;
709         ENTRY;
710
711         sec_hdr = (struct ptlrpcs_wire_hdr *) req->rq_reqbuf;
712
713         req->rq_reqmsg = (struct lustre_msg *) (req->rq_reqbuf + sizeof(*sec_hdr));
714         req->rq_reqlen = sec_hdr->msg_len;
715
716         msg.len = sec_hdr->msg_len;
717         msg.data = (__u8 *)req->rq_reqmsg;
718
719         mic.len = le32_to_cpu(*vp++);
720         mic.data = (unsigned char *)vp;
721         vlen -= 4;
722
723         if (mic.len > vlen) {
724                 CERROR("checksum len %d, while buffer len %d\n",
725                         mic.len, vlen);
726                 RETURN(GSS_S_CALL_BAD_STRUCTURE);
727         }
728
729         if (mic.len > 256) {
730                 CERROR("invalid mic len %d\n", mic.len);
731                 RETURN(GSS_S_CALL_BAD_STRUCTURE);
732         }
733
734         maj_stat = kgss_verify_mic(ctx, &msg, &mic, NULL);
735         if (maj_stat != GSS_S_COMPLETE) {
736                 CERROR("MIC verification error: major %x\n", maj_stat);
737                 RETURN(maj_stat);
738         }
739
740         if (gss_check_seq_num(&rsci->seqdata, gc->gc_seq)) {
741                 CERROR("discard replayed request %p(o%u,x"LPU64",t"LPU64")\n",
742                         req, req->rq_reqmsg->opc, req->rq_xid,
743                         req->rq_reqmsg->transno);
744                 RETURN(GSS_S_DUPLICATE_TOKEN);
745         }
746
747         RETURN(GSS_S_COMPLETE);
748 }
749
750 static int
751 gss_svc_unseal_request(struct ptlrpc_request *req,
752                        struct rsc *rsci,
753                        struct rpc_gss_wire_cred *gc,
754                        __u32 *vp, __u32 vlen)
755 {
756         struct ptlrpcs_wire_hdr *sec_hdr;
757         struct gss_ctx *ctx = rsci->mechctx;
758         rawobj_t cipher_text, plain_text;
759         __u32 major;
760         ENTRY;
761
762         sec_hdr = (struct ptlrpcs_wire_hdr *) req->rq_reqbuf;
763
764         if (vlen < 4) {
765                 CERROR("vlen only %u\n", vlen);
766                 RETURN(GSS_S_CALL_BAD_STRUCTURE);
767         }
768
769         cipher_text.len = le32_to_cpu(*vp++);
770         cipher_text.data = (__u8 *) vp;
771         vlen -= 4;
772         
773         if (cipher_text.len > vlen) {
774                 CERROR("cipher claimed %u while buf only %u\n",
775                         cipher_text.len, vlen);
776                 RETURN(GSS_S_CALL_BAD_STRUCTURE);
777         }
778
779         plain_text = cipher_text;
780
781         major = kgss_unwrap(ctx, GSS_C_QOP_DEFAULT, &cipher_text, &plain_text);
782         if (major) {
783                 CERROR("unwrap error 0x%x\n", major);
784                 RETURN(major);
785         }
786
787         if (gss_check_seq_num(&rsci->seqdata, gc->gc_seq)) {
788                 CERROR("discard replayed request %p(o%u,x"LPU64",t"LPU64")\n",
789                         req, req->rq_reqmsg->opc, req->rq_xid,
790                         req->rq_reqmsg->transno);
791                 RETURN(GSS_S_DUPLICATE_TOKEN);
792         }
793
794         req->rq_reqmsg = (struct lustre_msg *) (vp);
795         req->rq_reqlen = plain_text.len;
796
797         CDEBUG(D_SEC, "msg len %d\n", req->rq_reqlen);
798
799         RETURN(GSS_S_COMPLETE);
800 }
801
802 static int
803 gss_pack_err_notify(struct ptlrpc_request *req,
804                     __u32 major, __u32 minor)
805 {
806         struct gss_svc_data *svcdata = req->rq_sec_svcdata;
807         __u32 reslen, *resp, *reslenp;
808         char  nidstr[PTL_NALFMT_SIZE];
809         const __u32 secdata_len = 7 * 4;
810         int rc;
811         ENTRY;
812
813         OBD_FAIL_RETURN(OBD_FAIL_SVCGSS_ERR_NOTIFY|OBD_FAIL_ONCE, -EINVAL);
814
815         LASSERT(svcdata);
816         svcdata->is_err_notify = 1;
817         svcdata->reserve_len = 7 * 4;
818
819         rc = lustre_pack_reply(req, 0, NULL, NULL);
820         if (rc) {
821                 CERROR("could not pack reply, err %d\n", rc);
822                 RETURN(rc);
823         }
824
825         LASSERT(req->rq_reply_state);
826         LASSERT(req->rq_reply_state->rs_repbuf);
827         LASSERT(req->rq_reply_state->rs_repbuf_len >= secdata_len);
828         resp = (__u32 *) req->rq_reply_state->rs_repbuf;
829
830         /* header */
831         *resp++ = cpu_to_le32(PTLRPC_SEC_GSS);
832         *resp++ = cpu_to_le32(PTLRPC_SEC_TYPE_NONE);
833         *resp++ = cpu_to_le32(req->rq_replen);
834         reslenp = resp++;
835
836         /* skip lustre msg */
837         resp += req->rq_replen / 4;
838         reslen = svcdata->reserve_len;
839
840         /* gss replay:
841          * version, subflavor, notify, major, minor,
842          * obj1(fake), obj2(fake)
843          */
844         *resp++ = cpu_to_le32(PTLRPC_SEC_GSS_VERSION);
845         *resp++ = cpu_to_le32(PTLRPC_SEC_GSS_KRB5I);
846         *resp++ = cpu_to_le32(PTLRPC_GSS_PROC_ERR);
847         *resp++ = cpu_to_le32(major);
848         *resp++ = cpu_to_le32(minor);
849         *resp++ = 0;
850         *resp++ = 0;
851         reslen -= (4 * 4);
852         /* the actual sec data length */
853         *reslenp = cpu_to_le32(secdata_len);
854
855         req->rq_reply_state->rs_repdata_len += (secdata_len);
856         CDEBUG(D_SEC, "prepare gss error notify(0x%x/0x%x) to %s\n",
857                major, minor,
858                portals_nid2str(req->rq_peer.peer_ni->pni_number,
859                                req->rq_peer.peer_id.nid, nidstr));
860         RETURN(0);
861 }
862
863 static void dump_cache_head(struct cache_head *h)
864 {
865         CWARN("ref %d, fl %lx, n %p, t %ld, %ld\n",
866               atomic_read(&h->refcnt), h->flags, h->next,
867               h->expiry_time, h->last_refresh);
868 }
869 static void dump_rsi(struct rsi *rsi)
870 {
871         CWARN("dump rsi %p\n", rsi);
872         dump_cache_head(&rsi->h);
873         CWARN("%x,%x,%llx\n", rsi->naltype, rsi->netid, rsi->nid);
874         CWARN("len %d, d %p\n", rsi->in_handle.len, rsi->in_handle.data);
875         CWARN("len %d, d %p\n", rsi->in_token.len, rsi->in_token.data);
876         CWARN("len %d, d %p\n", rsi->out_handle.len, rsi->out_handle.data);
877         CWARN("len %d, d %p\n", rsi->out_token.len, rsi->out_token.data);
878 }
879
880 static int
881 gss_svcsec_handle_init(struct ptlrpc_request *req,
882                        struct rpc_gss_wire_cred *gc,
883                        __u32 *secdata, __u32 seclen,
884                        enum ptlrpcs_error *res)
885 {
886         struct gss_svc_data *svcdata = req->rq_sec_svcdata;
887         struct rsc          *rsci;
888         struct rsi          *rsikey, *rsip;
889         rawobj_t             tmpobj;
890         __u32 reslen,       *resp, *reslenp;
891         char                 nidstr[PTL_NALFMT_SIZE];
892         int                  rc;
893         ENTRY;
894
895         LASSERT(svcdata);
896
897         CDEBUG(D_SEC, "processing gss init(%d) request from %s\n", gc->gc_proc,
898                portals_nid2str(req->rq_peer.peer_ni->pni_number,
899                                req->rq_peer.peer_id.nid, nidstr));
900
901         *res = PTLRPCS_BADCRED;
902         OBD_FAIL_RETURN(OBD_FAIL_SVCGSS_INIT_REQ|OBD_FAIL_ONCE, SVC_DROP);
903
904         if (gc->gc_proc == RPC_GSS_PROC_INIT &&
905             gc->gc_ctx.len != 0) {
906                 CERROR("proc %d, ctx_len %d: not really init?\n",
907                 gc->gc_proc == RPC_GSS_PROC_INIT, gc->gc_ctx.len);
908                 RETURN(SVC_DROP);
909         }
910
911         OBD_ALLOC(rsikey, sizeof(*rsikey));
912         if (!rsikey) {
913                 CERROR("out of memory\n");
914                 RETURN(SVC_DROP);
915         }
916         cache_init(&rsikey->h);
917
918         /* obtain lustre svc type */
919         if (seclen < 4) {
920                 CERROR("sec size %d too small\n", seclen);
921                 GOTO(out_rsikey, rc = SVC_DROP);
922         }
923         rsikey->lustre_svc = le32_to_cpu(*secdata++);
924         seclen -= 4;
925
926         /* duplicate context handle. currently always 0 */
927         if (rawobj_dup(&rsikey->in_handle, &gc->gc_ctx)) {
928                 CERROR("fail to dup context handle\n");
929                 GOTO(out_rsikey, rc = SVC_DROP);
930         }
931
932         /* extract token */
933         *res = PTLRPCS_BADVERF;
934         if (rawobj_extract(&tmpobj, &secdata, &seclen)) {
935                 CERROR("can't extract token\n");
936                 GOTO(out_rsikey, rc = SVC_DROP);
937         }
938         if (rawobj_dup(&rsikey->in_token, &tmpobj)) {
939                 CERROR("can't duplicate token\n");
940                 GOTO(out_rsikey, rc = SVC_DROP);
941         }
942
943         rsikey->naltype = (__u32) req->rq_peer.peer_ni->pni_number;
944         rsikey->netid = 0;
945         rsikey->nid = (__u64) req->rq_peer.peer_id.nid;
946
947         rsip = gssd_upcall(rsikey, &my_chandle);
948         if (!rsip) {
949                 CERROR("error in gssd_upcall.\n");
950
951                 rc = SVC_COMPLETE;
952                 if (gss_pack_err_notify(req, GSS_S_FAILURE, 0))
953                         rc = SVC_DROP;
954
955                 GOTO(out_rsikey, rc);
956         }
957
958         rsci = gss_svc_searchbyctx(&rsip->out_handle);
959         if (!rsci) {
960                 CERROR("rsci still not mature yet?\n");
961
962                 rc = SVC_COMPLETE;
963                 if (gss_pack_err_notify(req, GSS_S_FAILURE, 0))
964                         rc = SVC_DROP;
965
966                 GOTO(out_rsip, rc);
967         }
968         CDEBUG(D_SEC, "svcsec create gss context %p(%u@%s)\n",
969                rsci, rsci->cred.vc_uid,
970                portals_nid2str(req->rq_peer.peer_ni->pni_number,
971                                req->rq_peer.peer_id.nid, nidstr));
972
973         svcdata->is_init = 1;
974         svcdata->reserve_len = 6 * 4 +
975                 size_round4(rsip->out_handle.len) +
976                 size_round4(rsip->out_token.len);
977
978         rc = lustre_pack_reply(req, 0, NULL, NULL);
979         if (rc) {
980                 CERROR("failed to pack reply, rc = %d\n", rc);
981                 set_bit(CACHE_NEGATIVE, &rsci->h.flags);
982                 GOTO(out, rc = SVC_DROP);
983         }
984
985         /* header */
986         resp = (__u32 *) req->rq_reply_state->rs_repbuf;
987         *resp++ = cpu_to_le32(PTLRPC_SEC_GSS);
988         *resp++ = cpu_to_le32(PTLRPC_SEC_TYPE_NONE);
989         *resp++ = cpu_to_le32(req->rq_replen);
990         reslenp = resp++;
991
992         resp += req->rq_replen / 4;
993         reslen = svcdata->reserve_len;
994
995         /* gss reply:
996          * status, major, minor, seq, out_handle, out_token
997          */
998         *resp++ = cpu_to_le32(PTLRPCS_OK);
999         *resp++ = cpu_to_le32(rsip->major_status);
1000         *resp++ = cpu_to_le32(rsip->minor_status);
1001         *resp++ = cpu_to_le32(GSS_SEQ_WIN);
1002         reslen -= (4 * 4);
1003         if (rawobj_serialize(&rsip->out_handle,
1004                              &resp, &reslen)) {
1005                 dump_rsi(rsip);
1006                 dump_rsi(rsikey);
1007                 LBUG();
1008         }
1009         if (rawobj_serialize(&rsip->out_token,
1010                              &resp, &reslen)) {
1011                 dump_rsi(rsip);
1012                 dump_rsi(rsikey);
1013                 LBUG();
1014         }
1015         /* the actual sec data length */
1016         *reslenp = cpu_to_le32(svcdata->reserve_len - reslen);
1017
1018         req->rq_reply_state->rs_repdata_len += le32_to_cpu(*reslenp);
1019         CDEBUG(D_SEC, "req %p: msgsize %d, authsize %d, "
1020                "total size %d\n", req, req->rq_replen,
1021                le32_to_cpu(*reslenp),
1022                req->rq_reply_state->rs_repdata_len);
1023
1024         *res = PTLRPCS_OK;
1025
1026         req->rq_auth_uid = rsci->cred.vc_uid;
1027         req->rq_remote_realm = rsci->remote_realm;
1028         req->rq_mapped_uid = rsci->mapped_uid;
1029
1030         /* This is simplified since right now we doesn't support
1031          * INIT_CONTINUE yet.
1032          */
1033         if (gc->gc_proc == RPC_GSS_PROC_INIT) {
1034                 struct ptlrpcs_wire_hdr *hdr;
1035
1036                 hdr = buf_to_sec_hdr(req->rq_reqbuf);
1037                 req->rq_reqmsg = buf_to_lustre_msg(req->rq_reqbuf);
1038                 req->rq_reqlen = hdr->msg_len;
1039
1040                 rc = SVC_LOGIN;
1041         } else
1042                 rc = SVC_COMPLETE;
1043
1044 out:
1045         rsc_put(&rsci->h, &rsc_cache);
1046 out_rsip:
1047         rsi_put(&rsip->h, &rsi_cache);
1048 out_rsikey:
1049         rsi_put(&rsikey->h, &rsi_cache);
1050
1051         RETURN(rc);
1052 }
1053
1054 static int
1055 gss_svcsec_handle_data(struct ptlrpc_request *req,
1056                        struct rpc_gss_wire_cred *gc,
1057                        __u32 *secdata, __u32 seclen,
1058                        enum ptlrpcs_error *res)
1059 {
1060         struct rsc          *rsci;
1061         char                 nidstr[PTL_NALFMT_SIZE];
1062         __u32                major;
1063         int                  rc;
1064         ENTRY;
1065
1066         *res = PTLRPCS_GSS_CREDPROBLEM;
1067
1068         rsci = gss_svc_searchbyctx(&gc->gc_ctx);
1069         if (!rsci) {
1070                 CWARN("Invalid gss context handle from %s\n",
1071                        portals_nid2str(req->rq_peer.peer_ni->pni_number,
1072                                        req->rq_peer.peer_id.nid, nidstr));
1073                 major = GSS_S_NO_CONTEXT;
1074                 goto notify_err;
1075         }
1076
1077         switch (gc->gc_svc) {
1078         case PTLRPC_GSS_SVC_INTEGRITY:
1079                 major = gss_svc_verify_request(req, rsci, gc, secdata, seclen);
1080                 if (major == GSS_S_COMPLETE)
1081                         break;
1082
1083                 CWARN("fail in verify:0x%x: ctx %p@%s\n", major, rsci,
1084                        portals_nid2str(req->rq_peer.peer_ni->pni_number,
1085                                        req->rq_peer.peer_id.nid, nidstr));
1086                 goto notify_err;
1087         case PTLRPC_GSS_SVC_PRIVACY:
1088                 major = gss_svc_unseal_request(req, rsci, gc, secdata, seclen);
1089                 if (major == GSS_S_COMPLETE)
1090                         break;
1091
1092                 CWARN("fail in decrypt:0x%x: ctx %p@%s\n", major, rsci,
1093                        portals_nid2str(req->rq_peer.peer_ni->pni_number,
1094                                        req->rq_peer.peer_id.nid, nidstr));
1095                 goto notify_err;
1096         default:
1097                 CERROR("unsupported gss service %d\n", gc->gc_svc);
1098                 GOTO(out, rc = SVC_DROP);
1099         }
1100
1101         req->rq_auth_uid = rsci->cred.vc_uid;
1102         req->rq_remote_realm = rsci->remote_realm;
1103         req->rq_mapped_uid = rsci->mapped_uid;
1104
1105         *res = PTLRPCS_OK;
1106         GOTO(out, rc = SVC_OK);
1107
1108 notify_err:
1109         if (gss_pack_err_notify(req, major, 0))
1110                 rc = SVC_DROP;
1111         else
1112                 rc = SVC_COMPLETE;
1113 out:
1114         if (rsci)
1115                 rsc_put(&rsci->h, &rsc_cache);
1116         RETURN(rc);
1117 }
1118
1119 static int
1120 gss_svcsec_handle_destroy(struct ptlrpc_request *req,
1121                           struct rpc_gss_wire_cred *gc,
1122                           __u32 *secdata, __u32 seclen,
1123                           enum ptlrpcs_error *res)
1124 {
1125         struct gss_svc_data *svcdata = req->rq_sec_svcdata;
1126         struct rsc          *rsci;
1127         char                 nidstr[PTL_NALFMT_SIZE];
1128         int                  rc;
1129         ENTRY;
1130
1131         LASSERT(svcdata);
1132         *res = PTLRPCS_GSS_CREDPROBLEM;
1133
1134         rsci = gss_svc_searchbyctx(&gc->gc_ctx);
1135         if (!rsci) {
1136                 CWARN("invalid gss context handle for destroy.\n");
1137                 RETURN(SVC_DROP);
1138         }
1139
1140         if (gc->gc_svc != PTLRPC_GSS_SVC_INTEGRITY) {
1141                 CERROR("service %d is not supported in destroy.\n",
1142                         gc->gc_svc);
1143                 GOTO(out, rc = SVC_DROP);
1144         }
1145
1146         *res = gss_svc_verify_request(req, rsci, gc, secdata, seclen);
1147         if (*res)
1148                 GOTO(out, rc = SVC_DROP);
1149
1150         /* compose reply, which is actually nothing */
1151         svcdata->is_fini = 1;
1152         if (lustre_pack_reply(req, 0, NULL, NULL))
1153                 GOTO(out, rc = SVC_DROP);
1154
1155         CDEBUG(D_SEC, "svcsec destroy gss context %p(%u@%s)\n",
1156                rsci, rsci->cred.vc_uid,
1157                portals_nid2str(req->rq_peer.peer_ni->pni_number,
1158                                req->rq_peer.peer_id.nid, nidstr));
1159
1160         set_bit(CACHE_NEGATIVE, &rsci->h.flags);
1161         *res = PTLRPCS_OK;
1162         rc = SVC_LOGOUT;
1163 out:
1164         rsc_put(&rsci->h, &rsc_cache);
1165         RETURN(rc);
1166 }
1167
1168 /*
1169  * let incomming request go through security check:
1170  *  o context establishment: invoke user space helper
1171  *  o data exchange: verify/decrypt
1172  *  o context destruction: mark context invalid
1173  *
1174  * in most cases, error will result to drop the packet silently.
1175  */
1176 static int
1177 gss_svcsec_accept(struct ptlrpc_request *req, enum ptlrpcs_error *res)
1178 {
1179         struct gss_svc_data *svcdata;
1180         struct rpc_gss_wire_cred *gc;
1181         struct ptlrpcs_wire_hdr *sec_hdr;
1182         __u32 seclen, *secdata, version;
1183         int rc;
1184         ENTRY;
1185
1186         CDEBUG(D_SEC, "request %p\n", req);
1187         LASSERT(req->rq_reqbuf);
1188         LASSERT(req->rq_reqbuf_len);
1189
1190         *res = PTLRPCS_BADCRED;
1191
1192         sec_hdr = buf_to_sec_hdr(req->rq_reqbuf);
1193         LASSERT(sec_hdr->flavor == PTLRPC_SEC_GSS);
1194
1195         seclen = req->rq_reqbuf_len - sizeof(*sec_hdr) - sec_hdr->msg_len;
1196         secdata = (__u32 *) buf_to_sec_data(req->rq_reqbuf);
1197
1198         if (sec_hdr->sec_len > seclen) {
1199                 CERROR("seclen %d, while max buf %d\n",
1200                         sec_hdr->sec_len, seclen);
1201                 RETURN(SVC_DROP);
1202         }
1203
1204         if (seclen < 6 * 4) {
1205                 CERROR("sec size %d too small\n", seclen);
1206                 RETURN(SVC_DROP);
1207         }
1208
1209         LASSERT(!req->rq_sec_svcdata);
1210         OBD_ALLOC(svcdata, sizeof(*svcdata));
1211         if (!svcdata) {
1212                 CERROR("fail to alloc svcdata\n");
1213                 RETURN(SVC_DROP);
1214         }
1215         req->rq_sec_svcdata = svcdata;
1216         gc = &svcdata->clcred;
1217
1218         /* Now secdata/seclen is what we want to parse
1219          */
1220         version = le32_to_cpu(*secdata++);      /* version */
1221         svcdata->subflavor = le32_to_cpu(*secdata++);    /* subflavor */
1222         gc->gc_proc = le32_to_cpu(*secdata++);  /* proc */
1223         gc->gc_seq = le32_to_cpu(*secdata++);   /* seq */
1224         gc->gc_svc = le32_to_cpu(*secdata++);   /* service */
1225         seclen -= 5 * 4;
1226
1227         CDEBUG(D_SEC, "wire gss_hdr: %u/%u/%u/%u/%u\n",
1228                version, svcdata->subflavor, gc->gc_proc,
1229                gc->gc_seq, gc->gc_svc);
1230
1231         if (version != PTLRPC_SEC_GSS_VERSION) {
1232                 CERROR("gss version mismatch: %d - %d\n",
1233                         version, PTLRPC_SEC_GSS_VERSION);
1234                 GOTO(err_free, rc = SVC_DROP);
1235         }
1236
1237         if (rawobj_extract(&gc->gc_ctx, &secdata, &seclen)) {
1238                 CERROR("fail to obtain gss context handle\n");
1239                 GOTO(err_free, rc = SVC_DROP);
1240         }
1241
1242         *res = PTLRPCS_BADVERF;
1243         switch(gc->gc_proc) {
1244         case RPC_GSS_PROC_INIT:
1245         case RPC_GSS_PROC_CONTINUE_INIT:
1246                 rc = gss_svcsec_handle_init(req, gc, secdata, seclen, res);
1247                 break;
1248         case RPC_GSS_PROC_DATA:
1249                 rc = gss_svcsec_handle_data(req, gc, secdata, seclen, res);
1250                 break;
1251         case RPC_GSS_PROC_DESTROY:
1252                 rc = gss_svcsec_handle_destroy(req, gc, secdata, seclen, res);
1253                 break;
1254         default:
1255                 rc = SVC_DROP;
1256                 LBUG();
1257         }
1258
1259 err_free:
1260         if (rc == SVC_DROP && req->rq_sec_svcdata) {
1261                 OBD_FREE(req->rq_sec_svcdata, sizeof(struct gss_svc_data));
1262                 req->rq_sec_svcdata = NULL;
1263         }
1264
1265         RETURN(rc);
1266 }
1267
1268 static int
1269 gss_svcsec_authorize(struct ptlrpc_request *req)
1270 {
1271         struct ptlrpc_reply_state *rs = req->rq_reply_state;
1272         struct gss_svc_data *gsd = (struct gss_svc_data *)req->rq_sec_svcdata;
1273         struct rpc_gss_wire_cred  *gc = &gsd->clcred;
1274         struct rsc                *rscp;
1275         struct ptlrpcs_wire_hdr   *sec_hdr;
1276         rawobj_buf_t               msg_buf;
1277         rawobj_t                   cipher_buf;
1278         __u32                     *vp, *vpsave, major, vlen, seclen;
1279         rawobj_t                   lmsg, mic;
1280         int                        ret;
1281         ENTRY;
1282
1283         LASSERT(rs);
1284         LASSERT(rs->rs_repbuf);
1285         LASSERT(gsd);
1286
1287         if (gsd->is_init || gsd->is_init_continue ||
1288             gsd->is_err_notify || gsd->is_fini) {
1289                 /* nothing to do in these cases */
1290                 CDEBUG(D_SEC, "req %p: init/fini/err\n", req);
1291                 RETURN(0);
1292         }
1293
1294         if (gc->gc_proc != RPC_GSS_PROC_DATA) {
1295                 CERROR("proc %d not support\n", gc->gc_proc);
1296                 RETURN(-EINVAL);
1297         }
1298
1299         rscp = gss_svc_searchbyctx(&gc->gc_ctx);
1300         if (!rscp) {
1301                 CERROR("ctx disapeared under us?\n");
1302                 RETURN(-EINVAL);
1303         }
1304
1305         sec_hdr = (struct ptlrpcs_wire_hdr *) rs->rs_repbuf;
1306         switch (gc->gc_svc) {
1307         case  PTLRPC_GSS_SVC_INTEGRITY:
1308                 /* prepare various pointers */
1309                 lmsg.len = req->rq_replen;
1310                 lmsg.data = (__u8 *) (rs->rs_repbuf + sizeof(*sec_hdr));
1311                 vp = (__u32 *) (lmsg.data + lmsg.len);
1312                 vlen = rs->rs_repbuf_len - sizeof(*sec_hdr) - lmsg.len;
1313                 seclen = vlen;
1314
1315                 sec_hdr->flavor = cpu_to_le32(PTLRPC_SEC_GSS);
1316                 sec_hdr->sectype = cpu_to_le32(PTLRPC_SEC_TYPE_AUTH);
1317                 sec_hdr->msg_len = cpu_to_le32(req->rq_replen);
1318
1319                 /* standard gss hdr */
1320                 LASSERT(vlen >= 7 * 4);
1321                 *vp++ = cpu_to_le32(PTLRPC_SEC_GSS_VERSION);
1322                 *vp++ = cpu_to_le32(PTLRPC_SEC_GSS_KRB5I);
1323                 *vp++ = cpu_to_le32(RPC_GSS_PROC_DATA);
1324                 *vp++ = cpu_to_le32(gc->gc_seq);
1325                 *vp++ = cpu_to_le32(PTLRPC_GSS_SVC_INTEGRITY);
1326                 *vp++ = 0;      /* fake ctx handle */
1327                 vpsave = vp++;  /* reserve size */
1328                 vlen -= 7 * 4;
1329
1330                 mic.len = vlen;
1331                 mic.data = (unsigned char *)vp;
1332
1333                 major = kgss_get_mic(rscp->mechctx, 0, &lmsg, &mic);
1334                 if (major) {
1335                         CERROR("fail to get MIC: 0x%x\n", major);
1336                         GOTO(out, ret = -EINVAL);
1337                 }
1338                 *vpsave = cpu_to_le32(mic.len);
1339                 seclen = seclen - vlen + mic.len;
1340                 sec_hdr->sec_len = cpu_to_le32(seclen);
1341                 rs->rs_repdata_len += size_round(seclen);
1342                 break;
1343         case  PTLRPC_GSS_SVC_PRIVACY:
1344                 vp = (__u32 *) (rs->rs_repbuf + sizeof(*sec_hdr));
1345                 vlen = rs->rs_repbuf_len - sizeof(*sec_hdr);
1346                 seclen = vlen;
1347
1348                 sec_hdr->flavor = cpu_to_le32(PTLRPC_SEC_GSS);
1349                 sec_hdr->sectype = cpu_to_le32(PTLRPC_SEC_TYPE_PRIV);
1350                 sec_hdr->msg_len = cpu_to_le32(0);
1351
1352                 /* standard gss hdr */
1353                 LASSERT(vlen >= 7 * 4);
1354                 *vp++ = cpu_to_le32(PTLRPC_SEC_GSS_VERSION);
1355                 *vp++ = cpu_to_le32(PTLRPC_SEC_GSS_KRB5I);
1356                 *vp++ = cpu_to_le32(RPC_GSS_PROC_DATA);
1357                 *vp++ = cpu_to_le32(gc->gc_seq);
1358                 *vp++ = cpu_to_le32(PTLRPC_GSS_SVC_PRIVACY);
1359                 *vp++ = 0;      /* fake ctx handle */
1360                 vpsave = vp++;  /* reserve size */
1361                 vlen -= 7 * 4;
1362
1363                 msg_buf.buf = (__u8 *) rs->rs_msg - GSS_PRIVBUF_PREFIX_LEN;
1364                 msg_buf.buflen = req->rq_replen + GSS_PRIVBUF_PREFIX_LEN +
1365                                  GSS_PRIVBUF_SUFFIX_LEN;
1366                 msg_buf.dataoff = GSS_PRIVBUF_PREFIX_LEN;
1367                 msg_buf.datalen = req->rq_replen;
1368
1369                 cipher_buf.data = (__u8 *) vp;
1370                 cipher_buf.len = vlen;
1371
1372                 major = kgss_wrap(rscp->mechctx, GSS_C_QOP_DEFAULT,
1373                                 &msg_buf, &cipher_buf);
1374                 if (major) {
1375                         CERROR("failed to wrap: 0x%x\n", major);
1376                         GOTO(out, ret = -EINVAL);
1377                 }
1378
1379                 *vpsave = cpu_to_le32(cipher_buf.len);
1380                 seclen = seclen - vlen + cipher_buf.len;
1381                 sec_hdr->sec_len = cpu_to_le32(seclen);
1382                 rs->rs_repdata_len += size_round(seclen);
1383                 break;
1384         default:
1385                 CERROR("Unknown service %d\n", gc->gc_svc);
1386                 GOTO(out, ret = -EINVAL);
1387         }
1388         ret = 0;
1389 out:
1390         rsc_put(&rscp->h, &rsc_cache);
1391
1392         RETURN(ret);
1393 }
1394
1395 static
1396 void gss_svcsec_cleanup_req(struct ptlrpc_svcsec *svcsec,
1397                             struct ptlrpc_request *req)
1398 {
1399         struct gss_svc_data *gsd = (struct gss_svc_data *) req->rq_sec_svcdata;
1400
1401         if (!gsd) {
1402                 CDEBUG(D_SEC, "no svc_data present. do nothing\n");
1403                 return;
1404         }
1405
1406         /* gsd->clclred.gc_ctx is NOT allocated, just set pointer
1407          * to the incoming packet buffer, so don't need free it
1408          */
1409         OBD_FREE(gsd, sizeof(*gsd));
1410         req->rq_sec_svcdata = NULL;
1411         return;
1412 }
1413
1414 static
1415 int gss_svcsec_est_payload(struct ptlrpc_svcsec *svcsec,
1416                            struct ptlrpc_request *req,
1417                            int msgsize)
1418 {
1419         struct gss_svc_data *svcdata = req->rq_sec_svcdata;
1420         ENTRY;
1421
1422         /* just return the pre-set reserve_len for init/fini/err cases.
1423          */
1424         LASSERT(svcdata);
1425         if (svcdata->is_init) {
1426                 CDEBUG(D_SEC, "is_init, reserver size %d(%d)\n",
1427                        size_round(svcdata->reserve_len),
1428                        svcdata->reserve_len);
1429                 LASSERT(svcdata->reserve_len);
1430                 LASSERT(svcdata->reserve_len % 4 == 0);
1431                 RETURN(size_round(svcdata->reserve_len));
1432         } else if (svcdata->is_err_notify) {
1433                 CDEBUG(D_SEC, "is_err_notify, reserver size %d(%d)\n",
1434                        size_round(svcdata->reserve_len),
1435                        svcdata->reserve_len);
1436                 RETURN(size_round(svcdata->reserve_len));
1437         } else if (svcdata->is_fini) {
1438                 CDEBUG(D_SEC, "is_fini, reserver size 0\n");
1439                 RETURN(0);
1440         } else {
1441                 if (svcdata->clcred.gc_svc == PTLRPC_GSS_SVC_NONE ||
1442                     svcdata->clcred.gc_svc == PTLRPC_GSS_SVC_INTEGRITY)
1443                         RETURN(size_round(GSS_MAX_AUTH_PAYLOAD));
1444                 else if (svcdata->clcred.gc_svc == PTLRPC_GSS_SVC_PRIVACY)
1445                         RETURN(size_round16(GSS_MAX_AUTH_PAYLOAD + msgsize +
1446                                             GSS_PRIVBUF_PREFIX_LEN +
1447                                             GSS_PRIVBUF_SUFFIX_LEN));
1448                 else {
1449                         CERROR("unknown gss svc %u\n", svcdata->clcred.gc_svc);
1450                         *((int *)0) = 0;
1451                         LBUG();
1452                 }
1453         }
1454         RETURN(0);
1455 }
1456
1457 int gss_svcsec_alloc_repbuf(struct ptlrpc_svcsec *svcsec,
1458                             struct ptlrpc_request *req,
1459                             int msgsize)
1460 {
1461         struct gss_svc_data *gsd = (struct gss_svc_data *) req->rq_sec_svcdata;
1462         struct ptlrpc_reply_state *rs;
1463         int msg_payload, sec_payload;
1464         int privacy, rc;
1465         ENTRY;
1466
1467         /* determine the security type: none/auth or priv, we have
1468          * different pack scheme for them.
1469          * init/fini/err will always be treated as none/auth.
1470          */
1471         LASSERT(gsd);
1472         if (!gsd->is_init && !gsd->is_init_continue &&
1473             !gsd->is_fini && !gsd->is_err_notify &&
1474             gsd->clcred.gc_svc == PTLRPC_GSS_SVC_PRIVACY)
1475                 privacy = 1;
1476         else
1477                 privacy = 0;
1478
1479         msg_payload = privacy ? 0 : msgsize;
1480         sec_payload = gss_svcsec_est_payload(svcsec, req, msgsize);
1481
1482         rc = svcsec_alloc_reply_state(req, msg_payload, sec_payload);
1483         if (rc)
1484                 RETURN(rc);
1485
1486         rs = req->rq_reply_state;
1487         LASSERT(rs);
1488         rs->rs_msg_len = msgsize;
1489
1490         if (privacy) {
1491                 /* we can choose to let msg simply point to the rear of the
1492                  * buffer, which lead to buffer overlap when doing encryption.
1493                  * usually it's ok and it indeed passed all existing tests.
1494                  * but not sure if there will be subtle problems in the future.
1495                  * so right now we choose to alloc another new buffer. we'll
1496                  * see how it works.
1497                  */
1498 #if 0
1499                 rs->rs_msg = (struct lustre_msg *)
1500                              (rs->rs_repbuf + rs->rs_repbuf_len -
1501                               msgsize - GSS_PRIVBUF_SUFFIX_LEN);
1502 #endif
1503                 char *msgbuf;
1504
1505                 msgsize += GSS_PRIVBUF_PREFIX_LEN + GSS_PRIVBUF_SUFFIX_LEN;
1506                 OBD_ALLOC(msgbuf, msgsize);
1507                 if (!msgbuf) {
1508                         CERROR("can't alloc %d\n", msgsize);
1509                         svcsec_free_reply_state(rs);
1510                         req->rq_reply_state = NULL;
1511                         RETURN(-ENOMEM);
1512                 }
1513                 rs->rs_msg = (struct lustre_msg *)
1514                                 (msgbuf + GSS_PRIVBUF_PREFIX_LEN);
1515         }
1516
1517         req->rq_repmsg = rs->rs_msg;
1518
1519         RETURN(0);
1520 }
1521
1522 static
1523 void gss_svcsec_free_repbuf(struct ptlrpc_svcsec *svcsec,
1524                             struct ptlrpc_reply_state *rs)
1525 {
1526         unsigned long p1 = (unsigned long) rs->rs_msg;
1527         unsigned long p2 = (unsigned long) rs->rs_buf;
1528
1529         LASSERT(rs->rs_buf);
1530         LASSERT(rs->rs_msg);
1531         LASSERT(rs->rs_msg_len);
1532
1533         if (p1 < p2 || p1 >= p2 + rs->rs_buf_len) {
1534                 char *start = (char*) rs->rs_msg - GSS_PRIVBUF_PREFIX_LEN;
1535                 int size = rs->rs_msg_len + GSS_PRIVBUF_PREFIX_LEN +
1536                            GSS_PRIVBUF_SUFFIX_LEN;
1537                 OBD_FREE(start, size);
1538         }
1539
1540         svcsec_free_reply_state(rs);
1541 }
1542
1543 struct ptlrpc_svcsec svcsec_gss = {
1544         .pss_owner              = THIS_MODULE,
1545         .pss_name               = "GSS_SVCSEC",
1546         .pss_flavor             = {PTLRPC_SEC_GSS, 0},
1547         .accept                 = gss_svcsec_accept,
1548         .authorize              = gss_svcsec_authorize,
1549         .alloc_repbuf           = gss_svcsec_alloc_repbuf,
1550         .free_repbuf            = gss_svcsec_free_repbuf,
1551         .cleanup_req            = gss_svcsec_cleanup_req,
1552 };
1553
1554 /* XXX hacking */
1555 void lgss_svc_cache_purge_all(void)
1556 {
1557         cache_purge(&rsi_cache);
1558         cache_purge(&rsc_cache);
1559 }
1560 EXPORT_SYMBOL(lgss_svc_cache_purge_all);
1561
1562 void lgss_svc_cache_flush(__u32 uid)
1563 {
1564         rsc_flush(uid);
1565 }
1566 EXPORT_SYMBOL(lgss_svc_cache_flush);
1567
1568 int gss_svc_init(void)
1569 {
1570         int rc;
1571
1572         rc = svcsec_register(&svcsec_gss);
1573         if (!rc) {
1574                 cache_register(&rsc_cache);
1575                 cache_register(&rsi_cache);
1576         }
1577         return rc;
1578 }
1579
1580 void gss_svc_exit(void)
1581 {
1582         int rc;
1583         if ((rc = cache_unregister(&rsi_cache)))
1584                 CERROR("unregister rsi cache: %d\n", rc);
1585         if ((rc = cache_unregister(&rsc_cache)))
1586                 CERROR("unregister rsc cache: %d\n", rc);
1587         if ((rc = svcsec_unregister(&svcsec_gss)))
1588                 CERROR("unregister svcsec_gss: %d\n", rc);
1589 }