2d09ed92f245038c97a1f923d730da78a0914ff3
[kopensolaris-gnu/glibc.git] / nscd / nscd_getgr_r.c
1 /* Copyright (C) 1998, 1999, 2000, 2002, 2003 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 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 <errno.h>
21 #include <grp.h>
22 #include <stdint.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <unistd.h>
27 #include <sys/socket.h>
28 #include <sys/uio.h>
29 #include <sys/un.h>
30
31 #include "nscd-client.h"
32 #include "nscd_proto.h"
33
34 int __nss_not_use_nscd_group;
35
36 static int nscd_getgr_r (const char *key, size_t keylen, request_type type,
37                          struct group *resultbuf, char *buffer,
38                          size_t buflen) internal_function;
39
40
41 int
42 __nscd_getgrnam_r (const char *name, struct group *resultbuf, char *buffer,
43                    size_t buflen)
44 {
45   return nscd_getgr_r (name, strlen (name) + 1, GETGRBYNAME, resultbuf,
46                        buffer, buflen);
47 }
48
49
50 int
51 __nscd_getgrgid_r (gid_t gid, struct group *resultbuf, char *buffer,
52                    size_t buflen)
53 {
54   char buf[12];
55   size_t n;
56
57   n = __snprintf (buf, sizeof (buf), "%d", gid) + 1;
58
59   return nscd_getgr_r (buf, n, GETGRBYGID, resultbuf, buffer, buflen);
60 }
61
62
63 /* Create a socket connected to a name. */
64 static int
65 open_socket (void)
66 {
67   struct sockaddr_un addr;
68   int sock;
69   int saved_errno = errno;
70
71   sock = __socket (PF_UNIX, SOCK_STREAM, 0);
72   if (sock < 0)
73     {
74       __set_errno (saved_errno);
75       return -1;
76     }
77
78   addr.sun_family = AF_UNIX;
79   strcpy (addr.sun_path, _PATH_NSCDSOCKET);
80   if (__connect (sock, (struct sockaddr *) &addr, sizeof (addr)) < 0)
81     {
82       __close (sock);
83       __set_errno (saved_errno);
84       return -1;
85     }
86
87   return sock;
88 }
89
90
91 static int
92 internal_function
93 nscd_getgr_r (const char *key, size_t keylen, request_type type,
94               struct group *resultbuf, char *buffer, size_t buflen)
95 {
96   int sock = open_socket ();
97   request_header req;
98   gr_response_header gr_resp;
99   ssize_t nbytes;
100   struct iovec vec[2];
101
102   if (sock == -1)
103     {
104       __nss_not_use_nscd_group = 1;
105       return -1;
106     }
107
108   req.version = NSCD_VERSION;
109   req.type = type;
110   req.key_len = keylen;
111
112   vec[0].iov_base = &req;
113   vec[0].iov_len = sizeof (request_header);
114   vec[1].iov_base = (void *) key;
115   vec[1].iov_len = keylen;
116
117   nbytes = TEMP_FAILURE_RETRY (__writev (sock, vec, 2));
118   if (nbytes != (ssize_t) (sizeof (request_header) + keylen))
119     {
120       __close (sock);
121       return -1;
122     }
123
124   nbytes = TEMP_FAILURE_RETRY (__read (sock, &gr_resp,
125                                        sizeof (gr_response_header)));
126   if (nbytes != (ssize_t) sizeof (gr_response_header))
127     {
128       __close (sock);
129       return -1;
130     }
131
132   if (gr_resp.found == -1)
133     {
134       /* The daemon does not cache this database.  */
135       __close (sock);
136       __nss_not_use_nscd_group = 1;
137       return -1;
138     }
139
140   if (gr_resp.found == 1)
141     {
142       uint32_t *len;
143       char *p = buffer;
144       size_t total_len;
145       uintptr_t align;
146       size_t cnt;
147
148       /* Now allocate the buffer the array for the group members.  We must
149          align the pointer.  */
150       align = ((__alignof__ (char *) - (p - ((char *) 0)))
151                & (__alignof__ (char *) - 1));
152       total_len = align + (1 + gr_resp.gr_mem_cnt) * sizeof (char *)
153                   + gr_resp.gr_name_len + gr_resp.gr_passwd_len;
154       if (buflen < total_len)
155         {
156         no_room:
157           __set_errno (ERANGE);
158           __close (sock);
159           return ERANGE;
160         }
161       buflen -= total_len;
162
163       p += align;
164       resultbuf->gr_mem = (char **) p;
165       p += (1 + gr_resp.gr_mem_cnt) * sizeof (char *);
166
167       /* Set pointers for strings.  */
168       resultbuf->gr_name = p;
169       p += gr_resp.gr_name_len;
170       resultbuf->gr_passwd = p;
171       p += gr_resp.gr_passwd_len;
172
173       /* Fill in what we know now.  */
174       resultbuf->gr_gid = gr_resp.gr_gid;
175
176       /* Allocate array to store lengths.  */
177       len = (uint32_t *) alloca (gr_resp.gr_mem_cnt * sizeof (uint32_t));
178
179       total_len = gr_resp.gr_mem_cnt * sizeof (uint32_t);
180       vec[0].iov_base = len;
181       vec[0].iov_len = total_len;
182       vec[1].iov_base = resultbuf->gr_name;
183       vec[1].iov_len = gr_resp.gr_name_len + gr_resp.gr_passwd_len;
184       total_len += gr_resp.gr_name_len + gr_resp.gr_passwd_len;
185
186       /* Get this data.  */
187       if ((size_t) TEMP_FAILURE_RETRY (__readv (sock, vec, 2)) != total_len)
188         {
189           __close (sock);
190           return -1;
191         }
192
193       /* Clear the terminating entry.  */
194       resultbuf->gr_mem[gr_resp.gr_mem_cnt] = NULL;
195
196       /* Prepare reading the group members.  */
197       total_len = 0;
198       for (cnt = 0; cnt < gr_resp.gr_mem_cnt; ++cnt)
199         {
200           resultbuf->gr_mem[cnt] = p;
201           total_len += len[cnt];
202           p += len[cnt];
203         }
204
205       if (total_len > buflen)
206         goto no_room;
207
208       if ((size_t) TEMP_FAILURE_RETRY (__read (sock, resultbuf->gr_mem[0],
209                                                total_len)) != total_len)
210         {
211           __close (sock);
212           /* The `errno' to some value != ERANGE.  */
213           __set_errno (ENOENT);
214           return ENOENT;
215         }
216
217       __close (sock);
218       return 0;
219     }
220   else
221     {
222       __close (sock);
223       /* The `errno' to some value != ERANGE.  */
224       __set_errno (ENOENT);
225       return ENOENT;
226     }
227 }