dirent/scandir-tail.c

   1 /* Logic guts of scandir*.
   2    Copyright (C) 1992-2017 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <http://www.gnu.org/licenses/>.  */
  18
  19 #include <dirent.h>
  20 #include <errno.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <libc-lock.h>
  24
  25 #ifndef SCANDIR_TAIL
  26 # define SCANDIR_TAIL   __scandir_tail
  27 # define READDIR        __readdir
  28 # define DIRENT_TYPE    struct dirent
  29 #endif
  30
  31 int
  32 SCANDIR_TAIL (DIR *dp,
  33               DIRENT_TYPE ***namelist,
  34               int (*select) (const DIRENT_TYPE *),
  35               int (*cmp) (const DIRENT_TYPE **, const DIRENT_TYPE **))
  36 {
  37   if (dp == NULL)
  38     return -1;
  39
  40   int save = errno;
  41   __set_errno (0);
  42
  43   int result;
  44   struct scandir_cancel_struct c = { .dp = dp };
  45   __libc_cleanup_push (&__scandir_cancel_handler, &c);
  46
  47   DIRENT_TYPE **v = NULL;
  48   size_t vsize = 0;
  49   DIRENT_TYPE *d;
  50   while ((d = READDIR (dp)) != NULL)
  51     {
  52       if (select != NULL)
  53         {
  54           int selected = (*select) (d);
  55
  56           /* The SELECT function might have changed errno.  It was
  57              zero before and it need to be again to make the later
  58              tests work.  */
  59           __set_errno (0);
  60
  61           if (!selected)
  62             continue;
  63         }
  64       else
  65         __set_errno (0);
  66
  67       if (__glibc_unlikely (c.cnt == vsize))
  68         {
  69           if (vsize == 0)
  70             vsize = 10;
  71           else
  72             vsize *= 2;
  73           DIRENT_TYPE **new = realloc (v, vsize * sizeof *v);
  74           if (new == NULL)
  75             break;
  76           c.v = v = new;
  77         }
  78
  79       size_t dsize = &d->d_name[_D_ALLOC_NAMLEN (d)] - (char *) d;
  80       DIRENT_TYPE *vnew = malloc (dsize);
  81       if (vnew == NULL)
  82         break;
  83       v[c.cnt++] = (DIRENT_TYPE *) memcpy (vnew, d, dsize);
  84     }
  85
  86   if (__glibc_likely (errno == 0))
  87     {
  88       __closedir (dp);
  89
  90       /* Sort the list if we have a comparison function to sort with.  */
  91       if (cmp != NULL)
  92         qsort (v, c.cnt, sizeof *v, (__compar_fn_t) cmp);
  93
  94       *namelist = v;
  95       result = c.cnt;
  96     }
  97   else
  98     {
  99       /* This frees everything and calls closedir.  */
 100       __scandir_cancel_handler (&c);
 101       result = -1;
 102     }
 103
 104   __libc_cleanup_pop (0);
 105
 106   if (result >= 0)
 107     __set_errno (save);
 108   return result;
 109 }