commit a102db417e1e177274a9912fa75b762efc1b8b55
author James Zern <jzern@google.com> Tue Apr 26 22:20:00 2022 -0700
committer James Zern <jzern@google.com> Fri Apr 29 21:56:33 2022 +0000
tree 4da609f68298b93c5dba2b3223d2a06741569b89
parent efe2a146db09013e9e70efaf3140b101e922dff4

y4m_input_open: check allocs

Bug: aomedia:3276
Change-Id: I99ee0ef3ab28a22923cb413ccf5935fdc38862be

common/y4minput.c

diff --git a/common/y4minput.c b/common/y4minput.c
index 8e20b49..d64b6e4 100644
--- a/common/y4minput.c
+++ b/common/y4minput.c
@@ -1141,9 +1141,15 @@
     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;
 }