读取图像文件时 FFmpeg 泄漏

FFmpeg leak while reading image files

在使用最新版本 FFmpeg 读取图像文件时,我遇到了内存泄漏问题,我无法追踪。

似乎在用 avcodec_send_packetavcodec_receive_frame 填充 AVFrame 之后,我对 av_frame_free 的调用实际上并没有释放 AVBuffer 对象框架。我唯一没有释放的是 AVCodecContext。如果我尝试这样做,我会崩溃。

我已经创建了这个示例程序,它非常简单。这将循环打开、读取然后关闭相同的图像文件。在我的系统上,这会以惊人的速度泄漏内存。

#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>

int main(int argc, char **argv) {
    av_register_all();

    while(1) {
        AVFormatContext *fmtCtx = NULL;

        if (avformat_open_input(&fmtCtx, "/path/to/test.jpg", NULL, NULL) == 0) {
            if (avformat_find_stream_info(fmtCtx, NULL) >= 0) {
                for (unsigned int i = 0u; i < fmtCtx -> nb_streams; ++i) {
                    AVStream *stream = fmtCtx -> streams[i];
                    AVCodecContext *codecCtx = stream -> codec;
                    AVCodec *codec = avcodec_find_decoder(codecCtx -> codec_id);

                    if (avcodec_open2(codecCtx, codec, NULL) == 0) {
                        AVPacket packet;

                        if (av_read_frame(fmtCtx, &packet) >= 0) {
                            if (avcodec_send_packet(codecCtx, &packet) == 0) {
                                AVFrame *frame = av_frame_alloc();

                                avcodec_receive_frame(codecCtx, frame);
                                av_frame_free(&frame);
                            }
                        }

                        av_packet_unref(&packet);
                    }
                }
            }

            avformat_close_input(&fmtCtx);
        }
    }

    return 0;
}

解决方案是创建文件打开时自动创建的 AVCodecContext 的副本,并在 avcodec_open2 中使用此副本。这允许使用 avcodec_free_context.

删除此副本

FFmpeg 的最新版本中,avcodec_copy_context 已被弃用并替换为 AVCodecParameters。在问题的示例程序中使用以下代码片段来填补漏洞:

AVCodecParameters *param = avcodec_parameters_alloc();
AVCodecContext *codecCtx = avcodec_alloc_context3(NULL);
AVCodec *codec = avcodec_find_decoder(stream -> codec -> codec_id);

avcodec_parameters_from_context(param, stream -> codec);
avcodec_parameters_to_context(codecCtx, param);
avcodec_parameters_free(&param);
[...]
avcodec_free_context(&codecCtx);