-/* $Id: metaproxy_prog.cpp,v 1.15 2008-02-26 23:56:05 adam Exp $
- Copyright (c) 2005-2008, Index Data.
-
-This file is part of Metaproxy.
+/* This file is part of Metaproxy.
+ Copyright (C) 2005-2012 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
for more details.
You should have received a copy of the GNU General Public License
-along with Metaproxy; see the file LICENSE. If not, write to the
-Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
-02111-1307, USA.
- */
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+*/
#include "config.hpp"
#include <yaz/options.h>
#include <yaz/daemon.h>
+#include <yaz/sc.h>
#include <iostream>
#include <stdexcept>
#include <libxml/xinclude.h>
-#include "filter.hpp"
-#include "package.hpp"
+#include <metaproxy/filter.hpp>
+#include <metaproxy/package.hpp>
+#include <metaproxy/util.hpp>
#include "router_flexml.hpp"
#include "factory_static.hpp"
+#if HAVE_UNISTD_H
+#include <unistd.h>
+#endif
+#include <signal.h>
+#ifdef WIN32
+#include <direct.h>
+#include <io.h>
+#include <process.h>
+#endif
+
namespace mp = metaproxy_1;
-static void handler(void *data)
+mp::RouterFleXML *routerp = 0;
+
+#if HAVE_UNISTD_H
+static pid_t process_group = 0;
+
+static void sig_usr1_handler(int s)
+{
+ yaz_log(YLOG_LOG, "metaproxy received SIGUSR1");
+ routerp->stop();
+}
+
+static void sig_term_handler(int s)
+{
+ yaz_log(YLOG_LOG, "metaproxy received SIGTERM");
+ yaz_log(YLOG_LOG, "metaproxy stop");
+ kill(-process_group, SIGTERM); /* kill all children processes as well */
+ _exit(0);
+}
+#endif
+
+static void work_common(void *data)
{
- mp::RouterFleXML *routerp = (mp::RouterFleXML*) data;
+#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::RouterFleXML*) data;
+ routerp->start();
mp::Package pack;
pack.router(*routerp).move();
+ yaz_log(YLOG_LOG, "metaproxy stop"); /* only for graceful stop */
+ _exit(0);
}
-
-int main(int argc, char **argv)
+
+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;
+
+ while ((ret = options("c{config}:Dh{help}l:p:tu: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}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"
+ " -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"
+#endif
+
+ << std::endl;
+ break;
+ case 'l':
+ yaz_log_init_file(arg);
+ break;
+ case 'p':
+ pidfile = arg;
+ break;
+ case 't':
+ test_config = true;
+ break;
+ case 'u':
+ uid = 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)
+#else
+ chdir(arg)
+#endif
+ )
{
- 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"
- " -X debug mode (no fork/daemon mode)\n"
- << 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 << "Metaproxy " VERSION "\n";
- break;
- case 'X':
- mode = YAZ_DAEMON_DEBUG;
- break;
- case -1:
- std::cerr << "bad option: " << arg << std::endl;
- std::exit(1);
+ 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;
+ return 1;
}
- if (!fname)
- {
- std::cerr << "No configuration given\n";
- std::exit(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)
- {
- std::cerr << "XML parsing failed\n";
- std::exit(1);
- }
- // and perform Xinclude then
- if (xmlXIncludeProcess(doc) > 0) {
- std::cerr << "processing XInclude directive\n";
- }
+ if (!test_config)
+ yaz_log(YLOG_LOG, "metaproxy start " VERSION
+#ifdef VERSION_SHA1
+ " " VERSION_SHA1
+#endif
+ );
+
+ 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::FactoryStatic factory;
- mp::RouterFleXML router(doc, factory, false);
+ mp::RouterFleXML *router =
+ new mp::RouterFleXML(doc, factory, test_config, wrbuf_cstr(base_path));
+ if (!test_config)
+ {
+
+ 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) {
- std::cerr << "std::logic error: " << e.what() << "\n";
- std::exit(1);
+ yaz_log(YLOG_FATAL,"std::logic error: %s" , e.what() );
+ ret = 1;
}
catch (std::runtime_error &e) {
- std::cerr << "std::runtime error: " << e.what() << "\n";
- std::exit(1);
+ yaz_log(YLOG_FATAL, "std::runtime error: %s" , e.what() );
+ ret = 1;
}
catch ( ... ) {
- std::cerr << "Unknown Exception" << std::endl;
- std::exit(1);
+ yaz_log(YLOG_FATAL, "Unknown Exception");
+ ret = 1;
}
- std::exit(0);
+ xmlFreeDoc(doc);
+ return ret;
+}
+
+static void sc_stop(yaz_sc_t s)
+{
+ return;
}
+int main(int argc, char **argv)
+{
+ int ret;
+ yaz_sc_t s = yaz_sc_create("metaproxy", "metaproxy");
+
+ ret = yaz_sc_program(s, argc, argv, sc_main, sc_stop);
+
+ yaz_sc_destroy(&s);
+ exit(ret);
+}
/*
* Local variables:
* c-basic-offset: 4
+ * c-file-style: "Stroustrup"
* indent-tabs-mode: nil
- * c-file-style: "stroustrup"
* End:
* vim: shiftwidth=4 tabstop=8 expandtab
*/
+