rstat_proc.c revision 1.2 1 /* @(#)rstat_proc.c 2.2 88/08/01 4.0 RPCSRC */
2 #ifndef lint
3 static char sccsid[] = "@(#)rpc.rstatd.c 1.1 86/09/25 Copyr 1984 Sun Micro";
4 #endif
5
6 /*
7 * Copyright (c) 1984 by Sun Microsystems, Inc.
8 */
9
10 /*
11 * Sun RPC is a product of Sun Microsystems, Inc. and is provided for
12 * unrestricted use provided that this legend is included on all tape
13 * media and as a part of the software program in whole or part. Users
14 * may copy or modify Sun RPC without charge, but are not authorized
15 * to license or distribute it to anyone else except as part of a product or
16 * program developed by the user.
17 *
18 * SUN RPC IS PROVIDED AS IS WITH NO WARRANTIES OF ANY KIND INCLUDING THE
19 * WARRANTIES OF DESIGN, MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
20 * PURPOSE, OR ARISING FROM A COURSE OF DEALING, USAGE OR TRADE PRACTICE.
21 *
22 * Sun RPC is provided with no support and without any obligation on the
23 * part of Sun Microsystems, Inc. to assist in its use, correction,
24 * modification or enhancement.
25 *
26 * SUN MICROSYSTEMS, INC. SHALL HAVE NO LIABILITY WITH RESPECT TO THE
27 * INFRINGEMENT OF COPYRIGHTS, TRADE SECRETS OR ANY PATENTS BY SUN RPC
28 * OR ANY PART THEREOF.
29 *
30 * In no event will Sun Microsystems, Inc. be liable for any lost revenue
31 * or profits or other special, indirect and consequential damages, even if
32 * Sun has been advised of the possibility of such damages.
33 *
34 * Sun Microsystems, Inc.
35 * 2550 Garcia Avenue
36 * Mountain View, California 94043
37 */
38
39 /*
40 * rstat service: built with rstat.x and derived from rpc.rstatd.c
41 */
42
43 #include <signal.h>
44 #include <stdio.h>
45 #include <rpc/rpc.h>
46 #include <sys/socket.h>
47 #include <nlist.h>
48 #include <syslog.h>
49 #include <sys/errno.h>
50 #include <sys/param.h>
51 #ifdef BSD
52 #include <sys/vmmeter.h>
53 #include <sys/dkstat.h>
54 #else
55 #include <sys/dk.h>
56 #endif
57 #include <net/if.h>
58 #include <rpcsvc/rstat.h>
59
60 struct nlist nl[] = {
61 #define X_CPTIME 0
62 { "_cp_time" },
63 #define X_SUM 1
64 { "_sum" },
65 #define X_IFNET 2
66 { "_ifnet" },
67 #define X_DKXFER 3
68 { "_dk_xfer" },
69 #define X_BOOTTIME 4
70 { "_boottime" },
71 #define X_HZ 5
72 { "_hz" },
73 #ifdef vax
74 #define X_AVENRUN 6
75 { "_avenrun" },
76 #endif
77 "",
78 };
79 int firstifnet, numintfs; /* chain of ethernet interfaces */
80 int stats_service();
81
82 extern int from_inetd;
83 int sincelastreq = 0; /* number of alarms since last request */
84 extern int closedown;
85
86 union {
87 struct stats s1;
88 struct statsswtch s2;
89 struct statstime s3;
90 } stats_all;
91
92 void updatestat();
93 static stat_is_init = 0;
94 extern int errno;
95
96 #ifndef FSCALE
97 #define FSCALE (1 << 8)
98 #endif
99
100 #ifndef BSD
101 /*
102 * BSD has the kvm facility for getting info from the
103 * kernel. If you aren't on BSD, this surfices.
104 */
105 int kmem;
106
107 kvm_read(off, addr, size)
108 unsigned long off, size;
109 char *addr;
110 {
111 int len;
112 if (lseek(kmem, (long)off, 0) == -1)
113 return(-1);
114 return(read(kmem, addr, size));
115 }
116
117 kvm_nlist(nl)
118 struct nlist *nl;
119 {
120 int n = nlist("/vmunix", nl);
121 if (nl[0].n_value == 0)
122 return(n);
123
124 if ((kmem = open("/dev/kmem", 0)) < 0)
125 return(-1);
126 return(0);
127 }
128 #endif
129
130 stat_init()
131 {
132 stat_is_init = 1;
133 setup();
134 updatestat();
135 (void) signal(SIGALRM, updatestat);
136 alarm(1);
137 }
138
139 statstime *
140 rstatproc_stats_3()
141 {
142 if (! stat_is_init)
143 stat_init();
144 sincelastreq = 0;
145 return(&stats_all.s3);
146 }
147
148 statsswtch *
149 rstatproc_stats_2()
150 {
151 if (! stat_is_init)
152 stat_init();
153 sincelastreq = 0;
154 return(&stats_all.s2);
155 }
156
157 stats *
158 rstatproc_stats_1()
159 {
160 if (! stat_is_init)
161 stat_init();
162 sincelastreq = 0;
163 return(&stats_all.s1);
164 }
165
166 u_int *
167 rstatproc_havedisk_3()
168 {
169 static u_int have;
170
171 if (! stat_is_init)
172 stat_init();
173 sincelastreq = 0;
174 have = havedisk();
175 return(&have);
176 }
177
178 u_int *
179 rstatproc_havedisk_2()
180 {
181 return(rstatproc_havedisk_3());
182 }
183
184 u_int *
185 rstatproc_havedisk_1()
186 {
187 return(rstatproc_havedisk_3());
188 }
189
190 void
191 updatestat()
192 {
193 int off, i, hz;
194 struct vmmeter sum;
195 struct ifnet ifnet;
196 double avrun[3];
197 struct timeval tm, btm;
198
199 #ifdef DEBUG
200 fprintf(stderr, "entering updatestat\n");
201 #endif
202 if (sincelastreq >= closedown) {
203 #ifdef DEBUG
204 fprintf(stderr, "about to closedown\n");
205 #endif
206 if (from_inetd)
207 exit(0);
208 else {
209 stat_is_init = 0;
210 return;
211 }
212 }
213 sincelastreq++;
214
215 if (kvm_read((long)nl[X_HZ].n_value, (char *)&hz, sizeof hz) != sizeof hz) {
216 syslog(LOG_ERR, "rstat: can't read hz from kmem\n");
217 exit(1);
218 }
219 if (kvm_read((long)nl[X_CPTIME].n_value, (char *)stats_all.s1.cp_time, sizeof (stats_all.s1.cp_time))
220 != sizeof (stats_all.s1.cp_time)) {
221 syslog(LOG_ERR, "rstat: can't read cp_time from kmem\n");
222 exit(1);
223 }
224 #ifdef vax
225 if (kvm_read((long)nl[X_AVENRUN].n_value, (char *)avrun, sizeof (avrun)) != sizeof (avrun)) {
226 syslog(LOG_ERR, "rstat: can't read avenrun from kmem\n");
227 exit(1);
228 }
229 #endif
230 #ifdef BSD
231 (void)getloadavg(avrun, sizeof(avrun) / sizeof(avrun[0]));
232 #endif
233 stats_all.s2.avenrun[0] = avrun[0] * FSCALE;
234 stats_all.s2.avenrun[1] = avrun[1] * FSCALE;
235 stats_all.s2.avenrun[2] = avrun[2] * FSCALE;
236 if (kvm_read((long)nl[X_BOOTTIME].n_value, (char *)&btm, sizeof (stats_all.s2.boottime))
237 != sizeof (stats_all.s2.boottime)) {
238 syslog(LOG_ERR, "rstat: can't read boottime from kmem\n");
239 exit(1);
240 }
241 stats_all.s2.boottime.tv_sec = btm.tv_sec;
242 stats_all.s2.boottime.tv_usec = btm.tv_usec;
243
244
245 #ifdef DEBUG
246 fprintf(stderr, "%d %d %d %d\n", stats_all.s1.cp_time[0],
247 stats_all.s1.cp_time[1], stats_all.s1.cp_time[2], stats_all.s1.cp_time[3]);
248 #endif
249
250 if (kvm_read((long)nl[X_SUM].n_value, (char *)&sum, sizeof sum) != sizeof sum) {
251 syslog(LOG_ERR, "rstat: can't read sum from kmem\n");
252 exit(1);
253 }
254 stats_all.s1.v_pgpgin = sum.v_pgpgin;
255 stats_all.s1.v_pgpgout = sum.v_pgpgout;
256 stats_all.s1.v_pswpin = sum.v_pswpin;
257 stats_all.s1.v_pswpout = sum.v_pswpout;
258 stats_all.s1.v_intr = sum.v_intr;
259 gettimeofday(&tm, (struct timezone *) 0);
260 stats_all.s1.v_intr -= hz*(tm.tv_sec - btm.tv_sec) +
261 hz*(tm.tv_usec - btm.tv_usec)/1000000;
262 stats_all.s2.v_swtch = sum.v_swtch;
263
264 if (kvm_read((long)nl[X_DKXFER].n_value, (char *)stats_all.s1.dk_xfer, sizeof (stats_all.s1.dk_xfer))
265 != sizeof (stats_all.s1.dk_xfer)) {
266 syslog(LOG_ERR, "rstat: can't read dk_xfer from kmem\n");
267 exit(1);
268 }
269
270 stats_all.s1.if_ipackets = 0;
271 stats_all.s1.if_opackets = 0;
272 stats_all.s1.if_ierrors = 0;
273 stats_all.s1.if_oerrors = 0;
274 stats_all.s1.if_collisions = 0;
275 for (off = firstifnet, i = 0; off && i < numintfs; i++) {
276 if (kvm_read(off, (char *)&ifnet, sizeof ifnet) != sizeof ifnet) {
277 syslog(LOG_ERR, "rstat: can't read ifnet from kmem\n");
278 exit(1);
279 }
280 stats_all.s1.if_ipackets += ifnet.if_ipackets;
281 stats_all.s1.if_opackets += ifnet.if_opackets;
282 stats_all.s1.if_ierrors += ifnet.if_ierrors;
283 stats_all.s1.if_oerrors += ifnet.if_oerrors;
284 stats_all.s1.if_collisions += ifnet.if_collisions;
285 off = (int) ifnet.if_next;
286 }
287 gettimeofday((struct timeval *)&stats_all.s3.curtime,
288 (struct timezone *) 0);
289 alarm(1);
290 }
291
292 setup()
293 {
294 struct ifnet ifnet;
295 int off;
296
297 if (kvm_nlist(nl) != 0) {
298 syslog(LOG_ERR, "rstatd: Can't get namelist.");
299 exit (1);
300 }
301
302 if (kvm_read((long)nl[X_IFNET].n_value, &firstifnet,
303 sizeof(int)) != sizeof(int)) {
304 syslog(LOG_ERR, "rstat: can't read firstifnet from kmem\n");
305 exit(1);
306 }
307
308 numintfs = 0;
309 for (off = firstifnet; off;) {
310 if (kvm_read(off, (char *)&ifnet, sizeof ifnet) != sizeof ifnet) {
311 syslog(LOG_ERR, "rstat: can't read ifnet from kmem\n");
312 exit(1);
313 }
314 numintfs++;
315 off = (int) ifnet.if_next;
316 }
317 }
318
319 /*
320 * returns true if have a disk
321 */
322 havedisk()
323 {
324 int i, cnt;
325 long xfer[DK_NDRIVE];
326
327 if (kvm_nlist(nl) != 0) {
328 syslog(LOG_ERR, "rstatd: Can't get namelist.");
329 exit (1);
330 }
331
332 if (kvm_read((long)nl[X_DKXFER].n_value, (char *)xfer, sizeof xfer)!= sizeof xfer) {
333 syslog(LOG_ERR, "rstat: can't read kmem\n");
334 exit(1);
335 }
336 cnt = 0;
337 for (i=0; i < DK_NDRIVE; i++)
338 cnt += xfer[i];
339 return (cnt != 0);
340 }
341
342 void
343 rstat_service(rqstp, transp)
344 struct svc_req *rqstp;
345 SVCXPRT *transp;
346 {
347 union {
348 int fill;
349 } argument;
350 char *result;
351 bool_t (*xdr_argument)(), (*xdr_result)();
352 char *(*local)();
353
354 switch (rqstp->rq_proc) {
355 case NULLPROC:
356 (void)svc_sendreply(transp, xdr_void, (char *)NULL);
357 goto leave;
358
359 case RSTATPROC_STATS:
360 xdr_argument = xdr_void;
361 xdr_result = xdr_statstime;
362 switch (rqstp->rq_vers) {
363 case RSTATVERS_ORIG:
364 local = (char *(*)()) rstatproc_stats_1;
365 break;
366 case RSTATVERS_SWTCH:
367 local = (char *(*)()) rstatproc_stats_2;
368 break;
369 case RSTATVERS_TIME:
370 local = (char *(*)()) rstatproc_stats_3;
371 break;
372 default:
373 svcerr_progvers(transp, RSTATVERS_ORIG, RSTATVERS_TIME);
374 goto leave;
375 /*NOTREACHED*/
376 }
377 break;
378
379 case RSTATPROC_HAVEDISK:
380 xdr_argument = xdr_void;
381 xdr_result = xdr_u_int;
382 switch (rqstp->rq_vers) {
383 case RSTATVERS_ORIG:
384 local = (char *(*)()) rstatproc_havedisk_1;
385 break;
386 case RSTATVERS_SWTCH:
387 local = (char *(*)()) rstatproc_havedisk_2;
388 break;
389 case RSTATVERS_TIME:
390 local = (char *(*)()) rstatproc_havedisk_3;
391 break;
392 default:
393 svcerr_progvers(transp, RSTATVERS_ORIG, RSTATVERS_TIME);
394 goto leave;
395 /*NOTREACHED*/
396 }
397 break;
398
399 default:
400 svcerr_noproc(transp);
401 goto leave;
402 }
403 bzero((char *)&argument, sizeof(argument));
404 if (!svc_getargs(transp, xdr_argument, &argument)) {
405 svcerr_decode(transp);
406 goto leave;
407 }
408 result = (*local)(&argument, rqstp);
409 if (result != NULL && !svc_sendreply(transp, xdr_result, result)) {
410 svcerr_systemerr(transp);
411 }
412 if (!svc_freeargs(transp, xdr_argument, &argument)) {
413 (void)fprintf(stderr, "unable to free arguments\n");
414 exit(1);
415 }
416 leave:
417 if (from_inetd)
418 exit(0);
419 }
420