coreutils/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\n"
  33         "Print information for USERNAME or the current user\n\n"
  34         "\t-g\tprints only the group ID\n"
  35         "\t-u\tprints only the user ID\n"
  36         "\t-r\tprints the real user ID instead of the effective ID (with -ug)\n\n";
  37
  38 extern int id_main(int argc, char **argv)
  39 {
  40         int no_user = 0, no_group = 0, print_real = 0;
  41         char *cp, *user, *group;
  42         gid_t gid;
  43
  44         cp = user = group = NULL;
  45
  46         argc--; argv++;
  47
  48         while (argc > 0) {
  49                 cp = *argv;
  50                 if (*cp == '-') {
  51                         switch (*++cp) {
  52                         case 'u': no_group = 1; break;
  53                         case 'g': no_user = 1; break;
  54                         case 'r': print_real = 1; break;
  55                         default: usage(id_usage);
  56                         }
  57                 } else {
  58                         user = cp;
  59                 }
  60                 argc--; argv++;
  61         }
  62
  63         if (no_user && no_group) usage(id_usage);
  64
  65         if (user == NULL) {
  66                 user = xmalloc(9);
  67                 group = xmalloc(9);
  68                 if (print_real) {
  69                         my_getpwuid(user, getuid());
  70                         my_getgrgid(group, getgid());
  71                 } else {
  72                         my_getpwuid(user, geteuid());
  73                         my_getgrgid(group, getegid());
  74                 }
  75         } else {
  76                 group = xmalloc(9);
  77             gid = my_getpwnamegid(user);
  78                 my_getgrgid(group, gid);
  79         }
  80
  81         if (no_group) printf("%u\n", my_getpwnam(user));
  82         else if (no_user) printf("%u\n", my_getgrnam(group));
  83         else
  84                 printf("uid=%u(%s) gid=%u(%s)\n",
  85                            my_getpwnam(user), user, my_getgrnam(group), group);
  86
  87
  88         exit(0);
  89 }
  90
  91
  92 /* END CODE */