-/*\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/encryption/Decrypter.h>\r
-#include <xmltooling/encryption/Encrypter.h>\r
-\r
-#include <fstream>\r
-#include <openssl/pem.h>\r
-#include <xercesc/util/XMLUniDefs.hpp>\r
-#include <xsec/dsig/DSIGReference.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 xmlencryption;\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 EncryptionTest : public CxxTest::TestSuite {\r
- XSECCryptoKey* m_key;\r
- vector<XSECCryptoX509*> m_certs;\r
-public:\r
- void 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
-\r
- void tearDown() {\r
- delete m_key;\r
- for_each(m_certs.begin(),m_certs.end(),xmltooling::cleanup<XSECCryptoX509>());\r
- }\r
-\r
- void testBasic() {\r
- TS_TRACE("testBasic");\r
-\r
- string path=data_path + "ComplexXMLObject.xml";\r
- ifstream fs(path.c_str());\r
- DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(fs);\r
- TS_ASSERT(doc!=NULL);\r
-\r
- try {\r
- Encrypter encrypter;\r
- Encrypter::EncryptionParams ep;\r
- Encrypter::KeyEncryptionParams kep(DSIGConstants::s_unicodeStrURIRSA_1_5,m_key->clone());\r
- auto_ptr<EncryptedData> encData(encrypter.encryptElement(doc->getDocumentElement(),ep,&kep));\r
-\r
- string buf;\r
- XMLHelper::serialize(encData->marshall(), buf);\r
- istringstream is(buf);\r
- DOMDocument* doc2=XMLToolingConfig::getConfig().getValidatingParser().parse(is);\r
- auto_ptr<EncryptedData> encData2(\r
- dynamic_cast<EncryptedData*>(XMLObjectBuilder::buildOneFromElement(doc2->getDocumentElement(),true))\r
- );\r
-\r
- Decrypter decrypter(new KeyResolver(m_key->clone()));\r
- DOMDocumentFragment* frag = decrypter.decryptData(encData2.get());\r
- XMLHelper::serialize(static_cast<DOMElement*>(frag->getFirstChild()), buf);\r
- //TS_TRACE(buf.c_str());\r
- TS_ASSERT(doc->getDocumentElement()->isEqualNode(frag->getFirstChild()));\r
- frag->release();\r
- doc->release();\r
- }\r
- catch (XMLToolingException& e) {\r
- TS_TRACE(e.what());\r
- doc->release();\r
- throw;\r
- }\r
- }\r
-\r
-};\r
+/**
+ * Licensed to the University Corporation for Advanced Internet
+ * Development, Inc. (UCAID) under one or more contributor license
+ * agreements. See the NOTICE file distributed with this work for
+ * additional information regarding copyright ownership.
+ *
+ * UCAID licenses this file to you 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/encryption/Decrypter.h>
+#include <xmltooling/encryption/Encrypter.h>
+#include <xmltooling/encryption/Encryption.h>
+#include <xmltooling/security/Credential.h>
+#include <xmltooling/security/CredentialCriteria.h>
+#include <xmltooling/security/CredentialResolver.h>
+
+#include <fstream>
+#include <xercesc/util/XMLUniDefs.hpp>
+#include <xsec/dsig/DSIGReference.hpp>
+
+using namespace xmlencryption;
+
+class EncryptionTest : public CxxTest::TestSuite {
+ CredentialResolver* m_resolver;
+public:
+ void setUp() {
+ m_resolver=nullptr;
+ 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(
+ CHAINING_CREDENTIAL_RESOLVER,doc->getDocumentElement()
+ );
+ XMLObjectBuilder::registerDefaultBuilder(new UnknownElementBuilder());
+ }
+
+ void tearDown() {
+ XMLObjectBuilder::deregisterDefaultBuilder();
+ delete m_resolver;
+ }
+
+ void testEncryption() {
+ string path=data_path + "ComplexXMLObject.xml";
+ ifstream fs(path.c_str());
+ DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(fs);
+ TS_ASSERT(doc!=nullptr);
+
+ try {
+ CredentialCriteria cc;
+ cc.setUsage(Credential::ENCRYPTION_CREDENTIAL);
+ Locker locker(m_resolver);
+ const Credential* cred=m_resolver->resolve(&cc);
+ TSM_ASSERT("Retrieved credential was null", cred!=nullptr);
+
+ Encrypter encrypter;
+ Encrypter::EncryptionParams ep;
+ Encrypter::KeyEncryptionParams kep(*cred);
+ auto_ptr<EncryptedData> encData(encrypter.encryptElement(doc->getDocumentElement(),ep,&kep));
+
+ string buf;
+ XMLHelper::serialize(encData->marshall(), buf);
+ //TS_TRACE(buf.c_str());
+ istringstream is(buf);
+ DOMDocument* doc2=XMLToolingConfig::getConfig().getValidatingParser().parse(is);
+ auto_ptr<EncryptedData> encData2(
+ dynamic_cast<EncryptedData*>(XMLObjectBuilder::buildOneFromElement(doc2->getDocumentElement(),true))
+ );
+
+ Decrypter decrypter(m_resolver);
+ DOMDocumentFragment* frag = decrypter.decryptData(*encData2.get());
+ XMLHelper::serialize(static_cast<DOMElement*>(frag->getFirstChild()), buf);
+ //TS_TRACE(buf.c_str());
+ TS_ASSERT(doc->getDocumentElement()->isEqualNode(frag->getFirstChild()));
+ frag->release();
+ doc->release();
+ }
+ catch (XMLToolingException& e) {
+ TS_TRACE(e.what());
+ doc->release();
+ throw;
+ }
+ }
+
+};