Check memory allocations.

AOMediaCodec · Nov 29, 2023 · ae06afc · ae06afc
1 parent 5b83d9f
commit ae06afc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/apps/avifgainmaputil/convert_command.cc b/apps/avifgainmaputil/convert_command.cc
@@ -79,6 +79,9 @@ avifResult ConvertCommand::Run() {
       depth = image->gainMap->metadata.alternateHdrHeadroomN == 0 ? 8 : 10;
     }
     ImagePtr new_base(avifImageCreateEmpty());
+    if (new_base == nullptr) {
+      return AVIF_RESULT_OUT_OF_MEMORY;
+    }
     const avifResult result =
         ChangeBase(*image, depth, image->yuvFormat, new_base.get());
     if (result != AVIF_RESULT_OK) {

diff --git a/apps/avifgainmaputil/swapbase_command.cc b/apps/avifgainmaputil/swapbase_command.cc
@@ -162,6 +162,9 @@ avifResult SwapBaseCommand::Run() {
   }
 
   ImagePtr new_base(avifImageCreateEmpty());
+  if (new_base == nullptr) {
+    return AVIF_RESULT_OUT_OF_MEMORY;
+  }
   result = ChangeBase(*image, depth, pixel_format, new_base.get());
   if (result != AVIF_RESULT_OK) {
     return result;