FFmpeg内存IO模式(内存区作输入或输出)

本文为作者原创，转载请注明出处：https://www.cnblogs.com/leisure_chn/p/10318145.html

所谓内存IO，在FFmpeg中叫作“buffered IO”或“custom IO”，指的是将一块内存缓冲区用作FFmpeg的输入或输出。与内存IO操作对应的是指定URL作为FFmpeg的输入或输出，比如URL可能是普通文件或网络流地址等。这两种输入输出模式我们暂且称作“内存IO模式”和“URL-IO模式”。

本文源码基于FFmpeg 4.1版本，为帮助理解，可参考FFmpeg工程examples中如下两份代码：
https://github.com/FFmpeg/FFmpeg/blob/n4.1/doc/examples/avio_reading.c
https://github.com/FFmpeg/FFmpeg/blob/n4.1/doc/examples/remuxing.c

1. 内存区作输入

1.1 用法

用法如示例中注释的步骤，如下：

// @opaque  : 是由用户提供的参数，指向用户数据
// @buf     : 作为FFmpeg的输入，此处由用户准备好buf中的数据
// @buf_size: buf的大小
// @return  : 本次IO数据量
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
{
    int fd = *((int *)opaque);
    int ret = read(fd, buf, buf_size);
    return ret;
}

int main()
{
    AVFormatContext *ifmt_ctx = NULL;
    AVIOContext *avio_in = NULL;
    uint8_t *ibuf = NULL;
    size_t ibuf_size = 4096;
    int fd = -1;

    // 打开一个FIFO文件的读端
    fd = open_fifo_for_read("/tmp/test_fifo");

    // 1. 分配缓冲区
    ibuf = av_malloc(ibuf_size);

    // 2. 分配AVIOContext，第三个参数write_flag为0
    avio_in = avio_alloc_context(ibuf, ibuf_size, 0, &fd, &read_packet, NULL, NULL);

    // 3. 分配AVFormatContext，并指定AVFormatContext.pb字段。必须在调用avformat_open_input()之前完成
    ifmt_ctx = avformat_alloc_context();
    ifmt_ctx->pb = avio_in;

    // 4. 打开输入(读取封装格式文件头)
    avformat_open_input(&ifmt_ctx, NULL, NULL, NULL);
    
    ......
}

当启用内存IO模式后(即ifmt_ctx->pb有效时)，将会忽略avformat_open_input()第二个参数url的值。在上述示例中，打开了FIFO的读端，并在回调函数中将FIFO中的数据填入内存缓冲区ibuf，内存缓冲区ibuf将作为FFmpeg的输入。在上述示例中，因为打开的是一个命名管道FIFO，FIFO的数据虽然在内存中，但FIFO有名字("/tmp/test_fifo")，所以此例也可以使用URL-IO模式，如下：

AVFormatContext *ifmt_ctx = NULL;
avformat_open_input(&ifmt_ctx, "/tmp/test_fifo", NULL, NULL);

而对于其他一些场合，当有效音视频数据位于内存，而这片内存并无一个URL属性可用时，则只能使用内存IO模式来取得输入数据。

1.2 回调时机

回调函数何时被回调呢？所有需要从输入源中读取数据的时刻，都将调用回调函数。和输入源是普通文件相比，只不过输入源变成了内存区，其他各种外在表现并无不同。

如下各函数在不同的阶段从输入源读数据，都会调用回调函数：
avformat_open_input() 从输入源读取封装格式文件头
avformat_find_stream_info() 从输入源读取一段数据，尝试解码，以获取流信息
av_read_frame() 从输入源读取数据包

2. 内存区作输出

2.1 用法

用法如示例中注释的步骤，如下：

// @opaque  : 是由用户提供的参数，指向用户数据
// @buf     : 作为FFmpeg的输出，此处FFmpeg已准备好buf中的数据
// @buf_size: buf的大小
// @return  : 本次IO数据量
static int write_packet(void *opaque, uint8_t *buf, int buf_size)
{
    int fd = *((int *)opaque);
    int ret = write(fd, buf, buf_size);
    return ret;
}
 
int main()
{
    AVFormatContext *ofmt_ctx = NULL;
    AVIOContext *avio_out =