samltest/security/AbstractPKIXTrustEngineTest.h

   1 /*\r
   2  *  Copyright 2001-2006 Internet2\r
   3  * \r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  *     http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 \r
  17 #include "internal.h"\r
  18 #include <saml/security/AbstractPKIXTrustEngine.h>\r
  19 #include <saml/saml2/metadata/MetadataProvider.h>\r
  20 \r
  21 using namespace opensaml::saml2;\r
  22 using namespace opensaml::saml2md;\r
  23 using namespace xmlsignature;\r
  24 \r
  25 namespace {\r
  26     class SampleTrustEngine : public AbstractPKIXTrustEngine {\r
  27     public:\r
  28         SampleTrustEngine() {}\r
  29         ~SampleTrustEngine() {}\r
  30         \r
  31         class SampleIterator : public PKIXValidationInfoIterator {\r
  32             vector<XSECCryptoX509CRL*> m_crls;\r
  33             KeyResolver::ResolvedCertificates m_certs;\r
  34             KeyResolver* m_resolver;\r
  35             bool m_done;\r
  36         public:\r
  37             SampleIterator() : m_resolver(NULL), m_done(false) {\r
  38                 string config = data_path + "security/FilesystemKeyResolver.xml";\r
  39                 ifstream in(config.c_str());\r
  40                 DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
  41                 XercesJanitor<DOMDocument> janitor(doc);\r
  42                 m_resolver = XMLToolingConfig::getConfig().KeyResolverManager.newPlugin(\r
  43                     FILESYSTEM_KEY_RESOLVER,doc->getDocumentElement()\r
  44                     );\r
  45                 m_resolver->resolveCertificates((KeyInfo*)NULL,m_certs);\r
  46             }\r
  47             \r
  48             ~SampleIterator() {\r
  49                 delete m_resolver;\r
  50             }\r
  51 \r
  52             bool next() {\r
  53                 if (m_done)\r
  54                     return false;\r
  55                 m_done = true;\r
  56                 return true;\r
  57             }\r
  58             \r
  59             int getVerificationDepth() const {\r
  60                 return 0;\r
  61             }\r
  62             \r
  63             const vector<XSECCryptoX509*>& getTrustAnchors() const {\r
  64                 return m_certs.v();\r
  65             }\r
  66             \r
  67             const vector<XSECCryptoX509CRL*>& getCRLs() const {\r
  68                 return m_crls;\r
  69             }\r
  70         };\r
  71     \r
  72         PKIXValidationInfoIterator* getPKIXValidationInfoIterator(const RoleDescriptor& role) const {\r
  73             return new SampleIterator();\r
  74         }\r
  75     };\r
  76 };\r
  77 \r
  78 class AbstractPKIXTrustEngineTest : public CxxTest::TestSuite, public SAMLObjectBaseTestCase {\r
  79 public:\r
  80     void setUp() {\r
  81         SAMLObjectBaseTestCase::setUp();\r
  82     }\r
  83     \r
  84     void tearDown() {\r
  85         SAMLObjectBaseTestCase::tearDown();\r
  86     }\r
  87 \r
  88     void testExplicitKeyTrustEngine() {\r
  89         string config = data_path + "security/FilesystemMetadataProvider.xml";\r
  90         ifstream in(config.c_str());\r
  91         DOMDocument* doc=XMLToolingConfig::getConfig().getParser().parse(in);\r
  92         XercesJanitor<DOMDocument> janitor(doc);\r
  93 \r
  94         auto_ptr_XMLCh path("path");\r
  95         string s = data_path + "security/example-metadata.xml";\r
  96         auto_ptr_XMLCh file(s.c_str());\r
  97         doc->getDocumentElement()->setAttributeNS(NULL,path.get(),file.get());\r
  98 \r
  99         // Build metadata provider.\r
 100         auto_ptr<MetadataProvider> metadataProvider(\r
 101             SAMLConfig::getConfig().MetadataProviderManager.newPlugin(FILESYSTEM_METADATA_PROVIDER,doc->getDocumentElement())\r
 102             );\r
 103         try {\r
 104             metadataProvider->init();\r
 105         }\r
 106         catch (XMLToolingException& ex) {\r
 107             TS_TRACE(ex.what());\r
 108             throw;\r
 109         }\r
 110         \r
 111         // Build trust engine.\r
 112         auto_ptr<opensaml::TrustEngine> trustEngine(new SampleTrustEngine());\r
 113         \r
 114         // Get signed assertion.\r
 115         config = data_path + "signature/SAML2Assertion.xml";\r
 116         ifstream in2(config.c_str());\r
 117         DOMDocument* doc2=XMLToolingConfig::getConfig().getParser().parse(in2);\r
 118         XercesJanitor<DOMDocument> janitor2(doc2);\r
 119         auto_ptr<Assertion> assertion(dynamic_cast<Assertion*>(XMLObjectBuilder::getBuilder(doc2->getDocumentElement())->buildFromDocument(doc2)));\r
 120         janitor2.release();\r
 121 \r
 122         Locker locker(metadataProvider.get());\r
 123         const EntityDescriptor* descriptor = metadataProvider->getEntityDescriptor("https://idp.example.org");\r
 124         TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);\r
 125         \r
 126         RoleDescriptor* role=descriptor->getIDPSSODescriptors().front();\r
 127         TSM_ASSERT("Role not present", role!=NULL);\r
 128         \r
 129         Signature* sig=assertion->getSignature();\r
 130         TSM_ASSERT("Signature not present", sig!=NULL);\r
 131         TSM_ASSERT("Signature failed to validate.", trustEngine->validate(*sig, *role, metadataProvider->getKeyResolver()));\r
 132 \r
 133         descriptor = metadataProvider->getEntityDescriptor("https://idp2.example.org");\r
 134         TSM_ASSERT("Retrieved entity descriptor was null", descriptor!=NULL);\r
 135         \r
 136         role=descriptor->getIDPSSODescriptors().front();\r
 137         TSM_ASSERT("Role not present", role!=NULL);\r
 138 \r
 139         TSM_ASSERT("Signature validated.", !trustEngine->validate(*sig, *role, metadataProvider->getKeyResolver()));\r
 140     }\r
 141 };\r