-/* $Id: yaz-proxy-main.cpp,v 1.4 2004-04-22 07:46:21 adam Exp $
- Copyright (c) 1998-2004, Index Data.
-
-This file is part of the yaz-proxy.
+/* This file is part of YAZ proxy
+ Copyright (C) 1998-2011 Index Data
YAZ proxy 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 YAZ proxy; 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
+*/
-#ifdef WIN32
-#else
#include <signal.h>
+#if HAVE_UNISTD_H
#include <unistd.h>
-#include <pwd.h>
#endif
+#if HAVE_SYS_TYPES_H
#include <sys/types.h>
-
-#include <stdarg.h>
-
-#if HAVE_GETRLIMIT
+#endif
+#if HAVE_SYS_TIME_H
#include <sys/time.h>
+#endif
+#if HAVE_SYS_RESOURCE_H
#include <sys/resource.h>
#endif
+#include <stdarg.h>
+#include <stdlib.h>
+
#include <yaz/log.h>
#include <yaz/options.h>
+#include <yaz/daemon.h>
-#include <yaz++/socket-manager.h>
-#include <yaz++/pdu-assoc.h>
+#include <yazpp/socket-manager.h>
+#include <yazpp/pdu-assoc.h>
#include <yazproxy/proxy.h>
-#if HAVE_XSLT
+#if YAZ_HAVE_XSLT
#include <libxml/parser.h>
#include <libxml/tree.h>
#include <libxslt/xsltutils.h>
#include <libxslt/transform.h>
#endif
+#if YAZ_HAVE_EXSLT
+#include <libexslt/exslt.h>
+#endif
+
+using namespace yazpp_1;
+
void usage(char *prog)
{
- fprintf (stderr, "%s: [-c config] [-l log] [-a log] [-v level] [-t target] "
- "[-u uid] [-p pidfile] @:port\n", prog);
+ fprintf (stderr, "%s: [-a log] [-c config] [-D]\n"
+ " [-i sec] [-l log] [-m num] [-n num] [-p pidfile]"
+ " [-t target] [-T sec] [-u uid]\n"
+ " [-v level] [-X] [-V] @:port\n", prog);
exit (1);
}
static char *pid_fname = 0;
static char *uid = 0;
static char *log_file = 0;
-static int debug = 0;
+static unsigned int daemon_flags = YAZ_DAEMON_KEEPALIVE;
static int no_limit_files = 0;
int args(Yaz_Proxy *proxy, int argc, char **argv)
char *prog = argv[0];
int ret;
- while ((ret = options("o:a:t:v:c:u:i:m:l:T:p:U:n:X",
- argv, argc, &arg)) != -2)
+ while ((ret = options("o:a:Dt:v:c:u:i:m:l:T:p:n:VX",
+ argv, argc, &arg)) != -2)
{
- int err;
+ int err;
switch (ret)
{
case 0:
if (addr)
- {
- usage(prog);
- return 1;
- }
- addr = arg;
+ {
+ usage(prog);
+ return 1;
+ }
+ addr = arg;
break;
- case 'c':
- err = proxy->set_config(arg);
- if (err == -2)
- {
- fprintf(stderr, "Config file support not enabled (proxy not compiled with libxml2 support)\n");
- exit(1);
- }
- else if (err == -1)
- {
- fprintf(stderr, "Bad or missing file %s\n", arg);
- exit(1);
- }
- break;
- case 'a':
- proxy->set_APDU_log(arg);
- break;
- case 't':
- proxy->set_default_target(arg);
- break;
- case 'U':
- proxy->set_proxy_authentication(arg);
+ case 'a':
+ proxy->set_APDU_log(arg);
+ break;
+ case 'c':
+ err = proxy->set_config(arg);
+ if (err == -2)
+ {
+ fprintf(stderr, "Config file support not enabled (not using libxslt & libxml2)\n");
+ exit(1);
+ }
+ else if (err == -1)
+ {
+ fprintf(stderr, "Bad or missing file %s\n", arg);
+ exit(1);
+ }
+ break;
+ case 'D':
+ daemon_flags |= YAZ_DAEMON_FORK;
break;
- case 'o':
- proxy->option("optimize", arg);
- break;
- case 'v':
- yaz_log_init_level (yaz_log_mask_str(arg));
- break;
- case 'l':
- yaz_log_init_file (arg);
- log_file = xstrdup(arg);
- break;
- case 'm':
- proxy->set_max_clients(atoi(arg));
- break;
case 'i':
- proxy->set_client_idletime(atoi(arg));
- break;
+ proxy->set_client_idletime(atoi(arg));
+ break;
+ case 'l':
+ yaz_log_init_file (arg);
+ log_file = xstrdup(arg);
+ break;
+ case 'm':
+ proxy->set_max_clients(atoi(arg));
+ break;
+ case 'n':
+ no_limit_files = atoi(arg);
+ break;
+ case 'o':
+ proxy->option("optimize", arg);
+ break;
+ case 'p':
+ if (!pid_fname)
+ pid_fname = xstrdup(arg);
+ break;
+ case 't':
+ proxy->set_default_target(arg);
+ break;
case 'T':
- proxy->set_target_idletime(atoi(arg));
- break;
- case 'n':
- no_limit_files = atoi(arg);
- break;
- case 'X':
- debug = 1;
- break;
- case 'p':
- if (!pid_fname)
- pid_fname = xstrdup(arg);
- break;
- case 'u':
- if (!uid)
- uid = xstrdup(arg);
- break;
+ proxy->set_target_idletime(atoi(arg));
+ break;
+ case 'u':
+ if (!uid)
+ uid = xstrdup(arg);
+ break;
+ case 'v':
+ yaz_log_init_level (yaz_log_mask_str(arg));
+ break;
+ case 'V':
+ puts(
+#ifdef VERSION
+ VERSION
+#else
+ "unknown"
+#endif
+ );
+ exit(0);
+ case 'X':
+ proxy->set_debug_mode(1);
+ daemon_flags = YAZ_DAEMON_DEBUG;
+ break;
default:
- usage(prog);
- return 1;
+ usage(prog);
+ return 1;
}
}
if (addr)
{
- if (proxy->server(addr))
- {
- yaz_log(LOG_FATAL|LOG_ERRNO, "listen %s", addr);
- exit(1);
- }
+ if (proxy->server(addr))
+ {
+ yaz_log(YLOG_FATAL|YLOG_ERRNO, "listen %s", addr);
+ exit(1);
+ }
}
else
{
- usage(prog);
- return 1;
+ usage(prog);
+ return 1;
}
return 0;
}
signal(SIGHUP, sighup_handler);
#endif
if (static_yaz_proxy)
- static_yaz_proxy->reconfig();
+ static_yaz_proxy->reconfig();
}
-#if HAVE_XSLT
+#if YAZ_HAVE_XSLT
static void proxy_xml_error_handler(void *ctx, const char *fmt, ...)
{
char buf[1024];
#else
vsnprintf(buf, sizeof(buf), fmt, ap);
#endif
- yaz_log(LOG_WARN, "%s: %s", (char*) ctx, buf);
+ yaz_log(YLOG_WARN, "%s: %s", (char*) ctx, buf);
va_end (ap);
}
#endif
-static void child_run(Yaz_SocketManager *m, int run)
+static void child_run(void *data)
{
+ SocketManager *m = (SocketManager *) data;
#ifdef WIN32
#else
signal(SIGHUP, sighup_handler);
#endif
-#if HAVE_XSLT
+#if YAZ_HAVE_XSLT
xmlSetGenericErrorFunc((void *) "XML", proxy_xml_error_handler);
xsltSetGenericErrorFunc((void *) "XSLT", proxy_xml_error_handler);
#endif
+
+#if YAZ_HAVE_EXSLT
+ exsltRegisterAll();
+#endif
#ifdef WIN32
#else
- yaz_log(LOG_LOG, "0 proxy run=%d pid=%ld", run, (long) getpid());
+ yaz_log(YLOG_LOG, "0 proxy pid=%ld", (long) getpid());
#endif
if (no_limit_files)
{
#if HAVE_SETRLIMIT
- struct rlimit limit_data;
- limit_data.rlim_cur = no_limit_files;
- limit_data.rlim_max = no_limit_files;
-
- yaz_log(LOG_LOG, "0 setrlimit NOFILE cur=%d max=%d",
- limit_data.rlim_cur, limit_data.rlim_max);
- if (setrlimit(RLIMIT_NOFILE, &limit_data))
- yaz_log(LOG_ERRNO|LOG_WARN, "setrlimit");
+ struct rlimit limit_data;
+ limit_data.rlim_cur = no_limit_files;
+ limit_data.rlim_max = no_limit_files;
+
+ yaz_log(YLOG_LOG, "0 setrlimit NOFILE cur=%ld max=%ld",
+ (long) limit_data.rlim_cur, (long) limit_data.rlim_max);
+ if (setrlimit(RLIMIT_NOFILE, &limit_data))
+ yaz_log(YLOG_ERRNO|YLOG_WARN, "setrlimit");
#else
- yaz_log(LOG_WARN, "setrlimit unavablable. Option -n ignored");
+ yaz_log(YLOG_WARN, "setrlimit unavablable. Option -n ignored");
#endif
}
-#ifdef WIN32
-#else
- if (pid_fname)
- {
- FILE *f = fopen(pid_fname, "w");
- if (!f)
- {
- yaz_log(LOG_ERRNO|LOG_FATAL, "Couldn't create %s", pid_fname);
- exit(0);
- }
- fprintf(f, "%ld", (long) getpid());
- fclose(f);
- xfree(pid_fname);
- }
- if (uid)
- {
- struct passwd *pw;
-
- if (!(pw = getpwnam(uid)))
- {
- yaz_log(LOG_FATAL, "%s: Unknown user", uid);
- exit(3);
- }
- if (log_file)
- {
- chown(log_file, pw->pw_uid, pw->pw_gid);
- xfree(log_file);
- }
- if (setuid(pw->pw_uid) < 0)
- {
- yaz_log(LOG_FATAL|LOG_ERRNO, "setuid");
- exit(4);
- }
- xfree(uid);
- }
-#endif
#if HAVE_GETRLIMIT
struct rlimit limit_data;
getrlimit(RLIMIT_NOFILE, &limit_data);
- yaz_log(LOG_LOG, "0 getrlimit NOFILE cur=%d max=%d",
- limit_data.rlim_cur, limit_data.rlim_max);
+ yaz_log(YLOG_LOG, "0 getrlimit NOFILE cur=%ld max=%ld",
+ (long) limit_data.rlim_cur, (long) limit_data.rlim_max);
#endif
-
+
while (m->processEvent() > 0)
- ;
+ ;
exit (0);
}
int main(int argc, char **argv)
{
-#if HAVE_XSLT
+#if YAZ_HAVE_XSLT
xmlInitMemory();
-
+
LIBXML_TEST_VERSION
#endif
- int cont = 1;
- int run = 1;
- Yaz_SocketManager mySocketManager;
- Yaz_Proxy proxy(new Yaz_PDU_Assoc(&mySocketManager));
+ SocketManager mySocketManager;
+ Yaz_Proxy proxy(new PDU_Assoc(&mySocketManager), &mySocketManager);
static_yaz_proxy = &proxy;
args(&proxy, argc, argv);
-#ifdef WIN32
- child_run(&mySocketManager, run);
-#else
- if (debug)
- {
- child_run(&mySocketManager, run);
- exit(0);
- }
- while (cont)
- {
- pid_t p = fork();
- if (p == (pid_t) -1)
- {
- yaz_log(LOG_FATAL|LOG_ERRNO, "fork");
- exit(1);
- }
- else if (p == 0)
- {
- child_run(&mySocketManager, run);
- }
- pid_t p1;
- int status;
- p1 = wait(&status);
-
- yaz_log_reopen();
-
- if (p1 != p)
- {
- yaz_log(LOG_FATAL, "p1=%d != p=%d", p1, p);
- exit(1);
- }
- if (WIFSIGNALED(status))
- {
- switch(WTERMSIG(status)) {
- case SIGILL:
- yaz_log(LOG_WARN, "Received SIGILL from child %ld", (long) p);
- cont = 1;
- break;
- case SIGABRT:
- yaz_log(LOG_WARN, "Received SIGABRT from child %ld", (long) p);
- cont = 1;
- break ;
- case SIGSEGV:
- yaz_log(LOG_WARN, "Received SIGSEGV from child %ld", (long) p);
- cont = 1;
- break;
- case SIGBUS:
- yaz_log(LOG_WARN, "Received SIGBUS from child %ld", (long) p);
- cont = 1;
- break;
- case SIGTERM:
- yaz_log(LOG_LOG, "Received SIGTERM from child %ld",
- (long) p);
- cont = 0;
- break;
- default:
- yaz_log(LOG_WARN, "Received SIG %d from child %ld",
- WTERMSIG(status), (long) p);
- cont = 0;
- }
- }
- else if (status == 0)
- cont = 0;
- else
- {
- yaz_log(LOG_LOG, "Exit %d from child %ld", status, (long) p);
- cont = 1;
- }
- if (cont)
- sleep(1 + run/5);
- run++;
- }
-#endif
+ yaz_daemon("yazproxy", daemon_flags,
+ child_run, &mySocketManager, pid_fname, uid);
exit (0);
return 0;
}
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * c-file-style: "Stroustrup"
+ * indent-tabs-mode: nil
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */
+