很多朋友都会问到rtsp如何通过rtmp协议推送到服务端,正好前段时间开发了这个功能写在这里,和大家分享下。
首先我想说的是:ffmpeg可以实现这个功能。ffmpeg支持rtsp协议,也支持rtmp。在这个案例中rtsp是输入,
rtmp是输出,ffmpeg实现了转码的功能。下面可出一个整体思路流程图。
图1
如图1所示:在获取都rtsp流以后,解复用(demux)获取ES流packet,最后将ES流封装成rtmp格式并发送
到服务端。
基本思路完毕,下面上代码。
一:初始化ffmpeg库
voidInit()
{
av_register_all();
avfilter_register_all();
avformat_network_init();
av_log_set_level(AV_LOG_ERROR);
}
二. 打开rtmp视频流
int OpenInput(char *fileName)
{
context=avformat_alloc_context();int ret = avformat_open_input(&context, fileName, nullptr,nullptr);if(ret < 0)
{returnret;
}
ret=avformat_find_stream_info(context,nullptr);
auto codecContext= context->streams[0]->codec;
ret= avcodec_open2(codecContext, avcodec_find_decoder(codecContext->codec_id), nullptr);returnret;
}
返回值小于0 表示打开流失败。
三. 创建输出流
int OpenOutput(char *fileName)
{int ret = 0;
ret= avformat_alloc_output_context2(&outputContext, nullptr, "flv", fileName);if(ret < 0)
{gotoError;
}
ret= avio_open2(&outputContext->pb, fileName, AVIO_FLAG_READ_WRITE,nullptr, nullptr);if(ret < 0)
{gotoError;
}for(int i = 0; i < context->nb_streams; i++)
{
AVStream* stream =avformat_new_stream(outputContext, nullptr);ret= avcodec_copy_context(stream->codec, context->streams[i]->codec);if(ret < 0)
{gotoError;
}
}
ret=avformat_write_header(outputContext, nullptr);if(ret < 0)
{gotoError;
}returnret ;
Error:if(outputContext)
{for(int i = 0; i < outputContext->nb_streams; i++)
{
avcodec_close(outputContext->streams[i]->codec);
}
avformat_close_input(&outputContext);
}returnret ;
}
返回值小于0,创建输出流失败,以rtmp格式发送视频流一定要以flv格式初始化output context。
四. 读取Packet
shared_ptrReadPacketFromSource()
{
shared_ptr packet(static_cast(av_malloc(sizeof(AVPacket))), [&](AVPacket *p) { av_free_packet(p); av_freep(&p);});
av_init_packet(packet.get());int ret = av_read_frame(context, packet.get());if(ret >= 0)
{returnpacket;
}else{returnnullptr;
}
}
五. 写Packet到服务端
av_interleaved_write_frame(outputContext, packet.get());
六. 简单demo
int _tmain(int argc, _TCHAR* argv[])
{
string fileInput= "rtsp://admin:admin@172.29.183.103/media/video1/multicast";
string fileOutput="rtmp://127.0.0.1/live/mystream";
Init();
if(OpenInput((char *)fileInput.c_str()) < 0)
{
cout << "Open file Input failed!" << endl;
this_thread::sleep_for(chrono::seconds(1000));
return 0;
}
if(OpenOutput((char *)fileOutput.c_str()) < 0)
{
cout << "Open file Output failed!" << endl;
this_thread::sleep_for(chrono::seconds(1000));
return 0;
}
auto timebase = av_q2d(context->streams[0]->time_base);
while(true)
{
auto packet = ReadPacketFromSource();
if(packet)
{
int ret = av_interleaved_write_frame(outputContext, packet.get());
}
else
{
cout <
break;
}
}
CloseInput();
CloseOutput();
cout <
this_thread::sleep_for(chrono::seconds(1000));
return 0;
}
七. 小结:
经测试,内网rtmp直播延时在1秒以内,公网延时在3,4秒左右。
如需交流,可以加QQ群766718184,或者QQ:350197870
视频教程 播放地址: http://www.iqiyi.com/u/1426749687