xref: /netbsd/usr.bin/w/w.c (revision c4a72b64) 
1 /*	$NetBSD: w.c,v 1.54 2002/10/21 10:18:23 enami 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.54 2002/10/21 10:18:23 enami 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/types.h>
58 #include <sys/time.h>
59 #include <sys/stat.h>
60 #include <sys/sysctl.h>
61 #include <sys/proc.h>
62 #include <sys/user.h>
63 #include <sys/ioctl.h>
64 #include <sys/socket.h>
65 
66 #include <netinet/in.h>
67 #include <arpa/inet.h>
68 
69 #include <ctype.h>
70 #include <err.h>
71 #include <errno.h>
72 #include <fcntl.h>
73 #include <kvm.h>
74 #include <limits.h>
75 #include <netdb.h>
76 #include <nlist.h>
77 #include <paths.h>
78 #include <stdio.h>
79 #include <stdlib.h>
80 #include <string.h>
81 #include <time.h>
82 #include <tzfile.h>
83 #include <unistd.h>
84 #ifdef SUPPORT_UTMP
85 #include <utmp.h>
86 #endif
87 #ifdef SUPPORT_UTMPX
88 #include <utmpx.h>
89 #endif
90 #include <vis.h>
91 
92 #include "extern.h"
93 
94 #define	max(a,b)	(((a)>(b))?(a):(b))
95 
96 struct timeval	boottime;
97 struct winsize	ws;
98 kvm_t	       *kd;
99 time_t		now;		/* the current time of day */
100 time_t		uptime;		/* time of last reboot & elapsed time since */
101 int		ttywidth;	/* width of tty */
102 int		argwidth;	/* width of tty left to print process args */
103 int		header = 1;	/* true if -h flag: don't print heading */
104 int		nflag;		/* true if -n flag: don't convert addrs */
105 int		sortidle;	/* sort bu idle time */
106 char	       *sel_user;	/* login of particular user selected */
107 char		domain[MAXHOSTNAMELEN + 1];
108 int maxname = 8, maxline = 3, maxhost = 16;
109 
110 /*
111  * One of these per active utmp entry.
112  */
113 struct	entry {
114 	struct	entry *next;
115 	char name[65];
116 	char line[65];
117 	char host[257];
118 	struct timeval tv;
119 	dev_t	tdev;			/* dev_t of terminal */
120 	time_t	idle;			/* idle time of terminal in seconds */
121 	struct	kinfo_proc2 *kp;	/* `most interesting' proc */
122 #ifdef SUPPORT_FTPD_UTMP
123 	pid_t	ftpd_pid;		/* pid as extracted from ftpd's entry */
124 #endif
125 } *ep, *ehead = NULL, **nextp = &ehead;
126 
127 static void	 pr_args(struct kinfo_proc2 *);
128 static void	 pr_header(time_t *, int);
129 #if defined(SUPPORT_UTMP) || defined(SUPPORT_UTMPX)
130 static struct stat *ttystat(char *);
131 static void	process(struct entry *);
132 #endif
133 static void	 usage(int);
134 int	main(int, char **);
135 
136 int
137 main(int argc, char **argv)
138 {
139 	struct kinfo_proc2 *kp;
140 	struct hostent *hp;
141 	struct in_addr l;
142 	int ch, i, nentries, nusers, wcmd;
143 	char *memf, *nlistf, *p, *x;
144 	time_t then;
145 #ifdef SUPPORT_UTMP
146 	struct utmp *ut;
147 #endif
148 #ifdef SUPPORT_UTMPX
149 	struct utmpx *utx;
150 #endif
151 	const char *progname;
152 	char buf[MAXHOSTNAMELEN], errbuf[_POSIX2_LINE_MAX];
153 
154 	/* Are we w(1) or uptime(1)? */
155 	progname = getprogname();
156 	if (*progname == '-')
157 		progname++;
158 	if (*progname == 'u') {
159 		wcmd = 0;
160 		p = "";
161 	} else {
162 		wcmd = 1;
163 		p = "hiflM:N:nsuw";
164 	}
165 
166 	memf = nlistf = NULL;
167 	while ((ch = getopt(argc, argv, p)) != -1)
168 		switch (ch) {
169 		case 'h':
170 			header = 0;
171 			break;
172 		case 'i':
173 			sortidle = 1;
174 			break;
175 		case 'M':
176 			header = 0;
177 			memf = optarg;
178 			break;
179 		case 'N':
180 			nlistf = optarg;
181 			break;
182 		case 'n':
183 			nflag = 1;
184 			break;
185 		case 'f': case 'l': case 's': case 'u': case 'w':
186 			warnx("[-flsuw] no longer supported");
187 			/* FALLTHROUGH */
188 		case '?':
189 		default:
190 			usage(wcmd);
191 		}
192 	argc -= optind;
193 	argv += optind;
194 
195 	if ((kd = kvm_openfiles(nlistf, memf, NULL,
196 	    memf == NULL ? KVM_NO_FILES : O_RDONLY, errbuf)) == NULL)
197 		errx(1, "%s", errbuf);
198 
199 	(void)time(&now);
200 
201 #ifdef SUPPORT_UTMPX
202 	setutxent();
203 #endif
204 #ifdef SUPPORT_UTMP
205 	setutent();
206 #endif
207 
208 	if (*argv)
209 		sel_user = *argv;
210 
211 	nusers = 0;
212 #ifdef SUPPORT_UTMPX
213 	while ((utx = getutxent()) != NULL) {
214 		if (utx->ut_type != USER_PROCESS)
215 			continue;
216 		++nusers;
217 		if (sel_user &&
218 		    strncmp(utx->ut_name, sel_user, sizeof(utx->ut_name) != 0))
219 			continue;
220 		if ((ep = calloc(1, sizeof(struct entry))) == NULL)
221 			err(1, NULL);
222 		(void)memcpy(ep->name, utx->ut_name, sizeof(utx->ut_name));
223 		(void)memcpy(ep->line, utx->ut_line, sizeof(utx->ut_line));
224 		ep->name[sizeof(utx->ut_name)] = '\0';
225 		ep->line[sizeof(utx->ut_line)] = '\0';
226 		if (!nflag || getnameinfo((struct sockaddr *)&utx->ut_ss,
227 		    utx->ut_ss.ss_len, ep->host, sizeof(ep->host), NULL, 0,
228 		    NI_NUMERICHOST) != 0) {
229 			(void)memcpy(ep->host, utx->ut_host,
230 			    sizeof(utx->ut_host));
231 			ep->host[sizeof(utx->ut_host)] = '\0';
232 		}
233 		ep->tv = utx->ut_tv;
234 		*nextp = ep;
235 		nextp = &(ep->next);
236 		if (wcmd != 0)
237 			process(ep);
238 	}
239 #endif
240 
241 #ifdef SUPPORT_UTMP
242 	while ((ut = getutent()) != NULL) {
243 		if (ut->ut_name[0] == '\0')
244 			continue;
245 
246 		if (sel_user &&
247 		    strncmp(ut->ut_name, sel_user, sizeof(ut->ut_name) != 0))
248 			continue;
249 
250 		/* Don't process entries that we have utmpx for */
251 		for (ep = ehead; ep != NULL; ep = ep->next) {
252 			if (strncmp(ep->line, ut->ut_line,
253 			    sizeof(ut->ut_line)) == 0)
254 				break;
255 		}
256 		if (ep != NULL)
257 			continue;
258 
259 		++nusers;
260 		if ((ep = calloc(1, sizeof(struct entry))) == NULL)
261 			err(1, NULL);
262 		(void)memcpy(ep->name, ut->ut_name, sizeof(ut->ut_name));
263 		(void)memcpy(ep->line, ut->ut_line, sizeof(ut->ut_line));
264 		(void)memcpy(ep->host, ut->ut_host, sizeof(ut->ut_host));
265 		ep->name[sizeof(ut->ut_name)] = '\0';
266 		ep->line[sizeof(ut->ut_line)] = '\0';
267 		ep->host[sizeof(ut->ut_host)] = '\0';
268 		ep->tv.tv_sec = ut->ut_time;
269 		ep->tv.tv_usec = 0;
270 		*nextp = ep;
271 		nextp = &(ep->next);
272 		if (wcmd != 0)
273 			process(ep);
274 	}
275 #endif
276 
277 #ifdef SUPPORT_UTMPX
278 	endutxent();
279 #endif
280 #ifdef SUPPORT_UTMP
281 	endutent();
282 #endif
283 
284 	if (header || wcmd == 0) {
285 		pr_header(&now, nusers);
286 		if (wcmd == 0)
287 			exit (0);
288 	}
289 
290 	if ((kp = kvm_getproc2(kd, KERN_PROC_ALL, 0,
291 	    sizeof(struct kinfo_proc2), &nentries)) == NULL)
292 		errx(1, "%s", kvm_geterr(kd));
293 
294 	/* Include trailing space because TTY header starts one column early. */
295 	for (i = 0; i < nentries; i++, kp++) {
296 
297 		if (kp->p_stat == SIDL || kp->p_stat == SZOMB)
298 			continue;
299 
300 		for (ep = ehead; ep != NULL; ep = ep->next) {
301 			if (ep->tdev == kp->p_tdev &&
302 			    kp->p__pgid == kp->p_tpgid) {
303 				/*
304 				 * Proc is in foreground of this terminal
305 				 */
306 				if (proc_compare(ep->kp, kp)) {
307 					ep->kp = kp;
308 				}
309 				break;
310 			}
311 #ifdef SUPPORT_FTPD_UTMP
312 			/*
313 			 * Hack to match process to ftp utmp entry.
314 			 */
315 			else if (ep->tdev == 0 && kp->p_tdev == NODEV &&
316 			    ep->ftpd_pid == kp->p_pid) {
317 				ep->kp = kp;
318 			}
319 #endif /* SUPPORT_FTPD_UTMP */
320 		}
321 	}
322 
323 	argwidth = printf("%-*sTTY %-*s %*s  IDLE WHAT\n",
324 	    maxname, "USER", maxhost, "FROM",
325 	    7 /* "dddhhXm" */, "LOGIN@");
326 	argwidth -= sizeof("WHAT\n") - 1 /* NUL */;
327 
328 	if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == -1 &&
329 	    ioctl(STDERR_FILENO, TIOCGWINSZ, &ws) == -1 &&
330 	    ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) || ws.ws_col == 0)
331 		ttywidth = 79;
332 	else
333 		ttywidth = ws.ws_col - 1;
334 	argwidth = ttywidth - argwidth;
335 	if (argwidth < 4)
336 		argwidth = 8;
337 	/* sort by idle time */
338 	if (sortidle && ehead != NULL) {
339 		struct entry *from = ehead, *save;
340 
341 		ehead = NULL;
342 		while (from != NULL) {
343 			for (nextp = &ehead;
344 			    (*nextp) && from->idle >= (*nextp)->idle;
345 			    nextp = &(*nextp)->next)
346 				continue;
347 			save = from;
348 			from = from->next;
349 			save->next = *nextp;
350 			*nextp = save;
351 		}
352 	}
353 #if defined(SUPPORT_UTMP) && defined(SUPPORT_UTMPX)
354 	else if (ehead != NULL) {
355 		struct entry *from = ehead, *save;
356 
357 		ehead = NULL;
358 		while (from != NULL) {
359 			for (nextp = &ehead;
360 			    (*nextp) && strcmp(from->line, (*nextp)->line) > 0;
361 			    nextp = &(*nextp)->next)
362 				continue;
363 			save = from;
364 			from = from->next;
365 			save->next = *nextp;
366 			*nextp = save;
367 		}
368 	}
369 #endif
370 
371 	if (!nflag) {
372 		int	rv;
373 
374 		rv = gethostname(domain, sizeof(domain));
375 		domain[sizeof(domain) - 1] = '\0';
376 		if (rv < 0 || (p = strchr(domain, '.')) == 0)
377 			domain[0] = '\0';
378 		else
379 			memmove(domain, p, strlen(p) + 1);
380 	}
381 
382 	for (ep = ehead; ep != NULL; ep = ep->next) {
383 		char host_buf[MAXHOSTNAMELEN + 1];
384 
385 		host_buf[MAXHOSTNAMELEN] = '\0';
386 		strncpy(host_buf, ep->host, MAXHOSTNAMELEN);
387 		p = *host_buf ? host_buf : "-";
388 
389 		for (x = p; x < p + MAXHOSTNAMELEN; x++)
390 			if (*x == '\0' || *x == ':')
391 				break;
392 		if (x == p + MAXHOSTNAMELEN || *x != ':')
393 			x = NULL;
394 		else
395 			*x++ = '\0';
396 
397 		if (!nflag && inet_aton(p, &l) &&
398 		    (hp = gethostbyaddr((char *)&l, sizeof(l), AF_INET))) {
399 			if (domain[0] != '\0') {
400 				p = hp->h_name;
401 				p += strlen(hp->h_name);
402 				p -= strlen(domain);
403 				if (p > hp->h_name &&
404 				    strcasecmp(p, domain) == 0)
405 					*p = '\0';
406 			}
407 			p = hp->h_name;
408 		}
409 		if (x) {
410 			(void)snprintf(buf, sizeof(buf), "%s:%s", p, x);
411 			p = buf;
412 		}
413 		if (ep->kp == NULL) {
414 			warnx("Stale utmp entry: %s %s %s",
415 			    ep->name, ep->line, ep->host);
416 			continue;
417 		}
418 		(void)printf("%-*s %-2.2s %-*.*s ",
419 		    maxname, ep->kp->p_login,
420 		    (strncmp(ep->line, "tty", 3) &&
421 		    strncmp(ep->line, "dty", 3)) ?
422 		    ep->line : ep->line + 3,
423 		    maxhost, maxhost, *p ? p : "-");
424 		then = (time_t)ep->tv.tv_sec;
425 		pr_attime(&then, &now);
426 		pr_idle(ep->idle);
427 		pr_args(ep->kp);
428 		(void)printf("\n");
429 	}
430 	exit(0);
431 }
432 
433 static void
434 pr_args(struct kinfo_proc2 *kp)
435 {
436 	char **argv;
437 	int left;
438 
439 	if (kp == 0)
440 		goto nothing;
441 	left = argwidth;
442 	argv = kvm_getargv2(kd, kp, argwidth);
443 	if (argv == 0)
444 		goto nothing;
445 	while (*argv) {
446 		fmt_puts(*argv, &left);
447 		argv++;
448 		fmt_putc(' ', &left);
449 	}
450 	return;
451 nothing:
452 	putchar('-');
453 }
454 
455 static void
456 pr_header(time_t *nowp, int nusers)
457 {
458 	double avenrun[3];
459 	time_t uptime;
460 	int days, hrs, i, mins;
461 	int mib[2];
462 	size_t size;
463 	char buf[256];
464 
465 	/*
466 	 * Print time of day.
467 	 *
468 	 * SCCS forces the string manipulation below, as it replaces
469 	 * %, M, and % in a character string with the file name.
470 	 */
471 	(void)strftime(buf, sizeof(buf), "%l:%" "M%p", localtime(nowp));
472 	buf[sizeof(buf) - 1] = '\0';
473 	(void)printf("%s ", buf);
474 
475 	/*
476 	 * Print how long system has been up.
477 	 * (Found by looking getting "boottime" from the kernel)
478 	 */
479 	mib[0] = CTL_KERN;
480 	mib[1] = KERN_BOOTTIME;
481 	size = sizeof(boottime);
482 	if (sysctl(mib, 2, &boottime, &size, NULL, 0) != -1 &&
483 	    boottime.tv_sec != 0) {
484 		uptime = now - boottime.tv_sec;
485 		uptime += 30;
486 		if (uptime > SECSPERMIN) {
487 			days = uptime / SECSPERDAY;
488 			uptime %= SECSPERDAY;
489 			hrs = uptime / SECSPERHOUR;
490 			uptime %= SECSPERHOUR;
491 			mins = uptime / SECSPERMIN;
492 			(void)printf(" up");
493 			if (days > 0)
494 				(void)printf(" %d day%s,", days,
495 				    days > 1 ? "s" : "");
496 			if (hrs > 0 && mins > 0)
497 				(void)printf(" %2d:%02d,", hrs, mins);
498 			else {
499 				if (hrs > 0)
500 					(void)printf(" %d hr%s,",
501 					    hrs, hrs > 1 ? "s" : "");
502 				if (mins > 0)
503 					(void)printf(" %d min%s,",
504 					    mins, mins > 1 ? "s" : "");
505 			}
506 		}
507 	}
508 
509 	/* Print number of users logged in to system */
510 	(void)printf(" %d user%s", nusers, nusers != 1 ? "s" : "");
511 
512 	/*
513 	 * Print 1, 5, and 15 minute load averages.
514 	 */
515 	if (getloadavg(avenrun, sizeof(avenrun) / sizeof(avenrun[0])) == -1)
516 		(void)printf(", no load average information available\n");
517 	else {
518 		(void)printf(", load averages:");
519 		for (i = 0; i < (sizeof(avenrun) / sizeof(avenrun[0])); i++) {
520 			if (i > 0)
521 				(void)printf(",");
522 			(void)printf(" %.2f", avenrun[i]);
523 		}
524 		(void)printf("\n");
525 	}
526 }
527 
528 #if defined(SUPPORT_UTMP) || defined(SUPPORT_UTMPX)
529 static struct stat *
530 ttystat(char *line)
531 {
532 	static struct stat sb;
533 	char ttybuf[MAXPATHLEN];
534 
535 	(void)snprintf(ttybuf, sizeof(ttybuf), "%s%s", _PATH_DEV, line);
536 	if (stat(ttybuf, &sb))
537 		return (NULL);
538 	return (&sb);
539 }
540 
541 static void
542 process(struct entry *ep)
543 {
544 	struct stat *stp;
545 	time_t touched;
546 	int max;
547 
548 	if ((max = strlen(ep->name)) > maxname)
549 		maxname = max;
550 	if ((max = strlen(ep->line)) > maxline)
551 		maxline = max;
552 	if ((max = strlen(ep->host)) > maxhost)
553 		maxhost = max;
554 
555 	if ((stp = ttystat(ep->line)) == NULL)
556 		return;
557 
558 #ifdef SUPPORT_FTPD_UTMP
559 	/*
560 	 * Hack to recognize and correctly parse
561 	 * ut entry made by ftpd. The "tty" used
562 	 * by ftpd is not a real tty, just identifier in
563 	 * form ftpSUPPORT_ID. Pid parsed from the "tty name"
564 	 * is used later to match corresponding process.
565 	 */
566 	if (strncmp(ep->line, "ftp", 3) == 0) {
567 		ep->ftpd_pid = strtol(ep->line + 3, NULL, 10);
568 
569 		return;
570 	}
571 #endif /* SUPPORT_FTPD_UTMP */
572 
573 	ep->tdev = stp->st_rdev;
574 	/*
575 	 * If this is the console device, attempt to ascertain
576 	 * the true console device dev_t.
577 	 */
578 	if (ep->tdev == 0) {
579 		int mib[2];
580 		size_t size;
581 
582 		mib[0] = CTL_KERN;
583 		mib[1] = KERN_CONSDEV;
584 		size = sizeof(dev_t);
585 		(void) sysctl(mib, 2, &ep->tdev, &size, NULL, 0);
586 	}
587 
588 	touched = stp->st_atime;
589 	if (touched < ep->tv.tv_sec) {
590 		/* tty untouched since before login */
591 		touched = ep->tv.tv_sec;
592 	}
593 	if ((ep->idle = now - touched) < 0)
594 		ep->idle = 0;
595 }
596 #endif
597 
598 static void
599 usage(int wcmd)
600 {
601 
602 	if (wcmd)
603 		(void)fprintf(stderr,
604 		    "usage: w: [-hin] [-M core] [-N system] [user]\n");
605 	else
606 		(void)fprintf(stderr, "uptime\n");
607 	exit(1);
608 }
609