mirror of
git://sourceware.org/git/glibc.git
synced 2024-12-15 04:20:28 +08:00
111 lines
2.8 KiB
C
111 lines
2.8 KiB
C
/* Logic guts of scandir*.
|
|
Copyright (C) 1992-2016 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 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
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with the GNU C Library; if not, see
|
|
<http://www.gnu.org/licenses/>. */
|
|
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <libc-lock.h>
|
|
|
|
#ifndef SCANDIR_TAIL
|
|
# define SCANDIR_TAIL __scandir_tail
|
|
# define READDIR __readdir
|
|
# define DIRENT_TYPE struct dirent
|
|
#endif
|
|
|
|
internal_function
|
|
int
|
|
SCANDIR_TAIL (DIR *dp,
|
|
DIRENT_TYPE ***namelist,
|
|
int (*select) (const DIRENT_TYPE *),
|
|
int (*cmp) (const DIRENT_TYPE **, const DIRENT_TYPE **))
|
|
{
|
|
if (dp == NULL)
|
|
return -1;
|
|
|
|
int save = errno;
|
|
__set_errno (0);
|
|
|
|
int result;
|
|
struct scandir_cancel_struct c = { .dp = dp };
|
|
__libc_cleanup_push (&__scandir_cancel_handler, &c);
|
|
|
|
DIRENT_TYPE **v = NULL;
|
|
size_t vsize = 0;
|
|
DIRENT_TYPE *d;
|
|
while ((d = READDIR (dp)) != NULL)
|
|
{
|
|
if (select != NULL)
|
|
{
|
|
int selected = (*select) (d);
|
|
|
|
/* The SELECT function might have changed errno. It was
|
|
zero before and it need to be again to make the later
|
|
tests work. */
|
|
__set_errno (0);
|
|
|
|
if (!selected)
|
|
continue;
|
|
}
|
|
else
|
|
__set_errno (0);
|
|
|
|
if (__glibc_unlikely (c.cnt == vsize))
|
|
{
|
|
if (vsize == 0)
|
|
vsize = 10;
|
|
else
|
|
vsize *= 2;
|
|
DIRENT_TYPE **new = realloc (v, vsize * sizeof *v);
|
|
if (new == NULL)
|
|
break;
|
|
c.v = v = new;
|
|
}
|
|
|
|
size_t dsize = &d->d_name[_D_ALLOC_NAMLEN (d)] - (char *) d;
|
|
DIRENT_TYPE *vnew = malloc (dsize);
|
|
if (vnew == NULL)
|
|
break;
|
|
v[c.cnt++] = (DIRENT_TYPE *) memcpy (vnew, d, dsize);
|
|
}
|
|
|
|
if (__glibc_likely (errno == 0))
|
|
{
|
|
__closedir (dp);
|
|
|
|
/* Sort the list if we have a comparison function to sort with. */
|
|
if (cmp != NULL)
|
|
qsort (v, c.cnt, sizeof *v, (__compar_fn_t) cmp);
|
|
|
|
*namelist = v;
|
|
result = c.cnt;
|
|
}
|
|
else
|
|
{
|
|
/* This frees everything and calls closedir. */
|
|
__scandir_cancel_handler (&c);
|
|
result = -1;
|
|
}
|
|
|
|
__libc_cleanup_pop (0);
|
|
|
|
if (result >= 0)
|
|
__set_errno (save);
|
|
return result;
|
|
}
|