FFmpeg源码：bytestream2_peek_be32函数分析

cuijiecheng20182024-06-22 13:43

一、bytestream2_peek_be32函数的声明

FFmpeg源码中bytestream2_peek_be32函数的声明如下：

cpp 复制代码

DEF(unsigned int, be32, 4, AV_RB32, AV_WB32)

该宏定义有多层。把它简化，其函数声明可以等价于：

cpp 复制代码

static unsigned int bytestream2_peek_be32(GetByteContext *g);

GetByteContext 结构体的定义为：

cpp 复制代码

typedef struct GetByteContext {
    const uint8_t *buffer, *buffer_end, *buffer_start;
} GetByteContext;

该结构体的用法一般是：成员buffer_start永久指向某个缓冲区的开头，成员buffer_end永久指向该缓冲区的末尾，成员buffer指向该缓冲区正在被读取的位置。

bytestream2_peek_be32函数作用是：如果g->buffer_start指向的缓冲区距离被读取到末尾小于4个字节（g->buffer_end - g->buffer < 4），返回0。否则返回"按照大端模式读取到的g->buffer指向的地方往后的4个字节"。

二、bytestream2_peek_be32函数的实现原理

FFmpeg源码目录下的libavcodec/bytestream.h中存在宏定义：

cpp 复制代码

DEF(unsigned int, be32, 4, AV_RB32, AV_WB32)

#define DEF(type, name, bytes, read, write)

static av_always_inline type bytestream2_peek_ ## name ## u(GetByteContext *g) \
{                                                                              \
    return read(g->buffer);                                                    \
}                          
                                                    \
static av_always_inline type bytestream2_peek_ ## name(GetByteContext *g)      \
{                                                                              \
    if (g->buffer_end - g->buffer < bytes)                                     \
        return 0;                                                              \
    return bytestream2_peek_ ## name ## u(g);                                  \
}

所以FFmpeg源码中的DEF(unsigned int, be32, 4, AV_RB32, AV_WB32) 等价于定义了函数：

cpp 复制代码

static __attribute__((always_inline)) inline unsigned int bytestream2_peek_be32u(GetByteContext *g)
{                                                                              
    return AV_RB32(g->buffer);                                                    
}

static __attribute__((always_inline)) inline unsigned int bytestream2_peek_be32(GetByteContext *g)
{                                                                             
    if (g->buffer_end - g->buffer < 4)                                     
        return 0;                                                              
    return bytestream2_peek_be32u(g);                                  \
}

可以看到bytestream2_peek_be32函数调用了 bytestream2_peek_be32u函数，而bytestream2_peek_be32u函数又调用了AV_RB32这个宏定义。而根据《FFmpeg源码：AV_RB32宏定义分析》可以知道，AV_RB32(g->buffer)的作用是按照大端模式读取g->buffer指向的缓冲区的前四个字节，并返回。从而可以了解到bytestream2_peek_be32函数的作用。