第5课 使用FFmpeg将rtmp流再转推到rtmp服务器

本课对应源文件下载链接:

https://download.csdn.net/download/XiBuQiuChong/88801992

通过前面的学习,我们已经可以正常播放网络rtmp流及本地mp4文件。这节课,我们将在前面的基础上实现一个常用的转推功能:读取rtmp流或mp4文件并转推到rtmp服务器上实现直播转发功能。

一、最终实现的效果

上面左图为播放的rtmp流并将该流转推到另一台rtmp服务器,右图为用vlc播放的另一台rtmp服务器上被转发的流。

二、使用FFmpeg将rtmp流转推到另一台rtmp服务器​的原理

使用FFmpeg API 转码推流的一般过程如下:

1.引入ffmpeg库:在代码中引入ffmpeg库,以便使用其提供的功能。

2.打开输入文件:使用avformat_open_input()函数打开输入文件,并获取输入文件的相关信息。

3.查找流信息:使用avformat_find_stream_info()函数查找输入文件中的流信息,并将其存储在AVFormatContext结构体中。

4.创建输出上下文:使用avformat_alloc_output_context2()函数创建输出上下文,并设置输出格式。

5.添加输出流:根据输入文件的流信息,使用avformat_new_stream()函数创建输出流,并将其添加到输出上下文中。

6.设置编码参数:为输出流设置编码参数,包括编码器、编码器参数等。

7.输出文件:使用avio_open()函数打开输出文件,并将输出文件的相关信息存储在输出上下文中。

8.写入文件头:使用avformat_write_header()函数写入输出文件的文件头。

9.转码推流:循环读取输入文件的数据包,使用av_read_frame()函数读取数据包,然后使用avcodec_send_frame()函数发送数据包给编码器进行编码,再使用avcodec_receive_packet()函数接收编码后的数据包,最后使用av_interleaved_write_frame()函数将编码后的数据包写入输出流。

10.写入文件尾:使用av_write_trailer()函数写入输出文件的文件尾。

11.释放资源:释放所有的上下文、流和其他资源,使用avformat_close_input()函数关闭输入文件。

三、转推功能的具体实现

与rtmp流或mp4文件播放功能相比,转推功能只是在原来的基础上又增加了编码功能和向rtmp服务器的推送功能。为了降低难度,这节课的转推功能不涉及编码部分,只实现将拉取来的rtmp流直接进行转推。

1.为了与上节课的播放功能区分,新建或直接复制fmlp(Flash Media Live Player)类为一个新的fmlt(Flash Media Live Transcoder)类,并修改主对话框相应代码使fmlt能正常工作并实现正常的播放功能。

修改如下:

cpp 复制代码
//#include "fmlp.h"
#include "fmlt.h"
..................
//fmlp *myFmlp = new fmlp();
fmlt *myFmlt = new fmlt();

2.与mp4文件相比,rtmp流转推实现起来相对容易,因此,我们先在原来的基础上实现rtmp流的转推功能。

(1)直接拷贝原fmlp.cpp的代码到fmt.cpp中,并定义转推到另一台rtmp服务器的流地址等变量:

cpp 复制代码
//转码输出初始化	
	const char *outFileName = "rtmp://192.168.0.101/hk/cctv";		
	AVOutputFormat *outFormat = NULL;
	AVFormatContext *outFormatCtx = NULL;
	//定义输出音视频流
	AVStream *videoStream = NULL;
	AVStream *audioStream = NULL;

(2)打开输出流:

cpp 复制代码
avformat_alloc_output_context2(&outFormatCtx, NULL, "flv", outFileName);

..................

         videoStream = avformat_new_stream(outFormatCtx, vDecodec);

         videoStream->id = outFormatCtx->nb_streams - 1;

         videoStream->codecpar->codec_tag = 0;

         avcodec_parameters_from_context(videoStream->codecpar, vDecodeCtx);

         audioStream = avformat_new_stream(outFormatCtx, aDecodec);

         audioStream->codecpar->codec_tag = 0;

         audioStream->id = outFormatCtx->nb_streams - 1;

         avcodec_parameters_from_context(audioStream->codecpar, aDecodeCtx);

         av_dump_format(outFormatCtx, 0, outFileName, 1);

         ret = avio_open2(&outFormatCtx->pb, outFileName, AVIO_FLAG_READ_WRITE, nullptr, nullptr);

         if (ret < 0){

                  return -1;

         }

         ret = avformat_write_header(outFormatCtx, NULL);

         if (ret < 0){

                  TRACE("ret:%d\n", ret);

                  return -1;

         }

(3)获取到原rtmp流并正常播放音视频后,将音视频包直接写入输出流:

cpp 复制代码
if (normalPkt.stream_index == videoIndex) {

                          //原视频播放代码

                          normalPkt.stream_index = videoStream->index;

                          normalPkt.pts = currentTime;

                          normalPkt.dts = currentTime;

                          normalPkt.duration = currentTime;

                          normalPkt.pos = -1;

                          //enVideoPacket.flags = AV_PKT_FLAG_KEY;

                          if (normalPkt.size > 0){

                                   ret = av_interleaved_write_frame(outFormatCtx, &normalPkt);

                          }

                          av_packet_unref(&normalPkt);

                  }

                  else if (normalPkt.stream_index == audioIndex)

                  {                       

                          //原音频播放代码

                          normalPkt.stream_index = audioStream->index;

                          normalPkt.pts = currentTime;

                          normalPkt.dts = currentTime;

                          normalPkt.duration = currentTime;

                          normalPkt.pos = -1;

                          //enVideoPacket.flags = AV_PKT_FLAG_KEY;

                          if (normalPkt.size > 0){

                                   ret = av_interleaved_write_frame(outFormatCtx, &normalPkt);

                          }

                          av_packet_unref(&normalPkt);

          }

这样,就实现了在播放音视频的同时又能将该流转推到另一台服务器的功能。

相关推荐
闻缺陷则喜何志丹11 分钟前
【计算几何 二分查找】P5485 [JLOI2010] 铁人双项比赛|普及+
c++·数学·二分查找·计算几何·洛谷
..空空的人15 分钟前
C++基于protobuf实现仿RabbitMQ消息队列---服务器模块认识1
服务器·开发语言·c++·分布式·rabbitmq·protobuf
晨非辰24 分钟前
基于Win32 API控制台的贪吃蛇游戏:从设计到C语言实现详解
c语言·c++·人工智能·后端·python·深度学习·游戏
小此方29 分钟前
Re: ゼロから学ぶ C++ 入門(六)类和对象·第三篇:运算符重载
开发语言·c++·后端
2301_7890156229 分钟前
每日精讲:环形链表、两个数组中的交集、随机链表的复制
c语言·数据结构·c++·算法·leetcode·链表·排序算法
2301_789015622 小时前
C++:二叉搜索树
c语言·开发语言·数据结构·c++·算法·排序算法
leiming69 小时前
C++ vector容器
开发语言·c++·算法
apocelipes11 小时前
从源码角度解析C++20新特性如何简化线程超时取消
c++·性能优化·golang·并发·c++20·linux编程
ozyzo11 小时前
求1~n的累加和
c++
charlie11451419112 小时前
现代C++嵌入式教程:C++98基础特性:从C到C++的演进(1)
c语言·开发语言·c++·笔记·学习·教程