《FFmpeg开发实战:从零基础到短视频上线》一书的"10.2.2 FFmpeg向网络推流"介绍了如何使用FFmpeg代码向网络推送视频流,当时的例程采用了RTSP方式推流,在向RTSP地址时推流是正常的,但向RTMP地址推流时出现了问题,下面就介绍了如何使用FFmpeg代码向RTMP地址推送视频流。
一、FFmpeg推流环境的区别
首先注意RTSP推流地址以"rtsp://"开头,且RTSP地址的默认端口号为8554。而RTMP推流地址以"rtmp://"开头,且RTMP地址的默认端口号为1935。
其次注意FFmpeg从6.1开始对RTMP协议做了增强支持,主要是支持HEVC、VP9和AV1等编码格式通过RTMP协议进行推流,所以建议将编译环境的FFmpeg版本升级到6.1或者更高版本。
二、FFmpeg推流代码的适配
FFmpeg推流代码对于RTSP地址和RTMP地址主要有下列两点适配区别:
1、调用avformat_alloc_output_context2函数分配音视频文件封装实例的时候,第三个输入参数对于RTSP地址而言要填rtsp,对于RTMP地址而言要填flv。比如以下代码通过判断推流地址的协议类型来决定avformat_alloc_output_context2的第三个参数要填何值。
int ret = 0;
// 分配音视频文件的封装实例(注意rtmp协议的第三个参数填flv,rtsp协议的第三个参数填rtsp)
if (strstr(dest_name, "rtmp") != NULL) {
ret = avformat_alloc_output_context2(&out_fmt_ctx, NULL, "flv", dest_name);
} else {
ret = avformat_alloc_output_context2(&out_fmt_ctx, NULL, "rtsp", dest_name);
}
if (ret < 0) {
av_log(NULL, AV_LOG_ERROR, "Can't alloc output_file %s.\n", dest_name);
return -1;
}
av_log(NULL, AV_LOG_INFO, "Success open push url %s.\n", dest_name);
2、调用avformat_alloc_output_context2函数之后,还要依据推流地址的协议类型来决定是否接着调用avio_open函数打开输出流,对于RTSP地址而言不必调用avio_open,对于RTMP地址而言必须调用avio_open。具体的判断与调用代码如下所示:
// 打开输出流(注意rtsp推流不要调用avio_open,但rtmp推流要调用avio_open)
if (strstr(dest_name, "rtmp") != NULL) {
ret = avio_open(&out_fmt_ctx->pb, dest_name, AVIO_FLAG_READ_WRITE);
if (ret < 0) {
av_log(NULL, AV_LOG_ERROR, "Can't open output_file %s.\n", dest_name);
return -1;
}
}
按照以上两点代码修改了《FFmpeg开发实战:从零基础到短视频上线》随书源码第十章的推流程序代码chapter10/pushvideo.c后,先按照之前的文章《详解MediaMTX的推拉流》启动电脑本地的流媒体服务器MediaMTX,再执行下面的编译命令。
gcc pushvideo.c -o pushvideo -I/usr/local/ffmpeg/include -L/usr/local/ffmpeg/lib -lavformat -lavdevice -lavfilter -lavcodec -lavutil -lswscale -lswresample -lpostproc -lm
编译完成后执行以下命令启动测试程序,期望把2018.mp4推给RTMP协议的推流地址rtmp://127.0.0.1:1935/stream。
./pushvideo ../file/2018.mp4 rtmp://127.0.0.1:1935/stream
接着打开另一个MSYS窗口,同样进入该书第十章的源码目录chapter10,执行下面的编译命令。
gcc pullvideo.c -o pullvideo -I/usr/local/ffmpeg/include -L/usr/local/ffmpeg/lib -I/usr/local/sdl2/include -L/usr/local/sdl2/lib -lsdl2 -lavformat -lavdevice -lavfilter -lavcodec -lavutil -lswscale -lswresample -lpostproc -lm
编译完成后执行以下命令启动测试程序,期望从http://127.0.0.1:8888/stream/index.m3u8拉取视频流并弹窗播放。
./pullvideo http://127.0.0.1:8888/stream/index.m3u8
然后果真弹出一个SDL窗口,正在播放从HLS服务拉取的视频画面,说明修改后的推流代码成功支持了RTMP协议。
更多详细的FFmpeg开发知识参见《FFmpeg开发实战:从零基础到短视频上线》一书。