X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavformat%2Ftests%2Ffifo_muxer.c;h=e20bd6e7b2b1c5ae0f48145d5a7a9fefa3f17949;hb=8b13492c9ecf85f1f674f99df370a19992b3910a;hp=9659198ff176d141ce39c7279f2073b3d90cf124;hpb=127cc6dd3d0b8ce8875aecacc5c2aaf45bf52243;p=ffmpeg

diff --git a/libavformat/tests/fifo_muxer.c b/libavformat/tests/fifo_muxer.c
index 9659198ff17..e20bd6e7b2b 100644
--- a/libavformat/tests/fifo_muxer.c
+++ b/libavformat/tests/fifo_muxer.c
@@ -25,6 +25,7 @@
 #include "libavutil/avassert.h"
 #include "libavformat/avformat.h"
 #include "libavformat/url.h"
+#include "libavformat/network.h"
 
 #define MAX_TST_PACKETS 128
 #define SLEEPTIME_50_MS 50000
@@ -150,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));