From 4b7de2299951cd13084305910d16939cd73ac817 Mon Sep 17 00:00:00 2001
From: Robert James Kaes <rjkaes@users.sourceforge.net>
Date: Mon, 15 Apr 2002 04:16:01 +0000
Subject: Added a test to make sure the "whole_buffer" could be allocated.

---
 src/sock.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/sock.c b/src/sock.c
index 004d0b6..cefe336 100644
--- a/src/sock.c
+++ b/src/sock.c
@@ -1,4 +1,4 @@
-/* $Id: sock.c,v 1.25 2002-04-13 19:03:18 rjkaes Exp $
+/* $Id: sock.c,v 1.26 2002-04-15 04:16:01 rjkaes Exp $
  *
  * Sockets are created and destroyed here. When a new connection comes in from
  * a client, we need to copy the socket and the create a second socket to the
@@ -477,8 +477,10 @@ readline(int fd, char **whole_buffer)
 	}
 
 	*whole_buffer = safemalloc(whole_buffer_len + 1);
-	if (!*whole_buffer)
-		return -ENOMEMORY;
+	if (!*whole_buffer) {
+		ret = -ENOMEMORY;
+		goto CLEANUP;
+	}
 
 	*(*whole_buffer + whole_buffer_len) = '\0';
 
-- 
cgit v1.2.3