nscd/nscd_initgroups.c

   1 /* Copyright (C) 2004 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3    Contributed by Ulrich Drepper <drepper@redhat.com>, 2004.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, write to the Free
  17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  18    02111-1307 USA.  */
  19
  20 #include <assert.h>
  21 #include <errno.h>
  22 #include <grp.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <not-cancel.h>
  27
  28 #include "nscd-client.h"
  29 #include "nscd_proto.h"
  30
  31
  32 libc_locked_map_ptr (map_handle);
  33 /* Note that we only free the structure if necessary.  The memory
  34    mapping is not removed since it is not visible to the malloc
  35    handling.  */
  36 libc_freeres_fn (gr_map_free)
  37 {
  38
  39   if (map_handle.mapped != NO_MAPPING)
  40     free (map_handle.mapped);
  41 }
  42
  43
  44 int
  45 __nscd_getgrouplist (const char *user, gid_t group, long int *size,
  46                      gid_t **groupsp, long int limit)
  47 {
  48   size_t userlen = strlen (user) + 1;
  49   const initgr_response_header *initgr_resp = NULL;
  50   char *respdata = NULL;
  51   int retval = -1;
  52   int sock = -1;
  53   int gc_cycle;
  54
  55   /* If the mapping is available, try to search there instead of
  56      communicating with the nscd.  */
  57   struct mapped_database *mapped = __nscd_get_map_ref (GETFDGR, "group",
  58                                                        &map_handle, &gc_cycle);
  59  retry:
  60   if (mapped != NO_MAPPING)
  61     {
  62       const struct datahead *found = __nscd_cache_search (INITGROUPS, user,
  63                                                           userlen, mapped);
  64       if (found != NULL)
  65         {
  66           initgr_resp = &found->data[0].initgrdata;
  67           respdata = (char *) (initgr_resp + 1);
  68           char *recend = (char *) found->data + found->recsize;
  69
  70           if (respdata + initgr_resp->ngrps * sizeof (int32_t) > recend)
  71             goto out;
  72         }
  73     }
  74
  75   /* If we do not have the cache mapped, try to get the data over the
  76      socket.  */
  77   initgr_response_header initgr_resp_mem;
  78   if (initgr_resp == NULL)
  79     {
  80       sock = __nscd_open_socket (user, userlen, INITGROUPS, &initgr_resp_mem,
  81                                  sizeof (initgr_resp_mem));
  82       if (sock == -1)
  83         /* nscd not running or wrong version or hosts caching disabled.  */
  84         __nss_not_use_nscd_group = 1;
  85
  86       initgr_resp = &initgr_resp_mem;
  87     }
  88
  89   if (initgr_resp->found == 1)
  90     {
  91       /* The following code assumes that gid_t and int32_t are the
  92          same size.  This is the case for al existing implementation.
  93          If this should change some code needs to be added which
  94          doesn't use memcpy but instead copies each array element one
  95          by one.  */
  96       assert (sizeof (int32_t) == sizeof (gid_t));
  97
  98       /* Make sure we have enough room.  We always count GROUP in even
  99          though we might not end up adding it.  */
 100       if (*size < initgr_resp->ngrps + 1)
 101         {
 102           gid_t *newp = realloc (*groupsp,
 103                                  (initgr_resp->ngrps + 1) * sizeof (gid_t));
 104           if (newp == NULL)
 105             /* We cannot increase the buffer size.  */
 106             goto out;
 107
 108           *groupsp = newp;
 109           *size = initgr_resp->ngrps + 1;
 110         }
 111
 112       if (respdata == NULL)
 113         {
 114           /* Read the data from the socket.  */
 115           if ((size_t) TEMP_FAILURE_RETRY (__read (sock, *groupsp,
 116                                                    initgr_resp->ngrps
 117                                                    * sizeof (gid_t)))
 118               == initgr_resp->ngrps * sizeof (gid_t))
 119             retval = initgr_resp->ngrps;
 120         }
 121       else
 122         {
 123           /* Just copy the data.  */
 124           retval = initgr_resp->ngrps;
 125           memcpy (*groupsp, respdata, retval * sizeof (gid_t));
 126         }
 127
 128       /* Check whether GROUP is part of the mix.  If not, add it.  */
 129       if (retval >= 0)
 130         {
 131           int cnt;
 132           for (cnt = 0; cnt < retval; ++cnt)
 133             if ((*groupsp)[cnt] == group)
 134               break;
 135
 136           if (cnt == retval)
 137             (*groupsp)[retval++] = group;
 138         }
 139     }
 140   else
 141     {
 142       /* The `errno' to some value != ERANGE.  */
 143       __set_errno (ENOENT);
 144       /* Even though we have not found anything, the result is zero.  */
 145       retval = 0;
 146     }
 147
 148   if (sock != -1)
 149     close_not_cancel_no_status (sock);
 150  out:
 151   if (__nscd_drop_map_ref (mapped, gc_cycle) != 0)
 152     /* When we come here this means there has been a GC cycle while we
 153        were looking for the data.  This means the data might have been
 154        inconsistent.  Retry.  */
 155     goto retry;
 156
 157   return retval;
 158 }