rtsp實時流通過rtmp推送到服務端

     很多朋友都會問到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