/*
- * Copyright (c) 1998-2003, Index Data.
+ * Copyright (c) 1998-2004, Index Data.
* See the file LICENSE for details.
*
- * $Id: yaz-proxy-main.cpp,v 1.26 2003-10-24 12:19:23 adam Exp $
+ * $Id: yaz-proxy-main.cpp,v 1.35 2004-03-17 10:49:58 adam Exp $
*/
#include <signal.h>
#include <unistd.h>
#include <pwd.h>
#include <sys/types.h>
+#include <stdarg.h>
+
+#if HAVE_GETRLIMIT
+#include <sys/time.h>
+#include <sys/resource.h>
+#endif
#include <yaz/log.h>
#include <yaz/options.h>
static char *pid_fname = 0;
static char *uid = 0;
static char *log_file = 0;
+static int debug = 0;
+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:", argv, argc, &arg)) != -2)
+ while ((ret = options("o:a:t:v:c:u:i:m:l:T:p:U:n:X",
+ argv, argc, &arg)) != -2)
{
int err;
switch (ret)
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);
}
if (addr)
{
- yaz_log(LOG_LOG, "Starting proxy pid=%ld", (long) getpid());
if (proxy->server(addr))
{
yaz_log(LOG_FATAL|LOG_ERRNO, "listen %s", addr);
static_yaz_proxy->reconfig();
}
-int main(int argc, char **argv)
+#if HAVE_XSLT
+static void proxy_xml_error_handler(void *ctx, const char *fmt, ...)
{
- static int mk_pid = 0;
- Yaz_SocketManager mySocketManager;
- Yaz_Proxy proxy(new Yaz_PDU_Assoc(&mySocketManager));
+ char buf[1024];
- static_yaz_proxy = &proxy;
+ va_list ap;
+ va_start(ap, fmt);
+
+ vsnprintf(buf, sizeof(buf), fmt, ap);
+
+ yaz_log(LOG_WARN, "%s: %s", (char*) ctx, buf);
+ va_end (ap);
+}
+#endif
+
+static void child_run(Yaz_SocketManager *m, int run)
+{
signal(SIGHUP, sighup_handler);
- args(&proxy, argc, argv);
+#if HAVE_XSLT
+ xmlSetGenericErrorFunc((void *) "XML", proxy_xml_error_handler);
+ xsltSetGenericErrorFunc((void *) "XSLT", proxy_xml_error_handler);
+#endif
+ yaz_log(LOG_LOG, "0 proxy run=%d pid=%ld", run, (long) getpid());
+ 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");
+#else
+ yaz_log(LOG_WARN, "setrlimit unavablable. Option -n ignored");
+#endif
+ }
if (pid_fname)
{
FILE *f = fopen(pid_fname, "w");
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");
}
xfree(uid);
}
-
- while (mySocketManager.processEvent() > 0)
+#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);
+#endif
+
+ while (m->processEvent() > 0)
;
exit (0);
+}
+
+int main(int argc, char **argv)
+{
+#if HAVE_XSLT
+ xmlInitMemory();
+
+ LIBXML_TEST_VERSION
+#endif
+ int cont = 1;
+ int run = 1;
+ Yaz_SocketManager mySocketManager;
+ Yaz_Proxy proxy(new Yaz_PDU_Assoc(&mySocketManager));
+
+ static_yaz_proxy = &proxy;
+
+ args(&proxy, argc, argv);
+
+ 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++;
+ }
+ exit (0);
return 0;
}