(__nscd_getgr_r): Adjust for protocol change. We now can do the job
[kopensolaris-gnu/glibc.git] / nscd / nscd_getgr_r.c
1 /* Copyright (C) 1998 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Thorsten Kukuk <kukuk@uni-paderborn.de>, 1998.
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 <errno.h>
21 #include <grp.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26 #include <sys/socket.h>
27 #include <sys/uio.h>
28 #include <sys/un.h>
29
30 #include "nscd.h"
31 #include "nscd_proto.h"
32
33 int __nss_not_use_nscd_group;
34
35 static int __nscd_getgr_r (const char *key, request_type type,
36                            struct group *resultbuf, char *buffer,
37                            size_t buflen);
38
39 int
40 __nscd_getgrnam_r (const char *name, struct group *resultbuf, char *buffer,
41                    size_t buflen)
42 {
43   if (name == NULL)
44     return 1;
45
46   return __nscd_getgr_r (name, GETGRBYNAME, resultbuf, buffer, buflen);
47 }
48
49 int
50 __nscd_getgrgid_r (gid_t gid, struct group *resultbuf, char *buffer,
51                    size_t buflen)
52 {
53   char *p = buffer;
54   int plen;
55
56   plen = __snprintf (buffer, buflen, "%d", gid);
57   if (plen == -1)
58     {
59       __set_errno (ERANGE);
60       return -1;
61     }
62   p = buffer + plen + 1;
63
64   return __nscd_getgr_r (buffer, GETGRBYGID, resultbuf, p, buflen - plen -1);
65 }
66
67 /* Create a socket connected to a name. */
68 static int
69 nscd_open_socket (void)
70 {
71   struct sockaddr_un addr;
72   int sock;
73   int saved_errno = errno;
74
75   sock = __socket (PF_UNIX, SOCK_STREAM, 0);
76   if (sock < 0)
77     {
78       __set_errno (saved_errno);
79       return -1;
80     }
81
82   addr.sun_family = AF_UNIX;
83   strcpy (addr.sun_path, _PATH_NSCDSOCKET);
84   if (__connect (sock, (struct sockaddr *) &addr, sizeof (addr)) < 0)
85     {
86       __close (sock);
87       __set_errno (saved_errno);
88       return -1;
89     }
90
91   return sock;
92 }
93
94 static int
95 __nscd_getgr_r (const char *key, request_type type, struct group *resultbuf,
96                 char *buffer, size_t buflen)
97 {
98   int sock = nscd_open_socket ();
99   request_header req;
100   gr_response_header gr_resp;
101   ssize_t nbytes;
102
103   if (sock == -1)
104     {
105       __nss_not_use_nscd_group = 1;
106       return 1;
107     }
108
109   req.version = NSCD_VERSION;
110   req.type = type;
111   req.key_len = strlen (key);
112   nbytes = __write (sock, &req, sizeof (request_header));
113   if (nbytes != sizeof (request_header))
114     {
115       __close (sock);
116       return 1;
117     }
118
119   nbytes = __write (sock, key, req.key_len);
120   if (nbytes != req.key_len)
121     {
122       __close (sock);
123       return 1;
124     }
125
126   nbytes = __read (sock, &gr_resp, sizeof (gr_response_header));
127   if (nbytes != sizeof (gr_response_header))
128     {
129       __close (sock);
130       return 1;
131     }
132
133   if (gr_resp.found == -1)
134     {
135       /* The daemon does not cache this database.  */
136       __close (sock);
137       __nss_not_use_nscd_group = 1;
138       return 1;
139     }
140
141   if (gr_resp.found == 1)
142     {
143       struct iovec *vec;
144       size_t *len;
145       char *p = buffer;
146       int nblocks;
147       size_t total_len;
148       uintptr_t align;
149
150       /* A first check whether the buffer is sufficently large is possible.  */
151       if (buflen < gr_resp.gr_name_len + 1 + gr_resp.gr_passwd_len + 1)
152         {
153           __set_errno (ERANGE);
154           __close (sock);
155           return -1;
156         }
157
158       /* Allocate the IOVEC.  */
159       vec = alloca ((2 + gr_resp.gr_mem_len) * sizeof (struct iovec));
160       len = alloca (gr_resp.gr_mem_len * sizeof (size_t));
161
162       vec[0].iov_base = resultbuf->gr_name = p;
163       vec[0].iov_len = gr_resp.gr_name_len;
164       total_len = gr_resp.gr_name_len;
165       p += gr_resp.gr_name_len + 1;
166
167       vec[1].iov_base = resultbuf->gr_passwd = p;
168       vec[1].iov_len = gr_resp.gr_passwd_len;
169       total_len += gr_resp.gr_passwd_len;
170       p += gr_resp.gr_passwd_len + 1;
171       buflen -= total_len;
172       nblocks = 2;
173
174       if (gr_resp.gr_mem_len > 0)
175         {
176           vec[2].iov_base = len;
177           vec[2].iov_len = gr_resp.gr_mem_len * sizeof (size_t);
178           total_len += gr_resp.gr_mem_len * sizeof (size_t);
179           nblocks = 3;
180         }
181
182       /* Get this data.  */
183       if (__readv (sock, vec, nblocks) != total_len)
184         {
185           __close (sock);
186           return 1;
187         }
188
189       /* Now we know the sizes.  First terminate the strings we just read. */
190       resultbuf->gr_name[gr_resp.gr_name_len] = '\0';
191       resultbuf->gr_passwd[gr_resp.gr_passwd_len] = '\0';
192
193       resultbuf->gr_gid = gr_resp.gr_gid;
194
195       /* Now allocate the buffer the array for the group members.  We must
196          align the pointer.  */
197       align = ((__alignof__ (char *) - (p - ((char *) 0)))
198                & (__alignof__ (char *) - 1));
199       if (align + (1 + gr_resp.gr_mem_len) * sizeof (char *) > buflen)
200         {
201           __set_errno (ERANGE);
202           __close (sock);
203           return -1;
204         }
205       p += align;
206       resultbuf->gr_mem = (char **) p;
207       p += (1 + gr_resp.gr_mem_len) * sizeof (char *);
208       buflen -= align + (1 + gr_resp.gr_mem_len) * sizeof (char *);
209
210       resultbuf->gr_mem[gr_resp.gr_mem_len] = NULL;
211
212       if (gr_resp.gr_mem_len > 0)
213         {
214           /* Prepare reading the group members.  */
215           size_t i;
216
217           total_len = 0;
218           for (i = 0; i < gr_resp.gr_mem_len; ++i)
219             {
220               if (len[i] >= buflen)
221                 {
222                   __set_errno (ERANGE);
223                   __close (sock);
224                   return -1;
225                 }
226
227               vec[i].iov_base = resultbuf->gr_mem[i] = p;
228               vec[i].iov_len = len[i];
229               total_len += len[i];
230               buflen -= len[i];
231               p += len[i];
232               *p++ = '\0';
233             }
234
235           if (__readv (sock, vec, gr_resp.gr_mem_len) != total_len)
236             {
237               __close (sock);
238               return -1;
239             }
240         }
241       __close (sock);
242       return 0;
243     }
244   else
245     {
246       __close (sock);
247       return -1;
248     }
249 }