/* This file is part of Metaproxy.
- Copyright (C) 2005-2010 Index Data
+ Copyright (C) Index Data
Metaproxy is free software; you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free
#include <stdexcept>
#include <libxml/xinclude.h>
-#include "filter.hpp"
-#include "package.hpp"
-#include "router_flexml.hpp"
-#include "factory_static.hpp"
+#include <metaproxy/filter.hpp>
+#include <metaproxy/package.hpp>
+#include <metaproxy/util.hpp>
+#include <metaproxy/router_xml.hpp>
#if HAVE_UNISTD_H
#include <unistd.h>
#endif
+#include <signal.h>
#ifdef WIN32
#include <direct.h>
#include <io.h>
namespace mp = metaproxy_1;
-static void handler(void *data)
+mp::RouterXML *routerp = 0;
+
+static void set_log_prefix(void)
+{
+#if HAVE_UNISTD_H
+ char str[80];
+
+ sprintf(str, "%lld", (long long) getpid());
+ yaz_log_init_prefix(str);
+#endif
+}
+
+#if HAVE_UNISTD_H
+static pid_t process_group = 0;
+static int sig_received = 0;
+
+static void sig_usr1_handler(int signo)
+{
+ sig_received = signo;
+ if (routerp)
+ routerp->stop(signo);
+}
+
+static void sig_term_handler(int signo)
+{
+ sig_received = signo;
+ if (routerp)
+ {
+ routerp->stop(signo);
+ }
+}
+#endif
+
+static void work_common(void *data)
{
- mp::RouterFleXML *routerp = (mp::RouterFleXML*) data;
+ set_log_prefix();
+#if HAVE_UNISTD_H
+ process_group = getpgid(0); // save process group ID
+
+ signal(SIGTERM, sig_term_handler);
+ signal(SIGUSR1, sig_usr1_handler);
+#endif
+ routerp = (mp::RouterXML*) data;
+ routerp->start();
mp::Package pack;
pack.router(*routerp).move();
+#if HAVE_UNISTD_H
+ switch (sig_received)
+ {
+ case SIGTERM:
+ yaz_log(YLOG_LOG, "metaproxy received SIGTERM");
+ break;
+ case SIGUSR1:
+ yaz_log(YLOG_LOG, "metaproxy received SIGUSR1");
+ break;
+ case 0:
+ break;
+ default:
+ yaz_log(YLOG_LOG, "metaproxy received signo=%d", sig_received);
+ }
+#endif
+ yaz_log(YLOG_LOG, "metaproxy stop");
+ delete routerp;
+ routerp = 0;
+#if HAVE_UNISTD_H
+ kill(-process_group, SIGTERM); /* kill all children processes as well */
+#endif
+ _exit(0);
}
-
+
+static void work_debug(void *data)
+{
+ work_common(data);
+}
+
+static void work_normal(void *data)
+{
+#if HAVE_UNISTD_H
+ /* make the current working process group leader */
+ setpgid(0, 0);
+#endif
+ work_common(data);
+}
+
static int sc_main(
yaz_sc_t s,
int argc, char **argv)
{
- try
+ bool test_config = false;
+ const char *fname = 0;
+ int ret;
+ char *arg;
+ unsigned mode = 0;
+ const char *pidfile = 0;
+ const char *uid = 0;
+
+ set_log_prefix();
+
+ while ((ret = options("c{config}:Dh{help}l:m:p:tu:v:V{version}w:X",
+ argv, argc, &arg)) != -2)
{
- const char *fname = 0;
- int ret;
- char *arg;
- unsigned mode = 0;
- const char *pidfile = 0;
- const char *uid = 0;
-
- while ((ret = options("c{config}:Dh{help}l:p:u:V{version}w:X",
- argv, argc, &arg)) != -2)
+ switch (ret)
{
- switch (ret)
- {
- case 'c':
- fname = arg;
- break;
- case 'D':
- mode = YAZ_DAEMON_FORK|YAZ_DAEMON_KEEPALIVE;
- break;
- case 'h':
- std::cerr << "metaproxy\n"
- " -h|--help help\n"
- " -V|--version version\n"
- " -c|--config f config filename\n"
- " -D daemon and keepalive operation\n"
- " -l f log file f\n"
- " -p f pid file f\n"
- " -u id change uid to id\n"
- " -w dir changes working directory to dir\n"
- " -X debug mode (no fork/daemon mode)\n"
+ case 'c':
+ fname = arg;
+ break;
+ case 'D':
+ mode = YAZ_DAEMON_FORK|YAZ_DAEMON_KEEPALIVE;
+ break;
+ case 'h':
+ std::cerr << "metaproxy\n"
+ " -h|--help help\n"
+ " -V|--version version\n"
+ " -v level\n"
+ " -c|--config f config filename\n"
+ " -D daemon and keepalive operation\n"
+ " -l f log file f\n"
+ " -m logformat log time format (strftime)\n"
+ " -p f pid file f\n"
+ " -t test configuration\n"
+ " -u id change uid to id\n"
+ " -w dir changes working directory to dir\n"
+ " -X debug mode (no fork/daemon mode)\n"
#ifdef WIN32
- " -install install windows service\n"
- " -remove remove windows service\n"
+ " -install install windows service\n"
+ " -remove remove windows service\n"
#endif
- << std::endl;
- break;
- case 'l':
- yaz_log_init_file(arg);
- break;
- case 'p':
- pidfile = arg;
- break;
- case 'u':
- uid = arg;
- break;
- case 'V':
- std::cout << VERSION "\n";
- return 0;
- break;
- case 'w':
- if (
+ << std::endl;
+ break;
+ case 'l':
+ yaz_log_init_file(arg);
+ break;
+ case 'm':
+ yaz_log_time_format(arg);
+ break;
+ case 'p':
+ pidfile = arg;
+ break;
+ case 't':
+ test_config = true;
+ break;
+ case 'u':
+ uid = arg;
+ break;
+ case 'v':
+ yaz_log_init_level(yaz_log_mask_str(arg));
+ break;
+ case 'V':
+ std::cout << VERSION;
+#ifdef VERSION_SHA1
+ std::cout << " " VERSION_SHA1;
+#endif
+ std::cout << "\n";
+ return 0;
+ break;
+ case 'w':
+ if (
#ifdef WIN32
- _chdir(arg)
+ _chdir(arg)
#else
- chdir(arg)
+ chdir(arg)
#endif
- )
- {
- std::cerr << "chdir " << arg << " failed" << std::endl;
- return 1;
- }
- case 'X':
- mode = YAZ_DAEMON_DEBUG;
- break;
- case -1:
- std::cerr << "bad option: " << arg << std::endl;
+ )
+ {
+ std::cerr << "chdir " << arg << " failed" << std::endl;
return 1;
}
- }
- if (!fname)
- {
- std::cerr << "No configuration given; use -h for help\n";
+ case 'X':
+ mode = YAZ_DAEMON_DEBUG;
+ break;
+ case -1:
+ std::cerr << "bad option: " << arg << std::endl;
return 1;
}
+ }
+ if (!fname)
+ {
+ std::cerr << "No configuration given; use -h for help\n";
+ return 1;
+ }
- yaz_log(YLOG_LOG, "Metaproxy " VERSION " started");
- xmlDocPtr doc = xmlReadFile(fname,
- NULL,
- XML_PARSE_XINCLUDE + XML_PARSE_NOBLANKS
- + XML_PARSE_NSCLEAN + XML_PARSE_NONET );
-
- if (!doc)
- {
- yaz_log (YLOG_FATAL,"XML parsing failed");
- return 1;
- }
- // and perform Xinclude then
- int r = xmlXIncludeProcess(doc);
- if (r == -1)
+ yaz_log(YLOG_LOG, "metaproxy %s " VERSION
+#ifdef VERSION_SHA1
+ " " VERSION_SHA1
+#endif
+ , test_config ? "test" : "start"
+ );
+
+ yaz_log_xml_errors(0, YLOG_LOG);
+ xmlDocPtr doc = xmlReadFile(fname,
+ NULL,
+ XML_PARSE_XINCLUDE + XML_PARSE_NOBLANKS
+ + XML_PARSE_NSCLEAN + XML_PARSE_NONET );
+
+ if (!doc)
+ {
+ yaz_log(YLOG_FATAL,"XML parsing failed");
+ return 1;
+ }
+ // and perform Xinclude then
+ int r = xmlXIncludeProcess(doc);
+ if (r == -1)
+ {
+ yaz_log(YLOG_FATAL, "XInclude processing failed");
+ return 1;
+ }
+ mp::wrbuf base_path;
+ const char *last_p = strrchr(fname,
+#ifdef WIN32
+ '\\'
+#else
+ '/'
+#endif
+ );
+ if (last_p)
+ wrbuf_write(base_path, fname, last_p - fname);
+ else
+ wrbuf_puts(base_path, ".");
+ ret = 0;
+ try {
+ mp::RouterXML *router =
+ new mp::RouterXML(doc, test_config, wrbuf_cstr(base_path));
+ if (!test_config)
{
- yaz_log(YLOG_FATAL, "XInclude processing failed");
- return 1;
- }
- mp::FactoryStatic factory;
- mp::RouterFleXML router(doc, factory, false);
- yaz_sc_running(s);
+ yaz_sc_running(s);
- yaz_daemon("metaproxy", mode, handler, &router, pidfile, uid);
+ yaz_daemon("metaproxy", mode, mode == YAZ_DAEMON_DEBUG ?
+ work_debug : work_normal, router, pidfile, uid);
+ }
}
catch (std::logic_error &e) {
- yaz_log (YLOG_FATAL,"std::logic error: %s" , e.what() );
- return 1;
+ yaz_log(YLOG_FATAL,"std::logic error: %s" , e.what() );
+ ret = 1;
}
catch (std::runtime_error &e) {
yaz_log(YLOG_FATAL, "std::runtime error: %s" , e.what() );
- return 1;
+ ret = 1;
}
catch ( ... ) {
yaz_log(YLOG_FATAL, "Unknown Exception");
- return 1;
+ ret = 1;
}
- return 0;
+ xmlFreeDoc(doc);
+ if (test_config)
+ yaz_log(YLOG_LOG, "metaproxy test exit code %d", ret);
+ return ret;
}
static void sc_stop(yaz_sc_t s)
exit(ret);
}
-
-
/*
* Local variables:
* c-basic-offset: 4