X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;f=mech_eap%2Futil_shib.cpp;h=b1464f7b59fd13716fed5450f129f6dec22734c0;hb=156cea08f449a20a965e5e004460bbd32361f106;hp=f1d68ac2d61de6f4681d7a219ae7b94b20c5492e;hpb=f9c13d9f7815f21b258b027279b9665a9199d2a7;p=moonshot.git diff --git a/mech_eap/util_shib.cpp b/mech_eap/util_shib.cpp index f1d68ac..b1464f7 100644 --- a/mech_eap/util_shib.cpp +++ b/mech_eap/util_shib.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2010, JANET(UK) + * Copyright (c) 2011, JANET(UK) * All rights reserved. * * Redistribution and use in source and binary forms, with or without @@ -31,7 +31,7 @@ */ /* * Copyright 2001-2009 Internet2 - * + * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at @@ -45,151 +45,163 @@ * limitations under the License. */ -#include "gssapiP_eap.h" +/* + * Local attribute provider implementation. + */ + +#include + +#include -#include #include -#include -#include -#include #include -#include -#include -#include -#include +#include -#include -#include -#include -#include -#include -#include +#include -#include "resolver.h" +#include "gssapiP_eap.h" using namespace shibsp; using namespace shibresolver; using namespace opensaml::saml2md; using namespace opensaml; -using namespace xmltooling::logging; using namespace xmltooling; -using namespace xercesc; using namespace std; +gss_eap_shib_attr_provider::gss_eap_shib_attr_provider(void) +{ + m_initialized = false; + m_authenticated = false; +} + +gss_eap_shib_attr_provider::~gss_eap_shib_attr_provider(void) +{ + for_each(m_attributes.begin(), + m_attributes.end(), + xmltooling::cleanup()) + ; +} + bool -gss_eap_shib_attr_provider::initFromExistingContext(const gss_eap_attr_ctx *manager, +gss_eap_shib_attr_provider::initWithExistingContext(const gss_eap_attr_ctx *manager, const gss_eap_attr_provider *ctx) { const gss_eap_shib_attr_provider *shib; - if (!gss_eap_attr_provider::initFromExistingContext(manager, ctx)) + if (!gss_eap_attr_provider::initWithExistingContext(manager, ctx)) { return false; + } + + m_authenticated = false; shib = static_cast(ctx); - if (shib != NULL) + if (shib != NULL) { m_attributes = duplicateAttributes(shib->getAttributes()); + m_authenticated = shib->authenticated(); + } + + m_initialized = true; return true; } -bool -addRadiusAttribute(const gss_eap_attr_provider *provider, - const gss_buffer_t attribute, - void *data) +static OM_uint32 +exportMechSecContext(OM_uint32 *minor, + gss_ctx_id_t gssCtx, + gss_buffer_t mechContext) { - const gss_eap_shib_attr_provider *shib; - const gss_eap_radius_attr_provider *radius; - int authenticated, complete, more = -1; - vector attributeIds(1); - SimpleAttribute *a; - - radius = static_cast(provider); - shib = static_cast(data); - - assert(radius != NULL && shib != NULL); - - string attributeName = - gss_eap_attr_ctx::composeAttributeName(ATTR_TYPE_RADIUS, attribute); - - attributeIds.push_back(attributeName); - a = new SimpleAttribute(attributeIds); - if (a == NULL) - return false; - - while (more != 0) { - gss_buffer_desc value = GSS_C_EMPTY_BUFFER; - OM_uint32 minor; - - if (!radius->getAttribute(attribute, - &authenticated, - &complete, - &value, - NULL, - &more)) - return false; - - string attributeValue((char *)value.value, value.length); - a->getValues().push_back(attributeValue); - - gss_release_buffer(&minor, &value); + OM_uint32 major; + gss_buffer_desc exportedCtx; + unsigned char *p; + + assert(gssCtx->mechanismUsed != GSS_C_NO_OID); + + major = gssEapExportSecContext(minor, gssCtx, &exportedCtx); + if (GSS_ERROR(major)) + return major; + + /* + * gss_import_sec_context expects the exported security context token + * to be tagged with the mechanism OID; in Heimdal and MIT, this is + * done by the mechglue, so if we are subverting the mechglue we need + * to add it ourselves. + */ + mechContext->length = 4 + gssCtx->mechanismUsed->length + exportedCtx.length; + mechContext->value = p = (unsigned char *)GSSEAP_MALLOC(mechContext->length); + if (mechContext->value == NULL) { + gss_release_buffer(minor, &exportedCtx); + throw std::bad_alloc(); } - shib->getAttributes().push_back(a); + p = store_oid(gssCtx->mechanismUsed, p); + memcpy(p, exportedCtx.value, exportedCtx.length); - return true; + gss_release_buffer(minor, &exportedCtx); + + return GSS_S_COMPLETE; } bool -gss_eap_shib_attr_provider::initFromGssContext(const gss_eap_attr_ctx *manager, +gss_eap_shib_attr_provider::initWithGssContext(const gss_eap_attr_ctx *manager, const gss_cred_id_t gssCred, const gss_ctx_id_t gssCtx) { - const gss_eap_saml_assertion_provider *saml; - const gss_eap_radius_attr_provider *radius; - gss_buffer_desc nameBuf = GSS_C_EMPTY_BUFFER; - ShibbolethResolver *resolver = NULL; - OM_uint32 minor; - - if (!gss_eap_attr_provider::initFromGssContext(manager, gssCred, gssCtx)) + if (!gss_eap_attr_provider::initWithGssContext(manager, gssCred, gssCtx)) return false; - saml = static_cast - (manager->getProvider(ATTR_TYPE_SAML_ASSERTION)); - radius = static_cast - (manager->getProvider(ATTR_TYPE_RADIUS)); + auto_ptr resolver(ShibbolethResolver::create()); + /* + * For now, leave ApplicationID defaulted. + * Later on, we could allow this via config option to the mechanism + * or rely on an SPRequest interface to pass in a URI identifying the + * acceptor. + */ +#if 0 + gss_buffer_desc nameBuf = GSS_C_EMPTY_BUFFER; if (gssCred != GSS_C_NO_CREDENTIAL && - gss_display_name(&minor, gssCred->name, &nameBuf, NULL) == GSS_S_COMPLETE) + gssEapDisplayName(&minor, gssCred->name, &nameBuf, NULL) == GSS_S_COMPLETE) { resolver->setApplicationID((const char *)nameBuf.value); + gss_release_buffer(&minor, &nameBuf); + } +#endif + + gss_buffer_desc mechContext = GSS_C_EMPTY_BUFFER; + OM_uint32 major, minor; + major = exportMechSecContext(&minor, gssCtx, &mechContext); + if (major == GSS_S_COMPLETE) { + resolver->addToken(&mechContext); + gss_release_buffer(&minor, &mechContext); + } - if (saml != NULL && saml->getAssertion() != NULL) + const gss_eap_saml_assertion_provider *saml; + saml = static_cast + (m_manager->getProvider(ATTR_TYPE_SAML_ASSERTION)); + if (saml != NULL && saml->getAssertion() != NULL) { resolver->addToken(saml->getAssertion()); + } - if (radius != NULL) - radius->getAttributeTypes(addRadiusAttribute, (void *)this); - - resolver->resolveAttributes(m_attributes); - - gss_release_buffer(&minor, &nameBuf); + try { + resolver->resolve(); + m_attributes = resolver->getResolvedAttributes(); + resolver->getResolvedAttributes().clear(); + } catch (exception &e) { + return false; + } - delete resolver; + m_authenticated = true; + m_initialized = true; return true; } -gss_eap_shib_attr_provider::~gss_eap_shib_attr_provider(void) -{ - for_each(m_attributes.begin(), - m_attributes.end(), - xmltooling::cleanup()) - ; -} - -int +ssize_t gss_eap_shib_attr_provider::getAttributeIndex(const gss_buffer_t attr) const { int i = 0; + assert(m_initialized); + for (vector::const_iterator a = m_attributes.begin(); a != m_attributes.end(); ++a) @@ -207,41 +219,51 @@ gss_eap_shib_attr_provider::getAttributeIndex(const gss_buffer_t attr) const return -1; } -void -gss_eap_shib_attr_provider::setAttribute(int complete, +bool +gss_eap_shib_attr_provider::setAttribute(int complete GSSEAP_UNUSED, const gss_buffer_t attr, const gss_buffer_t value) { string attrStr((char *)attr->value, attr->length); - vector ids(1); - - ids.push_back(attrStr); - + vector ids(1, attrStr); SimpleAttribute *a = new SimpleAttribute(ids); + assert(m_initialized); + if (value->length != 0) { string valueStr((char *)value->value, value->length); - a->getValues().push_back(valueStr); + a->getValues().push_back(valueStr); } m_attributes.push_back(a); + m_authenticated = false; + + return true; } -void +bool gss_eap_shib_attr_provider::deleteAttribute(const gss_buffer_t attr) { int i; + assert(m_initialized); + i = getAttributeIndex(attr); if (i >= 0) m_attributes.erase(m_attributes.begin() + i); + + m_authenticated = false; + + return true; } bool gss_eap_shib_attr_provider::getAttributeTypes(gss_eap_attr_enumeration_cb addAttribute, void *data) const { + assert(m_initialized); + for (vector::const_iterator a = m_attributes.begin(); a != m_attributes.end(); ++a) @@ -251,7 +273,7 @@ gss_eap_shib_attr_provider::getAttributeTypes(gss_eap_attr_enumeration_cb addAtt attribute.value = (void *)((*a)->getId()); attribute.length = strlen((char *)attribute.value); - if (!addAttribute(this, &attribute, data)) + if (!addAttribute(m_manager, this, &attribute, data)) return false; } @@ -263,6 +285,8 @@ gss_eap_shib_attr_provider::getAttribute(const gss_buffer_t attr) const { const Attribute *ret = NULL; + assert(m_initialized); + for (vector::const_iterator a = m_attributes.begin(); a != m_attributes.end(); ++a) @@ -293,38 +317,56 @@ gss_eap_shib_attr_provider::getAttribute(const gss_buffer_t attr, { const Attribute *shibAttr = NULL; gss_buffer_desc buf; + int nvalues, i = *more; + + assert(m_initialized); + + *more = 0; shibAttr = getAttribute(attr); if (shibAttr == NULL) return false; - if (*more == -1) { - *more = 0; - } else if (*more >= (int)shibAttr->valueCount()) { - *more = 0; - return true; - } + nvalues = shibAttr->valueCount(); - buf.value = (void *)shibAttr->getString(*more); + if (i == -1) + i = 0; + if (i >= nvalues) + return false; + + buf.value = (void *)shibAttr->getSerializedValues()[*more].c_str(); buf.length = strlen((char *)buf.value); - duplicateBuffer(buf, value); + if (buf.length != 0) { + if (value != NULL) + duplicateBuffer(buf, value); + + if (display_value != NULL) + duplicateBuffer(buf, display_value); + } - if (display_value != NULL) - duplicateBuffer(buf, display_value); - - *authenticated = TRUE; - *complete = FALSE; + if (authenticated != NULL) + *authenticated = m_authenticated; + if (complete != NULL) + *complete = false; + + if (nvalues > ++i) + *more = i; return true; } gss_any_t gss_eap_shib_attr_provider::mapToAny(int authenticated, - gss_buffer_t type_id) const + gss_buffer_t type_id GSSEAP_UNUSED) const { gss_any_t output; + assert(m_initialized); + + if (authenticated && !m_authenticated) + return (gss_any_t)NULL; + vector v = duplicateAttributes(m_attributes); output = (gss_any_t)new vector (v); @@ -333,62 +375,74 @@ gss_eap_shib_attr_provider::mapToAny(int authenticated, } void -gss_eap_shib_attr_provider::releaseAnyNameMapping(gss_buffer_t type_id, +gss_eap_shib_attr_provider::releaseAnyNameMapping(gss_buffer_t type_id GSSEAP_UNUSED, gss_any_t input) const { + assert(m_initialized); + vector *v = ((vector *)input); delete v; } -void -gss_eap_shib_attr_provider::exportToBuffer(gss_buffer_t buffer) const +const char * +gss_eap_shib_attr_provider::prefix(void) const +{ + return NULL; +} + +const char * +gss_eap_shib_attr_provider::name(void) const +{ + return "local"; +} + +JSONObject +gss_eap_shib_attr_provider::jsonRepresentation(void) const { - DDF attrs(NULL); + JSONObject obj; - buffer->length = 0; - buffer->value = NULL; + if (m_initialized == false) + return obj; /* don't export incomplete context */ + + JSONObject jattrs = JSONObject::array(); for (vector::const_iterator a = m_attributes.begin(); a != m_attributes.end(); ++a) { DDF attr = (*a)->marshall(); - attrs.add(attr); + JSONObject jattr = JSONObject::ddf(attr); + jattrs.append(jattr); } - ostringstream sink; - sink << attrs; - string str = sink.str(); + obj.set("attributes", jattrs); - duplicateBuffer(str, buffer); + obj.set("authenticated", m_authenticated); - attrs.destroy(); + return obj; } bool -gss_eap_shib_attr_provider::initFromBuffer(const gss_eap_attr_ctx *ctx, - const gss_buffer_t buffer) +gss_eap_shib_attr_provider::initWithJsonObject(const gss_eap_attr_ctx *ctx, + JSONObject &obj) { - if (!gss_eap_attr_provider::initFromBuffer(ctx, buffer)) + if (!gss_eap_attr_provider::initWithJsonObject(ctx, obj)) return false; - if (buffer->length == 0) - return true; - + assert(m_authenticated == false); assert(m_attributes.size() == 0); - DDF attrs(NULL); - string str((const char *)buffer->value, buffer->length); - istringstream source(str); + JSONObject jattrs = obj["attributes"]; + size_t nelems = jattrs.size(); - source >> attrs; + for (size_t i = 0; i < nelems; i++) { + JSONObject jattr = jattrs.get(i); - DDF attr = attrs.first(); - while (!attr.isnull()) { + DDF attr = jattr.ddf(); Attribute *attribute = Attribute::unmarshall(attr); m_attributes.push_back(attribute); - attr = attrs.next(); } - attrs.destroy(); + m_authenticated = obj["authenticated"].integer(); + m_initialized = true; return true; } @@ -396,12 +450,11 @@ gss_eap_shib_attr_provider::initFromBuffer(const gss_eap_attr_ctx *ctx, bool gss_eap_shib_attr_provider::init(void) { - if (!ShibbolethResolver::init()) + if (SPConfig::getConfig().getFeatures() == 0 && + ShibbolethResolver::init() == false) return false; - gss_eap_attr_ctx::registerProvider(ATTR_TYPE_LOCAL, - NULL, - gss_eap_shib_attr_provider::createAttrContext); + gss_eap_attr_ctx::registerProvider(ATTR_TYPE_LOCAL, createAttrContext); return true; } @@ -409,8 +462,32 @@ gss_eap_shib_attr_provider::init(void) void gss_eap_shib_attr_provider::finalize(void) { - ShibbolethResolver::term(); gss_eap_attr_ctx::unregisterProvider(ATTR_TYPE_LOCAL); + ShibbolethResolver::term(); +} + +OM_uint32 +gss_eap_shib_attr_provider::mapException(OM_uint32 *minor, + std::exception &e) const +{ + if (typeid(e) == typeid(AttributeException)) + *minor = GSSEAP_SHIB_ATTR_FAILURE; + else if (typeid(e) == typeid(AttributeExtractionException)) + *minor = GSSEAP_SHIB_ATTR_EXTRACT_FAILURE; + else if (typeid(e) == typeid(AttributeFilteringException)) + *minor = GSSEAP_SHIB_ATTR_FILTER_FAILURE; + else if (typeid(e) == typeid(AttributeResolutionException)) + *minor = GSSEAP_SHIB_ATTR_RESOLVE_FAILURE; + else if (typeid(e) == typeid(ConfigurationException)) + *minor = GSSEAP_SHIB_CONFIG_FAILURE; + else if (typeid(e) == typeid(ListenerException)) + *minor = GSSEAP_SHIB_LISTENER_FAILURE; + else + return GSS_S_CONTINUE_NEEDED; + + gssEapSaveStatusInfo(*minor, "%s", e.what()); + + return GSS_S_FAILURE; } gss_eap_attr_provider * @@ -422,10 +499,8 @@ gss_eap_shib_attr_provider::createAttrContext(void) Attribute * gss_eap_shib_attr_provider::duplicateAttribute(const Attribute *src) { - Attribute *attribute; - DDF obj = src->marshall(); - attribute = Attribute::unmarshall(obj); + Attribute *attribute = Attribute::unmarshall(obj); obj.destroy(); return attribute; @@ -443,3 +518,22 @@ gss_eap_shib_attr_provider::duplicateAttributes(const vector src) return dst; } + +OM_uint32 +gssEapLocalAttrProviderInit(OM_uint32 *minor) +{ + if (!gss_eap_shib_attr_provider::init()) { + *minor = GSSEAP_SHIB_INIT_FAILURE; + return GSS_S_FAILURE; + } + return GSS_S_COMPLETE; +} + +OM_uint32 +gssEapLocalAttrProviderFinalize(OM_uint32 *minor) +{ + gss_eap_shib_attr_provider::finalize(); + + *minor = 0; + return GSS_S_COMPLETE; +}