Use __readall and __readvall.
[kopensolaris-gnu/glibc.git] / nscd / nscd_helper.c
1 /* Copyright (C) 1998-2002, 2003, 2004, 2005 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Ulrich Drepper <drepper@cygnus.com>, 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 <assert.h>
21 #include <errno.h>
22 #include <fcntl.h>
23 #include <stdbool.h>
24 #include <unistd.h>
25 #include <sys/mman.h>
26 #include <sys/poll.h>
27 #include <sys/socket.h>
28 #include <sys/stat.h>
29 #include <sys/uio.h>
30 #include <sys/un.h>
31 #include <not-cancel.h>
32 #include <nis/rpcsvc/nis.h>
33
34 #include "nscd-client.h"
35
36
37 ssize_t
38 __readall (int fd, void *buf, size_t len)
39 {
40   size_t n = len;
41   ssize_t ret;
42   do
43     {
44       ret = TEMP_FAILURE_RETRY (__read (fd, buf, n));
45       if (ret <= 0)
46         break;
47       buf = (char *) buf + ret;
48       n -= ret;
49     }
50   while (n > 0);
51   return ret < 0 ? ret : len - n;
52 }
53
54
55 ssize_t
56 __readvall (int fd, const struct iovec *iov, int iovcnt)
57 {
58   ssize_t ret = TEMP_FAILURE_RETRY (__readv (fd, iov, iovcnt));
59   if (ret <= 0)
60     return ret;
61
62   size_t total = 0;
63   for (int i = 0; i < iovcnt; ++i)
64     total += iov[i].iov_len;
65
66   if (ret < total)
67     {
68       struct iovec iov_buf[iovcnt];
69       ssize_t r = ret;
70
71       struct iovec *iovp = memcpy (iov_buf, iov, iovcnt * sizeof (*iov));
72       do
73         {
74           while (iovp->iov_len <= r)
75             {
76               r -= iovp->iov_len;
77               --iovcnt;
78               ++iovp;
79             }
80           iovp->iov_base = (char *) iovp->iov_base + r;
81           iovp->iov_len -= r;
82           r = TEMP_FAILURE_RETRY (__readv (fd, iovp, iovcnt));
83           if (r <= 0)
84             break;
85           ret += r;
86         }
87       while (ret < total);
88       if (r < 0)
89         ret = r;
90     }
91   return ret;
92 }
93
94
95 static int
96 open_socket (void)
97 {
98   int sock = __socket (PF_UNIX, SOCK_STREAM, 0);
99   if (sock < 0)
100     return -1;
101
102   /* Make socket non-blocking.  */
103   int fl = __fcntl (sock, F_GETFL);
104   if (fl != -1)
105     __fcntl (sock, F_SETFL, fl | O_NONBLOCK);
106
107   struct sockaddr_un sun;
108   sun.sun_family = AF_UNIX;
109   strcpy (sun.sun_path, _PATH_NSCDSOCKET);
110   if (__connect (sock, (struct sockaddr *) &sun, sizeof (sun)) < 0
111       && errno != EINPROGRESS)
112     goto out;
113
114   struct pollfd fds[1];
115   fds[0].fd = sock;
116   fds[0].events = POLLOUT | POLLERR | POLLHUP;
117   if (__poll (fds, 1, 5 * 1000) > 0)
118     /* Success.  We do not check for success of the connect call here.
119        If it failed, the following operations will fail.  */
120     return sock;
121
122  out:
123   close_not_cancel_no_status (sock);
124
125   return -1;
126 }
127
128
129 void
130 __nscd_unmap (struct mapped_database *mapped)
131 {
132   assert (mapped->counter == 0);
133   __munmap ((void *) mapped->head, mapped->mapsize);
134   free (mapped);
135 }
136
137
138 /* Try to get a file descriptor for the shared meory segment
139    containing the database.  */
140 static struct mapped_database *
141 get_mapping (request_type type, const char *key,
142              struct mapped_database **mappedp)
143 {
144   struct mapped_database *result = NO_MAPPING;
145 #ifdef SCM_RIGHTS
146   const size_t keylen = strlen (key) + 1;
147   char resdata[keylen];
148   int saved_errno = errno;
149
150   int mapfd = -1;
151
152   /* Send the request.  */
153   struct iovec iov[2];
154   request_header req;
155
156   int sock = open_socket ();
157   if (sock < 0)
158     goto out;
159
160   req.version = NSCD_VERSION;
161   req.type = type;
162   req.key_len = keylen;
163
164   iov[0].iov_base = &req;
165   iov[0].iov_len = sizeof (req);
166   iov[1].iov_base = (void *) key;
167   iov[1].iov_len = keylen;
168
169   if (TEMP_FAILURE_RETRY (__writev (sock, iov, 2))
170       != iov[0].iov_len + iov[1].iov_len)
171     /* We cannot even write the request.  */
172     goto out_close2;
173
174   /* Room for the data sent along with the file descriptor.  We expect
175      the key name back.  */
176   iov[0].iov_base = resdata;
177   iov[0].iov_len = keylen;
178
179   char buf[CMSG_SPACE (sizeof (int))];
180   struct msghdr msg = { .msg_iov = iov, .msg_iovlen = 1,
181                         .msg_control = buf, .msg_controllen = sizeof (buf) };
182   struct cmsghdr *cmsg = CMSG_FIRSTHDR (&msg);
183
184   cmsg->cmsg_level = SOL_SOCKET;
185   cmsg->cmsg_type = SCM_RIGHTS;
186   cmsg->cmsg_len = CMSG_LEN (sizeof (int));
187
188   *(int *) CMSG_DATA (cmsg) = -1;
189
190   msg.msg_controllen = cmsg->cmsg_len;
191
192   struct pollfd fds[1];
193   fds[0].fd = sock;
194   fds[0].events = POLLIN | POLLERR | POLLHUP;
195   if (__poll (fds, 1, 5 * 1000) <= 0)
196     /* Failure or timeout.  */
197     goto out_close2;
198
199 #ifndef MSG_NOSIGNAL
200 # define MSG_NOSIGNAL 0
201 #endif
202   if (TEMP_FAILURE_RETRY (__recvmsg (sock, &msg, MSG_NOSIGNAL)) != keylen)
203     goto out_close2;
204
205   mapfd = *(int *) CMSG_DATA (cmsg);
206
207   if (CMSG_FIRSTHDR (&msg)->cmsg_len != CMSG_LEN (sizeof (int)))
208     goto out_close;
209
210   struct stat64 st;
211   if (strcmp (resdata, key) != 0
212       || fstat64 (mapfd, &st) != 0
213       || st.st_size < sizeof (struct database_pers_head))
214     goto out_close;
215
216   struct database_pers_head head;
217   if (TEMP_FAILURE_RETRY (__pread (mapfd, &head, sizeof (head), 0))
218       != sizeof (head))
219     goto out_close;
220
221   if (head.version != DB_VERSION || head.header_size != sizeof (head)
222       /* This really should not happen but who knows, maybe the update
223          thread got stuck.  */
224       || (! head.nscd_certainly_running
225           && head.timestamp + MAPPING_TIMEOUT < time (NULL)))
226     goto out_close;
227
228   size_t size = (sizeof (head) + roundup (head.module * sizeof (ref_t), ALIGN)
229                  + head.data_size);
230
231   if (st.st_size < size)
232     goto out_close;
233
234   /* The file is large enough, map it now.  */
235   void *mapping = __mmap (NULL, size, PROT_READ, MAP_SHARED, mapfd, 0);
236   if (mapping != MAP_FAILED)
237     {
238       /* Allocate a record for the mapping.  */
239       struct mapped_database *newp;
240
241       newp = malloc (sizeof (*newp));
242       if (newp == NULL)
243         {
244           /* Ugh, after all we went through the memory allocation failed.  */
245           __munmap (result, size);
246           goto out_close;
247         }
248
249       newp->head = mapping;
250       newp->data = ((char *) mapping + head.header_size
251                     + roundup (head.module * sizeof (ref_t), ALIGN));
252       newp->mapsize = size;
253       /* Set counter to 1 to show it is usable.  */
254       newp->counter = 1;
255
256       result = newp;
257     }
258
259  out_close:
260   __close (mapfd);
261  out_close2:
262   __close (sock);
263  out:
264   __set_errno (saved_errno);
265 #endif  /* SCM_RIGHTS */
266
267   struct mapped_database *oldval = *mappedp;
268   *mappedp = result;
269
270   if (oldval != NULL && atomic_decrement_val (&oldval->counter) == 0)
271     __nscd_unmap (oldval);
272
273   return result;
274 }
275
276
277 struct mapped_database *
278 __nscd_get_map_ref (request_type type, const char *name,
279                     struct locked_map_ptr *mapptr, int *gc_cyclep)
280 {
281   struct mapped_database *cur = mapptr->mapped;
282   if (cur == NO_MAPPING)
283     return cur;
284
285   int cnt = 0;
286   while (atomic_compare_and_exchange_val_acq (&mapptr->lock, 1, 0) != 0)
287     {
288       // XXX Best number of rounds?
289       if (++cnt > 5)
290         return NO_MAPPING;
291
292       atomic_delay ();
293     }
294
295   cur = mapptr->mapped;
296
297   if (__builtin_expect (cur != NO_MAPPING, 1))
298     {
299       /* If not mapped or timestamp not updated, request new map.  */
300       if (cur == NULL
301           || (cur->head->nscd_certainly_running == 0
302               && cur->head->timestamp + MAPPING_TIMEOUT < time (NULL)))
303         cur = get_mapping (type, name, &mapptr->mapped);
304
305       if (__builtin_expect (cur != NO_MAPPING, 1))
306         {
307           if (__builtin_expect (((*gc_cyclep = cur->head->gc_cycle) & 1) != 0,
308                                 0))
309             cur = NO_MAPPING;
310           else
311             atomic_increment (&cur->counter);
312         }
313     }
314
315   mapptr->lock = 0;
316
317   return cur;
318 }
319
320
321 const struct datahead *
322 __nscd_cache_search (request_type type, const char *key, size_t keylen,
323                      const struct mapped_database *mapped)
324 {
325   unsigned long int hash = __nis_hash (key, keylen) % mapped->head->module;
326
327   ref_t work = mapped->head->array[hash];
328   while (work != ENDREF)
329     {
330       struct hashentry *here = (struct hashentry *) (mapped->data + work);
331
332       if (type == here->type && keylen == here->len
333           && memcmp (key, mapped->data + here->key, keylen) == 0)
334         {
335           /* We found the entry.  Increment the appropriate counter.  */
336           const struct datahead *dh
337             = (struct datahead *) (mapped->data + here->packet);
338
339           /* See whether we must ignore the entry or whether something
340              is wrong because garbage collection is in progress.  */
341           if (dh->usable && ((char *) dh + dh->allocsize
342                              <= (char *) mapped->head + mapped->mapsize))
343             return dh;
344         }
345
346       work = here->next;
347     }
348
349   return NULL;
350 }
351
352
353 /* Create a socket connected to a name. */
354 int
355 __nscd_open_socket (const char *key, size_t keylen, request_type type,
356                     void *response, size_t responselen)
357 {
358   int saved_errno = errno;
359
360   int sock = open_socket ();
361   if (sock >= 0)
362     {
363       request_header req;
364       req.version = NSCD_VERSION;
365       req.type = type;
366       req.key_len = keylen;
367
368       struct iovec vec[2];
369       vec[0].iov_base = &req;
370       vec[0].iov_len = sizeof (request_header);
371       vec[1].iov_base = (void *) key;
372       vec[1].iov_len = keylen;
373
374       ssize_t nbytes = TEMP_FAILURE_RETRY (__writev (sock, vec, 2));
375       if (nbytes == (ssize_t) (sizeof (request_header) + keylen))
376         {
377           /* Wait for data.  */
378           struct pollfd fds[1];
379           fds[0].fd = sock;
380           fds[0].events = POLLIN | POLLERR | POLLHUP;
381           if (__poll (fds, 1, 5 * 1000) > 0)
382             {
383               nbytes = TEMP_FAILURE_RETRY (__read (sock, response,
384                                                    responselen));
385               if (nbytes == (ssize_t) responselen)
386                 return sock;
387             }
388         }
389
390       close_not_cancel_no_status (sock);
391     }
392
393   __set_errno (saved_errno);
394
395   return -1;
396 }