2 * The Shibboleth License, Version 1.
4 * University Corporation for Advanced Internet Development, Inc.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are met:
11 * Redistributions of source code must retain the above copyright notice, this
12 * list of conditions and the following disclaimer.
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.
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
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.
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.
51 * shar.cpp -- the SHAR "main" code. All the functionality is elsewhere
52 * (in case you want to turn this into a library later).
54 * Created By: Derek Atkins <derek@ihtfp.com>
59 // eventually we might be able to support autoconf via cygwin...
60 #if defined (_MSC_VER) || defined(__BORLANDC__)
61 # include "config_win32.h"
68 #include <sys/select.h>
75 #include "shar-utils.h"
76 #include <log4cpp/Category.hh>
80 using namespace shibboleth;
81 using namespace shibtarget;
82 using namespace log4cpp;
85 # define FD_SETSIZE 1024
88 extern "C" void shibrpc_prog_1(struct svc_req* rqstp, register SVCXPRT* transp);
91 const char* shar_config = NULL;
92 const char* shar_schemadir = NULL;
93 bool shar_checkonly = false;
94 static int unlink_socket = 0;
96 static bool new_connection(IListener::ShibSocket& listener, const Iterator<ShibRPCProtocols>& protos)
98 IListener::ShibSocket sock;
100 // Accept the connection.
101 if (!ShibTargetConfig::getConfig().getINI()->getListener()->accept(listener, sock))
104 // We throw away the result because the children manage themselves...
105 new SharChild(sock,protos);
109 static void shar_svc_run(IListener::ShibSocket& listener, const Iterator<ShibRPCProtocols>& protos)
111 NDC ndc("shar_svc_run");
112 Category& log=Category::getInstance("SHAR");
117 FD_SET(listener, &readfds);
118 struct timeval tv = { 0, 0 };
121 switch (select(FD_SETSIZE, &readfds, 0, 0, &tv)) {
123 if (errno == EINTR) continue;
124 SHARUtils::log_error();
131 if (!new_connection(listener, protos))
132 log.error("new_connection failed");
135 log.info("shar_svc_run ended");
140 int real_main(int preinit)
142 static IListener::ShibSocket sock;
143 ShibRPCProtocols protos[1] = {
144 { SHIBRPC_PROG, SHIBRPC_VERS_1, shibrpc_prog_1 }
147 ShibTargetConfig& conf=ShibTargetConfig::getConfig();
150 // initialize the shib-target library
152 ShibTargetConfig::Listener |
153 ShibTargetConfig::SessionCache |
154 ShibTargetConfig::Metadata |
155 ShibTargetConfig::Trust |
156 ShibTargetConfig::Credentials |
157 ShibTargetConfig::AAP |
158 ShibTargetConfig::SHARExtensions |
159 (shar_checkonly ? (ShibTargetConfig::SHIREExtensions | ShibTargetConfig::RequestMapper) : 0)
162 shar_config=getenv("SHIBCONFIG");
164 shar_schemadir=getenv("SHIBSCHEMAS");
166 shar_schemadir=SHIB_SCHEMAS;
168 shar_config=SHIB_CONFIG;
169 if (!conf.init(shar_schemadir,shar_config)) {
170 fprintf(stderr, "configuration is invalid, check log for specific problems\n");
174 // If just a test run, bail.
175 if (shar_checkonly) {
176 fprintf(stderr, "overall configuration is loadable, check log for non-fatal problems\n");
180 const IListener* listener=conf.getINI()->getListener();
182 // Create the SHAR listener socket
183 if (!listener->create(sock))
186 // Bind to the proper port
187 if (!listener->bind(sock))
190 // Initialize the SHAR Utilitites
195 if (!shar_checkonly) {
196 shar_svc_run(sock, ArrayIterator<ShibRPCProtocols>(protos,1));
197 fprintf(stderr,"shar_svc_run returned\n");
199 // Finalize the SHAR, close all clients
201 conf.getINI()->getListener()->close(sock);
205 fprintf(stderr, "shar shutdown complete\n");
212 static void term_handler(int arg)
217 static int setup_signals(void)
219 NDC ndc("setup_signals");
222 memset(&sa, 0, sizeof (sa));
223 sa.sa_handler = SIG_IGN;
224 sa.sa_flags = SA_RESTART;
226 if (sigaction(SIGPIPE, &sa, NULL) < 0) {
227 SHARUtils::log_error();
231 memset(&sa, 0, sizeof (sa));
232 sa.sa_handler = term_handler;
233 sa.sa_flags = SA_RESTART;
235 if (sigaction(SIGHUP, &sa, NULL) < 0) {
236 SHARUtils::log_error();
239 if (sigaction(SIGINT, &sa, NULL) < 0) {
240 SHARUtils::log_error();
243 if (sigaction(SIGQUIT, &sa, NULL) < 0) {
244 SHARUtils::log_error();
247 if (sigaction(SIGTERM, &sa, NULL) < 0) {
248 SHARUtils::log_error();
254 static void usage(char* whoami)
256 fprintf(stderr, "usage: %s [-fcdt]\n", whoami);
257 fprintf(stderr, " -c\tconfig file to use.\n");
258 fprintf(stderr, " -d\tschema directory to use.\n");
259 fprintf(stderr, " -t\tcheck configuration file for problems.\n");
260 fprintf(stderr, " -f\tforce removal of listener socket.\n");
261 fprintf(stderr, " -h\tprint this help message.\n");
265 static int parse_args(int argc, char* argv[])
269 while ((opt = getopt(argc, argv, "cdfFh")) > 0) {
275 shar_schemadir=optarg;
290 int main(int argc, char *argv[])
292 IListener::ShibSocket sock;
293 ShibRPCProtocols protos[] = {
294 { SHIBRPC_PROG, SHIBRPC_VERS_1, shibrpc_prog_1 }
297 if (setup_signals() != 0)
300 if (parse_args(argc, argv) != 0)
304 shar_config=getenv("SHIBCONFIG");
306 shar_schemadir=getenv("SHIBSCHEMAS");
308 shar_schemadir=SHIB_SCHEMAS;
310 shar_config=SHIB_CONFIG;
312 // initialize the shib-target library
313 ShibTargetConfig& conf=ShibTargetConfig::getConfig();
315 ShibTargetConfig::Listener |
316 ShibTargetConfig::SessionCache |
317 ShibTargetConfig::Metadata |
318 ShibTargetConfig::Trust |
319 ShibTargetConfig::Credentials |
320 ShibTargetConfig::AAP |
321 ShibTargetConfig::SHARExtensions |
322 (shar_checkonly ? (ShibTargetConfig::SHIREExtensions | ShibTargetConfig::RequestMapper) : 0)
324 if (!conf.init(shar_schemadir,shar_config)) {
325 fprintf(stderr, "configuration is invalid, check log for specific problems\n");
330 fprintf(stderr, "overall configuration is loadable, check log for non-fatal problems\n");
332 const IListener* listener=conf.getINI()->getListener();
334 // Create the SHAR listener socket
335 if (!listener->create(sock))
338 // Bind to the proper port
339 if (!listener->bind(sock, unlink_socket==1))
342 // Initialize the SHAR Utilitites
346 shar_svc_run(sock, ArrayIterator<ShibRPCProtocols>(protos,1));
348 /* Finalize the SHAR, close all clients */
350 fprintf(stderr, "shar utils finalized\n");
352 listener->close(sock);
353 fprintf(stderr, "shib socket closed\n");
357 fprintf(stderr, "shar shutdown complete\n");