4da585d1fb4e59c843dbea52b6724b9379dfd6e1
[moonshot.git] / mech_eap / util_attr.cpp
1 /*
2  * Copyright (c) 2011, 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 /*
34  * Attribute provider mechanism.
35  */
36
37 #include "gssapiP_eap.h"
38
39 #include <typeinfo>
40 #include <string>
41 #include <sstream>
42 #include <exception>
43 #include <stdexcept>
44 #include <new>
45
46 /* lazy initialisation */
47 static GSSEAP_THREAD_ONCE gssEapAttrProvidersInitOnce = GSSEAP_ONCE_INITIALIZER;
48 static OM_uint32 gssEapAttrProvidersInitStatus = GSS_S_UNAVAILABLE;
49
50 static void
51 gssEapAttrProvidersInitInternal(void)
52 {
53     OM_uint32 major, minor;
54
55     assert(gssEapAttrProvidersInitStatus == GSS_S_UNAVAILABLE);
56
57     major = gssEapRadiusAttrProviderInit(&minor);
58     if (GSS_ERROR(major))
59         goto cleanup;
60
61     major = gssEapSamlAttrProvidersInit(&minor);
62     if (GSS_ERROR(major))
63         goto cleanup;
64
65     /* Allow Shibboleth initialization failure to be non-fatal */
66     gssEapLocalAttrProviderInit(&minor);
67
68 cleanup:
69 #ifdef GSSEAP_DEBUG
70     assert(major == GSS_S_COMPLETE);
71 #endif
72
73     gssEapAttrProvidersInitStatus = major;
74 }
75
76 static OM_uint32
77 gssEapAttrProvidersInit(OM_uint32 *minor)
78 {
79     GSSEAP_ONCE(&gssEapAttrProvidersInitOnce, gssEapAttrProvidersInitInternal);
80
81     if (GSS_ERROR(gssEapAttrProvidersInitStatus))
82         *minor = GSSEAP_NO_ATTR_PROVIDERS;
83
84     return gssEapAttrProvidersInitStatus;
85 }
86
87 OM_uint32
88 gssEapAttrProvidersFinalize(OM_uint32 *minor)
89 {
90     OM_uint32 major = GSS_S_COMPLETE;
91
92     if (gssEapAttrProvidersInitStatus == GSS_S_COMPLETE) {
93         major = gssEapLocalAttrProviderFinalize(minor);
94         if (major == GSS_S_COMPLETE)
95             major = gssEapSamlAttrProvidersFinalize(minor);
96         if (major == GSS_S_COMPLETE)
97             major = gssEapRadiusAttrProviderFinalize(minor);
98
99         gssEapAttrProvidersInitStatus = GSS_S_UNAVAILABLE;
100     }
101
102     return major;
103 }
104
105 static gss_eap_attr_create_provider gssEapAttrFactories[ATTR_TYPE_MAX + 1];
106
107 /*
108  * Register a provider for a particular type and prefix
109  */
110 void
111 gss_eap_attr_ctx::registerProvider(unsigned int type,
112                                    gss_eap_attr_create_provider factory)
113 {
114     assert(type <= ATTR_TYPE_MAX);
115
116     assert(gssEapAttrFactories[type] == NULL);
117
118     gssEapAttrFactories[type] = factory;
119 }
120
121 /*
122  * Unregister a provider
123  */
124 void
125 gss_eap_attr_ctx::unregisterProvider(unsigned int type)
126 {
127     assert(type <= ATTR_TYPE_MAX);
128
129     gssEapAttrFactories[type] = NULL;
130 }
131
132 /*
133  * Create an attribute context, that manages instances of providers
134  */
135 gss_eap_attr_ctx::gss_eap_attr_ctx(void)
136 {
137     m_flags = 0;
138
139     for (unsigned int i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
140         gss_eap_attr_provider *provider;
141
142         if (gssEapAttrFactories[i] != NULL) {
143             provider = (gssEapAttrFactories[i])();
144         } else {
145             provider = NULL;
146         }
147
148         m_providers[i] = provider;
149     }
150 }
151
152 /*
153  * Convert an attribute prefix to a type
154  */
155 unsigned int
156 gss_eap_attr_ctx::attributePrefixToType(const gss_buffer_t prefix) const
157 {
158     unsigned int i;
159
160     for (i = ATTR_TYPE_MIN; i < ATTR_TYPE_MAX; i++) {
161         const char *pprefix;
162
163         if (!providerEnabled(i))
164             continue;
165
166         pprefix = m_providers[i]->prefix();
167         if (pprefix == NULL)
168             continue;
169
170         if (strlen(pprefix) == prefix->length &&
171             memcmp(pprefix, prefix->value, prefix->length) == 0)
172             return i;
173     }
174
175     return ATTR_TYPE_LOCAL;
176 }
177
178 /*
179  * Convert a type to an attribute prefix
180  */
181 gss_buffer_desc
182 gss_eap_attr_ctx::attributeTypeToPrefix(unsigned int type) const
183 {
184     gss_buffer_desc prefix = GSS_C_EMPTY_BUFFER;
185
186     if (type < ATTR_TYPE_MIN || type >= ATTR_TYPE_MAX)
187         return prefix;
188
189     if (!providerEnabled(type))
190         return prefix;
191
192     prefix.value = (void *)m_providers[type]->prefix();
193     if (prefix.value != NULL)
194         prefix.length = strlen((char *)prefix.value);
195
196     return prefix;
197 }
198
199 bool
200 gss_eap_attr_ctx::providerEnabled(unsigned int type) const
201 {
202     if (type == ATTR_TYPE_LOCAL &&
203         (m_flags & ATTR_FLAG_DISABLE_LOCAL))
204         return false;
205
206     if (m_providers[type] == NULL)
207         return false;
208
209     return true;
210 }
211
212 void
213 gss_eap_attr_ctx::releaseProvider(unsigned int type)
214 {
215     delete m_providers[type];
216     m_providers[type] = NULL;
217 }
218
219 /*
220  * Initialize a context from an existing context.
221  */
222 bool
223 gss_eap_attr_ctx::initFromExistingContext(const gss_eap_attr_ctx *manager)
224 {
225     bool ret = true;
226
227     m_flags = manager->m_flags;
228
229     for (unsigned int i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
230         gss_eap_attr_provider *provider;
231
232         if (!providerEnabled(i)) {
233             releaseProvider(i);
234             continue;
235         }
236
237         provider = m_providers[i];
238
239         ret = provider->initFromExistingContext(this,
240                                                 manager->m_providers[i]);
241         if (ret == false) {
242             releaseProvider(i);
243             break;
244         }
245     }
246
247     return ret;
248 }
249
250 /*
251  * Initialize a context from a GSS credential and context.
252  */
253 bool
254 gss_eap_attr_ctx::initFromGssContext(const gss_cred_id_t cred,
255                                      const gss_ctx_id_t ctx)
256 {
257     bool ret = true;
258
259     if (cred != GSS_C_NO_CREDENTIAL &&
260         (cred->flags & GSS_EAP_DISABLE_LOCAL_ATTRS_FLAG)) {
261         m_flags |= ATTR_FLAG_DISABLE_LOCAL;
262     }
263
264     for (unsigned int i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
265         gss_eap_attr_provider *provider;
266
267         if (!providerEnabled(i)) {
268             releaseProvider(i);
269             continue;
270         }
271
272         provider = m_providers[i];
273
274         ret = provider->initFromGssContext(this, cred, ctx);
275         if (ret == false) {
276             releaseProvider(i);
277             break;
278         }
279     }
280
281     return ret;
282 }
283
284 bool
285 gss_eap_attr_ctx::initWithJsonObject(JSONObject &obj)
286 {
287     bool ret = false;
288     bool foundSource[ATTR_TYPE_MAX + 1];
289     unsigned int type;
290
291     for (type = ATTR_TYPE_MIN; type <= ATTR_TYPE_MAX; type++)
292         foundSource[type] = false;
293
294     if (obj["version"].integer() != 1)
295         return false;
296
297     m_flags = obj["flags"].integer();
298
299     JSONObject sources = obj["sources"];
300
301     /* Initialize providers from serialized state */
302     for (type = ATTR_TYPE_MIN; type <= ATTR_TYPE_MAX; type++) {
303         gss_eap_attr_provider *provider;
304         const char *key;
305
306         if (!providerEnabled(type)) {
307             releaseProvider(type);
308             continue;
309         }
310
311         provider = m_providers[type];
312         key = provider->name();
313         if (key == NULL)
314             continue;
315
316         JSONObject source = sources.get(key);
317         if (!source.isnull() &&
318             !provider->initWithJsonObject(this, source)) {
319             releaseProvider(type);
320             return false;
321         }
322
323         foundSource[type] = true;
324     }
325
326     /* Initialize remaining providers from initialized providers */
327     for (type = ATTR_TYPE_MIN; type <= ATTR_TYPE_MAX; type++) {
328         gss_eap_attr_provider *provider;
329
330         if (foundSource[type] || !providerEnabled(type))
331             continue;
332
333         provider = m_providers[type];
334
335         ret = provider->initFromGssContext(this,
336                                            GSS_C_NO_CREDENTIAL,
337                                            GSS_C_NO_CONTEXT);
338         if (ret == false) {
339             releaseProvider(type);
340             return false;
341         }
342     }
343
344     return true;
345 }
346
347 JSONObject
348 gss_eap_attr_ctx::jsonRepresentation(void) const
349 {
350     JSONObject obj, sources;
351     unsigned int i;
352
353     obj.set("version", 1);
354     obj.set("flags", m_flags);
355
356     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
357         gss_eap_attr_provider *provider;
358         const char *key;
359
360         provider = m_providers[i];
361         if (provider == NULL)
362             continue; /* provider not initialised */
363
364         key = provider->name();
365         if (key == NULL)
366             continue; /* provider does not have state */
367
368         JSONObject source = provider->jsonRepresentation();
369         sources.set(key, source);
370     }
371
372     obj.set("sources", sources);
373
374     return obj;
375 }
376
377 /*
378  * Initialize a context from an exported context or name token
379  */
380 bool
381 gss_eap_attr_ctx::initFromBuffer(const gss_buffer_t buffer)
382 {
383     OM_uint32 major, minor;
384     bool ret;
385     char *s;
386     json_error_t error;
387
388     major = bufferToString(&minor, buffer, &s);
389     if (GSS_ERROR(major))
390         return false;
391
392     JSONObject obj = JSONObject::load(s, 0, &error);
393     if (!obj.isnull()) {
394         ret = initWithJsonObject(obj);
395     } else
396         ret = false;
397
398     GSSEAP_FREE(s);
399
400     return ret;
401 }
402
403 gss_eap_attr_ctx::~gss_eap_attr_ctx(void)
404 {
405     for (unsigned int i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++)
406         delete m_providers[i];
407 }
408
409 /*
410  * Locate provider for a given type
411  */
412 gss_eap_attr_provider *
413 gss_eap_attr_ctx::getProvider(unsigned int type) const
414 {
415     assert(type >= ATTR_TYPE_MIN && type <= ATTR_TYPE_MAX);
416     return m_providers[type];
417 }
418
419 /*
420  * Get primary provider. Only the primary provider is serialised when
421  * gss_export_sec_context() or gss_export_name_composite() is called.
422  */
423 gss_eap_attr_provider *
424 gss_eap_attr_ctx::getPrimaryProvider(void) const
425 {
426     return m_providers[ATTR_TYPE_MIN];
427 }
428
429 /*
430  * Set an attribute
431  */
432 bool
433 gss_eap_attr_ctx::setAttribute(int complete,
434                                const gss_buffer_t attr,
435                                const gss_buffer_t value)
436 {
437     gss_buffer_desc suffix = GSS_C_EMPTY_BUFFER;
438     unsigned int type;
439     gss_eap_attr_provider *provider;
440     bool ret = false;
441
442     decomposeAttributeName(attr, &type, &suffix);
443
444     provider = m_providers[type];
445     if (provider != NULL) {
446         ret = provider->setAttribute(complete,
447                                      (type == ATTR_TYPE_LOCAL) ? attr : &suffix,
448                                      value);
449     }
450
451     return ret;
452 }
453
454 /*
455  * Delete an attrbiute
456  */
457 bool
458 gss_eap_attr_ctx::deleteAttribute(const gss_buffer_t attr)
459 {
460     gss_buffer_desc suffix = GSS_C_EMPTY_BUFFER;
461     unsigned int type;
462     gss_eap_attr_provider *provider;
463     bool ret = false;
464
465     decomposeAttributeName(attr, &type, &suffix);
466
467     provider = m_providers[type];
468     if (provider != NULL) {
469         ret = provider->deleteAttribute(type == ATTR_TYPE_LOCAL ? attr : &suffix);
470     }
471
472     return ret;
473 }
474
475 /*
476  * Enumerate attribute types with callback
477  */
478 bool
479 gss_eap_attr_ctx::getAttributeTypes(gss_eap_attr_enumeration_cb cb, void *data) const
480 {
481     bool ret = false;
482     size_t i;
483
484     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
485         gss_eap_attr_provider *provider = m_providers[i];
486
487         if (provider == NULL)
488             continue;
489
490         ret = provider->getAttributeTypes(cb, data);
491         if (ret == false)
492             break;
493     }
494
495     return ret;
496 }
497
498 struct eap_gss_get_attr_types_args {
499     unsigned int type;
500     gss_buffer_set_t attrs;
501 };
502
503 static bool
504 addAttribute(const gss_eap_attr_ctx *manager,
505              const gss_eap_attr_provider *provider GSSEAP_UNUSED,
506              const gss_buffer_t attribute,
507              void *data)
508 {
509     eap_gss_get_attr_types_args *args = (eap_gss_get_attr_types_args *)data;
510     gss_buffer_desc qualified;
511     OM_uint32 major, minor;
512
513     if (args->type != ATTR_TYPE_LOCAL) {
514         manager->composeAttributeName(args->type, attribute, &qualified);
515         major = gss_add_buffer_set_member(&minor, &qualified, &args->attrs);
516         gss_release_buffer(&minor, &qualified);
517     } else {
518         major = gss_add_buffer_set_member(&minor, attribute, &args->attrs);
519     }
520
521     return GSS_ERROR(major) == false;
522 }
523
524 /*
525  * Enumerate attribute types, output is buffer set
526  */
527 bool
528 gss_eap_attr_ctx::getAttributeTypes(gss_buffer_set_t *attrs)
529 {
530     eap_gss_get_attr_types_args args;
531     OM_uint32 major, minor;
532     bool ret = false;
533     unsigned int i;
534
535     major = gss_create_empty_buffer_set(&minor, attrs);
536     if (GSS_ERROR(major))
537         throw new std::bad_alloc;
538
539     args.attrs = *attrs;
540
541     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
542         gss_eap_attr_provider *provider = m_providers[i];
543
544         args.type = i;
545
546         if (provider == NULL)
547             continue;
548
549         ret = provider->getAttributeTypes(addAttribute, (void *)&args);
550         if (ret == false)
551             break;
552     }
553
554     if (ret == false)
555         gss_release_buffer_set(&minor, attrs);
556
557     return ret;
558 }
559
560 /*
561  * Get attribute with given name
562  */
563 bool
564 gss_eap_attr_ctx::getAttribute(const gss_buffer_t attr,
565                                int *authenticated,
566                                int *complete,
567                                gss_buffer_t value,
568                                gss_buffer_t display_value,
569                                int *more) const
570 {
571     gss_buffer_desc suffix = GSS_C_EMPTY_BUFFER;
572     unsigned int type;
573     gss_eap_attr_provider *provider;
574     bool ret;
575
576     decomposeAttributeName(attr, &type, &suffix);
577
578     provider = m_providers[type];
579     if (provider == NULL)
580         return false;
581
582     ret = provider->getAttribute(type == ATTR_TYPE_LOCAL ? attr : &suffix,
583                                  authenticated, complete,
584                                  value, display_value, more);
585
586     return ret;
587 }
588
589 /*
590  * Map attribute context to C++ object
591  */
592 gss_any_t
593 gss_eap_attr_ctx::mapToAny(int authenticated,
594                            gss_buffer_t type_id) const
595 {
596     unsigned int type;
597     gss_eap_attr_provider *provider;
598     gss_buffer_desc suffix;
599
600     decomposeAttributeName(type_id, &type, &suffix);
601
602     provider = m_providers[type];
603     if (provider == NULL)
604         return (gss_any_t)NULL;
605
606     return provider->mapToAny(authenticated, &suffix);
607 }
608
609 /*
610  * Release mapped context
611  */
612 void
613 gss_eap_attr_ctx::releaseAnyNameMapping(gss_buffer_t type_id,
614                                         gss_any_t input) const
615 {
616     unsigned int type;
617     gss_eap_attr_provider *provider;
618     gss_buffer_desc suffix;
619
620     decomposeAttributeName(type_id, &type, &suffix);
621
622     provider = m_providers[type];
623     if (provider != NULL)
624         provider->releaseAnyNameMapping(&suffix, input);
625 }
626
627 /*
628  * Export attribute context to buffer
629  */
630 void
631 gss_eap_attr_ctx::exportToBuffer(gss_buffer_t buffer) const
632 {
633     OM_uint32 minor;
634     char *s;
635
636     JSONObject obj = jsonRepresentation();
637
638 #if 0
639     obj.dump(stdout, JSON_INDENT(3));
640 #endif
641
642     s = obj.dump(JSON_COMPACT);
643
644     if (GSS_ERROR(makeStringBuffer(&minor, s, buffer)))
645         throw new std::bad_alloc;
646 }
647
648 /*
649  * Return soonest expiry time of providers
650  */
651 time_t
652 gss_eap_attr_ctx::getExpiryTime(void) const
653 {
654     unsigned int i;
655     time_t expiryTime = 0;
656
657     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
658         gss_eap_attr_provider *provider = m_providers[i];
659         time_t providerExpiryTime;
660
661         if (provider == NULL)
662             continue;
663
664         providerExpiryTime = provider->getExpiryTime();
665         if (providerExpiryTime == 0)
666             continue;
667
668         if (expiryTime == 0 || providerExpiryTime < expiryTime)
669             expiryTime = providerExpiryTime;
670     }
671
672     return expiryTime;
673 }
674
675 OM_uint32
676 gss_eap_attr_ctx::mapException(OM_uint32 *minor, std::exception &e) const
677 {
678     unsigned int i;
679     OM_uint32 major;
680
681     /* Errors we handle ourselves */
682     major = GSS_S_FAILURE;
683
684     if (typeid(e) == typeid(std::bad_alloc)) {
685         *minor = ENOMEM;
686         goto cleanup;
687     } else if (typeid(e) == typeid(std::runtime_error)) {
688         major = GSS_S_BAD_NAME;
689         *minor = GSSEAP_BAD_ATTR_TOKEN;
690         goto cleanup;
691     }
692
693     /* Errors we delegate to providers */
694     major = GSS_S_CONTINUE_NEEDED;
695
696     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
697         gss_eap_attr_provider *provider = m_providers[i];
698
699         if (provider == NULL)
700             continue;
701
702         major = provider->mapException(minor, e);
703         if (major != GSS_S_CONTINUE_NEEDED)
704             break;
705     }
706
707     if (major == GSS_S_CONTINUE_NEEDED) {
708         *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
709         major = GSS_S_FAILURE;
710     }
711
712 cleanup:
713 #if 0
714     /* rethrow for now for debugging */
715     throw e;
716 #endif
717
718     assert(GSS_ERROR(major));
719
720     return major;
721 }
722
723 /*
724  * Decompose attribute name into prefix and suffix
725  */
726 void
727 gss_eap_attr_ctx::decomposeAttributeName(const gss_buffer_t attribute,
728                                          gss_buffer_t prefix,
729                                          gss_buffer_t suffix)
730 {
731     char *p = NULL;
732     size_t i;
733
734     for (i = 0; i < attribute->length; i++) {
735         if (((char *)attribute->value)[i] == ' ') {
736             p = (char *)attribute->value + i + 1;
737             break;
738         }
739     }
740
741     prefix->value = attribute->value;
742     prefix->length = i;
743
744     if (p != NULL && *p != '\0')  {
745         suffix->length = attribute->length - 1 - prefix->length;
746         suffix->value = p;
747     } else {
748         suffix->length = 0;
749         suffix->value = NULL;
750     }
751 }
752
753 /*
754  * Decompose attribute name into type and suffix
755  */
756 void
757 gss_eap_attr_ctx::decomposeAttributeName(const gss_buffer_t attribute,
758                                          unsigned int *type,
759                                          gss_buffer_t suffix) const
760 {
761     gss_buffer_desc prefix = GSS_C_EMPTY_BUFFER;
762
763     decomposeAttributeName(attribute, &prefix, suffix);
764     *type = attributePrefixToType(&prefix);
765 }
766
767 /*
768  * Compose attribute name from prefix, suffix; returns C++ string
769  */
770 std::string
771 gss_eap_attr_ctx::composeAttributeName(const gss_buffer_t prefix,
772                                        const gss_buffer_t suffix)
773 {
774     std::string str;
775
776     if (prefix == GSS_C_NO_BUFFER || prefix->length == 0)
777         return str;
778
779     str.append((const char *)prefix->value, prefix->length);
780
781     if (suffix != GSS_C_NO_BUFFER) {
782         str.append(" ");
783         str.append((const char *)suffix->value, suffix->length);
784     }
785
786     return str;
787 }
788
789 /*
790  * Compose attribute name from type, suffix; returns C++ string
791  */
792 std::string
793 gss_eap_attr_ctx::composeAttributeName(unsigned int type,
794                                        const gss_buffer_t suffix)
795 {
796     gss_buffer_desc prefix = attributeTypeToPrefix(type);
797
798     return composeAttributeName(&prefix, suffix);
799 }
800
801 /*
802  * Compose attribute name from prefix, suffix; returns GSS buffer
803  */
804 void
805 gss_eap_attr_ctx::composeAttributeName(const gss_buffer_t prefix,
806                                        const gss_buffer_t suffix,
807                                        gss_buffer_t attribute)
808 {
809     std::string str = composeAttributeName(prefix, suffix);
810
811     if (str.length() != 0) {
812         return duplicateBuffer(str, attribute);
813     } else {
814         attribute->length = 0;
815         attribute->value = NULL;
816     }
817 }
818
819 /*
820  * Compose attribute name from type, suffix; returns GSS buffer
821  */
822 void
823 gss_eap_attr_ctx::composeAttributeName(unsigned int type,
824                                        const gss_buffer_t suffix,
825                                        gss_buffer_t attribute) const
826 {
827     gss_buffer_desc prefix = attributeTypeToPrefix(type);
828
829     return composeAttributeName(&prefix, suffix, attribute);
830 }
831
832 /*
833  * C wrappers
834  */
835 OM_uint32
836 gssEapInquireName(OM_uint32 *minor,
837                   gss_name_t name,
838                   int *name_is_MN,
839                   gss_OID *MN_mech,
840                   gss_buffer_set_t *attrs)
841 {
842     OM_uint32 major;
843
844     if (name_is_MN != NULL)
845         *name_is_MN = (name->mechanismUsed != GSS_C_NULL_OID);
846
847     if (MN_mech != NULL) {
848         major = gssEapCanonicalizeOid(minor, name->mechanismUsed,
849                                       OID_FLAG_NULL_VALID, MN_mech);
850         if (GSS_ERROR(major))
851             return major;
852     }
853
854     if (name->attrCtx == NULL) {
855         *minor = GSSEAP_NO_ATTR_CONTEXT;
856         return GSS_S_UNAVAILABLE;
857     }
858
859     if (GSS_ERROR(gssEapAttrProvidersInit(minor))) {
860         return GSS_S_UNAVAILABLE;
861     }
862
863     try {
864         if (!name->attrCtx->getAttributeTypes(attrs)) {
865             *minor = GSSEAP_NO_ATTR_CONTEXT;
866             return GSS_S_UNAVAILABLE;
867         }
868     } catch (std::exception &e) {
869         return name->attrCtx->mapException(minor, e);
870     }
871
872     return GSS_S_COMPLETE;
873 }
874
875 OM_uint32
876 gssEapGetNameAttribute(OM_uint32 *minor,
877                        gss_name_t name,
878                        gss_buffer_t attr,
879                        int *authenticated,
880                        int *complete,
881                        gss_buffer_t value,
882                        gss_buffer_t display_value,
883                        int *more)
884 {
885     if (authenticated != NULL)
886         *authenticated = 0;
887     if (complete != NULL)
888         *complete = 0;
889
890     if (value != NULL) {
891         value->length = 0;
892         value->value = NULL;
893     }
894
895     if (display_value != NULL) {
896         display_value->length = 0;
897         display_value->value = NULL;
898     }
899
900     if (name->attrCtx == NULL) {
901         *minor = GSSEAP_NO_ATTR_CONTEXT;
902         return GSS_S_UNAVAILABLE;
903     }
904
905     if (GSS_ERROR(gssEapAttrProvidersInit(minor))) {
906         return GSS_S_UNAVAILABLE;
907     }
908
909     try {
910         if (!name->attrCtx->getAttribute(attr, authenticated, complete,
911                                          value, display_value, more)) {
912             *minor = GSSEAP_NO_SUCH_ATTR;
913             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
914                                  (int)attr->length, (char *)attr->value);
915             return GSS_S_UNAVAILABLE;
916         }
917     } catch (std::exception &e) {
918         return name->attrCtx->mapException(minor, e);
919     }
920
921     return GSS_S_COMPLETE;
922 }
923
924 OM_uint32
925 gssEapDeleteNameAttribute(OM_uint32 *minor,
926                           gss_name_t name,
927                           gss_buffer_t attr)
928 {
929     if (name->attrCtx == NULL) {
930         *minor = GSSEAP_NO_ATTR_CONTEXT;
931         return GSS_S_UNAVAILABLE;
932     }
933
934     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
935         return GSS_S_UNAVAILABLE;
936
937     try {
938         if (!name->attrCtx->deleteAttribute(attr)) {
939             *minor = GSSEAP_NO_SUCH_ATTR;
940             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
941                                  (int)attr->length, (char *)attr->value);
942             return GSS_S_UNAVAILABLE;
943         }
944     } catch (std::exception &e) {
945         return name->attrCtx->mapException(minor, e);
946     }
947
948     return GSS_S_COMPLETE;
949 }
950
951 OM_uint32
952 gssEapSetNameAttribute(OM_uint32 *minor,
953                        gss_name_t name,
954                        int complete,
955                        gss_buffer_t attr,
956                        gss_buffer_t value)
957 {
958     if (name->attrCtx == NULL) {
959         *minor = GSSEAP_NO_ATTR_CONTEXT;
960         return GSS_S_UNAVAILABLE;
961     }
962
963     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
964         return GSS_S_UNAVAILABLE;
965
966     try {
967         if (!name->attrCtx->setAttribute(complete, attr, value)) {
968              *minor = GSSEAP_NO_SUCH_ATTR;
969             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
970                                  (int)attr->length, (char *)attr->value);
971             return GSS_S_UNAVAILABLE;
972         }
973     } catch (std::exception &e) {
974         return name->attrCtx->mapException(minor, e);
975     }
976
977     return GSS_S_COMPLETE;
978 }
979
980 OM_uint32
981 gssEapExportAttrContext(OM_uint32 *minor,
982                         gss_name_t name,
983                         gss_buffer_t buffer)
984 {
985     if (name->attrCtx == NULL) {
986         buffer->length = 0;
987         buffer->value = NULL;
988
989         return GSS_S_COMPLETE;
990     }
991
992     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
993         return GSS_S_UNAVAILABLE;
994
995     try {
996         name->attrCtx->exportToBuffer(buffer);
997     } catch (std::exception &e) {
998         return name->attrCtx->mapException(minor, e);
999     }
1000
1001     return GSS_S_COMPLETE;
1002 }
1003
1004 OM_uint32
1005 gssEapImportAttrContext(OM_uint32 *minor,
1006                         gss_buffer_t buffer,
1007                         gss_name_t name)
1008 {
1009     gss_eap_attr_ctx *ctx = NULL;
1010
1011     assert(name->attrCtx == NULL);
1012
1013     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1014         return GSS_S_UNAVAILABLE;
1015
1016     if (buffer->length != 0) {
1017         try {
1018             ctx = new gss_eap_attr_ctx();
1019
1020             if (!ctx->initFromBuffer(buffer)) {
1021                 delete ctx;
1022                 *minor = GSSEAP_BAD_ATTR_TOKEN;
1023                 return GSS_S_BAD_NAME;
1024             }
1025             name->attrCtx = ctx;
1026         } catch (std::exception &e) {
1027             delete ctx;
1028             return name->attrCtx->mapException(minor, e);
1029         }
1030     }
1031
1032     return GSS_S_COMPLETE;
1033 }
1034
1035 OM_uint32
1036 gssEapDuplicateAttrContext(OM_uint32 *minor,
1037                            gss_name_t in,
1038                            gss_name_t out)
1039 {
1040     gss_eap_attr_ctx *ctx = NULL;
1041
1042     assert(out->attrCtx == NULL);
1043
1044     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1045         return GSS_S_UNAVAILABLE;
1046
1047     try {
1048         if (in->attrCtx != NULL) {
1049             ctx = new gss_eap_attr_ctx();
1050             if (!ctx->initFromExistingContext(in->attrCtx)) {
1051                 delete ctx;
1052                 *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
1053                 return GSS_S_FAILURE;
1054             }
1055             out->attrCtx = ctx;
1056         }
1057     } catch (std::exception &e) {
1058         delete ctx;
1059         return in->attrCtx->mapException(minor, e);
1060     }
1061
1062     return GSS_S_COMPLETE;
1063 }
1064
1065 OM_uint32
1066 gssEapMapNameToAny(OM_uint32 *minor,
1067                    gss_name_t name,
1068                    int authenticated,
1069                    gss_buffer_t type_id,
1070                    gss_any_t *output)
1071 {
1072     if (name->attrCtx == NULL) {
1073         *minor = GSSEAP_NO_ATTR_CONTEXT;
1074         return GSS_S_UNAVAILABLE;
1075     }
1076
1077     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1078         return GSS_S_UNAVAILABLE;
1079
1080     try {
1081         *output = name->attrCtx->mapToAny(authenticated, type_id);
1082     } catch (std::exception &e) {
1083         return name->attrCtx->mapException(minor, e);
1084     }
1085
1086     return GSS_S_COMPLETE;
1087 }
1088
1089 OM_uint32
1090 gssEapReleaseAnyNameMapping(OM_uint32 *minor,
1091                             gss_name_t name,
1092                             gss_buffer_t type_id,
1093                             gss_any_t *input)
1094 {
1095     if (name->attrCtx == NULL) {
1096         *minor = GSSEAP_NO_ATTR_CONTEXT;
1097         return GSS_S_UNAVAILABLE;
1098     }
1099
1100     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1101         return GSS_S_UNAVAILABLE;
1102
1103     try {
1104         if (*input != NULL)
1105             name->attrCtx->releaseAnyNameMapping(type_id, *input);
1106         *input = NULL;
1107     } catch (std::exception &e) {
1108         return name->attrCtx->mapException(minor, e);
1109     }
1110
1111     return GSS_S_COMPLETE;
1112 }
1113
1114 OM_uint32
1115 gssEapReleaseAttrContext(OM_uint32 *minor,
1116                          gss_name_t name)
1117 {
1118     if (name->attrCtx != NULL)
1119         delete name->attrCtx;
1120
1121     *minor = 0;
1122     return GSS_S_COMPLETE;
1123 }
1124
1125 /*
1126  * Public accessor for initialisng a context from a GSS context. Also
1127  * sets expiry time on GSS context as a side-effect.
1128  */
1129 OM_uint32
1130 gssEapCreateAttrContext(OM_uint32 *minor,
1131                         gss_cred_id_t gssCred,
1132                         gss_ctx_id_t gssCtx,
1133                         struct gss_eap_attr_ctx **pAttrContext,
1134                         time_t *pExpiryTime)
1135 {
1136     gss_eap_attr_ctx *ctx = NULL;
1137     OM_uint32 major;
1138
1139     assert(gssCtx != GSS_C_NO_CONTEXT);
1140
1141     major = gssEapAttrProvidersInit(minor);
1142     if (GSS_ERROR(major))
1143         return major;
1144
1145     *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
1146     major = GSS_S_FAILURE;
1147
1148     try {
1149         *pAttrContext = ctx = new gss_eap_attr_ctx();
1150         if (ctx->initFromGssContext(gssCred, gssCtx)) {
1151             *minor = 0;
1152             major = GSS_S_COMPLETE;
1153         }
1154     } catch (std::exception &e) {
1155         if (ctx != NULL)
1156             major = ctx->mapException(minor, e);
1157     }
1158
1159     if (major == GSS_S_COMPLETE) {
1160         *pExpiryTime = ctx->getExpiryTime();
1161     } else {
1162         delete ctx;
1163         *pAttrContext = NULL;
1164     }
1165
1166     return major;
1167 }