ef24319975e37d6988ac5c84fd124095bbbee69f
[shibboleth/cpp-opensaml.git] / saml / SAMLConfig.cpp
1
2 /*
3  *  Copyright 2001-2006 Internet2
4  * 
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 /**
19  * SAMLConfig.cpp
20  * 
21  * Library configuration 
22  */
23
24 #include "internal.h"
25 #include "exceptions.h"
26 #include "SAMLConfig.h"
27 #include "binding/ArtifactMap.h"
28 #include "binding/MessageDecoder.h"
29 #include "binding/MessageEncoder.h"
30 #include "binding/SAMLArtifact.h"
31 #include "binding/URLEncoder.h"
32 #include "saml1/core/Assertions.h"
33 #include "saml1/core/Protocols.h"
34 #include "saml2/core/Protocols.h"
35 #include "saml2/metadata/Metadata.h"
36 #include "saml2/metadata/MetadataProvider.h"
37 #include "security/TrustEngine.h"
38 #include "util/SAMLConstants.h"
39
40 #include <xmltooling/XMLToolingConfig.h>
41 #include <xmltooling/signature/Signature.h>
42 #include <xmltooling/util/NDC.h>
43
44 #include <log4cpp/Category.hh>
45 #include <xsec/enc/XSECCryptoException.hpp>
46 #include <xsec/enc/XSECCryptoProvider.hpp>
47 #include <xsec/utils/XSECPlatformUtils.hpp>
48 #include <openssl/err.h>
49
50 using namespace opensaml;
51 using namespace xmlsignature;
52 using namespace xmltooling;
53 using namespace log4cpp;
54 using namespace std;
55
56 // Expose entry points when used as an extension library
57
58 extern "C" int SAML_API xmltooling_extension_init(void*)
59 {
60     if (SAMLConfig::getConfig().init(false))
61         return 0;
62     return -1;
63 }
64
65 extern "C" void SAML_API xmltooling_extension_term()
66 {
67     SAMLConfig::getConfig().term(false);
68 }
69
70 DECL_EXCEPTION_FACTORY(ArtifactException,opensaml);
71 DECL_EXCEPTION_FACTORY(MetadataFilterException,opensaml::saml2md);
72 DECL_EXCEPTION_FACTORY(BindingException,opensaml);
73
74 namespace opensaml {
75    SAMLInternalConfig g_config;
76 }
77
78 SAMLConfig& SAMLConfig::getConfig()
79 {
80     return g_config;
81 }
82
83 SAMLInternalConfig& SAMLInternalConfig::getInternalConfig()
84 {
85     return g_config;
86 }
87
88 void SAMLConfig::setArtifactMap(ArtifactMap* artifactMap)
89 {
90     delete m_artifactMap;
91     m_artifactMap = artifactMap;
92 }
93
94 void SAMLConfig::setURLEncoder(URLEncoder* urlEncoder)
95 {
96     delete m_urlEncoder;
97     m_urlEncoder = urlEncoder;
98 }
99
100 bool SAMLInternalConfig::init(bool initXMLTooling)
101 {
102 #ifdef _DEBUG
103     xmltooling::NDC ndc("init");
104 #endif
105     Category& log=Category::getInstance(SAML_LOGCAT".SAMLConfig");
106     log.debug("library initialization started");
107
108     if (initXMLTooling) {
109         XMLToolingConfig::getConfig().init();
110         log.debug("XMLTooling library initialized");
111     }
112
113     REGISTER_EXCEPTION_FACTORY(ArtifactException,opensaml);
114     REGISTER_EXCEPTION_FACTORY(MetadataFilterException,opensaml::saml2md);
115     REGISTER_EXCEPTION_FACTORY(BindingException,opensaml);
116
117     saml1::registerAssertionClasses();
118     saml1p::registerProtocolClasses();
119     saml2::registerAssertionClasses();
120     saml2p::registerProtocolClasses();
121     saml2md::registerMetadataClasses();
122     saml2md::registerMetadataProviders();
123     saml2md::registerMetadataFilters();
124     registerSAMLArtifacts();
125     registerTrustEngines();
126     registerMessageEncoders();
127     registerMessageDecoders();
128     registerSecurityPolicyRules();
129     
130     m_urlEncoder = new URLEncoder();
131
132     log.info("library initialization complete");
133     return true;
134 }
135
136 void SAMLInternalConfig::term(bool termXMLTooling)
137 {
138 #ifdef _DEBUG
139     xmltooling::NDC ndc("term");
140 #endif
141     Category& log=Category::getInstance(SAML_LOGCAT".SAMLConfig");
142
143     MessageDecoderManager.deregisterFactories();
144     MessageEncoderManager.deregisterFactories();
145     TrustEngineManager.deregisterFactories();
146     SecurityPolicyRuleManager.deregisterFactories();
147     SAMLArtifactManager.deregisterFactories();
148     MetadataFilterManager.deregisterFactories();
149     MetadataProviderManager.deregisterFactories();
150
151     delete m_artifactMap;
152     m_artifactMap = NULL;
153     delete m_urlEncoder;
154     m_urlEncoder = NULL;
155
156     if (termXMLTooling) {
157         XMLToolingConfig::getConfig().term();
158         log.debug("XMLTooling library shut down");
159     }
160     log.info("library shutdown complete");
161 }
162
163 void SAMLInternalConfig::generateRandomBytes(void* buf, unsigned int len)
164 {
165     try {
166         if (XSECPlatformUtils::g_cryptoProvider->getRandom(reinterpret_cast<unsigned char*>(buf),len)<len)
167             throw XMLSecurityException("Unable to generate random data; was PRNG seeded?");
168     }
169     catch (XSECCryptoException& e) {
170         throw XMLSecurityException("Unable to generate random data: $1",params(1,e.getMsg()));
171     }
172 }
173
174 void SAMLInternalConfig::generateRandomBytes(std::string& buf, unsigned int len)
175 {
176     buf.erase();
177     auto_ptr<unsigned char> hold(new unsigned char[len]);
178     generateRandomBytes(hold.get(),len);
179     for (unsigned int i=0; i<len; i++)
180         buf+=(hold.get())[i];
181 }
182
183 XMLCh* SAMLInternalConfig::generateIdentifier()
184 {
185     unsigned char key[17];
186     generateRandomBytes(key,16);
187     
188     char hexform[34];
189     sprintf(hexform,"_%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
190             key[0],key[1],key[2],key[3],key[4],key[5],key[6],key[7],
191             key[8],key[9],key[10],key[11],key[12],key[13],key[14],key[15]);
192     hexform[33]=0;
193     return XMLString::transcode(hexform);
194 }
195
196 string SAMLInternalConfig::hashSHA1(const char* s, bool toHex)
197 {
198     static char DIGITS[] = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
199
200     auto_ptr<XSECCryptoHash> hasher(XSECPlatformUtils::g_cryptoProvider->hashSHA1());
201     if (hasher.get()) {
202         auto_ptr<char> dup(strdup(s));
203         unsigned char buf[21];
204         hasher->hash(reinterpret_cast<unsigned char*>(dup.get()),strlen(dup.get()));
205         if (hasher->finish(buf,20)==20) {
206             string ret;
207             if (toHex) {
208                 for (unsigned int i=0; i<20; i++) {
209                     ret+=(DIGITS[((unsigned char)(0xF0 & buf[i])) >> 4 ]);
210                     ret+=(DIGITS[0x0F & buf[i]]);
211                 }
212             }
213             else {
214                 for (unsigned int i=0; i<20; i++) {
215                     ret+=buf[i];
216                 }
217             }
218             return ret;
219         }
220     }
221     throw XMLSecurityException("Unable to generate SHA-1 hash.");
222 }
223
224 void opensaml::log_openssl()
225 {
226     const char* file;
227     const char* data;
228     int flags,line;
229
230     unsigned long code=ERR_get_error_line_data(&file,&line,&data,&flags);
231     while (code) {
232         Category& log=Category::getInstance("OpenSSL");
233         log.errorStream() << "error code: " << code << " in " << file << ", line " << line << CategoryStream::ENDLINE;
234         if (data && (flags & ERR_TXT_STRING))
235             log.errorStream() << "error data: " << data << CategoryStream::ENDLINE;
236         code=ERR_get_error_line_data(&file,&line,&data,&flags);
237     }
238 }
239
240 using namespace saml2md;
241
242 void opensaml::annotateException(XMLToolingException* e, const EntityDescriptor* entity, bool rethrow)
243 {
244     if (entity) {
245         auto_ptr_char id(entity->getEntityID());
246         e->addProperty("entityID",id.get());
247         const list<XMLObject*>& roles=entity->getOrderedChildren();
248         for (list<XMLObject*>::const_iterator child=roles.begin(); child!=roles.end(); ++child) {
249             const RoleDescriptor* role=dynamic_cast<RoleDescriptor*>(*child);
250             if (role && role->isValid()) {
251                 const vector<ContactPerson*>& contacts=role->getContactPersons();
252                 for (vector<ContactPerson*>::const_iterator c=contacts.begin(); c!=contacts.end(); ++c) {
253                     const XMLCh* ctype=(*c)->getContactType();
254                     if (ctype && (XMLString::equals(ctype,ContactPerson::CONTACT_SUPPORT)
255                             || XMLString::equals(ctype,ContactPerson::CONTACT_TECHNICAL))) {
256                         GivenName* fname=(*c)->getGivenName();
257                         SurName* lname=(*c)->getSurName();
258                         auto_ptr_char first(fname ? fname->getName() : NULL);
259                         auto_ptr_char last(lname ? lname->getName() : NULL);
260                         if (first.get() && last.get()) {
261                             string contact=string(first.get()) + ' ' + last.get();
262                             e->addProperty("contactName",contact.c_str());
263                         }
264                         else if (first.get())
265                             e->addProperty("contactName",first.get());
266                         else if (last.get())
267                             e->addProperty("contactName",last.get());
268                         const vector<EmailAddress*>& emails=const_cast<const ContactPerson*>(*c)->getEmailAddresss();
269                         if (!emails.empty()) {
270                             auto_ptr_char email(emails.front()->getAddress());
271                             if (email.get())
272                                 e->addProperty("contactEmail",email.get());
273                         }
274                         break;
275                     }
276                 }
277                 if (e->getProperty("contactName") || e->getProperty("contactEmail")) {
278                     auto_ptr_char eurl(role->getErrorURL());
279                     if (eurl.get()) {
280                         e->addProperty("errorURL",eurl.get());
281                     }
282                 }
283                 break;
284             }
285         }
286     }
287     
288     if (rethrow)
289         e->raise();
290 }
291
292 void opensaml::annotateException(XMLToolingException* e, const RoleDescriptor* role, bool rethrow)
293 {
294     if (role) {
295         auto_ptr_char id(dynamic_cast<EntityDescriptor*>(role->getParent())->getEntityID());
296         e->addProperty("entityID",id.get());
297
298         const vector<ContactPerson*>& contacts=role->getContactPersons();
299         for (vector<ContactPerson*>::const_iterator c=contacts.begin(); c!=contacts.end(); ++c) {
300             const XMLCh* ctype=(*c)->getContactType();
301             if (ctype && (XMLString::equals(ctype,ContactPerson::CONTACT_SUPPORT)
302                     || XMLString::equals(ctype,ContactPerson::CONTACT_TECHNICAL))) {
303                 GivenName* fname=(*c)->getGivenName();
304                 SurName* lname=(*c)->getSurName();
305                 auto_ptr_char first(fname ? fname->getName() : NULL);
306                 auto_ptr_char last(lname ? lname->getName() : NULL);
307                 if (first.get() && last.get()) {
308                     string contact=string(first.get()) + ' ' + last.get();
309                     e->addProperty("contactName",contact.c_str());
310                 }
311                 else if (first.get())
312                     e->addProperty("contactName",first.get());
313                 else if (last.get())
314                     e->addProperty("contactName",last.get());
315                 const vector<EmailAddress*>& emails=const_cast<const ContactPerson*>(*c)->getEmailAddresss();
316                 if (!emails.empty()) {
317                     auto_ptr_char email(emails.front()->getAddress());
318                     if (email.get())
319                         e->addProperty("contactEmail",email.get());
320                 }
321                 break;
322             }
323         }
324
325         auto_ptr_char eurl(role->getErrorURL());
326         if (eurl.get()) {
327             e->addProperty("errorURL",eurl.get());
328         }
329     }
330     
331     if (rethrow)
332         e->raise();
333 }