commit 2ee5c8651487377fe820ed917ebba030fc5455c8
author James Zern <jzern@google.com> Thu Apr 28 12:54:06 2022 -0700
committer James Zern <jzern@google.com> Tue May 17 01:41:45 2022 +0000
tree 7f72649d73460266ea865f04e70f119639593271
parent 723bbf9e82d4fa8fb5d0a56643f6d2a090c351ee

av1_pick_filter_restoration: add missing alloc check

Bug: aomedia:3276
Change-Id: I54b7dedead15a2e9bab2532e0733c27635e042e0

av1/encoder/pickrst.c

diff --git a/av1/encoder/pickrst.c b/av1/encoder/pickrst.c
index d01c11c..008c469 100644
--- a/av1/encoder/pickrst.c
+++ b/av1/encoder/pickrst.c
@@ -1756,8 +1756,10 @@
     ntiles[is_uv] = rest_tiles_in_plane(cm, is_uv);
 
   assert(ntiles[1] <= ntiles[0]);
-  RestUnitSearchInfo *rusi =
-      (RestUnitSearchInfo *)aom_memalign(16, sizeof(*rusi) * ntiles[0]);
+  RestUnitSearchInfo *rusi;
+  CHECK_MEM_ERROR(
+      cm, rusi,
+      (RestUnitSearchInfo *)aom_memalign(16, sizeof(*rusi) * ntiles[0]));
 
   // If the restoration unit dimensions are not multiples of
   // rsi->restoration_unit_size then some elements of the rusi array may be