]> git.saurik.com Git - apple/shell_cmds.git/blobdiff - who/who.c
shell_cmds-207.11.1.tar.gz
[apple/shell_cmds.git] / who / who.c
index f045888f13a6ba27a786536e21b74f3678fafee0..f0e356f22b246718bd138fe6826d1a8cd0c2d593 100644 (file)
--- a/who/who.c
+++ b/who/who.c
@@ -1,4 +1,4 @@
-/*     $NetBSD: who.c,v 1.6 1997/10/20 03:20:29 lukem Exp $    */
+/*     $NetBSD: who.c,v 1.23 2008/07/24 15:35:41 christos Exp $        */
 
 /*
  * Copyright (c) 1989, 1993
  * 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. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *     This product includes software developed by the University of
- *     California, Berkeley and its contributors.
- * 4. Neither the name of the University nor the names of its contributors
+ * 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.
  *
 
 #include <sys/cdefs.h>
 #ifndef lint
-__COPYRIGHT(
-"@(#) Copyright (c) 1989, 1993\n\
-       The Regents of the University of California.  All rights reserved.\n");
+__COPYRIGHT("@(#) Copyright (c) 1989, 1993\
+ The Regents of the University of California.  All rights reserved.");
 #endif /* not lint */
 
 #ifndef lint
 #if 0
 static char sccsid[] = "@(#)who.c      8.1 (Berkeley) 6/6/93";
 #endif
-__RCSID("$NetBSD: who.c,v 1.6 1997/10/20 03:20:29 lukem Exp $");
+__RCSID("$NetBSD: who.c,v 1.23 2008/07/24 15:35:41 christos Exp $");
 #endif /* not lint */
 
 #include <sys/types.h>
 #include <sys/stat.h>
+
 #include <err.h>
 #include <locale.h>
 #include <pwd.h>
@@ -60,45 +56,124 @@ __RCSID("$NetBSD: who.c,v 1.6 1997/10/20 03:20:29 lukem Exp $");
 #include <string.h>
 #include <time.h>
 #include <unistd.h>
+#ifdef SUPPORT_UTMP
 #include <utmp.h>
-
-void output __P((struct utmp *));
-void output_labels __P((void));
-void who_am_i __P((FILE *));
-FILE *file __P((char *));
-void usage __P((void));
-
-int show_term;                 /* show term state */
-int show_idle;                 /* show idle time */
-
-int main __P((int, char **));
+#endif
+#ifdef SUPPORT_UTMPX
+#include <utmpx.h>
+#endif
+#ifdef __APPLE__
+#include <limits.h>
+#include <paths.h>
+#include <stdint.h>
+#endif /* __APPLE__ */
+
+#include "utmpentry.h"
+
+#ifdef __APPLE__
+#define __UNCONST(a)   ((void *)(unsigned long)(const void *)(a))
+#endif /* __APPLE__ */
+
+static void output_labels(void);
+static void who_am_i(const char *, int);
+static void usage(void) __dead;
+static void process(const char *, int);
+static void eprint(const struct utmpentry *);
+static void print(const char *, const char *, time_t, const char *, pid_t pid,
+    uint16_t term, uint16_t xit, uint16_t sess, uint16_t type);
+static void quick(const char *);
+
+static int show_term;                  /* show term state */
+static int show_idle;                  /* show idle time */
+#ifndef __APPLE__
+static int show_details;               /* show exit status etc. */
+#endif /* !__APPLE__ */
+
+struct ut_type_names {
+  int type;
+  const char *name;
+} ut_type_names[] = {
+#ifdef SUPPORT_UTMPX
+  { EMPTY, "empty" }, 
+  { RUN_LVL, "run level" }, 
+  { BOOT_TIME, "boot time" }, 
+  { OLD_TIME, "old time" }, 
+  { NEW_TIME, "new time" }, 
+  { INIT_PROCESS, "init process" }, 
+  { LOGIN_PROCESS, "login process" }, 
+  { USER_PROCESS, "user process" }, 
+  { DEAD_PROCESS, "dead process" }, 
+#if defined(_NETBSD_SOURCE)
+  { ACCOUNTING, "accounting" }, 
+  { SIGNATURE, "signature" },
+  { DOWN_TIME, "down time" },
+#endif /* _NETBSD_SOURCE */
+#endif /* SUPPORT_UTMPX */
+  { -1, "unknown" }
+};
 
 int
-main(argc, argv)
-       int argc;
-       char **argv;
+main(int argc, char *argv[])
 {
-       struct utmp usr;
-       FILE *ufp;
-       int c, only_current_term, show_labels;
+       int c, only_current_term, show_labels, quick_mode, default_mode;
+       int et = 0;
 
        setlocale(LC_ALL, "");
 
        only_current_term = show_term = show_idle = show_labels = 0;
-       while ((c = getopt(argc, argv, "mTuH")) != -1) {
+       quick_mode = default_mode = 0;
+
+       while ((c = getopt(argc, argv, "abdHlmpqrsTtuv")) != -1) {
                switch (c) {
+               case 'a':
+                       et = -1;
+#ifdef __APPLE__
+                       show_idle = 1;
+#else /* !__APPLE__ */
+                       show_idle = show_details = 1;
+#endif /* __APPLE__ */
+                       break;
+               case 'b':
+                       et |= (1 << BOOT_TIME);
+                       break;
+               case 'd':
+                       et |= (1 << DEAD_PROCESS);
+                       break;
+               case 'H':
+                       show_labels = 1;
+                       break;
+               case 'l':
+                       et |= (1 << LOGIN_PROCESS);
+                       break;
                case 'm':
                        only_current_term = 1;
                        break;
+               case 'p':
+                       et |= (1 << INIT_PROCESS);
+                       break;
+               case 'q':
+                       quick_mode = 1;
+                       break;
+               case 'r':
+                       et |= (1 << RUN_LVL);
+                       break;
+               case 's':
+                       default_mode = 1;
+                       break;
                case 'T':
                        show_term = 1;
                        break;
+               case 't':
+                       et |= (1 << NEW_TIME);
+                       break;
                case 'u':
                        show_idle = 1;
                        break;
-               case 'H':
-                       show_labels = 1;
+#ifndef __APPLE__
+               case 'v':
+                       show_details = 1;
                        break;
+#endif /* !__APPLE__ */
                default:
                        usage();
                        /* NOTREACHED */
@@ -107,115 +182,189 @@ main(argc, argv)
        argc -= optind;
        argv += optind;
 
+       if (et != 0)
+               etype = et;
+
+#ifndef __APPLE__
        if (chdir("/dev")) {
-               err(1, "cannot change directory to /dev");
+               err(EXIT_FAILURE, "cannot change directory to /dev");
                /* NOTREACHED */
        }
+#endif /* !__APPLE__ */
 
-       if (show_labels)
-               output_labels();
+       if (default_mode)
+               only_current_term = show_term = show_idle = 0;
 
        switch (argc) {
        case 0:                                 /* who */
-               ufp = file(_PATH_UTMP);
-
-               if (only_current_term) {
-                       who_am_i(ufp);
+               if (quick_mode) {
+                       quick(NULL);
+               } else if (only_current_term) {
+                       who_am_i(NULL, show_labels);
                } else {
-                       /* only entries with both name and line fields */
-                       while (fread((char *)&usr, sizeof(usr), 1, ufp) == 1)
-                               if (*usr.ut_name && *usr.ut_line)
-                                       output(&usr);
+                       process(NULL, show_labels);
                }
                break;
        case 1:                                 /* who utmp_file */
-               ufp = file(*argv);
-
-               if (only_current_term) {
-                       who_am_i(ufp);
+               if (quick_mode) {
+                       quick(*argv);
+               } else if (only_current_term) {
+                       who_am_i(*argv, show_labels);
                } else {
-                       /* all entries */
-                       while (fread((char *)&usr, sizeof(usr), 1, ufp) == 1)
-                               output(&usr);
+                       process(*argv, show_labels);
                }
                break;
        case 2:                                 /* who am i */
-               ufp = file(_PATH_UTMP);
-               who_am_i(ufp);
+               who_am_i(NULL, show_labels);
                break;
        default:
                usage();
                /* NOTREACHED */
        }
-       exit(0);
+
+       return 0;
+}
+
+static char *
+strrstr(const char *str, const char *pat)
+{
+       const char *estr;
+       size_t len;
+       if (*pat == '\0')
+               return __UNCONST(str);
+
+       len = strlen(pat);
+
+       for (estr = str + strlen(str); str < estr; estr--)
+               if (strncmp(estr, pat, len) == 0)
+                       return __UNCONST(estr);
+       return NULL;
 }
 
-void
-who_am_i(ufp)
-       FILE *ufp;
+static void
+who_am_i(const char *fname, int show_labels)
 {
-       struct utmp usr;
        struct passwd *pw;
-       char *p;
+       const char *p;
        char *t;
+       time_t now;
+       struct utmpentry *ehead, *ep;
 
        /* search through the utmp and find an entry for this tty */
-       if ((p = ttyname(0)) != NULL) {
-               /* strip any directory component */
-               if ((t = strrchr(p, '/')) != NULL)
+       if ((p = ttyname(STDIN_FILENO)) != NULL) {
+
+               /* strip directory prefixes for ttys */
+               if ((t = strrstr(p, "/pts/")) != NULL ||
+                   (t = strrchr(p, '/')) != NULL)
                        p = t + 1;
-               while (fread((char *)&usr, sizeof(usr), 1, ufp) == 1)
-                       if (usr.ut_name && !strcmp(usr.ut_line, p)) {
-                               output(&usr);
+
+               (void)getutentries(fname, &ehead);
+               for (ep = ehead; ep; ep = ep->next)
+                       if (strcmp(ep->line, p) == 0) {
+                               if (show_labels)
+                                       output_labels();
+                               eprint(ep);
                                return;
                        }
-               /* well, at least we know what the tty is */
-               (void)strncpy(usr.ut_line, p, UT_LINESIZE);
        } else
-               (void)strcpy(usr.ut_line, "tty??");
+               p = "tty??";
 
+       (void)time(&now);
        pw = getpwuid(getuid());
-       (void)strncpy(usr.ut_name, pw ? pw->pw_name : "?", UT_NAMESIZE);
-       (void)time(&usr.ut_time);
-       *usr.ut_host = '\0';
-       output(&usr);
+       if (show_labels)
+               output_labels();
+       print(pw ? pw->pw_name : "?", p, now, "", getpid(), 0, 0, 0, 0);
+}
+
+static void
+process(const char *fname, int show_labels)
+{
+       struct utmpentry *ehead, *ep;
+       (void)getutentries(fname, &ehead);
+       if (show_labels)
+               output_labels();
+       for (ep = ehead; ep != NULL; ep = ep->next)
+               eprint(ep);
+#ifdef __APPLE__
+       if ((etype & (1 << RUN_LVL)) != 0) {
+               printf("   .       run-level 3\n");
+       }
+#endif /* __APPLE__ */
+}
+
+static void
+eprint(const struct utmpentry *ep)
+{
+       print(ep->name, ep->line, (time_t)ep->tv.tv_sec, ep->host, ep->pid,
+#ifdef __APPLE__
+           0, 0, 0, ep->type);
+#else /* !__APPLE__ */
+           ep->term, ep->exit, ep->sess, ep->type);
+#endif /* __APPLE__ */
 }
 
-void
-output(up)
-       struct utmp *up;
+static void
+print(const char *name, const char *line, time_t t, const char *host,
+    pid_t pid, uint16_t term, uint16_t xit, uint16_t sess, uint16_t type)
 {
        struct stat sb;
-       char line[sizeof (up->ut_line) + 1];
        char state;
        static time_t now = 0;
        time_t idle;
+       const char *types = NULL;
+       size_t i;
 
        state = '?';
        idle = 0;
 
+       for (i = 0; ut_type_names[i].type >= 0; i++) {
+               types = ut_type_names[i].name;
+               if (ut_type_names[i].type == type)
+                       break;
+       }
+       
        if (show_term || show_idle) {
                if (now == 0)
                        time(&now);
                
-               strncpy(line, up->ut_line, sizeof (up->ut_line));
-               line[sizeof (up->ut_line)] = '\0';
-
+#ifdef __APPLE__
+               char tty[PATH_MAX + 1];
+               snprintf(tty, sizeof(tty), "%s%s", _PATH_DEV, line);
+               if (stat(tty, &sb) == 0) {
+#else /* !__APPLE__ */
                if (stat(line, &sb) == 0) {
+#endif /* __APPLE__ */
                        state = (sb.st_mode & 020) ? '+' : '-';
                        idle = now - sb.st_atime;
                }
                
        }
 
-       (void)printf("%-*.*s ", UT_NAMESIZE, UT_NAMESIZE, up->ut_name);
+#ifdef __APPLE__
+       switch (type) {
+       case LOGIN_PROCESS:
+               (void)printf("%-*.*s ", maxname, maxname, "LOGIN");
+               break;
+       case BOOT_TIME:
+               (void)printf("%-*.*s ", maxname, maxname, "reboot");
+               break;
+       default:
+               (void)printf("%-*.*s ", maxname, maxname, name);
+               break;
+       }
+#else /* !__APPLE__ */
+       (void)printf("%-*.*s ", maxname, maxname, name);
+#endif /* __APPLE__ */
 
-       if (show_term) {
+       if (show_term)
                (void)printf("%c ", state);
-       }
 
-       (void)printf("%-*.*s ", UT_LINESIZE, UT_LINESIZE, up->ut_line);
-       (void)printf("%.12s ", ctime(&up->ut_time) + 4);
+#ifdef __APPLE__
+       (void)printf("%-*.*s ", maxline, maxline, type == BOOT_TIME ? "~" : line);
+#else /* !__APPLE__ */
+       (void)printf("%-*.*s ", maxline, maxline, line);
+#endif /* __APPLE__ */
+       (void)printf("%.12s ", ctime(&t) + 4);
 
        if (show_idle) {
                if (idle < 60) 
@@ -226,48 +375,79 @@ output(up)
                                     (long)(idle % (60 * 60)) / 60);
                else
                        (void)printf(" old  ");
+
+               (void)printf("\t%6d", pid);
+               
+#ifndef __APPLE__
+               if (show_details) {
+                       if (type == RUN_LVL)
+                               (void)printf("\tnew=%c old=%c", term, xit);
+                       else
+                               (void)printf("\tterm=%d exit=%d", term, xit);
+                       (void)printf(" sess=%d", sess);
+                       (void)printf(" type=%s ", types);
+               }
+#endif /* !__APPLE__ */
        }
        
-       if (*up->ut_host)
-               printf("\t(%.*s)", UT_HOSTSIZE, up->ut_host);
+#ifdef __APPLE__
+       /* 6179576 */
+       if (type == DEAD_PROCESS)
+               (void)printf("\tterm=%d exit=%d", 0, 0);
+#endif /* __APPLE__ */
+
+       if (*host)
+               (void)printf("\t(%.*s)", maxhost, host);
        (void)putchar('\n');
 }
 
-void
-output_labels()
+static void
+output_labels(void)
 {
-       (void)printf("%-*.*s ", UT_NAMESIZE, UT_NAMESIZE, "USER");
+       (void)printf("%-*.*s ", maxname, maxname, "USER");
 
        if (show_term)
                (void)printf("S ");
        
-       (void)printf("%-*.*s ", UT_LINESIZE, UT_LINESIZE, "LINE");
+       (void)printf("%-*.*s ", maxline, maxline, "LINE");
        (void)printf("WHEN         ");
 
-       if (show_idle)
+       if (show_idle) {
                (void)printf("IDLE  ");
+               (void)printf("\t   PID");
        
-       (void)printf("\t%.*s", UT_HOSTSIZE, "FROM");
+               (void)printf("\tCOMMENT");
+       }               
 
        (void)putchar('\n');
 }
 
-FILE *
-file(name)
-       char *name;
+static void
+quick(const char *fname)
 {
-       FILE *ufp;
-
-       if (!(ufp = fopen(name, "r"))) {
-               err(1, "%s", name);
-               /* NOTREACHED */
+       struct utmpentry *ehead, *ep;
+       int num = 0;
+
+       (void)getutentries(fname, &ehead);
+       for (ep = ehead; ep != NULL; ep = ep->next) {
+               (void)printf("%-*s ", maxname, ep->name);
+               if ((++num % 8) == 0)
+                       (void)putchar('\n');
        }
-       return (ufp);
+       if (num % 8)
+               (void)putchar('\n');
+
+       (void)printf("# users = %d\n", num);
 }
 
-void
-usage()
+static void
+usage(void)
 {
-       (void)fprintf(stderr, "usage: who [-mTuH] [ file ]\n       who am i\n");
-       exit(1);
+#ifdef __APPLE__
+       (void)fprintf(stderr, "Usage: %s [-abdHlmpqrsTtu] [file]\n\t%s am i\n",
+#else /* !__APPLE__ */
+       (void)fprintf(stderr, "Usage: %s [-abdHlmqrsTtuv] [file]\n\t%s am i\n",
+#endif /* __APPLE__ */
+           getprogname(), getprogname());
+       exit(EXIT_FAILURE);
 }