id.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini id implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 2000 by Randolph Chung <tausq@debian.org>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  21  *
  22  */
  23
  24 #include "internal.h"
  25 #include <stdio.h>
  26 #include <unistd.h>
  27 #include <pwd.h>
  28 #include <grp.h>
  29 #include <sys/types.h>
  30
  31 static const char id_usage[] =
  32         "id [OPTIONS]... [USERNAME]\n"
  33 #ifndef BB_FEATURE_TRIVIAL_HELP
  34         "\nPrint information for USERNAME or the current user\n\n"
  35         "\t-g\tprints only the group ID\n"
  36         "\t-u\tprints only the user ID\n"
  37         "\t-r\tprints the real user ID instead of the effective ID (with -ug)\n\n"
  38 #endif
  39         ;
  40
  41 extern int id_main(int argc, char **argv)
  42 {
  43         int no_user = 0, no_group = 0, print_real = 0;
  44         char *cp, *user, *group;
  45         gid_t gid;
  46
  47         cp = user = group = NULL;
  48
  49         argc--; argv++;
  50
  51         while (argc > 0) {
  52                 cp = *argv;
  53                 if (*cp == '-') {
  54                         switch (*++cp) {
  55                         case 'u': no_group = 1; break;
  56                         case 'g': no_user = 1; break;
  57                         case 'r': print_real = 1; break;
  58                         default: usage(id_usage);
  59                         }
  60                 } else {
  61                         user = cp;
  62                 }
  63                 argc--; argv++;
  64         }
  65
  66         if (no_user && no_group) usage(id_usage);
  67
  68         if (user == NULL) {
  69                 user = xmalloc(9);
  70                 group = xmalloc(9);
  71                 if (print_real) {
  72                         my_getpwuid(user, getuid());
  73                         my_getgrgid(group, getgid());
  74                 } else {
  75                         my_getpwuid(user, geteuid());
  76                         my_getgrgid(group, getegid());
  77                 }
  78         } else {
  79                 group = xmalloc(9);
  80             gid = my_getpwnamegid(user);
  81                 my_getgrgid(group, gid);
  82         }
  83
  84         if (no_group) printf("%u\n", my_getpwnam(user));
  85         else if (no_user) printf("%u\n", my_getgrnam(group));
  86         else
  87                 printf("uid=%u(%s) gid=%u(%s)\n",
  88                            my_getpwnam(user), user, my_getgrnam(group), group);
  89
  90
  91         exit(0);
  92 }
  93
  94
  95 /* END CODE */