Утечка памяти в ffmpeg в методе avcodec_open2 - PullRequest
0 голосов
/ 11 июня 2018

Я разработал приложение, которое обрабатывает потоковое видео в реальном времени.Проблема в том, что он должен работать как сервис, и со временем я замечаю некоторое увеличение памяти.Когда я проверяю приложение с помощью valgrind - оно не обнаружило проблем, связанных с утечкой.Поэтому я проверил это с помощью инструментов профиля Google.Это результат (вычитание одного из первых дампов из последних) после приблизительно 6-часового прогона:

   30.0  35.7%  35.7%     30.0  35.7% av_malloc
    28.9  34.4%  70.2%     28.9  34.4% av_reallocp
    24.5  29.2%  99.4%     24.5  29.2% x264_malloc

Когда я проверяю память на графике, я вижу, что эти выделения связаны с avcodec_open2.Код клиента:

`           g_EncoderMutex.lock();
            ffmpeg_encoder_start(OutFileName.c_str(), AV_CODEC_ID_H264, m_FPS, width, height);
            for (pts = 0; pts < VideoImages.size(); pts++) {                
                m_frame->pts = pts;
                ffmpeg_encoder_encode_frame(VideoImages[pts].RGBimage[0]);
            }
            ffmpeg_encoder_finish();
            g_EncoderMutex.unlock()

Метод ffmpeg_encoder_start:

 void VideoEncoder::ffmpeg_encoder_start(const char *filename, int codec_id, int fps, int width, int height)
        {
            int ret;
            m_FPS=fps;
            AVOutputFormat * fmt = av_guess_format(filename, NULL, NULL);
            m_oc = NULL;
            avformat_alloc_output_context2(&m_oc, NULL, NULL, filename);

            m_stream = avformat_new_stream(m_oc, 0);
            AVCodec *codec=NULL;

            codec =  avcodec_find_encoder(codec_id);    
            if (!codec) 
            {
                fprintf(stderr, "Codec not found\n");
                return; //-1
            }

            m_c=m_stream->codec;

            avcodec_get_context_defaults3(m_c, codec);

            m_c->bit_rate = 400000;
            m_c->width = width;
            m_c->height = height;
            m_c->time_base.num = 1;
            m_c->time_base.den = m_FPS;
            m_c->gop_size = 10;
            m_c->max_b_frames = 1;
            m_c->pix_fmt = AV_PIX_FMT_YUV420P;
            if (codec_id == AV_CODEC_ID_H264)
                av_opt_set(m_c->priv_data, "preset", "ultrafast", 0);

            if (m_oc->oformat->flags & AVFMT_GLOBALHEADER) 
                m_c->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
            avcodec_open2( m_c, codec, NULL );

            m_stream->time_base=(AVRational){1, m_FPS};

            if (avio_open(&m_oc->pb, filename, AVIO_FLAG_WRITE) < 0)
            {
                printf( "Could not open '%s'\n", filename);
                exit(1);
            }            

            avformat_write_header(m_oc, NULL);
            m_frame = av_frame_alloc();
            if (!m_frame) {
                printf( "Could not allocate video frame\n");
                exit(1);
            }
            m_frame->format = m_c->pix_fmt;
            m_frame->width  = m_c->width;
            m_frame->height = m_c->height;
            ret = av_image_alloc(m_frame->data, m_frame->linesize, m_c->width, m_c->height, m_c->pix_fmt, 32);
            if (ret < 0) {
                printf("Could not allocate raw picture buffer\n");
                exit(1);
            }
        }

Код ffmpeg_encoder_encode_frame:

void VideoEncoder::ffmpeg_encoder_encode_frame(uint8_t *rgb) 
{
    int ret, got_output;
    ffmpeg_encoder_set_frame_yuv_from_rgb(rgb);
    av_init_packet(&m_pkt);
    m_pkt.data = NULL;
    m_pkt.size = 0;

    ret = avcodec_encode_video2(m_c, &m_pkt, m_frame, &got_output);
    if (ret < 0) {
        printf("Error encoding frame\n");
        exit(1);
    }
    if (got_output) 
    {

         av_packet_rescale_ts(&m_pkt,
                        (AVRational){1, m_FPS}, m_stream->time_base);
        m_pkt.stream_index = m_stream->index;
        int ret = av_interleaved_write_frame(m_oc, &m_pkt);

        av_packet_unref(&m_pkt);

    }

}

ffmpeg_encoder_finish код:

**

Этот код выполняется несколько раз в цикле.Итак, мой вопрос - что я делаю не так?может быть, ffmpeg использует какую-то внутреннюю буферизацию?Если так, как это отключить?Потому что такое увеличение использования памяти вообще недопустимо.

Ответы [ 3 ]

0 голосов
/ 20 июня 2018

Вы не закрыли контекст кодера.Добавьте avcodec_close(m_c) к ffmpeg_encoder_finish().

См. ffmpeg.org

Пользователь должен вызвать avcodec_close () и avformat_free_context () для очистки выделения с помощью avformat_new_stream ().

Плюс я не вижу, как распределяется m_c.Обычно он назначается с avcodec_alloc_context и должен быть освобожден с av_free (после закрытия курса).

0 голосов
/ 21 августа 2019

Достаточно вызвать 'avcodec_free_context (m_c)', эта процедура вызывает 'avcodec_close', а также отменяет выделение 'extradata' (если она была выделена) и 'subtitle_header' (если она была выделена).

0 голосов
/ 12 июня 2018

Не используйте valgrind для проверки утечек памяти для ваших собственных проектов, используйте дезинфицирующие средства, с помощью которых вы можете точно определить источник утечки.Проверьте это: Многопоточный видеодекодер теряет память

Надеюсь, это поможет.

...