From b29a549cdd85d42a1697041ab04f0ae4eddd23ca Mon Sep 17 00:00:00 2001
From: Simo Sorce <idra@samba.org>
Date: Sun, 5 Aug 2001 10:10:16 +0000
Subject: Some fixes about malloc/Realloc and mem leak thanks to andreas
 moroder

---
 source/lib/messages.c | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

(limited to 'source/lib/messages.c')

diff --git a/source/lib/messages.c b/source/lib/messages.c
index 78cd3e22bc2..54f6321041a 100644
--- a/source/lib/messages.c
+++ b/source/lib/messages.c
@@ -322,12 +322,19 @@ void message_register(int msg_type,
 
 	dfn = (struct dispatch_fns *)malloc(sizeof(*dfn));
 
-	ZERO_STRUCTP(dfn);
+	if (dfn != NULL) {
 
-	dfn->msg_type = msg_type;
-	dfn->fn = fn;
+		ZERO_STRUCTPN(dfn);
 
-	DLIST_ADD(dispatch_fns, dfn);
+		dfn->msg_type = msg_type;
+		dfn->fn = fn;
+
+		DLIST_ADD(dispatch_fns, dfn);
+	}
+	else {
+	
+		DEBUG(0,("message_register: Not enough memory. malloc failed!\n"));
+	}
 }
 
 /****************************************************************************
-- 
cgit