From 6552b0558a15b05c49976b393f2d3123c317df3a Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Wed, 6 Aug 2014 01:45:29 +0200 Subject: avutil/pixelutils: check for malloc failure Found-by: Daemon404 Signed-off-by: Michael Niedermayer --- libavutil/pixelutils.c | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'libavutil') diff --git a/libavutil/pixelutils.c b/libavutil/pixelutils.c index ca424d89e4..10ff7e801a 100644 --- 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; -- cgit v1.2.3