From: Linus Torvalds <torvalds@osdl.org>
Date: Fri, 12 Jan 2007 04:37:38 +0000 (-0800)
Subject: Fix up totally buggered read_or_die()
X-Git-Tag: v1.5.0-rc2~118
X-Git-Url: https://git.tokkee.org/?a=commitdiff_plain;h=4494c656e2e29c468c48c9c2b20595342056e9dc;p=git.git

Fix up totally buggered read_or_die()

The "read_or_die()" function would silently NOT die for a partial read,
and since it was of type "void" it obviously couldn't even return the
partial number of bytes read.

IOW, it was totally broken. This hopefully fixes it up.

Signed-off-by: Linus Torvalds <torvalds@osdl.org>
Signed-off-by: Junio C Hamano <junkio@cox.net>
---

diff --git a/write_or_die.c b/write_or_die.c
index 1224cac5d..4e8183e93 100644
--- a/write_or_die.c
+++ b/write_or_die.c
@@ -4,16 +4,11 @@ int read_in_full(int fd, void *buf, size_t count)
 {
 	char *p = buf;
 	ssize_t total = 0;
-	ssize_t loaded = 0;
 
 	while (count > 0) {
-		loaded = xread(fd, p, count);
-		if (loaded <= 0) {
-			if (total)
-				return total;
-			else
-				return loaded;
-		}
+		ssize_t loaded = xread(fd, p, count);
+		if (loaded <= 0)
+			return total ? total : loaded;
 		count -= loaded;
 		p += loaded;
 		total += loaded;
@@ -26,13 +21,12 @@ void read_or_die(int fd, void *buf, size_t count)
 {
 	ssize_t loaded;
 
-	if (!count)
-		return;
 	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));
+	if (loaded != count) {
+		if (loaded < 0)
+			die("read error (%s)", strerror(errno));
+		die("read error: end of file");
+	}
 }
 
 int write_in_full(int fd, const void *buf, size_t count)