Cache network interface information
[platform/upstream/glibc.git] / nscd / hstcache.c
index 3ceb671..4d68ade 100644 (file)
@@ -1,5 +1,5 @@
 /* Cache handling for host lookup.
-   Copyright (C) 1998-2005, 2006, 2007, 2008 Free Software Foundation, Inc.
+   Copyright (C) 1998-2008, 2009, 2011 Free Software Foundation, Inc.
    This file is part of the GNU C Library.
    Contributed by Ulrich Drepper <drepper@cygnus.com>, 1998.
 
@@ -77,14 +77,27 @@ static const hst_response_header notfound =
 };
 
 
-static void
+/* This is the standard reply in case there are temporary problems.  */
+static const hst_response_header tryagain =
+{
+  .version = NSCD_VERSION,
+  .found = 0,
+  .h_name_len = 0,
+  .h_aliases_cnt = 0,
+  .h_addrtype = -1,
+  .h_length = -1,
+  .h_addr_list_cnt = 0,
+  .error = TRY_AGAIN
+};
+
+
+static time_t
 cache_addhst (struct database_dyn *db, int fd, request_header *req,
              const void *key, struct hostent *hst, uid_t owner,
-             struct hashentry *he, struct datahead *dh, int errval,
+             struct hashentry *const he, struct datahead *dh, int errval,
              int32_t ttl)
 {
-  ssize_t total;
-  ssize_t written;
+  bool all_written = true;
   time_t t = time (NULL);
 
   /* We allocate all data in one memory block: the iov vector,
@@ -98,6 +111,7 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
 
   assert (offsetof (struct dataset, resp) == offsetof (struct datahead, data));
 
+  time_t timeout = MAX_TIMEOUT_VALUE;
   if (hst == NULL)
     {
       if (he != NULL && errval == EAGAIN)
@@ -109,22 +123,33 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
            /* Do not reset the value if we never not reload the record.  */
            dh->nreloads = reload_count - 1;
 
-         written = total = 0;
+         /* Reload with the same time-to-live value.  */
+         timeout = dh->timeout = t + dh->ttl;
        }
       else
        {
          /* We have no data.  This means we send the standard reply for this
-            case.  */
-         written = total = sizeof (notfound);
+            case.  Possibly this is only temporary.  */
+         ssize_t total = sizeof (notfound);
+         assert (sizeof (notfound) == sizeof (tryagain));
 
-         if (fd != -1)
-           written = TEMP_FAILURE_RETRY (send (fd, &notfound, total,
-                                               MSG_NOSIGNAL));
+         const hst_response_header *resp = (errval == EAGAIN
+                                            ? &tryagain : &notfound);
+
+         if (fd != -1 &&
+             TEMP_FAILURE_RETRY (send (fd, resp, total,
+                                       MSG_NOSIGNAL)) != total)
+           all_written = false;
 
-         dataset = mempool_alloc (db, sizeof (struct dataset) + req->key_len,
-                                  IDX_result_data);
          /* If we cannot permanently store the result, so be it.  */
-         if (dataset != NULL)
+         if (__builtin_expect (db->negtimeout == 0, 0))
+           {
+             /* Mark the old entry as obsolete.  */
+             if (dh != NULL)
+               dh->usable = false;
+           }
+         else if ((dataset = mempool_alloc (db, (sizeof (struct dataset)
+                                                 + req->key_len), 1)) != NULL)
            {
              dataset->head.allocsize = sizeof (struct dataset) + req->key_len;
              dataset->head.recsize = total;
@@ -133,11 +158,11 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
              dataset->head.usable = true;
 
              /* Compute the timeout time.  */
-             dataset->head.timeout = t + (ttl == INT32_MAX
-                                          ? db->negtimeout : ttl);
+             dataset->head.ttl = ttl == INT32_MAX ? db->negtimeout : ttl;
+             timeout = dataset->head.timeout = t + dataset->head.ttl;
 
              /* This is the reply.  */
-             memcpy (&dataset->resp, &notfound, total);
+             memcpy (&dataset->resp, resp, total);
 
              /* Copy the key data.  */
              memcpy (dataset->strdata, key, req->key_len);
@@ -152,11 +177,8 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                         + sizeof (struct dataset) + req->key_len, MS_ASYNC);
                }
 
-             /* Now get the lock to safely insert the records.  */
-             pthread_rwlock_rdlock (&db->lock);
-
              (void) cache_add (req->type, &dataset->strdata, req->key_len,
-                               &dataset->head, true, db, owner);
+                               &dataset->head, true, db, owner, he == NULL);
 
              pthread_rwlock_unlock (&db->lock);
 
@@ -164,8 +186,6 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
              if (dh != NULL)
                dh->usable = false;
            }
-         else
-           ++db->head->addfailed;
        }
     }
   else
@@ -175,12 +195,12 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
       size_t h_aliases_cnt;
       uint32_t *h_aliases_len;
       size_t h_addr_list_cnt;
-      int addr_list_type;
       char *addresses;
       char *aliases;
       char *key_copy = NULL;
       char *cp;
       size_t cnt;
+      ssize_t total;
 
       /* Determine the number of aliases.  */
       h_aliases_cnt = 0;
@@ -202,13 +222,12 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
 
       if (h_addr_list_cnt == 0)
        /* Invalid entry.  */
-       return;
+       return MAX_TIMEOUT_VALUE;
 
       total += (sizeof (struct dataset)
                + h_name_len
                + h_aliases_cnt * sizeof (uint32_t)
                + h_addr_list_cnt * hst->h_length);
-      written = total;
 
       /* If we refill the cache, first assume the reconrd did not
         change.  Allocate memory on the cache since it is likely
@@ -223,13 +242,8 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
         questionable whether it is worthwhile complicating the cache
         handling just for handling such a special case. */
       if (he == NULL && h_addr_list_cnt == 1)
-       {
-         dataset = (struct dataset *) mempool_alloc (db,
-                                                     total + req->key_len,
-                                                     IDX_result_data);
-         if (dataset == NULL)
-           ++db->head->addfailed;
-       }
+       dataset = (struct dataset *) mempool_alloc (db, total + req->key_len,
+                                                   1);
 
       if (dataset == NULL)
        {
@@ -249,7 +263,8 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
       dataset->head.usable = true;
 
       /* Compute the timeout time.  */
-      dataset->head.timeout = t + (ttl == INT32_MAX ? db->postimeout : ttl);
+      dataset->head.ttl = ttl == INT32_MAX ? db->postimeout : ttl;
+      timeout = dataset->head.timeout = t + dataset->head.ttl;
 
       dataset->resp.version = NSCD_VERSION;
       dataset->resp.found = 1;
@@ -260,6 +275,9 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
       dataset->resp.h_addr_list_cnt = h_addr_list_cnt;
       dataset->resp.error = NETDB_SUCCESS;
 
+      /* Make sure there is no gap.  */
+      assert ((char *) (&dataset->resp.error + 1) == dataset->strdata);
+
       cp = dataset->strdata;
 
       cp = mempcpy (cp, hst->h_name, h_name_len);
@@ -286,6 +304,8 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
         we explicitly add the name here.  */
       key_copy = memcpy (cp, key, req->key_len);
 
+      assert ((char *) &dataset->resp + dataset->head.recsize == cp);
+
       /* Now we can determine whether on refill we have to create a new
         record or not.  */
       if (he != NULL)
@@ -301,6 +321,7 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                 timeout value.  Note that the new record has been
                 allocated on the stack and need not be freed.  */
              assert (h_addr_list_cnt == 1);
+             dh->ttl = dataset->head.ttl;
              dh->timeout = dataset->head.timeout;
              ++dh->nreloads;
            }
@@ -313,7 +334,7 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                  struct dataset *newp
                    = (struct dataset *) mempool_alloc (db,
                                                        total + req->key_len,
-                                                       IDX_result_data);
+                                                       1);
                  if (newp != NULL)
                    {
                      /* Adjust pointers into the memory block.  */
@@ -326,8 +347,6 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                      dataset = memcpy (newp, dataset, total + req->key_len);
                      alloca_used = false;
                    }
-                 else
-                   ++db->head->addfailed;
                }
 
              /* Mark the old record as obsolete.  */
@@ -346,25 +365,32 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
            {
              assert (db->wr_fd != -1);
              assert ((char *) &dataset->resp > (char *) db->data);
-             assert ((char *) &dataset->resp - (char *) db->head
+             assert ((char *) dataset - (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);
+             ssize_t written = sendfileall (fd, db->wr_fd,
+                                            (char *) &dataset->resp
+                                            - (char *) db->head,
+                                            dataset->head.recsize);
+             if (written != dataset->head.recsize)
+               {
 # ifndef __ASSUME_SENDFILE
-             if (written == -1 && errno == ENOSYS)
-               goto use_write;
+                 if (written == -1 && errno == ENOSYS)
+                   goto use_write;
 # endif
+                 all_written = false;
+               }
            }
          else
 # ifndef __ASSUME_SENDFILE
          use_write:
 # endif
 #endif
-           written = writeall (fd, &dataset->resp, total);
+           if (writeall (fd, &dataset->resp, dataset->head.recsize)
+               != dataset->head.recsize)
+             all_written = false;
        }
 
       /* Add the record to the database.  But only if it has not been
@@ -387,12 +413,6 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                     + total + req->key_len, MS_ASYNC);
            }
 
-         addr_list_type = (hst->h_length == NS_INADDRSZ
-                           ? GETHOSTBYADDR : GETHOSTBYADDRv6);
-
-         /* Now get the lock to safely insert the records.  */
-         pthread_rwlock_rdlock (&db->lock);
-
          /* NB: the following code is really complicated.  It has
             seemlingly duplicated code paths which do the same.  The
             problem is that we always must add the hash table entry
@@ -408,18 +428,20 @@ cache_addhst (struct database_dyn *db, int fd, request_header *req,
                  || req->type == GETHOSTBYADDRv6);
 
          (void) cache_add (req->type, key_copy, req->key_len,
-                           &dataset->head, true, db, owner);
+                           &dataset->head, true, db, owner, he == NULL);
 
          pthread_rwlock_unlock (&db->lock);
        }
     }
 
-  if (__builtin_expect (written != total, 0) && debug_level > 0)
+  if (__builtin_expect (!all_written, 0) && debug_level > 0)
     {
       char buf[256];
       dbg_log (_("short write in %s: %s"),  __FUNCTION__,
               strerror_r (errno, buf, sizeof (buf)));
     }
+
+  return timeout;
 }
 
 
@@ -441,7 +463,7 @@ lookup (int type, void *key, struct hostent *resultbufp, char *buffer,
 }
 
 
-static void
+static time_t
 addhstbyX (struct database_dyn *db, int fd, request_header *req,
           void *key, uid_t uid, struct hashentry *he, struct datahead *dh)
 {
@@ -495,6 +517,7 @@ addhstbyX (struct database_dyn *db, int fd, request_header *req,
              /* We set the error to indicate this is (possibly) a
                 temporary error and that it does not mean the entry
                 is not available at all.  */
+             h_errno = TRY_AGAIN;
              errval = EAGAIN;
              break;
            }
@@ -506,11 +529,13 @@ addhstbyX (struct database_dyn *db, int fd, request_header *req,
        buffer = (char *) extend_alloca (buffer, buflen, 2 * buflen);
     }
 
-  cache_addhst (db, fd, req, key, hst, uid, he, dh,
-               h_errno == TRY_AGAIN ? errval : 0, ttl);
+  time_t timeout = cache_addhst (db, fd, req, key, hst, uid, he, dh,
+                                h_errno == TRY_AGAIN ? errval : 0, ttl);
 
   if (use_malloc)
     free (buffer);
+
+  return timeout;
 }
 
 
@@ -522,7 +547,7 @@ addhstbyname (struct database_dyn *db, int fd, request_header *req,
 }
 
 
-void
+time_t
 readdhstbyname (struct database_dyn *db, struct hashentry *he,
                struct datahead *dh)
 {
@@ -532,7 +557,7 @@ readdhstbyname (struct database_dyn *db, struct hashentry *he,
       .key_len = he->len
     };
 
-  addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
+  return addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
 }
 
 
@@ -544,7 +569,7 @@ addhstbyaddr (struct database_dyn *db, int fd, request_header *req,
 }
 
 
-void
+time_t
 readdhstbyaddr (struct database_dyn *db, struct hashentry *he,
                struct datahead *dh)
 {
@@ -554,7 +579,7 @@ readdhstbyaddr (struct database_dyn *db, struct hashentry *he,
       .key_len = he->len
     };
 
-  addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
+  return addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
 }
 
 
@@ -566,7 +591,7 @@ addhstbynamev6 (struct database_dyn *db, int fd, request_header *req,
 }
 
 
-void
+time_t
 readdhstbynamev6 (struct database_dyn *db, struct hashentry *he,
                  struct datahead *dh)
 {
@@ -576,7 +601,7 @@ readdhstbynamev6 (struct database_dyn *db, struct hashentry *he,
       .key_len = he->len
     };
 
-  addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
+  return addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
 }
 
 
@@ -588,7 +613,7 @@ addhstbyaddrv6 (struct database_dyn *db, int fd, request_header *req,
 }
 
 
-void
+time_t
 readdhstbyaddrv6 (struct database_dyn *db, struct hashentry *he,
                  struct datahead *dh)
 {
@@ -598,5 +623,5 @@ readdhstbyaddrv6 (struct database_dyn *db, struct hashentry *he,
       .key_len = he->len
     };
 
-  addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
+  return addhstbyX (db, -1, &req, db->data + he->key, he->owner, he, dh);
 }