lib/fclose.c

   1 /* fclose replacement.
   2    Copyright (C) 2008-2011 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 <stdio.h>
  21
  22 #include <errno.h>
  23 #include <unistd.h>
  24
  25 #include "freading.h"
  26
  27 /* Override fclose() to call the overridden fflush() or close().  */
  28
  29 int
  30 rpl_fclose (FILE *fp)
  31 #undef fclose
  32 {
  33   int saved_errno = 0;
  34   int fd;
  35   int result = 0;
  36
  37   /* Don't change behavior on memstreams.  */
  38   fd = fileno (fp);
  39   if (fd < 0)
  40     return fclose (fp);
  41
  42   /* We only need to flush the file if it is not reading or if it is
  43      seekable.  This only guarantees the file position of input files
  44      if the fflush module is also in use.  */
  45   if ((!freading (fp) || lseek (fileno (fp), 0, SEEK_CUR) != -1)
  46       && fflush (fp))
  47     saved_errno = errno;
  48
  49   /* fclose() calls close(), but we need to also invoke all hooks that our
  50      overridden close() function invokes.  See lib/close.c.  */
  51 #if WINDOWS_SOCKETS
  52   /* Call the overridden close(), then the original fclose().
  53      Note about multithread-safety: There is a race condition where some
  54      other thread could open fd between our close and fclose.  */
  55   if (close (fd) < 0 && saved_errno == 0)
  56     saved_errno = errno;
  57
  58   fclose (fp); /* will fail with errno = EBADF, if we did not lose a race */
  59
  60 #else /* !WINDOWS_SOCKETS */
  61   /* Call fclose() and invoke all hooks of the overridden close().  */
  62
  63 # if REPLACE_FCHDIR
  64   /* Note about multithread-safety: There is a race condition here as well.
  65      Some other thread could open fd between our calls to fclose and
  66      _gl_unregister_fd.  */
  67   result = fclose (fp);
  68   if (result == 0)
  69     _gl_unregister_fd (fd);
  70 # else
  71   /* No race condition here.  */
  72   result = fclose (fp);
  73 # endif
  74
  75 #endif /* !WINDOWS_SOCKETS */
  76
  77   if (saved_errno != 0)
  78     {
  79       errno = saved_errno;
  80       result = EOF;
  81     }
  82
  83   return result;
  84 }