很多朋友都會問到rtsp如何通過rtmp協議推送到服務端,正好前段時間開發了這個功能寫在這里,和大家分享下。
首先我想說的是:ffmpeg可以實現這個功能。ffmpeg支持rtsp協議,也支持rtmp。在這個案例中rtsp是輸入,
rtmp是輸出,ffmpeg實現了轉碼的功能。下面可出一個整體思路流程圖。
圖1
如圖1所示:在獲取都rtsp流以后,解復用(demux)獲取ES流packet,最后將ES流封裝成rtmp格式並發送
到服務端。
基本思路完畢,下面上代碼。
一:初始化ffmpeg庫
void Init() { 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) { return ret; } ret = avformat_find_stream_info(context,nullptr); auto codecContext = context->streams[0]->codec; ret = avcodec_open2(codecContext, avcodec_find_decoder(codecContext->codec_id), nullptr); return ret; }
返回值小於0 表示打開流失敗。
三. 創建輸出流
int OpenOutput(char *fileName) { int ret = 0; ret = avformat_alloc_output_context2(&outputContext, nullptr, "flv", fileName); if(ret < 0) { goto Error; } ret = avio_open2(&outputContext->pb, fileName, AVIO_FLAG_READ_WRITE,nullptr, nullptr); if(ret < 0) { goto Error; } 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) { goto Error; } } ret = avformat_write_header(outputContext, nullptr); if(ret < 0) { goto Error; } return ret ; Error: if(outputContext) { for(int i = 0; i < outputContext->nb_streams; i++) { avcodec_close(outputContext->streams[i]->codec); } avformat_close_input(&outputContext); } return ret ; }
返回值小於0,創建輸出流失敗,以rtmp格式發送視頻流一定要以flv格式初始化output context。
四. 讀取Packet
shared_ptr<AVPacket> ReadPacketFromSource() { shared_ptr<AVPacket> packet(static_cast<AVPacket*>(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) { return packet; } else { return nullptr; } }
五. 寫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 <<"write packet end!"<< endl; break; } } CloseInput(); CloseOutput(); cout <<"Transcode file end!" << endl; this_thread::sleep_for(chrono::seconds(1000)); return 0; }
七. 小結:
經測試,內網rtmp直播延時在1秒以內,公網延時在3,4秒左右。
如需交流,可以加QQ群766718184,或者QQ:350197870
視頻教程 播放地址: http://www.iqiyi.com/u/1426749687
視頻下載地址:http://www.chungen90.com/?news_3/
Demo下載地址: http://www.chungen90.com/?news_2