(internal_setent): If opening of file failed with EGAIN return
[kopensolaris-gnu/glibc.git] / nss / nss_files / files-XXX.c
index da4ad1f..eb0f6fe 100644 (file)
@@ -1,26 +1,28 @@
 /* Common code for file-based databases in nss_files module.
-Copyright (C) 1996 Free Software Foundation, Inc.
-This file is part of the GNU C Library.
+   Copyright (C) 1996, 1997 Free Software Foundation, Inc.
+   This file is part of the GNU C Library.
 
-The GNU C Library is free software; you can redistribute it and/or
-modify it under the terms of the GNU Library General Public License as
-published by the Free Software Foundation; either version 2 of the
-License, or (at your option) any later version.
+   The GNU C Library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Library General Public License as
+   published by the Free Software Foundation; either 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,
-but WITHOUT ANY WARRANTY; without even the implied warranty of
-MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
-Library General Public License for more details.
+   The GNU C Library 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
+   Library General Public License for more details.
 
-You should have received a copy of the GNU Library General Public
-License along with the GNU C Library; see the file COPYING.LIB.  If
-not, write to the Free Software Foundation, Inc., 675 Mass Ave,
-Cambridge, MA 02139, USA.  */
+   You should have received a copy of the GNU Library General Public
+   License along with the GNU C Library; see the file COPYING.LIB.  If not,
+   write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+   Boston, MA 02111-1307, USA.  */
 
 #include <stdio.h>
 #include <ctype.h>
 #include <assert.h>
-#include <libc-lock.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <bits/libc-lock.h>
 #include "nsswitch.h"
 
 /* These symbols are defined by the including source file:
@@ -39,13 +41,14 @@ Cambridge, MA 02139, USA.  */
 #define DATAFILE       "/etc/" DATABASE
 
 #ifdef NEED_H_ERRNO
-#define H_ERRNO_PROTO  , int *herrnop
-#define H_ERRNO_ARG    , herrnop
-#define H_ERRNO_SET(val) (*herrnop = (val))
+# include <netdb.h>
+# define H_ERRNO_PROTO , int *herrnop
+# define H_ERRNO_ARG   , herrnop
+# define H_ERRNO_SET(val) (*herrnop = (val))
 #else
-#define H_ERRNO_PROTO
-#define H_ERRNO_ARG
-#define H_ERRNO_SET(val) ((void) 0)
+# define H_ERRNO_PROTO
+# define H_ERRNO_ARG
+# define H_ERRNO_SET(val) ((void) 0)
 #endif
 
 /* Locks the static variables in this file.  */
@@ -54,20 +57,42 @@ __libc_lock_define_initialized (static, lock)
 /* Maintenance of the shared stream open on the database file.  */
 
 static FILE *stream;
+static fpos_t position;
+static enum { none, getent, getby } last_use;
 static int keep_stream;
 
 /* Open database file if not already opened.  */
-static int
+static enum nss_status
 internal_setent (int stayopen)
 {
-  int status = NSS_STATUS_SUCCESS;
+  enum nss_status status = NSS_STATUS_SUCCESS;
 
   if (stream == NULL)
     {
       stream = fopen (DATAFILE, "r");
 
       if (stream == NULL)
-       status = NSS_STATUS_UNAVAIL;
+       status = errno == EAGAIN ? NSS_STATUS_TRYAGAIN : NSS_STATUS_UNAVAIL;
+      else
+       {
+         /* We have to make sure the file is  `closed on exec'.  */
+         int result, flags;
+
+         result = flags = fcntl (fileno (stream), F_GETFD, 0);
+         if (result >= 0)
+           {
+             flags |= FD_CLOEXEC;
+             result = fcntl (fileno (stream), F_SETFD, flags);
+           }
+         if (result < 0)
+           {
+             /* Something went wrong.  Close the stream and return a
+                failure.  */
+             fclose (stream);
+             stream = NULL;
+             status = NSS_STATUS_UNAVAIL;
+           }
+       }
     }
   else
     rewind (stream);
@@ -81,15 +106,24 @@ internal_setent (int stayopen)
 
 
 /* Thread-safe, exported version of that.  */
-int
+enum nss_status
 CONCAT(_nss_files_set,ENTNAME) (int stayopen)
 {
-  int status;
+  enum nss_status status;
 
   __libc_lock_lock (lock);
 
   status = internal_setent (stayopen);
 
+  if (status == NSS_STATUS_SUCCESS && fgetpos (stream, &position) < 0)
+    {
+      fclose (stream);
+      stream = NULL;
+      status = NSS_STATUS_UNAVAIL;
+    }
+
+  last_use = getent;
+
   __libc_lock_unlock (lock);
 
   return status;
@@ -109,7 +143,7 @@ internal_endent (void)
 
 
 /* Thread-safe, exported version of that.  */
-int
+enum nss_status
 CONCAT(_nss_files_end,ENTNAME) (void)
 {
   __libc_lock_lock (lock);
@@ -133,25 +167,20 @@ internal_getent (struct STRUCTURE *result,
   char *p;
   struct parser_data *data = (void *) buffer;
   int linebuflen = buffer + buflen - data->linebuffer;
-
-  /* Be prepared that the set*ent function was not called before.  */
-  if (stream == NULL)
-    {
-      enum nss_status status;
-
-      status = internal_setent (0);
-      if (status != NSS_STATUS_SUCCESS)
-       return status;
-    }
+  int parse_result;
 
   if (buflen < (int) sizeof *data + 1)
     {
-      errno = ERANGE;
+      __set_errno (ERANGE);
+      H_ERRNO_SET (NETDB_INTERNAL);
       return NSS_STATUS_TRYAGAIN;
     }
 
   do
     {
+      /* Terminate the line so that we can test for overflow.  */
+      data->linebuffer[linebuflen - 1] = '\0';
+
       p = fgets (data->linebuffer, linebuflen, stream);
       if (p == NULL)
        {
@@ -159,34 +188,67 @@ internal_getent (struct STRUCTURE *result,
          H_ERRNO_SET (HOST_NOT_FOUND);
          return NSS_STATUS_NOTFOUND;
        }
-
-      /* Terminate the line for any case.  */
-      data->linebuffer[linebuflen - 1] = '\0';
+      else if (data->linebuffer[linebuflen - 1] != '\0')
+       {
+         /* The line is too long.  Give the user the opportunity to
+            enlarge the buffer.  */
+         __set_errno (ERANGE);
+         H_ERRNO_SET (NETDB_INTERNAL);
+         return NSS_STATUS_TRYAGAIN;
+       }
 
       /* Skip leading blanks.  */
       while (isspace (*p))
        ++p;
-    } while (*p == '\0' || *p == '#' ||        /* Ignore empty and comment lines.  */
-            /* Parse the line.  If it is invalid, loop to
-               get the next line of the file to parse.  */
-            ! parse_line (p, result, data, buflen));
+    }
+  while (*p == '\0' || *p == '#' /* Ignore empty and comment lines.  */
+        /* Parse the line.  If it is invalid, loop to get the next
+           line of the file to parse.  */
+        || ! (parse_result = parse_line (p, result, data, buflen)));
 
   /* Filled in RESULT with the next entry from the database file.  */
-  return NSS_STATUS_SUCCESS;
+  return parse_result == -1 ? NSS_STATUS_TRYAGAIN : NSS_STATUS_SUCCESS;
 }
 
 
 /* Return the next entry from the database file, doing locking.  */
-int
+enum nss_status
 CONCAT(_nss_files_get,ENTNAME_r) (struct STRUCTURE *result,
-                                 char *buffer, int buflen H_ERRNO_PROTO)
+                                 char *buffer, size_t buflen H_ERRNO_PROTO)
 {
   /* Return next entry in host file.  */
-  int status;
+  enum nss_status status = NSS_STATUS_SUCCESS;
 
   __libc_lock_lock (lock);
 
-  status = internal_getent (result, buffer, buflen H_ERRNO_ARG);
+  /* Be prepared that the set*ent function was not called before.  */
+  if (stream == NULL)
+    status = internal_setent (0);
+
+  if (status == NSS_STATUS_SUCCESS)
+    {
+      /* If the last use was not by the getent function we need the
+        position the stream.  */
+      if (last_use != getent)
+       if (fsetpos (stream, &position) < 0)
+         status = NSS_STATUS_UNAVAIL;
+       else
+         last_use = getent;
+
+      if (status == NSS_STATUS_SUCCESS)
+       {
+         status = internal_getent (result, buffer, buflen H_ERRNO_ARG);
+
+         /* Remember this position if we were successful.  If the
+            operation failed we give the user a chance to repeat the
+            operation (perhaps the buffer was too small).  */
+         if (status == NSS_STATUS_SUCCESS)
+           fgetpos (stream, &position);
+         else
+           /* We must make sure we reposition the stream the next call.  */
+           last_use = none;
+       }
+    }
 
   __libc_lock_unlock (lock);
 
@@ -209,21 +271,27 @@ CONCAT(_nss_files_get,ENTNAME_r) (struct STRUCTURE *result,
 enum nss_status                                                                      \
 _nss_files_get##name##_r (proto,                                             \
                          struct STRUCTURE *result,                           \
-                         char *buffer, int buflen H_ERRNO_PROTO)             \
+                         char *buffer, size_t buflen H_ERRNO_PROTO)          \
 {                                                                            \
   enum nss_status status;                                                    \
                                                                              \
   __libc_lock_lock (lock);                                                   \
                                                                              \
   /* Reset file pointer to beginning or open file.  */                       \
-  internal_setent (keep_stream);                                             \
+  status = internal_setent (keep_stream);                                    \
+                                                                             \
+  if (status == NSS_STATUS_SUCCESS)                                          \
+    {                                                                        \
+      /* Tell getent function that we have repositioned the file pointer.  */ \
+      last_use = getby;                                                              \
                                                                              \
-  while ((status = internal_getent (result, buffer, buflen H_ERRNO_ARG))      \
-        == NSS_STATUS_SUCCESS)                                               \
-    { break_if_match }                                                       \
+      while ((status = internal_getent (result, buffer, buflen H_ERRNO_ARG))  \
+            == NSS_STATUS_SUCCESS)                                           \
+       { break_if_match }                                                    \
                                                                              \
-  if (! keep_stream)                                                         \
-    internal_endent ();                                                              \
+      if (! keep_stream)                                                     \
+       internal_endent ();                                                   \
+    }                                                                        \
                                                                              \
   __libc_lock_unlock (lock);                                                 \
                                                                              \