src/users.c

   1 /* GNU's users.
   2    Copyright (C) 1992-2004 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 #include <config.h>
  21 #include <getopt.h>
  22 #include <stdio.h>
  23
  24 #include <sys/types.h>
  25 #include "system.h"
  26
  27 #include "error.h"
  28 #include "long-options.h"
  29 #include "quote.h"
  30 #include "readutmp.h"
  31
  32 /* The official name of this program (e.g., no `g' prefix).  */
  33 #define PROGRAM_NAME "users"
  34
  35 #define AUTHORS "Joseph Arceneaux", "David MacKenzie"
  36
  37 /* The name this program was run with. */
  38 char *program_name;
  39
  40 static int
  41 userid_compare (const void *v_a, const void *v_b)
  42 {
  43   char **a = (char **) v_a;
  44   char **b = (char **) v_b;
  45   return strcmp (*a, *b);
  46 }
  47
  48 static void
  49 list_entries_users (size_t n, const STRUCT_UTMP *this)
  50 {
  51   char **u = xnmalloc (n, sizeof *u);
  52   size_t i;
  53   size_t n_entries = 0;
  54
  55   while (n--)
  56     {
  57       if (UT_USER (this) [0]
  58 #ifdef USER_PROCESS
  59           && this->ut_type == USER_PROCESS
  60 #endif
  61           )
  62         {
  63           char *trimmed_name;
  64
  65           trimmed_name = extract_trimmed_name (this);
  66
  67           u[n_entries] = trimmed_name;
  68           ++n_entries;
  69         }
  70       this++;
  71     }
  72
  73   qsort (u, n_entries, sizeof (u[0]), userid_compare);
  74
  75   for (i = 0; i < n_entries; i++)
  76     {
  77       char c = (i < n_entries - 1 ? ' ' : '\n');
  78       fputs (u[i], stdout);
  79       putchar (c);
  80     }
  81
  82   for (i = 0; i < n_entries; i++)
  83     free (u[i]);
  84   free (u);
  85 }
  86
  87 /* Display a list of users on the system, according to utmp file FILENAME. */
  88
  89 static void
  90 users (const char *filename)
  91 {
  92   size_t n_users;
  93   STRUCT_UTMP *utmp_buf;
  94
  95   if (read_utmp (filename, &n_users, &utmp_buf) != 0)
  96     error (EXIT_FAILURE, errno, "%s", filename);
  97
  98   list_entries_users (n_users, utmp_buf);
  99
 100   free (utmp_buf);
 101 }
 102
 103 void
 104 usage (int status)
 105 {
 106   if (status != EXIT_SUCCESS)
 107     fprintf (stderr, _("Try `%s --help' for more information.\n"),
 108              program_name);
 109   else
 110     {
 111       printf (_("Usage: %s [OPTION]... [ FILE ]\n"), program_name);
 112       printf (_("\
 113 Output who is currently logged in according to FILE.\n\
 114 If FILE is not specified, use %s.  %s as FILE is common.\n\
 115 \n\
 116 "),
 117               UTMP_FILE, WTMP_FILE);
 118       fputs (HELP_OPTION_DESCRIPTION, stdout);
 119       fputs (VERSION_OPTION_DESCRIPTION, stdout);
 120       printf (_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT);
 121     }
 122   exit (status);
 123 }
 124
 125 int
 126 main (int argc, char **argv)
 127 {
 128   initialize_main (&argc, &argv);
 129   program_name = argv[0];
 130   setlocale (LC_ALL, "");
 131   bindtextdomain (PACKAGE, LOCALEDIR);
 132   textdomain (PACKAGE);
 133
 134   atexit (close_stdout);
 135
 136   parse_long_options (argc, argv, PROGRAM_NAME, GNU_PACKAGE, VERSION,
 137                       usage, AUTHORS, (char const *) NULL);
 138   if (getopt_long (argc, argv, "", NULL, NULL) != -1)
 139     usage (EXIT_FAILURE);
 140
 141   switch (argc - optind)
 142     {
 143     case 0:                     /* users */
 144       users (UTMP_FILE);
 145       break;
 146
 147     case 1:                     /* users <utmp file> */
 148       users (argv[optind]);
 149       break;
 150
 151     default:                    /* lose */
 152       error (0, 0, _("extra operand %s"), quote (argv[optind + 1]));
 153       usage (EXIT_FAILURE);
 154     }
 155
 156   exit (EXIT_SUCCESS);
 157 }