diff options
Diffstat (limited to 'libexec/rpc.sprayd/sprayd.c')
| -rw-r--r-- | libexec/rpc.sprayd/sprayd.c | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/libexec/rpc.sprayd/sprayd.c b/libexec/rpc.sprayd/sprayd.c new file mode 100644 index 00000000000..b5f8e98eb7b --- /dev/null +++ b/libexec/rpc.sprayd/sprayd.c @@ -0,0 +1,155 @@ +/* + * Copyright (c) 1994 Christos Zoulas + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. All advertising materials mentioning features or use of this software + * must display the following acknowledgement: + * This product includes software developed by Christos Zoulas. + * 4. The name of the author may not be used to endorse or promote products + * derived from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR + * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. + * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, + * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF + * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + * + * $Id: sprayd.c,v 1.1 1995/10/18 08:43:22 deraadt Exp $ + */ + +#ifndef lint +static char rcsid[] = "$Id: sprayd.c,v 1.1 1995/10/18 08:43:22 deraadt Exp $"; +#endif /* not lint */ + +#include <stdio.h> +#include <signal.h> +#include <rpc/rpc.h> +#include <sys/time.h> +#include <syslog.h> +#include <rpcsvc/spray.h> + +static void spray_service __P((struct svc_req *, SVCXPRT *)); + +static int from_inetd = 1; + +#define TIMEOUT 120 + +void +cleanup() +{ + (void) pmap_unset(SPRAYPROG, SPRAYVERS); + exit(0); +} + +void +die() +{ + exit(0); +} + +int +main(argc, argv) + int argc; + char *argv[]; +{ + SVCXPRT *transp; + int sock = 0; + int proto = 0; + struct sockaddr_in from; + int fromlen; + + /* + * See if inetd started us + */ + fromlen = sizeof(from); + if (getsockname(0, (struct sockaddr *)&from, &fromlen) < 0) { + from_inetd = 0; + sock = RPC_ANYSOCK; + proto = IPPROTO_UDP; + } + + if (!from_inetd) { + daemon(0, 0); + + (void) pmap_unset(SPRAYPROG, SPRAYVERS); + + (void) signal(SIGINT, cleanup); + (void) signal(SIGTERM, cleanup); + (void) signal(SIGHUP, cleanup); + } else { + (void) signal(SIGALRM, die); + alarm(TIMEOUT); + } + + openlog("rpc.sprayd", LOG_CONS|LOG_PID, LOG_DAEMON); + + transp = svcudp_create(sock); + if (transp == NULL) { + syslog(LOG_ERR, "cannot create udp service."); + return 1; + } + if (!svc_register(transp, SPRAYPROG, SPRAYVERS, spray_service, proto)) { + syslog(LOG_ERR, + "unable to register (SPRAYPROG, SPRAYVERS, %s).", + proto ? "udp" : "(inetd)"); + return 1; + } + + svc_run(); + syslog(LOG_ERR, "svc_run returned"); + return 1; +} + + +static void +spray_service(rqstp, transp) + struct svc_req *rqstp; + SVCXPRT *transp; +{ + static spraycumul scum; + static struct timeval clear, get; + + switch (rqstp->rq_proc) { + case SPRAYPROC_CLEAR: + scum.counter = 0; + (void) gettimeofday(&clear, 0); + /*FALLTHROUGH*/ + + case NULLPROC: + (void)svc_sendreply(transp, xdr_void, (char *)NULL); + return; + + case SPRAYPROC_SPRAY: + scum.counter++; + return; + + case SPRAYPROC_GET: + (void) gettimeofday(&get, 0); + timersub(&get, &clear, &get); + scum.clock.sec = get.tv_sec; + scum.clock.usec = get.tv_usec; + break; + + default: + svcerr_noproc(transp); + return; + } + + if (!svc_sendreply(transp, xdr_spraycumul, (caddr_t)&scum)) { + svcerr_systemerr(transp); + syslog(LOG_ERR, "bad svc_sendreply"); + } +} |