xmltooling/XMLToolingConfig.cpp

   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 /**\r
  18  * XMLToolingConfig.cpp\r
  19  * \r
  20  * Library configuration \r
  21  */\r
  22 \r
  23 #include "internal.h"\r
  24 #include "XMLToolingConfig.h"\r
  25 #include "impl/UnknownElement.h"\r
  26 #include "util/NDC.h"\r
  27 \r
  28 #ifdef HAVE_DLFCN_H\r
  29 # include <dlfcn.h>\r
  30 #endif\r
  31 \r
  32 #include <log4cpp/Category.hh>\r
  33 #include <log4cpp/PropertyConfigurator.hh>\r
  34 #include <log4cpp/OstreamAppender.hh>\r
  35 #include <xercesc/util/PlatformUtils.hpp>\r
  36 #include <xsec/framework/XSECProvider.hpp>\r
  37 \r
  38 #include <stdexcept>\r
  39 \r
  40 using namespace log4cpp;\r
  41 using namespace xmltooling;\r
  42 using namespace std;\r
  43 \r
  44 namespace {\r
  45    XMLToolingInternalConfig g_config;\r
  46 }\r
  47 \r
  48 XMLToolingConfig& XMLToolingConfig::getConfig()\r
  49 {\r
  50     return g_config;\r
  51 }\r
  52 \r
  53 XMLToolingInternalConfig& XMLToolingInternalConfig::getInternalConfig()\r
  54 {\r
  55     return g_config;\r
  56 }\r
  57 \r
  58 bool XMLToolingInternalConfig::log_config(const char* config)\r
  59 {\r
  60     try {\r
  61         if (!config && !*config)\r
  62             config=getenv("XMLTOOLING_LOG_CONFIG");\r
  63         if (!config && !*config)\r
  64             config="WARN";\r
  65         \r
  66         bool level=false;\r
  67         Category& root = Category::getRoot();\r
  68         if (!strcmp(config,"DEBUG")) {\r
  69             root.setPriority(Priority::DEBUG);\r
  70             level=true;\r
  71         }\r
  72         else if (!strcmp(config,"INFO")) {\r
  73             root.setPriority(Priority::INFO);\r
  74             level=true;\r
  75         }\r
  76         else if (!strcmp(config,"NOTICE")) {\r
  77             root.setPriority(Priority::NOTICE);\r
  78             level=true;\r
  79         }\r
  80         else if (!strcmp(config,"WARN")) {\r
  81             root.setPriority(Priority::WARN);\r
  82             level=true;\r
  83         }\r
  84         else if (!strcmp(config,"ERROR")) {\r
  85             root.setPriority(Priority::ERROR);\r
  86             level=true;\r
  87         }\r
  88         else if (!strcmp(config,"CRIT")) {\r
  89             root.setPriority(Priority::CRIT);\r
  90             level=true;\r
  91         }\r
  92         else if (!strcmp(config,"ALERT")) {\r
  93             root.setPriority(Priority::ALERT);\r
  94             level=true;\r
  95         }\r
  96         else if (!strcmp(config,"EMERG")) {\r
  97             root.setPriority(Priority::EMERG);\r
  98             level=true;\r
  99         }\r
 100         else if (!strcmp(config,"FATAL")) {\r
 101             root.setPriority(Priority::FATAL);\r
 102             level=true;\r
 103         }\r
 104         if (level)\r
 105             root.setAppender(new OstreamAppender("default",&cerr));\r
 106         else\r
 107             PropertyConfigurator::configure(config);\r
 108     }\r
 109     catch (const ConfigureFailure& e) {\r
 110         Category::getInstance(XMLTOOLING_LOGCAT".Logging").crit("failed to initialize log4cpp: %s", e.what());\r
 111         return false;\r
 112     }\r
 113     \r
 114     return true;\r
 115 }\r
 116 \r
 117 bool XMLToolingInternalConfig::init()\r
 118 {\r
 119 #ifdef _DEBUG\r
 120     xmltooling::NDC ndc("init");\r
 121 #endif\r
 122     Category& log=Category::getInstance(XMLTOOLING_LOGCAT".XMLToolingConfig");\r
 123     try {\r
 124         log.debug("library initialization started");\r
 125 \r
 126         xercesc::XMLPlatformUtils::Initialize();\r
 127         log.debug("Xerces initialization complete");\r
 128 \r
 129         XSECPlatformUtils::Initialise();\r
 130         //m_xsec=new XSECProvider();\r
 131         log.debug("XMLSec initialization complete");\r
 132         \r
 133         m_parserPool=new ParserPool();\r
 134         m_lock=xercesc::XMLPlatformUtils::makeMutex();\r
 135 \r
 136         // default registrations\r
 137         XMLObjectBuilder::registerDefaultBuilder(new UnknownElementBuilder());\r
 138         Marshaller::registerDefaultMarshaller(new UnknownElementMarshaller());\r
 139         Unmarshaller::registerDefaultUnmarshaller(new UnknownElementUnmarshaller());\r
 140     }\r
 141     catch (const xercesc::XMLException&) {\r
 142         log.fatal("caught exception while initializing Xerces");\r
 143         return false;\r
 144     }\r
 145 \r
 146     log.info("library initialization complete");\r
 147     return true;\r
 148 }\r
 149 \r
 150 void XMLToolingInternalConfig::term()\r
 151 {\r
 152     for (vector<void*>::reverse_iterator i=m_libhandles.rbegin(); i!=m_libhandles.rend(); i++) {\r
 153 #if defined(WIN32)\r
 154         FARPROC fn=GetProcAddress(static_cast<HMODULE>(*i),"xmltooling_extension_term");\r
 155         if (fn)\r
 156             fn();\r
 157         FreeLibrary(static_cast<HMODULE>(*i));\r
 158 #elif defined(HAVE_DLFCN_H)\r
 159         void (*fn)()=(void (*)())dlsym(*i,"xmltooling_extension_term");\r
 160         if (fn)\r
 161             fn();\r
 162         dlclose(*i);\r
 163 #else\r
 164 # error "Don't know about dynamic loading on this platform!"\r
 165 #endif\r
 166     }\r
 167     m_libhandles.clear();\r
 168     \r
 169     delete m_parserPool;\r
 170     m_parserPool=NULL;\r
 171 \r
 172     //delete m_xsec; m_xsec=NULL;\r
 173     XSECPlatformUtils::Terminate();\r
 174     xercesc::XMLPlatformUtils::closeMutex(m_lock);\r
 175     m_lock=NULL;\r
 176     xercesc::XMLPlatformUtils::Terminate();\r
 177 \r
 178  #ifdef _DEBUG\r
 179     xmltooling::NDC ndc("term");\r
 180 #endif\r
 181    Category::getInstance(XMLTOOLING_LOGCAT".XMLToolingConfig").info("library shutdown complete");\r
 182 }\r
 183 \r
 184 ILockable& XMLToolingInternalConfig::lock()\r
 185 {\r
 186     xercesc::XMLPlatformUtils::lockMutex(m_lock);\r
 187     return *this;\r
 188 }\r
 189 \r
 190 void XMLToolingInternalConfig::unlock()\r
 191 {\r
 192     xercesc::XMLPlatformUtils::unlockMutex(m_lock);\r
 193 }\r
 194 \r
 195 bool XMLToolingInternalConfig::load_library(const char* path, void* context)\r
 196 {\r
 197 #ifdef _DEBUG\r
 198     xmltooling::NDC ndc("LoadLibrary");\r
 199 #endif\r
 200     Category& log=Category::getInstance(XMLTOOLING_LOGCAT".XMLToolingConfig");\r
 201     log.info("loading extension: %s", path);\r
 202 \r
 203     Locker locker(this);\r
 204 \r
 205 #if defined(WIN32)\r
 206     HMODULE handle=NULL;\r
 207     char* fixed=const_cast<char*>(path);\r
 208     if (strchr(fixed,'/')) {\r
 209         fixed=strdup(path);\r
 210         char* p=fixed;\r
 211         while (p=strchr(p,'/'))\r
 212             *p='\\';\r
 213     }\r
 214 \r
 215     UINT em=SetErrorMode(SEM_FAILCRITICALERRORS);\r
 216     try {\r
 217         handle=LoadLibraryEx(fixed,NULL,LOAD_WITH_ALTERED_SEARCH_PATH);\r
 218         if (!handle)\r
 219              handle=LoadLibraryEx(fixed,NULL,0);\r
 220         if (!handle)\r
 221             throw runtime_error(string("unable to load extension library: ") + fixed);\r
 222         FARPROC fn=GetProcAddress(handle,"xmltooling_extension_init");\r
 223         if (!fn)\r
 224             throw runtime_error(string("unable to locate xmltooling_extension_init entry point: ") + fixed);\r
 225         if (reinterpret_cast<int(*)(void*)>(fn)(context)!=0)\r
 226             throw runtime_error(string("detected error in xmltooling_extension_init: ") + fixed);\r
 227         if (fixed!=path)\r
 228             free(fixed);\r
 229         SetErrorMode(em);\r
 230     }\r
 231     catch(runtime_error& e) {\r
 232         log.error(e.what());\r
 233         if (handle)\r
 234             FreeLibrary(handle);\r
 235         SetErrorMode(em);\r
 236         if (fixed!=path)\r
 237             free(fixed);\r
 238         return false;\r
 239     }\r
 240 \r
 241 #elif defined(HAVE_DLFCN_H)\r
 242     void* handle=dlopen(path,RTLD_LAZY);\r
 243     if (!handle)\r
 244         throw runtime_error(string("unable to load extension library '") + path + "': " + dlerror());\r
 245     int (*fn)(void*)=(int (*)(void*))(dlsym(handle,"xmltooling_extension_init"));\r
 246     if (!fn) {\r
 247         dlclose(handle);\r
 248         throw runtime_error(\r
 249             string("unable to locate xmltooling_extension_init entry point in '") + path + "': " +\r
 250                 (dlerror() ? dlerror() : "unknown error")\r
 251             );\r
 252     }\r
 253     try {\r
 254         if (fn(context)!=0)\r
 255             throw runtime_error(string("detected error in xmltooling_extension_init in ") + path);\r
 256     }\r
 257     catch(runtime_error& e) {\r
 258         log.error(e.what());\r
 259         if (handle)\r
 260             dlclose(handle);\r
 261         return false;\r
 262     }\r
 263 #else\r
 264 # error "Don't know about dynamic loading on this platform!"\r
 265 #endif\r
 266     m_libhandles.push_back(handle);\r
 267     log.info("loaded extension: %s", path);\r
 268     return true;\r
 269 }\r