使用ffmpeg推送视频流至流媒体服务器(c语言)

最近研究ffmpeg推送, 研究了ffmpeg内部函数调用顺序。写了一个小demo。

/* * 学习ffmpeg使用例子. * 如有不足之处，请指出。在此谢谢各位 * 源码链接：http://ffmpeg.org/doxygen/2.8/index.html * ffmpeg使用版本 3.3.2 */#include <stdio.h>#include <stdlib.h>#include "libavformat/avformat.h"#include "libavutil/mathematics.h"#include "libavutil/time.h"#include "libavcodec/avcodec.h"//输入对应的format contextstatic AVFormatContext *input_format_context = NULL;//输出对应的format contextstatic AVFormatContext *output_format_context = NULL;//输出流static AVStream *input_stream = NULL;//输出流static AVStream *output_stream = NULL;//包static AVPacket pkt;const char *input_file = "1.flv"; //带路径的视频文件(绝对路径或相对路径，本例子使用相对路径，1.flv表示与c文件同一目录下)const char *output_file = "rtmp://localhost:1935/live/hello";//输出 URL（Output URL）[RTMP], 例子使用red5作为流媒体服务器static int video_index = 0;static int frame_index = 0;void init_register_and_network_for_ffmpeg(){    // 第一步av_register_all();    // 注册复用器，编码器等。注册所有的muxers、demuxers和protocols    av_register_all();    //Network,Do global initialization of network components.    //初始化网络容器。    avformat_network_init();}/* * 创建输入流 */AVStream *create_input_avstream(){    int ret; // ret保存函数调用结果    // 第一步av_register_all();    // 注册复用器，编码器等。注册所有的muxers、demuxers和protocols    //av_register_all();    //利用ffmpeg默认方法,初始化context。此步骤可以忽略，input_format_context可以为NULL    //第二步中avformat_open_input()函数中有判断，若为NULL自动调用此函数。    input_format_context = avformat_alloc_context();    printf("-------------%s---------------\n", input_format_context->format_whitelist);        //第二步,avformat_open_input();    //打开多媒体数据并且获得一些相关的信息    if ((ret = avformat_open_input(&input_format_context, input_file, 0, 0)) < 0)    {        printf("Can't open input file.");        return NULL;    }    //第三步,avformat_find_stream_info();    //读取一部分视音频数据并且获得一些相关的信息    if ((ret = avformat_find_stream_info(input_format_context, 0)) < 0) {        printf( "Failed to retrieve input stream information");        return NULL;    }    //初始化完成后，AVFormatContext中存放有AVStream的引用    //通过AVStream中存放的AVCodecContext引用获取 （enum AVMediaType）枚举中视频对应的值。    //   ***enum AVMediaType    //AVMEDIA_TYPE_UNKNOWN,AVMEDIA_TYPE_VIDEO,AVMEDIA_TYPE_AUDIO,AVMEDIA_TYPE_DATA,AVMEDIA_TYPE_SUBTITLE,AVMEDIA_TYPE_ATTACHMENT,AVMEDIA_TYPE_NB    //例：首先通过循环遍历获取视频    for (int i = 0; i < input_format_context->nb_streams; i++) //nb_streams为stream数量    {        //如果stream的codec为AVCodecContext变量，AVCodecContext变量中的codec为AVMediaType枚举值        if (input_format_context->streams[i]->codec->codec == AVMEDIA_TYPE_VIDEO)        {            input_stream = input_format_context->streams[i];            video_index = i;            break;        }    }    //Print detailed information about the input or output format, such as duration, bitrate, streams, container, programs, metadata, side data, codec and time base    //打印输入输出format的详细信息，该函数可选调用。    av_dump_format(input_format_context, 0, input_file, 0);    return input_stream;}/* * 创建输出流 */AVStream *create_output_avstream(){    int ret;    // 第一步av_register_all();    // 注册复用器，编码器等。注册所有的muxers、demuxers和protocols    //av_register_all();    // 第二步avformat_alloc_output_context2();    //基于FFmpeg的视音频编码器程序中，该函数通常是第一个调用的函数（除了组件注册函数av_register_all()）    // return >= 0 in case of success, 返回值大于等于0表示成功    avformat_alloc_output_context2(&output_format_context, NULL, "flv", output_file);    if (!output_format_context) {        printf( "Could not create output context\n");        ret = AVERROR_UNKNOWN;        return NULL;    }    //根据输入流创建输出流（Create output AVStream according to input AVStream）    AVStream *output_stream = avformat_new_stream(output_format_context, input_stream->codec->codec);    if (!output_stream) {        printf( "Failed allocating output stream\n");        ret = AVERROR_UNKNOWN;        return NULL;    }    //复制AVCodecContext的设置（Copy the settings of AVCodecContext）    ret = avcodec_copy_context(output_stream->codec, input_stream->codec);    if (ret < 0)    {        printf( "Failed to copy context from input to output stream codec context\n");        return NULL;    }    output_stream->codec->codec_tag = 0;    if (output_format_context->oformat->flags & AVFMT_GLOBALHEADER)        output_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;    //Print detailed information about the input or output format, such as duration, bitrate, streams, container, programs, metadata, side data, codec and time base    //打印输入输出format的详细信息，该函数可选调用。    av_dump_format(output_format_context, 0, output_file, 1);    return output_stream;}int open_url_and_write(){    int ret;    int64_t start_time = 0; // 开始时间    //打开输出URL（Open output URL）    if (!(output_format_context->oformat->flags & AVFMT_NOFILE)) {        ret = avio_open2(&output_format_context->pb, output_file, AVIO_FLAG_WRITE, NULL, NULL); //根据源代码，调用avio_open同效,avio_open中调用avio_open2。        //ret = avio_open(&output_format_context->pb, output_file, AVIO_FLAG_WRITE);        if (ret < 0) {            printf( "Could not open output URL '%s'", output_file);            return -1;        }    }    //写文件头（Write file header    //0 on success, negative AVERROR on failure.    ret = avformat_write_header(output_format_context, NULL);    if (ret < 0) {        printf( "Error occurred when opening output URL\n");        return -1;    }    start_time = av_gettime();    while (1) {        AVStream *in_stream, *out_stream;        //获取一个AVPacket（Get an AVPacket）        ret = av_read_frame(input_format_context, &pkt);        if (ret < 0)            break;        //FIX：No PTS (Example: Raw H.264)        //Simple Write PTS        if(pkt.pts == AV_NOPTS_VALUE)        {            //Write PTS            //AVRational time_base1=input_format_context->streams[video_index]->time_base;            AVRational time_base1 = input_format_context->streams[video_index]->time_base;            //Duration between 2 frames (us)            int64_t calc_duration = (double)AV_TIME_BASE/av_q2d(input_format_context->streams[video_index]->r_frame_rate);            //Parameters            pkt.pts = (double)(frame_index*calc_duration) / (double)(av_q2d(time_base1)*AV_TIME_BASE);            pkt.dts = pkt.pts;            pkt.duration = (double)calc_duration / (double)(av_q2d(time_base1)*AV_TIME_BASE);        }        //Important:Delay        if(pkt.stream_index == video_index)        {            AVRational time_base = input_format_context->streams[video_index]->time_base;            AVRational time_base_q = {1, AV_TIME_BASE};            int64_t pts_time = av_rescale_q(pkt.dts, time_base, time_base_q);            int64_t now_time = av_gettime() - start_time;            if (pts_time > now_time)                av_usleep(pts_time - now_time);        }        in_stream  = input_format_context->streams[pkt.stream_index];        out_stream = output_format_context->streams[pkt.stream_index];        /* copy packet */        //转换PTS/DTS（Convert PTS/DTS）        pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));        pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));        pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);        pkt.pos = -1;        //Print to Screen        if(pkt.stream_index == video_index){            printf("Send %8d video frames to output URL\n",frame_index);            frame_index++;        }        //ret = av_write_frame(ofmt_ctx, &pkt);        ret = av_interleaved_write_frame(output_format_context, &pkt);        if (ret < 0) {            printf( "Error muxing packet\n");            break;        }        av_free_packet(&pkt);    }    //写文件尾（Write file trailer）    av_write_trailer(output_format_context);    return 0;}int close_input_and_output(){    int ret;    // close input    avformat_close_input(&input_format_context);    /* close output */    if (output_format_context && !(output_format_context->oformat->flags & AVFMT_NOFILE))        avio_close(output_format_context->pb);    avformat_free_context(output_format_context);    if (ret < 0 && ret != AVERROR_EOF) {        printf( "Error occurred.\n");        return -1;    }    return 0;}int main(int argc, char *argv[]){    init_register_and_network_for_ffmpeg();    input_stream = create_input_avstream(); //创建输入流、初始化输入context    if (NULL == input_stream)    {        close_input_and_output(); //创建失败,调用close关闭输入输出。        return -1;    }    output_stream = create_output_avstream();//创建输出流、初始化输出context    if (NULL == output_stream)    {        close_input_and_output(); //创建失败,调用close关闭输入输出。        return -1;    }    if(0 != open_url_and_write())    {        close_input_and_output(); //调用close关闭输入输出。        return -1;    }    //close    close_input_and_output();}

本人使用的所有文件目录如下：

通过命令编译:

gcc -I/Users/marco/Desktop/libs/include -L/Users/marco/Desktop/libs/lib -lavcodec -lavdevice -lavfilter -lavformat -lavresample -lavutil -lpostproc -lswresample -lswscale pusher.c

如果编译中出现以下情况：

请找到libavutil/mathematics.h下的AVRounding。将其前面加上typedef。

修改前:

修改后: