plug leak
[mech_eap.git] / util_radius.cpp
1 /*
2  * Copyright (c) 2010, JANET(UK)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * 3. Neither the name of JANET(UK) nor the names of its contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  */
32
33 #include "gssapiP_eap.h"
34
35 /* stuff that should be provided by libradsec/libfreeradius-radius */
36 #define VENDORATTR(vendor, attr)            (((vendor) << 16) | (attr))
37
38 #ifndef ATTRID
39 #define ATTRID(attr)                        ((attr) & 0xFFFF)
40 #endif
41
42 static gss_buffer_desc radiusUrnPrefix = {
43     sizeof("urn:x-radius:") - 1,
44     (void *)"urn:x-radius:"
45 };
46
47 static VALUE_PAIR *copyAvps(const VALUE_PAIR *src);
48
49 gss_eap_radius_attr_provider::gss_eap_radius_attr_provider(void)
50 {
51     m_vps = NULL;
52     m_authenticated = false;
53 }
54
55 gss_eap_radius_attr_provider::~gss_eap_radius_attr_provider(void)
56 {
57     if (m_vps != NULL)
58         pairfree(&m_vps);
59 }
60
61 bool
62 gss_eap_radius_attr_provider::initFromExistingContext(const gss_eap_attr_ctx *manager,
63                                                       const gss_eap_attr_provider *ctx)
64 {
65     const gss_eap_radius_attr_provider *radius;
66
67     if (!gss_eap_attr_provider::initFromExistingContext(manager, ctx))
68         return false;
69
70     radius = static_cast<const gss_eap_radius_attr_provider *>(ctx);
71
72     if (radius->m_vps != NULL)
73         m_vps = copyAvps(const_cast<VALUE_PAIR *>(radius->getAvps()));
74
75     m_authenticated = radius->m_authenticated;
76
77     return true;
78 }
79
80 bool
81 gss_eap_radius_attr_provider::initFromGssContext(const gss_eap_attr_ctx *manager,
82                                                  const gss_cred_id_t gssCred,
83                                                  const gss_ctx_id_t gssCtx)
84 {
85     if (!gss_eap_attr_provider::initFromGssContext(manager, gssCred, gssCtx))
86         return false;
87
88     if (gssCtx != GSS_C_NO_CONTEXT) {
89         if (gssCtx->acceptorCtx.vps != NULL) {
90             m_vps = copyAvps(gssCtx->acceptorCtx.vps);
91             if (m_vps == NULL)
92                 return false;
93
94             /* We assume libradsec validated this for us */
95             assert(pairfind(m_vps, PW_MESSAGE_AUTHENTICATOR) != NULL);
96             m_authenticated = true;
97         }
98     }
99
100     return true;
101 }
102
103 static bool
104 alreadyAddedAttributeP(std::vector <std::string> &attrs, VALUE_PAIR *vp)
105 {
106     for (std::vector<std::string>::const_iterator a = attrs.begin();
107          a != attrs.end();
108          ++a) {
109         if (strcmp(vp->name, (*a).c_str()) == 0)
110             return true;
111     }
112
113     return false;
114 }
115
116 static bool
117 isSecretAttributeP(uint16_t attrid, uint16_t vendor)
118 {
119     bool ret = false;
120
121     switch (vendor) {
122     case VENDORPEC_MS:
123         switch (attrid) {
124         case PW_MS_MPPE_SEND_KEY:
125         case PW_MS_MPPE_RECV_KEY:
126             ret = true;
127             break;
128         default:
129             break;
130         }
131     default:
132         break;
133     }
134
135     return ret;
136 }
137
138 static bool
139 isSecretAttributeP(uint32_t attribute)
140 {
141     return isSecretAttributeP(ATTRID(attribute), VENDOR(attribute));
142 }
143
144 static bool
145 isHiddenAttributeP(uint16_t attrid, uint16_t vendor)
146 {
147     bool ret = false;
148
149     /* should have been filtered */
150     assert(!isSecretAttributeP(attrid, vendor));
151
152     switch (vendor) {
153     case VENDORPEC_UKERNA:
154         ret = true;
155         break;
156     default:
157         break;
158     }
159
160     return ret;
161 }
162
163 static bool
164 isHiddenAttributeP(uint32_t attribute)
165 {
166     return isHiddenAttributeP(ATTRID(attribute), VENDOR(attribute));
167 }
168
169 /*
170  * Copy AVP list, same as paircopy except it filters out attributes
171  * containing keys.
172  */
173 static VALUE_PAIR *
174 copyAvps(const VALUE_PAIR *src)
175 {
176     const VALUE_PAIR *vp;
177     VALUE_PAIR *dst = NULL, **pDst = &dst;
178
179     for (vp = src; vp != NULL; vp = vp->next) {
180         VALUE_PAIR *vpcopy;
181
182         if (isSecretAttributeP(vp->attribute))
183             continue;
184
185         vpcopy = paircopyvp(vp);
186         if (vpcopy == NULL) {
187             pairfree(&dst);
188             throw new std::bad_alloc;
189             return NULL;
190         }
191         *pDst = vpcopy;
192         pDst = &vpcopy->next;
193      }
194
195     return dst;
196 }
197
198 bool
199 gss_eap_radius_attr_provider::getAttributeTypes(gss_eap_attr_enumeration_cb addAttribute, void *data) const
200 {
201     VALUE_PAIR *vp;
202     std::vector <std::string> seen;
203
204     for (vp = m_vps; vp != NULL; vp = vp->next) {
205         gss_buffer_desc attribute;
206         char attrid[64];
207
208         if (isHiddenAttributeP(vp->attribute))
209             continue;
210
211         if (alreadyAddedAttributeP(seen, vp))
212             continue;
213
214         snprintf(attrid, sizeof(attrid), "%s%d",
215             (char *)radiusUrnPrefix.value, vp->attribute);
216
217         attribute.value = attrid;
218         attribute.length = strlen(attrid);
219
220         if (!addAttribute(this, &attribute, data))
221             return false;
222
223         seen.push_back(std::string(vp->name));
224     }
225
226     return true;
227 }
228
229 uint32_t
230 getAttributeId(const gss_buffer_t attr)
231 {
232     OM_uint32 tmpMinor;
233     gss_buffer_desc strAttr = GSS_C_EMPTY_BUFFER;
234     DICT_ATTR *da;
235     char *s;
236     uint32_t attrid = 0;
237
238     if (attr->length < radiusUrnPrefix.length ||
239         memcmp(attr->value, radiusUrnPrefix.value, radiusUrnPrefix.length) != 0)
240         return 0;
241
242     /* need to duplicate because attr may not be NUL terminated */
243     duplicateBuffer(*attr, &strAttr);
244     s = (char *)strAttr.value + radiusUrnPrefix.length;
245
246     if (isdigit(*s)) {
247         attrid = strtoul(s, NULL, 10);
248     } else {
249         da = dict_attrbyname(s);
250         if (da != NULL)
251             attrid = da->attr;
252     }
253
254     gss_release_buffer(&tmpMinor, &strAttr);
255
256     return attrid;
257 }
258
259 bool
260 gss_eap_radius_attr_provider::setAttribute(int complete,
261                                            uint32_t attrid,
262                                            const gss_buffer_t value)
263 {
264     OM_uint32 major = GSS_S_UNAVAILABLE, minor;
265
266     if (!isSecretAttributeP(attrid) &&
267         !isHiddenAttributeP(attrid)) {
268         deleteAttribute(attrid);
269
270         major = gssEapRadiusAddAvp(&minor, &m_vps,
271                                    ATTRID(attrid), VENDOR(attrid), 
272                                    value);
273     }
274
275     return !GSS_ERROR(major);
276 }
277
278 bool
279 gss_eap_radius_attr_provider::setAttribute(int complete,
280                                            const gss_buffer_t attr,
281                                            const gss_buffer_t value)
282 {
283     uint32_t attrid = getAttributeId(attr);
284
285     if (!attrid)
286         return false;
287
288     return setAttribute(complete, attrid, value);
289 }
290
291 bool
292 gss_eap_radius_attr_provider::deleteAttribute(uint32_t attrid)
293 {
294     if (isSecretAttributeP(attrid) || isHiddenAttributeP(attrid) ||
295         pairfind(m_vps, attrid) == NULL)
296         return false;
297
298     pairdelete(&m_vps, attrid);
299
300     return true;
301 }
302
303 bool
304 gss_eap_radius_attr_provider::deleteAttribute(const gss_buffer_t attr)
305 {
306     uint32_t attrid = getAttributeId(attr);
307
308     if (!attrid)
309         return false;
310
311     return deleteAttribute(attrid);
312 }
313
314 bool
315 gss_eap_radius_attr_provider::getAttribute(const gss_buffer_t attr,
316                                            int *authenticated,
317                                            int *complete,
318                                            gss_buffer_t value,
319                                            gss_buffer_t display_value,
320                                            int *more) const
321 {
322     uint32_t attrid;
323
324     attrid = getAttributeId(attr);
325     if (!attrid)
326         return false;
327
328     return getAttribute(attrid, authenticated, complete,
329                         value, display_value, more);
330 }
331
332 bool
333 gss_eap_radius_attr_provider::getAttribute(uint32_t attrid,
334                                            int *authenticated,
335                                            int *complete,
336                                            gss_buffer_t value,
337                                            gss_buffer_t display_value,
338                                            int *more) const
339 {
340     VALUE_PAIR *vp;
341     int i = *more, count = 0;
342
343     *more = 0;
344
345     if (isHiddenAttributeP(attrid))
346         return false;
347
348     if (i == -1)
349         i = 0;
350
351     for (vp = pairfind(m_vps, attrid);
352          vp != NULL;
353          vp = pairfind(vp->next, attrid)) {
354         if (count++ == i) {
355             if (pairfind(vp->next, attrid) != NULL)
356                 *more = count;
357             break;
358         }
359     }
360
361     if (vp == NULL && *more == 0)
362         return false;
363
364     if (value != GSS_C_NO_BUFFER) {
365         gss_buffer_desc valueBuf;
366
367         valueBuf.value = (void *)vp->vp_octets;
368         valueBuf.length = vp->length;
369
370         duplicateBuffer(valueBuf, value);
371     }
372
373     if (display_value != GSS_C_NO_BUFFER) {
374         char displayString[MAX_STRING_LEN];
375         gss_buffer_desc displayBuf;
376
377         displayBuf.length = vp_prints_value(displayString,
378                                             sizeof(displayString), vp, 0);
379         displayBuf.value = (void *)displayString;
380
381         duplicateBuffer(displayBuf, display_value);
382     }
383
384     if (authenticated != NULL)
385         *authenticated = m_authenticated;
386     if (complete != NULL)
387         *complete = true;
388
389     return true;
390 }
391
392 bool
393 gss_eap_radius_attr_provider::getFragmentedAttribute(uint16_t attribute,
394                                                      uint16_t vendor,
395                                                      int *authenticated,
396                                                      int *complete,
397                                                      gss_buffer_t value) const
398 {
399     OM_uint32 major, minor;
400
401     major = gssEapRadiusGetAvp(&minor, m_vps, attribute, vendor, value, TRUE);
402
403     if (authenticated != NULL)
404         *authenticated = m_authenticated;
405     if (complete != NULL)
406         *complete = true;
407
408     return !GSS_ERROR(major);
409 }
410
411 bool
412 gss_eap_radius_attr_provider::getAttribute(uint16_t attribute,
413                                            uint16_t vendor,
414                                            int *authenticated,
415                                            int *complete,
416                                            gss_buffer_t value,
417                                            gss_buffer_t display_value,
418                                            int *more) const
419 {
420
421     return getAttribute(VENDORATTR(attribute, vendor),
422                         authenticated, complete,
423                         value, display_value, more);
424 }
425
426 gss_any_t
427 gss_eap_radius_attr_provider::mapToAny(int authenticated,
428                                        gss_buffer_t type_id) const
429 {
430     if (authenticated && !m_authenticated)
431         return (gss_any_t)NULL;
432
433     return (gss_any_t)copyAvps(m_vps);
434 }
435
436 void
437 gss_eap_radius_attr_provider::releaseAnyNameMapping(gss_buffer_t type_id,
438                                                     gss_any_t input) const
439 {
440     pairfree((VALUE_PAIR **)&input);
441 }
442
443 bool
444 gss_eap_radius_attr_provider::init(void)
445 {
446     gss_eap_attr_ctx::registerProvider(ATTR_TYPE_RADIUS,
447                                        "urn:ietf:params:gss-eap:radius-avp",
448                                        gss_eap_radius_attr_provider::createAttrContext);
449     return true;
450 }
451
452 void
453 gss_eap_radius_attr_provider::finalize(void)
454 {
455     gss_eap_attr_ctx::unregisterProvider(ATTR_TYPE_RADIUS);
456 }
457
458 gss_eap_attr_provider *
459 gss_eap_radius_attr_provider::createAttrContext(void)
460 {
461     return new gss_eap_radius_attr_provider;
462 }
463
464 OM_uint32
465 gssEapRadiusAddAvp(OM_uint32 *minor,
466                    VALUE_PAIR **vps,
467                    uint16_t attribute,
468                    uint16_t vendor,
469                    const gss_buffer_t buffer)
470 {
471     uint32_t attrid = VENDORATTR(vendor, attribute);
472     unsigned char *p = (unsigned char *)buffer->value;
473     size_t remain = buffer->length;
474
475     do {
476         VALUE_PAIR *vp;
477         size_t n = remain;
478
479         if (n > MAX_STRING_LEN)
480             n = MAX_STRING_LEN;
481
482         vp = paircreate(attrid, PW_TYPE_OCTETS);
483         if (vp == NULL) {
484             *minor = ENOMEM;
485             return GSS_S_FAILURE;
486         }
487
488         memcpy(vp->vp_octets, p, n);
489         vp->length = n;
490
491         pairadd(vps, vp);
492
493         p += n;
494         remain -= n;
495     } while (remain != 0);
496
497     return GSS_S_COMPLETE;
498 }
499
500 OM_uint32
501 gssEapRadiusGetRawAvp(OM_uint32 *minor,
502                       VALUE_PAIR *vps,
503                       uint16_t attribute,
504                       uint16_t vendor,
505                       VALUE_PAIR **vp)
506 {
507     uint32_t attr = VENDORATTR(vendor, attribute);
508
509     *vp = pairfind(vps, attr);
510
511     return (*vp == NULL) ? GSS_S_UNAVAILABLE : GSS_S_COMPLETE;
512 }
513
514 OM_uint32
515 gssEapRadiusGetAvp(OM_uint32 *minor,
516                    VALUE_PAIR *vps,
517                    uint16_t attribute,
518                    uint16_t vendor,
519                    gss_buffer_t buffer,
520                    int concat)
521 {
522     VALUE_PAIR *vp;
523     unsigned char *p;
524     uint32_t attr = VENDORATTR(vendor, attribute);
525
526     buffer->length = 0;
527     buffer->value = NULL;
528
529     vp = pairfind(vps, attr);
530     if (vp == NULL)
531         return GSS_S_UNAVAILABLE;
532
533     do {
534         buffer->length += vp->length;
535     } while (concat && (vp = pairfind(vp->next, attr)) != NULL);
536
537     buffer->value = GSSEAP_MALLOC(buffer->length);
538     if (buffer->value == NULL) {
539         *minor = ENOMEM;
540         return GSS_S_FAILURE;
541     }
542
543     p = (unsigned char *)buffer->value;
544
545     for (vp = pairfind(vps, attr);
546          concat && vp != NULL;
547          vp = pairfind(vp->next, attr)) {
548         memcpy(p, vp->vp_octets, vp->length);
549         p += vp->length;
550     }
551
552     *minor = 0;
553     return GSS_S_COMPLETE;
554 }
555
556 OM_uint32
557 gssEapRadiusFreeAvps(OM_uint32 *minor,
558                      VALUE_PAIR **vps)
559 {
560     pairfree(vps);
561     *minor = 0;
562     return GSS_S_COMPLETE;
563 }
564
565 OM_uint32
566 gssEapRadiusAttrProviderInit(OM_uint32 *minor)
567 {
568     if (!gss_eap_radius_attr_provider::init()) {
569         *minor = GSSEAP_RADSEC_INIT_FAILURE;
570         return GSS_S_FAILURE;
571     }
572
573     return GSS_S_COMPLETE;
574 }
575
576 OM_uint32
577 gssEapRadiusAttrProviderFinalize(OM_uint32 *minor)
578 {
579     gss_eap_radius_attr_provider::finalize();
580     return GSS_S_COMPLETE;
581 }
582
583 /*
584  * Encoding is:
585  * 4 octet NBO attribute ID | 4 octet attribute length | attribute data
586  */
587 static size_t
588 avpSize(const VALUE_PAIR *vp)
589 {
590     size_t size = 4 + 1;
591
592     if (vp != NULL)
593         size += vp->length;
594
595     return size;
596 }
597
598 static bool
599 avpExport(const VALUE_PAIR *vp,
600           unsigned char **pBuffer,
601           size_t *pRemain)
602 {
603     unsigned char *p = *pBuffer;
604     size_t remain = *pRemain;
605
606     assert(remain >= avpSize(vp));
607
608     store_uint32_be(vp->attribute, p);
609
610     switch (vp->type) {
611     case PW_TYPE_INTEGER:
612     case PW_TYPE_IPADDR:
613     case PW_TYPE_DATE:
614         p[4] = 4;
615         store_uint32_be(vp->lvalue, p + 5);
616         break;
617     default:
618         assert(vp->length <= MAX_STRING_LEN);
619         p[4] = (uint8_t)vp->length;
620         memcpy(p + 5, vp->vp_octets, vp->length);
621         break;
622     }
623
624     *pBuffer += 5 + p[4];
625     *pRemain -= 5 + p[4];
626
627     return true;
628
629 }
630
631 static bool
632 avpImport(VALUE_PAIR **pVp,
633           unsigned char **pBuffer,
634           size_t *pRemain)
635 {
636     unsigned char *p = *pBuffer;
637     size_t remain = *pRemain;
638     VALUE_PAIR *vp = NULL;
639     DICT_ATTR *da;
640     uint32_t attrid;
641
642     if (remain < avpSize(NULL))
643         goto fail;
644
645     attrid = load_uint32_be(p);
646     p += 4;
647     remain -= 4;
648
649     da = dict_attrbyvalue(attrid);
650     if (da == NULL)
651         goto fail;
652
653     vp = pairalloc(da);
654     if (vp == NULL) {
655         throw new std::bad_alloc;
656         goto fail;
657     }
658
659     if (remain < p[0])
660         goto fail;
661
662     switch (vp->type) {
663     case PW_TYPE_INTEGER:
664     case PW_TYPE_IPADDR:
665     case PW_TYPE_DATE:
666         if (p[0] != 4)
667             goto fail;
668
669         vp->length = 4;
670         vp->lvalue = load_uint32_be(p + 1);
671         p += 5;
672         remain -= 5;
673         break;
674     case PW_TYPE_STRING:
675         /* check enough room to NUL terminate */
676         if (p[0] == MAX_STRING_LEN)
677             goto fail;
678         else
679         /* fallthrough */
680     default:
681         if (p[0] > MAX_STRING_LEN)
682             goto fail;
683
684         vp->length = (uint32_t)p[0];
685         memcpy(vp->vp_octets, p + 1, vp->length);
686
687         if (vp->type == PW_TYPE_STRING)
688             vp->vp_strvalue[vp->length] = '\0';
689
690         p += 1 + vp->length;
691         remain -= 1 + vp->length;
692         break;
693     }
694
695     *pVp = vp;
696     *pBuffer = p;
697     *pRemain = remain;
698
699     return true;
700
701 fail:
702     pairbasicfree(vp);
703     return false;
704 }
705
706 bool
707 gss_eap_radius_attr_provider::initFromBuffer(const gss_eap_attr_ctx *ctx,
708                                              const gss_buffer_t buffer)
709 {
710     unsigned char *p = (unsigned char *)buffer->value;
711     size_t remain = buffer->length;
712     VALUE_PAIR **pNext = &m_vps;
713
714     if (!gss_eap_attr_provider::initFromBuffer(ctx, buffer))
715         return false;
716
717     do {
718         VALUE_PAIR *attr;
719
720         if (!avpImport(&attr, &p, &remain))
721             return false;
722
723         *pNext = attr;
724         pNext = &attr->next;
725     } while (remain != 0);
726
727     return true;
728 }
729
730 void
731 gss_eap_radius_attr_provider::exportToBuffer(gss_buffer_t buffer) const
732 {
733     VALUE_PAIR *vp;
734     unsigned char *p;
735     size_t remain = 0;
736
737     for (vp = m_vps; vp != NULL; vp = vp->next) {
738         remain += avpSize(vp);
739     }
740
741     buffer->value = GSSEAP_MALLOC(remain);
742     if (buffer->value == NULL) {
743         throw new std::bad_alloc;
744         return;
745     }
746     buffer->length = remain;
747
748     p = (unsigned char *)buffer->value;
749
750     for (vp = m_vps; vp != NULL; vp = vp->next) {
751         avpExport(vp, &p, &remain);
752     }
753
754     assert(remain == 0);
755 }
756
757 time_t
758 gss_eap_radius_attr_provider::getExpiryTime(void) const
759 {
760     VALUE_PAIR *vp;
761
762     vp = pairfind(m_vps, PW_SESSION_TIMEOUT);
763     if (vp == NULL || vp->lvalue == 0)
764         return 0;
765
766     return time(NULL) + vp->lvalue;
767 }
768
769 OM_uint32
770 gssEapRadiusMapError(OM_uint32 *minor,
771                      struct rs_error *err)
772 {
773     int code;
774
775     assert(err != NULL);
776
777     code = rs_err_code(err, 0);
778
779     if (code == RSE_OK) {
780         *minor = 0;
781         return GSS_S_COMPLETE;
782     }
783
784     *minor = ERROR_TABLE_BASE_rse + code;
785
786     gssEapSaveStatusInfo(*minor, "%s", rs_err_msg(err, 0));
787     rs_err_free(err);
788
789     return GSS_S_FAILURE;
790 }