X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=libustcomm%2Fustcomm.c;h=0d2ab339485cb666ba033b24d9896840a7d1c311;hb=da000ba4a6841b1bd55344c75c595aabdd830cd7;hp=e1a2d5b9554425283b04a283d93403a5d846ff96;hpb=972af454fdacdeda8f6e7980ce6261f1c92f4034;p=ust.git

diff --git a/libustcomm/ustcomm.c b/libustcomm/ustcomm.c
index e1a2d5b..0d2ab33 100644
--- a/libustcomm/ustcomm.c
+++ b/libustcomm/ustcomm.c
@@ -157,11 +157,12 @@ int ustcomm_request_consumer(pid_t pid, const char *channel)
 }
 
 /* returns 1 to indicate a message was received
- * returns 0 to indicate no message was received (cannot happen)
+ * returns 0 to indicate no message was received (end of stream)
  * returns -1 to indicate an error
  */
 
 #define RECV_INCREMENT 1
+#define RECV_INITIAL_BUF_SIZE 10
 
 static int recv_message_fd(int fd, char **msg)
 {
@@ -170,13 +171,20 @@ static int recv_message_fd(int fd, char **msg)
 	char *buf = NULL;
 	int buf_used_size = 0;
 
-	buf = malloc(10);
-	buf_alloc_size = 16;
+	buf = malloc(RECV_INITIAL_BUF_SIZE);
+	buf_alloc_size = RECV_INITIAL_BUF_SIZE;
 
 	for(;;) {
 		if(buf_used_size + RECV_INCREMENT > buf_alloc_size) {
+			char *new_buf;
 			buf_alloc_size *= 2;
-			buf = (char *) realloc(buf, buf_alloc_size);
+			new_buf = (char *) realloc(buf, buf_alloc_size);
+			if(new_buf == NULL) {
+				ERR("realloc returned NULL");
+				free(buf);
+				return -1;
+			}
+			buf = new_buf;
 		}
 
 		/* FIXME: this is really inefficient; but with count>1 we would
@@ -197,7 +205,6 @@ static int recv_message_fd(int fd, char **msg)
 			}
 		}
 
-
 		buf_used_size += result;
 
 		if(buf[buf_used_size-1] == 0) {