gnulib/import/closedir.c

   1 /* Stop reading the entries of a directory.
   2    Copyright (C) 2006-2016 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program 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
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  16
  17 #include <config.h>
  18
  19 /* Specification.  */
  20 #include <dirent.h>
  21
  22 #if REPLACE_FCHDIR
  23 # include <unistd.h>
  24 #endif
  25
  26 #if HAVE_CLOSEDIR
  27
  28 /* Override closedir(), to keep track of the open file descriptors.
  29    Needed because there is a function dirfd().  */
  30
  31 #else
  32
  33 # include <stdlib.h>
  34
  35 # include "dirent-private.h"
  36
  37 #endif
  38
  39 int
  40 closedir (DIR *dirp)
  41 {
  42 # if REPLACE_FCHDIR || REPLACE_DIRFD
  43   int fd = dirfd (dirp);
  44 # endif
  45   int retval;
  46
  47 #if HAVE_CLOSEDIR
  48 # undef closedir
  49
  50   retval = closedir (dirp);
  51
  52 # ifdef __KLIBC__
  53   if (!retval)
  54     _gl_unregister_dirp_fd (fd);
  55 # endif
  56 #else
  57
  58   if (dirp->current != INVALID_HANDLE_VALUE)
  59     FindClose (dirp->current);
  60   free (dirp);
  61
  62   retval = 0;
  63
  64 #endif
  65
  66 #if REPLACE_FCHDIR
  67   if (retval >= 0)
  68     _gl_unregister_fd (fd);
  69 #endif
  70   return retval;
  71 }