361 lines
8.0 KiB
C
361 lines
8.0 KiB
C
/* $NetBSD: rup.c,v 1.18 1999/07/30 01:29:30 hubertf Exp $ */
|
||
|
||
/*-
|
||
* Copyright (c) 1993, John Brezak
|
||
* 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 the University of
|
||
* California, Berkeley and its contributors.
|
||
* 4. Neither the name of the University nor the names of its contributors
|
||
* may be used to endorse or promote products derived from this software
|
||
* without specific prior written permission.
|
||
*
|
||
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``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 REGENTS OR CONTRIBUTORS 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.
|
||
*/
|
||
|
||
#include <sys/cdefs.h>
|
||
#ifndef lint
|
||
__RCSID("$NetBSD: rup.c,v 1.18 1999/07/30 01:29:30 hubertf Exp $");
|
||
#endif /* not lint */
|
||
|
||
#include <sys/types.h>
|
||
#include <sys/socket.h>
|
||
#include <netinet/in.h>
|
||
#include <arpa/inet.h>
|
||
|
||
#include <err.h>
|
||
#include <netdb.h>
|
||
#include <rpc/rpc.h>
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <time.h>
|
||
#include <unistd.h>
|
||
|
||
#undef FSHIFT /* Use protocol's shift and scale values */
|
||
#undef FSCALE
|
||
#include <rpcsvc/rstat.h>
|
||
|
||
#define HOST_WIDTH 24
|
||
|
||
int printtime; /* print the remote host(s)'s time */
|
||
|
||
struct host_list {
|
||
struct host_list *next;
|
||
struct in_addr addr;
|
||
} *hosts;
|
||
|
||
int search_host __P((struct in_addr));
|
||
void remember_host __P((struct in_addr));
|
||
|
||
int
|
||
search_host(addr)
|
||
struct in_addr addr;
|
||
{
|
||
struct host_list *hp;
|
||
|
||
if (!hosts)
|
||
return(0);
|
||
|
||
for (hp = hosts; hp != NULL; hp = hp->next) {
|
||
if (hp->addr.s_addr == addr.s_addr)
|
||
return(1);
|
||
}
|
||
return(0);
|
||
}
|
||
|
||
void
|
||
remember_host(addr)
|
||
struct in_addr addr;
|
||
{
|
||
struct host_list *hp;
|
||
|
||
if (!(hp = (struct host_list *)malloc(sizeof(struct host_list)))) {
|
||
err(1, "malloc");
|
||
/* NOTREACHED */
|
||
}
|
||
hp->addr.s_addr = addr.s_addr;
|
||
hp->next = hosts;
|
||
hosts = hp;
|
||
}
|
||
|
||
|
||
|
||
struct rup_data {
|
||
char *host;
|
||
struct statstime statstime;
|
||
};
|
||
struct rup_data *rup_data;
|
||
int rup_data_idx = 0;
|
||
int rup_data_max = 0;
|
||
|
||
enum sort_type {
|
||
SORT_NONE,
|
||
SORT_HOST,
|
||
SORT_LDAV,
|
||
SORT_UPTIME
|
||
};
|
||
enum sort_type sort_type;
|
||
|
||
int compare __P((struct rup_data *, struct rup_data *));
|
||
void remember_rup_data __P((char *, struct statstime *));
|
||
int rstat_reply __P((char *, struct sockaddr_in *));
|
||
int print_rup_data __P((char *, statstime *));
|
||
void onehost __P((char *));
|
||
void allhosts __P((void));
|
||
int main __P((int, char *[]));
|
||
void usage __P((void));
|
||
|
||
int
|
||
compare(d1, d2)
|
||
struct rup_data *d1;
|
||
struct rup_data *d2;
|
||
{
|
||
switch(sort_type) {
|
||
case SORT_HOST:
|
||
return strcmp(d1->host, d2->host);
|
||
case SORT_LDAV:
|
||
return d1->statstime.avenrun[0]
|
||
- d2->statstime.avenrun[0];
|
||
case SORT_UPTIME:
|
||
return d1->statstime.boottime.tv_sec
|
||
- d2->statstime.boottime.tv_sec;
|
||
default:
|
||
/* something's really wrong here */
|
||
abort();
|
||
}
|
||
}
|
||
|
||
void
|
||
remember_rup_data(host, st)
|
||
char *host;
|
||
struct statstime *st;
|
||
{
|
||
if (rup_data_idx >= rup_data_max) {
|
||
rup_data_max += 16;
|
||
rup_data = realloc (rup_data,
|
||
rup_data_max * sizeof(struct rup_data));
|
||
if (rup_data == NULL) {
|
||
err (1, "realloc");
|
||
/* NOTREACHED */
|
||
}
|
||
}
|
||
|
||
rup_data[rup_data_idx].host = strdup(host);
|
||
rup_data[rup_data_idx].statstime = *st;
|
||
rup_data_idx++;
|
||
}
|
||
|
||
|
||
int
|
||
rstat_reply(replyp, raddrp)
|
||
char *replyp;
|
||
struct sockaddr_in *raddrp;
|
||
{
|
||
struct hostent *hp;
|
||
char *host;
|
||
statstime *host_stat = (statstime *)replyp;
|
||
|
||
if (!search_host(raddrp->sin_addr)) {
|
||
hp = gethostbyaddr((char *)&raddrp->sin_addr.s_addr,
|
||
sizeof(struct in_addr), AF_INET);
|
||
|
||
if (hp)
|
||
host = hp->h_name;
|
||
else
|
||
host = inet_ntoa(raddrp->sin_addr);
|
||
|
||
remember_host(raddrp->sin_addr);
|
||
|
||
if (sort_type != SORT_NONE) {
|
||
remember_rup_data(host, host_stat);
|
||
} else {
|
||
print_rup_data(host, host_stat);
|
||
}
|
||
}
|
||
|
||
return (0);
|
||
}
|
||
|
||
|
||
int
|
||
print_rup_data(host, host_stat)
|
||
char *host;
|
||
statstime *host_stat;
|
||
{
|
||
struct tm *tmp_time;
|
||
struct tm host_time;
|
||
unsigned ups=0,upm=0,uph=0,upd=0;
|
||
|
||
char days_buf[16];
|
||
char hours_buf[16];
|
||
|
||
if (printtime)
|
||
printf("%-*.*s", HOST_WIDTH-4, HOST_WIDTH-4, host);
|
||
else
|
||
printf("%-*.*s", HOST_WIDTH, HOST_WIDTH, host);
|
||
|
||
tmp_time = localtime((time_t *)&host_stat->curtime.tv_sec);
|
||
host_time = *tmp_time;
|
||
|
||
host_stat->curtime.tv_sec -= host_stat->boottime.tv_sec;
|
||
|
||
ups=host_stat->curtime.tv_sec;
|
||
upd=ups/(3600*24);
|
||
ups-=upd*3600*24;
|
||
uph=ups/3600;
|
||
ups-=uph*3600;
|
||
upm=ups/60;
|
||
|
||
if (upd != 0)
|
||
sprintf(days_buf, "%3u day%s, ", upd,
|
||
(upd > 1) ? "s" : "");
|
||
else
|
||
days_buf[0] = '\0';
|
||
|
||
if (uph != 0)
|
||
sprintf(hours_buf, "%2u:%02u, ",
|
||
uph, upm);
|
||
else
|
||
if (upm != 0)
|
||
sprintf(hours_buf, "%2u min%s ", upm,
|
||
(upm == 1) ? ", " : "s,");
|
||
else
|
||
hours_buf[0] = '\0';
|
||
if (printtime)
|
||
printf(" %2d:%02d%cm",
|
||
(host_time.tm_hour % 12) ? (host_time.tm_hour % 12) : 12,
|
||
host_time.tm_min, (host_time.tm_hour >= 12) ? 'p' : 'a');
|
||
|
||
printf(" up %9.9s%9.9s load average: %.2f %.2f %.2f\n",
|
||
days_buf, hours_buf,
|
||
(double)host_stat->avenrun[0]/FSCALE,
|
||
(double)host_stat->avenrun[1]/FSCALE,
|
||
(double)host_stat->avenrun[2]/FSCALE);
|
||
|
||
return(0);
|
||
}
|
||
|
||
|
||
void
|
||
onehost(host)
|
||
char *host;
|
||
{
|
||
CLIENT *rstat_clnt;
|
||
statstime host_stat;
|
||
static struct timeval timeout = {25, 0};
|
||
|
||
rstat_clnt = clnt_create(host, RSTATPROG, RSTATVERS_TIME, "udp");
|
||
if (rstat_clnt == NULL) {
|
||
warnx("%s", clnt_spcreateerror(host));
|
||
return;
|
||
}
|
||
|
||
memset((char *)&host_stat, 0, sizeof(host_stat));
|
||
if (clnt_call(rstat_clnt, RSTATPROC_STATS, xdr_void, NULL, xdr_statstime, &host_stat, timeout) != RPC_SUCCESS) {
|
||
warnx("%s", clnt_sperror(rstat_clnt, host));
|
||
return;
|
||
}
|
||
|
||
print_rup_data(host, &host_stat);
|
||
clnt_destroy(rstat_clnt);
|
||
}
|
||
|
||
void
|
||
allhosts()
|
||
{
|
||
statstime host_stat;
|
||
enum clnt_stat clnt_stat;
|
||
size_t i;
|
||
|
||
if (sort_type != SORT_NONE) {
|
||
printf("collecting responses...");
|
||
fflush(stdout);
|
||
}
|
||
|
||
clnt_stat = clnt_broadcast(RSTATPROG, RSTATVERS_TIME, RSTATPROC_STATS,
|
||
xdr_void, NULL,
|
||
xdr_statstime, (char*)&host_stat, rstat_reply);
|
||
if (clnt_stat != RPC_SUCCESS && clnt_stat != RPC_TIMEDOUT) {
|
||
warnx("%s", clnt_sperrno(clnt_stat));
|
||
exit(1);
|
||
}
|
||
|
||
if (sort_type != SORT_NONE) {
|
||
putchar('\n');
|
||
qsort(rup_data, rup_data_idx, sizeof(struct rup_data),
|
||
(int (*)(const void*, const void*))compare);
|
||
|
||
for (i = 0; i < rup_data_idx; i++) {
|
||
print_rup_data(rup_data[i].host, &rup_data[i].statstime);
|
||
}
|
||
}
|
||
}
|
||
|
||
int
|
||
main(argc, argv)
|
||
int argc;
|
||
char *argv[];
|
||
{
|
||
int ch;
|
||
extern int optind;
|
||
|
||
sort_type = SORT_NONE;
|
||
while ((ch = getopt(argc, argv, "dhlt")) != -1)
|
||
switch (ch) {
|
||
case 'd':
|
||
printtime = 1;
|
||
break;
|
||
case 'h':
|
||
sort_type = SORT_HOST;
|
||
break;
|
||
case 'l':
|
||
sort_type = SORT_LDAV;
|
||
break;
|
||
case 't':
|
||
sort_type = SORT_UPTIME;
|
||
break;
|
||
default:
|
||
usage();
|
||
/*NOTREACHED*/
|
||
}
|
||
|
||
setlinebuf(stdout);
|
||
|
||
if (argc == optind)
|
||
allhosts();
|
||
else {
|
||
for (; optind < argc; optind++)
|
||
onehost(argv[optind]);
|
||
}
|
||
|
||
exit(0);
|
||
}
|
||
|
||
void
|
||
usage()
|
||
{
|
||
fprintf(stderr, "Usage: rup [-dhlt] [hosts ...]\n");
|
||
exit(1);
|
||
}
|