X-Git-Url: http://www.project-moonshot.org/gitweb/?p=shibboleth%2Fcpp-opensaml.git;a=blobdiff_plain;f=saml%2Fsaml2%2Fbinding%2Fimpl%2FSAML2ArtifactEncoder.cpp;h=80626f5ed39560771bfe997d8d6b39cdab52efd7;hp=048eeb74790d52f3435c47b84a181fe4e8aae404;hb=54b7006826fd06e8e2bc251aaba38cccc85b3936;hpb=b0c11eb8bfd9daa93de903f1ab94f0f34df46c2e diff --git a/saml/saml2/binding/impl/SAML2ArtifactEncoder.cpp b/saml/saml2/binding/impl/SAML2ArtifactEncoder.cpp index 048eeb7..80626f5 100644 --- a/saml/saml2/binding/impl/SAML2ArtifactEncoder.cpp +++ b/saml/saml2/binding/impl/SAML2ArtifactEncoder.cpp @@ -1,5 +1,5 @@ /* - * Copyright 2001-2006 Internet2 + * 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. @@ -17,60 +17,97 @@ /** * SAML2ArtifactEncoder.cpp * - * SAML 2.0 HTTP-Artifact binding message encoder + * SAML 2.0 HTTP-Artifact binding message encoder. */ #include "internal.h" #include "exceptions.h" #include "binding/ArtifactMap.h" -#include "binding/URLEncoder.h" +#include "binding/MessageEncoder.h" #include "saml2/binding/SAML2Artifact.h" -#include "saml2/binding/SAML2ArtifactEncoder.h" #include "saml2/core/Protocols.h" +#include "saml2/metadata/Metadata.h" +#include "signature/ContentReference.h" #include #include -#include +#include +#include +#include +#include #include +#include #include +#include using namespace opensaml::saml2p; +using namespace opensaml::saml2md; using namespace opensaml; using namespace xmlsignature; +using namespace xmltooling::logging; using namespace xmltooling; -using namespace log4cpp; using namespace std; namespace opensaml { namespace saml2p { - MessageEncoder* SAML_DLLLOCAL SAML2ArtifactEncoderFactory(const DOMElement* const & e) + class SAML_DLLLOCAL SAML2ArtifactEncoder : public MessageEncoder { - return new SAML2ArtifactEncoder(e); + public: + SAML2ArtifactEncoder(const DOMElement* e, const XMLCh* ns); + virtual ~SAML2ArtifactEncoder() {} + + long encode( + GenericResponse& genericResponse, + XMLObject* xmlObject, + const char* destination, + const EntityDescriptor* recipient=NULL, + const char* relayState=NULL, + const ArtifactGenerator* artifactGenerator=NULL, + const Credential* credential=NULL, + const XMLCh* signatureAlg=NULL, + const XMLCh* digestAlg=NULL + ) const; + + private: + bool m_post; + string m_template; + }; + + MessageEncoder* SAML_DLLLOCAL SAML2ArtifactEncoderFactory(const pair& p) + { + return new SAML2ArtifactEncoder(p.first, p.second); } }; -}; -static const XMLCh templat[] = UNICODE_LITERAL_8(t,e,m,p,l,a,t,e); + static const XMLCh _template[] = UNICODE_LITERAL_8(t,e,m,p,l,a,t,e); + static const XMLCh postArtifact[] = UNICODE_LITERAL_12(p,o,s,t,A,r,t,i,f,a,c,t); +}; -SAML2ArtifactEncoder::SAML2ArtifactEncoder(const DOMElement* e) +SAML2ArtifactEncoder::SAML2ArtifactEncoder(const DOMElement* e, const XMLCh* ns) : m_post(false) { if (e) { - auto_ptr_char t(e->getAttributeNS(NULL, templat)); - if (t.get()) - m_template = t.get(); + const XMLCh* flag = e->getAttributeNS(ns, postArtifact); + m_post = (flag && (*flag==chLatin_t || *flag==chDigit_1)); + if (m_post) { + auto_ptr_char t(e->getAttributeNS(ns, _template)); + if (t.get() && *t.get()) { + m_template = t.get(); + XMLToolingConfig::getConfig().getPathResolver()->resolve(m_template, PathResolver::XMLTOOLING_CFG_FILE); + } + } } } -SAML2ArtifactEncoder::~SAML2ArtifactEncoder() {} - long SAML2ArtifactEncoder::encode( - HTTPResponse& httpResponse, - xmltooling::XMLObject* xmlObject, + GenericResponse& genericResponse, + XMLObject* xmlObject, const char* destination, - const char* recipientID, + const EntityDescriptor* recipient, const char* relayState, - const CredentialResolver* credResolver, - const XMLCh* sigAlgorithm + const ArtifactGenerator* artifactGenerator, + const Credential* credential, + const XMLCh* signatureAlg, + const XMLCh* digestAlg ) const { #ifdef _DEBUG @@ -78,31 +115,36 @@ long SAML2ArtifactEncoder::encode( #endif Category& log = Category::getInstance(SAML_LOGCAT".MessageEncoder.SAML2Artifact"); log.debug("validating input"); - + if (!destination) + throw BindingException("Encoding response requires a destination."); + HTTPResponse* httpResponse=dynamic_cast(&genericResponse); + if (!httpResponse) + throw BindingException("Unable to cast response interface to HTTPResponse type."); if (relayState && strlen(relayState)>80) throw BindingException("RelayState cannot exceed 80 bytes in length."); - if (xmlObject->getParent()) throw BindingException("Cannot encode XML content with parent."); StatusResponseType* response = NULL; RequestAbstractType* request = dynamic_cast(xmlObject); - if (!request) + if (!request) { response = dynamic_cast(xmlObject); - if (!response) - throw BindingException("XML content for SAML 2.0 HTTP-Artifact Encoder must be a SAML 2.0 protocol message."); + if (!response) + throw BindingException("XML content for SAML 2.0 HTTP-Artifact Encoder must be a SAML 2.0 protocol message."); + } ArtifactMap* mapper = SAMLConfig::getConfig().getArtifactMap(); if (!mapper) throw BindingException("SAML 2.0 HTTP-Artifact Encoder requires ArtifactMap be set in configuration."); // Obtain a fresh artifact. - if (!m_artifactGenerator) + if (!artifactGenerator) throw BindingException("SAML 2.0 HTTP-Artifact Encoder requires an ArtifactGenerator instance."); - log.debug("obtaining new artifact for relying party (%s)", recipientID ? recipientID : "unknown"); - auto_ptr artifact(m_artifactGenerator->generateSAML2Artifact(recipientID)); + auto_ptr_char recipientID(recipient ? recipient->getEntityID() : NULL); + log.debug("obtaining new artifact for relying party (%s)", recipientID.get() ? recipientID.get() : "unknown"); + auto_ptr artifact(artifactGenerator->generateSAML2Artifact(recipient)); - if (credResolver) { + if (credential) { // Signature based on native XML signing. if (request ? request->getSignature() : response->getSignature()) { log.debug("message already signed, skipping signature operation"); @@ -111,31 +153,39 @@ long SAML2ArtifactEncoder::encode( log.debug("signing the message"); // Build a Signature. - Signature* sig = buildSignature(credResolver, sigAlgorithm); - - // Append Signature. + Signature* sig = SignatureBuilder::buildSignature(); request ? request->setSignature(sig) : response->setSignature(sig); - + if (signatureAlg) + sig->setSignatureAlgorithm(signatureAlg); + if (digestAlg) { + opensaml::ContentReference* cr = dynamic_cast(sig->getContentReference()); + if (cr) + cr->setDigestAlgorithm(digestAlg); + } + // Sign response while marshalling. vector sigs(1,sig); - xmlObject->marshall((DOMDocument*)NULL,&sigs); + xmlObject->marshall((DOMDocument*)NULL,&sigs,credential); } } + + if (log.isDebugEnabled()) + log.debugStream() << "marshalled message:" << logging::eol << *xmlObject << logging::eol; // Store the message. Last step in storage will be to delete the XML. log.debug("storing artifact and content in map"); - mapper->storeContent(xmlObject, artifact.get(), recipientID); + mapper->storeContent(xmlObject, artifact.get(), recipientID.get()); if (m_template.empty()) { // Generate redirect. string loc = destination; loc += (strchr(destination,'?') ? '&' : '?'); - URLEncoder* escaper = SAMLConfig::getConfig().getURLEncoder(); + const URLEncoder* escaper = XMLToolingConfig::getConfig().getURLEncoder(); loc = loc + "SAMLart=" + escaper->encode(artifact->encode().c_str()); - if (relayState) + if (relayState && *relayState) loc = loc + "&RelayState=" + escaper->encode(relayState); log.debug("message encoded, sending redirect to client"); - return httpResponse.sendRedirect(loc.c_str()); + return httpResponse->sendRedirect(loc.c_str()); } else { // Push message into template and send result to client. @@ -143,16 +193,21 @@ long SAML2ArtifactEncoder::encode( TemplateEngine* engine = XMLToolingConfig::getConfig().getTemplateEngine(); if (!engine) throw BindingException("Encoding artifact using POST requires a TemplateEngine instance."); + HTTPResponse::sanitizeURL(destination); ifstream infile(m_template.c_str()); if (!infile) throw BindingException("Failed to open HTML template for POST response ($1).", params(1,m_template.c_str())); - map params; - params["action"] = destination; - params["SAMLart"] = artifact->encode(); - if (relayState) - params["RelayState"] = relayState; + TemplateEngine::TemplateParameters params; + params.m_map["action"] = destination; + params.m_map["SAMLart"] = artifact->encode(); + if (relayState && *relayState) + params.m_map["RelayState"] = relayState; stringstream s; engine->run(infile, s, params); - return httpResponse.sendResponse(s); + httpResponse->setContentType("text/html"); + httpResponse->setResponseHeader("Expires", "01-Jan-1997 12:00:00 GMT"); + httpResponse->setResponseHeader("Cache-Control", "no-cache, no-store, must-revalidate, private"); + httpResponse->setResponseHeader("Pragma", "no-cache"); + return httpResponse->sendResponse(s); } }