.
[kopensolaris-gnu/glibc.git] / dirent / scandir.c
index 6907f08..89bd9e9 100644 (file)
-/* Copyright (C) 1992, 1993, 1994, 1995, 1996 Free Software Foundation, Inc.
-This file is part of the GNU C Library.
+/* Copyright (C) 1992-1998, 2000, 2002, 2003 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 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.
 
-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
+   Lesser 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 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.  */
 
-#include <ansidecl.h>
 #include <dirent.h>
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
+#include <bits/libc-lock.h>
+
+#ifndef SCANDIR
+#define SCANDIR scandir
+#define READDIR __readdir
+#define DIRENT_TYPE struct dirent
+#endif
+
+#ifndef SCANDIR_CANCEL
+#define SCANDIR_CANCEL
+struct scandir_cancel_struct
+{
+  DIR *dp;
+  void *v;
+  size_t cnt;
+};
+
+static void
+cancel_handler (void *arg)
+{
+  struct scandir_cancel_struct *cp = arg;
+  size_t i;
+  void **v = cp->v;
+
+  for (i = 0; i < cp->cnt; ++i)
+    free (v[i]);
+  free (v);
+  (void) __closedir (cp->dp);
+}
+#endif
+
 
 int
-DEFUN(scandir, (dir, namelist, select, cmp),
-      CONST char *dir AND
-      struct dirent ***namelist AND
-      int EXFUN((*select), (struct dirent *)) AND
-      int EXFUN((*cmp), (CONST PTR, CONST PTR)))
+SCANDIR (dir, namelist, select, cmp)
+     const char *dir;
+     DIRENT_TYPE ***namelist;
+     int (*select) (const DIRENT_TYPE *);
+     int (*cmp) (const void *, const void *);
 {
-  DIR *dp = opendir (dir);
-  struct dirent **v = NULL;
-  size_t vsize = 0, i;
-  struct dirent *d;
+  DIR *dp = __opendir (dir);
+  DIRENT_TYPE **v = NULL;
+  size_t vsize = 0;
+  struct scandir_cancel_struct c;
+  DIRENT_TYPE *d;
   int save;
 
   if (dp == NULL)
     return -1;
 
   save = errno;
-  errno = 0;
-
-  i = 0;
-  while ((d = readdir (dp)) != NULL)
-    if (select == NULL || (*select) (d))
-      {
-       size_t dsize;
-
-       if (i == vsize)
-         {
-           struct dirent **new;
-           if (vsize == 0)
-             vsize = 10;
-           else
-             vsize *= 2;
-           new = (struct dirent **) realloc (v, vsize * sizeof (*v));
-           if (new == NULL)
-             {
-             lose:
-               errno = ENOMEM;
+  __set_errno (0);
+
+  c.dp = dp;
+  c.v = NULL;
+  c.cnt = 0;
+  __libc_cleanup_push (cancel_handler, &c);
+
+  while ((d = READDIR (dp)) != NULL)
+    {
+      int use_it = select == NULL;
+
+      if (! use_it)
+       {
+         use_it = select (d);
+         /* The select function might have changed errno.  It was
+            zero before and it need to be again to make the latter
+            tests work.  */
+         __set_errno (0);
+       }
+
+      if (use_it)
+       {
+         DIRENT_TYPE *vnew;
+         size_t dsize;
+
+         /* Ignore errors from select or readdir */
+         __set_errno (0);
+
+         if (__builtin_expect (c.cnt == vsize, 0))
+           {
+             DIRENT_TYPE **new;
+             if (vsize == 0)
+               vsize = 10;
+             else
+               vsize *= 2;
+             new = (DIRENT_TYPE **) realloc (v, vsize * sizeof (*v));
+             if (new == NULL)
                break;
-             }
-           v = new;
-         }
+             v = new;
+             c.v = (void *) v;
+           }
 
-       dsize = &d->d_name[_D_ALLOC_NAMLEN (d)] - (char *) d;
-       v[i] = (struct dirent *) malloc (dsize);
-       if (v[i] == NULL)
-         goto lose;
+         dsize = &d->d_name[_D_ALLOC_NAMLEN (d)] - (char *) d;
+         vnew = (DIRENT_TYPE *) malloc (dsize);
+         if (vnew == NULL)
+           break;
 
-       memcpy (v[i++], d, dsize);
-      }
+         v[c.cnt++] = (DIRENT_TYPE *) memcpy (vnew, d, dsize);
+       }
+    }
 
-  if (errno != 0)
+  if (__builtin_expect (errno, 0) != 0)
     {
       save = errno;
-      (void) closedir (dp);
-      while (i > 0)
-       free (v[--i]);
+
+      while (c.cnt > 0)
+       free (v[--c.cnt]);
       free (v);
-      errno = save;
-      return -1;
+      c.cnt = -1;
     }
+  else
+    {
+      /* Sort the list if we have a comparison function to sort with.  */
+      if (cmp != NULL)
+       qsort (v, c.cnt, sizeof (*v), cmp);
+
+      *namelist = v;
+    }
+
+  __libc_cleanup_pop (0);
 
-  (void) closedir (dp);
-  errno = save;
+  (void) __closedir (dp);
+  __set_errno (save);
 
-  /* Sort the list if we have a comparison function to sort with.  */
-  if (cmp != NULL)
-    qsort (v, i, sizeof (*v), cmp);
-  *namelist = v;
-  return i;
+  return c.cnt;
 }