dirent/scandir.c

   1 /* Copyright (C) 1992, 1993, 1994, 1995, 1996 Free Software Foundation, Inc.
   2 This file is part of the GNU C Library.
   3
   4 The GNU C Library is free software; you can redistribute it and/or
   5 modify it under the terms of the GNU Library General Public License as
   6 published by the Free Software Foundation; either version 2 of the
   7 License, or (at your option) any later version.
   8
   9 The GNU C Library is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 Library General Public License for more details.
  13
  14 You should have received a copy of the GNU Library General Public
  15 License along with the GNU C Library; see the file COPYING.LIB.  If
  16 not, write to the Free Software Foundation, Inc., 675 Mass Ave,
  17 Cambridge, MA 02139, USA.  */
  18
  19 #include <dirent.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22 #include <errno.h>
  23
  24 int
  25 scandir (dir, namelist, select, cmp)
  26      const char *dir;
  27      struct dirent ***namelist;
  28      int (*select) __P ((struct dirent *));
  29      int (*cmp) __P ((const void *, const void *));
  30 {
  31   DIR *dp = opendir (dir);
  32   struct dirent **v = NULL;
  33   size_t vsize = 0, i;
  34   struct dirent *d;
  35   int save;
  36
  37   if (dp == NULL)
  38     return -1;
  39
  40   save = errno;
  41   __set_errno (0);
  42
  43   i = 0;
  44   while ((d = readdir (dp)) != NULL)
  45     if (select == NULL || (*select) (d))
  46       {
  47         size_t dsize;
  48
  49         if (i == vsize)
  50           {
  51             struct dirent **new;
  52             if (vsize == 0)
  53               vsize = 10;
  54             else
  55               vsize *= 2;
  56             new = (struct dirent **) realloc (v, vsize * sizeof (*v));
  57             if (new == NULL)
  58               {
  59               lose:
  60                 __set_errno (ENOMEM);
  61                 break;
  62               }
  63             v = new;
  64           }
  65
  66         dsize = &d->d_name[_D_ALLOC_NAMLEN (d)] - (char *) d;
  67         v[i] = (struct dirent *) malloc (dsize);
  68         if (v[i] == NULL)
  69           goto lose;
  70
  71         memcpy (v[i++], d, dsize);
  72       }
  73
  74   if (errno != 0)
  75     {
  76       save = errno;
  77       (void) closedir (dp);
  78       while (i > 0)
  79         free (v[--i]);
  80       free (v);
  81       __set_errno (save);
  82       return -1;
  83     }
  84
  85   (void) closedir (dp);
  86   __set_errno (save);
  87
  88   /* Sort the list if we have a comparison function to sort with.  */
  89   if (cmp != NULL)
  90     qsort (v, i, sizeof (*v), cmp);
  91   *namelist = v;
  92   return i;
  93 }