shibd/shibd.cpp

   1 /*\r
   2  *  Copyright 2001-2007 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  * shar.cpp -- the shibd "main" code.  All the functionality is elsewhere\r
  19  *\r
  20  * Created By:  Derek Atkins <derek@ihtfp.com>\r
  21  *\r
  22  * $Id: shar.cpp 2164 2007-02-11 05:26:18 +0000 (Sun, 11 Feb 2007) cantor $\r
  23  */\r
  24 \r
  25 \r
  26 // eventually we might be able to support autoconf via cygwin...\r
  27 #if defined (_MSC_VER) || defined(__BORLANDC__)\r
  28 # include "config_win32.h"\r
  29 #else\r
  30 # include "config.h"\r
  31 #endif\r
  32 \r
  33 #ifdef WIN32\r
  34 # define _CRT_NONSTDC_NO_DEPRECATE 1\r
  35 # define _CRT_SECURE_NO_DEPRECATE 1\r
  36 #endif\r
  37 \r
  38 #include <shibsp/SPConfig.h>\r
  39 \r
  40 #ifdef HAVE_UNISTD_H\r
  41 #include <unistd.h>\r
  42 #include <sys/select.h>\r
  43 #endif\r
  44 \r
  45 #include <stdio.h>\r
  46 #include <signal.h>\r
  47 #include <log4cpp/Category.hh>\r
  48 #include <shibsp/ServiceProvider.h>\r
  49 #include <shibsp/remoting/ListenerService.h>\r
  50 #include <xercesc/util/XMLUniDefs.hpp>\r
  51 #include <xmltooling/XMLToolingConfig.h>\r
  52 #include <xmltooling/util/XMLHelper.h>\r
  53 \r
  54 using namespace shibsp;\r
  55 using namespace log4cpp;\r
  56 using namespace xmltooling;\r
  57 using namespace std;\r
  58 \r
  59 bool shibd_shutdown = false;\r
  60 const char* shar_config = NULL;\r
  61 const char* shar_schemadir = NULL;\r
  62 bool shar_checkonly = false;\r
  63 static int unlink_socket = 0;\r
  64 const char* pidfile = NULL;\r
  65 \r
  66 #ifdef WIN32\r
  67 \r
  68 //#include <CRTDBG.H>\r
  69 \r
  70 #define nNoMansLandSize 4\r
  71 typedef struct _CrtMemBlockHeader\r
  72 {\r
  73         struct _CrtMemBlockHeader * pBlockHeaderNext;\r
  74         struct _CrtMemBlockHeader * pBlockHeaderPrev;\r
  75         char *                      szFileName;\r
  76         int                         nLine;\r
  77         size_t                      nDataSize;\r
  78         int                         nBlockUse;\r
  79         long                        lRequest;\r
  80         unsigned char               gap[nNoMansLandSize];\r
  81         /* followed by:\r
  82          *  unsigned char           data[nDataSize];\r
  83          *  unsigned char           anotherGap[nNoMansLandSize];\r
  84          */\r
  85 } _CrtMemBlockHeader;\r
  86 \r
  87 /*\r
  88 int MyAllocHook(int nAllocType, void *pvData,\r
  89       size_t nSize, int nBlockUse, long lRequest,\r
  90       const unsigned char * szFileName, int nLine)\r
  91 {\r
  92     if ( nBlockUse == _CRT_BLOCK )\r
  93       return( TRUE );\r
  94     if (nAllocType == _HOOK_FREE) {\r
  95         _CrtMemBlockHeader* ptr = (_CrtMemBlockHeader*)(((_CrtMemBlockHeader *)pvData)-1);\r
  96         if (ptr->nDataSize == 8192)\r
  97             fprintf(stderr,"free  request %u size %u\n", ptr->lRequest, ptr->nDataSize);\r
  98     }\r
  99     else if (nAllocType == _HOOK_ALLOC && nSize == 8192)\r
 100         fprintf(stderr,"%s request %u size %u\n", ((nAllocType == _HOOK_ALLOC) ? "alloc" : "realloc"), lRequest, nSize);\r
 101     return (TRUE);\r
 102 }\r
 103 */\r
 104 \r
 105 int real_main(int preinit)\r
 106 {\r
 107     SPConfig& conf=SPConfig::getConfig();\r
 108     if (preinit) {\r
 109 \r
 110         // Initialize the SP library.\r
 111         conf.setFeatures(\r
 112             SPConfig::Listener |\r
 113             SPConfig::Caching |\r
 114             SPConfig::Metadata |\r
 115             SPConfig::Trust |\r
 116             SPConfig::Credentials |\r
 117             SPConfig::AttributeResolver |\r
 118             SPConfig::OutOfProcess |\r
 119             (shar_checkonly ? (SPConfig::InProcess | SPConfig::RequestMapping) : SPConfig::Logging)\r
 120             );\r
 121         if (!shar_config)\r
 122             shar_config=getenv("SHIBCONFIG");\r
 123         if (!shar_schemadir)\r
 124             shar_schemadir=getenv("SHIBSCHEMAS");\r
 125         if (!shar_schemadir)\r
 126             shar_schemadir=SHIBSP_SCHEMAS;\r
 127         if (!shar_config)\r
 128             shar_config=SHIBSP_CONFIG;\r
 129         if (!conf.init(shar_schemadir)) {\r
 130             fprintf(stderr, "configuration is invalid, see console for specific problems\n");\r
 131             return -1;\r
 132         }\r
 133         \r
 134         try {\r
 135             fprintf(stderr, "loading configuration file: %s\n", shar_config);\r
 136             static const XMLCh path[] = UNICODE_LITERAL_4(p,a,t,h);\r
 137             static const XMLCh validate[] = UNICODE_LITERAL_8(v,a,l,i,d,a,t,e);\r
 138             DOMDocument* dummydoc=XMLToolingConfig::getConfig().getParser().newDocument();\r
 139             XercesJanitor<DOMDocument> docjanitor(dummydoc);\r
 140             DOMElement* dummy = dummydoc->createElementNS(NULL,path);\r
 141             auto_ptr_XMLCh src(shar_config);\r
 142             dummy->setAttributeNS(NULL,path,src.get());\r
 143             dummy->setAttributeNS(NULL,validate,xmlconstants::XML_ONE);\r
 144     \r
 145             conf.setServiceProvider(conf.ServiceProviderManager.newPlugin(XML_SERVICE_PROVIDER,dummy));\r
 146             conf.getServiceProvider()->init();\r
 147         }\r
 148         catch (exception& ex) {\r
 149             fprintf(stderr, "caught exception while loading configuration: %s\n", ex.what());\r
 150             conf.term();\r
 151             return -2;\r
 152         }\r
 153 \r
 154         // If just a test run, bail.\r
 155         if (shar_checkonly) {\r
 156             fprintf(stdout, "overall configuration is loadable, check console for non-fatal problems\n");\r
 157             return 0;\r
 158         }\r
 159     }\r
 160     else {\r
 161 \r
 162         //_CrtSetAllocHook(MyAllocHook);\r
 163 \r
 164         // Run the listener\r
 165         if (!shar_checkonly) {\r
 166 \r
 167             // Run the listener.\r
 168             if (!conf.getServiceProvider()->getListenerService()->run(&shibd_shutdown)) {\r
 169                 fprintf(stderr, "listener failed to enter listen loop\n");\r
 170                 return -3;\r
 171             }\r
 172         }\r
 173 \r
 174         conf.term();\r
 175     }\r
 176     return 0;\r
 177 }\r
 178 \r
 179 #else\r
 180 \r
 181 static void term_handler(int arg)\r
 182 {\r
 183     shibd_shutdown = true;\r
 184 }\r
 185 \r
 186 static int setup_signals(void)\r
 187 {\r
 188     struct sigaction sa;\r
 189     memset(&sa, 0, sizeof (sa));\r
 190     sa.sa_handler = SIG_IGN;\r
 191     sa.sa_flags = SA_RESTART;\r
 192 \r
 193     if (sigaction(SIGPIPE, &sa, NULL) < 0) {\r
 194         return -1;\r
 195     }\r
 196 \r
 197     memset(&sa, 0, sizeof (sa));\r
 198     sa.sa_handler = term_handler;\r
 199     sa.sa_flags = SA_RESTART;\r
 200 \r
 201     if (sigaction(SIGHUP, &sa, NULL) < 0) {\r
 202         return -1;\r
 203     }\r
 204     if (sigaction(SIGINT, &sa, NULL) < 0) {\r
 205         return -1;\r
 206     }\r
 207     if (sigaction(SIGQUIT, &sa, NULL) < 0) {\r
 208         return -1;\r
 209     }\r
 210     if (sigaction(SIGTERM, &sa, NULL) < 0) {\r
 211         return -1;\r
 212     }\r
 213     return 0;\r
 214 }\r
 215 \r
 216 static void usage(char* whoami)\r
 217 {\r
 218     fprintf(stderr, "usage: %s [-fcdt]\n", whoami);\r
 219     fprintf(stderr, "  -c\tconfig file to use.\n");\r
 220     fprintf(stderr, "  -d\tschema directory to use.\n");\r
 221     fprintf(stderr, "  -t\tcheck configuration file for problems.\n");\r
 222     fprintf(stderr, "  -f\tforce removal of listener socket.\n");\r
 223     fprintf(stderr, "  -p\tpid file to use.\n");\r
 224     fprintf(stderr, "  -h\tprint this help message.\n");\r
 225     exit(1);\r
 226 }\r
 227 \r
 228 static int parse_args(int argc, char* argv[])\r
 229 {\r
 230     int opt;\r
 231 \r
 232     while ((opt = getopt(argc, argv, "c:d:p:fth")) > 0) {\r
 233         switch (opt) {\r
 234             case 'c':\r
 235                 shar_config=optarg;\r
 236                 break;\r
 237             case 'd':\r
 238                 shar_schemadir=optarg;\r
 239                 break;\r
 240             case 'f':\r
 241                 unlink_socket = 1;\r
 242                 break;\r
 243             case 't':\r
 244                 shar_checkonly=true;\r
 245                 break;\r
 246             case 'p':\r
 247                 pidfile=optarg;\r
 248                 break;\r
 249             default:\r
 250                 return -1;\r
 251         }\r
 252     }\r
 253     return 0;\r
 254 }\r
 255 \r
 256 int main(int argc, char *argv[])\r
 257 {\r
 258     if (setup_signals() != 0)\r
 259         return -1;\r
 260 \r
 261     if (parse_args(argc, argv) != 0)\r
 262         usage(argv[0]);\r
 263 \r
 264     if (!shar_config)\r
 265         shar_config=getenv("SHIBCONFIG");\r
 266     if (!shar_schemadir)\r
 267         shar_schemadir=getenv("SHIBSCHEMAS");\r
 268     if (!shar_schemadir)\r
 269         shar_schemadir=SHIBSP_SCHEMAS;\r
 270     if (!shar_config)\r
 271         shar_config=SHIBSP_CONFIG;\r
 272 \r
 273     // initialize the shib-target library\r
 274     SPConfig& conf=SPConfig::getConfig();\r
 275     conf.setFeatures(\r
 276         SPConfig::Listener |\r
 277         SPConfig::Caching |\r
 278         SPConfig::Metadata |\r
 279         SPConfig::Trust |\r
 280         SPConfig::Credentials |\r
 281         SPConfig::AttributeResolver |\r
 282         SPConfig::OutOfProcess |\r
 283         (shar_checkonly ? (SPConfig::InProcess | SPConfig::RequestMapping) : SPConfig::Logging)\r
 284         );\r
 285     if (!conf.init(shar_schemadir)) {\r
 286         fprintf(stderr, "configuration is invalid, check console for specific problems\n");\r
 287         return -1;\r
 288     }\r
 289 \r
 290     try {\r
 291         fprintf(stderr, "loading configuration file: %s\n", shar_config);\r
 292         static const XMLCh path[] = UNICODE_LITERAL_4(p,a,t,h);\r
 293         static const XMLCh validate[] = UNICODE_LITERAL_8(v,a,l,i,d,a,t,e);\r
 294         DOMDocument* dummydoc=XMLToolingConfig::getConfig().getParser().newDocument();\r
 295         XercesJanitor<DOMDocument> docjanitor(dummydoc);\r
 296         DOMElement* dummy = dummydoc->createElementNS(NULL,path);\r
 297         auto_ptr_XMLCh src(shar_config);\r
 298         dummy->setAttributeNS(NULL,path,src.get());\r
 299         dummy->setAttributeNS(NULL,validate,xmlconstants::XML_ONE);\r
 300 \r
 301         conf.setServiceProvider(conf.ServiceProviderManager.newPlugin(XML_SERVICE_PROVIDER,dummy));\r
 302         conf.getServiceProvider()->init();\r
 303     }\r
 304     catch (exception& ex) {\r
 305         fprintf(stderr, "caught exception while loading configuration: %s\n", ex.what());\r
 306         conf.term();\r
 307         return -2;\r
 308     }\r
 309 \r
 310     if (shar_checkonly)\r
 311         fprintf(stderr, "overall configuration is loadable, check console for non-fatal problems\n");\r
 312     else {\r
 313 \r
 314         // Write the pid file\r
 315         if (pidfile) {\r
 316             FILE* pidf = fopen(pidfile, "w");\r
 317             if (pidf) {\r
 318                 fprintf(pidf, "%d\n", getpid());\r
 319                 fclose(pidf);\r
 320             } else {\r
 321                 perror(pidfile);  // keep running though\r
 322             }\r
 323         }\r
 324     \r
 325         // Run the listener\r
 326         if (!conf.getServiceProvider()->getListenerService()->run(&shibd_shutdown)) {\r
 327             fprintf(stderr, "listener failed to enter listen loop\n");\r
 328             return -3;\r
 329         }\r
 330     }\r
 331 \r
 332     conf.term();\r
 333     if (pidfile)\r
 334         unlink(pidfile);\r
 335     return 0;\r
 336 }\r
 337 \r
 338 #endif\r