Add initgroups support.
[kopensolaris-gnu/glibc.git] / hesiod / nss_hesiod / hesiod-grp.c
1 /* Copyright (C) 1997, 1999 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Mark Kettenis <kettenis@phys.uva.nl>, 1997.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Library General Public License as
7    published by the Free Software Foundation; either version 2 of the
8    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    Library General Public License for more details.
14
15    You should have received a copy of the GNU Library General Public
16    License along with the GNU C Library; see the file COPYING.LIB.  If not,
17    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18    Boston, MA 02111-1307, USA.  */
19
20 #include <ctype.h>
21 #include <errno.h>
22 #include <grp.h>
23 #include <hesiod.h>
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <nss.h>
28 #include <bits/libc-lock.h>
29
30 /* Get the declaration of the parser function.  */
31 #define ENTNAME grent
32 #define STRUCTURE group
33 #define EXTERN_PARSER
34 #include <nss/nss_files/files-parse.c>
35
36 /* Locks the static variables in this file.  */
37 __libc_lock_define_initialized (static, lock);
38
39 static void *context = NULL;
40
41 static enum nss_status
42 internal_setgrent (void)
43 {
44   if (!context)
45     {
46       if (hesiod_init (&context) == -1)
47         return NSS_STATUS_UNAVAIL;
48     }
49
50   return NSS_STATUS_SUCCESS;
51 }
52
53 enum nss_status
54 _nss_hesiod_setgrent (void)
55 {
56   enum nss_status status;
57
58   __libc_lock_lock (lock);
59
60   status = internal_setgrent ();
61
62   __libc_lock_unlock (lock);
63
64   return status;
65 }
66
67 enum nss_status
68 _nss_hesiod_endgrent (void)
69 {
70   __libc_lock_lock (lock);
71
72   if (context)
73     {
74       hesiod_end (context);
75       context = NULL;
76     }
77
78   __libc_lock_unlock (lock);
79
80   return NSS_STATUS_SUCCESS;
81 }
82
83 static enum nss_status
84 lookup (const char *name, const char *type, struct group *grp,
85         char *buffer, size_t buflen, int *errnop)
86 {
87   enum nss_status status;
88   struct parser_data *data = (void *) buffer;
89   size_t linebuflen;
90   char **list;
91   int parse_res;
92   size_t len;
93
94   status = internal_setgrent ();
95   if (status != NSS_STATUS_SUCCESS)
96     return status;
97
98   list = hesiod_resolve (context, name, type);
99   if (list == NULL)
100     return errno == ENOENT ? NSS_STATUS_NOTFOUND : NSS_STATUS_UNAVAIL;
101
102   linebuflen = buffer + buflen - data->linebuffer;
103   len = strlen (*list) + 1;
104   if (linebuflen < len)
105     {
106       hesiod_free_list (context, list);
107       *errnop = ERANGE;
108       return NSS_STATUS_TRYAGAIN;
109     }
110
111   memcpy (data->linebuffer, *list, len);
112   hesiod_free_list (context, list);
113
114   parse_res = _nss_files_parse_grent (buffer, grp, data, buflen, errnop);
115   if (parse_res < 1)
116     return parse_res == -1 ? NSS_STATUS_TRYAGAIN : NSS_STATUS_NOTFOUND;
117
118   return NSS_STATUS_SUCCESS;
119 }
120
121 enum nss_status
122 _nss_hesiod_getgrnam_r (const char *name, struct group *grp,
123                         char *buffer, size_t buflen, int *errnop)
124 {
125   enum nss_status status;
126
127   __libc_lock_lock (lock);
128
129   status = lookup (name, "group", grp, buffer, buflen, errnop);
130
131   __libc_lock_unlock (lock);
132
133   return status;
134 }
135
136 enum nss_status
137 _nss_hesiod_getgrgid_r (gid_t gid, struct group *grp,
138                         char *buffer, size_t buflen, int *errnop)
139 {
140   enum nss_status status = NSS_STATUS_UNAVAIL;
141   char gidstr[21];      /* We will probably never have a gid_t with more
142                            than 64 bits.  */
143
144   snprintf (gidstr, sizeof gidstr, "%d", gid);
145
146   __libc_lock_lock (lock);
147
148   status = lookup (gidstr, "gid", grp, buffer, buflen, errnop);
149
150   __libc_lock_unlock (lock);
151
152   return status;
153 }
154
155 static int
156 internal_gid_in_list (const gid_t *list, const gid_t g, long int len)
157 {
158   while (len > 0)
159     {
160       if (*list == g)
161         return 1;
162       --len;
163       ++list;
164     }
165   return 0;
166 }
167
168 static enum nss_status
169 internal_gid_from_group (void *context, const char *groupname, gid_t *group)
170 {
171   char **grp_res;
172   enum nss_status status = NSS_STATUS_NOTFOUND;
173
174   grp_res = hesiod_resolve (context, groupname, "group");
175   if (grp_res != NULL && *grp_res != NULL)
176     {
177       char *p = *grp_res;
178
179       while (*p != '\0' && *p != ':')
180         ++p;
181       while (*p != '\0' && *p == ':')
182         ++p;
183       while (*p != '\0' && *p != ':')
184         ++p;
185       while (*p != '\0' && *p == ':')
186         ++p;
187       if (*p == ':')
188         {
189           char *endp;
190           char *q = ++p;
191
192           q = p;
193           while (*q != '\0' && *q != ':')
194             ++q;
195
196           *group = strtol (p, &endp, 10);
197           if (endp == q && endp != p)
198             status = NSS_STATUS_SUCCESS;
199         }
200       hesiod_free_list (context, grp_res);
201     }
202   return status;
203 }
204
205 enum nss_status
206 _nss_hesiod_initgroups (const char *user, gid_t group, long int *start,
207                         long int *size, gid_t *groups, long int limit,
208                         int *errnop)
209 {
210   enum nss_status status = NSS_STATUS_SUCCESS;
211   char **list = NULL;
212   char *p;
213   void *context;
214
215   if (hesiod_init (&context) == -1)
216     return NSS_STATUS_UNAVAIL;
217
218   list = hesiod_resolve (context, user, "grplist");
219
220   if (list == NULL)
221     {
222       hesiod_end(context);
223       return errno == ENOENT ? NSS_STATUS_NOTFOUND : NSS_STATUS_UNAVAIL;
224     }
225
226   if (!internal_gid_in_list (groups, group, *start) && *start < limit)
227     groups[(*start)++] = group;
228
229   p = *list;
230   while (*p != '\0' && *start < limit)
231     {
232       char *endp;
233       char *q;
234
235       status = NSS_STATUS_NOTFOUND;
236
237       q = p;
238       while (*q != '\0' && *q != ':')
239         ++q;
240
241       if (*q != '\0')
242         *q++ = '\0';
243
244       group = strtol (p, &endp, 10);
245       if (*endp == '\0' && endp != p)
246         status = NSS_STATUS_SUCCESS;
247       else
248         status = internal_gid_from_group (context, p, &group);
249
250       if (status == NSS_STATUS_SUCCESS
251           && !internal_gid_in_list (groups, group, *start))
252         groups[(*start)++] = group;
253
254       p = q;
255     }
256
257   hesiod_free_list (context, list);
258   hesiod_end(context);
259
260   return NSS_STATUS_SUCCESS;
261 }