Updated to fedora-glibc-20071212T1051
[kopensolaris-gnu/glibc.git] / nscd / pwdcache.c
index 2354b6d..bc1b6ba 100644 (file)
@@ -1,22 +1,21 @@
 /* Cache handling for passwd lookup.
-   Copyright (C) 1998-2002, 2003, 2004 Free Software Foundation, Inc.
+   Copyright (C) 1998-2005, 2006, 2007 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
 
-   The GNU C Library is free software; you can redistribute it and/or
-   modify it under the terms of the GNU Lesser General Public
-   License as published by the Free Software Foundation; either
-   version 2.1 of the License, or (at your option) any later version.
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published
+   by the Free Software Foundation; version 2 of the License, or
+   (at your option) any later version.
 
-   The GNU C Library is distributed in the hope that it will be useful,
+   This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
-   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-   Lesser General Public License for more details.
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
 
-   You should have received a copy of the GNU Lesser General Public
-   License along with the GNU C Library; if not, write to the Free
-   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
-   02111-1307 USA.  */
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software Foundation,
+   Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
 
 #include <alloca.h>
 #include <assert.h>
 #include <time.h>
 #include <unistd.h>
 #include <sys/mman.h>
+#include <sys/socket.h>
 #include <stackinfo.h>
 
 #include "nscd.h"
 #include "dbg_log.h"
+#ifdef HAVE_SENDFILE
+# include <kernel-features.h>
+#endif
 
 /* This is the standard reply in case the service is disabled.  */
 static const pw_response_header disabled =
@@ -114,7 +117,8 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
          written = total = sizeof (notfound);
 
          if (fd != -1)
-           written = TEMP_FAILURE_RETRY (write (fd, &notfound, total));
+           written = TEMP_FAILURE_RETRY (send (fd, &notfound, total,
+                                               MSG_NOSIGNAL));
 
          dataset = mempool_alloc (db, sizeof (struct dataset) + req->key_len);
          /* If we cannot permanently store the result, so be it.  */
@@ -135,6 +139,16 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
              /* Copy the key data.  */
              char *key_copy = memcpy (dataset->strdata, key, req->key_len);
 
+             /* If necessary, we also propagate the data to disk.  */
+             if (db->persistent)
+               {
+                 // XXX async OK?
+                 uintptr_t pval = (uintptr_t) dataset & ~pagesize_m1;
+                 msync ((void *) pval,
+                        ((uintptr_t) dataset & pagesize_m1)
+                        + sizeof (struct dataset) + req->key_len, MS_ASYNC);
+               }
+
              /* Now get the lock to safely insert the records.  */
              pthread_rwlock_rdlock (&db->lock);
 
@@ -245,7 +259,7 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
              && memcmp (&dataset->resp, dh->data,
                         dh->allocsize - offsetof (struct dataset, resp)) == 0)
            {
-             /* The sata has not changed.  We will just bump the
+             /* The data has not changed.  We will just bump the
                 timeout value.  Note that the new record has been
                 allocated on the stack and need not be freed.  */
              dh->timeout = dataset->head.timeout;
@@ -261,10 +275,13 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
                {
                  /* Adjust pointer into the memory block.  */
                  cp = (char *) newp + (cp - (char *) dataset);
+                 key_copy = (char *) newp + (key_copy - (char *) dataset);
 
                  dataset = memcpy (newp, dataset, total + n);
                  alloca_used = false;
                }
+             else
+               ++db->head->addfailed;
 
              /* Mark the old record as obsolete.  */
              dh->usable = false;
@@ -277,7 +294,30 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
             unnecessarily let the receiver wait.  */
          assert (fd != -1);
 
-         written = TEMP_FAILURE_RETRY (write (fd, &dataset->resp, total));
+#ifdef HAVE_SENDFILE
+         if (__builtin_expect (db->mmap_used, 1) && !alloca_used)
+           {
+             assert (db->wr_fd != -1);
+             assert ((char *) &dataset->resp > (char *) db->data);
+             assert ((char *) &dataset->resp - (char *) db->head
+                     + total
+                     <= (sizeof (struct database_pers_head)
+                          + db->head->module * sizeof (ref_t)
+                          + db->head->data_size));
+             written = sendfileall (fd, db->wr_fd,
+                                    (char *) &dataset->resp
+                                    - (char *) db->head, total);
+# ifndef __ASSUME_SENDFILE
+             if (written == -1 && errno == ENOSYS)
+               goto use_write;
+# endif
+           }
+         else
+# ifndef __ASSUME_SENDFILE
+         use_write:
+# endif
+#endif
+           written = writeall (fd, &dataset->resp, total);
        }
 
 
@@ -302,10 +342,10 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
             marked with FIRST first.  Otherwise we end up with
             dangling "pointers" in case a latter hash entry cannot be
             added.  */
-         bool first = req->type == GETPWBYNAME;
+         bool first = true;
 
          /* If the request was by UID, add that entry first.  */
-         if (req->type != GETPWBYNAME)
+         if (req->type == GETPWBYUID)
            {
              if (cache_add (GETPWBYUID, cp, key_offset, &dataset->head, true,
                             db, owner) < 0)
@@ -315,12 +355,14 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
                  dataset->head.usable = false;
                  goto out;
                }
+
+             first = false;
            }
          /* If the key is different from the name add a separate entry.  */
          else if (strcmp (key_copy, dataset->strdata) != 0)
            {
              if (cache_add (GETPWBYNAME, key_copy, key_len + 1,
-                            &dataset->head, first, db, owner) < 0)
+                            &dataset->head, true, db, owner) < 0)
                {
                  /* Could not allocate memory.  Make sure the data gets
                     discarded.  */
@@ -332,11 +374,12 @@ cache_addpw (struct database_dyn *db, int fd, request_header *req,
            }
 
          /* We have to add the value for both, byname and byuid.  */
-         if (__builtin_expect (cache_add (GETPWBYNAME, dataset->strdata,
-                                          pw_name_len, &dataset->head, first,
-                                          db, owner) == 0, 1))
+         if ((req->type == GETPWBYNAME || db->propagate)
+             && __builtin_expect (cache_add (GETPWBYNAME, dataset->strdata,
+                                             pw_name_len, &dataset->head,
+                                             first, db, owner) == 0, 1))
            {
-             if (req->type == GETPWBYNAME)
+             if (req->type == GETPWBYNAME && db->propagate)
                (void) cache_add (GETPWBYUID, cp, key_offset, &dataset->head,
                                  req->type != GETPWBYNAME, db, owner);
            }
@@ -390,7 +433,6 @@ addpwbyX (struct database_dyn *db, int fd, request_header *req,
   char *buffer = (char *) alloca (buflen);
   struct passwd resultbuf;
   struct passwd *pwd;
-  uid_t oldeuid = 0;
   bool use_malloc = false;
   int errval = 0;
 
@@ -402,22 +444,15 @@ addpwbyX (struct database_dyn *db, int fd, request_header *req,
        dbg_log (_("Reloading \"%s\" in password cache!"), keystr);
     }
 
-  if (db->secure)
-    {
-      oldeuid = geteuid ();
-      seteuid (c_uid);
-    }
-
   while (lookup (req->type, key, &resultbuf, buffer, buflen, &pwd) != 0
         && (errval = errno) == ERANGE)
     {
-      char *old_buffer = buffer;
       errno = 0;
-#define INCR 1024
 
       if (__builtin_expect (buflen > 32768, 0))
        {
-         buflen += INCR;
+         char *old_buffer = buffer;
+         buflen *= 2;
          buffer = (char *) realloc (use_malloc ? buffer : NULL, buflen);
          if (buffer == NULL)
            {
@@ -438,12 +473,9 @@ addpwbyX (struct database_dyn *db, int fd, request_header *req,
       else
        /* Allocate a new buffer on the stack.  If possible combine it
           with the previously allocated buffer.  */
-       buffer = (char *) extend_alloca (buffer, buflen, buflen + INCR);
+       buffer = (char *) extend_alloca (buffer, buflen, 2 * buflen);
     }
 
-  if (db->secure)
-    seteuid (oldeuid);
-
   /* Add the entry to the cache.  */
   cache_addpw (db, fd, req, keystr, pwd, c_uid, he, dh, errval);