From: Howard Chu <hyc@highlandsun.com>
Date: Tue, 8 Nov 2011 10:14:21 +0000 (-0800)
Subject: Check for malloc failure in prev commit
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=30fcf46fc82f96ca41b710fc38bbc15f2489795e;p=rtmpdump

Check for malloc failure in prev commit
---

diff --git a/rtmpdump.c b/rtmpdump.c
index 892a8bc..c37def2 100644
--- a/rtmpdump.c
+++ b/rtmpdump.c
@@ -1155,6 +1155,8 @@ main(int argc, char **argv)
 	  tcUrl.av_len = strlen(RTMPProtocolStringsLower[protocol]) +
 	  	hostname.av_len + app.av_len + sizeof("://:65535/");
       tcUrl.av_val = (char *) malloc(tcUrl.av_len);
+	  if (!tcUrl.av_val)
+	    return RD_FAILED;
       tcUrl.av_len = snprintf(tcUrl.av_val, tcUrl.av_len, "%s://%.*s:%d/%.*s",
 	  	   RTMPProtocolStringsLower[protocol], hostname.av_len,
 		   hostname.av_val, port, app.av_len, app.av_val);