shithub: dav1d

Info  •  Files  •  Log  •  Branches

memory allocation error checks

--- a/src/decode.c

+++ b/src/decode.c

@@ -2637,6 +2637,7 @@

                 dav1d_alloc_aligned(sizeof(int32_t) * 3 *

                                     f->sb128w * f->sb128h * 128 * 128, 32);

             if (!f->frame_thread.b || !f->frame_thread.pal_idx ||

+                !f->frame_thread.pal || !f->frame_thread.cbi ||

                 !f->frame_thread.cf)

             {

                 goto error;

@@ -3242,6 +3243,10 @@

             // put the new values. Allocate them here (the data

             // actually gets set elsewhere)

             f->cur_segmap_ref = dav1d_ref_create(f->b4_stride * 32 * f->sb128h);

+            if (!f->cur_segmap_ref) {

+                res = -ENOMEM;

+                goto error;

+            }

             f->cur_segmap = f->cur_segmap_ref->data;

         } else if (f->prev_segmap_ref) {

             // We're not updating an existing map, and we have a valid

@@ -3252,6 +3257,10 @@

         } else {

             // We need to make a new map. Allocate one here and zero it out.

             f->cur_segmap_ref = dav1d_ref_create(f->b4_stride * 32 * f->sb128h);

+            if (!f->cur_segmap_ref) {

+                res = -ENOMEM;

+                goto error;

+            }

             f->cur_segmap = f->cur_segmap_ref->data;

             memset(f->cur_segmap_ref->data, 0, f->b4_stride * 32 * f->sb128h);

         }