From: James Zern <jzern@google.com>
Date: Wed, 27 Apr 2022 05:20:00 +0000 (-0700)
Subject: y4m_input_open: check allocs
X-Git-Tag: v1.12.0-rc1~27^2
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=8baaa7b5a3fcea958261bda667a871c163e93bf9;p=libvpx

y4m_input_open: check allocs

Change-Id: I99ee0ef3ab28a22923cb413ccf5935fdc38862be
---

diff --git a/y4minput.c b/y4minput.c
index 9a4bdbd7b..7d3c03a7f 100644
--- a/y4minput.c
+++ b/y4minput.c
@@ -1087,9 +1087,15 @@ int y4m_input_open(y4m_input *y4m_ctx, FILE *file, char *skip_buffer,
     y4m_ctx->dst_buf = (unsigned char *)malloc(y4m_ctx->dst_buf_sz);
   else
     y4m_ctx->dst_buf = (unsigned char *)malloc(2 * y4m_ctx->dst_buf_sz);
+  if (!y4m_ctx->dst_buf) return -1;
 
-  if (y4m_ctx->aux_buf_sz > 0)
+  if (y4m_ctx->aux_buf_sz > 0) {
     y4m_ctx->aux_buf = (unsigned char *)malloc(y4m_ctx->aux_buf_sz);
+    if (!y4m_ctx->aux_buf) {
+      free(y4m_ctx->dst_buf);
+      return -1;
+    }
+  }
   return 0;
 }