使用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 context
static AVFormatContext *input_format_context = NULL;
//输出对应的format context
static 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。

修改前:

修改后: