shibsp/attribute/resolver/impl/QueryAttributeResolver.cpp

   1 /*
   2  *  Copyright 2001-2007 Internet2
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 /**
  18  * QueryAttributeResolver.cpp
  19  *
  20  * AttributeResolver based on SAML queries.
  21  */
  22
  23 #include "internal.h"
  24 #include "Application.h"
  25 #include "ServiceProvider.h"
  26 #include "SessionCache.h"
  27 #include "attribute/Attribute.h"
  28 #include "attribute/filtering/AttributeFilter.h"
  29 #include "attribute/filtering/BasicFilteringContext.h"
  30 #include "attribute/resolver/AttributeExtractor.h"
  31 #include "attribute/resolver/AttributeResolver.h"
  32 #include "attribute/resolver/ResolutionContext.h"
  33 #include "binding/SOAPClient.h"
  34 #include "util/SPConstants.h"
  35
  36 #include <saml/exceptions.h>
  37 #include <saml/binding/SecurityPolicy.h>
  38 #include <saml/saml1/binding/SAML1SOAPClient.h>
  39 #include <saml/saml1/core/Assertions.h>
  40 #include <saml/saml1/core/Protocols.h>
  41 #include <saml/saml1/profile/AssertionValidator.h>
  42 #include <saml/saml2/binding/SAML2SOAPClient.h>
  43 #include <saml/saml2/core/Protocols.h>
  44 #include <saml/saml2/metadata/Metadata.h>
  45 #include <saml/saml2/metadata/MetadataProvider.h>
  46 #include <saml/saml2/profile/AssertionValidator.h>
  47 #include <xmltooling/util/NDC.h>
  48 #include <xmltooling/util/XMLHelper.h>
  49 #include <xercesc/util/XMLUniDefs.hpp>
  50
  51 using namespace shibsp;
  52 using namespace opensaml::saml1;
  53 using namespace opensaml::saml1p;
  54 using namespace opensaml::saml2;
  55 using namespace opensaml::saml2p;
  56 using namespace opensaml::saml2md;
  57 using namespace opensaml;
  58 using namespace xmltooling;
  59 using namespace std;
  60
  61 namespace shibsp {
  62
  63     class SHIBSP_DLLLOCAL QueryContext : public ResolutionContext
  64     {
  65     public:
  66         QueryContext(const Application& application, const Session& session)
  67                 : m_query(true), m_app(application), m_session(&session), m_metadata(NULL), m_entity(NULL), m_nameid(NULL) {
  68             m_protocol = XMLString::transcode(session.getProtocol());
  69             m_class = XMLString::transcode(session.getAuthnContextClassRef());
  70             m_decl = XMLString::transcode(session.getAuthnContextDeclRef());
  71         }
  72
  73         QueryContext(
  74             const Application& application,
  75             const EntityDescriptor* issuer,
  76             const XMLCh* protocol,
  77             const NameID* nameid,
  78             const XMLCh* authncontext_class=NULL,
  79             const XMLCh* authncontext_decl=NULL,
  80             const vector<const opensaml::Assertion*>* tokens=NULL,
  81             const vector<Attribute*>* attributes=NULL
  82             ) : m_query(true), m_app(application), m_session(NULL), m_metadata(NULL), m_entity(issuer),
  83                 m_protocol(protocol), m_nameid(nameid), m_class(authncontext_class), m_decl(authncontext_decl) {
  84
  85             if (tokens) {
  86                 for (vector<const opensaml::Assertion*>::const_iterator t = tokens->begin(); t!=tokens->end(); ++t) {
  87                     const saml2::Assertion* token2 = dynamic_cast<const saml2::Assertion*>(*t);
  88                     if (token2 && !token2->getAttributeStatements().empty()) {
  89                         m_query = false;
  90                     }
  91                     else {
  92                         const saml1::Assertion* token1 = dynamic_cast<const saml1::Assertion*>(*t);
  93                         if (token1 && !token1->getAttributeStatements().empty()) {
  94                             m_query = false;
  95                         }
  96                     }
  97                 }
  98             }
  99         }
 100
 101         ~QueryContext() {
 102             if (m_session) {
 103                 XMLString::release((XMLCh**)&m_protocol);
 104                 XMLString::release((XMLCh**)&m_class);
 105                 XMLString::release((XMLCh**)&m_decl);
 106             }
 107             if (m_metadata)
 108                 m_metadata->unlock();
 109             for_each(m_attributes.begin(), m_attributes.end(), xmltooling::cleanup<shibsp::Attribute>());
 110             for_each(m_assertions.begin(), m_assertions.end(), xmltooling::cleanup<opensaml::Assertion>());
 111         }
 112
 113         bool doQuery() const {
 114             return m_query;
 115         }
 116
 117         const Application& getApplication() const {
 118             return m_app;
 119         }
 120         const EntityDescriptor* getEntityDescriptor() const {
 121             if (m_entity)
 122                 return m_entity;
 123             if (m_session && m_session->getEntityID()) {
 124                 m_metadata = m_app.getMetadataProvider();
 125                 if (m_metadata) {
 126                     m_metadata->lock();
 127                     return m_entity = m_metadata->getEntityDescriptor(m_session->getEntityID());
 128                 }
 129             }
 130             return NULL;
 131         }
 132         const XMLCh* getProtocol() const {
 133             return m_protocol;
 134         }
 135         const NameID* getNameID() const {
 136             return m_session ? m_session->getNameID() : m_nameid;
 137         }
 138         const XMLCh* getClassRef() const {
 139             return m_class;
 140         }
 141         const XMLCh* getDeclRef() const {
 142             return m_decl;
 143         }
 144         const Session* getSession() const {
 145             return m_session;
 146         }
 147         vector<shibsp::Attribute*>& getResolvedAttributes() {
 148             return m_attributes;
 149         }
 150         vector<opensaml::Assertion*>& getResolvedAssertions() {
 151             return m_assertions;
 152         }
 153
 154     private:
 155         bool m_query;
 156         const Application& m_app;
 157         const Session* m_session;
 158         mutable MetadataProvider* m_metadata;
 159         mutable const EntityDescriptor* m_entity;
 160         const XMLCh* m_protocol;
 161         const NameID* m_nameid;
 162         const XMLCh* m_class;
 163         const XMLCh* m_decl;
 164         vector<shibsp::Attribute*> m_attributes;
 165         vector<opensaml::Assertion*> m_assertions;
 166     };
 167
 168     class SHIBSP_DLLLOCAL QueryResolver : public AttributeResolver
 169     {
 170     public:
 171         QueryResolver(const DOMElement* e);
 172         ~QueryResolver() {
 173             for_each(m_SAML1Designators.begin(), m_SAML1Designators.end(), xmltooling::cleanup<AttributeDesignator>());
 174             for_each(m_SAML2Designators.begin(), m_SAML2Designators.end(), xmltooling::cleanup<saml2::Attribute>());
 175         }
 176
 177         Lockable* lock() {return this;}
 178         void unlock() {}
 179
 180         ResolutionContext* createResolutionContext(
 181             const Application& application,
 182             const EntityDescriptor* issuer,
 183             const XMLCh* protocol,
 184             const NameID* nameid,
 185             const XMLCh* authncontext_class=NULL,
 186             const XMLCh* authncontext_decl=NULL,
 187             const vector<const opensaml::Assertion*>* tokens=NULL,
 188             const vector<shibsp::Attribute*>* attributes=NULL
 189             ) const {
 190             return new QueryContext(application,issuer,protocol,nameid,authncontext_class,authncontext_decl,tokens,attributes);
 191         }
 192
 193         ResolutionContext* createResolutionContext(const Application& application, const Session& session) const {
 194             return new QueryContext(application,session);
 195         }
 196
 197         void resolveAttributes(ResolutionContext& ctx) const;
 198
 199         void getAttributeIds(vector<string>& attributes) const {
 200             // Nothing to do, only the extractor would actually generate them.
 201         }
 202
 203     private:
 204         bool SAML1Query(QueryContext& ctx) const;
 205         bool SAML2Query(QueryContext& ctx) const;
 206
 207         Category& m_log;
 208         vector<AttributeDesignator*> m_SAML1Designators;
 209         vector<saml2::Attribute*> m_SAML2Designators;
 210     };
 211
 212     AttributeResolver* SHIBSP_DLLLOCAL QueryResolverFactory(const DOMElement* const & e)
 213     {
 214         return new QueryResolver(e);
 215     }
 216
 217 };
 218
 219 void SHIBSP_API shibsp::registerAttributeResolvers()
 220 {
 221     SPConfig::getConfig().AttributeResolverManager.registerFactory(QUERY_ATTRIBUTE_RESOLVER, QueryResolverFactory);
 222 }
 223
 224 QueryResolver::QueryResolver(const DOMElement* e) : m_log(Category::getInstance(SHIBSP_LOGCAT".AttributeResolver"))
 225 {
 226 #ifdef _DEBUG
 227     xmltooling::NDC ndc("QueryResolver");
 228 #endif
 229
 230     DOMElement* child = XMLHelper::getFirstChildElement(e);
 231     while (child) {
 232         try {
 233             if (XMLHelper::isNodeNamed(e, samlconstants::SAML20_NS, saml2::Attribute::LOCAL_NAME)) {
 234                 auto_ptr<XMLObject> obj(saml2::AttributeBuilder::buildOneFromElement(child));
 235                 saml2::Attribute* down = dynamic_cast<saml2::Attribute*>(obj.get());
 236                 if (down) {
 237                     m_SAML2Designators.push_back(down);
 238                     obj.release();
 239                 }
 240             }
 241             else if (XMLHelper::isNodeNamed(e, samlconstants::SAML1P_NS, AttributeDesignator::LOCAL_NAME)) {
 242                 auto_ptr<XMLObject> obj(AttributeDesignatorBuilder::buildOneFromElement(child));
 243                 AttributeDesignator* down = dynamic_cast<AttributeDesignator*>(obj.get());
 244                 if (down) {
 245                     m_SAML1Designators.push_back(down);
 246                     obj.release();
 247                 }
 248             }
 249         }
 250         catch (exception& ex) {
 251             m_log.error("exception loading attribute designator: %s", ex.what());
 252         }
 253         child = XMLHelper::getNextSiblingElement(child);
 254     }
 255 }
 256
 257 bool QueryResolver::SAML1Query(QueryContext& ctx) const
 258 {
 259 #ifdef _DEBUG
 260     xmltooling::NDC ndc("query");
 261 #endif
 262
 263     int version = XMLString::equals(ctx.getProtocol(), samlconstants::SAML11_PROTOCOL_ENUM) ? 1 : 0;
 264     const AttributeAuthorityDescriptor* AA = ctx.getEntityDescriptor()->getAttributeAuthorityDescriptor(ctx.getProtocol());
 265     if (!AA) {
 266         m_log.warn("no SAML 1.%d AttributeAuthority role found in metadata", version);
 267         return false;
 268     }
 269
 270     shibsp::SecurityPolicy policy(ctx.getApplication());
 271     MetadataCredentialCriteria mcc(*AA);
 272     shibsp::SOAPClient soaper(policy);
 273     const PropertySet* policySettings =
 274         ctx.getApplication().getServiceProvider().getPolicySettings(ctx.getApplication().getString("policyId").second);
 275     pair<bool,bool> signedAssertions = policySettings->getBool("signedAssertions");
 276
 277     auto_ptr_XMLCh binding(samlconstants::SAML1_BINDING_SOAP);
 278     saml1p::Response* response=NULL;
 279     const vector<AttributeService*>& endpoints=AA->getAttributeServices();
 280     for (vector<AttributeService*>::const_iterator ep=endpoints.begin(); !response && ep!=endpoints.end(); ++ep) {
 281         try {
 282             if (!XMLString::equals((*ep)->getBinding(),binding.get()))
 283                 continue;
 284             auto_ptr_char loc((*ep)->getLocation());
 285             auto_ptr_XMLCh issuer(ctx.getApplication().getString("entityID").second);
 286             NameIdentifier* nameid = NameIdentifierBuilder::buildNameIdentifier();
 287             nameid->setName(ctx.getNameID()->getName());
 288             nameid->setFormat(ctx.getNameID()->getFormat());
 289             nameid->setNameQualifier(ctx.getNameID()->getNameQualifier());
 290             saml1::Subject* subject = saml1::SubjectBuilder::buildSubject();
 291             subject->setNameIdentifier(nameid);
 292             saml1p::AttributeQuery* query = saml1p::AttributeQueryBuilder::buildAttributeQuery();
 293             query->setSubject(subject);
 294             query->setResource(issuer.get());
 295             for (vector<AttributeDesignator*>::const_iterator ad = m_SAML1Designators.begin(); ad!=m_SAML1Designators.end(); ++ad)
 296                 query->getAttributeDesignators().push_back((*ad)->cloneAttributeDesignator());
 297             Request* request = RequestBuilder::buildRequest();
 298             request->setAttributeQuery(query);
 299             request->setMinorVersion(version);
 300
 301             SAML1SOAPClient client(soaper, false);
 302             client.sendSAML(request, mcc, loc.get());
 303             response = client.receiveSAML();
 304         }
 305         catch (exception& ex) {
 306             m_log.error("exception making SAML query: %s", ex.what());
 307             soaper.reset();
 308         }
 309     }
 310
 311     if (!response) {
 312         m_log.error("unable to obtain a SAML response from attribute authority");
 313         return false;
 314     }
 315     else if (!response->getStatus() || !response->getStatus()->getStatusCode() || response->getStatus()->getStatusCode()->getValue()==NULL ||
 316             *(response->getStatus()->getStatusCode()->getValue()) != saml1p::StatusCode::SUCCESS) {
 317         delete response;
 318         m_log.error("attribute authority returned a SAML error");
 319         return true;
 320     }
 321
 322     const vector<saml1::Assertion*>& assertions = const_cast<const saml1p::Response*>(response)->getAssertions();
 323     if (assertions.empty()) {
 324         delete response;
 325         m_log.warn("response from attribute authority was empty");
 326         return true;
 327     }
 328     else if (assertions.size()>1)
 329         m_log.warn("simple resolver only supports one assertion in the query response");
 330
 331     auto_ptr<saml1p::Response> wrapper(response);
 332     saml1::Assertion* newtoken = assertions.front();
 333
 334     if (!newtoken->getSignature() && signedAssertions.first && signedAssertions.second) {
 335         m_log.error("assertion unsigned, rejecting it based on signedAssertions policy");
 336         return true;
 337     }
 338
 339     try {
 340         policy.evaluate(*newtoken);
 341         if (!policy.isSecure())
 342             throw SecurityPolicyException("Security of SAML 1.x query result not established.");
 343         saml1::AssertionValidator tokval(ctx.getApplication().getAudiences(), time(NULL));
 344         tokval.validateAssertion(*newtoken);
 345     }
 346     catch (exception& ex) {
 347         m_log.error("assertion failed policy/validation: %s", ex.what());
 348         return true;
 349     }
 350
 351     newtoken->detach();
 352     wrapper.release();
 353     ctx.getResolvedAssertions().push_back(newtoken);
 354
 355     // Finally, extract and filter the result.
 356     try {
 357         AttributeExtractor* extractor = ctx.getApplication().getAttributeExtractor();
 358         if (extractor) {
 359             Locker extlocker(extractor);
 360             extractor->extractAttributes(ctx.getApplication(), AA, *newtoken, ctx.getResolvedAttributes());
 361         }
 362
 363         AttributeFilter* filter = ctx.getApplication().getAttributeFilter();
 364         if (filter) {
 365             BasicFilteringContext fc(ctx.getApplication(), ctx.getResolvedAttributes(), AA, ctx.getClassRef(), ctx.getDeclRef());
 366             Locker filtlocker(filter);
 367             filter->filterAttributes(fc, ctx.getResolvedAttributes());
 368         }
 369     }
 370     catch (exception& ex) {
 371         m_log.error("caught exception extracting/filtering attributes from query result: %s", ex.what());
 372         for_each(ctx.getResolvedAttributes().begin(), ctx.getResolvedAttributes().end(), xmltooling::cleanup<shibsp::Attribute>());
 373         ctx.getResolvedAttributes().clear();
 374     }
 375
 376     return true;
 377 }
 378
 379 bool QueryResolver::SAML2Query(QueryContext& ctx) const
 380 {
 381 #ifdef _DEBUG
 382     xmltooling::NDC ndc("query");
 383 #endif
 384
 385     const AttributeAuthorityDescriptor* AA = ctx.getEntityDescriptor()->getAttributeAuthorityDescriptor(samlconstants::SAML20P_NS);
 386     if (!AA) {
 387         m_log.warn("no SAML 2 AttributeAuthority role found in metadata");
 388         return false;
 389     }
 390
 391     shibsp::SecurityPolicy policy(ctx.getApplication());
 392     MetadataCredentialCriteria mcc(*AA);
 393     shibsp::SOAPClient soaper(policy);
 394     const PropertySet* policySettings =
 395         ctx.getApplication().getServiceProvider().getPolicySettings(ctx.getApplication().getString("policyId").second);
 396     pair<bool,bool> signedAssertions = policySettings->getBool("signedAssertions");
 397
 398     const PropertySet* relyingParty = ctx.getApplication().getRelyingParty(ctx.getEntityDescriptor());
 399     pair<bool,const char*> encryption = relyingParty->getString("encryption");
 400
 401     auto_ptr_XMLCh binding(samlconstants::SAML20_BINDING_SOAP);
 402     saml2p::StatusResponseType* srt=NULL;
 403     const vector<AttributeService*>& endpoints=AA->getAttributeServices();
 404     for (vector<AttributeService*>::const_iterator ep=endpoints.begin(); !srt && ep!=endpoints.end(); ++ep) {
 405         try {
 406             if (!XMLString::equals((*ep)->getBinding(),binding.get()))
 407                 continue;
 408             auto_ptr_char loc((*ep)->getLocation());
 409             auto_ptr_XMLCh issuer(ctx.getApplication().getString("entityID").second);
 410
 411             auto_ptr<saml2::Subject> subject(saml2::SubjectBuilder::buildSubject());
 412
 413             // Encrypt the NameID?
 414             if (encryption.first && (!strcmp(encryption.second, "true") || !strcmp(encryption.second, "back"))) {
 415                 auto_ptr<EncryptedID> encrypted(EncryptedIDBuilder::buildEncryptedID());
 416                 MetadataCredentialCriteria mcc(*AA);
 417                 encrypted->encrypt(
 418                     *ctx.getNameID(),
 419                     *(ctx.getApplication().getMetadataProvider()),
 420                     mcc,
 421                     false,
 422                     relyingParty->getXMLString("encryptionAlg").second
 423                     );
 424                 subject->setEncryptedID(encrypted.release());
 425             }
 426             else {
 427                 subject->setNameID(ctx.getNameID()->cloneNameID());
 428             }
 429
 430             saml2p::AttributeQuery* query = saml2p::AttributeQueryBuilder::buildAttributeQuery();
 431             query->setSubject(subject.release());
 432             Issuer* iss = IssuerBuilder::buildIssuer();
 433             iss->setName(issuer.get());
 434             query->setIssuer(iss);
 435             for (vector<saml2::Attribute*>::const_iterator ad = m_SAML2Designators.begin(); ad!=m_SAML2Designators.end(); ++ad)
 436                 query->getAttributes().push_back((*ad)->cloneAttribute());
 437
 438             SAML2SOAPClient client(soaper, false);
 439             client.sendSAML(query, mcc, loc.get());
 440             srt = client.receiveSAML();
 441         }
 442         catch (exception& ex) {
 443             m_log.error("exception making SAML query: %s", ex.what());
 444             soaper.reset();
 445         }
 446     }
 447
 448     if (!srt) {
 449         m_log.error("unable to obtain a SAML response from attribute authority");
 450         return false;
 451     }
 452     saml2p::Response* response = dynamic_cast<saml2p::Response*>(srt);
 453     if (!response) {
 454         delete srt;
 455         m_log.error("message was not a samlp:Response");
 456         return true;
 457     }
 458     else if (!response->getStatus() || !response->getStatus()->getStatusCode() ||
 459             !XMLString::equals(response->getStatus()->getStatusCode()->getValue(), saml2p::StatusCode::SUCCESS)) {
 460         delete srt;
 461         m_log.error("attribute authority returned a SAML error");
 462         return true;
 463     }
 464
 465     const vector<saml2::Assertion*>& assertions = const_cast<const saml2p::Response*>(response)->getAssertions();
 466     if (assertions.empty()) {
 467         delete srt;
 468         m_log.warn("response from attribute authority was empty");
 469         return true;
 470     }
 471     else if (assertions.size()>1)
 472         m_log.warn("simple resolver only supports one assertion in the query response");
 473
 474     auto_ptr<saml2p::StatusResponseType> wrapper(srt);
 475     saml2::Assertion* newtoken = assertions.front();
 476
 477     if (!newtoken->getSignature() && signedAssertions.first && signedAssertions.second) {
 478         m_log.error("assertion unsigned, rejecting it based on signedAssertions policy");
 479         return true;
 480     }
 481
 482     try {
 483         policy.evaluate(*newtoken);
 484         if (!policy.isSecure())
 485             throw SecurityPolicyException("Security of SAML 2.0 query result not established.");
 486         saml2::AssertionValidator tokval(ctx.getApplication().getAudiences(), time(NULL));
 487         tokval.validateAssertion(*newtoken);
 488     }
 489     catch (exception& ex) {
 490         m_log.error("assertion failed policy/validation: %s", ex.what());
 491         return true;
 492     }
 493
 494     newtoken->detach();
 495     wrapper.release();
 496     ctx.getResolvedAssertions().push_back(newtoken);
 497
 498     // Finally, extract and filter the result.
 499     try {
 500         AttributeExtractor* extractor = ctx.getApplication().getAttributeExtractor();
 501         if (extractor) {
 502             Locker extlocker(extractor);
 503             extractor->extractAttributes(ctx.getApplication(), AA, *newtoken, ctx.getResolvedAttributes());
 504         }
 505
 506         AttributeFilter* filter = ctx.getApplication().getAttributeFilter();
 507         if (filter) {
 508             BasicFilteringContext fc(ctx.getApplication(), ctx.getResolvedAttributes(), AA, ctx.getClassRef(), ctx.getDeclRef());
 509             Locker filtlocker(filter);
 510             filter->filterAttributes(fc, ctx.getResolvedAttributes());
 511         }
 512     }
 513     catch (exception& ex) {
 514         m_log.error("caught exception extracting/filtering attributes from query result: %s", ex.what());
 515         for_each(ctx.getResolvedAttributes().begin(), ctx.getResolvedAttributes().end(), xmltooling::cleanup<shibsp::Attribute>());
 516         ctx.getResolvedAttributes().clear();
 517     }
 518
 519     return true;
 520 }
 521
 522 void QueryResolver::resolveAttributes(ResolutionContext& ctx) const
 523 {
 524 #ifdef _DEBUG
 525     xmltooling::NDC ndc("resolveAttributes");
 526 #endif
 527
 528     QueryContext& qctx = dynamic_cast<QueryContext&>(ctx);
 529     if (!qctx.doQuery()) {
 530         m_log.debug("found AttributeStatement in input to new session, skipping query");
 531         return;
 532     }
 533
 534     if (qctx.getNameID() && qctx.getEntityDescriptor()) {
 535         if (XMLString::equals(qctx.getProtocol(), samlconstants::SAML20P_NS)) {
 536             m_log.debug("attempting SAML 2.0 attribute query");
 537             SAML2Query(qctx);
 538         }
 539         else if (XMLString::equals(qctx.getProtocol(), samlconstants::SAML11_PROTOCOL_ENUM) ||
 540                 XMLString::equals(qctx.getProtocol(), samlconstants::SAML10_PROTOCOL_ENUM)) {
 541             m_log.debug("attempting SAML 1.x attribute query");
 542             SAML1Query(qctx);
 543         }
 544         else
 545             m_log.warn("SSO protocol does not allow for attribute query");
 546     }
 547     else
 548         m_log.warn("can't attempt attribute query, either no NameID or no metadata to use");
 549 }