2 * Copyright (c) 1995-1997, Index Data
3 * See the file LICENSE for details.
4 * Sebastian Hammer, Adam Dickmeiss
6 * NT server based on threads by
7 * Chas Woodfield, Fretwell Downing Datasystem.
10 * Revision 1.38 1997-09-04 14:19:14 adam
13 * Revision 1.37 1997/09/01 08:53:01 adam
14 * New windows NT/95 port using MSV5.0. The test server 'ztest' was
15 * moved a separate directory. MSV5.0 project server.dsp created.
16 * As an option, the server can now operate as an NT service.
18 * Revision 1.36 1996/07/06 19:58:36 quinn
19 * System headerfiles gathered in yconfig
21 * Revision 1.35 1996/05/29 10:03:28 quinn
24 * Revision 1.34 1996/02/21 13:12:07 quinn
25 * *** empty log message ***
27 * Revision 1.33 1996/02/10 12:23:49 quinn
28 * Enable inetd operations fro TCP/IP stack
30 * Revision 1.32 1996/01/19 15:41:52 quinn
31 * *** empty log message ***
33 * Revision 1.31 1995/11/17 11:09:39 adam
34 * Added new option '-c' to specify configuration name in control block.
36 * Revision 1.30 1995/11/01 13:54:59 quinn
39 * Revision 1.29 1995/10/30 12:41:29 quinn
40 * Added hostname lookup for server.
42 * Revision 1.28 1995/09/29 17:12:30 quinn
45 * Revision 1.27 1995/09/27 15:03:02 quinn
46 * Modified function heads & prototypes.
48 * Revision 1.26 1995/08/29 14:44:51 quinn
51 * Revision 1.25 1995/08/29 11:18:02 quinn
52 * Added code to receive close
54 * Revision 1.24 1995/06/16 10:31:39 quinn
55 * Added session timeout.
57 * Revision 1.23 1995/06/15 12:30:48 quinn
60 * Revision 1.22 1995/06/15 07:45:17 quinn
63 * Revision 1.21 1995/06/06 08:15:40 quinn
66 * Revision 1.20 1995/05/29 08:12:09 quinn
69 * Revision 1.19 1995/05/16 09:37:27 quinn
72 * Revision 1.18 1995/05/16 08:51:09 quinn
73 * License, documentation, and memory fixes
75 * Revision 1.17 1995/05/15 11:56:42 quinn
76 * Asynchronous facilities. Restructuring of seshigh code.
78 * Revision 1.16 1995/04/10 10:23:40 quinn
79 * Some work to add scan and other things.
81 * Revision 1.15 1995/03/31 10:16:51 quinn
84 * Revision 1.14 1995/03/31 09:18:58 quinn
87 * Revision 1.13 1995/03/30 16:08:39 quinn
90 * Revision 1.12 1995/03/30 13:29:02 quinn
93 * Revision 1.11 1995/03/30 12:18:17 quinn
96 * Revision 1.10 1995/03/29 15:40:16 quinn
97 * Ongoing work. Statserv is now dynamic by default
99 * Revision 1.9 1995/03/27 08:34:30 quinn
100 * Added dynamic server functionality.
101 * Released bindings to session.c (is now redundant)
103 * Revision 1.8 1995/03/20 09:46:26 quinn
106 * Revision 1.7 1995/03/16 13:29:04 quinn
107 * Partitioned server.
109 * Revision 1.6 1995/03/15 15:18:52 quinn
110 * Little changes to better support nonblocking I/O
113 * Revision 1.5 1995/03/15 08:37:45 quinn
114 * Now we're pretty much set for nonblocking I/O.
116 * Revision 1.4 1995/03/14 16:59:48 quinn
119 * Revision 1.3 1995/03/14 11:30:15 quinn
122 * Revision 1.2 1995/03/14 10:28:03 quinn
123 * More work on demo server.
125 * Revision 1.1 1995/03/10 18:22:45 quinn
126 * The rudiments of an asynchronous server.
131 * Simple, static server. I wouldn't advise a static server unless you
132 * really have to, but it's great for debugging memory management. :)
152 #include <comstack.h>
158 #include <statserv.h>
160 static IOCHAN pListener;
162 static char *me = "statserver";
166 static statserv_options_block control_block = {
167 1, /* dynamic mode */
168 LOG_DEFAULT_LEVEL, /* log level */
170 "", /* diagnostic output to stderr */
171 "tcp:@:9999", /* default listener port */
172 PROTO_Z3950, /* default application protocol */
173 60, /* idle timeout (minutes) */
174 1024*1024, /* maximum PDU size (approx.) to allow */
175 "default-config", /* configuration name to pass to backend */
180 * handle incoming connect requests.
181 * The dynamic mode is a bit tricky mostly because we want to avoid
182 * doing all of the listening and accepting in the parent - it's
187 typedef struct _ThreadList ThreadList;
189 typedef struct _ThreadList
196 static ThreadList *pFirstThread;
197 static CRITICAL_SECTION Thread_CritSect;
198 static BOOL bInitialized = FALSE;
200 static void ThreadList_Initialize()
202 /* Initialize the critical Sections */
203 InitializeCriticalSection(&Thread_CritSect);
205 /* Set the first thraed */
208 /* we have been initialized */
212 static void statserv_add(HANDLE hThread, IOCHAN pIOChannel)
214 /* Only one thread can go through this section at a time */
215 EnterCriticalSection(&Thread_CritSect);
218 /* Lets create our new object */
219 ThreadList *pNewThread = (ThreadList *)malloc(sizeof(ThreadList));
220 pNewThread->hThread = hThread;
221 pNewThread->pIOChannel = pIOChannel;
222 pNewThread->pNext = pFirstThread;
223 pFirstThread = pNewThread;
225 /* Lets let somebody else create a new object now */
226 LeaveCriticalSection(&Thread_CritSect);
230 void statserv_remove(IOCHAN pIOChannel)
232 /* Only one thread can go through this section at a time */
233 EnterCriticalSection(&Thread_CritSect);
236 ThreadList *pCurrentThread = pFirstThread;
237 ThreadList *pNextThread;
238 ThreadList *pPrevThread =NULL;
240 /* Step through alll the threads */
241 for (; pCurrentThread != NULL; pCurrentThread = pNextThread)
243 /* We only need to compare on the IO Channel */
244 if (pCurrentThread->pIOChannel == pIOChannel)
246 /* We have found the thread we want to delete */
247 /* First of all reset the next pointers */
248 if (pPrevThread == NULL)
249 pFirstThread = pCurrentThread->pNext;
251 pPrevThread->pNext = pCurrentThread->pNext;
253 /* All we need todo now is delete the memory */
254 free(pCurrentThread);
256 /* No need to look at any more threads */
261 /* We need to look at another thread */
262 pNextThread = pCurrentThread->pNext;
266 /* Lets let somebody else remove an object now */
267 LeaveCriticalSection(&Thread_CritSect);
271 void statserv_closedown()
273 /* Shouldn't do anything if we are not initialized */
277 HANDLE *pThreadHandles = NULL;
279 /* We need to stop threads adding and removing while we start the closedown process */
280 EnterCriticalSection(&Thread_CritSect);
283 /* We have exclusive access to the thread stuff now */
284 /* Y didn't i use a semaphore - Oh well never mind */
285 ThreadList *pCurrentThread = pFirstThread;
287 /* Before we do anything else, we need to shutdown the listener */
288 if (pListener != NULL)
289 iochan_destroy(pListener);
291 for (; pCurrentThread != NULL; pCurrentThread = pCurrentThread->pNext)
293 /* Just destroy the IOCHAN, that should do the trick */
294 iochan_destroy(pCurrentThread->pIOChannel);
296 /* Keep a running count of our handles */
302 HANDLE *pCurrentHandle ;
304 /* Allocate the thread handle array */
305 pThreadHandles = (HANDLE *)malloc(sizeof(HANDLE) * iHandles);
306 pCurrentHandle = pThreadHandles;
308 for (pCurrentThread = pFirstThread;
309 pCurrentThread != NULL;
310 pCurrentThread = pCurrentThread->pNext, pCurrentHandle++)
312 /* Just the handle */
313 *pCurrentHandle = pCurrentThread->hThread;
317 /* We can now leave the critical section */
318 LeaveCriticalSection(&Thread_CritSect);
321 /* Now we can really do something */
324 /* This will now wait, until all the threads close */
325 WaitForMultipleObjects(iHandles, pThreadHandles, TRUE, INFINITE);
327 /* Free the memory we allocated for the handle array */
328 free(pThreadHandles);
331 /* No longer require the critical section, since all threads are dead */
332 DeleteCriticalSection(&Thread_CritSect);
336 static void listener(IOCHAN h, int event)
338 COMSTACK line = (COMSTACK) iochan_getdata(h);
343 if (event == EVENT_INPUT)
345 if ((res = cs_listen(line, 0, 0)) < 0)
347 logf(LOG_FATAL, "cs_listen failed.");
352 logf(LOG_DEBUG, "listen ok");
353 iochan_setevent(h, EVENT_OUTPUT);
354 iochan_setflags(h, EVENT_OUTPUT | EVENT_EXCEPT); /* set up for acpt */
356 else if (event == EVENT_OUTPUT)
363 if (!(new_line = cs_accept(line)))
365 logf(LOG_FATAL, "Accept failed.");
366 iochan_setflags(h, EVENT_INPUT | EVENT_EXCEPT); /* reset listener */
369 logf(LOG_DEBUG, "accept ok");
371 if (!(new_chan = iochan_create(cs_fileno(new_line), ir_session, EVENT_INPUT)))
373 logf(LOG_FATAL, "Failed to create iochan");
377 if (!(newas = create_association(new_chan, new_line)))
379 logf(LOG_FATAL, "Failed to create new assoc.");
383 iochan_setdata(new_chan, newas);
384 iochan_settimeout(new_chan, control_block.idle_timeout * 60);
385 a = cs_addrstr(new_line);
386 logf(LOG_LOG, "Accepted connection from %s", a ? a : "[Unknown]");
388 /* Now what we need todo is create a new thread with this iochan as the parameter */
389 /* if (CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)event_loop, new_chan, 0, &ThreadId) == NULL)
391 /* Somehow, somewhere we need to store this thread id, otherwise we won't be able to close cleanly */
392 NewHandle = (HANDLE)_beginthreadex(NULL, 0, event_loop, new_chan, 0, &ThreadId);
393 if (NewHandle == (HANDLE)-1)
395 logf(LOG_FATAL, "Failed to create new thread.");
400 /* We successfully created the thread, so add it to the list */
401 statserv_add(NewHandle, new_chan);
403 logf(LOG_DEBUG, "Created new thread, iochan %p", new_chan);
404 iochan_setflags(h, EVENT_INPUT | EVENT_EXCEPT); /* reset listener */
408 logf(LOG_FATAL, "Bad event on listener.");
416 /* To save having an #ifdef in event_loop we need to define this empty function */
417 void statserv_remove(IOCHAN pIOChannel)
421 void statserv_closedown()
423 /* We don't need todoanything here - or do we */
424 if (pListener != NULL)
425 iochan_destroy(pListener);
428 static void listener(IOCHAN h, int event)
430 COMSTACK line = (COMSTACK) iochan_getdata(h);
433 static int child = 0;
436 if (event == EVENT_INPUT)
438 if (control_block.dynamic && !child)
444 logf(LOG_FATAL|LOG_ERRNO, "pipe");
448 if ((res = fork()) < 0)
450 logf(LOG_FATAL|LOG_ERRNO, "fork");
454 else if (res == 0) /* child */
461 for (pp = iochan_getchan(); pp; pp = iochan_getnext(pp))
465 COMSTACK l = iochan_getdata(pp);
470 sprintf(nbuf, "%s(%d)", me, getpid());
471 log_init(control_block.loglevel, nbuf, 0);
476 /* wait for child to take the call */
482 if ((res = read(hand[0], dummy, 1)) < 0 && errno != EINTR)
484 logf(LOG_FATAL|LOG_ERRNO, "handshake read");
490 logf(LOG_DEBUG, "P: Child has taken the call");
495 if ((res = cs_listen(line, 0, 0)) < 0)
497 logf(LOG_FATAL, "cs_listen failed.");
502 logf(LOG_DEBUG, "listen ok");
503 iochan_setevent(h, EVENT_OUTPUT);
504 iochan_setflags(h, EVENT_OUTPUT | EVENT_EXCEPT); /* set up for acpt */
506 /* in dynamic mode, only the child ever comes down here */
507 else if (event == EVENT_OUTPUT)
513 if (!(new_line = cs_accept(line)))
515 logf(LOG_FATAL, "Accept failed.");
516 iochan_setflags(h, EVENT_INPUT | EVENT_EXCEPT); /* reset listener */
519 logf(LOG_DEBUG, "accept ok");
520 if (control_block.dynamic)
523 /* close our half of the listener socket */
524 for (pp = iochan_getchan(); pp; pp = iochan_getnext(pp))
526 COMSTACK l = iochan_getdata(pp);
531 logf(LOG_DEBUG, "Releasing parent");
535 iochan_setflags(h, EVENT_INPUT | EVENT_EXCEPT); /* reset listener */
537 if (!(new_chan = iochan_create(cs_fileno(new_line), ir_session,
540 logf(LOG_FATAL, "Failed to create iochan");
544 if (!(newas = create_association(new_chan, new_line)))
546 logf(LOG_FATAL, "Failed to create new assoc.");
550 iochan_setdata(new_chan, newas);
551 iochan_settimeout(new_chan, control_block.idle_timeout * 60);
552 a = cs_addrstr(new_line);
553 logf(LOG_LOG, "Accepted connection from %s", a ? a : "[Unknown]");
557 logf(LOG_FATAL, "Bad event on listener.");
565 static void inetd_connection(int what)
572 if ((line = cs_createbysocket(0, tcpip_type, 0, what)))
574 if ((chan = iochan_create(cs_fileno(line), ir_session, EVENT_INPUT)))
576 if ((assoc = create_association(chan, line)))
578 iochan_setdata(chan, assoc);
579 iochan_settimeout(chan, control_block.idle_timeout * 60);
580 addr = cs_addrstr(line);
581 logf(LOG_LOG, "Inetd association from %s", addr ? addr : "[UNKNOWN]");
585 logf(LOG_FATAL, "Failed to create association structure");
590 logf(LOG_FATAL, "Failed to create iochan");
595 logf(LOG_ERRNO|LOG_FATAL, "Failed to create comstack on socket 0");
600 * Set up a listening endpoint, and give it to the event-handler.
602 static void add_listener(char *where, int what)
606 char mode[100], addr[100];
610 if (!where || sscanf(where, "%[^:]:%s", mode, addr) != 2)
612 fprintf(stderr, "%s: Address format: ('tcp'|'osi')':'<address>.\n",
615 if (!strcmp(mode, "tcp"))
617 if (!(ap = tcpip_strtoaddr(addr)))
619 fprintf(stderr, "Address resolution failed for TCP.\n");
623 else if (!strcmp(mode, "osi"))
626 if (!(ap = mosi_strtoaddr(addr)))
628 fprintf(stderr, "Address resolution failed for TCP.\n");
632 fprintf(stderr, "OSI Transport not allowed by configuration.\n");
637 fprintf(stderr, "You must specify either 'osi:' or 'tcp:'.\n");
639 logf(LOG_LOG, "Adding %s %s listener on %s",
640 control_block.dynamic ? "dynamic" : "static",
641 what == PROTO_SR ? "SR" : "Z3950", where);
642 if (!(l = cs_create(type, 0, what)))
644 logf(LOG_FATAL|LOG_ERRNO, "Failed to create listener");
646 if (cs_bind(l, ap, CS_SERVER) < 0)
648 logf(LOG_FATAL|LOG_ERRNO, "Failed to bind to %s", where);
650 if (!(lst = iochan_create(cs_fileno(l), listener, EVENT_INPUT |
653 logf(LOG_FATAL|LOG_ERRNO, "Failed to create IOCHAN-type");
655 iochan_setdata(lst, l);
657 /* Ensure our listener chain is setup properly */
658 lst->next = pListener;
663 /* For windows we don't need to catch the signals */
664 static void catchchld(int num)
666 while (waitpid(-1, 0, WNOHANG) > 0)
668 signal(SIGCHLD, catchchld);
672 statserv_options_block *statserv_getcontrol(void)
674 static statserv_options_block cb;
676 memcpy(&cb, &control_block, sizeof(cb));
680 void statserv_setcontrol(statserv_options_block *block)
682 memcpy(&control_block, block, sizeof(*block));
685 int statserv_main(int argc, char **argv)
687 int ret, listeners = 0, inetd = 0, r;
689 int protocol = control_block.default_proto;
692 /* We need to initialize the thread list */
693 ThreadList_Initialize();
697 while ((ret = options("a:iszSl:v:u:c:w:t:k:", argv, argc, &arg)) != -2)
702 add_listener(arg, protocol);
705 case 'z': protocol = PROTO_Z3950; break;
706 case 's': protocol = PROTO_SR; break;
707 case 'S': control_block.dynamic = 0; break;
709 strcpy(control_block.logfile, arg ? arg : "");
710 log_init(control_block.loglevel, me, control_block.logfile);
713 control_block.loglevel = log_mask_str(arg);
714 log_init(control_block.loglevel, me, control_block.logfile);
717 strcpy(control_block.apdufile, arg ? arg : ""); break;
719 strcpy(control_block.setuid, arg ? arg : ""); break;
721 strcpy(control_block.configname, arg ? arg : ""); break;
723 if (!arg || !(r = atoi(arg)))
725 fprintf(stderr, "%s: Specify positive timeout for -t.\n",
729 control_block.idle_timeout = r;
732 if (!arg || !(r = atoi(arg)))
734 fprintf(stderr, "%s: Specify positive timeout for -t.\n",
738 control_block.maxrecordsize = r * 1024;
751 fprintf(stderr, "Usage: %s [ -i -a <pdufile> -v <loglevel>"
752 " -l <logfile> -u <user> -c <config> -t <minutes>"
754 " -zsS <listener-addr> -w <directory> ... ]\n", me);
760 log_init(control_block.loglevel, NULL, control_block.logfile);
763 if ((pListener == NULL) && *control_block.default_listen)
764 add_listener(control_block.default_listen, protocol);
768 inetd_connection(protocol);
771 if (control_block.dynamic)
772 signal(SIGCHLD, catchchld);
774 if (*control_block.setuid)
778 if (!(pw = getpwnam(control_block.setuid)))
780 logf(LOG_FATAL, "%s: Unknown user", control_block.setuid);
783 if (setuid(pw->pw_uid) < 0)
785 logf(LOG_FATAL|LOG_ERRNO, "setuid");
791 logf(LOG_LOG, "Entering event loop.");
793 if (pListener == NULL)
796 return event_loop(pListener);