最简单的基于FFmpeg的推流器以推送RTMP为例

/**
 * 最简单的基于FFmpeg的推流器（推送RTMP）
 * Simplest FFmpeg Streamer （Send RTMP）
 * 
 * 雷霄骅 Lei Xiaohua
 * leixiaohua1020@126.com
 * 中国传媒大学/数字电视技术
 * Communication University of China / Digital TV Technology
 * http://blog.csdn.net/leixiaohua1020
 * 
 * 本例子实现了推送本地视频至流媒体服务器（以RTMP为例）。
 * 是使用FFmpeg进行流媒体推送最简单的教程。
 *
 * This example stream local media files to streaming media 
 * server （Use RTMP as example）. 
 * It‘s the simplest FFmpeg streamer.
 * 
 */

#include 

extern “C“
{
#include “libavformat/avformat.h“
#include “libavutil/mathematics.h“
#include “libavutil/time.h“
};


int main（int argc char* argv[]）
{
	AVOutputFormat *ofmt = NULL;
	//输入对应一个AVFormatContext，输出对应一个AVFormatContext
	//（Input AVFormatContext and Output AVFormatContext）
	AVFormatContext *ifmt_ctx = NULL *ofmt_ctx = NULL;
	AVPacket pkt;
	const char *in_filename *out_filename;
	int ret i;
	
	//in_filename  = “cuc_ieschool.mov“;
	//in_filename  = “cuc_ieschool.mkv“;
	//in_filename  = “cuc_ieschool.ts“;
	//in_filename  = “cuc_ieschool.mp4“;
	//in_filename  = “cuc_ieschool.h264“;
	in_filename  = “cuc_ieschool.flv“;//输入URL（Input file URL）
	out_filename = “rtmp://localhost/publishlive/livestream“;//输出 URL（Output URL）[RTMP]
	//out_filename = “rtp://233.233.233.233:6666“;//输出 URL（Output URL）[UDP]

	av_register_all（）;
	//Network
	avformat_network_init（）;
	//输入（Input）
	if （（ret = avformat_open_input（&ifmt_ctx in_filename 0 0）） < 0） {
		printf（ “Could not open input file.“）;
		goto end;
	}
	if （（ret = avformat_find_stream_info（ifmt_ctx 0）） < 0） {
		printf（ “Failed to retrieve input stream information“）;
		goto end;
	}

	int videoindex=-1;
	for（i=0; inb_streams; i++） 
		if（ifmt_ctx->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO）{
			videoindex=i;
			break;
		}

	av_dump_format（ifmt_ctx 0 in_filename 0）;

	//输出（Output）
	
	avformat_alloc_output_context2（&ofmt_ctx NULL “flv“ out_filename）; //RTMP
	//avformat_alloc_output_context2（&ofmt_ctx NULL “mpegts“ out_filename）;//UDP

	if （!ofmt_ctx） {
		printf（ “Could not create output context\n“）;
		ret = AVERROR_UNKNOWN;
		goto end;
	}
	ofmt = ofmt_ctx->oformat;
	for （i = 0; i < ifmt_ctx->nb_streams; i++） {
		//根据输入流创建输出流（Create output AVStream according to input AVStream）
		AVStream *in_stream = ifmt_ctx->streams[i];
		AVStream *out_stream = avformat_new_stream（ofmt_ctx in_stream->codec->codec）;
		if （!out_stream） {
			printf（ “Failed allocating output stream\n“）;
			ret = AVERROR_UNKNOWN;
			goto end;
		}
		//复制AVCodecContext的设置（Copy the settings of AVCodecContext）
		ret = avcodec_copy_context（out_stream->codec in_stream->codec）;
		if （ret < 0） {
			printf（ “Failed to copy context from input to output stream codec context\n“）;
			goto end;
		}
		out_stream->codec->codec_tag = 0;
		if （ofmt_ctx->oformat->flags & AVFMT_GLOBALHEADER）
			out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;
	}
	//Dump Format----------------