nfsd: pass client principal name in rsc downcall
[linux-2.6] / net / sunrpc / auth_gss / svcauth_gss.c
1 /*
2  * Neil Brown <neilb@cse.unsw.edu.au>
3  * J. Bruce Fields <bfields@umich.edu>
4  * Andy Adamson <andros@umich.edu>
5  * Dug Song <dugsong@monkey.org>
6  *
7  * RPCSEC_GSS server authentication.
8  * This implements RPCSEC_GSS as defined in rfc2203 (rpcsec_gss) and rfc2078
9  * (gssapi)
10  *
11  * The RPCSEC_GSS involves three stages:
12  *  1/ context creation
13  *  2/ data exchange
14  *  3/ context destruction
15  *
16  * Context creation is handled largely by upcalls to user-space.
17  *  In particular, GSS_Accept_sec_context is handled by an upcall
18  * Data exchange is handled entirely within the kernel
19  *  In particular, GSS_GetMIC, GSS_VerifyMIC, GSS_Seal, GSS_Unseal are in-kernel.
20  * Context destruction is handled in-kernel
21  *  GSS_Delete_sec_context is in-kernel
22  *
23  * Context creation is initiated by a RPCSEC_GSS_INIT request arriving.
24  * The context handle and gss_token are used as a key into the rpcsec_init cache.
25  * The content of this cache includes some of the outputs of GSS_Accept_sec_context,
26  * being major_status, minor_status, context_handle, reply_token.
27  * These are sent back to the client.
28  * Sequence window management is handled by the kernel.  The window size if currently
29  * a compile time constant.
30  *
31  * When user-space is happy that a context is established, it places an entry
32  * in the rpcsec_context cache. The key for this cache is the context_handle.
33  * The content includes:
34  *   uid/gidlist - for determining access rights
35  *   mechanism type
36  *   mechanism specific information, such as a key
37  *
38  */
39
40 #include <linux/types.h>
41 #include <linux/module.h>
42 #include <linux/pagemap.h>
43
44 #include <linux/sunrpc/auth_gss.h>
45 #include <linux/sunrpc/gss_err.h>
46 #include <linux/sunrpc/svcauth.h>
47 #include <linux/sunrpc/svcauth_gss.h>
48 #include <linux/sunrpc/cache.h>
49
50 #ifdef RPC_DEBUG
51 # define RPCDBG_FACILITY        RPCDBG_AUTH
52 #endif
53
54 /* The rpcsec_init cache is used for mapping RPCSEC_GSS_{,CONT_}INIT requests
55  * into replies.
56  *
57  * Key is context handle (\x if empty) and gss_token.
58  * Content is major_status minor_status (integers) context_handle, reply_token.
59  *
60  */
61
62 static int netobj_equal(struct xdr_netobj *a, struct xdr_netobj *b)
63 {
64         return a->len == b->len && 0 == memcmp(a->data, b->data, a->len);
65 }
66
67 #define RSI_HASHBITS    6
68 #define RSI_HASHMAX     (1<<RSI_HASHBITS)
69 #define RSI_HASHMASK    (RSI_HASHMAX-1)
70
71 struct rsi {
72         struct cache_head       h;
73         struct xdr_netobj       in_handle, in_token;
74         struct xdr_netobj       out_handle, out_token;
75         int                     major_status, minor_status;
76 };
77
78 static struct cache_head *rsi_table[RSI_HASHMAX];
79 static struct cache_detail rsi_cache;
80 static struct rsi *rsi_update(struct rsi *new, struct rsi *old);
81 static struct rsi *rsi_lookup(struct rsi *item);
82
83 static void rsi_free(struct rsi *rsii)
84 {
85         kfree(rsii->in_handle.data);
86         kfree(rsii->in_token.data);
87         kfree(rsii->out_handle.data);
88         kfree(rsii->out_token.data);
89 }
90
91 static void rsi_put(struct kref *ref)
92 {
93         struct rsi *rsii = container_of(ref, struct rsi, h.ref);
94         rsi_free(rsii);
95         kfree(rsii);
96 }
97
98 static inline int rsi_hash(struct rsi *item)
99 {
100         return hash_mem(item->in_handle.data, item->in_handle.len, RSI_HASHBITS)
101              ^ hash_mem(item->in_token.data, item->in_token.len, RSI_HASHBITS);
102 }
103
104 static int rsi_match(struct cache_head *a, struct cache_head *b)
105 {
106         struct rsi *item = container_of(a, struct rsi, h);
107         struct rsi *tmp = container_of(b, struct rsi, h);
108         return netobj_equal(&item->in_handle, &tmp->in_handle)
109                 && netobj_equal(&item->in_token, &tmp->in_token);
110 }
111
112 static int dup_to_netobj(struct xdr_netobj *dst, char *src, int len)
113 {
114         dst->len = len;
115         dst->data = (len ? kmemdup(src, len, GFP_KERNEL) : NULL);
116         if (len && !dst->data)
117                 return -ENOMEM;
118         return 0;
119 }
120
121 static inline int dup_netobj(struct xdr_netobj *dst, struct xdr_netobj *src)
122 {
123         return dup_to_netobj(dst, src->data, src->len);
124 }
125
126 static void rsi_init(struct cache_head *cnew, struct cache_head *citem)
127 {
128         struct rsi *new = container_of(cnew, struct rsi, h);
129         struct rsi *item = container_of(citem, struct rsi, h);
130
131         new->out_handle.data = NULL;
132         new->out_handle.len = 0;
133         new->out_token.data = NULL;
134         new->out_token.len = 0;
135         new->in_handle.len = item->in_handle.len;
136         item->in_handle.len = 0;
137         new->in_token.len = item->in_token.len;
138         item->in_token.len = 0;
139         new->in_handle.data = item->in_handle.data;
140         item->in_handle.data = NULL;
141         new->in_token.data = item->in_token.data;
142         item->in_token.data = NULL;
143 }
144
145 static void update_rsi(struct cache_head *cnew, struct cache_head *citem)
146 {
147         struct rsi *new = container_of(cnew, struct rsi, h);
148         struct rsi *item = container_of(citem, struct rsi, h);
149
150         BUG_ON(new->out_handle.data || new->out_token.data);
151         new->out_handle.len = item->out_handle.len;
152         item->out_handle.len = 0;
153         new->out_token.len = item->out_token.len;
154         item->out_token.len = 0;
155         new->out_handle.data = item->out_handle.data;
156         item->out_handle.data = NULL;
157         new->out_token.data = item->out_token.data;
158         item->out_token.data = NULL;
159
160         new->major_status = item->major_status;
161         new->minor_status = item->minor_status;
162 }
163
164 static struct cache_head *rsi_alloc(void)
165 {
166         struct rsi *rsii = kmalloc(sizeof(*rsii), GFP_KERNEL);
167         if (rsii)
168                 return &rsii->h;
169         else
170                 return NULL;
171 }
172
173 static void rsi_request(struct cache_detail *cd,
174                        struct cache_head *h,
175                        char **bpp, int *blen)
176 {
177         struct rsi *rsii = container_of(h, struct rsi, h);
178
179         qword_addhex(bpp, blen, rsii->in_handle.data, rsii->in_handle.len);
180         qword_addhex(bpp, blen, rsii->in_token.data, rsii->in_token.len);
181         (*bpp)[-1] = '\n';
182 }
183
184
185 static int rsi_parse(struct cache_detail *cd,
186                     char *mesg, int mlen)
187 {
188         /* context token expiry major minor context token */
189         char *buf = mesg;
190         char *ep;
191         int len;
192         struct rsi rsii, *rsip = NULL;
193         time_t expiry;
194         int status = -EINVAL;
195
196         memset(&rsii, 0, sizeof(rsii));
197         /* handle */
198         len = qword_get(&mesg, buf, mlen);
199         if (len < 0)
200                 goto out;
201         status = -ENOMEM;
202         if (dup_to_netobj(&rsii.in_handle, buf, len))
203                 goto out;
204
205         /* token */
206         len = qword_get(&mesg, buf, mlen);
207         status = -EINVAL;
208         if (len < 0)
209                 goto out;
210         status = -ENOMEM;
211         if (dup_to_netobj(&rsii.in_token, buf, len))
212                 goto out;
213
214         rsip = rsi_lookup(&rsii);
215         if (!rsip)
216                 goto out;
217
218         rsii.h.flags = 0;
219         /* expiry */
220         expiry = get_expiry(&mesg);
221         status = -EINVAL;
222         if (expiry == 0)
223                 goto out;
224
225         /* major/minor */
226         len = qword_get(&mesg, buf, mlen);
227         if (len <= 0)
228                 goto out;
229         rsii.major_status = simple_strtoul(buf, &ep, 10);
230         if (*ep)
231                 goto out;
232         len = qword_get(&mesg, buf, mlen);
233         if (len <= 0)
234                 goto out;
235         rsii.minor_status = simple_strtoul(buf, &ep, 10);
236         if (*ep)
237                 goto out;
238
239         /* out_handle */
240         len = qword_get(&mesg, buf, mlen);
241         if (len < 0)
242                 goto out;
243         status = -ENOMEM;
244         if (dup_to_netobj(&rsii.out_handle, buf, len))
245                 goto out;
246
247         /* out_token */
248         len = qword_get(&mesg, buf, mlen);
249         status = -EINVAL;
250         if (len < 0)
251                 goto out;
252         status = -ENOMEM;
253         if (dup_to_netobj(&rsii.out_token, buf, len))
254                 goto out;
255         rsii.h.expiry_time = expiry;
256         rsip = rsi_update(&rsii, rsip);
257         status = 0;
258 out:
259         rsi_free(&rsii);
260         if (rsip)
261                 cache_put(&rsip->h, &rsi_cache);
262         else
263                 status = -ENOMEM;
264         return status;
265 }
266
267 static struct cache_detail rsi_cache = {
268         .owner          = THIS_MODULE,
269         .hash_size      = RSI_HASHMAX,
270         .hash_table     = rsi_table,
271         .name           = "auth.rpcsec.init",
272         .cache_put      = rsi_put,
273         .cache_request  = rsi_request,
274         .cache_parse    = rsi_parse,
275         .match          = rsi_match,
276         .init           = rsi_init,
277         .update         = update_rsi,
278         .alloc          = rsi_alloc,
279 };
280
281 static struct rsi *rsi_lookup(struct rsi *item)
282 {
283         struct cache_head *ch;
284         int hash = rsi_hash(item);
285
286         ch = sunrpc_cache_lookup(&rsi_cache, &item->h, hash);
287         if (ch)
288                 return container_of(ch, struct rsi, h);
289         else
290                 return NULL;
291 }
292
293 static struct rsi *rsi_update(struct rsi *new, struct rsi *old)
294 {
295         struct cache_head *ch;
296         int hash = rsi_hash(new);
297
298         ch = sunrpc_cache_update(&rsi_cache, &new->h,
299                                  &old->h, hash);
300         if (ch)
301                 return container_of(ch, struct rsi, h);
302         else
303                 return NULL;
304 }
305
306
307 /*
308  * The rpcsec_context cache is used to store a context that is
309  * used in data exchange.
310  * The key is a context handle. The content is:
311  *  uid, gidlist, mechanism, service-set, mech-specific-data
312  */
313
314 #define RSC_HASHBITS    10
315 #define RSC_HASHMAX     (1<<RSC_HASHBITS)
316 #define RSC_HASHMASK    (RSC_HASHMAX-1)
317
318 #define GSS_SEQ_WIN     128
319
320 struct gss_svc_seq_data {
321         /* highest seq number seen so far: */
322         int                     sd_max;
323         /* for i such that sd_max-GSS_SEQ_WIN < i <= sd_max, the i-th bit of
324          * sd_win is nonzero iff sequence number i has been seen already: */
325         unsigned long           sd_win[GSS_SEQ_WIN/BITS_PER_LONG];
326         spinlock_t              sd_lock;
327 };
328
329 struct rsc {
330         struct cache_head       h;
331         struct xdr_netobj       handle;
332         struct svc_cred         cred;
333         struct gss_svc_seq_data seqdata;
334         struct gss_ctx          *mechctx;
335         char                    *client_name;
336 };
337
338 static struct cache_head *rsc_table[RSC_HASHMAX];
339 static struct cache_detail rsc_cache;
340 static struct rsc *rsc_update(struct rsc *new, struct rsc *old);
341 static struct rsc *rsc_lookup(struct rsc *item);
342
343 static void rsc_free(struct rsc *rsci)
344 {
345         kfree(rsci->handle.data);
346         if (rsci->mechctx)
347                 gss_delete_sec_context(&rsci->mechctx);
348         if (rsci->cred.cr_group_info)
349                 put_group_info(rsci->cred.cr_group_info);
350         kfree(rsci->client_name);
351 }
352
353 static void rsc_put(struct kref *ref)
354 {
355         struct rsc *rsci = container_of(ref, struct rsc, h.ref);
356
357         rsc_free(rsci);
358         kfree(rsci);
359 }
360
361 static inline int
362 rsc_hash(struct rsc *rsci)
363 {
364         return hash_mem(rsci->handle.data, rsci->handle.len, RSC_HASHBITS);
365 }
366
367 static int
368 rsc_match(struct cache_head *a, struct cache_head *b)
369 {
370         struct rsc *new = container_of(a, struct rsc, h);
371         struct rsc *tmp = container_of(b, struct rsc, h);
372
373         return netobj_equal(&new->handle, &tmp->handle);
374 }
375
376 static void
377 rsc_init(struct cache_head *cnew, struct cache_head *ctmp)
378 {
379         struct rsc *new = container_of(cnew, struct rsc, h);
380         struct rsc *tmp = container_of(ctmp, struct rsc, h);
381
382         new->handle.len = tmp->handle.len;
383         tmp->handle.len = 0;
384         new->handle.data = tmp->handle.data;
385         tmp->handle.data = NULL;
386         new->mechctx = NULL;
387         new->cred.cr_group_info = NULL;
388         new->client_name = NULL;
389 }
390
391 static void
392 update_rsc(struct cache_head *cnew, struct cache_head *ctmp)
393 {
394         struct rsc *new = container_of(cnew, struct rsc, h);
395         struct rsc *tmp = container_of(ctmp, struct rsc, h);
396
397         new->mechctx = tmp->mechctx;
398         tmp->mechctx = NULL;
399         memset(&new->seqdata, 0, sizeof(new->seqdata));
400         spin_lock_init(&new->seqdata.sd_lock);
401         new->cred = tmp->cred;
402         tmp->cred.cr_group_info = NULL;
403         new->client_name = tmp->client_name;
404         tmp->client_name = NULL;
405 }
406
407 static struct cache_head *
408 rsc_alloc(void)
409 {
410         struct rsc *rsci = kmalloc(sizeof(*rsci), GFP_KERNEL);
411         if (rsci)
412                 return &rsci->h;
413         else
414                 return NULL;
415 }
416
417 static int rsc_parse(struct cache_detail *cd,
418                      char *mesg, int mlen)
419 {
420         /* contexthandle expiry [ uid gid N <n gids> mechname ...mechdata... ] */
421         char *buf = mesg;
422         int len, rv;
423         struct rsc rsci, *rscp = NULL;
424         time_t expiry;
425         int status = -EINVAL;
426         struct gss_api_mech *gm = NULL;
427
428         memset(&rsci, 0, sizeof(rsci));
429         /* context handle */
430         len = qword_get(&mesg, buf, mlen);
431         if (len < 0) goto out;
432         status = -ENOMEM;
433         if (dup_to_netobj(&rsci.handle, buf, len))
434                 goto out;
435
436         rsci.h.flags = 0;
437         /* expiry */
438         expiry = get_expiry(&mesg);
439         status = -EINVAL;
440         if (expiry == 0)
441                 goto out;
442
443         rscp = rsc_lookup(&rsci);
444         if (!rscp)
445                 goto out;
446
447         /* uid, or NEGATIVE */
448         rv = get_int(&mesg, &rsci.cred.cr_uid);
449         if (rv == -EINVAL)
450                 goto out;
451         if (rv == -ENOENT)
452                 set_bit(CACHE_NEGATIVE, &rsci.h.flags);
453         else {
454                 int N, i;
455
456                 /* gid */
457                 if (get_int(&mesg, &rsci.cred.cr_gid))
458                         goto out;
459
460                 /* number of additional gid's */
461                 if (get_int(&mesg, &N))
462                         goto out;
463                 status = -ENOMEM;
464                 rsci.cred.cr_group_info = groups_alloc(N);
465                 if (rsci.cred.cr_group_info == NULL)
466                         goto out;
467
468                 /* gid's */
469                 status = -EINVAL;
470                 for (i=0; i<N; i++) {
471                         gid_t gid;
472                         if (get_int(&mesg, &gid))
473                                 goto out;
474                         GROUP_AT(rsci.cred.cr_group_info, i) = gid;
475                 }
476
477                 /* mech name */
478                 len = qword_get(&mesg, buf, mlen);
479                 if (len < 0)
480                         goto out;
481                 gm = gss_mech_get_by_name(buf);
482                 status = -EOPNOTSUPP;
483                 if (!gm)
484                         goto out;
485
486                 status = -EINVAL;
487                 /* mech-specific data: */
488                 len = qword_get(&mesg, buf, mlen);
489                 if (len < 0)
490                         goto out;
491                 status = gss_import_sec_context(buf, len, gm, &rsci.mechctx);
492                 if (status)
493                         goto out;
494
495                 /* get client name */
496                 len = qword_get(&mesg, buf, mlen);
497                 if (len > 0) {
498                         rsci.client_name = kstrdup(buf, GFP_KERNEL);
499                         if (!rsci.client_name)
500                                 goto out;
501                 }
502
503         }
504         rsci.h.expiry_time = expiry;
505         rscp = rsc_update(&rsci, rscp);
506         status = 0;
507 out:
508         gss_mech_put(gm);
509         rsc_free(&rsci);
510         if (rscp)
511                 cache_put(&rscp->h, &rsc_cache);
512         else
513                 status = -ENOMEM;
514         return status;
515 }
516
517 static struct cache_detail rsc_cache = {
518         .owner          = THIS_MODULE,
519         .hash_size      = RSC_HASHMAX,
520         .hash_table     = rsc_table,
521         .name           = "auth.rpcsec.context",
522         .cache_put      = rsc_put,
523         .cache_parse    = rsc_parse,
524         .match          = rsc_match,
525         .init           = rsc_init,
526         .update         = update_rsc,
527         .alloc          = rsc_alloc,
528 };
529
530 static struct rsc *rsc_lookup(struct rsc *item)
531 {
532         struct cache_head *ch;
533         int hash = rsc_hash(item);
534
535         ch = sunrpc_cache_lookup(&rsc_cache, &item->h, hash);
536         if (ch)
537                 return container_of(ch, struct rsc, h);
538         else
539                 return NULL;
540 }
541
542 static struct rsc *rsc_update(struct rsc *new, struct rsc *old)
543 {
544         struct cache_head *ch;
545         int hash = rsc_hash(new);
546
547         ch = sunrpc_cache_update(&rsc_cache, &new->h,
548                                  &old->h, hash);
549         if (ch)
550                 return container_of(ch, struct rsc, h);
551         else
552                 return NULL;
553 }
554
555
556 static struct rsc *
557 gss_svc_searchbyctx(struct xdr_netobj *handle)
558 {
559         struct rsc rsci;
560         struct rsc *found;
561
562         memset(&rsci, 0, sizeof(rsci));
563         if (dup_to_netobj(&rsci.handle, handle->data, handle->len))
564                 return NULL;
565         found = rsc_lookup(&rsci);
566         rsc_free(&rsci);
567         if (!found)
568                 return NULL;
569         if (cache_check(&rsc_cache, &found->h, NULL))
570                 return NULL;
571         return found;
572 }
573
574 /* Implements sequence number algorithm as specified in RFC 2203. */
575 static int
576 gss_check_seq_num(struct rsc *rsci, int seq_num)
577 {
578         struct gss_svc_seq_data *sd = &rsci->seqdata;
579
580         spin_lock(&sd->sd_lock);
581         if (seq_num > sd->sd_max) {
582                 if (seq_num >= sd->sd_max + GSS_SEQ_WIN) {
583                         memset(sd->sd_win,0,sizeof(sd->sd_win));
584                         sd->sd_max = seq_num;
585                 } else while (sd->sd_max < seq_num) {
586                         sd->sd_max++;
587                         __clear_bit(sd->sd_max % GSS_SEQ_WIN, sd->sd_win);
588                 }
589                 __set_bit(seq_num % GSS_SEQ_WIN, sd->sd_win);
590                 goto ok;
591         } else if (seq_num <= sd->sd_max - GSS_SEQ_WIN) {
592                 goto drop;
593         }
594         /* sd_max - GSS_SEQ_WIN < seq_num <= sd_max */
595         if (__test_and_set_bit(seq_num % GSS_SEQ_WIN, sd->sd_win))
596                 goto drop;
597 ok:
598         spin_unlock(&sd->sd_lock);
599         return 1;
600 drop:
601         spin_unlock(&sd->sd_lock);
602         return 0;
603 }
604
605 static inline u32 round_up_to_quad(u32 i)
606 {
607         return (i + 3 ) & ~3;
608 }
609
610 static inline int
611 svc_safe_getnetobj(struct kvec *argv, struct xdr_netobj *o)
612 {
613         int l;
614
615         if (argv->iov_len < 4)
616                 return -1;
617         o->len = svc_getnl(argv);
618         l = round_up_to_quad(o->len);
619         if (argv->iov_len < l)
620                 return -1;
621         o->data = argv->iov_base;
622         argv->iov_base += l;
623         argv->iov_len -= l;
624         return 0;
625 }
626
627 static inline int
628 svc_safe_putnetobj(struct kvec *resv, struct xdr_netobj *o)
629 {
630         u8 *p;
631
632         if (resv->iov_len + 4 > PAGE_SIZE)
633                 return -1;
634         svc_putnl(resv, o->len);
635         p = resv->iov_base + resv->iov_len;
636         resv->iov_len += round_up_to_quad(o->len);
637         if (resv->iov_len > PAGE_SIZE)
638                 return -1;
639         memcpy(p, o->data, o->len);
640         memset(p + o->len, 0, round_up_to_quad(o->len) - o->len);
641         return 0;
642 }
643
644 /*
645  * Verify the checksum on the header and return SVC_OK on success.
646  * Otherwise, return SVC_DROP (in the case of a bad sequence number)
647  * or return SVC_DENIED and indicate error in authp.
648  */
649 static int
650 gss_verify_header(struct svc_rqst *rqstp, struct rsc *rsci,
651                   __be32 *rpcstart, struct rpc_gss_wire_cred *gc, __be32 *authp)
652 {
653         struct gss_ctx          *ctx_id = rsci->mechctx;
654         struct xdr_buf          rpchdr;
655         struct xdr_netobj       checksum;
656         u32                     flavor = 0;
657         struct kvec             *argv = &rqstp->rq_arg.head[0];
658         struct kvec             iov;
659
660         /* data to compute the checksum over: */
661         iov.iov_base = rpcstart;
662         iov.iov_len = (u8 *)argv->iov_base - (u8 *)rpcstart;
663         xdr_buf_from_iov(&iov, &rpchdr);
664
665         *authp = rpc_autherr_badverf;
666         if (argv->iov_len < 4)
667                 return SVC_DENIED;
668         flavor = svc_getnl(argv);
669         if (flavor != RPC_AUTH_GSS)
670                 return SVC_DENIED;
671         if (svc_safe_getnetobj(argv, &checksum))
672                 return SVC_DENIED;
673
674         if (rqstp->rq_deferred) /* skip verification of revisited request */
675                 return SVC_OK;
676         if (gss_verify_mic(ctx_id, &rpchdr, &checksum) != GSS_S_COMPLETE) {
677                 *authp = rpcsec_gsserr_credproblem;
678                 return SVC_DENIED;
679         }
680
681         if (gc->gc_seq > MAXSEQ) {
682                 dprintk("RPC:       svcauth_gss: discarding request with "
683                                 "large sequence number %d\n", gc->gc_seq);
684                 *authp = rpcsec_gsserr_ctxproblem;
685                 return SVC_DENIED;
686         }
687         if (!gss_check_seq_num(rsci, gc->gc_seq)) {
688                 dprintk("RPC:       svcauth_gss: discarding request with "
689                                 "old sequence number %d\n", gc->gc_seq);
690                 return SVC_DROP;
691         }
692         return SVC_OK;
693 }
694
695 static int
696 gss_write_null_verf(struct svc_rqst *rqstp)
697 {
698         __be32     *p;
699
700         svc_putnl(rqstp->rq_res.head, RPC_AUTH_NULL);
701         p = rqstp->rq_res.head->iov_base + rqstp->rq_res.head->iov_len;
702         /* don't really need to check if head->iov_len > PAGE_SIZE ... */
703         *p++ = 0;
704         if (!xdr_ressize_check(rqstp, p))
705                 return -1;
706         return 0;
707 }
708
709 static int
710 gss_write_verf(struct svc_rqst *rqstp, struct gss_ctx *ctx_id, u32 seq)
711 {
712         __be32                  xdr_seq;
713         u32                     maj_stat;
714         struct xdr_buf          verf_data;
715         struct xdr_netobj       mic;
716         __be32                  *p;
717         struct kvec             iov;
718
719         svc_putnl(rqstp->rq_res.head, RPC_AUTH_GSS);
720         xdr_seq = htonl(seq);
721
722         iov.iov_base = &xdr_seq;
723         iov.iov_len = sizeof(xdr_seq);
724         xdr_buf_from_iov(&iov, &verf_data);
725         p = rqstp->rq_res.head->iov_base + rqstp->rq_res.head->iov_len;
726         mic.data = (u8 *)(p + 1);
727         maj_stat = gss_get_mic(ctx_id, &verf_data, &mic);
728         if (maj_stat != GSS_S_COMPLETE)
729                 return -1;
730         *p++ = htonl(mic.len);
731         memset((u8 *)p + mic.len, 0, round_up_to_quad(mic.len) - mic.len);
732         p += XDR_QUADLEN(mic.len);
733         if (!xdr_ressize_check(rqstp, p))
734                 return -1;
735         return 0;
736 }
737
738 struct gss_domain {
739         struct auth_domain      h;
740         u32                     pseudoflavor;
741 };
742
743 static struct auth_domain *
744 find_gss_auth_domain(struct gss_ctx *ctx, u32 svc)
745 {
746         char *name;
747
748         name = gss_service_to_auth_domain_name(ctx->mech_type, svc);
749         if (!name)
750                 return NULL;
751         return auth_domain_find(name);
752 }
753
754 static struct auth_ops svcauthops_gss;
755
756 u32 svcauth_gss_flavor(struct auth_domain *dom)
757 {
758         struct gss_domain *gd = container_of(dom, struct gss_domain, h);
759
760         return gd->pseudoflavor;
761 }
762
763 EXPORT_SYMBOL_GPL(svcauth_gss_flavor);
764
765 int
766 svcauth_gss_register_pseudoflavor(u32 pseudoflavor, char * name)
767 {
768         struct gss_domain       *new;
769         struct auth_domain      *test;
770         int                     stat = -ENOMEM;
771
772         new = kmalloc(sizeof(*new), GFP_KERNEL);
773         if (!new)
774                 goto out;
775         kref_init(&new->h.ref);
776         new->h.name = kstrdup(name, GFP_KERNEL);
777         if (!new->h.name)
778                 goto out_free_dom;
779         new->h.flavour = &svcauthops_gss;
780         new->pseudoflavor = pseudoflavor;
781
782         stat = 0;
783         test = auth_domain_lookup(name, &new->h);
784         if (test != &new->h) { /* Duplicate registration */
785                 auth_domain_put(test);
786                 kfree(new->h.name);
787                 goto out_free_dom;
788         }
789         return 0;
790
791 out_free_dom:
792         kfree(new);
793 out:
794         return stat;
795 }
796
797 EXPORT_SYMBOL_GPL(svcauth_gss_register_pseudoflavor);
798
799 static inline int
800 read_u32_from_xdr_buf(struct xdr_buf *buf, int base, u32 *obj)
801 {
802         __be32  raw;
803         int     status;
804
805         status = read_bytes_from_xdr_buf(buf, base, &raw, sizeof(*obj));
806         if (status)
807                 return status;
808         *obj = ntohl(raw);
809         return 0;
810 }
811
812 /* It would be nice if this bit of code could be shared with the client.
813  * Obstacles:
814  *      The client shouldn't malloc(), would have to pass in own memory.
815  *      The server uses base of head iovec as read pointer, while the
816  *      client uses separate pointer. */
817 static int
818 unwrap_integ_data(struct xdr_buf *buf, u32 seq, struct gss_ctx *ctx)
819 {
820         int stat = -EINVAL;
821         u32 integ_len, maj_stat;
822         struct xdr_netobj mic;
823         struct xdr_buf integ_buf;
824
825         integ_len = svc_getnl(&buf->head[0]);
826         if (integ_len & 3)
827                 return stat;
828         if (integ_len > buf->len)
829                 return stat;
830         if (xdr_buf_subsegment(buf, &integ_buf, 0, integ_len))
831                 BUG();
832         /* copy out mic... */
833         if (read_u32_from_xdr_buf(buf, integ_len, &mic.len))
834                 BUG();
835         if (mic.len > RPC_MAX_AUTH_SIZE)
836                 return stat;
837         mic.data = kmalloc(mic.len, GFP_KERNEL);
838         if (!mic.data)
839                 return stat;
840         if (read_bytes_from_xdr_buf(buf, integ_len + 4, mic.data, mic.len))
841                 goto out;
842         maj_stat = gss_verify_mic(ctx, &integ_buf, &mic);
843         if (maj_stat != GSS_S_COMPLETE)
844                 goto out;
845         if (svc_getnl(&buf->head[0]) != seq)
846                 goto out;
847         stat = 0;
848 out:
849         kfree(mic.data);
850         return stat;
851 }
852
853 static inline int
854 total_buf_len(struct xdr_buf *buf)
855 {
856         return buf->head[0].iov_len + buf->page_len + buf->tail[0].iov_len;
857 }
858
859 static void
860 fix_priv_head(struct xdr_buf *buf, int pad)
861 {
862         if (buf->page_len == 0) {
863                 /* We need to adjust head and buf->len in tandem in this
864                  * case to make svc_defer() work--it finds the original
865                  * buffer start using buf->len - buf->head[0].iov_len. */
866                 buf->head[0].iov_len -= pad;
867         }
868 }
869
870 static int
871 unwrap_priv_data(struct svc_rqst *rqstp, struct xdr_buf *buf, u32 seq, struct gss_ctx *ctx)
872 {
873         u32 priv_len, maj_stat;
874         int pad, saved_len, remaining_len, offset;
875
876         rqstp->rq_splice_ok = 0;
877
878         priv_len = svc_getnl(&buf->head[0]);
879         if (rqstp->rq_deferred) {
880                 /* Already decrypted last time through! The sequence number
881                  * check at out_seq is unnecessary but harmless: */
882                 goto out_seq;
883         }
884         /* buf->len is the number of bytes from the original start of the
885          * request to the end, where head[0].iov_len is just the bytes
886          * not yet read from the head, so these two values are different: */
887         remaining_len = total_buf_len(buf);
888         if (priv_len > remaining_len)
889                 return -EINVAL;
890         pad = remaining_len - priv_len;
891         buf->len -= pad;
892         fix_priv_head(buf, pad);
893
894         /* Maybe it would be better to give gss_unwrap a length parameter: */
895         saved_len = buf->len;
896         buf->len = priv_len;
897         maj_stat = gss_unwrap(ctx, 0, buf);
898         pad = priv_len - buf->len;
899         buf->len = saved_len;
900         buf->len -= pad;
901         /* The upper layers assume the buffer is aligned on 4-byte boundaries.
902          * In the krb5p case, at least, the data ends up offset, so we need to
903          * move it around. */
904         /* XXX: This is very inefficient.  It would be better to either do
905          * this while we encrypt, or maybe in the receive code, if we can peak
906          * ahead and work out the service and mechanism there. */
907         offset = buf->head[0].iov_len % 4;
908         if (offset) {
909                 buf->buflen = RPCSVC_MAXPAYLOAD;
910                 xdr_shift_buf(buf, offset);
911                 fix_priv_head(buf, pad);
912         }
913         if (maj_stat != GSS_S_COMPLETE)
914                 return -EINVAL;
915 out_seq:
916         if (svc_getnl(&buf->head[0]) != seq)
917                 return -EINVAL;
918         return 0;
919 }
920
921 struct gss_svc_data {
922         /* decoded gss client cred: */
923         struct rpc_gss_wire_cred        clcred;
924         /* save a pointer to the beginning of the encoded verifier,
925          * for use in encryption/checksumming in svcauth_gss_release: */
926         __be32                          *verf_start;
927         struct rsc                      *rsci;
928 };
929
930 char *svc_gss_principal(struct svc_rqst *rqstp)
931 {
932         struct gss_svc_data *gd = (struct gss_svc_data *)rqstp->rq_auth_data;
933
934         if (gd && gd->rsci)
935                 return gd->rsci->client_name;
936         return NULL;
937 }
938
939 static int
940 svcauth_gss_set_client(struct svc_rqst *rqstp)
941 {
942         struct gss_svc_data *svcdata = rqstp->rq_auth_data;
943         struct rsc *rsci = svcdata->rsci;
944         struct rpc_gss_wire_cred *gc = &svcdata->clcred;
945         int stat;
946
947         /*
948          * A gss export can be specified either by:
949          *      export  *(sec=krb5,rw)
950          * or by
951          *      export gss/krb5(rw)
952          * The latter is deprecated; but for backwards compatibility reasons
953          * the nfsd code will still fall back on trying it if the former
954          * doesn't work; so we try to make both available to nfsd, below.
955          */
956         rqstp->rq_gssclient = find_gss_auth_domain(rsci->mechctx, gc->gc_svc);
957         if (rqstp->rq_gssclient == NULL)
958                 return SVC_DENIED;
959         stat = svcauth_unix_set_client(rqstp);
960         if (stat == SVC_DROP)
961                 return stat;
962         return SVC_OK;
963 }
964
965 static inline int
966 gss_write_init_verf(struct svc_rqst *rqstp, struct rsi *rsip)
967 {
968         struct rsc *rsci;
969         int        rc;
970
971         if (rsip->major_status != GSS_S_COMPLETE)
972                 return gss_write_null_verf(rqstp);
973         rsci = gss_svc_searchbyctx(&rsip->out_handle);
974         if (rsci == NULL) {
975                 rsip->major_status = GSS_S_NO_CONTEXT;
976                 return gss_write_null_verf(rqstp);
977         }
978         rc = gss_write_verf(rqstp, rsci->mechctx, GSS_SEQ_WIN);
979         cache_put(&rsci->h, &rsc_cache);
980         return rc;
981 }
982
983 /*
984  * Having read the cred already and found we're in the context
985  * initiation case, read the verifier and initiate (or check the results
986  * of) upcalls to userspace for help with context initiation.  If
987  * the upcall results are available, write the verifier and result.
988  * Otherwise, drop the request pending an answer to the upcall.
989  */
990 static int svcauth_gss_handle_init(struct svc_rqst *rqstp,
991                         struct rpc_gss_wire_cred *gc, __be32 *authp)
992 {
993         struct kvec *argv = &rqstp->rq_arg.head[0];
994         struct kvec *resv = &rqstp->rq_res.head[0];
995         struct xdr_netobj tmpobj;
996         struct rsi *rsip, rsikey;
997         int ret;
998
999         /* Read the verifier; should be NULL: */
1000         *authp = rpc_autherr_badverf;
1001         if (argv->iov_len < 2 * 4)
1002                 return SVC_DENIED;
1003         if (svc_getnl(argv) != RPC_AUTH_NULL)
1004                 return SVC_DENIED;
1005         if (svc_getnl(argv) != 0)
1006                 return SVC_DENIED;
1007
1008         /* Martial context handle and token for upcall: */
1009         *authp = rpc_autherr_badcred;
1010         if (gc->gc_proc == RPC_GSS_PROC_INIT && gc->gc_ctx.len != 0)
1011                 return SVC_DENIED;
1012         memset(&rsikey, 0, sizeof(rsikey));
1013         if (dup_netobj(&rsikey.in_handle, &gc->gc_ctx))
1014                 return SVC_DROP;
1015         *authp = rpc_autherr_badverf;
1016         if (svc_safe_getnetobj(argv, &tmpobj)) {
1017                 kfree(rsikey.in_handle.data);
1018                 return SVC_DENIED;
1019         }
1020         if (dup_netobj(&rsikey.in_token, &tmpobj)) {
1021                 kfree(rsikey.in_handle.data);
1022                 return SVC_DROP;
1023         }
1024
1025         /* Perform upcall, or find upcall result: */
1026         rsip = rsi_lookup(&rsikey);
1027         rsi_free(&rsikey);
1028         if (!rsip)
1029                 return SVC_DROP;
1030         switch (cache_check(&rsi_cache, &rsip->h, &rqstp->rq_chandle)) {
1031         case -EAGAIN:
1032         case -ETIMEDOUT:
1033         case -ENOENT:
1034                 /* No upcall result: */
1035                 return SVC_DROP;
1036         case 0:
1037                 ret = SVC_DROP;
1038                 /* Got an answer to the upcall; use it: */
1039                 if (gss_write_init_verf(rqstp, rsip))
1040                         goto out;
1041                 if (resv->iov_len + 4 > PAGE_SIZE)
1042                         goto out;
1043                 svc_putnl(resv, RPC_SUCCESS);
1044                 if (svc_safe_putnetobj(resv, &rsip->out_handle))
1045                         goto out;
1046                 if (resv->iov_len + 3 * 4 > PAGE_SIZE)
1047                         goto out;
1048                 svc_putnl(resv, rsip->major_status);
1049                 svc_putnl(resv, rsip->minor_status);
1050                 svc_putnl(resv, GSS_SEQ_WIN);
1051                 if (svc_safe_putnetobj(resv, &rsip->out_token))
1052                         goto out;
1053         }
1054         ret = SVC_COMPLETE;
1055 out:
1056         cache_put(&rsip->h, &rsi_cache);
1057         return ret;
1058 }
1059
1060 /*
1061  * Accept an rpcsec packet.
1062  * If context establishment, punt to user space
1063  * If data exchange, verify/decrypt
1064  * If context destruction, handle here
1065  * In the context establishment and destruction case we encode
1066  * response here and return SVC_COMPLETE.
1067  */
1068 static int
1069 svcauth_gss_accept(struct svc_rqst *rqstp, __be32 *authp)
1070 {
1071         struct kvec     *argv = &rqstp->rq_arg.head[0];
1072         struct kvec     *resv = &rqstp->rq_res.head[0];
1073         u32             crlen;
1074         struct gss_svc_data *svcdata = rqstp->rq_auth_data;
1075         struct rpc_gss_wire_cred *gc;
1076         struct rsc      *rsci = NULL;
1077         __be32          *rpcstart;
1078         __be32          *reject_stat = resv->iov_base + resv->iov_len;
1079         int             ret;
1080
1081         dprintk("RPC:       svcauth_gss: argv->iov_len = %zd\n",
1082                         argv->iov_len);
1083
1084         *authp = rpc_autherr_badcred;
1085         if (!svcdata)
1086                 svcdata = kmalloc(sizeof(*svcdata), GFP_KERNEL);
1087         if (!svcdata)
1088                 goto auth_err;
1089         rqstp->rq_auth_data = svcdata;
1090         svcdata->verf_start = NULL;
1091         svcdata->rsci = NULL;
1092         gc = &svcdata->clcred;
1093
1094         /* start of rpc packet is 7 u32's back from here:
1095          * xid direction rpcversion prog vers proc flavour
1096          */
1097         rpcstart = argv->iov_base;
1098         rpcstart -= 7;
1099
1100         /* credential is:
1101          *   version(==1), proc(0,1,2,3), seq, service (1,2,3), handle
1102          * at least 5 u32s, and is preceeded by length, so that makes 6.
1103          */
1104
1105         if (argv->iov_len < 5 * 4)
1106                 goto auth_err;
1107         crlen = svc_getnl(argv);
1108         if (svc_getnl(argv) != RPC_GSS_VERSION)
1109                 goto auth_err;
1110         gc->gc_proc = svc_getnl(argv);
1111         gc->gc_seq = svc_getnl(argv);
1112         gc->gc_svc = svc_getnl(argv);
1113         if (svc_safe_getnetobj(argv, &gc->gc_ctx))
1114                 goto auth_err;
1115         if (crlen != round_up_to_quad(gc->gc_ctx.len) + 5 * 4)
1116                 goto auth_err;
1117
1118         if ((gc->gc_proc != RPC_GSS_PROC_DATA) && (rqstp->rq_proc != 0))
1119                 goto auth_err;
1120
1121         *authp = rpc_autherr_badverf;
1122         switch (gc->gc_proc) {
1123         case RPC_GSS_PROC_INIT:
1124         case RPC_GSS_PROC_CONTINUE_INIT:
1125                 return svcauth_gss_handle_init(rqstp, gc, authp);
1126         case RPC_GSS_PROC_DATA:
1127         case RPC_GSS_PROC_DESTROY:
1128                 /* Look up the context, and check the verifier: */
1129                 *authp = rpcsec_gsserr_credproblem;
1130                 rsci = gss_svc_searchbyctx(&gc->gc_ctx);
1131                 if (!rsci)
1132                         goto auth_err;
1133                 switch (gss_verify_header(rqstp, rsci, rpcstart, gc, authp)) {
1134                 case SVC_OK:
1135                         break;
1136                 case SVC_DENIED:
1137                         goto auth_err;
1138                 case SVC_DROP:
1139                         goto drop;
1140                 }
1141                 break;
1142         default:
1143                 *authp = rpc_autherr_rejectedcred;
1144                 goto auth_err;
1145         }
1146
1147         /* now act upon the command: */
1148         switch (gc->gc_proc) {
1149         case RPC_GSS_PROC_DESTROY:
1150                 if (gss_write_verf(rqstp, rsci->mechctx, gc->gc_seq))
1151                         goto auth_err;
1152                 rsci->h.expiry_time = get_seconds();
1153                 set_bit(CACHE_NEGATIVE, &rsci->h.flags);
1154                 if (resv->iov_len + 4 > PAGE_SIZE)
1155                         goto drop;
1156                 svc_putnl(resv, RPC_SUCCESS);
1157                 goto complete;
1158         case RPC_GSS_PROC_DATA:
1159                 *authp = rpcsec_gsserr_ctxproblem;
1160                 svcdata->verf_start = resv->iov_base + resv->iov_len;
1161                 if (gss_write_verf(rqstp, rsci->mechctx, gc->gc_seq))
1162                         goto auth_err;
1163                 rqstp->rq_cred = rsci->cred;
1164                 get_group_info(rsci->cred.cr_group_info);
1165                 *authp = rpc_autherr_badcred;
1166                 switch (gc->gc_svc) {
1167                 case RPC_GSS_SVC_NONE:
1168                         break;
1169                 case RPC_GSS_SVC_INTEGRITY:
1170                         /* placeholders for length and seq. number: */
1171                         svc_putnl(resv, 0);
1172                         svc_putnl(resv, 0);
1173                         if (unwrap_integ_data(&rqstp->rq_arg,
1174                                         gc->gc_seq, rsci->mechctx))
1175                                 goto garbage_args;
1176                         break;
1177                 case RPC_GSS_SVC_PRIVACY:
1178                         /* placeholders for length and seq. number: */
1179                         svc_putnl(resv, 0);
1180                         svc_putnl(resv, 0);
1181                         if (unwrap_priv_data(rqstp, &rqstp->rq_arg,
1182                                         gc->gc_seq, rsci->mechctx))
1183                                 goto garbage_args;
1184                         break;
1185                 default:
1186                         goto auth_err;
1187                 }
1188                 svcdata->rsci = rsci;
1189                 cache_get(&rsci->h);
1190                 rqstp->rq_flavor = gss_svc_to_pseudoflavor(
1191                                         rsci->mechctx->mech_type, gc->gc_svc);
1192                 ret = SVC_OK;
1193                 goto out;
1194         }
1195 garbage_args:
1196         ret = SVC_GARBAGE;
1197         goto out;
1198 auth_err:
1199         /* Restore write pointer to its original value: */
1200         xdr_ressize_check(rqstp, reject_stat);
1201         ret = SVC_DENIED;
1202         goto out;
1203 complete:
1204         ret = SVC_COMPLETE;
1205         goto out;
1206 drop:
1207         ret = SVC_DROP;
1208 out:
1209         if (rsci)
1210                 cache_put(&rsci->h, &rsc_cache);
1211         return ret;
1212 }
1213
1214 static __be32 *
1215 svcauth_gss_prepare_to_wrap(struct xdr_buf *resbuf, struct gss_svc_data *gsd)
1216 {
1217         __be32 *p;
1218         u32 verf_len;
1219
1220         p = gsd->verf_start;
1221         gsd->verf_start = NULL;
1222
1223         /* If the reply stat is nonzero, don't wrap: */
1224         if (*(p-1) != rpc_success)
1225                 return NULL;
1226         /* Skip the verifier: */
1227         p += 1;
1228         verf_len = ntohl(*p++);
1229         p += XDR_QUADLEN(verf_len);
1230         /* move accept_stat to right place: */
1231         memcpy(p, p + 2, 4);
1232         /* Also don't wrap if the accept stat is nonzero: */
1233         if (*p != rpc_success) {
1234                 resbuf->head[0].iov_len -= 2 * 4;
1235                 return NULL;
1236         }
1237         p++;
1238         return p;
1239 }
1240
1241 static inline int
1242 svcauth_gss_wrap_resp_integ(struct svc_rqst *rqstp)
1243 {
1244         struct gss_svc_data *gsd = (struct gss_svc_data *)rqstp->rq_auth_data;
1245         struct rpc_gss_wire_cred *gc = &gsd->clcred;
1246         struct xdr_buf *resbuf = &rqstp->rq_res;
1247         struct xdr_buf integ_buf;
1248         struct xdr_netobj mic;
1249         struct kvec *resv;
1250         __be32 *p;
1251         int integ_offset, integ_len;
1252         int stat = -EINVAL;
1253
1254         p = svcauth_gss_prepare_to_wrap(resbuf, gsd);
1255         if (p == NULL)
1256                 goto out;
1257         integ_offset = (u8 *)(p + 1) - (u8 *)resbuf->head[0].iov_base;
1258         integ_len = resbuf->len - integ_offset;
1259         BUG_ON(integ_len % 4);
1260         *p++ = htonl(integ_len);
1261         *p++ = htonl(gc->gc_seq);
1262         if (xdr_buf_subsegment(resbuf, &integ_buf, integ_offset,
1263                                 integ_len))
1264                 BUG();
1265         if (resbuf->tail[0].iov_base == NULL) {
1266                 if (resbuf->head[0].iov_len + RPC_MAX_AUTH_SIZE > PAGE_SIZE)
1267                         goto out_err;
1268                 resbuf->tail[0].iov_base = resbuf->head[0].iov_base
1269                                                 + resbuf->head[0].iov_len;
1270                 resbuf->tail[0].iov_len = 0;
1271                 resv = &resbuf->tail[0];
1272         } else {
1273                 resv = &resbuf->tail[0];
1274         }
1275         mic.data = (u8 *)resv->iov_base + resv->iov_len + 4;
1276         if (gss_get_mic(gsd->rsci->mechctx, &integ_buf, &mic))
1277                 goto out_err;
1278         svc_putnl(resv, mic.len);
1279         memset(mic.data + mic.len, 0,
1280                         round_up_to_quad(mic.len) - mic.len);
1281         resv->iov_len += XDR_QUADLEN(mic.len) << 2;
1282         /* not strictly required: */
1283         resbuf->len += XDR_QUADLEN(mic.len) << 2;
1284         BUG_ON(resv->iov_len > PAGE_SIZE);
1285 out:
1286         stat = 0;
1287 out_err:
1288         return stat;
1289 }
1290
1291 static inline int
1292 svcauth_gss_wrap_resp_priv(struct svc_rqst *rqstp)
1293 {
1294         struct gss_svc_data *gsd = (struct gss_svc_data *)rqstp->rq_auth_data;
1295         struct rpc_gss_wire_cred *gc = &gsd->clcred;
1296         struct xdr_buf *resbuf = &rqstp->rq_res;
1297         struct page **inpages = NULL;
1298         __be32 *p, *len;
1299         int offset;
1300         int pad;
1301
1302         p = svcauth_gss_prepare_to_wrap(resbuf, gsd);
1303         if (p == NULL)
1304                 return 0;
1305         len = p++;
1306         offset = (u8 *)p - (u8 *)resbuf->head[0].iov_base;
1307         *p++ = htonl(gc->gc_seq);
1308         inpages = resbuf->pages;
1309         /* XXX: Would be better to write some xdr helper functions for
1310          * nfs{2,3,4}xdr.c that place the data right, instead of copying: */
1311         if (resbuf->tail[0].iov_base) {
1312                 BUG_ON(resbuf->tail[0].iov_base >= resbuf->head[0].iov_base
1313                                                         + PAGE_SIZE);
1314                 BUG_ON(resbuf->tail[0].iov_base < resbuf->head[0].iov_base);
1315                 if (resbuf->tail[0].iov_len + resbuf->head[0].iov_len
1316                                 + 2 * RPC_MAX_AUTH_SIZE > PAGE_SIZE)
1317                         return -ENOMEM;
1318                 memmove(resbuf->tail[0].iov_base + RPC_MAX_AUTH_SIZE,
1319                         resbuf->tail[0].iov_base,
1320                         resbuf->tail[0].iov_len);
1321                 resbuf->tail[0].iov_base += RPC_MAX_AUTH_SIZE;
1322         }
1323         if (resbuf->tail[0].iov_base == NULL) {
1324                 if (resbuf->head[0].iov_len + 2*RPC_MAX_AUTH_SIZE > PAGE_SIZE)
1325                         return -ENOMEM;
1326                 resbuf->tail[0].iov_base = resbuf->head[0].iov_base
1327                         + resbuf->head[0].iov_len + RPC_MAX_AUTH_SIZE;
1328                 resbuf->tail[0].iov_len = 0;
1329         }
1330         if (gss_wrap(gsd->rsci->mechctx, offset, resbuf, inpages))
1331                 return -ENOMEM;
1332         *len = htonl(resbuf->len - offset);
1333         pad = 3 - ((resbuf->len - offset - 1)&3);
1334         p = (__be32 *)(resbuf->tail[0].iov_base + resbuf->tail[0].iov_len);
1335         memset(p, 0, pad);
1336         resbuf->tail[0].iov_len += pad;
1337         resbuf->len += pad;
1338         return 0;
1339 }
1340
1341 static int
1342 svcauth_gss_release(struct svc_rqst *rqstp)
1343 {
1344         struct gss_svc_data *gsd = (struct gss_svc_data *)rqstp->rq_auth_data;
1345         struct rpc_gss_wire_cred *gc = &gsd->clcred;
1346         struct xdr_buf *resbuf = &rqstp->rq_res;
1347         int stat = -EINVAL;
1348
1349         if (gc->gc_proc != RPC_GSS_PROC_DATA)
1350                 goto out;
1351         /* Release can be called twice, but we only wrap once. */
1352         if (gsd->verf_start == NULL)
1353                 goto out;
1354         /* normally not set till svc_send, but we need it here: */
1355         /* XXX: what for?  Do we mess it up the moment we call svc_putu32
1356          * or whatever? */
1357         resbuf->len = total_buf_len(resbuf);
1358         switch (gc->gc_svc) {
1359         case RPC_GSS_SVC_NONE:
1360                 break;
1361         case RPC_GSS_SVC_INTEGRITY:
1362                 stat = svcauth_gss_wrap_resp_integ(rqstp);
1363                 if (stat)
1364                         goto out_err;
1365                 break;
1366         case RPC_GSS_SVC_PRIVACY:
1367                 stat = svcauth_gss_wrap_resp_priv(rqstp);
1368                 if (stat)
1369                         goto out_err;
1370                 break;
1371         default:
1372                 goto out_err;
1373         }
1374
1375 out:
1376         stat = 0;
1377 out_err:
1378         if (rqstp->rq_client)
1379                 auth_domain_put(rqstp->rq_client);
1380         rqstp->rq_client = NULL;
1381         if (rqstp->rq_gssclient)
1382                 auth_domain_put(rqstp->rq_gssclient);
1383         rqstp->rq_gssclient = NULL;
1384         if (rqstp->rq_cred.cr_group_info)
1385                 put_group_info(rqstp->rq_cred.cr_group_info);
1386         rqstp->rq_cred.cr_group_info = NULL;
1387         if (gsd->rsci)
1388                 cache_put(&gsd->rsci->h, &rsc_cache);
1389         gsd->rsci = NULL;
1390
1391         return stat;
1392 }
1393
1394 static void
1395 svcauth_gss_domain_release(struct auth_domain *dom)
1396 {
1397         struct gss_domain *gd = container_of(dom, struct gss_domain, h);
1398
1399         kfree(dom->name);
1400         kfree(gd);
1401 }
1402
1403 static struct auth_ops svcauthops_gss = {
1404         .name           = "rpcsec_gss",
1405         .owner          = THIS_MODULE,
1406         .flavour        = RPC_AUTH_GSS,
1407         .accept         = svcauth_gss_accept,
1408         .release        = svcauth_gss_release,
1409         .domain_release = svcauth_gss_domain_release,
1410         .set_client     = svcauth_gss_set_client,
1411 };
1412
1413 int
1414 gss_svc_init(void)
1415 {
1416         int rv = svc_auth_register(RPC_AUTH_GSS, &svcauthops_gss);
1417         if (rv)
1418                 return rv;
1419         rv = cache_register(&rsc_cache);
1420         if (rv)
1421                 goto out1;
1422         rv = cache_register(&rsi_cache);
1423         if (rv)
1424                 goto out2;
1425         return 0;
1426 out2:
1427         cache_unregister(&rsc_cache);
1428 out1:
1429         svc_auth_unregister(RPC_AUTH_GSS);
1430         return rv;
1431 }
1432
1433 void
1434 gss_svc_shutdown(void)
1435 {
1436         cache_unregister(&rsc_cache);
1437         cache_unregister(&rsi_cache);
1438         svc_auth_unregister(RPC_AUTH_GSS);
1439 }