Convert from NULL macro to nullptr, remove unused zlib code.
[shibboleth/cpp-opensaml.git] / samltest / signature / SAML1ResponseTest.h
index e63856f..f4a5bb1 100644 (file)
-/*\r
- *  Copyright 2001-2005 Internet2\r
- * \r
- * Licensed under the Apache License, Version 2.0 (the "License");\r
- * you may not use this file except in compliance with the License.\r
- * You may obtain a copy of the License at\r
- *\r
- *     http://www.apache.org/licenses/LICENSE-2.0\r
- *\r
- * Unless required by applicable law or agreed to in writing, software\r
- * distributed under the License is distributed on an "AS IS" BASIS,\r
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
- * See the License for the specific language governing permissions and\r
- * limitations under the License.\r
- */\r
-\r
-#include "internal.h"\r
-#include <saml/saml1/core/Protocols.h>\r
-#include <saml/signature/SignatureProfileValidator.h>\r
-\r
-#include <xmltooling/signature/Signature.h>\r
-\r
-#include <fstream>\r
-#include <openssl/pem.h>\r
-#include <xercesc/util/XMLUniDefs.hpp>\r
-#include <xsec/enc/XSECKeyInfoResolverDefault.hpp>\r
-#include <xsec/enc/OpenSSL/OpenSSLCryptoX509.hpp>\r
-#include <xsec/enc/OpenSSL/OpenSSLCryptoKeyRSA.hpp>\r
-#include <xsec/enc/XSECCryptoException.hpp>\r
-#include <xsec/framework/XSECException.hpp>\r
-\r
-using namespace opensaml::saml1;\r
-using namespace xmlsignature;\r
-\r
-class TestValidator : public Validator\r
-{\r
-public:\r
-    TestValidator() {}\r
-    virtual ~TestValidator() {}\r
-\r
-    Validator* clone() const {\r
-        return new TestValidator();\r
-    }\r
-\r
-    void validate(const XMLObject* xmlObject) const {\r
-        DSIGSignature* sig=dynamic_cast<const Signature*>(xmlObject)->getXMLSignature();\r
-        if (!sig)\r
-            throw SignatureException("Only a marshalled Signature object can be verified.");\r
-        XSECKeyInfoResolverDefault resolver;\r
-        sig->setKeyInfoResolver(&resolver); // It will clone the resolver for us.\r
-        try {\r
-            if (!sig->verify())\r
-                throw SignatureException("Signature did not verify.");\r
-        }\r
-        catch(XSECException& e) {\r
-            auto_ptr_char temp(e.getMsg());\r
-            throw SignatureException(string("Caught an XMLSecurity exception verifying signature: ") + temp.get());\r
-        }\r
-        catch(XSECCryptoException& e) {\r
-            throw SignatureException(string("Caught an XMLSecurity exception verifying signature: ") + e.getMsg());\r
-        }\r
-    }\r
-};\r
-\r
-class _addcert : public std::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
-class SAML1ResponseTest : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
-    XSECCryptoKey* m_key;\r
-    vector<XSECCryptoX509*> m_certs;\r
-public:\r
-    void setUp() {\r
-        childElementsFile  = data_path + "signature/SAML1Response.xml";\r
-        SAMLObjectBaseTestCase::setUp();\r
-        string keypath=data_path + "key.pem";\r
-        BIO* in=BIO_new(BIO_s_file_internal());\r
-        if (in && BIO_read_filename(in,keypath.c_str())>0) {\r
-            EVP_PKEY* pkey=PEM_read_bio_PrivateKey(in, NULL, NULL, NULL);\r
-            if (pkey) {\r
-                m_key=new OpenSSLCryptoKeyRSA(pkey);\r
-                EVP_PKEY_free(pkey);\r
-            }\r
-        }\r
-        if (in) BIO_free(in);\r
-        TS_ASSERT(m_key!=NULL);\r
-\r
-        string certpath=data_path + "cert.pem";\r
-        in=BIO_new(BIO_s_file_internal());\r
-        if (in && BIO_read_filename(in,certpath.c_str())>0) {\r
-            X509* x=NULL;\r
-            while (x=PEM_read_bio_X509(in,NULL,NULL,NULL)) {\r
-                m_certs.push_back(new OpenSSLCryptoX509(x));\r
-                X509_free(x);\r
-            }\r
-        }\r
-        if (in) BIO_free(in);\r
-        TS_ASSERT(m_certs.size()>0);\r
-    }\r
-\r
-    void tearDown() {\r
-        SAMLObjectBaseTestCase::tearDown();\r
-        delete m_key;\r
-        for_each(m_certs.begin(),m_certs.end(),xmltooling::cleanup<XSECCryptoX509>());\r
-    }\r
-\r
-    void testSignature() {\r
-        auto_ptr_XMLCh issuer("issuer");\r
-        auto_ptr_XMLCh issueInstant("1970-01-02T01:01:02.100Z");\r
-        auto_ptr_XMLCh aid("aident");\r
-        auto_ptr_XMLCh rid("rident");\r
-        auto_ptr_XMLCh method("method");\r
-        auto_ptr_XMLCh nameid("John Doe");\r
-        \r
-        NameIdentifier* n=NameIdentifierBuilder::buildNameIdentifier();\r
-        n->setName(nameid.get());        \r
-        Subject* subject=SubjectBuilder::buildSubject();\r
-        subject->setNameIdentifier(n);\r
-\r
-        AuthenticationStatement* statement=AuthenticationStatementBuilder::buildAuthenticationStatement();\r
-        statement->setAuthenticationInstant(issueInstant.get());\r
-        statement->setAuthenticationMethod(method.get());\r
-        statement->setSubject(subject);\r
-        \r
-        Assertion* assertion=AssertionBuilder::buildAssertion();\r
-        assertion->setAssertionID(aid.get());\r
-        assertion->setIssueInstant(issueInstant.get());\r
-        assertion->setIssuer(issuer.get());\r
-        assertion->getAuthenticationStatements().push_back(statement);\r
-\r
-        // Append a Signature.\r
-        assertion->setSignature(SignatureBuilder::buildSignature());\r
-        assertion->getSignature()->setSigningKey(m_key->clone());\r
-\r
-        // Build KeyInfo.\r
-        KeyInfo* keyInfo=KeyInfoBuilder::buildKeyInfo();\r
-        X509Data* x509Data=X509DataBuilder::buildX509Data();\r
-        keyInfo->getX509Datas().push_back(x509Data);\r
-        for_each(m_certs.begin(),m_certs.end(),bind1st(_addcert(),x509Data));\r
-        assertion->getSignature()->setKeyInfo(keyInfo);\r
-\r
-        // Sign assertion while marshalling.\r
-        vector<Signature*> sigs(1,assertion->getSignature());\r
-        DOMElement* rootElement = NULL;\r
-        try {\r
-            rootElement=assertion->marshall((DOMDocument*)NULL,&sigs);\r
-        }\r
-        catch (XMLToolingException& e) {\r
-            TS_TRACE(e.what());\r
-            delete assertion;\r
-            throw;\r
-        }\r
-\r
-        StatusCode* sc=StatusCodeBuilder::buildStatusCode();\r
-        sc->setValue(&StatusCode::SUCCESS);\r
-        Status* status=StatusBuilder::buildStatus();\r
-        status->setStatusCode(sc);\r
-\r
-        auto_ptr<Response> response(ResponseBuilder::buildResponse());\r
-        response->setResponseID(rid.get());\r
-        response->setIssueInstant(issueInstant.get());\r
-        response->setStatus(status);\r
-        response->getAssertions().push_back(assertion);\r
-        response->setSignature(SignatureBuilder::buildSignature());\r
-        response->getSignature()->setSigningKey(m_key->clone());\r
-        response->getSignature()->setKeyInfo(keyInfo->cloneKeyInfo());\r
-\r
-        // Sign response while marshalling.\r
-        sigs.clear();\r
-        sigs.push_back(response->getSignature());\r
-        rootElement = NULL;\r
-        try {\r
-            rootElement=response->marshall((DOMDocument*)NULL,&sigs);\r
-        }\r
-        catch (XMLToolingException& e) {\r
-            TS_TRACE(e.what());\r
-            throw;\r
-        }\r
-\r
-        string buf;\r
-        XMLHelper::serialize(rootElement, buf);\r
-        istringstream in(buf);\r
-        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
-        const XMLObjectBuilder* b = XMLObjectBuilder::getBuilder(doc->getDocumentElement());\r
-        \r
-        assertEquals(expectedChildElementsDOM, b->buildFromDocument(doc));\r
-        \r
-        try {\r
-            assertion->getSignature()->registerValidator(new SignatureProfileValidator());\r
-            assertion->getSignature()->registerValidator(new TestValidator());\r
-            assertion->getSignature()->validate(true);\r
-            response->getSignature()->registerValidator(new SignatureProfileValidator());\r
-            response->getSignature()->registerValidator(new TestValidator());\r
-            response->getSignature()->validate(true);\r
-        }\r
-        catch (XMLToolingException& e) {\r
-            TS_TRACE(e.what());\r
-            throw;\r
-        }\r
-    }\r
-\r
-};\r
+/*
+ *  Copyright 2001-2010 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
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "signature/SAMLSignatureTestBase.h"
+
+#include <saml/saml1/core/Assertions.h>
+#include <saml/saml1/core/Protocols.h>
+
+#include <fstream>
+
+using namespace opensaml::saml1p;
+using namespace opensaml::saml1;
+
+class SAML1ResponseTest : public CxxTest::TestSuite, public SAMLSignatureTestBase {
+public:
+    void setUp() {
+        childElementsFile  = data_path + "signature/SAML1Response.xml";
+        SAMLSignatureTestBase::setUp();
+    }
+
+    void tearDown() {
+        SAMLSignatureTestBase::tearDown();
+    }
+
+    void testSignature() {
+        auto_ptr_XMLCh issuer("issuer");
+        auto_ptr_XMLCh issueInstant("1970-01-02T01:01:02.100Z");
+        auto_ptr_XMLCh aid("aident");
+        auto_ptr_XMLCh rid("rident");
+        auto_ptr_XMLCh method("method");
+        auto_ptr_XMLCh nameid("John Doe");
+        
+        NameIdentifier* n=NameIdentifierBuilder::buildNameIdentifier();
+        n->setName(nameid.get());        
+        Subject* subject=SubjectBuilder::buildSubject();
+        subject->setNameIdentifier(n);
+
+        AuthenticationStatement* statement=AuthenticationStatementBuilder::buildAuthenticationStatement();
+        statement->setAuthenticationInstant(issueInstant.get());
+        statement->setAuthenticationMethod(method.get());
+        statement->setSubject(subject);
+        
+        Assertion* assertion=AssertionBuilder::buildAssertion();
+        assertion->setAssertionID(aid.get());
+        assertion->setIssueInstant(issueInstant.get());
+        assertion->setIssuer(issuer.get());
+        assertion->getAuthenticationStatements().push_back(statement);
+
+        // Append a Signature.
+        assertion->setSignature(SignatureBuilder::buildSignature());
+
+        // Sign assertion while marshalling.
+        vector<Signature*> sigs(1,assertion->getSignature());
+        CredentialCriteria cc;
+        cc.setUsage(Credential::SIGNING_CREDENTIAL);
+        Locker locker(m_resolver);
+        const Credential* cred = m_resolver->resolve(&cc);
+        TSM_ASSERT("Retrieved credential was null", cred!=nullptr);
+
+        DOMElement* rootElement = nullptr;
+        try {
+            rootElement=assertion->marshall((DOMDocument*)nullptr,&sigs,cred);
+        }
+        catch (XMLToolingException& e) {
+            TS_TRACE(e.what());
+            delete assertion;
+            throw;
+        }
+
+        StatusCode* sc=StatusCodeBuilder::buildStatusCode();
+        sc->setValue(&StatusCode::SUCCESS);
+        Status* status=StatusBuilder::buildStatus();
+        status->setStatusCode(sc);
+
+        auto_ptr<Response> response(ResponseBuilder::buildResponse());
+        response->setResponseID(rid.get());
+        response->setIssueInstant(issueInstant.get());
+        response->setStatus(status);
+        response->getAssertions().push_back(assertion);
+        response->setSignature(SignatureBuilder::buildSignature());
+
+        // Sign response while marshalling.
+        sigs.clear();
+        sigs.push_back(response->getSignature());
+        rootElement = nullptr;
+        try {
+            rootElement=response->marshall((DOMDocument*)nullptr,&sigs,cred);
+        }
+        catch (XMLToolingException& e) {
+            TS_TRACE(e.what());
+            throw;
+        }
+
+        string buf;
+        XMLHelper::serialize(rootElement, buf);
+        istringstream in(buf);
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        const XMLObjectBuilder* b = XMLObjectBuilder::getBuilder(doc->getDocumentElement());
+        
+        auto_ptr<XMLObject> response2(b->buildFromDocument(doc));
+        assertEquals("Unmarshalled response does not match", expectedChildElementsDOM, response2.get(), false);
+        
+        try {
+            opensaml::SignatureProfileValidator spv;
+            spv.validate(dynamic_cast<Response*>(response2.get())->getAssertions().front()->getSignature());
+            spv.validate(dynamic_cast<Response*>(response2.get())->getSignature());
+
+            SignatureValidator sv(cred);
+            sv.validate(dynamic_cast<Response*>(response2.get())->getAssertions().front()->getSignature());
+            sv.validate(dynamic_cast<Response*>(response2.get())->getSignature());
+        }
+        catch (XMLToolingException& e) {
+            TS_TRACE(e.what());
+            throw;
+        }
+    }
+
+};