音视频入门基础:FLV专题(11)——FFmpeg源码中,解析SCRIPTDATASTRING类型的ScriptDataValue的实现
一、引言
从《音视频入门基础:FLV专题(9)——Script Tag简介》中可以知道,根据《video_file_format_spec_v10_1.pdf》第80到81页,SCRIPTDATAVALUE类型由一个8位(1字节)的Type和一个ScriptDataValue组成。其中Type属性用来指定ScriptDataValue的类型,根据Type值的不同,ScriptDataValue的类型也不同。当Type的值为2时,ScriptDataValue为SCRIPTDATASTRING类型:
根据《video_file_format_spec_v10_1.pdf》第83页,SCRIPTDATASTRING类型用来记录存贮不超过65535个字符的字符串数据。SCRIPTDATASTRING类型由一个16位(2字节)的StringLength属性和一个存贮字符数据的StringData属性组成。其中StringLength属性用来指定要存贮的字符串的长度,单位为字节;StringData属性存贮实际的字符串数据(不超过65535个字符):
FFmpeg源码中通过amf_get_string函数解析SCRIPTDATASTRING类型的ScriptDataValue。
二、amf_get_string函数的定义
amf_get_string函数定义在FFmpeg源码(本文演示用的FFmpeg源码版本为7.0.1)的源文件libavformat/flvdec.c中:
static int amf_get_string(AVIOContext *ioc, char *buffer, int buffsize)
{
int ret;
int length = avio_rb16(ioc);
if (length >= buffsize) {
avio_skip(ioc, length);
return -1;
}
ret = avio_read(ioc, buffer, length);
if (ret < 0)
return ret;
if (ret < length)
return AVERROR_INVALIDDATA;
buffer[length] = '\0';
return length;
}
该函数的作用就是解析SCRIPTDATASTRING类型的ScriptDataValue。
形参ioc:既是输入型参数也是输出型参数。指向一个AVIOContext(字节流上下文结构体)变量。关于AVIOContext结构体可以参考:《FFmpeg源码:avio_r8、avio_rl16、avio_rl24、avio_rl32、avio_rl64函数分析》。执行amf_get_string函数前,ioc->buf_ptr必须指向SCRIPTDATASTRING类型的ScriptDataValue的第一个字节。
形参buffer:输出型参数。执行amf_get_string函数后,形参buffer指向的缓冲区会得到从SCRIPTDATASTRING类型的ScriptDataValue中解析出来的实际的字符串数据,也就是SCRIPTDATASTRING的StringData属性中存贮的实际的字符串数据。
形参buffsize:输入型参数。形参buffer指向的缓冲区的大小,单位为字节。该值一般为32。
返回值:解析失败返回一个负数,解析成功返回“SCRIPTDATASTRING的StringData属性中存贮的实际的字符串数据的长度,单位为字节”。
三、amf_get_string函数的内部实现原理
amf_get_string函数中,首先通过avio_rb16函数得到SCRIPTDATASTRING的StringLength属性,也就是SCRIPTDATASTRING的StringData属性中存贮的实际的字符串数据的长度,赋值给局部变量length。关于avio_rb16函数的用法可以参考:《FFmpeg源码:avio_r8、avio_rl16、avio_rl24、avio_rl32、avio_rl64函数分析》:
int length = avio_rb16(ioc);
虽然FLV的官方文档《video_file_format_spec_v10_1.pdf》写着SCRIPTDATASTRING类型可以用来记录存贮不超过65535个字符的字符串数据。但是FFmpeg源码中做了限制,为了避免形参buffer指向的缓冲区不够大无法容纳解析出来的字符串(FFmpeg源码中形参buffer指向的缓冲区的大小一般为32字节),导致数组溢出,如果判断出表达式“length >= buffsize”为真,通过avio_skip函数让AVIOContext文件位置指针跳过length个字节,并且让amf_get_string函数返回-1。关于avio_skip函数的用法可以参考:《FFmpeg源码:avio_skip函数分析》:
if (length >= buffsize) {
avio_skip(ioc, length);
return -1;
}
通过avio_read函数从AVIOContext输入缓冲区/文件描述符去读取FLV文件中的数据,总共读取length个字节,从而得到SCRIPTDATASTRING的StringData属性中存贮的实际的字符串数据。关于avio_read函数的用法可以参考:《FFmpeg源码:avio_read函数分析》。如果ret小于0,表示调用avio_read函数出错,返回ret的值;如果ret < length表示实际读取到的字节数小于SCRIPTDATASTRING的StringLength属性,也表示出错了, 返回AVERROR_INVALIDDATA:
ret = avio_read(ioc, buffer, length);
if (ret < 0)
return ret;
if (ret < length)
return AVERROR_INVALIDDATA;
在形参buffer指向的缓冲区的有效数据之后添加字符'\0',用于标记字符串的结束:
buffer[length] = '\0';
解析成功返回“SCRIPTDATASTRING的StringData属性中存贮的实际的字符串数据的长度,单位为字节”:
return length;