From d1f3e475f9807b445ba37ff2fd23f71c4645de79 Mon Sep 17 00:00:00 2001 From: Steven Liu Date: Fri, 6 Jan 2017 06:35:32 +0800 Subject: [PATCH] avformat/test/fifo_muxer: add check for FailingMuxerPacketData alloc CID: 1396257 Reviewed-by: Michael Niedermayer Signed-off-by: Steven Liu --- libavformat/tests/fifo_muxer.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/libavformat/tests/fifo_muxer.c b/libavformat/tests/fifo_muxer.c index dc62965a15..e20bd6e7b2 100644 --- a/libavformat/tests/fifo_muxer.c +++ b/libavformat/tests/fifo_muxer.c @@ -151,6 +151,9 @@ static int prepare_packet(AVPacket *pkt,const FailingMuxerPacketData *pkt_data, { int ret; FailingMuxerPacketData *data = av_malloc(sizeof(*data)); + if (!data) { + return AVERROR(ENOMEM); + } memcpy(data, pkt_data, sizeof(FailingMuxerPacketData)); ret = av_packet_from_data(pkt, (uint8_t*) data, sizeof(*data));