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);
 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     if (typeid(e) == typeid(std::bad_alloc)) {
 683         major = GSS_S_FAILURE;
 684         *minor = ENOMEM;
 685         goto cleanup;
 686     } else if (typeid(e) == typeid(std::runtime_error)) {
 687         major = GSS_S_BAD_NAME;
 688         *minor = GSSEAP_BAD_ATTR_TOKEN;
 689         goto cleanup;
 690     } else if (this == NULL) {
 691         major = GSS_S_FAILURE;
 692         goto cleanup;
 693     }
 694
 695     /* Errors we delegate to providers */
 696     major = GSS_S_CONTINUE_NEEDED;
 697
 698     for (i = ATTR_TYPE_MIN; i <= ATTR_TYPE_MAX; i++) {
 699         gss_eap_attr_provider *provider = m_providers[i];
 700
 701         if (provider == NULL)
 702             continue;
 703
 704         major = provider->mapException(minor, e);
 705         if (major != GSS_S_CONTINUE_NEEDED)
 706             break;
 707     }
 708
 709     if (major == GSS_S_CONTINUE_NEEDED) {
 710         *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
 711         major = GSS_S_FAILURE;
 712     }
 713
 714 cleanup:
 715 #if 0
 716     /* rethrow for now for debugging */
 717     throw e;
 718 #endif
 719
 720     assert(GSS_ERROR(major));
 721
 722     return major;
 723 }
 724
 725 /*
 726  * Decompose attribute name into prefix and suffix
 727  */
 728 void
 729 gss_eap_attr_ctx::decomposeAttributeName(const gss_buffer_t attribute,
 730                                          gss_buffer_t prefix,
 731                                          gss_buffer_t suffix)
 732 {
 733     char *p = NULL;
 734     size_t i;
 735
 736     for (i = 0; i < attribute->length; i++) {
 737         if (((char *)attribute->value)[i] == ' ') {
 738             p = (char *)attribute->value + i + 1;
 739             break;
 740         }
 741     }
 742
 743     prefix->value = attribute->value;
 744     prefix->length = i;
 745
 746     if (p != NULL && *p != '\0')  {
 747         suffix->length = attribute->length - 1 - prefix->length;
 748         suffix->value = p;
 749     } else {
 750         suffix->length = 0;
 751         suffix->value = NULL;
 752     }
 753 }
 754
 755 /*
 756  * Decompose attribute name into type and suffix
 757  */
 758 void
 759 gss_eap_attr_ctx::decomposeAttributeName(const gss_buffer_t attribute,
 760                                          unsigned int *type,
 761                                          gss_buffer_t suffix) const
 762 {
 763     gss_buffer_desc prefix = GSS_C_EMPTY_BUFFER;
 764
 765     decomposeAttributeName(attribute, &prefix, suffix);
 766     *type = attributePrefixToType(&prefix);
 767 }
 768
 769 /*
 770  * Compose attribute name from prefix, suffix; returns C++ string
 771  */
 772 std::string
 773 gss_eap_attr_ctx::composeAttributeName(const gss_buffer_t prefix,
 774                                        const gss_buffer_t suffix)
 775 {
 776     std::string str;
 777
 778     if (prefix == GSS_C_NO_BUFFER || prefix->length == 0)
 779         return str;
 780
 781     str.append((const char *)prefix->value, prefix->length);
 782
 783     if (suffix != GSS_C_NO_BUFFER) {
 784         str.append(" ");
 785         str.append((const char *)suffix->value, suffix->length);
 786     }
 787
 788     return str;
 789 }
 790
 791 /*
 792  * Compose attribute name from type, suffix; returns C++ string
 793  */
 794 std::string
 795 gss_eap_attr_ctx::composeAttributeName(unsigned int type,
 796                                        const gss_buffer_t suffix)
 797 {
 798     gss_buffer_desc prefix = attributeTypeToPrefix(type);
 799
 800     return composeAttributeName(&prefix, suffix);
 801 }
 802
 803 /*
 804  * Compose attribute name from prefix, suffix; returns GSS buffer
 805  */
 806 void
 807 gss_eap_attr_ctx::composeAttributeName(const gss_buffer_t prefix,
 808                                        const gss_buffer_t suffix,
 809                                        gss_buffer_t attribute)
 810 {
 811     std::string str = composeAttributeName(prefix, suffix);
 812
 813     if (str.length() != 0) {
 814         return duplicateBuffer(str, attribute);
 815     } else {
 816         attribute->length = 0;
 817         attribute->value = NULL;
 818     }
 819 }
 820
 821 /*
 822  * Compose attribute name from type, suffix; returns GSS buffer
 823  */
 824 void
 825 gss_eap_attr_ctx::composeAttributeName(unsigned int type,
 826                                        const gss_buffer_t suffix,
 827                                        gss_buffer_t attribute) const
 828 {
 829     gss_buffer_desc prefix = attributeTypeToPrefix(type);
 830
 831     return composeAttributeName(&prefix, suffix, attribute);
 832 }
 833
 834 /*
 835  * C wrappers
 836  */
 837 OM_uint32
 838 gssEapInquireName(OM_uint32 *minor,
 839                   gss_name_t name,
 840                   int *name_is_MN,
 841                   gss_OID *MN_mech,
 842                   gss_buffer_set_t *attrs)
 843 {
 844     OM_uint32 major;
 845
 846     if (name_is_MN != NULL)
 847         *name_is_MN = (name->mechanismUsed != GSS_C_NULL_OID);
 848
 849     if (MN_mech != NULL) {
 850         major = gssEapCanonicalizeOid(minor, name->mechanismUsed,
 851                                       OID_FLAG_NULL_VALID, MN_mech);
 852         if (GSS_ERROR(major))
 853             return major;
 854     }
 855
 856     if (name->attrCtx == NULL) {
 857         *minor = GSSEAP_NO_ATTR_CONTEXT;
 858         return GSS_S_UNAVAILABLE;
 859     }
 860
 861     if (GSS_ERROR(gssEapAttrProvidersInit(minor))) {
 862         return GSS_S_UNAVAILABLE;
 863     }
 864
 865     try {
 866         if (!name->attrCtx->getAttributeTypes(attrs)) {
 867             *minor = GSSEAP_NO_ATTR_CONTEXT;
 868             return GSS_S_UNAVAILABLE;
 869         }
 870     } catch (std::exception &e) {
 871         return name->attrCtx->mapException(minor, e);
 872     }
 873
 874     return GSS_S_COMPLETE;
 875 }
 876
 877 OM_uint32
 878 gssEapGetNameAttribute(OM_uint32 *minor,
 879                        gss_name_t name,
 880                        gss_buffer_t attr,
 881                        int *authenticated,
 882                        int *complete,
 883                        gss_buffer_t value,
 884                        gss_buffer_t display_value,
 885                        int *more)
 886 {
 887     if (authenticated != NULL)
 888         *authenticated = 0;
 889     if (complete != NULL)
 890         *complete = 0;
 891
 892     if (value != NULL) {
 893         value->length = 0;
 894         value->value = NULL;
 895     }
 896
 897     if (display_value != NULL) {
 898         display_value->length = 0;
 899         display_value->value = NULL;
 900     }
 901
 902     if (name->attrCtx == NULL) {
 903         *minor = GSSEAP_NO_ATTR_CONTEXT;
 904         return GSS_S_UNAVAILABLE;
 905     }
 906
 907     if (GSS_ERROR(gssEapAttrProvidersInit(minor))) {
 908         return GSS_S_UNAVAILABLE;
 909     }
 910
 911     try {
 912         if (!name->attrCtx->getAttribute(attr, authenticated, complete,
 913                                          value, display_value, more)) {
 914             *minor = GSSEAP_NO_SUCH_ATTR;
 915             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
 916                                  (int)attr->length, (char *)attr->value);
 917             return GSS_S_UNAVAILABLE;
 918         }
 919     } catch (std::exception &e) {
 920         return name->attrCtx->mapException(minor, e);
 921     }
 922
 923     return GSS_S_COMPLETE;
 924 }
 925
 926 OM_uint32
 927 gssEapDeleteNameAttribute(OM_uint32 *minor,
 928                           gss_name_t name,
 929                           gss_buffer_t attr)
 930 {
 931     if (name->attrCtx == NULL) {
 932         *minor = GSSEAP_NO_ATTR_CONTEXT;
 933         return GSS_S_UNAVAILABLE;
 934     }
 935
 936     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
 937         return GSS_S_UNAVAILABLE;
 938
 939     try {
 940         if (!name->attrCtx->deleteAttribute(attr)) {
 941             *minor = GSSEAP_NO_SUCH_ATTR;
 942             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
 943                                  (int)attr->length, (char *)attr->value);
 944             return GSS_S_UNAVAILABLE;
 945         }
 946     } catch (std::exception &e) {
 947         return name->attrCtx->mapException(minor, e);
 948     }
 949
 950     return GSS_S_COMPLETE;
 951 }
 952
 953 OM_uint32
 954 gssEapSetNameAttribute(OM_uint32 *minor,
 955                        gss_name_t name,
 956                        int complete,
 957                        gss_buffer_t attr,
 958                        gss_buffer_t value)
 959 {
 960     if (name->attrCtx == NULL) {
 961         *minor = GSSEAP_NO_ATTR_CONTEXT;
 962         return GSS_S_UNAVAILABLE;
 963     }
 964
 965     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
 966         return GSS_S_UNAVAILABLE;
 967
 968     try {
 969         if (!name->attrCtx->setAttribute(complete, attr, value)) {
 970              *minor = GSSEAP_NO_SUCH_ATTR;
 971             gssEapSaveStatusInfo(*minor, "Unknown naming attribute %.*s",
 972                                  (int)attr->length, (char *)attr->value);
 973             return GSS_S_UNAVAILABLE;
 974         }
 975     } catch (std::exception &e) {
 976         return name->attrCtx->mapException(minor, e);
 977     }
 978
 979     return GSS_S_COMPLETE;
 980 }
 981
 982 OM_uint32
 983 gssEapExportAttrContext(OM_uint32 *minor,
 984                         gss_name_t name,
 985                         gss_buffer_t buffer)
 986 {
 987     if (name->attrCtx == NULL) {
 988         buffer->length = 0;
 989         buffer->value = NULL;
 990
 991         return GSS_S_COMPLETE;
 992     }
 993
 994     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
 995         return GSS_S_UNAVAILABLE;
 996
 997     try {
 998         name->attrCtx->exportToBuffer(buffer);
 999     } catch (std::exception &e) {
1000         return name->attrCtx->mapException(minor, e);
1001     }
1002
1003     return GSS_S_COMPLETE;
1004 }
1005
1006 OM_uint32
1007 gssEapImportAttrContext(OM_uint32 *minor,
1008                         gss_buffer_t buffer,
1009                         gss_name_t name)
1010 {
1011     gss_eap_attr_ctx *ctx = NULL;
1012
1013     assert(name->attrCtx == NULL);
1014
1015     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1016         return GSS_S_UNAVAILABLE;
1017
1018     if (buffer->length != 0) {
1019         try {
1020             ctx = new gss_eap_attr_ctx();
1021
1022             if (!ctx->initFromBuffer(buffer)) {
1023                 delete ctx;
1024                 *minor = GSSEAP_BAD_ATTR_TOKEN;
1025                 return GSS_S_BAD_NAME;
1026             }
1027             name->attrCtx = ctx;
1028         } catch (std::exception &e) {
1029             delete ctx;
1030             return name->attrCtx->mapException(minor, e);
1031         }
1032     }
1033
1034     return GSS_S_COMPLETE;
1035 }
1036
1037 OM_uint32
1038 gssEapDuplicateAttrContext(OM_uint32 *minor,
1039                            gss_name_t in,
1040                            gss_name_t out)
1041 {
1042     gss_eap_attr_ctx *ctx = NULL;
1043
1044     assert(out->attrCtx == NULL);
1045
1046     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1047         return GSS_S_UNAVAILABLE;
1048
1049     try {
1050         if (in->attrCtx != NULL) {
1051             ctx = new gss_eap_attr_ctx();
1052             if (!ctx->initFromExistingContext(in->attrCtx)) {
1053                 delete ctx;
1054                 *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
1055                 return GSS_S_FAILURE;
1056             }
1057             out->attrCtx = ctx;
1058         }
1059     } catch (std::exception &e) {
1060         delete ctx;
1061         return in->attrCtx->mapException(minor, e);
1062     }
1063
1064     return GSS_S_COMPLETE;
1065 }
1066
1067 OM_uint32
1068 gssEapMapNameToAny(OM_uint32 *minor,
1069                    gss_name_t name,
1070                    int authenticated,
1071                    gss_buffer_t type_id,
1072                    gss_any_t *output)
1073 {
1074     if (name->attrCtx == NULL) {
1075         *minor = GSSEAP_NO_ATTR_CONTEXT;
1076         return GSS_S_UNAVAILABLE;
1077     }
1078
1079     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1080         return GSS_S_UNAVAILABLE;
1081
1082     try {
1083         *output = name->attrCtx->mapToAny(authenticated, type_id);
1084     } catch (std::exception &e) {
1085         return name->attrCtx->mapException(minor, e);
1086     }
1087
1088     return GSS_S_COMPLETE;
1089 }
1090
1091 OM_uint32
1092 gssEapReleaseAnyNameMapping(OM_uint32 *minor,
1093                             gss_name_t name,
1094                             gss_buffer_t type_id,
1095                             gss_any_t *input)
1096 {
1097     if (name->attrCtx == NULL) {
1098         *minor = GSSEAP_NO_ATTR_CONTEXT;
1099         return GSS_S_UNAVAILABLE;
1100     }
1101
1102     if (GSS_ERROR(gssEapAttrProvidersInit(minor)))
1103         return GSS_S_UNAVAILABLE;
1104
1105     try {
1106         if (*input != NULL)
1107             name->attrCtx->releaseAnyNameMapping(type_id, *input);
1108         *input = NULL;
1109     } catch (std::exception &e) {
1110         return name->attrCtx->mapException(minor, e);
1111     }
1112
1113     return GSS_S_COMPLETE;
1114 }
1115
1116 OM_uint32
1117 gssEapReleaseAttrContext(OM_uint32 *minor,
1118                          gss_name_t name)
1119 {
1120     if (name->attrCtx != NULL)
1121         delete name->attrCtx;
1122
1123     *minor = 0;
1124     return GSS_S_COMPLETE;
1125 }
1126
1127 /*
1128  * Public accessor for initialisng a context from a GSS context. Also
1129  * sets expiry time on GSS context as a side-effect.
1130  */
1131 OM_uint32
1132 gssEapCreateAttrContext(OM_uint32 *minor,
1133                         gss_cred_id_t gssCred,
1134                         gss_ctx_id_t gssCtx,
1135                         struct gss_eap_attr_ctx **pAttrContext,
1136                         time_t *pExpiryTime)
1137 {
1138     gss_eap_attr_ctx *ctx = NULL;
1139     OM_uint32 major;
1140
1141     assert(gssCtx != GSS_C_NO_CONTEXT);
1142
1143     major = gssEapAttrProvidersInit(minor);
1144     if (GSS_ERROR(major))
1145         return major;
1146
1147     *minor = GSSEAP_ATTR_CONTEXT_FAILURE;
1148     major = GSS_S_FAILURE;
1149
1150     try {
1151         *pAttrContext = ctx = new gss_eap_attr_ctx();
1152         if (ctx->initFromGssContext(gssCred, gssCtx)) {
1153             *minor = 0;
1154             major = GSS_S_COMPLETE;
1155         }
1156     } catch (std::exception &e) {
1157         if (ctx != NULL)
1158             major = ctx->mapException(minor, e);
1159     }
1160
1161     if (major == GSS_S_COMPLETE) {
1162         *pExpiryTime = ctx->getExpiryTime();
1163     } else {
1164         delete ctx;
1165         *pAttrContext = NULL;
1166     }
1167
1168     return major;
1169 }