FFmpeg开发笔记（八十一）FFmpeg代码对RTSP和RTMP的推流区别

《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开发实战：从零基础到短视频上线》一书。