c33c76dfb397e9bd9bffdf4bfb941f5bb9948720
[kopensolaris-gnu/glibc.git] / 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   int gc_cycle;
50
51   /* If the mapping is available, try to search there instead of
52      communicating with the nscd.  */
53   struct mapped_database *mapped;
54   mapped = __nscd_get_map_ref (GETFDGR, "group", &map_handle, &gc_cycle);
55
56  retry:;
57   const initgr_response_header *initgr_resp = NULL;
58   char *respdata = NULL;
59   int retval = -1;
60   int sock = -1;
61
62   if (mapped != NO_MAPPING)
63     {
64       const struct datahead *found = __nscd_cache_search (INITGROUPS, user,
65                                                           userlen, mapped);
66       if (found != NULL)
67         {
68           initgr_resp = &found->data[0].initgrdata;
69           respdata = (char *) (initgr_resp + 1);
70           char *recend = (char *) found->data + found->recsize;
71
72           if (respdata + initgr_resp->ngrps * sizeof (int32_t) > recend)
73             goto out;
74         }
75     }
76
77   /* If we do not have the cache mapped, try to get the data over the
78      socket.  */
79   initgr_response_header initgr_resp_mem;
80   if (initgr_resp == NULL)
81     {
82       sock = __nscd_open_socket (user, userlen, INITGROUPS, &initgr_resp_mem,
83                                  sizeof (initgr_resp_mem));
84       if (sock == -1)
85         /* nscd not running or wrong version or hosts caching disabled.  */
86         __nss_not_use_nscd_group = 1;
87
88       initgr_resp = &initgr_resp_mem;
89     }
90
91   if (initgr_resp->found == 1)
92     {
93       /* The following code assumes that gid_t and int32_t are the
94          same size.  This is the case for al existing implementation.
95          If this should change some code needs to be added which
96          doesn't use memcpy but instead copies each array element one
97          by one.  */
98       assert (sizeof (int32_t) == sizeof (gid_t));
99
100       /* Make sure we have enough room.  We always count GROUP in even
101          though we might not end up adding it.  */
102       if (*size < initgr_resp->ngrps + 1)
103         {
104           gid_t *newp = realloc (*groupsp,
105                                  (initgr_resp->ngrps + 1) * sizeof (gid_t));
106           if (newp == NULL)
107             /* We cannot increase the buffer size.  */
108             goto out;
109
110           *groupsp = newp;
111           *size = initgr_resp->ngrps + 1;
112         }
113
114       if (respdata == NULL)
115         {
116           /* Read the data from the socket.  */
117           if ((size_t) TEMP_FAILURE_RETRY (__read (sock, *groupsp,
118                                                    initgr_resp->ngrps
119                                                    * sizeof (gid_t)))
120               == initgr_resp->ngrps * sizeof (gid_t))
121             retval = initgr_resp->ngrps;
122         }
123       else
124         {
125           /* Just copy the data.  */
126           retval = initgr_resp->ngrps;
127           memcpy (*groupsp, respdata, retval * sizeof (gid_t));
128         }
129
130       /* Check whether GROUP is part of the mix.  If not, add it.  */
131       if (retval >= 0)
132         {
133           int cnt;
134           for (cnt = 0; cnt < retval; ++cnt)
135             if ((*groupsp)[cnt] == group)
136               break;
137
138           if (cnt == retval)
139             (*groupsp)[retval++] = group;
140         }
141     }
142   else
143     {
144       /* The `errno' to some value != ERANGE.  */
145       __set_errno (ENOENT);
146       /* Even though we have not found anything, the result is zero.  */
147       retval = 0;
148     }
149
150   if (sock != -1)
151     close_not_cancel_no_status (sock);
152  out:
153   if (__nscd_drop_map_ref (mapped, &gc_cycle) != 0 && retval != -1)
154     {
155       /* When we come here this means there has been a GC cycle while we
156          were looking for the data.  This means the data might have been
157          inconsistent.  Retry if possible.  */
158       if ((gc_cycle & 1) != 0)
159         {
160           /* nscd is just running gc now.  Disable using the mapping.  */
161           __nscd_unmap (mapped);
162           mapped = NO_MAPPING;
163         }
164
165       goto retry;
166     }
167
168   return retval;
169 }