-/* $Id: metaproxy_prog.cpp,v 1.6 2006-06-10 14:29:12 adam Exp $
- Copyright (c) 2005-2006, Index Data.
+/* This file is part of Metaproxy.
+ Copyright (C) 2005-2010 Index Data
- See the LICENSE file for details
- */
+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
+Software Foundation; either version 2, or (at your option) any later
+version.
+
+Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+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 <boost/program_options.hpp>
-namespace po = boost::program_options;
+#include <yaz/log.h>
+#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 "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;
-int main(int argc, char **argv)
+mp::RouterFleXML *routerp = 0;
+
+#if HAVE_UNISTD_H
+static pid_t process_group = 0;
+
+static void sig_term_handler(int s)
+{
+ kill(-process_group, SIGTERM); /* kill all children processes as well */
+ _exit(0);
+}
+#endif
+
+static void handler(void *data)
+{
+ routerp = (mp::RouterFleXML*) data;
+
+#if HAVE_UNISTD_H
+ /* make the current working process group leader */
+ setpgid(0, 0);
+ process_group = getpgid(0); // save process group ID
+
+ signal(SIGTERM, sig_term_handler);
+#endif
+
+ mp::Package pack;
+ pack.router(*routerp).move(); /* should never exit */
+}
+
+static int sc_main(
+ yaz_sc_t s,
+ int argc, char **argv)
{
try
{
- po::options_description desc("Allowed options");
- desc.add_options()
- ("help,h", "produce help message")
- ("version,V", "show version")
- ("config", po::value< std::vector<std::string> >(), "xml config")
- ;
-
- po::positional_options_description p;
- p.add("config", -1);
-
- po::variables_map vm;
- po::store(po::command_line_parser(argc, argv).
- options(desc).positional(p).run(), vm);
- po::notify(vm);
-
- if (vm.count("help")) {
- std::cout << desc << "\n";
- return 1;
- }
- if (vm.count("version")) {
- std::cout << "Metaproxy " VERSION "\n";
- return 0;
- }
- xmlDocPtr doc = 0;
- if (vm.count("config"))
- {
- std::vector<std::string> config_fnames =
- vm["config"].as< std::vector<std::string> >();
+ const char *fname = 0;
+ int ret;
+ char *arg;
+ unsigned mode = 0;
+ const char *pidfile = 0;
+ const char *uid = 0;
- if (config_fnames.size() != 1)
- {
- std::cerr << "Only one configuration must be given\n";
- std::exit(1);
- }
-
- doc = xmlParseFile(config_fnames[0].c_str());
- if (!doc)
+ while ((ret = options("c{config}:Dh{help}l:p:u:V{version}w:X",
+ argv, argc, &arg)) != -2)
+ {
+ switch (ret)
{
- std::cerr << "xmlParseFile failed\n";
- std::exit(1);
+ 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"
+#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 '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
+ )
+ {
+ 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;
}
}
- else
+ if (!fname)
{
- std::cerr << "No configuration given\n";
- std::exit(1);
+ std::cerr << "No configuration given; use -h for help\n";
+ return 1;
}
- if (doc)
+
+ yaz_log(YLOG_LOG, "Metaproxy start " VERSION
+#ifdef VERSION_SHA1
+ " " VERSION_SHA1
+#endif
+ );
+
+ xmlDocPtr doc = xmlReadFile(fname,
+ NULL,
+ XML_PARSE_XINCLUDE + XML_PARSE_NOBLANKS
+ + XML_PARSE_NSCLEAN + XML_PARSE_NONET );
+
+ if (!doc)
{
- try {
- mp::FactoryStatic factory;
- mp::RouterFleXML router(doc, factory);
- mp::Package pack;
- pack.router(router).move();
- }
- catch (std::runtime_error &e) {
- std::cerr << "std::runtime error: " << e.what() << "\n";
- std::exit(1);
- }
- xmlFreeDoc(doc);
+ yaz_log (YLOG_FATAL,"XML parsing failed");
+ return 1;
}
- }
- catch (po::unknown_option &e) {
- std::cerr << e.what() << "; use --help for list of options\n";
- std::exit(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,
+#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);
+
+ yaz_sc_running(s);
+
+ yaz_daemon("metaproxy", mode, handler, 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() );
+ return 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() );
+ return 1;
}
catch ( ... ) {
- std::cerr << "Unknown Exception" << std::endl;
- std::exit(1);
+ yaz_log(YLOG_FATAL, "Unknown Exception");
+ return 1;
}
- std::exit(0);
+ return 0;
+}
+
+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
*/
+