From 8baaa7b5a3fcea958261bda667a871c163e93bf9 Mon Sep 17 00:00:00 2001 From: James Zern Date: Tue, 26 Apr 2022 22:20:00 -0700 Subject: [PATCH] y4m_input_open: check allocs Change-Id: I99ee0ef3ab28a22923cb413ccf5935fdc38862be --- y4minput.c | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/y4minput.c b/y4minput.c index 9a4bdbd7b..7d3c03a7f 100644 --- a/y4minput.c +++ b/y4minput.c @@ -1087,9 +1087,15 @@ int y4m_input_open(y4m_input *y4m_ctx, FILE *file, char *skip_buffer, y4m_ctx->dst_buf = (unsigned char *)malloc(y4m_ctx->dst_buf_sz); else y4m_ctx->dst_buf = (unsigned char *)malloc(2 * y4m_ctx->dst_buf_sz); + if (!y4m_ctx->dst_buf) return -1; - if (y4m_ctx->aux_buf_sz > 0) + if (y4m_ctx->aux_buf_sz > 0) { y4m_ctx->aux_buf = (unsigned char *)malloc(y4m_ctx->aux_buf_sz); + if (!y4m_ctx->aux_buf) { + free(y4m_ctx->dst_buf); + return -1; + } + } return 0; } -- 2.50.0