FFmpeg视频播放的内存管理

在写这个播放器的时候,遇到了一些内存管理的问题,虽然棘手但是也让我对此有了比较完善的理解,而且很多相关资料并没有跟随FFmpeg的更新,比如缓冲池AVBufferPool的使用。

使用ffmpeg版本是3.4

AVFrame和AVPacket的内存管理策略

对AVFrame:

  • av_frame_alloc只是给AVFrame分配了内存,它内部的buf还是空的,就相当于造了一个箱子,但箱子里是空的。
  • av_frame_ref对src的buf增加一个引用,即使用同一个数据,只是这个数据引用计数+1.av_frame_unref把自身对buf的引用释放掉,数据的引用计数-1。
  • av_frame_free内部还是调用了unref,只是把传入的frame也置空。

发现还缺了一个buffer初始化的方法,初始化就在解码函数avcodec_send_packetavcodec_receive_frame内部。

然后对于解码有个坑,对avcodec_receive_frame函数:

Note that the function will always call
av_frame_unref(frame) before doing anything else.

如果你使用同一个frame,每次去接收解码后的数据,那么每次传进去就会把前面的数据释放掉,导致就只有一个frame是有用的。

如果你觉得frame的alloc花费很大,想节省资源,然后又没注意到这个注释的话,很可能就会这么做。

对此有两种方案:

  1. 继续只使用一个frame来接收,但是在传递给下一步(渲染、播放等)的时候,下一步的模块使用一个新的frame和av_frame_ref来接收,而不是直接的赋值。
  2. 每次解码前构建一个新的AVFrame,把它传给avcodec_receive_frame,这样每次都是新的frame,互不干扰。但是在整个流程结束时,要释放这个frame.

方便来说,是第二种方案好;但从模块化角度说,是第一种的更好,单解码这一步,要自己管理好自己的内存,即buffer的alloc和unref配套。这样内存的管理在当前的模块内部是完善的,如果出了问题,也只是其他模块出了问题。相比而言,第一种就是把内存的释放依赖在了其他模块的处理上。

AVPacket基本和AVFrame一致,只是获取packet的函数av_read_frame它并不会执行unref操作,而是直接把buf设为null。使用上面的两个方案之一也都可以规避这个问题。

不管怎样,直接的frame1=frame2这样的赋值是不可取的。当然要具体问题具体分析,时刻注意它内部是用引用计数的方式管理buf内的数据。

一点都没释放

最开始是播放停止后的内存几乎没有下降,解码后的AVFrame是用一个缓冲区来管理的,里面的frame是暂存没释放的,我以为是这个缓冲区里有留存,然后给它添加了释放方法,结束后每个frame都调用av_packet_free,然后奇怪的事出现了。

很明确每个frame都调用了free或者unref,但是内存却没什么改变。哪怕释放不干净,至少要少一点吧。难道是av_packet_free不起作用?我试着把播放完的frame的free取消,但内存在播放的时候就飙涨了,说明这个是有用的。

然后缓冲区有个最大数量限制,调大这个数量,内存就上涨,调小就下降。这可以理解,因为这里面的frame都是存在的,所以肯定会占内存。

结合上面一起就是:在结束播放后,缓冲区里的frame集体没有释放,一个都没有!

怎么查?看源码。

av_frame_free看,这个里面起作用的还是av_frame_unref,它的源码:

void av_buffer_unref(AVBufferRef **buf)
    {
        if (!buf || !*buf)
            return;
    
        buffer_replace(buf, NULL);
    }
    
     static void buffer_replace(AVBufferRef **dst, AVBufferRef **src)
    {
        AVBuffer *b;
    
        b = (*dst)->buffer;
    
        if (src) {
            **dst = **src;
            av_freep(src);
        } else
            av_freep(dst);
    
        if (atomic_fetch_add_explicit(&b->refcount, -1, memory_order_acq_rel) == 1) {
            b->free(b->opaque, b->data);
            av_freep(&b);
        }
    }

所以关键点就是atomic_fetch_add_explicit,这个函数有一个系列,就是进行原子性的加减乘除的,这个函数是先fetchadd,先查询再增加,所以返回的值是修改之前的。

atomic_fetch_add_explicit(&b->refcount, -1, memory_order_acq_rel) == 1整句代码就是:如果当前引用计数为1,就释放数据,因为加-1,所以条件等价于引用计数为0。

AVFrame和AVPacket的重量级数据都存在它们的buf里,data和extend_data都是从数据里引用过来的,buf是AVBufferRef类型,表示一个对于AVBuffer的引用,多一个引用,AVBuffer的引用计数就+1,少一个就-1,没有引用就释放,AVBuffer是数据的真身。对于AVFrame和AVPacket的内存管理就是依赖av_xxx_refav_xxx_unref这一套函数。

然后就是看一下b->free(b->opaque, b->data);这个具体调用了什么函数。在AVBuffer的文档里有个void av_buffer_default_free(void *opaque, uint8_t *data);,说是默认的释放函数,在释放AVBuffer时调用这个函数。这个函数就是调用了av_free,而av_free就是调用了free,也就是单纯的释放内存罢了。

如果b->free(b->opaque, b->data);真的是调用了这个默认的释放函数,那么内存一定会下降的。这里有个帮助很大但不知道原理的东西,就是Synbolic断点可以自动定位到源码,而且可以查看调用栈数据,相关知识只能查到这个。这样就可以在运行的时候直接看到b->free是什么东西了,它是pool_release_buffer!!!

static void pool_release_buffer(void *opaque, uint8_t *data)
{
   BufferPoolEntry *buf = opaque;
   AVBufferPool *pool = buf->pool;
   ...
   if (atomic_fetch_add_explicit(&pool->refcount, -1, memory_order_acq_rel) == 1)
       buffer_pool_free(pool);

这里面根本没有释放data的地方,同样是引用计数操作,然后到buffer_pool_free

/*
* This function gets called when the pool has been uninited and
* all the buffers returned to it.
*/
static void buffer_pool_free(AVBufferPool *pool)
{
   while (pool->pool) {
       BufferPoolEntry *buf = pool->pool;
       pool->pool = buf->next;

       buf->free(buf->opaque, buf->data);
       av_freep(&buf);
   }
   ff_mutex_destroy(&pool->mutex);

   if (pool->pool_free)
       pool->pool_free(pool->opaque);

   av_freep(&pool);
}

结合这个函数、pool这个名字还有上面那两行注释,以及我的测试可以得出:

  • pool是一个缓冲池,管理者众多的内存缓冲区(AVBuffer)
  • 从池里生成的buffer,在释放的时候,是再回到池里,并且池的引用计数-1。也就是这是一个循环使用的缓冲池,使用引用计数来标记内部的缓冲区。
  • 池构建(av_buffer_pool_init)的时候,引用计数为初始值1,调用av_buffer_pool_uninit标记为可销毁,引用计数减1,这两者刚好匹配。
  • 内部每生成一个buffer,引用计数+1,回收一个buffer,引用计数-1。这两者也是匹配的。
  • 结合上两点,只要合理操作,内存就可以得到释放。而没有释放,至少有一个没做到。
  • 循环缓冲池的作用是为了避免频繁的、大量的内存分配和释放,特别是视频帧数据,一帧就上百k。同时也解释了为什么内存一点都没有释放,使用了池,要么全部释放,要么一点都不释放。

从内部再回到外部,先检查是否有frame没有释放。这时确实是有的,就在:

retval = avcodec_receive_frame(decoder->codecCtx, frame);
            
if (retval != 0) {
    TFCheckRetval("avcodec receive frame");
    av_frame_free(&frame);//漏掉了这里
    continue;
 }

在解码失败后,就直接continue了。在意识里,好像这里的frame是无用的,没数据的,所以就直接忽略了,接下一个。就死在了这里。

在把这种的frame都释放时候,还是有问题,就剩下av_buffer_pool_uninit这个了。这个函数的调用里用户使用的外层很远,最终查到是从avcodec_close这里进入的。在逻辑也是合理的,解码结束了,才需要把分配的内存销毁。但是不要直接调用avcodec_close,而是使用avcodec_free_context,把codec相关的其他东西一并释放了。

到这,终于内存释放了。重点在于认识到有个pool的存在,这个在网上资料并不多。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 220,548评论 6 513
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,069评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 166,985评论 0 357
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,305评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,324评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,030评论 1 308
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,639评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,552评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,081评论 1 319
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,194评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,327评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,004评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,688评论 3 332
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,188评论 0 23
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,307评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,667评论 3 375
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,337评论 2 358

推荐阅读更多精彩内容