lib/getgroups.c

   1 /* provide consistent interface to getgroups for systems that don't allow N==0
   2
   3    Copyright (C) 1996, 1999, 2003, 2006, 2007, 2008, 2009 Free
   4    Software Foundation, Inc.
   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 3 of the License, or
   9    (at your option) 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, see <http://www.gnu.org/licenses/>.  */
  18
  19 /* written by Jim Meyering */
  20
  21 #include <config.h>
  22
  23 #include <unistd.h>
  24
  25 #include <errno.h>
  26 #include <stdlib.h>
  27
  28 #undef getgroups
  29
  30 /* On at least Ultrix 4.3 and NextStep 3.2, getgroups (0, NULL) always
  31    fails.  On other systems, it returns the number of supplemental
  32    groups for the process.  This function handles that special case
  33    and lets the system-provided function handle all others.  However,
  34    it can fail with ENOMEM if memory is tight.  It is unspecified
  35    whether the effective group id is included in the list.  */
  36
  37 int
  38 rpl_getgroups (int n, GETGROUPS_T *group)
  39 {
  40   int n_groups;
  41   GETGROUPS_T *gbuf;
  42   int saved_errno;
  43
  44   if (n != 0)
  45     return getgroups (n, group);
  46
  47   n = 20;
  48   while (1)
  49     {
  50       /* No need to worry about address arithmetic overflow here,
  51          since the ancient systems that we're running on have low
  52          limits on the number of secondary groups.  */
  53       gbuf = malloc (n * sizeof *gbuf);
  54       if (!gbuf)
  55         return -1;
  56       n_groups = getgroups (n, gbuf);
  57       if (n_groups == -1 ? errno != EINVAL : n_groups < n)
  58         break;
  59       free (gbuf);
  60       n *= 2;
  61     }
  62
  63   saved_errno = errno;
  64   free (gbuf);
  65   errno = saved_errno;
  66
  67   return n_groups;
  68 }