lib/getugroups.c

   1 /* getugroups.c -- return a list of the groups a user is in
   2
   3    Copyright (C) 1990, 1991, 1998, 1999, 2000, 2003, 2004, 2005 Free
   4    Software Foundation.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2, or (at your option)
   9    any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software Foundation,
  18    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
  19
  20 /* Written by David MacKenzie. */
  21
  22 #ifdef HAVE_CONFIG_H
  23 # include <config.h>
  24 #endif
  25
  26 #include <sys/types.h>
  27 #include <stdio.h> /* grp.h on alpha OSF1 V2.0 uses "FILE *". */
  28 #include <grp.h>
  29
  30 #include <unistd.h>
  31
  32 #include <errno.h>
  33 #ifndef EOVERFLOW
  34 # define EOVERFLOW EINVAL
  35 #endif
  36
  37 /* setgrent, getgrent, and endgrent are not specified by POSIX.1,
  38    so header files might not declare them.
  39    If you don't have them at all, we can't implement this function.
  40    You lose!  */
  41 struct group *getgrent ();
  42
  43 #include <string.h>
  44
  45 #define STREQ(s1, s2) ((strcmp (s1, s2) == 0))
  46
  47 /* Like `getgroups', but for user USERNAME instead of for the current
  48    process.  Store at most MAXCOUNT group IDs in the GROUPLIST array.
  49    If GID is not -1, store it first (if possible).  GID should be the
  50    group ID (pw_gid) obtained from getpwuid, in case USERNAME is not
  51    listed in /etc/groups.
  52    Always return the number of groups of which USERNAME is a member.  */
  53
  54 int
  55 getugroups (int maxcount, GETGROUPS_T *grouplist, char *username, gid_t gid)
  56 {
  57   struct group *grp;
  58   register char **cp;
  59   register int count = 0;
  60
  61   if (gid != (gid_t) -1)
  62     {
  63       if (maxcount != 0)
  64         grouplist[count] = gid;
  65       ++count;
  66     }
  67
  68   setgrent ();
  69   while ((grp = getgrent ()) != 0)
  70     {
  71       for (cp = grp->gr_mem; *cp; ++cp)
  72         {
  73           int n;
  74
  75           if ( ! STREQ (username, *cp))
  76             continue;
  77
  78           /* See if this group number is already on the list.  */
  79           for (n = 0; n < count; ++n)
  80             if (grouplist && grouplist[n] == grp->gr_gid)
  81               break;
  82
  83           /* If it's a new group number, then try to add it to the list.  */
  84           if (n == count)
  85             {
  86               if (maxcount != 0)
  87                 {
  88                   if (count >= maxcount)
  89                     {
  90                       endgrent ();
  91                       return count;
  92                     }
  93                   grouplist[count] = grp->gr_gid;
  94                 }
  95               count++;
  96               if (count < 0)
  97                 {
  98                   errno = EOVERFLOW;
  99                   return -1;
 100                 }
 101             }
 102         }
 103     }
 104   endgrent ();
 105
 106   return count;
 107 }