[FFmpeg-devel] [PATCH] libavformat/whip: Add av_freep() to avoid potential memory leak

[Jiasheng Jiang]

Nicolas George (HE12025-08-07)
> Jiasheng Jiang (HE12025-08-07):
>> Add av_freep() if avio_alloc_context() fails to avoid potential memory leak.
>> 
>> Fixes: 167e343bbe ("avformat/whip: Add WHIP muxer support for subsecond latency streaming")
>> Signed-off-by: Jiasheng Jiang <jiashengjiangcool at gmail.com>
>> ---
>>  libavformat/whip.c | 1 +
>>  1 file changed, 1 insertion(+)
>> 
>> diff --git a/libavformat/whip.c b/libavformat/whip.c
>> index 256ea14d2c..82a5a51029 100644
>> --- a/libavformat/whip.c
>> +++ b/libavformat/whip.c
>> @@ -1553,6 +1553,7 @@ static int create_rtp_muxer(AVFormatContext *s)
>>  
>>          rtp_ctx->pb = avio_alloc_context(buffer, buffer_size, 1, s, NULL, on_rtp_write_packet, NULL);
>>          if (!rtp_ctx->pb) {
>
>> +            av_freep(&buffer);
>>              ret = AVERROR(ENOMEM);
>>              goto end;
>
> That belong to after end, otherwise you need to make the change
> everywhere.
>
>>          }
>
> Regards,

I found that after the success of avio_alloc_context(), buffer is assigned to rtp_ctx->pb->buffer and will be freed in whip_deinit(). Therefore, I think moving av_freep() after end would cause a double free.

- Jiasheng