src/who.c

   1 /* GNU's who.
   2    Copyright (C) 92, 93, 94, 95, 96, 1997 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software Foundation,
  16    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
  17
  18 /* Written by jla; revised by djm */
  19
  20 /* Output format:
  21    name [state] line time [idle] host
  22    state: -T
  23    name, line, time: not -q
  24    idle: -u
  25 */
  26
  27 #include <config.h>
  28 #include <getopt.h>
  29
  30 #include "system.h"
  31 #include "error.h"
  32 #include "readutmp.h"
  33
  34 #ifndef MAXHOSTNAMELEN
  35 # define MAXHOSTNAMELEN 64
  36 #endif
  37
  38 #ifndef S_IWGRP
  39 # define S_IWGRP 020
  40 #endif
  41
  42 int gethostname ();
  43 char *ttyname ();
  44 char *xmalloc ();
  45
  46 /* The name this program was run with. */
  47 char *program_name;
  48
  49 /* If nonzero, display usage information and exit.  */
  50 static int show_help;
  51
  52 /* If nonzero, print the version on standard output and exit.  */
  53 static int show_version;
  54
  55 /* If nonzero, attempt to canonicalize hostnames via a DNS lookup. */
  56 static int do_lookup;
  57
  58 /* If nonzero, display only a list of usernames and count of
  59    the users logged on.
  60    Ignored for `who am i'. */
  61 static int short_list;
  62
  63 /* If nonzero, display the hours:minutes since each user has touched
  64    the keyboard, or "." if within the last minute, or "old" if
  65    not within the last day. */
  66 static int include_idle;
  67
  68 /* If nonzero, display a line at the top describing each field. */
  69 static int include_heading;
  70
  71 /* If nonzero, display a `+' for each user if mesg y, a `-' if mesg n,
  72    or a `?' if their tty cannot be statted. */
  73 static int include_mesg;
  74
  75 static struct option const longopts[] =
  76 {
  77   {"count", no_argument, NULL, 'q'},
  78   {"idle", no_argument, NULL, 'u'},
  79   {"heading", no_argument, NULL, 'H'},
  80   {"lookup", no_argument, NULL, 'l'},
  81   {"message", no_argument, NULL, 'T'},
  82   {"mesg", no_argument, NULL, 'T'},
  83   {"writable", no_argument, NULL, 'T'},
  84   {"help", no_argument, &show_help, 1},
  85   {"version", no_argument, &show_version, 1},
  86   {NULL, 0, NULL, 0}
  87 };
  88
  89 /* Return a string representing the time between WHEN and the time
  90    that this function is first run. */
  91
  92 static const char *
  93 idle_string (time_t when)
  94 {
  95   static time_t now = 0;
  96   static char idle_hhmm[10];
  97   time_t seconds_idle;
  98
  99   if (now == 0)
 100     time (&now);
 101
 102   seconds_idle = now - when;
 103   if (seconds_idle < 60)        /* One minute. */
 104     return "  .  ";
 105   if (seconds_idle < (24 * 60 * 60)) /* One day. */
 106     {
 107       sprintf (idle_hhmm, "%02d:%02d",
 108                (int) (seconds_idle / (60 * 60)),
 109                (int) ((seconds_idle % (60 * 60)) / 60));
 110       return (const char *) idle_hhmm;
 111     }
 112   return _(" old ");
 113 }
 114
 115 /* Display a line of information about UTMP_ENT. */
 116
 117 static void
 118 print_entry (const STRUCT_UTMP *utmp_ent)
 119 {
 120   struct stat stats;
 121   time_t last_change;
 122   char mesg;
 123
 124 #define DEV_DIR_WITH_TRAILING_SLASH "/dev/"
 125 #define DEV_DIR_LEN (sizeof (DEV_DIR_WITH_TRAILING_SLASH) - 1)
 126
 127   char line[sizeof (utmp_ent->ut_line) + DEV_DIR_LEN + 1];
 128   time_t tm;
 129
 130   /* Copy ut_line into LINE, prepending `/dev/' if ut_line is not
 131      already an absolute pathname.  Some system may put the full,
 132      absolute pathname in ut_line.  */
 133   if (utmp_ent->ut_line[0] == '/')
 134     {
 135       strncpy (line, utmp_ent->ut_line, sizeof (utmp_ent->ut_line));
 136       line[sizeof (utmp_ent->ut_line)] = '\0';
 137     }
 138   else
 139     {
 140       strcpy (line, DEV_DIR_WITH_TRAILING_SLASH);
 141       strncpy (line + DEV_DIR_LEN, utmp_ent->ut_line, sizeof (utmp_ent->ut_line));
 142       line[DEV_DIR_LEN + sizeof (utmp_ent->ut_line)] = '\0';
 143     }
 144
 145   if (stat (line, &stats) == 0)
 146     {
 147       mesg = (stats.st_mode & S_IWGRP) ? '+' : '-';
 148       last_change = stats.st_atime;
 149     }
 150   else
 151     {
 152       mesg = '?';
 153       last_change = 0;
 154     }
 155
 156   printf ("%-8.*s", (int) sizeof (utmp_ent->ut_name), utmp_ent->ut_name);
 157   if (include_mesg)
 158     printf ("  %c  ", mesg);
 159   printf (" %-8.*s", (int) sizeof (utmp_ent->ut_line), utmp_ent->ut_line);
 160
 161   /* Don't take the address of UT_TIME_MEMBER directly.
 162      Ulrich Drepper wrote:
 163      ``... GNU libc (and perhaps other libcs as well) have extended
 164      utmp file formats which do not use a simple time_t ut_time field.
 165      In glibc, ut_time is a macro which selects for backward compatibility
 166      the tv_sec member of a struct timeval value.''  */
 167   tm = UT_TIME_MEMBER (utmp_ent);
 168   printf (" %-12.12s", ctime (&tm) + 4);
 169
 170   if (include_idle)
 171     {
 172       if (last_change)
 173         printf (" %s", idle_string (last_change));
 174       else
 175         printf ("   .  ");
 176     }
 177 #ifdef HAVE_UT_HOST
 178   if (utmp_ent->ut_host[0] && do_lookup)
 179     {
 180       extern char *canon_host ();
 181       char ut_host[sizeof (utmp_ent->ut_host) + 1];
 182       char *host = 0, *display = 0;
 183
 184       /* Copy the host name into UT_HOST, and ensure it's nul terminated. */
 185       strncpy (ut_host, utmp_ent->ut_host, (int) sizeof (utmp_ent->ut_host));
 186       ut_host[sizeof (utmp_ent->ut_host)] = '\0';
 187
 188       /* Look for an X display.  */
 189       display = strrchr (ut_host, ':');
 190       if (display)
 191         *display++ = '\0';
 192
 193       if (*ut_host)
 194         /* See if we can canonicalize it.  */
 195         host = canon_host (ut_host);
 196       if (! host)
 197         host = ut_host;
 198
 199       if (display)
 200         printf (" (%s:%s)", host, display);
 201       else
 202         printf (" (%s)", host);
 203     }
 204 #endif
 205
 206   putchar ('\n');
 207 }
 208
 209 /* Print the username of each valid entry and the number of valid entries
 210    in UTMP_BUF, which should have N elements. */
 211
 212 static void
 213 list_entries_who (int n, const STRUCT_UTMP *utmp_buf)
 214 {
 215   int entries;
 216
 217   entries = 0;
 218   while (n--)
 219     {
 220       if (utmp_buf->ut_name[0]
 221 #ifdef USER_PROCESS
 222           && utmp_buf->ut_type == USER_PROCESS
 223 #endif
 224          )
 225         {
 226           char *trimmed_name;
 227
 228           trimmed_name = extract_trimmed_name (utmp_buf);
 229
 230           printf ("%s ", trimmed_name);
 231           free (trimmed_name);
 232           entries++;
 233         }
 234       utmp_buf++;
 235     }
 236   printf (_("\n# users=%u\n"), entries);
 237 }
 238
 239 static void
 240 print_heading (void)
 241 {
 242   printf ("%-8s ", _("USER"));
 243   if (include_mesg)
 244     printf (_("MESG "));
 245   printf ("%-8s ", _("LINE"));
 246   printf (_("LOGIN-TIME   "));
 247   if (include_idle)
 248     printf (_("IDLE  "));
 249   printf (_("FROM\n"));
 250 }
 251
 252 /* Display UTMP_BUF, which should have N entries. */
 253
 254 static void
 255 scan_entries (int n, const STRUCT_UTMP *utmp_buf)
 256 {
 257   if (include_heading)
 258     print_heading ();
 259
 260   while (n--)
 261     {
 262       if (utmp_buf->ut_name[0]
 263 #ifdef USER_PROCESS
 264           && utmp_buf->ut_type == USER_PROCESS
 265 #endif
 266          )
 267         print_entry (utmp_buf);
 268       utmp_buf++;
 269     }
 270 }
 271
 272 /* Display a list of who is on the system, according to utmp file FILENAME. */
 273
 274 static void
 275 who (const char *filename)
 276 {
 277   int n_users;
 278   STRUCT_UTMP *utmp_buf;
 279   int fail = read_utmp (filename, &n_users, &utmp_buf);
 280
 281   if (fail)
 282     error (1, errno, "%s", filename);
 283
 284   if (short_list)
 285     list_entries_who (n_users, utmp_buf);
 286   else
 287     scan_entries (n_users, utmp_buf);
 288 }
 289
 290 /* Search UTMP_CONTENTS, which should have N entries, for
 291    an entry with a `ut_line' field identical to LINE.
 292    Return the first matching entry found, or NULL if there
 293    is no matching entry. */
 294
 295 static const STRUCT_UTMP *
 296 search_entries (int n, const STRUCT_UTMP *utmp_buf, const char *line)
 297 {
 298   while (n--)
 299     {
 300       if (utmp_buf->ut_name[0]
 301 #ifdef USER_PROCESS
 302           && utmp_buf->ut_type == USER_PROCESS
 303 #endif
 304           && !strncmp (line, utmp_buf->ut_line, sizeof (utmp_buf->ut_line)))
 305         return utmp_buf;
 306       utmp_buf++;
 307     }
 308   return NULL;
 309 }
 310
 311 /* Display the entry in utmp file FILENAME for this tty on standard input,
 312    or nothing if there is no entry for it. */
 313
 314 static void
 315 who_am_i (const char *filename)
 316 {
 317   const STRUCT_UTMP *utmp_entry;
 318   STRUCT_UTMP *utmp_buf;
 319   char hostname[MAXHOSTNAMELEN + 1];
 320   char *tty;
 321   int fail;
 322   int n_users;
 323
 324   if (gethostname (hostname, MAXHOSTNAMELEN + 1))
 325     *hostname = 0;
 326
 327   if (include_heading)
 328     {
 329       printf ("%*s ", (int) strlen (hostname), " ");
 330       print_heading ();
 331     }
 332
 333   tty = ttyname (0);
 334   if (tty == NULL)
 335     return;
 336   tty += 5;                     /* Remove "/dev/".  */
 337
 338   fail = read_utmp (filename, &n_users, &utmp_buf);
 339
 340   if (fail)
 341     error (1, errno, "%s", filename);
 342
 343   utmp_entry = search_entries (n_users, utmp_buf, tty);
 344   if (utmp_entry == NULL)
 345     return;
 346
 347   printf ("%s!", hostname);
 348   print_entry (utmp_entry);
 349 }
 350
 351 static void
 352 usage (int status)
 353 {
 354   if (status != 0)
 355     fprintf (stderr, _("Try `%s --help' for more information.\n"),
 356              program_name);
 357   else
 358     {
 359       printf (_("Usage: %s [OPTION]... [ FILE | ARG1 ARG2 ]\n"), program_name);
 360       printf (_("\
 361 \n\
 362   -H, --heading     print line of column headings\n\
 363   -i, -u, --idle    add user idle time as HOURS:MINUTES, . or old\n\
 364   -l, --lookup      attempt to canonicalize hostnames via DNS\n\
 365   -m                only hostname and user associated with stdin\n\
 366   -q, --count       all login names and number of users logged on\n\
 367   -s                (ignored)\n\
 368   -T, -w, --mesg    add user's message status as +, - or ?\n\
 369       --message     same as -T\n\
 370       --writable    same as -T\n\
 371       --help        display this help and exit\n\
 372       --version     output version information and exit\n\
 373 \n\
 374 If FILE is not specified, use %s.  %s as FILE is common.\n\
 375 If ARG1 ARG2 given, -m presumed: `am i' or `mom likes' are usual.\n\
 376 "), UTMP_FILE, WTMP_FILE);
 377       puts (_("\nReport bugs to <sh-utils-bugs@gnu.org>."));
 378     }
 379   exit (status);
 380 }
 381
 382 int
 383 main (int argc, char **argv)
 384 {
 385   int optc, longind;
 386   int my_line_only = 0;
 387
 388   program_name = argv[0];
 389   setlocale (LC_ALL, "");
 390   bindtextdomain (PACKAGE, LOCALEDIR);
 391   textdomain (PACKAGE);
 392
 393   while ((optc = getopt_long (argc, argv, "ilmqsuwHT", longopts, &longind)) != -1)
 394     {
 395       switch (optc)
 396         {
 397         case 0:
 398           break;
 399
 400         case 'm':
 401           my_line_only = 1;
 402           break;
 403
 404         case 'l':
 405           do_lookup = 1;
 406           break;
 407
 408         case 'q':
 409           short_list = 1;
 410           break;
 411
 412         case 's':
 413           break;
 414
 415         case 'i':
 416         case 'u':
 417           include_idle = 1;
 418           break;
 419
 420         case 'H':
 421           include_heading = 1;
 422           break;
 423
 424         case 'w':
 425         case 'T':
 426           include_mesg = 1;
 427           break;
 428
 429         default:
 430           usage (1);
 431         }
 432     }
 433
 434   if (show_version)
 435     {
 436       printf ("who (%s) %s\n", GNU_PACKAGE, VERSION);
 437       exit (0);
 438     }
 439
 440   if (show_help)
 441     usage (0);
 442
 443   switch (argc - optind)
 444     {
 445     case 0:                     /* who */
 446       if (my_line_only)
 447         who_am_i (UTMP_FILE);
 448       else
 449         who (UTMP_FILE);
 450       break;
 451
 452     case 1:                     /* who <utmp file> */
 453       if (my_line_only)
 454         who_am_i (argv[optind]);
 455       else
 456         who (argv[optind]);
 457       break;
 458
 459     case 2:                     /* who <blurf> <glop> */
 460       who_am_i (UTMP_FILE);
 461       break;
 462
 463     default:                    /* lose */
 464       error (0, 0, _("too many arguments"));
 465       usage (1);
 466     }
 467
 468   exit (0);
 469 }