w.c revision 1.35 1 /* $NetBSD: w.c,v 1.35 2000/05/26 03:10:31 simonb Exp $ */
2
3 /*-
4 * Copyright (c) 1980, 1991, 1993, 1994
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 * must display the following acknowledgement:
17 * This product includes software developed by the University of
18 * California, Berkeley and its contributors.
19 * 4. Neither the name of the University nor the names of its contributors
20 * may be used to endorse or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36 #include <sys/cdefs.h>
37 #ifndef lint
38 __COPYRIGHT("@(#) Copyright (c) 1980, 1991, 1993, 1994\n\
39 The Regents of the University of California. All rights reserved.\n");
40 #endif /* not lint */
41
42 #ifndef lint
43 #if 0
44 static char sccsid[] = "@(#)w.c 8.6 (Berkeley) 6/30/94";
45 #else
46 __RCSID("$NetBSD: w.c,v 1.35 2000/05/26 03:10:31 simonb Exp $");
47 #endif
48 #endif /* not lint */
49
50 /*
51 * w - print system status (who and what)
52 *
53 * This program is similar to the systat command on Tenex/Tops 10/20
54 *
55 */
56 #include <sys/param.h>
57 #include <sys/time.h>
58 #include <sys/stat.h>
59 #include <sys/sysctl.h>
60 #include <sys/proc.h>
61 #include <sys/user.h>
62 #include <sys/ioctl.h>
63 #include <sys/socket.h>
64
65 #include <netinet/in.h>
66 #include <arpa/inet.h>
67
68 #include <ctype.h>
69 #include <err.h>
70 #include <errno.h>
71 #include <fcntl.h>
72 #include <kvm.h>
73 #include <limits.h>
74 #include <netdb.h>
75 #include <nlist.h>
76 #include <paths.h>
77 #include <stdio.h>
78 #include <stdlib.h>
79 #include <string.h>
80 #include <time.h>
81 #include <tzfile.h>
82 #include <unistd.h>
83 #include <utmp.h>
84 #include <vis.h>
85
86 #include "extern.h"
87
88 struct timeval boottime;
89 struct utmp utmp;
90 struct winsize ws;
91 kvm_t *kd;
92 time_t now; /* the current time of day */
93 time_t uptime; /* time of last reboot & elapsed time since */
94 int ttywidth; /* width of tty */
95 int argwidth; /* width of tty */
96 int header = 1; /* true if -h flag: don't print heading */
97 int nflag; /* true if -n flag: don't convert addrs */
98 int sortidle; /* sort bu idle time */
99 char *sel_user; /* login of particular user selected */
100 char domain[MAXHOSTNAMELEN + 1];
101
102 /*
103 * One of these per active utmp entry.
104 */
105 struct entry {
106 struct entry *next;
107 struct utmp utmp;
108 dev_t tdev; /* dev_t of terminal */
109 time_t idle; /* idle time of terminal in seconds */
110 struct kinfo_proc2 *kp; /* `most interesting' proc */
111 } *ep, *ehead = NULL, **nextp = &ehead;
112
113 static void pr_args __P((struct kinfo_proc2 *));
114 static void pr_header __P((time_t *, int));
115 static struct stat
116 *ttystat __P((char *));
117 static void usage __P((int));
118 int main __P((int, char **));
119
120 int
121 main(argc, argv)
122 int argc;
123 char **argv;
124 {
125 extern char *__progname;
126 struct kinfo_proc2 *kp;
127 struct hostent *hp;
128 struct stat *stp;
129 FILE *ut;
130 struct in_addr l;
131 int ch, i, nentries, nusers, wcmd;
132 char *memf, *nlistf, *p, *x;
133 char buf[MAXHOSTNAMELEN], errbuf[_POSIX2_LINE_MAX];
134
135 /* Are we w(1) or uptime(1)? */
136 p = __progname;
137 if (*p == '-')
138 p++;
139 if (*p == 'u') {
140 wcmd = 0;
141 p = "";
142 } else {
143 wcmd = 1;
144 p = "hiflM:N:nsuw";
145 }
146
147 memf = nlistf = NULL;
148 while ((ch = getopt(argc, argv, p)) != -1)
149 switch (ch) {
150 case 'h':
151 header = 0;
152 break;
153 case 'i':
154 sortidle = 1;
155 break;
156 case 'M':
157 header = 0;
158 memf = optarg;
159 break;
160 case 'N':
161 nlistf = optarg;
162 break;
163 case 'n':
164 nflag = 1;
165 break;
166 case 'f': case 'l': case 's': case 'u': case 'w':
167 warnx("[-flsuw] no longer supported");
168 /* FALLTHROUGH */
169 case '?':
170 default:
171 usage(wcmd);
172 }
173 argc -= optind;
174 argv += optind;
175
176 if ((kd = kvm_openfiles(nlistf, memf, NULL,
177 memf == NULL ? KVM_NO_FILES : O_RDONLY, errbuf)) == NULL)
178 errx(1, "%s", errbuf);
179
180 (void)time(&now);
181 if ((ut = fopen(_PATH_UTMP, "r")) == NULL)
182 err(1, "%s", _PATH_UTMP);
183
184 if (*argv)
185 sel_user = *argv;
186
187 for (nusers = 0; fread(&utmp, sizeof(utmp), 1, ut);) {
188 if (utmp.ut_name[0] == '\0')
189 continue;
190 ++nusers;
191 if (wcmd == 0 || (sel_user &&
192 strncmp(utmp.ut_name, sel_user, UT_NAMESIZE) != 0))
193 continue;
194 if ((ep = calloc(1, sizeof(struct entry))) == NULL)
195 err(1, NULL);
196 *nextp = ep;
197 nextp = &(ep->next);
198 memmove(&(ep->utmp), &utmp, sizeof(struct utmp));
199 if (!(stp = ttystat(ep->utmp.ut_line)))
200 continue;
201 ep->tdev = stp->st_rdev;
202 #ifdef CPU_CONSDEV
203 /*
204 * If this is the console device, attempt to ascertain
205 * the true console device dev_t.
206 */
207 if (ep->tdev == 0) {
208 int mib[2];
209 size_t size;
210
211 mib[0] = CTL_MACHDEP;
212 mib[1] = CPU_CONSDEV;
213 size = sizeof(dev_t);
214 (void) sysctl(mib, 2, &ep->tdev, &size, NULL, 0);
215 }
216 #endif
217 if ((ep->idle = now - stp->st_atime) < 0)
218 ep->idle = 0;
219 }
220 (void)fclose(ut);
221
222 if (header || wcmd == 0) {
223 pr_header(&now, nusers);
224 if (wcmd == 0)
225 exit (0);
226 }
227
228 #define HEADER "USER TTY FROM LOGIN@ IDLE WHAT\n"
229 #define WUSED (sizeof (HEADER) - sizeof ("WHAT\n"))
230 (void)printf(HEADER);
231
232 if ((kp = kvm_getproc2(kd, KERN_PROC_ALL, 0, sizeof(struct kinfo_proc),
233 &nentries)) == NULL)
234 errx(1, "%s", kvm_geterr(kd));
235 for (i = 0; i < nentries; i++, kp++) {
236
237 if (kp->p_stat == SIDL || kp->p_stat == SZOMB)
238 continue;
239 for (ep = ehead; ep != NULL; ep = ep->next) {
240 if (ep->tdev == kp->p_tdev &&
241 kp->p__pgid == kp->p_tpgid) {
242 /*
243 * Proc is in foreground of this terminal
244 */
245 if (proc_compare(ep->kp, kp))
246 ep->kp = kp;
247 break;
248 }
249 }
250 }
251 if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == -1 &&
252 ioctl(STDERR_FILENO, TIOCGWINSZ, &ws) == -1 &&
253 ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) || ws.ws_col == 0)
254 ttywidth = 79;
255 else
256 ttywidth = ws.ws_col - 1;
257 argwidth = ttywidth - WUSED;
258 if (argwidth < 4)
259 argwidth = 8;
260 /* sort by idle time */
261 if (sortidle && ehead != NULL) {
262 struct entry *from = ehead, *save;
263
264 ehead = NULL;
265 while (from != NULL) {
266 for (nextp = &ehead;
267 (*nextp) && from->idle >= (*nextp)->idle;
268 nextp = &(*nextp)->next)
269 continue;
270 save = from;
271 from = from->next;
272 save->next = *nextp;
273 *nextp = save;
274 }
275 }
276
277 if (!nflag) {
278 int rv;
279
280 rv = gethostname(domain, sizeof(domain));
281 domain[sizeof(domain) - 1] = '\0';
282 if (rv < 0 || (p = strchr(domain, '.')) == 0)
283 domain[0] = '\0';
284 else
285 memmove(domain, p, strlen(p) + 1);
286 }
287
288 for (ep = ehead; ep != NULL; ep = ep->next) {
289 p = *ep->utmp.ut_host ? ep->utmp.ut_host : "-";
290 for (x = p; x < p + UT_HOSTSIZE; x++)
291 if (*x == '\0' || *x == ':')
292 break;
293 if (x == p + UT_HOSTSIZE || *x != ':')
294 x = NULL;
295 else
296 *x++ = '\0';
297
298 if (!nflag && inet_aton(p, &l) &&
299 (hp = gethostbyaddr((char *)&l, sizeof(l), AF_INET))) {
300 if (domain[0] != '\0') {
301 p = hp->h_name;
302 p += strlen(hp->h_name);
303 p -= strlen(domain);
304 if (p > hp->h_name && strcmp(p, domain) == 0)
305 *p = '\0';
306 }
307 p = hp->h_name;
308 }
309 if (x) {
310 (void)snprintf(buf, sizeof(buf), "%s:%.*s", p,
311 (int)(ep->utmp.ut_host + UT_HOSTSIZE - x), x);
312 p = buf;
313 }
314 (void)printf("%-*.*s %-2.2s %-*.*s ",
315 UT_NAMESIZE, UT_NAMESIZE, ep->utmp.ut_name,
316 (strncmp(ep->utmp.ut_line, "tty", 3) &&
317 strncmp(ep->utmp.ut_line, "dty", 3)) ?
318 ep->utmp.ut_line : ep->utmp.ut_line + 3,
319 UT_HOSTSIZE, UT_HOSTSIZE, *p ? p : "-");
320 pr_attime(&ep->utmp.ut_time, &now);
321 pr_idle(ep->idle);
322 pr_args(ep->kp);
323 printf("\n");
324 }
325 exit(0);
326 }
327
328 static void
329 pr_args(kp)
330 struct kinfo_proc2 *kp;
331 {
332 char **argv;
333 int left;
334
335 if (kp == 0)
336 goto nothing;
337 left = argwidth;
338 argv = kvm_getargv2(kd, kp, argwidth);
339 if (argv == 0)
340 goto nothing;
341 while (*argv) {
342 fmt_puts(*argv, &left);
343 argv++;
344 fmt_putc(' ', &left);
345 }
346 return;
347 nothing:
348 putchar('-');
349 }
350
351 static void
352 pr_header(nowp, nusers)
353 time_t *nowp;
354 int nusers;
355 {
356 double avenrun[3];
357 time_t uptime;
358 int days, hrs, i, mins;
359 int mib[2];
360 size_t size;
361 char buf[256];
362
363 /*
364 * Print time of day.
365 *
366 * SCCS forces the string manipulation below, as it replaces
367 * %, M, and % in a character string with the file name.
368 */
369 (void)strftime(buf, sizeof(buf),
370 __CONCAT("%l:%","M%p"), localtime(nowp));
371 buf[sizeof(buf) - 1] = '\0';
372 (void)printf("%s ", buf);
373
374 /*
375 * Print how long system has been up.
376 * (Found by looking getting "boottime" from the kernel)
377 */
378 mib[0] = CTL_KERN;
379 mib[1] = KERN_BOOTTIME;
380 size = sizeof(boottime);
381 if (sysctl(mib, 2, &boottime, &size, NULL, 0) != -1 &&
382 boottime.tv_sec != 0) {
383 uptime = now - boottime.tv_sec;
384 uptime += 30;
385 if (uptime > SECSPERMIN) {
386 days = uptime / SECSPERDAY;
387 uptime %= SECSPERDAY;
388 hrs = uptime / SECSPERHOUR;
389 uptime %= SECSPERHOUR;
390 mins = uptime / SECSPERMIN;
391 (void)printf(" up");
392 if (days > 0)
393 (void)printf(" %d day%s,", days,
394 days > 1 ? "s" : "");
395 if (hrs > 0 && mins > 0)
396 (void)printf(" %2d:%02d,", hrs, mins);
397 else {
398 if (hrs > 0)
399 (void)printf(" %d hr%s,",
400 hrs, hrs > 1 ? "s" : "");
401 if (mins > 0)
402 (void)printf(" %d min%s,",
403 mins, mins > 1 ? "s" : "");
404 }
405 }
406 }
407
408 /* Print number of users logged in to system */
409 (void)printf(" %d user%s", nusers, nusers != 1 ? "s" : "");
410
411 /*
412 * Print 1, 5, and 15 minute load averages.
413 */
414 if (getloadavg(avenrun, sizeof(avenrun) / sizeof(avenrun[0])) == -1)
415 (void)printf(", no load average information available\n");
416 else {
417 (void)printf(", load averages:");
418 for (i = 0; i < (sizeof(avenrun) / sizeof(avenrun[0])); i++) {
419 if (i > 0)
420 (void)printf(",");
421 (void)printf(" %.2f", avenrun[i]);
422 }
423 (void)printf("\n");
424 }
425 }
426
427 static struct stat *
428 ttystat(line)
429 char *line;
430 {
431 static struct stat sb;
432 char ttybuf[MAXPATHLEN];
433
434 (void)snprintf(ttybuf, sizeof(ttybuf), "%s/%s", _PATH_DEV, line);
435 if (stat(ttybuf, &sb))
436 return (NULL);
437 return (&sb);
438 }
439
440 static void
441 usage(wcmd)
442 int wcmd;
443 {
444 if (wcmd)
445 (void)fprintf(stderr,
446 "usage: w: [-hin] [-M core] [-N system] [user]\n");
447 else
448 (void)fprintf(stderr, "uptime\n");
449 exit(1);
450 }
451