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