Shift policy responsibility to ArtifactResolver, make msg rules more forgiving.
[shibboleth/cpp-opensaml.git] / samltest / saml1 / binding / SAML1ArtifactTest.h
index decc34f..05ea3cb 100644 (file)
 #include "binding.h"\r
 \r
 #include <saml/binding/ArtifactMap.h>\r
+#include <saml/saml1/core/Assertions.h>\r
 #include <saml/saml1/core/Protocols.h>\r
 #include <saml/saml1/binding/SAMLArtifactType0001.h>\r
+#include <xmltooling/validation/ValidatorSuite.h>\r
 \r
 using namespace opensaml::saml1p;\r
 using namespace opensaml::saml1;\r
 \r
+namespace {\r
+    class SAML_DLLLOCAL _addcert : public binary_function<X509Data*,XSECCryptoX509*,void> {\r
+    public:\r
+        void operator()(X509Data* bag, XSECCryptoX509* cert) const {\r
+            safeBuffer& buf=cert->getDEREncodingSB();\r
+            X509Certificate* x=X509CertificateBuilder::buildX509Certificate();\r
+            x->setValue(buf.sbStrToXMLCh());\r
+            bag->getX509Certificates().push_back(x);\r
+        }\r
+    };\r
+};\r
+\r
 class SAML1ArtifactTest : public CxxTest::TestSuite,\r
-    public SAMLBindingBaseTestCase, public MessageEncoder::ArtifactGenerator, public MessageDecoder::ArtifactResolver {\r
+        public SAMLBindingBaseTestCase, public MessageEncoder::ArtifactGenerator, public MessageDecoder::ArtifactResolver {\r
 public:\r
     void setUp() {\r
-        m_fields.clear();\r
         SAMLBindingBaseTestCase::setUp();\r
     }\r
 \r
     void tearDown() {\r
-        m_fields.clear();\r
         SAMLBindingBaseTestCase::tearDown();\r
     }\r
 \r
     void testSAML1Artifact() {\r
         try {\r
+            QName idprole(samlconstants::SAML20MD_NS, IDPSSODescriptor::LOCAL_NAME);\r
+            SecurityPolicy policy(m_rules1, m_metadata, &idprole, m_trust);\r
+\r
             // Read message to use from file.\r
             string path = data_path + "saml1/binding/SAML1Assertion.xml";\r
             ifstream in(path.c_str());\r
@@ -49,37 +64,33 @@ public:
             janitor.release();\r
 \r
             // Encode message.\r
-            auto_ptr<MessageEncoder> encoder(SAMLConfig::getConfig().MessageEncoderManager.newPlugin(SAML1_ARTIFACT_ENCODER, NULL));\r
+            auto_ptr<MessageEncoder> encoder(\r
+                SAMLConfig::getConfig().MessageEncoderManager.newPlugin(samlconstants::SAML1_PROFILE_BROWSER_ARTIFACT, NULL)\r
+                );\r
             encoder->setArtifactGenerator(this);\r
-            encoder->encode(m_fields,toSend.get(),"https://sp.example.org/","state",m_creds);\r
+            encoder->encode(*this,toSend.get(),"https://sp.example.org/SAML/SSO","https://sp.example.org/","state",m_creds);\r
             toSend.release();\r
             \r
             // Decode message.\r
             string relayState;\r
-            const RoleDescriptor* issuer=NULL;\r
-            bool trusted=false;\r
-            QName idprole(SAMLConstants::SAML20MD_NS, IDPSSODescriptor::LOCAL_NAME);\r
-            auto_ptr<MessageDecoder> decoder(SAMLConfig::getConfig().MessageDecoderManager.newPlugin(SAML1_ARTIFACT_DECODER, NULL));\r
+            auto_ptr<MessageDecoder> decoder(\r
+                SAMLConfig::getConfig().MessageDecoderManager.newPlugin(samlconstants::SAML1_PROFILE_BROWSER_ARTIFACT, NULL)\r
+                );\r
             decoder->setArtifactResolver(this);\r
             Locker locker(m_metadata);\r
-            auto_ptr<Response> response(\r
-                dynamic_cast<Response*>(\r
-                    decoder->decode(relayState,issuer,trusted,*this,m_metadata,&idprole,m_trust)\r
-                    )\r
-                );\r
+            auto_ptr<Response> response(dynamic_cast<Response*>(decoder->decode(relayState,*this,policy)));\r
             \r
             // Test the results.\r
             TSM_ASSERT_EQUALS("TARGET was not the expected result.", relayState, "state");\r
             TSM_ASSERT("SAML Response not decoded successfully.", response.get());\r
-            TSM_ASSERT("Message was not verified.", issuer && trusted);\r
-            auto_ptr_char entityID(dynamic_cast<const EntityDescriptor*>(issuer->getParent())->getEntityID());\r
+            TSM_ASSERT("Message was not verified.", policy.isSecure());\r
+            auto_ptr_char entityID(policy.getIssuer()->getName());\r
             TSM_ASSERT("Issuer was not expected.", !strcmp(entityID.get(),"https://idp.example.org/"));\r
             TSM_ASSERT_EQUALS("Assertion count was not correct.", response->getAssertions().size(), 1);\r
 \r
             // Trigger a replay.\r
-            TSM_ASSERT_THROWS("Did not catch the replay.", \r
-                decoder->decode(relayState,issuer,trusted,*this,m_metadata,&idprole,m_trust),\r
-                BindingException);\r
+            policy.reset();\r
+            TSM_ASSERT_THROWS("Did not catch the replay.", decoder->decode(relayState,*this,policy), BindingException);\r
         }\r
         catch (XMLToolingException& ex) {\r
             TS_TRACE(ex.what());\r
@@ -87,18 +98,6 @@ public:
         }\r
     }\r
 \r
-    const char* getMethod() const {\r
-        return "GET";\r
-    } \r
-\r
-    const char* getRequestURL() const {\r
-        return "https://sp.example.org/SAML/Artifact";\r
-    }\r
-    \r
-    const char* getQueryString() const {\r
-        return NULL;\r
-    }\r
-    \r
     SAMLArtifact* generateSAML1Artifact(const char* relyingParty) const {\r
         return new SAMLArtifactType0001(SAMLConfig::getConfig().hashSHA1("https://idp.example.org/"));\r
     }\r
@@ -107,11 +106,29 @@ public:
         throw BindingException("Not implemented.");\r
     }\r
     \r
+    Signature* buildSignature(const CredentialResolver* credResolver) const\r
+    {\r
+        // Build a Signature.\r
+        Signature* sig = SignatureBuilder::buildSignature();\r
+        sig->setSigningKey(credResolver->getKey());\r
+\r
+        // Build KeyInfo.\r
+        const vector<XSECCryptoX509*>& certs = credResolver->getCertificates();\r
+        if (!certs.empty()) {\r
+            KeyInfo* keyInfo=KeyInfoBuilder::buildKeyInfo();\r
+            X509Data* x509Data=X509DataBuilder::buildX509Data();\r
+            keyInfo->getX509Datas().push_back(x509Data);\r
+            for_each(certs.begin(),certs.end(),bind1st(_addcert(),x509Data));\r
+            sig->setKeyInfo(keyInfo);\r
+        }\r
+        \r
+        return sig;\r
+    }\r
+\r
     Response* resolve(\r
-        bool& authenticated,\r
         const vector<SAMLArtifact*>& artifacts,\r
         const IDPSSODescriptor& idpDescriptor,\r
-        const X509TrustEngine* trustEngine=NULL\r
+        SecurityPolicy& policy\r
         ) const {\r
         TSM_ASSERT_EQUALS("Too many artifacts.", artifacts.size(), 1);\r
         XMLObject* xmlObject =\r
@@ -125,17 +142,18 @@ public:
         StatusCode* sc = StatusCodeBuilder::buildStatusCode();\r
         status->setStatusCode(sc);\r
         sc->setValue(&StatusCode::SUCCESS);\r
-        response->marshall();\r
+        response->setSignature(buildSignature(m_creds));\r
+        vector<Signature*> sigs(1,response->getSignature());\r
+        response->marshall((DOMDocument*)NULL,&sigs);\r
         SchemaValidators.validate(response.get());\r
-        authenticated = true;\r
+        policy.evaluate(*(response.get()), this);\r
         return response.release();\r
     }\r
 \r
-    XMLObject* resolve(\r
-        bool& authenticated,\r
+    saml2p::ArtifactResponse* resolve(\r
         const saml2p::SAML2Artifact& artifact,\r
         const SSODescriptorType& ssoDescriptor,\r
-        const X509TrustEngine* trustEngine=NULL\r
+        SecurityPolicy& policy\r
         ) const {\r
         throw BindingException("Not implemented.");\r
     }\r