lib/safe-write.c

   1 /* An interface to write() that retries after interrupts.
   2    Copyright (C) 1993, 1994, 1998, 2002 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 2, or (at your option)
   7    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, write to the Free Software Foundation,
  16    Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.  */
  17
  18 #if HAVE_CONFIG_H
  19 # include <config.h>
  20 #endif
  21
  22 /* Specification.  */
  23 #include "safe-write.h"
  24
  25 /* Get ssize_t.  */
  26 #include <sys/types.h>
  27 #if HAVE_UNISTD_H
  28 # include <unistd.h>
  29 #endif
  30
  31 #include <errno.h>
  32 #ifndef errno
  33 extern int errno;
  34 #endif
  35
  36 #include <limits.h>
  37
  38 /* We don't pass an nbytes count > SSIZE_MAX to write() - POSIX says the
  39    effect would be implementation-defined.  Also we don't pass an nbytes
  40    count > INT_MAX but <= SSIZE_MAX to write() - this triggers a bug in
  41    Tru64 5.1.  */
  42 #define MAX_BYTES_TO_READ INT_MAX
  43
  44 /* Write up to COUNT bytes at BUF to descriptor FD, retrying if interrupted.
  45    Return the actual number of bytes written, zero for EOF, or (size_t) -1
  46    for an error.  */
  47 size_t
  48 safe_write (int fd, const void *buf, size_t count)
  49 {
  50   size_t total_written = 0;
  51
  52   if (count > 0)
  53     {
  54       const char *ptr = (const char *) buf;
  55       do
  56         {
  57           size_t nbytes_to_write = count;
  58           ssize_t result;
  59
  60           /* Limit the number of bytes to write in one round, to avoid running
  61              into unspecified behaviour.  But keep the file pointer block
  62              aligned when doing so.  */
  63           if (nbytes_to_write > MAX_BYTES_TO_READ)
  64             nbytes_to_write = MAX_BYTES_TO_READ & ~8191;
  65
  66           result = write (fd, ptr, nbytes_to_write);
  67           if (result < 0)
  68             {
  69 #ifdef EINTR
  70               if (errno == EINTR)
  71                 continue;
  72 #endif
  73               return result;
  74             }
  75           total_written += result;
  76           ptr += result;
  77           count -= result;
  78         }
  79       while (count > 0);
  80     }
  81
  82   return total_written;
  83 }