shib-target/shib-config.cpp

   1 /*
   2  * The Shibboleth License, Version 1.
   3  * Copyright (c) 2002
   4  * University Corporation for Advanced Internet Development, Inc.
   5  * All rights reserved
   6  *
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions are met:
  10  *
  11  * Redistributions of source code must retain the above copyright notice, this
  12  * list of conditions and the following disclaimer.
  13  *
  14  * Redistributions in binary form must reproduce the above copyright notice,
  15  * this list of conditions and the following disclaimer in the documentation
  16  * and/or other materials provided with the distribution, if any, must include
  17  * the following acknowledgment: "This product includes software developed by
  18  * the University Corporation for Advanced Internet Development
  19  * <http://www.ucaid.edu>Internet2 Project. Alternately, this acknowledegement
  20  * may appear in the software itself, if and wherever such third-party
  21  * acknowledgments normally appear.
  22  *
  23  * Neither the name of Shibboleth nor the names of its contributors, nor
  24  * Internet2, nor the University Corporation for Advanced Internet Development,
  25  * Inc., nor UCAID may be used to endorse or promote products derived from this
  26  * software without specific prior written permission. For written permission,
  27  * please contact shibboleth@shibboleth.org
  28  *
  29  * Products derived from this software may not be called Shibboleth, Internet2,
  30  * UCAID, or the University Corporation for Advanced Internet Development, nor
  31  * may Shibboleth appear in their name, without prior written permission of the
  32  * University Corporation for Advanced Internet Development.
  33  *
  34  *
  35  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  36  * AND WITH ALL FAULTS. ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  37  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
  38  * PARTICULAR PURPOSE, AND NON-INFRINGEMENT ARE DISCLAIMED AND THE ENTIRE RISK
  39  * OF SATISFACTORY QUALITY, PERFORMANCE, ACCURACY, AND EFFORT IS WITH LICENSEE.
  40  * IN NO EVENT SHALL THE COPYRIGHT OWNER, CONTRIBUTORS OR THE UNIVERSITY
  41  * CORPORATION FOR ADVANCED INTERNET DEVELOPMENT, INC. BE LIABLE FOR ANY DIRECT,
  42  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  43  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  44  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  45  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  46  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  47  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  48  */
  49
  50 /*
  51  * shib-config.cpp -- ShibTarget initialization and finalization routines
  52  *
  53  * Created By:  Derek Atkins <derek@ihtfp.com>
  54  *
  55  * $Id$
  56  */
  57
  58 #include "internal.h"
  59
  60 #include <log4cpp/OstreamAppender.hh>
  61
  62 using namespace std;
  63 using namespace log4cpp;
  64 using namespace saml;
  65 using namespace shibboleth;
  66 using namespace shibtarget;
  67
  68 namespace {
  69     STConfig g_Config;
  70 }
  71
  72 const XMLCh ShibTargetConfig::SHIBTARGET_NS[] = // urn:mace:shibboleth:target:config:1.0
  73 { chLatin_u, chLatin_r, chLatin_n, chColon, chLatin_m, chLatin_a, chLatin_c, chLatin_e, chColon,
  74   chLatin_s, chLatin_h, chLatin_i, chLatin_b, chLatin_b, chLatin_o, chLatin_l, chLatin_e, chLatin_t, chLatin_h, chColon,
  75   chLatin_t, chLatin_a, chLatin_r, chLatin_g, chLatin_e, chLatin_t, chColon,
  76   chLatin_c, chLatin_o, chLatin_n, chLatin_f, chLatin_i, chLatin_g, chColon,
  77   chDigit_1, chPeriod, chDigit_0, chNull
  78 };
  79
  80 // Factories for built-in plugins we can manufacture. Actual definitions
  81 // will be with the actual object implementation.
  82 #ifndef WIN32
  83 PlugManager::Factory UnixListenerFactory;
  84 #endif
  85 PlugManager::Factory TCPListenerFactory;
  86 PlugManager::Factory MemoryCacheFactory;
  87 PlugManager::Factory XMLRequestMapFactory;
  88 //PlugManager::Factory htaccessFactory;
  89
  90 ShibTargetConfig& ShibTargetConfig::getConfig()
  91 {
  92     return g_Config;
  93 }
  94
  95 bool STConfig::init(const char* schemadir, const char* config)
  96 {
  97     // With new build of log4cpp, we need to establish a "default"
  98     // logging appender to stderr up front.
  99     const char* loglevel=getenv("SHIB_LOGGING");
 100     if (!loglevel)
 101         loglevel = SHIB_LOGGING;
 102     Category& root = Category::getRoot();
 103     if (!strcmp(loglevel,"DEBUG"))
 104         root.setPriority(Priority::DEBUG);
 105     else if (!strcmp(loglevel,"INFO"))
 106         root.setPriority(Priority::INFO);
 107     else if (!strcmp(loglevel,"NOTICE"))
 108         root.setPriority(Priority::NOTICE);
 109     else if (!strcmp(loglevel,"WARN"))
 110         root.setPriority(Priority::WARN);
 111     else if (!strcmp(loglevel,"ERROR"))
 112         root.setPriority(Priority::ERROR);
 113     else if (!strcmp(loglevel,"CRIT"))
 114         root.setPriority(Priority::CRIT);
 115     else if (!strcmp(loglevel,"ALERT"))
 116         root.setPriority(Priority::ALERT);
 117     else if (!strcmp(loglevel,"EMERG"))
 118         root.setPriority(Priority::EMERG);
 119     else if (!strcmp(loglevel,"FATAL"))
 120         root.setPriority(Priority::FATAL);
 121     root.setAppender(new OstreamAppender("default",&cerr));
 122
 123     saml::NDC ndc("init");
 124     Category& log = Category::getInstance("shibtarget.STConfig");
 125
 126     if (!schemadir || !config) {
 127         log.fatal("schema directory or config file not supplied");
 128         return false;
 129     }
 130
 131     // This will cause some extra console logging, but for now,
 132     // initialize the underlying libraries.
 133     SAMLConfig& samlConf=SAMLConfig::getConfig();
 134     if (schemadir)
 135         samlConf.schema_dir = schemadir;
 136     SAMLSOAPBinding::version=string("Shibboleth: ") + PACKAGE_VERSION;
 137     try {
 138         if (!samlConf.init()) {
 139             log.fatal("Failed to initialize SAML Library");
 140             return false;
 141         }
 142     }
 143     catch (...) {
 144         log.fatal("Died initializing SAML Library");
 145         return false;
 146     }
 147
 148     ShibConfig& shibConf=ShibConfig::getConfig();
 149     try {
 150         if (!shibConf.init()) {
 151             log.fatal("Failed to initialize Shib library");
 152             samlConf.term();
 153             return false;
 154         }
 155     }
 156     catch (...) {
 157         log.fatal("Died initializing Shib library.");
 158         samlConf.term();
 159         return false;
 160     }
 161
 162 #ifndef _DEBUG
 163     try {
 164 #endif
 165         // Register plugin types.
 166 #ifndef WIN32
 167         samlConf.m_plugMgr.regFactory(shibtarget::XML::UnixListenerType,&UnixListenerFactory);
 168 #endif
 169         samlConf.m_plugMgr.regFactory(shibtarget::XML::TCPListenerType,&TCPListenerFactory);
 170         samlConf.m_plugMgr.regFactory(shibtarget::XML::MemorySessionCacheType,&MemoryCacheFactory);
 171         samlConf.m_plugMgr.regFactory(shibtarget::XML::RequestMapType,&XMLRequestMapFactory);
 172         //shibConf.m_plugMgr.regFactory(shibtarget::XML::htaccessType,&htaccessFactory);
 173         saml::XML::registerSchema(ShibTargetConfig::SHIBTARGET_NS,shibtarget::XML::SHIBTARGET_SCHEMA_ID);
 174
 175         log.info("loading configuration file: %s", config);
 176         static const XMLCh uri[] = { chLatin_u, chLatin_r, chLatin_i, chNull };
 177         DOMImplementation* impl=DOMImplementationRegistry::getDOMImplementation(NULL);
 178         DOMDocument* dummydoc=impl->createDocument();
 179         DOMElement* dummy = dummydoc->createElementNS(NULL,XML::Literals::ShibbolethTargetConfig);
 180         auto_ptr_XMLCh src(config);
 181         dummy->setAttributeNS(NULL,uri,src.get());
 182         m_ini=ShibTargetConfigFactory(dummy);
 183         dummydoc->release();
 184
 185         pair<bool,unsigned int> skew=m_ini->getUnsignedInt("clockSkew");
 186         samlConf.clock_skew_secs=skew.first ? skew.second : 180;
 187
 188         m_tranLog=new FixedContextCategory(SHIBTRAN_LOGCAT);
 189         m_tranLog->info("opened transaction log");
 190         m_tranLogLock = Mutex::create();
 191
 192         m_rpcpool = new RPCHandlePool;
 193 #ifndef _DEBUG
 194     }
 195     catch (...) {
 196         log.fatal("caught exception while loading/initializing configuration");
 197         delete m_ini;
 198         delete m_rpcpool;
 199         shibConf.term();
 200         samlConf.term();
 201         return false;
 202     }
 203 #endif
 204
 205     log.info("finished initializing");
 206
 207     return true;
 208 }
 209
 210 void STConfig::shutdown()
 211 {
 212     saml::NDC ndc("shutdown");
 213     Category& log = Category::getInstance("shibtarget.STConfig");
 214     log.info("shutting down the library");
 215     delete m_rpcpool;
 216     m_rpcpool = NULL;
 217     delete m_tranLogLock;
 218     m_tranLogLock = NULL;
 219     //delete m_tranLog; // This is crashing for some reason, but we're shutting down anyway.
 220     delete m_ini;
 221     m_ini = NULL;
 222     ShibConfig::getConfig().term();
 223     SAMLConfig::getConfig().term();
 224     log.info("library shutdown complete");
 225 }