Home | History | Annotate | Line # | Download | only in rpc
svc_run.c revision 1.23
      1 /*	$NetBSD: svc_run.c,v 1.23 2015/11/06 19:34:13 christos Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2010, Oracle America, Inc.
      5  *
      6  * Redistribution and use in source and binary forms, with or without
      7  * modification, are permitted provided that the following conditions are
      8  * met:
      9  *
     10  *     * Redistributions of source code must retain the above copyright
     11  *       notice, this list of conditions and the following disclaimer.
     12  *     * Redistributions in binary form must reproduce the above
     13  *       copyright notice, this list of conditions and the following
     14  *       disclaimer in the documentation and/or other materials
     15  *       provided with the distribution.
     16  *     * Neither the name of the "Oracle America, Inc." nor the names of its
     17  *       contributors may be used to endorse or promote products derived
     18  *       from this software without specific prior written permission.
     19  *
     20  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     21  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     22  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
     23  *   FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
     24  *   COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
     25  *   INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     26  *   DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
     27  *   GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     28  *   INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     29  *   WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
     30  *   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     32  */
     33 
     34 #include <sys/cdefs.h>
     35 #if defined(LIBC_SCCS) && !defined(lint)
     36 #if 0
     37 static char *sccsid = "@(#)svc_run.c 1.1 87/10/13 Copyr 1984 Sun Micro";
     38 static char *sccsid = "@(#)svc_run.c	2.1 88/07/29 4.0 RPCSRC";
     39 #else
     40 __RCSID("$NetBSD: svc_run.c,v 1.23 2015/11/06 19:34:13 christos Exp $");
     41 #endif
     42 #endif
     43 
     44 /*
     45  * This is the rpc server side idle loop
     46  * Wait for input, call server program.
     47  */
     48 #include "namespace.h"
     49 #include "reentrant.h"
     50 #include <err.h>
     51 #include <errno.h>
     52 #include <stdio.h>
     53 #include <stdlib.h>
     54 #include <string.h>
     55 #include <unistd.h>
     56 
     57 #include <rpc/rpc.h>
     58 
     59 #include "svc_fdset.h"
     60 #include "rpc_internal.h"
     61 
     62 #ifdef __weak_alias
     63 __weak_alias(svc_run,_svc_run)
     64 __weak_alias(svc_exit,_svc_exit)
     65 #endif
     66 
     67 void
     68 svc_run(void)
     69 {
     70 	fd_set *readfds, *cleanfds;
     71 	struct timeval timeout;
     72 	int maxfd, fdsize;
     73 #ifndef RUMP_RPC
     74 	int probs = 0;
     75 #endif
     76 #ifdef _REENTRANT
     77 	extern rwlock_t svc_fd_lock;
     78 #endif
     79 
     80 	readfds = NULL;
     81 	cleanfds = NULL;
     82 	fdsize = 0;
     83 	timeout.tv_sec = 30;
     84 	timeout.tv_usec = 0;
     85 
     86 	for (;;) {
     87 		rwlock_rdlock(&svc_fd_lock);
     88 		if (fdsize != svc_fdset_getsize(0)) {
     89 			fdsize = svc_fdset_getsize(0);
     90 			free(readfds);
     91 			readfds = svc_fdset_copy(svc_fdset_get());
     92 			free(cleanfds);
     93 			cleanfds = svc_fdset_copy(svc_fdset_get());
     94 		}
     95 		maxfd = *svc_fdset_getmax();
     96 		rwlock_unlock(&svc_fd_lock);
     97 		switch (select(maxfd + 1, readfds, NULL, NULL, &timeout)) {
     98 		case -1:
     99 #ifndef RUMP_RPC
    100 			if ((errno == EINTR || errno == EBADF) && probs < 100) {
    101 				probs++;
    102 				continue;
    103 			}
    104 #endif
    105 			if (errno == EINTR) {
    106 				continue;
    107 			}
    108 			warn("%s: select failed", __func__);
    109 			goto out;
    110 		case 0:
    111 			__svc_clean_idle(cleanfds, 30, FALSE);
    112 			continue;
    113 		default:
    114 			svc_getreqset2(readfds, fdsize);
    115 #ifndef RUMP_RPC
    116 			probs = 0;
    117 #endif
    118 		}
    119 	}
    120 out:
    121 	free(readfds);
    122 	free(cleanfds);
    123 }
    124 
    125 /*
    126  *      This function causes svc_run() to exit by telling it that it has no
    127  *      more work to do.
    128  */
    129 void
    130 svc_exit(void)
    131 {
    132 #ifdef _REENTRANT
    133 	extern rwlock_t svc_fd_lock;
    134 #endif
    135 
    136 	rwlock_wrlock(&svc_fd_lock);
    137 	svc_fdset_zero();
    138 	rwlock_unlock(&svc_fd_lock);
    139 }
    140