[FFmpeg-devel] [PATCH 3/4] avio: Check for memory allocation failure of private data

[Derek Buitenhuis]

Signed-off-by: Derek Buitenhuis <derek.buitenhuis at gmail.com>
---
 libavformat/avio.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libavformat/avio.c b/libavformat/avio.c
index 2c7a35e..558e3e8 100644
--- a/libavformat/avio.c
+++ b/libavformat/avio.c
@@ -143,6 +143,10 @@ static int url_alloc_for_protocol (URLContext **puc, struct URLProtocol *up,
     uc->max_packet_size = 0; /* default: stream file */
     if (up->priv_data_size) {
         uc->priv_data = av_mallocz(up->priv_data_size);
+        if (!uc->priv_data) {
+            err = AVERROR(ENOMEM);
+            goto fail;
+        }
         if (up->priv_data_class) {
             int proto_len= strlen(up->name);
             char *start = strchr(uc->filename, ',');
-- 
1.8.4.rc3