Merge commit 'f8ab9f2fe3730a0e8c3c1eb7c486f7c0041fb3fc'

[ffmpeg] / libavutil / pixelutils.c

diff --git a/libavutil/pixelutils.c b/libavutil/pixelutils.c
index ca424d89e463e93f3d3329e1fca6bfc5462edea8..10ff7e801a6be31428447d8d2952f2f7db724a98 100644 (file)
--- a/libavutil/pixelutils.c
+++ b/libavutil/pixelutils.c
@@ -128,6 +128,12 @@ int main(void)
     uint8_t *buf2 = av_malloc(W2*H2);
     uint32_t state = 0;
 
+    if (!buf1 || !buf2) {
+        fprintf(stderr, "malloc failure\n");
+        ret = 1;
+        goto end;
+    }
+
     for (i = 0; i < W1*H1; i++) {
         state = state * 1664525 + 1013904223;
         buf1[i] = state>>24;