X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;ds=sidebyside;f=src%2Fyaz-proxy-main.cpp;h=9daed93cd25c9dff9ce023af4a709aa6323bdc63;hb=cf75ac8890f84648d5960d4b07fea7aab9a6eff1;hp=904f9155be34fbb611eaed2f11b8f08ab4aa29bb;hpb=5c45ffbb2b430a6f41277c303a5e9b94242dab96;p=yazpp-moved-to-github.git

diff --git a/src/yaz-proxy-main.cpp b/src/yaz-proxy-main.cpp
index 904f915..9daed93 100644
--- a/src/yaz-proxy-main.cpp
+++ b/src/yaz-proxy-main.cpp
@@ -1,41 +1,121 @@
 /*
- * Copyright (c) 1998-1999, Index Data.
+ * Copyright (c) 1998-2003, Index Data.
  * See the file LICENSE for details.
- * Sebastian Hammer, Adam Dickmeiss
  * 
- * $Log: yaz-proxy-main.cpp,v $
- * Revision 1.5  1999-04-21 12:09:01  adam
- * Many improvements. Modified to proxy server to work with "sessions"
- * based on cookies.
- *
- * Revision 1.4  1999/04/09 11:46:57  adam
- * Added object Yaz_Z_Assoc. Much more functional client.
- *
- * Revision 1.3  1999/02/02 14:01:21  adam
- * First WIN32 port of YAZ++.
- *
- * Revision 1.2  1999/01/28 13:08:45  adam
- * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
- * yaz-socket-manager.cc.
- *
- * Revision 1.1.1.1  1999/01/28 09:41:07  adam
- * First implementation of YAZ++.
- *
+ * $Id: yaz-proxy-main.cpp,v 1.20 2003-10-09 12:11:10 adam Exp $
  */
 
-#include <log.h>
+#include <signal.h>
+#include <yaz/log.h>
+#include <yaz/options.h>
 
-#include <yaz-socket-manager.h>
-#include <yaz-pdu-assoc.h>
-#include <yaz-proxy.h>
+#include <yaz++/socket-manager.h>
+#include <yaz++/pdu-assoc.h>
+#include <yaz++/proxy.h>
+
+void usage(char *prog)
+{
+    fprintf (stderr, "%s: [-c config] [-a log] [-m num] [-v level] [-t target] [-i sec] "
+             "[-u auth] [-o optlevel] @:port\n", prog);
+    exit (1);
+}
+
+int args(Yaz_Proxy *proxy, int argc, char **argv)
+{
+    char *addr = 0;
+    char *arg;
+    char *prog = argv[0];
+    int ret;
+
+    while ((ret = options("o:a:t:v:c:u:i:m:l:T:", argv, argc, &arg)) != -2)
+    {
+	int err;
+        switch (ret)
+        {
+        case 0:
+            if (addr)
+	    {
+		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);
+            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);
+	    break;
+	case 'm':
+	    proxy->set_max_clients(atoi(arg));
+	    break;
+        case 'i':
+	    proxy->set_client_idletime(atoi(arg));
+	    break;
+        case 'T':
+	    proxy->set_target_idletime(atoi(arg));
+	    break;
+        default:
+	    usage(prog);
+	    return 1;
+        }
+    }
+    if (addr)
+    {
+	proxy->server(addr);
+    }
+    else
+    {
+	usage(prog);
+	return 1;
+    }
+    return 0;
+}
+
+static Yaz_Proxy *static_yaz_proxy = 0;
+static void sighup_handler(int num)
+{
+    if (static_yaz_proxy)
+	static_yaz_proxy->reconfig();
+}
 
 int main(int argc, char **argv)
 {
     Yaz_SocketManager mySocketManager;
-    Yaz_Proxy proxy(new Yaz_PDU_Assoc(&mySocketManager, 0));
+    Yaz_Proxy proxy(new Yaz_PDU_Assoc(&mySocketManager));
+
+    static_yaz_proxy = &proxy;
+
+    signal(SIGHUP, sighup_handler);
 
-    proxy.server(argc < 2 ? "@:9000" : argv[1]);
+    args(&proxy, argc, argv);
     while (mySocketManager.processEvent() > 0)
 	;
+    exit (0);
     return 0;
 }