From 93d26e4cb9cec2eb8abb4f37e6dda2c86fcceeac Mon Sep 17 00:00:00 2001 From: Andy Whitcroft Date: Mon, 8 Jan 2007 15:58:08 +0000 Subject: short i/o: fix calls to read to use xread or read_in_full We have a number of badly checked read() calls. Often we are expecting read() to read exactly the size we requested or fail, this fails to handle interrupts or short reads. Add a read_in_full() providing those semantics. Otherwise we at a minimum need to check for EINTR and EAGAIN, where this is appropriate use xread(). Signed-off-by: Andy Whitcroft Signed-off-by: Junio C Hamano --- write_or_die.c | 29 +++++++++++++++++++++++------ 1 file changed, 23 insertions(+), 6 deletions(-) (limited to 'write_or_die.c') diff --git a/write_or_die.c b/write_or_die.c index 613c0c3f6e..e7f82639ff 100644 --- a/write_or_die.c +++ b/write_or_die.c @@ -1,19 +1,36 @@ #include "cache.h" -void read_or_die(int fd, void *buf, size_t count) +int read_in_full(int fd, void *buf, size_t count) { char *p = buf; - ssize_t loaded; + ssize_t total = 0; + ssize_t loaded = 0; while (count > 0) { loaded = xread(fd, p, count); - if (loaded == 0) - die("unexpected end of file"); - else if (loaded < 0) - die("read error (%s)", strerror(errno)); + if (loaded <= 0) { + if (total) + return total; + else + return loaded; + } count -= loaded; p += loaded; + total += loaded; } + + return total; +} + +void read_or_die(int fd, void *buf, size_t count) +{ + ssize_t loaded; + + loaded = read_in_full(fd, buf, count); + if (loaded == 0) + die("unexpected end of file"); + else if (loaded < 0) + die("read error (%s)", strerror(errno)); } void write_or_die(int fd, const void *buf, size_t count) -- cgit v1.3