commit db562b242496799d170753d17da7aa8a1f569408
author James Zern <jzern@google.com> Thu Apr 28 12:54:06 2022 -0700
committer James Zern <jzern@google.com> Wed May 18 00:13:28 2022 +0000
tree fa7e9c1cdc242218e67a2a675a03d3d4ee83e581
parent 7738e55654de9e098154527bca5afced0d1a55e5

tpl_model,mode_estimation: add missing alloc checks

Bug: aomedia:3276
Change-Id: If39cb118d54576f5ef078df96bf2d6e49c0d49bd

av1/encoder/tpl_model.c

diff --git a/av1/encoder/tpl_model.c b/av1/encoder/tpl_model.c
index ef322ae..238be92 100644
--- a/av1/encoder/tpl_model.c
+++ b/av1/encoder/tpl_model.c
@@ -503,6 +503,16 @@
   int64_t recon_error = 1;
   int64_t pred_error = 1;
 
+  if (!(predictor8 && src_diff && coeff && qcoeff && dqcoeff)) {
+    aom_free(predictor8);
+    aom_free(src_diff);
+    aom_free(coeff);
+    aom_free(qcoeff);
+    aom_free(dqcoeff);
+    aom_internal_error(cm->error, AOM_CODEC_MEM_ERROR,
+                       "Error allocating tpl data");
+  }
+
   memset(tpl_stats, 0, sizeof(*tpl_stats));
   tpl_stats->ref_frame_index[0] = -1;
   tpl_stats->ref_frame_index[1] = -1;