diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c
index 2b14d48ff58755251940ccfb58dfee6c5437bd20..fa63ddf2b901361a0bba7e1190133c0c68fac0ae 100644
--- a/libavformat/aviobuf.c
+++ b/libavformat/aviobuf.c
@@ -113,6 +113,8 @@ AVIOContext *avio_alloc_context(
                   int64_t (*seek)(void *opaque, int64_t offset, int whence))
 {
     AVIOContext *s = av_mallocz(sizeof(AVIOContext));
+    if (!s)
+        return NULL;
     ffio_init_context(s, buffer, buffer_size, write_flag, opaque,
                   read_packet, write_packet, seek);
     return s;