w.c revision 1.50 1 /* $NetBSD: w.c,v 1.50 2002/08/22 14:52:55 christos 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.50 2002/08/22 14:52:55 christos 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 if (wcmd) {
201 #ifdef SUPPORT_UTMP
202 setutent();
203 #endif
204 #ifdef SUPPORT_UTMP
205 setutxent();
206 #endif
207 }
208
209 if (*argv)
210 sel_user = *argv;
211
212 nusers = 0;
213 #ifdef SUPPORT_UTMPX
214 while ((utx = getutxent()) != NULL) {
215 if (utx->ut_type != USER_PROCESS)
216 continue;
217 ++nusers;
218 if (wcmd == 0 || (sel_user &&
219 strncmp(utx->ut_name, sel_user, sizeof(utx->ut_name) != 0)))
220 continue;
221 if ((ep = calloc(1, sizeof(struct entry))) == NULL)
222 err(1, NULL);
223 (void)memcpy(ep->name, utx->ut_name, sizeof(utx->ut_name));
224 (void)memcpy(ep->line, utx->ut_line, sizeof(utx->ut_line));
225 (void)memcpy(ep->host, utx->ut_host, sizeof(utx->ut_host));
226 ep->name[sizeof(utx->ut_name)] = '\0';
227 ep->line[sizeof(utx->ut_line)] = '\0';
228 ep->host[sizeof(utx->ut_host)] = '\0';
229 ep->tv = utx->ut_tv;
230 *nextp = ep;
231 nextp = &(ep->next);
232 process(ep);
233 }
234 #endif
235
236 #ifdef SUPPORT_UTMP
237 while ((ut = getutent()) != NULL) {
238 if (ut->ut_name[0] == '\0')
239 continue;
240 if (wcmd == 0 || (sel_user &&
241 strncmp(ut->ut_name, sel_user, sizeof(ut->ut_name) != 0)))
242 continue;
243
244 /* Don't process entries that we have utmpx for */
245 for (ep = ehead; ep != NULL; ep = ep->next) {
246 if (strncmp(ep->line, ut->ut_line,
247 sizeof(ut->ut_line)) == 0)
248 break;
249 }
250 if (ep != NULL)
251 continue;
252
253 ++nusers;
254 if ((ep = calloc(1, sizeof(struct entry))) == NULL)
255 err(1, NULL);
256 (void)memcpy(ep->name, ut->ut_name, sizeof(ut->ut_name));
257 (void)memcpy(ep->line, ut->ut_line, sizeof(ut->ut_line));
258 (void)memcpy(ep->host, ut->ut_host, sizeof(ut->ut_host));
259 ep->name[sizeof(ut->ut_name)] = '\0';
260 ep->line[sizeof(ut->ut_line)] = '\0';
261 ep->host[sizeof(ut->ut_host)] = '\0';
262 ep->tv.tv_sec = ut->ut_time;
263 ep->tv.tv_usec = 0;
264 *nextp = ep;
265 nextp = &(ep->next);
266 process(ep);
267 }
268 #endif
269
270 if (wcmd) {
271 #ifdef SUPPORT_UTMPX
272 endutxent();
273 #endif
274 #ifdef SUPPORT_UTMP
275 endutent();
276 #endif
277 } else
278 nusers = 1;
279
280 if (header || wcmd == 0) {
281 pr_header(&now, nusers);
282 if (wcmd == 0)
283 exit (0);
284 }
285
286 if ((kp = kvm_getproc2(kd, KERN_PROC_ALL, 0,
287 sizeof(struct kinfo_proc2), &nentries)) == NULL)
288 errx(1, "%s", kvm_geterr(kd));
289
290 /* Include trailing space because TTY header starts one column early. */
291 for (i = 0; i < nentries; i++, kp++) {
292
293 if (kp->p_stat == SIDL || kp->p_stat == SZOMB)
294 continue;
295
296 for (ep = ehead; ep != NULL; ep = ep->next) {
297 if (ep->tdev == kp->p_tdev &&
298 kp->p__pgid == kp->p_tpgid) {
299 /*
300 * Proc is in foreground of this terminal
301 */
302 if (proc_compare(ep->kp, kp)) {
303 ep->kp = kp;
304 }
305 break;
306 }
307 #ifdef SUPPORT_FTPD_UTMP
308 /*
309 * Hack to match process to ftp utmp entry.
310 */
311 else if (ep->tdev == 0 && kp->p_tdev == NODEV &&
312 ep->ftpd_pid == kp->p_pid) {
313 ep->kp = kp;
314 }
315 #endif /* SUPPORT_FTPD_UTMP */
316 }
317 }
318
319 argwidth = printf("%-*sTTY %-*s %*s IDLE WHAT\n",
320 maxname, "USER", maxhost, "FROM",
321 7 /* "dddhhXm" */, "LOGIN@");
322 argwidth -= sizeof("WHAT\n") - 1 /* NUL */;
323
324 if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == -1 &&
325 ioctl(STDERR_FILENO, TIOCGWINSZ, &ws) == -1 &&
326 ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) || ws.ws_col == 0)
327 ttywidth = 79;
328 else
329 ttywidth = ws.ws_col - 1;
330 argwidth = ttywidth - argwidth;
331 if (argwidth < 4)
332 argwidth = 8;
333 /* sort by idle time */
334 if (sortidle && ehead != NULL) {
335 struct entry *from = ehead, *save;
336
337 ehead = NULL;
338 while (from != NULL) {
339 for (nextp = &ehead;
340 (*nextp) && from->idle >= (*nextp)->idle;
341 nextp = &(*nextp)->next)
342 continue;
343 save = from;
344 from = from->next;
345 save->next = *nextp;
346 *nextp = save;
347 }
348 }
349 #if defined(SUPPORT_UTMP) && defined(SUPPORT_UTMPX)
350 else if (ehead != NULL) {
351 struct entry *from = ehead, *save;
352
353 ehead = NULL;
354 while (from != NULL) {
355 for (nextp = &ehead;
356 (*nextp) && strcmp(from->line, (*nextp)->line) > 0;
357 nextp = &(*nextp)->next)
358 continue;
359 save = from;
360 from = from->next;
361 save->next = *nextp;
362 *nextp = save;
363 }
364 }
365 #endif
366
367 if (!nflag) {
368 int rv;
369
370 rv = gethostname(domain, sizeof(domain));
371 domain[sizeof(domain) - 1] = '\0';
372 if (rv < 0 || (p = strchr(domain, '.')) == 0)
373 domain[0] = '\0';
374 else
375 memmove(domain, p, strlen(p) + 1);
376 }
377
378 for (ep = ehead; ep != NULL; ep = ep->next) {
379 char host_buf[MAXHOSTNAMELEN + 1];
380
381 host_buf[MAXHOSTNAMELEN] = '\0';
382 strncpy(host_buf, ep->host, MAXHOSTNAMELEN);
383 p = *host_buf ? host_buf : "-";
384
385 for (x = p; x < p + MAXHOSTNAMELEN; x++)
386 if (*x == '\0' || *x == ':')
387 break;
388 if (x == p + MAXHOSTNAMELEN || *x != ':')
389 x = NULL;
390 else
391 *x++ = '\0';
392
393 if (!nflag && inet_aton(p, &l) &&
394 (hp = gethostbyaddr((char *)&l, sizeof(l), AF_INET))) {
395 if (domain[0] != '\0') {
396 p = hp->h_name;
397 p += strlen(hp->h_name);
398 p -= strlen(domain);
399 if (p > hp->h_name &&
400 strcasecmp(p, domain) == 0)
401 *p = '\0';
402 }
403 p = hp->h_name;
404 }
405 if (x) {
406 (void)snprintf(buf, sizeof(buf), "%s:%s", p, x);
407 p = buf;
408 }
409 if (ep->kp == NULL) {
410 warnx("Stale utmp entry: %s %s %s",
411 ep->name, ep->line, ep->host);
412 continue;
413 }
414 (void)printf("%-*s %-2.2s %-*.*s ",
415 maxname, ep->kp->p_login,
416 (strncmp(ep->line, "tty", 3) &&
417 strncmp(ep->line, "dty", 3)) ?
418 ep->line : ep->line + 3,
419 maxhost, maxhost, *p ? p : "-");
420 then = (time_t)ep->tv.tv_sec;
421 pr_attime(&then, &now);
422 pr_idle(ep->idle);
423 pr_args(ep->kp);
424 (void)printf("\n");
425 }
426 exit(0);
427 }
428
429 static void
430 pr_args(struct kinfo_proc2 *kp)
431 {
432 char **argv;
433 int left;
434
435 if (kp == 0)
436 goto nothing;
437 left = argwidth;
438 argv = kvm_getargv2(kd, kp, argwidth);
439 if (argv == 0)
440 goto nothing;
441 while (*argv) {
442 fmt_puts(*argv, &left);
443 argv++;
444 fmt_putc(' ', &left);
445 }
446 return;
447 nothing:
448 putchar('-');
449 }
450
451 static void
452 pr_header(time_t *nowp, int nusers)
453 {
454 double avenrun[3];
455 time_t uptime;
456 int days, hrs, i, mins;
457 int mib[2];
458 size_t size;
459 char buf[256];
460
461 /*
462 * Print time of day.
463 *
464 * SCCS forces the string manipulation below, as it replaces
465 * %, M, and % in a character string with the file name.
466 */
467 (void)strftime(buf, sizeof(buf), "%l:%" "M%p", localtime(nowp));
468 buf[sizeof(buf) - 1] = '\0';
469 (void)printf("%s ", buf);
470
471 /*
472 * Print how long system has been up.
473 * (Found by looking getting "boottime" from the kernel)
474 */
475 mib[0] = CTL_KERN;
476 mib[1] = KERN_BOOTTIME;
477 size = sizeof(boottime);
478 if (sysctl(mib, 2, &boottime, &size, NULL, 0) != -1 &&
479 boottime.tv_sec != 0) {
480 uptime = now - boottime.tv_sec;
481 uptime += 30;
482 if (uptime > SECSPERMIN) {
483 days = uptime / SECSPERDAY;
484 uptime %= SECSPERDAY;
485 hrs = uptime / SECSPERHOUR;
486 uptime %= SECSPERHOUR;
487 mins = uptime / SECSPERMIN;
488 (void)printf(" up");
489 if (days > 0)
490 (void)printf(" %d day%s,", days,
491 days > 1 ? "s" : "");
492 if (hrs > 0 && mins > 0)
493 (void)printf(" %2d:%02d,", hrs, mins);
494 else {
495 if (hrs > 0)
496 (void)printf(" %d hr%s,",
497 hrs, hrs > 1 ? "s" : "");
498 if (mins > 0)
499 (void)printf(" %d min%s,",
500 mins, mins > 1 ? "s" : "");
501 }
502 }
503 }
504
505 /* Print number of users logged in to system */
506 (void)printf(" %d user%s", nusers, nusers != 1 ? "s" : "");
507
508 /*
509 * Print 1, 5, and 15 minute load averages.
510 */
511 if (getloadavg(avenrun, sizeof(avenrun) / sizeof(avenrun[0])) == -1)
512 (void)printf(", no load average information available\n");
513 else {
514 (void)printf(", load averages:");
515 for (i = 0; i < (sizeof(avenrun) / sizeof(avenrun[0])); i++) {
516 if (i > 0)
517 (void)printf(",");
518 (void)printf(" %.2f", avenrun[i]);
519 }
520 (void)printf("\n");
521 }
522 }
523
524 #if defined(SUPPORT_UTMP) || defined(SUPPORT_UTMPX)
525 static struct stat *
526 ttystat(char *line)
527 {
528 static struct stat sb;
529 char ttybuf[MAXPATHLEN];
530
531 (void)snprintf(ttybuf, sizeof(ttybuf), "%s%s", _PATH_DEV, line);
532 if (stat(ttybuf, &sb))
533 return (NULL);
534 return (&sb);
535 }
536
537 static void
538 process(struct entry *ep)
539 {
540 struct stat *stp;
541 time_t touched;
542 int max;
543
544 if ((max = strlen(ep->name)) > maxname)
545 maxname = max;
546 if ((max = strlen(ep->line)) > maxline)
547 maxline = max;
548 if ((max = strlen(ep->host)) > maxhost)
549 maxhost = max;
550
551 if ((stp = ttystat(ep->line)) == NULL)
552 return;
553
554 #ifdef SUPPORT_FTPD_UTMP
555 /*
556 * Hack to recognize and correctly parse
557 * ut entry made by ftpd. The "tty" used
558 * by ftpd is not a real tty, just identifier in
559 * form ftpSUPPORT_ID. Pid parsed from the "tty name"
560 * is used later to match corresponding process.
561 */
562 if (strncmp(ep->line, "ftp", 3) == 0) {
563 ep->ftpd_pid = strtol(ep->line + 3, NULL, 10);
564
565 return;
566 }
567 #endif /* SUPPORT_FTPD_UTMP */
568
569 ep->tdev = stp->st_rdev;
570 /*
571 * If this is the console device, attempt to ascertain
572 * the true console device dev_t.
573 */
574 if (ep->tdev == 0) {
575 int mib[2];
576 size_t size;
577
578 mib[0] = CTL_KERN;
579 mib[1] = KERN_CONSDEV;
580 size = sizeof(dev_t);
581 (void) sysctl(mib, 2, &ep->tdev, &size, NULL, 0);
582 }
583
584 touched = stp->st_atime;
585 if (touched < ep->tv.tv_sec) {
586 /* tty untouched since before login */
587 touched = ep->tv.tv_sec;
588 }
589 if ((ep->idle = now - touched) < 0)
590 ep->idle = 0;
591 }
592 #endif
593
594 static void
595 usage(int wcmd)
596 {
597
598 if (wcmd)
599 (void)fprintf(stderr,
600 "usage: w: [-hin] [-M core] [-N system] [user]\n");
601 else
602 (void)fprintf(stderr, "uptime\n");
603 exit(1);
604 }
605