lib/safe-read.c

   1 /* An interface to read 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-read.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 #ifndef CHAR_BIT
  39 # define CHAR_BIT 8
  40 #endif
  41
  42 /* The extra casts work around common compiler bugs.  */
  43 #define TYPE_SIGNED(t) (! ((t) 0 < (t) -1))
  44 /* The outer cast is needed to work around a bug in Cray C 5.0.3.0.
  45    It is necessary at least when t == time_t.  */
  46 #define TYPE_MINIMUM(t) ((t) (TYPE_SIGNED (t) \
  47                               ? ~ (t) 0 << (sizeof (t) * CHAR_BIT - 1) : (t) 0))
  48 #define TYPE_MAXIMUM(t) ((t) (~ (t) 0 - TYPE_MINIMUM (t)))
  49
  50 #ifndef INT_MAX
  51 # define INT_MAX TYPE_MAXIMUM (int)
  52 #endif
  53
  54 /* We don't pass an nbytes count > SSIZE_MAX to read() - POSIX says the
  55    effect would be implementation-defined.  Also we don't pass an nbytes
  56    count > INT_MAX but <= SSIZE_MAX to read() - this triggers a bug in
  57    Tru64 5.1.  */
  58 #define MAX_BYTES_TO_READ INT_MAX
  59
  60 /* Read up to COUNT bytes at BUF from descriptor FD, retrying if interrupted.
  61    Return the actual number of bytes read, zero for EOF, or SAFE_READ_ERROR
  62    upon error.  */
  63 size_t
  64 safe_read (int fd, void *buf, size_t count)
  65 {
  66   size_t total_read = 0;
  67   char *ptr = buf;
  68
  69   while (count > 0)
  70     {
  71       size_t nbytes_to_read = count;
  72       ssize_t result;
  73
  74       /* Limit the number of bytes to read in one round, to avoid running
  75          into unspecified behaviour.  But keep the file pointer block
  76          aligned when doing so.  */
  77       if (nbytes_to_read > MAX_BYTES_TO_READ)
  78         nbytes_to_read = MAX_BYTES_TO_READ & ~8191;
  79
  80       result = read (fd, ptr, nbytes_to_read);
  81       if (result == 0)
  82         break;
  83       if (result < 0)
  84         {
  85 #ifdef EINTR
  86           if (errno == EINTR)
  87             continue;
  88 #endif
  89           return SAFE_READ_ERROR;
  90         }
  91       total_read += result;
  92       ptr += result;
  93       count -= result;
  94     }
  95
  96   return total_read;
  97 }