X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=server%2Fstatserv.c;h=6bfee9311d8e8f0de1b2f63d781a7e79fff013ec;hp=e77787965ceba125e0b3ac079a6b8bdaaf1bc40c;hb=bf7ee634e855d15e955d06e7b225d2045abd518a;hpb=c4159a200151b1704984ec01f48356fbc757372d diff --git a/server/statserv.c b/server/statserv.c index e777879..6bfee93 100644 --- a/server/statserv.c +++ b/server/statserv.c @@ -1,13 +1,59 @@ /* - * Copyright (c) 1995-1997, Index Data + * Copyright (c) 1995-1999, Index Data * See the file LICENSE for details. * Sebastian Hammer, Adam Dickmeiss * * NT server based on threads by - * Chas Woodfield, Fretwell Downing Datasystem. + * Chas Woodfield, Fretwell Downing Datasystems. * * $Log: statserv.c,v $ - * Revision 1.42 1997-10-27 14:03:02 adam + * Revision 1.54 1999-04-16 14:45:55 adam + * Added interface for tcpd wrapper for access control. + * + * Revision 1.53 1999/02/02 13:57:39 adam + * Uses preprocessor define WIN32 instead of WINDOWS to build code + * for Microsoft WIN32. + * + * Revision 1.52 1998/08/21 14:13:34 adam + * Added GNU Configure script to build Makefiles. + * + * Revision 1.51 1998/07/07 15:51:03 adam + * Changed server so that it stops if bind fails - "address already in + * use" typically causes this. + * + * Revision 1.50 1998/06/22 11:32:39 adam + * Added 'conditional cs_listen' feature. + * + * Revision 1.49 1998/02/27 14:04:55 adam + * Fixed bug in statserv_remove. + * + * Revision 1.48 1998/02/11 11:53:36 adam + * Changed code so that it compiles as C++. + * + * Revision 1.47 1998/02/10 10:28:57 adam + * Added app_name, service_dependencies, service_display_name and + * options_func. options_func allows us to specify a different function + * to interogate the command line arguments. The other members allow us + * to pass the full service details accross to the service manager (CW). + * + * + * Revision 1.46 1998/01/30 15:24:57 adam + * Fixed bug in inetd code. The server listened on tcp:@:9999 even + * though it was started in inetd mode. + * + * Revision 1.45 1998/01/29 13:30:23 adam + * Better event handle system for NT/Unix. + * + * Revision 1.44 1997/11/07 13:31:52 adam + * Added NT Service name part of statserv_options_block. Moved NT + * service utility to server library. + * + * Revision 1.43 1997/10/31 12:20:09 adam + * Improved memory debugging for xmalloc/nmem.c. References to NMEM + * instead of ODR in n ESPEC-1 handling in source d1_espec.c. + * Bug fix: missing fclose in data1_read_espec1. + * + * Revision 1.42 1997/10/27 14:03:02 adam * Added new member to statserver_options_block, pre_init, which * specifies a callback to be invoked after command line parsing and * before the server listens for the first time. @@ -144,13 +190,13 @@ * */ -#include #include #include -#ifdef WINDOWS +#ifdef WIN32 #include #include #include +#include "service.h" #else #include #include @@ -159,24 +205,25 @@ #include #include -#include -#include "eventl.h" -#include "session.h" #include #include +#include #ifdef USE_XTIMOSI #include #endif #include +#include "eventl.h" +#include "session.h" #include -static IOCHAN pListener; +static IOCHAN pListener = NULL; static char *me = "statserver"; /* * default behavior. */ -static statserv_options_block control_block = { +int check_options(int argc, char **argv); +statserv_options_block control_block = { 1, /* dynamic mode */ LOG_DEFAULT_LEVEL, /* log level */ "", /* no PDUs */ @@ -187,7 +234,18 @@ static statserv_options_block control_block = { 1024*1024, /* maximum PDU size (approx.) to allow */ "default-config", /* configuration name to pass to backend */ "", /* set user id */ - NULL + NULL, /* pre init handler */ + check_options, /* Default routine, for checking the run-time arguments */ + check_ip_tcpd, + "", + 0 /* default value for inet deamon */ + +#ifdef WIN32 + ,"Z39.50 Server", /* NT Service Name */ + "Server", /* NT application Name */ + "", /* NT Service Dependencies */ + "Z39.50 Server" /* NT Service Display Name */ +#endif /* WIN32 */ }; /* @@ -196,16 +254,16 @@ static statserv_options_block control_block = { * doing all of the listening and accepting in the parent - it's * safer that way. */ -#ifdef WINDOWS +#ifdef WIN32 typedef struct _ThreadList ThreadList; -typedef struct _ThreadList +struct _ThreadList { HANDLE hThread; IOCHAN pIOChannel; ThreadList *pNext; -} ThreadList; +}; static ThreadList *pFirstThread; static CRITICAL_SECTION Thread_CritSect; @@ -214,13 +272,13 @@ static BOOL bInitialized = FALSE; static void ThreadList_Initialize() { /* Initialize the critical Sections */ - InitializeCriticalSection(&Thread_CritSect); + InitializeCriticalSection(&Thread_CritSect); /* Set the first thraed */ - pFirstThread = NULL; + pFirstThread = NULL; - /* we have been initialized */ - bInitialized = TRUE; + /* we have been initialized */ + bInitialized = TRUE; } static void statserv_add(HANDLE hThread, IOCHAN pIOChannel) @@ -274,6 +332,7 @@ void statserv_remove(IOCHAN pIOChannel) { /* We need to look at another thread */ pNextThread = pCurrentThread->pNext; + pPrevThread = pCurrentThread; } } @@ -348,6 +407,11 @@ void statserv_closedown() } } +int __stdcall event_loop_thread (IOCHAN iochan) +{ + return event_loop (&iochan); +} + static void listener(IOCHAN h, int event) { COMSTACK line = (COMSTACK) iochan_getdata(h); @@ -359,7 +423,7 @@ static void listener(IOCHAN h, int event) { if ((res = cs_listen(line, 0, 0)) < 0) { - logf(LOG_FATAL, "cs_listen failed."); + logf(LOG_FATAL, "cs_listen failed"); return; } else if (res == 1) @@ -383,7 +447,8 @@ static void listener(IOCHAN h, int event) } logf(LOG_DEBUG, "Accept ok"); - if (!(new_chan = iochan_create(cs_fileno(new_line), ir_session, EVENT_INPUT))) + if (!(new_chan = iochan_create(cs_fileno(new_line), ir_session, + EVENT_INPUT))) { logf(LOG_FATAL, "Failed to create iochan"); iochan_destroy(h); @@ -400,19 +465,22 @@ static void listener(IOCHAN h, int event) logf(LOG_DEBUG, "Setting timeout %d", control_block.idle_timeout); iochan_setdata(new_chan, newas); iochan_settimeout(new_chan, control_block.idle_timeout * 60); +#ifndef WIN32 logf(LOG_DEBUG, "Determining client address"); a = cs_addrstr(new_line); logf(LOG_LOG, "Accepted connection from %s", a ? a : "[Unknown]"); - /* Now what we need todo is create a new thread with this iochan as - the parameter */ - /* if (CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)event_loop, new_chan, - 0, &ThreadId) == NULL) */ - /* Somehow, somewhere we need to store this thread id, otherwise we won't be - able to close cleanly */ - NewHandle = (HANDLE)_beginthreadex(NULL, 0, event_loop, new_chan, 0, &ThreadId); +#endif + /* Now what we need todo is create a new thread with this iochan as + the parameter */ + /* if (CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)event_loop_thread, + new_chan, 0, &ThreadId) == NULL) */ + /* Somehow, somewhere we need to store this thread id, otherwise we + won't be able to close cleanly */ + NewHandle = (HANDLE)_beginthreadex(NULL, 0, event_loop_thread, + new_chan, 0, &ThreadId); if (NewHandle == (HANDLE)-1) { - + logf(LOG_FATAL|LOG_ERRNO, "Failed to create new thread."); iochan_destroy(h); return; @@ -431,7 +499,7 @@ static void listener(IOCHAN h, int event) } } -#else /* WINDOWS */ +#else /* WIN32 */ /* To save having an #ifdef in event_loop we need to define this empty function */ void statserv_remove(IOCHAN pIOChannel) @@ -440,9 +508,9 @@ void statserv_remove(IOCHAN pIOChannel) void statserv_closedown() { - /* We don't need todoanything here - or do we */ - if (pListener != NULL) - iochan_destroy(pListener); + IOCHAN p; + for (p = pListener; p; p = p->next) + iochan_destroy(p); } static void listener(IOCHAN h, int event) @@ -478,11 +546,11 @@ static void listener(IOCHAN h, int event) close(hand[0]); child = 1; - for (pp = iochan_getchan(); pp; pp = iochan_getnext(pp)) + for (pp = pListener; pp; pp = iochan_getnext(pp)) { if (pp != h) { - COMSTACK l = iochan_getdata(pp); + COMSTACK l = (COMSTACK)iochan_getdata(pp); cs_close(l); iochan_destroy(pp); } @@ -512,9 +580,10 @@ static void listener(IOCHAN h, int event) return; } } - if ((res = cs_listen(line, 0, 0)) < 0) + if ((res = cs_listen_check(line, 0, 0, control_block.check_ip, + control_block.daemon_name)) < 0) { - logf(LOG_FATAL, "cs_listen failed."); + logf(LOG_WARN, "cs_listen failed"); return; } else if (res == 1) @@ -541,9 +610,9 @@ static void listener(IOCHAN h, int event) { IOCHAN pp; /* close our half of the listener socket */ - for (pp = iochan_getchan(); pp; pp = iochan_getnext(pp)) + for (pp = pListener; pp; pp = iochan_getnext(pp)) { - COMSTACK l = iochan_getdata(pp); + COMSTACK l = (COMSTACK)iochan_getdata(pp); cs_close(l); iochan_destroy(pp); } @@ -553,7 +622,7 @@ static void listener(IOCHAN h, int event) } else iochan_setflags(h, EVENT_INPUT | EVENT_EXCEPT); /* reset listener */ - + if (!(new_chan = iochan_create(cs_fileno(new_line), ir_session, EVENT_INPUT))) { @@ -561,6 +630,8 @@ static void listener(IOCHAN h, int event) iochan_destroy(h); return; } + new_chan->next = pListener; + pListener = new_chan; if (!(newas = create_association(new_chan, new_line))) { logf(LOG_FATAL, "Failed to create new assoc."); @@ -580,7 +651,7 @@ static void listener(IOCHAN h, int event) } } -#endif /* WINDOWS */ +#endif /* WIN32 */ static void inetd_connection(int what) { @@ -604,6 +675,8 @@ static void inetd_connection(int what) { logf(LOG_FATAL, "Failed to create association structure"); } + chan->next = pListener; + pListener = chan; } else { @@ -654,6 +727,7 @@ static void add_listener(char *where, int what) if (!(l = cs_create(type, 0, what))) { logf(LOG_FATAL|LOG_ERRNO, "Failed to create listener"); + return; } ap = cs_straddr (l, addr); if (!ap) @@ -665,11 +739,15 @@ static void add_listener(char *where, int what) if (cs_bind(l, ap, CS_SERVER) < 0) { logf(LOG_FATAL|LOG_ERRNO, "Failed to bind to %s", where); + cs_close (l); + return; } if (!(lst = iochan_create(cs_fileno(l), listener, EVENT_INPUT | EVENT_EXCEPT))) { logf(LOG_FATAL|LOG_ERRNO, "Failed to create IOCHAN-type"); + cs_close (l); + return; } iochan_setdata(lst, l); @@ -678,7 +756,7 @@ static void add_listener(char *where, int what) pListener = lst; } -#ifndef WINDOWS +#ifndef WIN32 /* For windows we don't need to catch the signals */ static void catchchld(int num) { @@ -686,7 +764,7 @@ static void catchchld(int num) ; signal(SIGCHLD, catchchld); } -#endif /* WINDOWS */ +#endif /* WIN32 */ statserv_options_block *statserv_getcontrol(void) { @@ -701,19 +779,17 @@ void statserv_setcontrol(statserv_options_block *block) memcpy(&control_block, block, sizeof(*block)); } -int statserv_main(int argc, char **argv) +int statserv_start(int argc, char **argv) { - int ret, listeners = 0, inetd = 0, r; - char *arg; - int protocol = control_block.default_proto; + int ret; nmem_init (); -#ifdef WINDOWS +#ifdef WIN32 /* We need to initialize the thread list */ ThreadList_Initialize(); -#endif /* WINDOWS */ +#endif /* WIN32 */ -#ifdef WINDOWS +#ifdef WIN32 if ((me = strrchr (argv[0], '\\'))) me++; else @@ -721,74 +797,12 @@ int statserv_main(int argc, char **argv) #else me = argv[0]; #endif - while ((ret = options("a:iszSl:v:u:c:w:t:k:", argv, argc, &arg)) != -2) - { - switch (ret) - { - case 0: - add_listener(arg, protocol); - listeners++; - break; - case 'z': protocol = PROTO_Z3950; break; - case 's': protocol = PROTO_SR; break; - case 'S': control_block.dynamic = 0; break; - case 'l': - strcpy(control_block.logfile, arg ? arg : ""); - log_init(control_block.loglevel, me, control_block.logfile); - break; - case 'v': - control_block.loglevel = log_mask_str(arg); - log_init(control_block.loglevel, me, control_block.logfile); - break; - case 'a': - strcpy(control_block.apdufile, arg ? arg : ""); break; - case 'u': - strcpy(control_block.setuid, arg ? arg : ""); break; - case 'c': - strcpy(control_block.configname, arg ? arg : ""); break; - case 't': - if (!arg || !(r = atoi(arg))) - { - fprintf(stderr, "%s: Specify positive timeout for -t.\n", - me); - return(1); - } - control_block.idle_timeout = r; - break; - case 'k': - if (!arg || !(r = atoi(arg))) - { - fprintf(stderr, "%s: Specify positive timeout for -t.\n", - me); - return(1); - } - control_block.maxrecordsize = r * 1024; - break; - case 'i': - inetd = 1; break; - case 'w': - if (chdir(arg)) - { - perror(arg); - - return(1); - } - break; - default: - fprintf(stderr, "Usage: %s [ -i -a -v " - " -l -u -c -t " - " -k " - " -zsS -w ... ]\n", me); - return(1); - } - } + if (control_block.options_func(argc, argv)) + return(1); - if ((pListener == NULL) && *control_block.default_listen) - add_listener(control_block.default_listen, protocol); - -#ifndef WINDOWS - if (inetd) - inetd_connection(protocol); +#ifndef WIN32 + if (control_block.inetd) + inetd_connection(control_block.default_proto); else { if (control_block.pre_init) @@ -811,14 +825,156 @@ int statserv_main(int argc, char **argv) exit(1); } } -#endif /* WINDOWS */ +#endif /* WIN32 */ - logf(LOG_LOG, "Entering event loop."); + if ((pListener == NULL) && *control_block.default_listen) + add_listener(control_block.default_listen, + control_block.default_proto); if (pListener == NULL) ret = 1; else - ret = event_loop(pListener); + { + logf(LOG_LOG, "Entering event loop."); + ret = event_loop(&pListener); + } nmem_exit (); return ret; } + +int check_options(int argc, char **argv) +{ + int ret = 0, r; + char *arg; + + while ((ret = options("a:iszSl:v:u:c:w:t:k:d:", argv, argc, &arg)) != -2) + { + switch (ret) + { + case 0: + add_listener(arg, control_block.default_proto); + break; + case 'z': + control_block.default_proto = PROTO_Z3950; + break; + case 's': + control_block.default_proto = PROTO_SR; + break; + case 'S': + control_block.dynamic = 0; + break; + case 'l': + strcpy(control_block.logfile, arg ? arg : ""); + log_init(control_block.loglevel, me, control_block.logfile); + break; + case 'v': + control_block.loglevel = log_mask_str(arg); + log_init(control_block.loglevel, me, control_block.logfile); + break; + case 'a': + strcpy(control_block.apdufile, arg ? arg : ""); + break; + case 'u': + strcpy(control_block.setuid, arg ? arg : ""); + break; + case 'c': + strcpy(control_block.configname, arg ? arg : ""); + break; + case 'd': + strcpy(control_block.daemon_name, arg ? arg : ""); + break; + case 't': + if (!arg || !(r = atoi(arg))) + { + fprintf(stderr, "%s: Specify positive timeout for -t.\n", me); + return(1); + } + control_block.idle_timeout = r; + break; + case 'k': + if (!arg || !(r = atoi(arg))) + { + fprintf(stderr, "%s: Specify positive size for -k.\n", me); + return(1); + } + control_block.maxrecordsize = r * 1024; + break; + case 'i': + control_block.inetd = 1; + break; + case 'w': + if (chdir(arg)) + { + perror(arg); + return(1); + } + break; + default: + fprintf(stderr, "Usage: %s [ -i -a -v " + " -l -u -c -t " + " -k -d " + " -zsS -w ... ]\n", me); + return(1); + } + } + return 0; +} + +#ifdef WIN32 +typedef struct _Args +{ + char **argv; + int argc; +} Args; + +static Args ArgDetails; + +/* name of the executable */ +#define SZAPPNAME "server" + +/* list of service dependencies - "dep1\0dep2\0\0" */ +#define SZDEPENDENCIES "" + +int statserv_main(int argc, char **argv) +{ + statserv_options_block *cb = statserv_getcontrol(); + + /* Lets setup the Arg structure */ + ArgDetails.argc = argc; + ArgDetails.argv = argv; + + /* Now setup the service with the service controller */ + SetupService(argc, argv, &ArgDetails, SZAPPNAME, + cb->service_name, /* internal service name */ + cb->service_name, /* displayed name of the service */ + SZDEPENDENCIES); + return 0; +} + +int StartAppService(void *pHandle, int argc, char **argv) +{ + /* Initializes the App */ + return 1; +} + +void RunAppService(void *pHandle) +{ + Args *pArgs = (Args *)pHandle; + + /* Starts the app running */ + statserv_start(pArgs->argc, pArgs->argv); +} + +void StopAppService(void *pHandle) +{ + /* Stops the app */ + statserv_closedown(); +} +#else +int statserv_main(int argc, char **argv) +{ + int ret = statserv_start (argc, argv); + statserv_closedown (); + return ret; +} +#endif