[idms-dbma-devel]commit - r51 - trunk/modules/protocols/smtp

[<idm...@li...>]

Author: nkukard
Date: 2004-10-24 19:10:50 +0200 (Sun, 24 Oct 2004)
New Revision: 51

Modified:
   trunk/modules/protocols/smtp/smtp.c
Log:
* Added some memory allocation checking


Modified: trunk/modules/protocols/smtp/smtp.c
===================================================================
--- trunk/modules/protocols/smtp/smtp.c	2004-10-24 15:41:27 UTC (rev 50)
+++ trunk/modules/protocols/smtp/smtp.c	2004-10-24 17:10:50 UTC (rev 51)
@@ -92,9 +92,18 @@
 	data->envelopeTo = NULL;
 	
 	data->tmpBufSize = SMTP_RESP_LEN;
-	data->tmpBuf = malloc(data->tmpBufSize);
-	data->tmpCmd = malloc(sizeof(struct smtp_command_t));
+
+	/* Check if allocation succeeded */	
+	if (!(data->tmpBuf = malloc(data->tmpBufSize)))
+		return 1;
 	
+	/* Same here... */	
+	if (!(data->tmpCmd = malloc(sizeof(struct smtp_command_t))))
+	{
+		free(data->tmpBuf);
+		return 1;
+	}
+	
 	data->cmdHandle = NULL;
 	
 	fprintf(stderr,"smtp startup\n");