y4m output: make sure image widths are valid

strukturag · Oct 20, 2024 · b56f7c1 · b56f7c1
1 parent 09881e1
commit b56f7c1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/heifio/encoder_y4m.cc b/heifio/encoder_y4m.cc
@@ -64,6 +64,10 @@ bool Y4MEncoder::Encode(const struct heif_image_handle* handle,
   int cw = heif_image_get_width(image, heif_channel_Cb);
   int ch = heif_image_get_height(image, heif_channel_Cb);
 
+  if (yw < 0 || cw < 0) {
+    return false;
+  }
+
   fprintf(fp, "YUV4MPEG2 W%d H%d F30:1\nFRAME\n", yw, yh);
 
   for (int y = 0; y < yh; y++) {