gcc const fix, converted linefeeds
[shibboleth/cpp-xmltooling.git] / xmltoolingtest / SignatureTest.h
index c1ca7ec..a733767 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 "XMLObjectBaseTestCase.h"\r
-\r
-#include <xmltooling/signature/CredentialResolver.h>\r
-#include <xmltooling/signature/KeyInfo.h>\r
-#include <xmltooling/signature/SignatureValidator.h>\r
-\r
-#include <fstream>\r
-#include <xercesc/util/XMLUniDefs.hpp>\r
-#include <xsec/dsig/DSIGReference.hpp>\r
-\r
-class TestContext : public ContentReference\r
-{\r
-    XMLCh* m_uri;\r
-    \r
-public:\r
-    TestContext(const XMLCh* uri) {\r
-        m_uri=XMLString::replicate(uri);\r
-    }\r
-    \r
-    virtual ~TestContext() {\r
-        XMLString::release(&m_uri);\r
-    }\r
-\r
-    void createReferences(DSIGSignature* sig) {\r
-        DSIGReference* ref=sig->createReference(m_uri);\r
-        ref->appendEnvelopedSignatureTransform();\r
-        ref->appendCanonicalizationTransform(CANON_C14NE_NOC);\r
-    }\r
-};\r
-\r
-class TestValidator : public SignatureValidator\r
-{\r
-    XMLCh* m_uri;\r
-    \r
-public:\r
-    TestValidator(const XMLCh* uri) : SignatureValidator(XMLToolingConfig::getConfig().KeyResolverManager.newPlugin(INLINE_KEY_RESOLVER,NULL)) {\r
-        m_uri=XMLString::replicate(uri);\r
-    }\r
-    \r
-    virtual ~TestValidator() {\r
-        XMLString::release(&m_uri);\r
-    }\r
-\r
-    void validate(const Signature* sigObj) const {\r
-        DSIGSignature* sig=sigObj->getXMLSignature();\r
-        if (!sig)\r
-            throw SignatureException("Only a marshalled Signature object can be verified.");\r
-        const XMLCh* uri=sig->getReferenceList()->item(0)->getURI();\r
-        TSM_ASSERT_SAME_DATA("Reference URI does not match.",uri,m_uri,XMLString::stringLen(uri));\r
-        SignatureValidator::validate(sigObj);\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 SignatureTest : public CxxTest::TestSuite {\r
-    CredentialResolver* m_resolver;\r
-public:\r
-    void setUp() {\r
-        m_resolver=NULL;\r
-        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);\r
-        QName qtype(SimpleXMLObject::NAMESPACE,SimpleXMLObject::TYPE_NAME);\r
-        XMLObjectBuilder::registerBuilder(qname, new SimpleXMLObjectBuilder());\r
-        XMLObjectBuilder::registerBuilder(qtype, new SimpleXMLObjectBuilder());\r
-\r
-        string config = data_path + "FilesystemCredentialResolver.xml";\r
-        ifstream in(config.c_str());\r
-        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
-        XercesJanitor<DOMDocument> janitor(doc);\r
-        m_resolver = XMLToolingConfig::getConfig().CredentialResolverManager.newPlugin(\r
-            FILESYSTEM_CREDENTIAL_RESOLVER,doc->getDocumentElement()\r
-            );\r
-    }\r
-\r
-    void tearDown() {\r
-        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);\r
-        QName qtype(SimpleXMLObject::NAMESPACE,SimpleXMLObject::TYPE_NAME);\r
-        XMLObjectBuilder::deregisterBuilder(qname);\r
-        XMLObjectBuilder::deregisterBuilder(qtype);\r
-        delete m_resolver;\r
-    }\r
-\r
-    void testSignature() {\r
-        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);\r
-        const SimpleXMLObjectBuilder* b=dynamic_cast<const SimpleXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(qname));\r
-        TS_ASSERT(b!=NULL);\r
-        \r
-        auto_ptr<SimpleXMLObject> sxObject(b->buildObject());\r
-        TS_ASSERT(sxObject.get()!=NULL);\r
-        VectorOf(SimpleXMLObject) kids=sxObject->getSimpleXMLObjects();\r
-        kids.push_back(b->buildObject());\r
-        kids.push_back(b->buildObject());\r
-        \r
-        // Test some collection stuff\r
-        auto_ptr_XMLCh foo("Foo");\r
-        auto_ptr_XMLCh bar("Bar");\r
-        kids.begin()->setId(foo.get());\r
-        kids[1]->setValue(bar.get());\r
-        \r
-        // Append a Signature.\r
-        Signature* sig=SignatureBuilder::buildSignature();\r
-        sxObject->setSignature(sig);\r
-        sig->setContentReference(new TestContext(&chNull));\r
-\r
-        Locker locker(m_resolver);\r
-        sig->setSigningKey(m_resolver->getKey());\r
-        \r
-        // Build KeyInfo.\r
-        KeyInfo* keyInfo=KeyInfoBuilder::buildKeyInfo();\r
-        X509Data* x509Data=X509DataBuilder::buildX509Data();\r
-        keyInfo->getX509Datas().push_back(x509Data);\r
-        for_each(m_resolver->getCertificates().begin(),m_resolver->getCertificates().end(),bind1st(_addcert(),x509Data));\r
-        sig->setKeyInfo(keyInfo);\r
-        \r
-        DOMElement* rootElement = NULL;\r
-        try {\r
-            rootElement=sxObject->marshall((DOMDocument*)NULL);\r
-            sig->sign();\r
-        }\r
-        catch (XMLToolingException& e) {\r
-            TS_TRACE(e.what());\r
-            throw;\r
-        }\r
-        \r
-        string buf;\r
-        XMLHelper::serialize(rootElement, buf);\r
-        //TS_TRACE(buf.c_str());\r
-\r
-        istringstream in(buf);\r
-        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
-        auto_ptr<SimpleXMLObject> sxObject2(dynamic_cast<SimpleXMLObject*>(b->buildFromDocument(doc)));\r
-        TS_ASSERT(sxObject2.get()!=NULL);\r
-        TS_ASSERT(sxObject2->getSignature()!=NULL);\r
-        \r
-        try {\r
-            TestValidator tv(&chNull);\r
-            tv.validate(sxObject2->getSignature());\r
-        }\r
-        catch (XMLToolingException& e) {\r
-            TS_TRACE(e.what());\r
-            throw;\r
-        }\r
-    }\r
-\r
-};\r
+/*
+ *  Copyright 2001-2005 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 "XMLObjectBaseTestCase.h"
+
+#include <xmltooling/signature/CredentialResolver.h>
+#include <xmltooling/signature/KeyInfo.h>
+#include <xmltooling/signature/SignatureValidator.h>
+
+#include <fstream>
+#include <xercesc/util/XMLUniDefs.hpp>
+#include <xsec/dsig/DSIGReference.hpp>
+
+class TestContext : public ContentReference
+{
+    XMLCh* m_uri;
+    
+public:
+    TestContext(const XMLCh* uri) {
+        m_uri=XMLString::replicate(uri);
+    }
+    
+    virtual ~TestContext() {
+        XMLString::release(&m_uri);
+    }
+
+    void createReferences(DSIGSignature* sig) {
+        DSIGReference* ref=sig->createReference(m_uri);
+        ref->appendEnvelopedSignatureTransform();
+        ref->appendCanonicalizationTransform(CANON_C14NE_NOC);
+    }
+};
+
+class TestValidator : public SignatureValidator
+{
+    XMLCh* m_uri;
+    
+public:
+    TestValidator(const XMLCh* uri) : SignatureValidator(XMLToolingConfig::getConfig().KeyResolverManager.newPlugin(INLINE_KEY_RESOLVER,NULL)) {
+        m_uri=XMLString::replicate(uri);
+    }
+    
+    virtual ~TestValidator() {
+        XMLString::release(&m_uri);
+    }
+
+    void validate(const Signature* sigObj) const {
+        DSIGSignature* sig=sigObj->getXMLSignature();
+        if (!sig)
+            throw SignatureException("Only a marshalled Signature object can be verified.");
+        const XMLCh* uri=sig->getReferenceList()->item(0)->getURI();
+        TSM_ASSERT_SAME_DATA("Reference URI does not match.",uri,m_uri,XMLString::stringLen(uri));
+        SignatureValidator::validate(sigObj);
+    }
+};
+
+class _addcert : public std::binary_function<X509Data*,XSECCryptoX509*,void> {
+public:
+    void operator()(X509Data* bag, XSECCryptoX509* cert) const {
+        safeBuffer& buf=cert->getDEREncodingSB();
+        X509Certificate* x=X509CertificateBuilder::buildX509Certificate();
+        x->setValue(buf.sbStrToXMLCh());
+        bag->getX509Certificates().push_back(x);
+    }
+};
+
+class SignatureTest : public CxxTest::TestSuite {
+    CredentialResolver* m_resolver;
+public:
+    void setUp() {
+        m_resolver=NULL;
+        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);
+        QName qtype(SimpleXMLObject::NAMESPACE,SimpleXMLObject::TYPE_NAME);
+        XMLObjectBuilder::registerBuilder(qname, new SimpleXMLObjectBuilder());
+        XMLObjectBuilder::registerBuilder(qtype, new SimpleXMLObjectBuilder());
+
+        string config = data_path + "FilesystemCredentialResolver.xml";
+        ifstream in(config.c_str());
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        XercesJanitor<DOMDocument> janitor(doc);
+        m_resolver = XMLToolingConfig::getConfig().CredentialResolverManager.newPlugin(
+            FILESYSTEM_CREDENTIAL_RESOLVER,doc->getDocumentElement()
+            );
+    }
+
+    void tearDown() {
+        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);
+        QName qtype(SimpleXMLObject::NAMESPACE,SimpleXMLObject::TYPE_NAME);
+        XMLObjectBuilder::deregisterBuilder(qname);
+        XMLObjectBuilder::deregisterBuilder(qtype);
+        delete m_resolver;
+    }
+
+    void testSignature() {
+        QName qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME);
+        const SimpleXMLObjectBuilder* b=dynamic_cast<const SimpleXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(qname));
+        TS_ASSERT(b!=NULL);
+        
+        auto_ptr<SimpleXMLObject> sxObject(b->buildObject());
+        TS_ASSERT(sxObject.get()!=NULL);
+        VectorOf(SimpleXMLObject) kids=sxObject->getSimpleXMLObjects();
+        kids.push_back(b->buildObject());
+        kids.push_back(b->buildObject());
+        
+        // Test some collection stuff
+        auto_ptr_XMLCh foo("Foo");
+        auto_ptr_XMLCh bar("Bar");
+        kids.begin()->setId(foo.get());
+        kids[1]->setValue(bar.get());
+        
+        // Append a Signature.
+        Signature* sig=SignatureBuilder::buildSignature();
+        sxObject->setSignature(sig);
+        sig->setContentReference(new TestContext(&chNull));
+
+        Locker locker(m_resolver);
+        sig->setSigningKey(m_resolver->getKey());
+        
+        // Build KeyInfo.
+        KeyInfo* keyInfo=KeyInfoBuilder::buildKeyInfo();
+        X509Data* x509Data=X509DataBuilder::buildX509Data();
+        keyInfo->getX509Datas().push_back(x509Data);
+        for_each(m_resolver->getCertificates().begin(),m_resolver->getCertificates().end(),bind1st(_addcert(),x509Data));
+        sig->setKeyInfo(keyInfo);
+        
+        DOMElement* rootElement = NULL;
+        try {
+            rootElement=sxObject->marshall((DOMDocument*)NULL);
+            sig->sign();
+        }
+        catch (XMLToolingException& e) {
+            TS_TRACE(e.what());
+            throw;
+        }
+        
+        string buf;
+        XMLHelper::serialize(rootElement, buf);
+        //TS_TRACE(buf.c_str());
+
+        istringstream in(buf);
+        DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);
+        auto_ptr<SimpleXMLObject> sxObject2(dynamic_cast<SimpleXMLObject*>(b->buildFromDocument(doc)));
+        TS_ASSERT(sxObject2.get()!=NULL);
+        TS_ASSERT(sxObject2->getSignature()!=NULL);
+        
+        try {
+            TestValidator tv(&chNull);
+            tv.validate(sxObject2->getSignature());
+        }
+        catch (XMLToolingException& e) {
+            TS_TRACE(e.what());
+            throw;
+        }
+    }
+
+};