Collapse gd_astpending and gd_reqpri together into gd_reqflags. gd_reqflags
[dragonfly.git] / libexec / rpc.sprayd / sprayd.c
CommitLineData
984263bc
MD
1/*
2 * Copyright (c) 1994 Christos Zoulas
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by Christos Zoulas.
16 * 4. The name of the author may not be used to endorse or promote products
17 * derived from this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
20 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
21 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
22 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
24 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
1de703da
MD
29 *
30 * $FreeBSD: src/libexec/rpc.sprayd/sprayd.c,v 1.5 1999/08/28 00:09:59 peter Exp $
31 * $DragonFly: src/libexec/rpc.sprayd/sprayd.c,v 1.2 2003/06/17 04:27:07 dillon Exp $
984263bc
MD
32 */
33
984263bc
MD
34#include <rpc/rpc.h>
35#include <rpc/pmap_clnt.h>
36#include <rpcsvc/spray.h>
37#include <signal.h>
38#include <stdio.h>
39#include <stdlib.h>
40#include <sys/time.h>
41#include <sys/socket.h>
42#include <syslog.h>
43#include <unistd.h>
44
45static void spray_service __P((struct svc_req *, SVCXPRT *));
46
47static int from_inetd = 1;
48
49#define timersub(tvp, uvp, vvp) \
50 do { \
51 (vvp)->tv_sec = (tvp)->tv_sec - (uvp)->tv_sec; \
52 (vvp)->tv_usec = (tvp)->tv_usec - (uvp)->tv_usec; \
53 if ((vvp)->tv_usec < 0) { \
54 (vvp)->tv_sec--; \
55 (vvp)->tv_usec += 1000000; \
56 } \
57 } while (0)
58
59#define TIMEOUT 120
60
61void
62cleanup()
63{
64 (void) pmap_unset(SPRAYPROG, SPRAYVERS);
65 exit(0);
66}
67
68void
69die()
70{
71 exit(0);
72}
73
74int
75main(argc, argv)
76 int argc;
77 char *argv[];
78{
79 SVCXPRT *transp;
80 int sock = 0;
81 int proto = 0;
82 struct sockaddr_in from;
83 int fromlen;
84
85 /*
86 * See if inetd started us
87 */
88 fromlen = sizeof(from);
89 if (getsockname(0, (struct sockaddr *)&from, &fromlen) < 0) {
90 from_inetd = 0;
91 sock = RPC_ANYSOCK;
92 proto = IPPROTO_UDP;
93 }
94
95 if (!from_inetd) {
96 daemon(0, 0);
97
98 (void) pmap_unset(SPRAYPROG, SPRAYVERS);
99
100 (void) signal(SIGINT, cleanup);
101 (void) signal(SIGTERM, cleanup);
102 (void) signal(SIGHUP, cleanup);
103 } else {
104 (void) signal(SIGALRM, die);
105 alarm(TIMEOUT);
106 }
107
108 openlog("rpc.sprayd", LOG_CONS|LOG_PID, LOG_DAEMON);
109
110 transp = svcudp_create(sock);
111 if (transp == NULL) {
112 syslog(LOG_ERR, "cannot create udp service");
113 return 1;
114 }
115 if (!svc_register(transp, SPRAYPROG, SPRAYVERS, spray_service, proto)) {
116 syslog(LOG_ERR,
117 "unable to register (SPRAYPROG, SPRAYVERS, %s)",
118 proto ? "udp" : "(inetd)");
119 return 1;
120 }
121
122 svc_run();
123 syslog(LOG_ERR, "svc_run returned");
124 return 1;
125}
126
127
128static void
129spray_service(rqstp, transp)
130 struct svc_req *rqstp;
131 SVCXPRT *transp;
132{
133 static spraycumul scum;
134 static struct timeval clear, get;
135
136 switch (rqstp->rq_proc) {
137 case SPRAYPROC_CLEAR:
138 scum.counter = 0;
139 (void) gettimeofday(&clear, 0);
140 /*FALLTHROUGH*/
141
142 case NULLPROC:
143 (void)svc_sendreply(transp, xdr_void, (char *)NULL);
144 return;
145
146 case SPRAYPROC_SPRAY:
147 scum.counter++;
148 return;
149
150 case SPRAYPROC_GET:
151 (void) gettimeofday(&get, 0);
152 timersub(&get, &clear, &get);
153 scum.clock.sec = get.tv_sec;
154 scum.clock.usec = get.tv_usec;
155 break;
156
157 default:
158 svcerr_noproc(transp);
159 return;
160 }
161
162 if (!svc_sendreply(transp, xdr_spraycumul, (caddr_t)&scum)) {
163 svcerr_systemerr(transp);
164 syslog(LOG_ERR, "bad svc_sendreply");
165 }
166}