From 186094e3033991b7e8a46cc4b716cd2d06e5231e Mon Sep 17 00:00:00 2001
From: des <des@d4fa192b-c00b-0410-8231-f00ffab90ce4>
Date: Tue, 3 Jul 2007 11:34:16 +0000
Subject: [PATCH] Additional paranoia.

git-svn-id: svn+ssh://projects.linpro.no/svn/varnish/trunk@1625 d4fa192b-c00b-0410-8231-f00ffab90ce4
---
 .../bin/varnishreplay/varnishreplay.c         | 34 +++++++++++--------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/varnish-cache/bin/varnishreplay/varnishreplay.c b/varnish-cache/bin/varnishreplay/varnishreplay.c
index 16461d8b..45726a96 100644
--- a/varnish-cache/bin/varnishreplay/varnishreplay.c
+++ b/varnish-cache/bin/varnishreplay/varnishreplay.c
@@ -385,12 +385,14 @@ receive_response(int sock)
 	int status;
 
 	/* Read header */
-	while (1) {
+	for (;;) {
 		line_len = read_line(&line, sock);
+		if (line_len < 0)
+			return (-1);
 		end = line + line_len;
 
-		if (*line == '\r' && *(line + 1) == '\n') {
-			free(line);
+		if (line_len >= 2 && line[0] == '\r' && line[1] == '\n') {
+			freez(line);
 			break;
 		}
 
@@ -405,7 +407,7 @@ receive_response(int sock)
 		else if (isprefix(line, "connection:", end, &next))
 			close_connection = (strstr(next, "close") != NULL);
 
-		free(line);
+		freez(line);
 	}
 
 	thread_log(1, "status: %d\n", status);
@@ -422,30 +424,32 @@ receive_response(int sock)
 	} else if (chunked) {
 		/* Chunked encoding, read size and bytes until no more */
 		thread_log(1, "chunked encoding\n");
-		while (1) {
-			line_len = read_line(&line, sock);
+		for (;;) {
+			if ((line_len = read_line(&line, sock)) < 0)
+				return (-1);
 			end = line + line_len;
 			block_len = strtol(line, &end, 16);
-			if (block_len == 0) {
+			freez(line);
+			if (block_len == 0)
 				break;
-			}
-			n = read_block(block_len, sock);
+			if ((n = read_block(block_len, sock)) < 0)
+				return (-1);
 			thread_log(1, "size of body: %d\n", (int)block_len);
 			thread_log(1, "bytes read: %d\n", n);
-			free(line);
-			n = read_line(&line, sock);
-			free(line);
+			if ((n = read_line(&line, sock)) < 0)
+				return (-1);
+			freez(line);
 		}
 		n = read_line(&line, sock);
-		free(line);
+		freez(line);
 	} else if ((content_length <= 0 && !chunked) || req_failed) {
 		/* No body --> stop reading. */
 		thread_log(1, "no body\n");
-		return (1);
+		return (-1);
 	} else {
 		/* Unhandled case. */
 		thread_log(0, "An error occured\n");
-		return (1);
+		return (-1);
 	}
 
 	return close_connection;
-- 
2.39.5