/*- * Copyright (c) 1980, 1991, 1993, 1994 * The Regents of the University of California. All rights reserved. * Portions copyright (c) 2007 Apple Inc. 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. 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 #ifndef __APPLE__ __FBSDID("$FreeBSD: src/usr.bin/w/w.c,v 1.58 2005/06/04 23:40:09 gad Exp $"); #endif #ifndef lint static const char copyright[] = "@(#) Copyright (c) 1980, 1991, 1993, 1994\n\ The Regents of the University of California. All rights reserved.\n"; #endif #ifndef lint static const char sccsid[] = "@(#)w.c 8.4 (Berkeley) 4/16/94"; #endif /* * w - print system status (who and what) * * This program is similar to the systat command on Tenex/Tops 10/20 * */ #include #include #include #include #include #include #include #include #include #ifndef __APPLE__ #include #endif #include #include #include #include #include #include #include #if HAVE_KVM #include #endif #include #include #include #include #include #include #include #include #include #include #include #include #include #if HAVE_UTMPX #include /* use utmp values so formatting is the same */ #define UT_NAMESIZE 8 #define UT_LINESIZE 8 #else /* HAVE_UTMPX */ #include #endif /* HAVE_UTMPX */ #include #include #include "extern.h" struct timeval boottime; #if !HAVE_UTMPX struct utmp utmp; #endif struct winsize ws; #if HAVE_KVM kvm_t *kd; #endif time_t now; /* the current time of day */ int ttywidth; /* width of tty */ int argwidth; /* width of tty */ int header = 1; /* true if -h flag: don't print heading */ #if !HAVE_UTMPX int nflag; /* true if -n flag: don't convert addrs */ #endif #ifndef __APPLE__ int dflag; /* true if -d flag: output debug info */ #endif int sortidle; /* sort by idle time */ int use_ampm; /* use AM/PM time */ int use_comma; /* use comma as floats separator */ char **sel_users; /* login array of particular users selected */ /* * One of these per active utmp entry. */ struct entry { struct entry *next; #if HAVE_UTMPX struct utmpx utmp; #else struct utmp utmp; #endif dev_t tdev; /* dev_t of terminal */ time_t idle; /* idle time of terminal in seconds */ struct kinfo_proc *kp; /* `most interesting' proc */ char *args; /* arg list of interesting process */ struct kinfo_proc *dkp; /* debug option proc list */ } *ep, *ehead = NULL, **nextp = &ehead; #ifndef __APPLE__ #define debugproc(p) *((struct kinfo_proc **)&(p)->ki_udata) #else #define debugproc(p) *((struct kinfo_proc **)&(p)->ki_spare[0]) #endif /* W_DISPHOSTSIZE should not be greater than UT_HOSTSIZE */ #define W_DISPHOSTSIZE 16 static void pr_header(time_t *, int); static struct stat *ttystat(char *, int); static void usage(int); static int this_is_uptime(const char *s); #if !HAVE_KVM static void w_getargv(void); #endif char *fmt_argv(char **, char *, int); /* ../../bin/ps/fmt.c */ int main(int argc, char *argv[]) { struct kinfo_proc *kp; struct kinfo_proc *kprocbuf; struct kinfo_proc *dkp; struct stat *stp; #if HAVE_UTMPX struct utmpx *ux; #else FILE *ut; #endif time_t touched; #if HAVE_KVM int ch, i, nentries, nusers, wcmd, longidle, dropgid; const char *memf, *nlistf, *p; #else int ch, i, nentries, nusers, wcmd, longidle; const char *p; #endif /* HAVE_KVM */ char *x_suffix; #ifdef __APPLE__ char buf[MAXHOSTNAMELEN]; #else char buf[MAXHOSTNAMELEN], errbuf[_POSIX2_LINE_MAX]; char fn[MAXHOSTNAMELEN]; #endif /* __APPLE__ */ char *dot; #if !HAVE_KVM int local_error = 0, retry_count = 0; size_t bufSize = 0; size_t orig_bufSize = 0; int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_ALL, 0 }; #endif (void)setlocale(LC_ALL, ""); #ifndef __APPLE__ use_ampm = (*nl_langinfo(T_FMT_AMPM) != '\0'); use_comma = (*nl_langinfo(RADIXCHAR) != ','); #endif /* Are we w(1) or uptime(1)? */ if (this_is_uptime(argv[0]) == 0) { wcmd = 0; p = ""; } else { wcmd = 1; p = "dhiflM:N:nsuw"; } #if HAVE_KVM dropgid = 0; memf = nlistf = _PATH_DEVNULL; #endif while ((ch = getopt(argc, argv, p)) != -1) switch (ch) { #ifndef __APPLE__ case 'd': dflag = 1; break; #endif case 'h': header = 0; break; case 'i': sortidle = 1; break; #if HAVE_KVM case 'M': header = 0; memf = optarg; dropgid = 1; break; case 'N': nlistf = optarg; dropgid = 1; break; #endif /* HAVE_KVM */ #if !HAVE_UTMPX case 'n': nflag = 1; break; #else /* !HAVE_UTMPX */ case 'n': #endif /* !HAVE_UTMPX */ case 'f': case 'l': case 's': case 'u': case 'w': #if !HAVE_KVM case 'M': case 'N': #endif #ifdef __APPLE__ case 'd': warnx("[-MNdflnsuw] no longer supported"); #else warnx("[-flsuw] no longer supported"); #endif /* FALLTHROUGH */ case '?': default: usage(wcmd); } argc -= optind; argv += optind; #if !TARGET_OS_EMBEDDED if (!(_res.options & RES_INIT)) res_init(); _res.retrans = 2; /* resolver timeout to 2 seconds per try */ _res.retry = 1; /* only try once.. */ #endif #if HAVE_KVM /* * Discard setgid privileges if not the running kernel so that bad * guys can't print interesting stuff from kernel memory. */ if (dropgid) setgid(getgid()); if ((kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, errbuf)) == NULL) errx(1, "%s", errbuf); #endif (void)time(&now); #if HAVE_UTMPX setutxent(); #else if ((ut = fopen(_PATH_UTMP, "r")) == NULL) err(1, "%s", _PATH_UTMP); #endif if (*argv) sel_users = argv; #if HAVE_UTMPX for (nusers = 0; (ux = getutxent());) { if (ux->ut_user[0] == '\0' || ux->ut_type != USER_PROCESS) continue; if (!(stp = ttystat(ux->ut_line, sizeof(ux->ut_line)))) continue; /* corrupted record */ #else for (nusers = 0; fread(&utmp, sizeof(utmp), 1, ut);) { if (utmp.ut_name[0] == '\0') continue; if (!(stp = ttystat(utmp.ut_line, UT_LINESIZE))) continue; /* corrupted record */ #endif ++nusers; if (wcmd == 0) continue; if (sel_users) { int usermatch; char **user; usermatch = 0; for (user = sel_users; !usermatch && *user; user++) #if HAVE_UTMPX if (!strncmp(ux->ut_user, *user, sizeof(ux->ut_user))) #else if (!strncmp(utmp.ut_name, *user, UT_NAMESIZE)) #endif usermatch = 1; if (!usermatch) continue; } if ((ep = calloc(1, sizeof(struct entry))) == NULL) errx(1, "calloc"); *nextp = ep; nextp = &ep->next; #if HAVE_UTMPX memmove(&ep->utmp, ux, sizeof(*ux)); #else memmove(&ep->utmp, &utmp, sizeof(struct utmp)); #endif ep->tdev = stp->st_rdev; #ifdef CPU_CONSDEV /* * If this is the console device, attempt to ascertain * the true console device dev_t. */ if (ep->tdev == 0) { int mib[2]; size_t size; mib[0] = CTL_MACHDEP; mib[1] = CPU_CONSDEV; size = sizeof(dev_t); (void)sysctl(mib, 2, &ep->tdev, &size, NULL, 0); } #endif touched = stp->st_atime; #ifdef __APPLE__ if (touched < ep->utmp.ut_tv.tv_sec) { /* tty untouched since before login */ touched = ep->utmp.ut_tv.tv_sec; } #else if (touched < ep->utmp.ut_time) { /* tty untouched since before login */ touched = ep->utmp.ut_time; } #endif if ((ep->idle = now - touched) < 0) ep->idle = 0; } #if HAVE_UTMPX endutxent(); #else (void)fclose(ut); #endif if (header || wcmd == 0) { pr_header(&now, nusers); if (wcmd == 0) { #if HAVE_KVM (void)kvm_close(kd); #endif exit(0); } #define HEADER_USER "USER" #define HEADER_TTY "TTY" #define HEADER_FROM "FROM" #define HEADER_LOGIN_IDLE "LOGIN@ IDLE " #define HEADER_WHAT "WHAT\n" #define WUSED (UT_NAMESIZE + UT_LINESIZE + W_DISPHOSTSIZE + \ sizeof(HEADER_LOGIN_IDLE) + 3) /* header width incl. spaces */ (void)printf("%-*.*s %-*.*s %-*.*s %s", UT_NAMESIZE, UT_NAMESIZE, HEADER_USER, UT_LINESIZE, UT_LINESIZE, HEADER_TTY, W_DISPHOSTSIZE, W_DISPHOSTSIZE, HEADER_FROM, HEADER_LOGIN_IDLE HEADER_WHAT); } #if HAVE_KVM if ((kp = kvm_getprocs(kd, KERN_PROC_ALL, 0, &nentries)) == NULL) err(1, "%s", kvm_geterr(kd)); #else mib[0] = CTL_KERN; mib[1] = KERN_PROC; mib[2] = KERN_PROC_ALL; mib[3] = 0; if (sysctl(mib, 4, NULL, &bufSize, NULL, 0) < 0) { perror("Failure calling sysctl"); exit(1); } kprocbuf = kp = (struct kinfo_proc *)malloc(bufSize); retry_count = 0; orig_bufSize = bufSize; for (retry_count = 0; ; retry_count++) { local_error = 0; bufSize = orig_bufSize; if ((local_error = sysctl(mib, 4, kp, &bufSize, NULL, 0)) < 0) { if (retry_count < 1000) { sleep(1); continue; } perror("Failure calling sysctl"); exit(1); } else if (local_error == 0) { break; } sleep(1); } nentries = bufSize / sizeof(struct kinfo_proc); #endif /* !HAVE_KVM */ #if !HAVE_KVM #define ki_stat kp_proc.p_stat #define ki_pgid kp_eproc.e_pgid #define ki_tpgid kp_eproc.e_tpgid #define ki_tdev kp_eproc.e_tdev #endif /* !HAVE_KVM */ for (i = 0; i < nentries; i++, kp++) { if (kp->ki_stat == SIDL || kp->ki_stat == SZOMB) continue; for (ep = ehead; ep != NULL; ep = ep->next) { if (ep->tdev == kp->ki_tdev) { /* * proc is associated with this terminal */ if (ep->kp == NULL && kp->ki_pgid == kp->ki_tpgid) { /* * Proc is 'most interesting' */ if (proc_compare(ep->kp, kp)) ep->kp = kp; } /* * Proc debug option info; add to debug * list using kinfo_proc ki_spare[0] * as next pointer; ptr to ptr avoids the * ptr = long assumption. */ dkp = ep->dkp; ep->dkp = kp; #ifndef __APPLE__ debugproc(kp) = dkp; #endif } } } if ((ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == -1 && ioctl(STDERR_FILENO, TIOCGWINSZ, &ws) == -1 && ioctl(STDIN_FILENO, TIOCGWINSZ, &ws) == -1) || ws.ws_col == 0) ttywidth = 79; else ttywidth = ws.ws_col - 1; argwidth = ttywidth - WUSED; if (argwidth < 4) argwidth = 8; for (ep = ehead; ep != NULL; ep = ep->next) { if (ep->kp == NULL) { ep->args = strdup("-"); continue; } #if HAVE_KVM ep->args = fmt_argv(kvm_getargv(kd, ep->kp, argwidth), ep->kp->ki_comm, MAXCOMLEN); #else w_getargv(); #endif /* HAVE_KVM */ if (ep->args == NULL) err(1, NULL); } /* sort by idle time */ if (sortidle && ehead != NULL) { struct entry *from, *save; from = ehead; ehead = NULL; while (from != NULL) { for (nextp = &ehead; (*nextp) && from->idle >= (*nextp)->idle; nextp = &(*nextp)->next) continue; save = from; from = from->next; save->next = *nextp; *nextp = save; } } for (ep = ehead; ep != NULL; ep = ep->next) { #if HAVE_UTMPX char host_buf[sizeof(ep->utmp.ut_host) + 1]; strlcpy(host_buf, ep->utmp.ut_host, sizeof(host_buf)); #else char host_buf[UT_HOSTSIZE + 1]; struct sockaddr_storage ss; struct sockaddr *sa = (struct sockaddr *)&ss; struct sockaddr_in *lsin = (struct sockaddr_in *)&ss; struct sockaddr_in6 *lsin6 = (struct sockaddr_in6 *)&ss; time_t t; int isaddr; host_buf[UT_HOSTSIZE] = '\0'; strncpy(host_buf, ep->utmp.ut_host, UT_HOSTSIZE); #endif /* HAVE_UTMPX */ p = *host_buf ? host_buf : "-"; if ((x_suffix = strrchr(p, ':')) != NULL) { if ((dot = strchr(x_suffix, '.')) != NULL && strchr(dot+1, '.') == NULL) *x_suffix++ = '\0'; else x_suffix = NULL; } #if !HAVE_UTMPX if (!nflag) { /* Attempt to change an IP address into a name */ isaddr = 0; memset(&ss, '\0', sizeof(ss)); if (inet_pton(AF_INET6, p, &lsin6->sin6_addr) == 1) { lsin6->sin6_len = sizeof(*lsin6); lsin6->sin6_family = AF_INET6; isaddr = 1; } else if (inet_pton(AF_INET, p, &lsin->sin_addr) == 1) { lsin->sin_len = sizeof(*lsin); lsin->sin_family = AF_INET; isaddr = 1; } if (isaddr && realhostname_sa(fn, sizeof(fn), sa, sa->sa_len) == HOSTNAME_FOUND) p = fn; } #endif /* !HAVE_UTMPX */ if (x_suffix) { (void)snprintf(buf, sizeof(buf), "%s:%s", p, x_suffix); p = buf; } #ifndef __APPLE__ if (dflag) { for (dkp = ep->dkp; dkp != NULL; dkp = debugproc(dkp)) { const char *ptr; ptr = fmt_argv(kvm_getargv(kd, dkp, argwidth), dkp->ki_comm, MAXCOMLEN); if (ptr == NULL) ptr = "-"; (void)printf("\t\t%-9d %s\n", dkp->ki_pid, ptr); } } #endif /* !__APPLE__ */ (void)printf("%-*.*s %-*.*s %-*.*s ", #if HAVE_UTMPX UT_NAMESIZE, (int)sizeof(ep->utmp.ut_user), ep->utmp.ut_user, UT_LINESIZE, (int)sizeof(ep->utmp.ut_line), #else UT_NAMESIZE, UT_NAMESIZE, ep->utmp.ut_name, UT_LINESIZE, UT_LINESIZE, #endif strncmp(ep->utmp.ut_line, "tty", 3) && strncmp(ep->utmp.ut_line, "cua", 3) ? ep->utmp.ut_line : ep->utmp.ut_line + 3, W_DISPHOSTSIZE, W_DISPHOSTSIZE, *p ? p : "-"); #ifdef __APPLE__ pr_attime(&ep->utmp.ut_tv.tv_sec, &now); #else t = _time_to_time32(ep->utmp.ut_time); pr_attime(&t, &now); #endif longidle = pr_idle(ep->idle); (void)printf("%.*s\n", argwidth - longidle, ep->args); #ifdef __APPLE__ free(ep->args); #endif } #if HAVE_KVM (void)kvm_close(kd); #else free(kprocbuf); #endif /* HAVE_KVM */ exit(0); } static void pr_header(time_t *nowp, int nusers) { double avenrun[3]; time_t uptime; int days, hrs, i, mins, secs; int mib[2]; size_t size; char buf[256]; /* * Print time of day. */ if (strftime(buf, sizeof(buf), use_ampm ? "%l:%M%p" : "%k:%M", localtime(nowp)) != 0) (void)printf("%s ", buf); /* * Print how long system has been up. * (Found by looking getting "boottime" from the kernel) */ mib[0] = CTL_KERN; mib[1] = KERN_BOOTTIME; size = sizeof(boottime); if (sysctl(mib, 2, &boottime, &size, NULL, 0) != -1 && boottime.tv_sec != 0) { uptime = now - boottime.tv_sec; if (uptime > 60) uptime += 30; days = uptime / 86400; uptime %= 86400; hrs = uptime / 3600; uptime %= 3600; mins = uptime / 60; secs = uptime % 60; (void)printf(" up"); if (days > 0) (void)printf(" %d day%s,", days, days > 1 ? "s" : ""); if (hrs > 0 && mins > 0) (void)printf(" %2d:%02d,", hrs, mins); else if (hrs > 0) (void)printf(" %d hr%s,", hrs, hrs > 1 ? "s" : ""); else if (mins > 0) (void)printf(" %d min%s,", mins, mins > 1 ? "s" : ""); else (void)printf(" %d sec%s,", secs, secs > 1 ? "s" : ""); } /* Print number of users logged in to system */ (void)printf(" %d user%s", nusers, nusers == 1 ? "" : "s"); /* * Print 1, 5, and 15 minute load averages. */ if (getloadavg(avenrun, sizeof(avenrun) / sizeof(avenrun[0])) == -1) (void)printf(", no load average information available\n"); else { (void)printf(", load averages:"); for (i = 0; i < (int)(sizeof(avenrun) / sizeof(avenrun[0])); i++) { if (use_comma && i > 0) (void)printf(","); (void)printf(" %.2f", avenrun[i]); } (void)printf("\n"); } } static struct stat * ttystat(char *line, int sz) { static struct stat sb; char ttybuf[MAXPATHLEN]; (void)snprintf(ttybuf, sizeof(ttybuf), "%s%.*s", _PATH_DEV, sz, line); if (stat(ttybuf, &sb) == 0) { return (&sb); } else { warn("%s", ttybuf); return (NULL); } } static void usage(int wcmd) { if (wcmd) (void)fprintf(stderr, "usage: w [hi] [user ...]\n"); else (void)fprintf(stderr, "usage: uptime\n"); exit(1); } static int this_is_uptime(const char *s) { const char *u; if ((u = strrchr(s, '/')) != NULL) ++u; else u = s; if (strcmp(u, "uptime") == 0) return (0); return (-1); } #if !HAVE_KVM static void w_getargv(void) { int mib[3], argmax; size_t size; char *procargs, *sp, *np, *cp; mib[0] = CTL_KERN; mib[1] = KERN_ARGMAX; size = sizeof(argmax); if (sysctl(mib, 2, &argmax, &size, NULL, 0) == -1) { goto ERROR; } procargs = malloc(argmax); if (procargs == NULL) { goto ERROR; } mib[0] = CTL_KERN; mib[1] = KERN_PROCARGS; mib[2] = KI_PROC(ep)->p_pid; size = (size_t)argmax; if (sysctl(mib, 3, procargs, &size, NULL, 0) == -1) { goto ERROR_FREE; } for (cp = procargs; cp < &procargs[size]; cp++) { if (*cp == '\0') { break; } } if (cp == &procargs[size]) { goto ERROR_FREE; } sp = cp; for (np = NULL; cp < &procargs[size]; cp++) { if (*cp == '\0') { if (np != NULL) { *np = ' '; } np = cp; } else if (*cp == '=') { break; } } for (np = sp; (np < &procargs[size]) && (*np == ' '); np++); ep->args = strdup(np); free(procargs); return; ERROR_FREE: free(procargs); ERROR: /* ep->args = malloc(2); ep->args[0] = '-'; ep->args[1] = '\0'; */ asprintf(&ep->args, "%s", KI_PROC(ep)->p_comm); return; } #endif /* HAVE_KVM */