FFmpeg資料結構AVBuffer

作者： 葉餘 來源： https://www.cnblogs.com/leisure_chn/p/10399048.html

AVBuffer是FFmpeg中很常用的一種緩沖區，緩沖區使用引用計數(reference-counted)機制。

AVBufferRef則對AVBuffer緩沖區提供了一層封裝，最主要的是作引用計數處理，實作了一種安全機制。使用者不應直接通路AVBuffer，應通過AVBufferRef來通路AVBuffer，以保證安全。

FFmpeg中很多基礎的資料結構都包含了AVBufferRef成員，來間接使用AVBuffer緩沖區。

本文使用的FFmpeg版本号為FFmpeg 4.1。

AVBuffer和AVBufferRef結構體定義及操作函數位于libavutil中的buffer.h、buffer_internal.h、buffer.c三個檔案中。需要關注的要點是AVBufferRef和AVBuffer的關系以及緩沖區引用計數的概念。

1. 資料結構定義

1.1 struct AVBuffer

struct AVBuffer定義于“libavutil/buffer_internal.h”，buffer_internal.h位于FFmpeg工程源碼中，而FFmpeg提供的開發庫頭檔案中并無此檔案，是以這是一個内部資料結構，不向使用者開放，使用者不應直接通路AVBuffer，應通過AVBufferRef來通路AVBuffer，以保證安全。

struct AVBuffer {
    uint8_t *data; /**< data described by this buffer */
    int      size; /**< size of data in bytes */

    /**
     *  number of existing AVBufferRef instances referring to this buffer
     */
    atomic_uint refcount;

    /**
     * a callback for freeing the data
     */
    void (*free)(void *opaque, uint8_t *data);

    /**
     * an opaque pointer, to be used by the freeing callback
     */
    void *opaque;

    /**
     * A combination of BUFFER_FLAG_*
     */
    int flags;
};      

• data: 緩沖區位址
• size: 緩沖區大小
• refcount: 引用計數值
• free: 用于釋放緩沖區記憶體的回調函數
• opaque: 提供給free回調函數的參數
• flags: 緩沖區标志

1.2 struct AVBufferRef

struct AVBufferRef定義于buffer.h中：

/**
 * A reference to a data buffer.
 *
 * The size of this struct is not a part of the public ABI and it is not meant
 * to be allocated directly.
 */
typedef struct AVBufferRef {
    AVBuffer *buffer;
    /**
     * The data buffer. It is considered writable if and only if
     * this is the only reference to the buffer, in which case
     * av_buffer_is_writable() returns 1.
     */
    uint8_t *data;
    /**
     * Size of data in bytes.
     */
    int      size;
} AVBufferRef;      

• buffer: AVBuffer
• data: 緩沖區位址，實際等于buffer->data
• size: 緩沖區大小，實際等于buffer->size

2. 關鍵函數實作

2.1 av_buffer_alloc()

AVBufferRef *av_buffer_alloc(int size)
{
    AVBufferRef *ret = NULL;
    uint8_t    *data = NULL;
    data = av_malloc(size);
    if (!data)
        return NULL;
    ret = av_buffer_create(data, size, av_buffer_default_free, NULL, 0);
    if (!ret)
        av_freep(&data);
    return ret;
}      

av_buffer_alloc()作了如下處理:

a) 使用av_malloc配置設定緩沖區

b) 調用av_buffer_create()建立

AVBuffer AVBufferRef::*buffer

成員，用于管理AVBuffer緩沖區

c) 傳回

AVBufferRef *

對象

2.2 av_buffer_create()

AVBufferRef *av_buffer_create(uint8_t *data, int size,
                              void (*free)(void *opaque, uint8_t *data),
                              void *opaque, int flags)
{
    AVBufferRef *ref = NULL;
    AVBuffer    *buf = NULL;

    buf = av_mallocz(sizeof(*buf));
    if (!buf)
        return NULL;

    buf->data     = data;
    buf->size     = size;
    buf->free     = free ? free : av_buffer_default_free;
    buf->opaque   = opaque;

    atomic_init(&buf->refcount, 1);

    if (flags & AV_BUFFER_FLAG_READONLY)
        buf->flags |= BUFFER_FLAG_READONLY;

    ref = av_mallocz(sizeof(*ref));
    if (!ref) {
        av_freep(&buf);
        return NULL;
    }

    ref->buffer = buf;
    ref->data   = data;
    ref->size   = size;

    return ref;
}      

av_buffer_create()是一個比較核心的函數，從其實作代碼很容易看出AVBufferRef和AVBuffer這間的關系。

函數主要功能就是初始化

AVBuffer AVBufferRef::*buffer

成員，即為上述清單

ref->buffer

各字段指派，最終，

AVBufferRef *ref

全部構造完畢，将之傳回。

其中

void (*free)(void *opaque, uint8_t *data)

參數指派為

av_buffer_default_free

，實作如下。其實就是直接調用了

av_free

回收記憶體。

void av_buffer_default_free(void *opaque, uint8_t *data)
{
    av_free(data);
}      

2.3 av_buffer_ref()

AVBufferRef *av_buffer_ref(AVBufferRef *buf)
{
    AVBufferRef *ret = av_mallocz(sizeof(*ret));

    if (!ret)
        return NULL;

    *ret = *buf;

    atomic_fetch_add_explicit(&buf->buffer->refcount, 1, memory_order_relaxed);

    return ret;
}      

av_buffer_ref()處理如下：

a)

*ret = *buf;

一句将buf各成員值指派給ret中對應成員，buf和ret将共用同一份AVBuffer緩沖區

b)

atomic_fetch_add_explicit(...);

一句将AVBuffer緩沖區引用計數加1

注意此處的關鍵點：共用緩沖區(緩沖區不拷貝)，緩沖區引用計數加1

2.4 av_buffer_unref()

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);
    }
}

void av_buffer_unref(AVBufferRef **buf)
{
    if (!buf || !*buf)
        return;

    buffer_replace(buf, NULL);
}      

av_buffer_unref()處理如下：

a) 回收

AVBufferRef **buf

記憶體

b) 将

(*buf)->buffer

(即AVBAVBufferRef的成員AVBuffer)的引用計數減1，若引用計數為0，則通過

b->free(b->opaque, b->data);

調用回調函數回收AVBuffer緩沖區記憶體

注意此處的關鍵點：銷毀一個AVBufferRef時，将其AVBuffer緩沖區引用計數減1，若緩沖區引用計數變為0，則将緩沖區也回收，這很容易了解，隻有當緩沖區不被任何對象引用時，緩沖區才能被銷毀

3. 修改記錄

2018-12-13 V1.0 初稿

「視訊雲技術」你最值得關注的音視訊技術公衆号，每周推送來自阿裡雲一線的實踐技術文章，在這裡與音視訊領域一流工程師交流切磋。