X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fmetaproxy_prog.cpp;h=481901bb6e15cba207f66951f264f7a98b919ace;hb=0098e12fb155644a8befc6c5c5a3d04308128e72;hp=11b04b23073902299a620f22074176d859c270d0;hpb=b078c89a9ee8e512b53560ff48cd041b25ac2cdf;p=metaproxy-moved-to-github.git diff --git a/src/metaproxy_prog.cpp b/src/metaproxy_prog.cpp index 11b04b2..481901b 100644 --- a/src/metaproxy_prog.cpp +++ b/src/metaproxy_prog.cpp @@ -1,5 +1,5 @@ /* 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 @@ -29,8 +29,8 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #include #include -#include "router_flexml.hpp" -#include "factory_static.hpp" +#include +#include #if HAVE_UNISTD_H #include @@ -44,158 +44,240 @@ Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA namespace mp = metaproxy_1; -mp::RouterFleXML *routerp = 0; +mp::RouterXML *routerp = 0; -static void sig_term_handler(int s) +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) { - delete routerp; + routerp->stop(signo); } - exit(0); } +#endif -static void handler(void *data) +static void work_common(void *data) { - 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(); + 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_FATAL, "XInclude processing failed"); - return 1; - } - WRBUF base_path = wrbuf_alloc(); - const char *last_p = strrchr(fname, + 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); - - mp::FactoryStatic factory; - mp::RouterFleXML *router = - new mp::RouterFleXML(doc, factory, false, wrbuf_cstr(base_path)); - wrbuf_destroy(base_path); + ); + 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_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) @@ -214,8 +296,6 @@ int main(int argc, char **argv) exit(ret); } - - /* * Local variables: * c-basic-offset: 4