commit 1e92154cd925794ebcf20e8d110d89fd1231cbff
author James Zern <jzern@google.com> Thu Apr 28 12:54:06 2022 -0700
committer James Zern <jzern@google.com> Fri May 06 18:35:17 2022 +0000
tree eee63f9cbf326dacb81da01a3329e810c2fc6acb
parent 8ff2856bffb74a1dd5d721a6453a51284f477b99

av1_alloc_txb_buf: add missing alloc checks

Bug: aomedia:3276
Change-Id: Ia2e1c6e596b2ad8ccf158fd8d07fa95488644654

av1/encoder/encodetxb.c

diff --git a/av1/encoder/encodetxb.c b/av1/encoder/encodetxb.c
index c059366..4ea4f4c 100644
--- a/av1/encoder/encodetxb.c
+++ b/av1/encoder/encodetxb.c
@@ -44,14 +44,17 @@
 
   av1_free_txb_buf(cpi);
   // TODO(jingning): This should be further reduced.
-  cpi->coeff_buffer_base = aom_malloc(sizeof(*cpi->coeff_buffer_base) * size);
+  CHECK_MEM_ERROR(cm, cpi->coeff_buffer_base,
+                  aom_malloc(sizeof(*cpi->coeff_buffer_base) * size));
   CHECK_MEM_ERROR(
       cm, coeff_buf_pool->tcoeff,
       aom_memalign(32, sizeof(*coeff_buf_pool->tcoeff) * num_tcoeffs));
-  coeff_buf_pool->eobs =
-      aom_malloc(sizeof(*coeff_buf_pool->eobs) * num_tcoeffs / txb_unit_size);
-  coeff_buf_pool->entropy_ctx = aom_malloc(
-      sizeof(*coeff_buf_pool->entropy_ctx) * num_tcoeffs / txb_unit_size);
+  CHECK_MEM_ERROR(
+      cm, coeff_buf_pool->eobs,
+      aom_malloc(sizeof(*coeff_buf_pool->eobs) * num_tcoeffs / txb_unit_size));
+  CHECK_MEM_ERROR(cm, coeff_buf_pool->entropy_ctx,
+                  aom_malloc(sizeof(*coeff_buf_pool->entropy_ctx) *
+                             num_tcoeffs / txb_unit_size));
 
   tran_low_t *tcoeff_ptr = coeff_buf_pool->tcoeff;
   uint16_t *eob_ptr = coeff_buf_pool->eobs;