]> mj.ucw.cz Git - nwho.git/blob - nwho.c
Renamed rywho_pkt to nwho_pkt
[nwho.git] / nwho.c
1 /*
2  *      The Remote User Information Lister
3  *
4  *      (c) 1997--2010 Martin Mares <mj@ucw.cz>
5  */
6
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <string.h>
10 #include <fcntl.h>
11 #include <unistd.h>
12 #include <dirent.h>
13 #include <time.h>
14 #include <netinet/in.h>
15
16 #include "net.h"
17
18 static int is_uptime;
19 static time_t now;
20
21 static void
22 puttime(int s)
23 {
24   int d, h, m;
25   if (s < 100)
26     printf("%4ds", s);
27   else
28     {
29       m = s/60;
30       s %= 60;
31       h = m/60;
32       m %= 60;
33       if (h < 100)
34         printf("%02d.%02d", h, m);
35       else
36         {
37           d = h/24;
38           h %= 24;
39           if (d < 100)
40             printf("%2dd%02d", d, h);
41           else
42             printf("%4dd", d);
43         }
44     }
45 }
46
47 static void
48 show_uptime(char *name, struct nwho_pkt *p)
49 {
50   int i;
51
52   if (now - ntohl(p->server_time) >= DEFAULT_DOWN_TIME)
53     {
54       printf("%-16s down\n", name);
55       return;
56     }
57   printf("%-16s up ", name);
58   puttime(ntohl(p->uptime));
59   printf("  load");
60   for(i=0; i<3; i++)
61     {
62       int l = ntohl(p->avl[i]);
63       printf(" %2d.%02d", l/100, l%100);
64     }
65   printf(" %3d users\n", (int) ntohl(p->num_users));
66 }
67
68 static void
69 show_users(char *name, struct nwho_pkt *p)
70 {
71   int u;
72   int m = ntohl(p->num_users);
73   struct userinfo *i;
74
75   if (now - ntohl(p->server_time) >= DEFAULT_DOWN_TIME)
76     return;
77   for(u=0; u<m; u++)
78     {
79       i = &p->users[u];
80       printf("%-8.8s %-7s %c %-16s ", i->name, i->con, (i->mesg_y ? ' ' : '-'), name);
81       puttime(ntohl(i->login_time));
82       putchar(' ');
83       puttime(ntohl(i->idle_time));
84       putchar('\n');
85     }
86   if (m == MAX_USERS)
87     printf("%s: MAX_USERS reached!\n", name);
88 }
89
90 static void
91 scan(void)
92 {
93   DIR *d;
94   struct dirent *e;
95   struct nwho_pkt pkt;
96   int fd, r;
97   int is = 0;
98
99   if (chdir(NWHO_SPOOL_DIR) < 0)
100     {
101       fprintf(stderr, "chdir(" NWHO_SPOOL_DIR "): %m\n");
102       exit(1);
103     }
104   d = opendir(".");
105   if (!d)
106     {
107       perror("opendir");
108       exit(1);
109     }
110   while (e = readdir(d))
111     if (e->d_name[0] != '.')
112       {
113         fd = open(e->d_name, O_RDONLY);
114         if (fd < 0)
115           {
116             fprintf(stderr, "%s: %m\n", e->d_name);
117             continue;
118           }
119         r = read(fd, &pkt, sizeof(pkt));
120         close(fd);
121         if (r < sizeof(struct nwho_pkt) - MAX_USERS*sizeof(struct userinfo)
122             || pkt.magic != htonl(NWHO_MAGIC)
123             || r != sizeof(struct nwho_pkt) - (MAX_USERS - ntohl(pkt.num_users))*sizeof(struct userinfo))
124           {
125             fprintf(stderr, "%s: Malformed record\n", e->d_name);
126             continue;
127           }
128         (is_uptime ? show_uptime : show_users)(e->d_name, &pkt);
129         is = 1;
130       }
131   closedir(d);
132   if (!is)
133     puts("No data available.");
134 }
135
136 int
137 main(int argc, char **argv)
138 {
139   if (strstr(argv[0], "uptime"))
140     is_uptime = 1;
141   if (argc != 1)
142     {
143       fprintf(stderr, "Usage: %s\n", argv[0]);
144       return 1;
145     }
146   if (!is_uptime)
147     puts("Name     Li      M Where            LogT  IdleT");
148   now = time(NULL);
149   scan();
150   return 0;
151 }