Use _CALL_DL_FCT.
[kopensolaris-gnu/glibc.git] / grp / initgroups.c
1 /* Copyright (C) 1989, 91, 93, 96, 97, 98 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3
4    The GNU C Library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Library General Public License as
6    published by the Free Software Foundation; either version 2 of the
7    License, or (at your option) any later version.
8
9    The GNU C Library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Library General Public License for more details.
13
14    You should have received a copy of the GNU Library General Public
15    License along with the GNU C Library; see the file COPYING.LIB.  If not,
16    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17    Boston, MA 02111-1307, USA.  */
18
19 #include <alloca.h>
20 #include <errno.h>
21 #include <grp.h>
22 #include <limits.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <sys/types.h>
26 #include <nsswitch.h>
27
28
29 /* Type of the lookup function.  */
30 typedef enum nss_status (*initgroups_function) (const char *, gid_t,
31                                                 long int *, long int *,
32                                                 gid_t *, long int, int *);
33 /* Prototype for the setgrent functions we use here.  */
34 typedef enum nss_status (*set_function) (void);
35
36 /* Prototype for the endgrent functions we use here.  */
37 typedef enum nss_status (*end_function) (void);
38
39 /* Prototype for the setgrent functions we use here.  */
40 typedef enum nss_status (*get_function) (struct group *, char *,
41                                          size_t, int *);
42
43 /* The lookup function for the first entry of this service.  */
44 extern int __nss_group_lookup (service_user **nip, const char *name,
45                                    void **fctp);
46 extern void *__nss_lookup_function (service_user *ni, const char *fct_name);
47
48 extern service_user *__nss_group_database;
49
50 static enum nss_status
51 compat_call (service_user *nip, const char *user, gid_t group, long int *start,
52              long int *size, gid_t *groups, long int limit, int *errnop)
53 {
54   struct group grpbuf, *g;
55   size_t buflen = sysconf (_SC_GETPW_R_SIZE_MAX);
56   char *tmpbuf;
57   enum nss_status status;
58   set_function setgrent_fct;
59   get_function getgrent_fct;
60   end_function endgrent_fct;
61
62   setgrent_fct = __nss_lookup_function (nip, "setgrent");
63   status = _CALL_DL_FCT (setgrent_fct, ());
64   if (status != NSS_STATUS_SUCCESS)
65     return status;
66
67   getgrent_fct = __nss_lookup_function (nip, "getgrent_r");
68   endgrent_fct = __nss_lookup_function (nip, "endgrent");
69
70   tmpbuf = __alloca (buflen);
71
72   do
73     {
74       while ((status = _CALL_DL_FCT (getgrent_fct,
75                                      (&grpbuf, tmpbuf, buflen, errnop)),
76               status == NSS_STATUS_TRYAGAIN)
77              && *errnop == ERANGE)
78         {
79           buflen *= 2;
80           tmpbuf = __alloca (buflen);
81         }
82
83       if (status != NSS_STATUS_SUCCESS)
84         goto done;
85
86       g = &grpbuf;
87       if (g->gr_gid != group)
88         {
89           char **m;
90
91           for (m = g->gr_mem; *m != NULL; ++m)
92             if (strcmp (*m, user) == 0)
93               {
94                 /* Matches user.  Insert this group.  */
95                 if (*start == *size && limit <= 0)
96                   {
97                     /* Need a bigger buffer.  */
98                     groups = realloc (groups, *size * sizeof (*groups));
99                     if (groups == NULL)
100                       goto done;
101                     *size *= 2;
102                   }
103
104                 groups[*start] = g->gr_gid;
105                 *start += 1;
106
107                 if (*start == limit)
108                   /* Can't take any more groups; stop searching.  */
109                   goto done;
110
111                 break;
112               }
113         }
114     }
115   while (status == NSS_STATUS_SUCCESS);
116
117  done:
118   _CALL_DL_FCT (endgrent_fct, ());
119
120   return NSS_STATUS_SUCCESS;
121 }
122
123 /* Initialize the group set for the current user
124    by reading the group database and using all groups
125    of which USER is a member.  Also include GROUP.  */
126 int
127 initgroups (user, group)
128      const char *user;
129      gid_t group;
130 {
131 #if defined NGROUPS_MAX && NGROUPS_MAX == 0
132
133   /* No extra groups allowed.  */
134   return 0;
135
136 #else
137
138   service_user *nip = NULL;
139   initgroups_function fct;
140   enum nss_status status = NSS_STATUS_UNAVAIL;
141   int no_more;
142   /* Start is one, because we have the first group as parameter.  */
143   long int start = 1;
144   long int size;
145   gid_t *groups;
146 #ifdef NGROUPS_MAX
147 # define limit NGROUPS_MAX
148
149   size = limit;
150 #else
151   long int limit = sysconf (_SC_NGROUPS_MAX);
152
153   if (limit > 0)
154     size = limit;
155   else
156     /* No fixed limit on groups.  Pick a starting buffer size.  */
157     size = 16;
158 #endif
159
160   groups = malloc (size * sizeof (gid_t *));
161
162   groups[0] = group;
163
164   if (__nss_group_database != NULL)
165     {
166       no_more = 0;
167       nip = __nss_group_database;
168     }
169   else
170     no_more = __nss_database_lookup ("group", NULL,
171                                      "compat [NOTFOUND=return] files", &nip);
172
173   while (! no_more)
174     {
175       fct = __nss_lookup_function (nip, "initgroups");
176
177       if (fct == NULL)
178         status = compat_call (nip, user, group, &start, &size, groups,
179                               limit, &errno);
180       else
181         status = _CALL_DL_FCT (fct, (user, group, &start, &size, groups, limit,
182                                      &errno));
183
184       if (nip->next == NULL)
185         no_more = -1;
186       else
187         nip = nip->next;
188     }
189
190   return setgroups (start, groups);
191 #endif
192 }