From 6c3cb02a742f0ce32a85e86738a18e3d6d711d59 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Thu, 15 Jan 2015 16:51:08 +0100 Subject: [PATCH] avcodec/flac_parser: Check for av_malloc() failure Signed-off-by: Michael Niedermayer --- libavcodec/flac_parser.c | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/libavcodec/flac_parser.c b/libavcodec/flac_parser.c index bc874c558b..d16af29efd 100644 --- a/libavcodec/flac_parser.c +++ b/libavcodec/flac_parser.c @@ -192,6 +192,11 @@ static int find_headers_search_validate(FLACParseContext *fpc, int offset) (*end_handle)->offset = offset; (*end_handle)->link_penalty = av_malloc(sizeof(int) * FLAC_MAX_SEQUENTIAL_HEADERS); + if (!(*end_handle)->link_penalty) { + av_freep(end_handle); + return AVERROR(ENOMEM); + } + for (i = 0; i < FLAC_MAX_SEQUENTIAL_HEADERS; i++) (*end_handle)->link_penalty[i] = FLAC_HEADER_NOT_PENALIZED_YET; -- 2.11.0