2 * Copyright 2001-2007 Internet2
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
18 * QueryAttributeResolver.cpp
20 * AttributeResolver based on SAML queries.
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"
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>
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;
63 class SHIBSP_DLLLOCAL QueryContext : public ResolutionContext
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());
74 const Application& application,
75 const EntityDescriptor* issuer,
76 const XMLCh* protocol,
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) {
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()) {
92 const saml1::Assertion* token1 = dynamic_cast<const saml1::Assertion*>(*t);
93 if (token1 && !token1->getAttributeStatements().empty()) {
103 XMLString::release((XMLCh**)&m_protocol);
104 XMLString::release((XMLCh**)&m_class);
105 XMLString::release((XMLCh**)&m_decl);
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>());
113 bool doQuery() const {
117 const Application& getApplication() const {
120 const EntityDescriptor* getEntityDescriptor() const {
123 if (m_session && m_session->getEntityID()) {
124 m_metadata = m_app.getMetadataProvider();
127 return m_entity = m_metadata->getEntityDescriptor(m_session->getEntityID());
132 const XMLCh* getProtocol() const {
135 const NameID* getNameID() const {
136 return m_session ? m_session->getNameID() : m_nameid;
138 const XMLCh* getClassRef() const {
141 const XMLCh* getDeclRef() const {
144 const Session* getSession() const {
147 vector<shibsp::Attribute*>& getResolvedAttributes() {
150 vector<opensaml::Assertion*>& getResolvedAssertions() {
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;
164 vector<shibsp::Attribute*> m_attributes;
165 vector<opensaml::Assertion*> m_assertions;
168 class SHIBSP_DLLLOCAL QueryResolver : public AttributeResolver
171 QueryResolver(const DOMElement* e);
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>());
177 Lockable* lock() {return this;}
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
190 return new QueryContext(application,issuer,protocol,nameid,authncontext_class,authncontext_decl,tokens,attributes);
193 ResolutionContext* createResolutionContext(const Application& application, const Session& session) const {
194 return new QueryContext(application,session);
197 void resolveAttributes(ResolutionContext& ctx) const;
199 void getAttributeIds(vector<string>& attributes) const {
200 // Nothing to do, only the extractor would actually generate them.
204 bool SAML1Query(QueryContext& ctx) const;
205 bool SAML2Query(QueryContext& ctx) const;
208 vector<AttributeDesignator*> m_SAML1Designators;
209 vector<saml2::Attribute*> m_SAML2Designators;
212 AttributeResolver* SHIBSP_DLLLOCAL QueryResolverFactory(const DOMElement* const & e)
214 return new QueryResolver(e);
219 void SHIBSP_API shibsp::registerAttributeResolvers()
221 SPConfig::getConfig().AttributeResolverManager.registerFactory(QUERY_ATTRIBUTE_RESOLVER, QueryResolverFactory);
224 QueryResolver::QueryResolver(const DOMElement* e) : m_log(Category::getInstance(SHIBSP_LOGCAT".AttributeResolver"))
227 xmltooling::NDC ndc("QueryResolver");
230 DOMElement* child = XMLHelper::getFirstChildElement(e);
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());
237 m_SAML2Designators.push_back(down);
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());
245 m_SAML1Designators.push_back(down);
250 catch (exception& ex) {
251 m_log.error("exception loading attribute designator: %s", ex.what());
253 child = XMLHelper::getNextSiblingElement(child);
257 bool QueryResolver::SAML1Query(QueryContext& ctx) const
260 xmltooling::NDC ndc("query");
263 int version = XMLString::equals(ctx.getProtocol(), samlconstants::SAML11_PROTOCOL_ENUM) ? 1 : 0;
264 const AttributeAuthorityDescriptor* AA = ctx.getEntityDescriptor()->getAttributeAuthorityDescriptor(ctx.getProtocol());
266 m_log.warn("no SAML 1.%d AttributeAuthority role found in metadata", version);
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");
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) {
282 if (!XMLString::equals((*ep)->getBinding(),binding.get()))
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);
301 SAML1SOAPClient client(soaper, false);
302 client.sendSAML(request, mcc, loc.get());
303 response = client.receiveSAML();
305 catch (exception& ex) {
306 m_log.error("exception making SAML query: %s", ex.what());
312 m_log.error("unable to obtain a SAML response from attribute authority");
315 else if (!response->getStatus() || !response->getStatus()->getStatusCode() || response->getStatus()->getStatusCode()->getValue()==NULL ||
316 *(response->getStatus()->getStatusCode()->getValue()) != saml1p::StatusCode::SUCCESS) {
318 m_log.error("attribute authority returned a SAML error");
322 const vector<saml1::Assertion*>& assertions = const_cast<const saml1p::Response*>(response)->getAssertions();
323 if (assertions.empty()) {
325 m_log.warn("response from attribute authority was empty");
328 else if (assertions.size()>1)
329 m_log.warn("simple resolver only supports one assertion in the query response");
331 auto_ptr<saml1p::Response> wrapper(response);
332 saml1::Assertion* newtoken = assertions.front();
334 if (!newtoken->getSignature() && signedAssertions.first && signedAssertions.second) {
335 m_log.error("assertion unsigned, rejecting it based on signedAssertions policy");
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);
346 catch (exception& ex) {
347 m_log.error("assertion failed policy/validation: %s", ex.what());
353 ctx.getResolvedAssertions().push_back(newtoken);
355 // Finally, extract and filter the result.
357 AttributeExtractor* extractor = ctx.getApplication().getAttributeExtractor();
359 Locker extlocker(extractor);
360 extractor->extractAttributes(ctx.getApplication(), AA, *newtoken, ctx.getResolvedAttributes());
363 AttributeFilter* filter = ctx.getApplication().getAttributeFilter();
365 BasicFilteringContext fc(ctx.getApplication(), ctx.getResolvedAttributes(), AA, ctx.getClassRef(), ctx.getDeclRef());
366 Locker filtlocker(filter);
367 filter->filterAttributes(fc, ctx.getResolvedAttributes());
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();
379 bool QueryResolver::SAML2Query(QueryContext& ctx) const
382 xmltooling::NDC ndc("query");
385 const AttributeAuthorityDescriptor* AA = ctx.getEntityDescriptor()->getAttributeAuthorityDescriptor(samlconstants::SAML20P_NS);
387 m_log.warn("no SAML 2 AttributeAuthority role found in metadata");
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");
398 auto_ptr_XMLCh binding(samlconstants::SAML20_BINDING_SOAP);
399 saml2p::StatusResponseType* srt=NULL;
400 const vector<AttributeService*>& endpoints=AA->getAttributeServices();
401 for (vector<AttributeService*>::const_iterator ep=endpoints.begin(); !srt && ep!=endpoints.end(); ++ep) {
403 if (!XMLString::equals((*ep)->getBinding(),binding.get()))
405 auto_ptr_char loc((*ep)->getLocation());
406 auto_ptr_XMLCh issuer(ctx.getApplication().getString("entityID").second);
407 saml2::Subject* subject = saml2::SubjectBuilder::buildSubject();
408 subject->setNameID(ctx.getNameID()->cloneNameID());
409 saml2p::AttributeQuery* query = saml2p::AttributeQueryBuilder::buildAttributeQuery();
410 query->setSubject(subject);
411 Issuer* iss = IssuerBuilder::buildIssuer();
412 iss->setName(issuer.get());
413 query->setIssuer(iss);
414 for (vector<saml2::Attribute*>::const_iterator ad = m_SAML2Designators.begin(); ad!=m_SAML2Designators.end(); ++ad)
415 query->getAttributes().push_back((*ad)->cloneAttribute());
417 SAML2SOAPClient client(soaper, false);
418 client.sendSAML(query, mcc, loc.get());
419 srt = client.receiveSAML();
421 catch (exception& ex) {
422 m_log.error("exception making SAML query: %s", ex.what());
428 m_log.error("unable to obtain a SAML response from attribute authority");
431 saml2p::Response* response = dynamic_cast<saml2p::Response*>(srt);
434 m_log.error("message was not a samlp:Response");
437 else if (!response->getStatus() || !response->getStatus()->getStatusCode() ||
438 !XMLString::equals(response->getStatus()->getStatusCode()->getValue(), saml2p::StatusCode::SUCCESS)) {
440 m_log.error("attribute authority returned a SAML error");
444 const vector<saml2::Assertion*>& assertions = const_cast<const saml2p::Response*>(response)->getAssertions();
445 if (assertions.empty()) {
447 m_log.warn("response from attribute authority was empty");
450 else if (assertions.size()>1)
451 m_log.warn("simple resolver only supports one assertion in the query response");
453 auto_ptr<saml2p::StatusResponseType> wrapper(srt);
454 saml2::Assertion* newtoken = assertions.front();
456 if (!newtoken->getSignature() && signedAssertions.first && signedAssertions.second) {
457 m_log.error("assertion unsigned, rejecting it based on signedAssertions policy");
462 policy.evaluate(*newtoken);
463 if (!policy.isSecure())
464 throw SecurityPolicyException("Security of SAML 2.0 query result not established.");
465 saml2::AssertionValidator tokval(ctx.getApplication().getAudiences(), time(NULL));
466 tokval.validateAssertion(*newtoken);
468 catch (exception& ex) {
469 m_log.error("assertion failed policy/validation: %s", ex.what());
475 ctx.getResolvedAssertions().push_back(newtoken);
477 // Finally, extract and filter the result.
479 AttributeExtractor* extractor = ctx.getApplication().getAttributeExtractor();
481 Locker extlocker(extractor);
482 extractor->extractAttributes(ctx.getApplication(), AA, *newtoken, ctx.getResolvedAttributes());
485 AttributeFilter* filter = ctx.getApplication().getAttributeFilter();
487 BasicFilteringContext fc(ctx.getApplication(), ctx.getResolvedAttributes(), AA, ctx.getClassRef(), ctx.getDeclRef());
488 Locker filtlocker(filter);
489 filter->filterAttributes(fc, ctx.getResolvedAttributes());
492 catch (exception& ex) {
493 m_log.error("caught exception extracting/filtering attributes from query result: %s", ex.what());
494 for_each(ctx.getResolvedAttributes().begin(), ctx.getResolvedAttributes().end(), xmltooling::cleanup<shibsp::Attribute>());
495 ctx.getResolvedAttributes().clear();
501 void QueryResolver::resolveAttributes(ResolutionContext& ctx) const
504 xmltooling::NDC ndc("resolveAttributes");
507 QueryContext& qctx = dynamic_cast<QueryContext&>(ctx);
508 if (!qctx.doQuery()) {
509 m_log.debug("found AttributeStatement in input to new session, skipping query");
513 if (qctx.getNameID() && qctx.getEntityDescriptor()) {
514 if (XMLString::equals(qctx.getProtocol(), samlconstants::SAML20P_NS)) {
515 m_log.debug("attempting SAML 2.0 attribute query");
518 else if (XMLString::equals(qctx.getProtocol(), samlconstants::SAML11_PROTOCOL_ENUM) ||
519 XMLString::equals(qctx.getProtocol(), samlconstants::SAML10_PROTOCOL_ENUM)) {
520 m_log.debug("attempting SAML 1.x attribute query");
524 m_log.warn("SSO protocol does not allow for attribute query");
527 m_log.warn("can't attempt attribute query, either no NameID or no metadata to use");