python rtsp rtmp_rtsp实时流通过rtmp推送到服务端

很多朋友都会问到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