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 初稿