如何使用 ffmpeg 和 c++ 将自制流发布到 rtmp 服务器？答案

【问题标题】：How to publish selfmade stream with ffmpeg and c++ to rtmp server?如何使用 ffmpeg 和 c++ 将自制流发布到 rtmp 服务器？
【发布时间】：2013-02-15 03:19:50
【问题描述】：

祝大家有美好的一天！

我正在为 Windows 编写一个应用程序，它将捕获屏幕并通过 rtmp（用于广播）将流发送到 Wowza 服务器。我的应用程序使用 ffmpeg 和 Qt。我使用 WinApi 捕获屏幕，将缓冲区转换为 YUV444（因为它最简单）并按照文件 decoder_encoding.c（来自 FFmpeg 示例）中的描述对帧进行编码：

///////////////////////////
//Encoder initialization
///////////////////////////
avcodec_register_all();
codec=avcodec_find_encoder(AV_CODEC_ID_H264);
c = avcodec_alloc_context3(codec);
c->width=scr_width;
c->height=scr_height;
c->bit_rate = 400000;
int base_num=1;
int base_den=1;//for one frame per second
c->time_base= (AVRational){base_num,base_den};
c->gop_size = 10;
c->max_b_frames=1;
c->pix_fmt = AV_PIX_FMT_YUV444P;
av_opt_set(c->priv_data, "preset", "slow", 0);

frame = avcodec_alloc_frame();
frame->format = c->pix_fmt;
frame->width  = c->width;
frame->height = c->height;

for(int counter=0;counter<10;counter++)
{
///////////////////////////
//Capturing Screen
///////////////////////////
    GetCapScr(shotbuf,scr_width,scr_height);//result: shotbuf is filled by screendata from HBITMAP
///////////////////////////
//Convert buffer to YUV444 (standard formula)
//It's handmade function because of problems with prepare buffer to swscale from HBITMAP
///////////////////////////
    RGBtoYUV(shotbuf,frame->linesize,frame->data,scr_width,scr_height);//result in frame->data
///////////////////////////
//Encode Screenshot
///////////////////////////
    av_init_packet(&pkt);
    pkt.data = NULL;    // packet data will be allocated by the encoder
    pkt.size = 0;
    frame->pts = counter;
    avcodec_encode_video2(c, &pkt, frame, &got_output);
    if (got_output) 
    {
        //I think that  sending packet by rtmp  must be here!
        av_free_packet(&pkt);             

    }

}
// Get the delayed frames
for (int got_output = 1,i=0; got_output; i++)
{
    ret = avcodec_encode_video2(c, &pkt, NULL, &got_output);
    if (ret < 0)
        {
            fprintf(stderr, "Error encoding frame\n");
            exit(1);
        }
        if (got_output)
        {
        //I think that  sending packet by rtmp  must be here!
        av_free_packet(&pkt);      
        }
}

///////////////////////////
//Deinitialize encoder
///////////////////////////
avcodec_close(c);
av_free(c);
av_freep(&frame->data[0]);
avcodec_free_frame(&frame);

我需要将此代码生成的视频流发送到 RTMP 服务器。换句话说，我需要这个命令的 c++/c 模拟：

ffmpeg -re -i "sample.h264" -f flv rtmp://sample.url.com/screen/test_stream

它很有用，但我不想将流保存到文件，我想使用 ffmpeg 库对屏幕捕获进行实时编码，并将编码帧发送到我自己的应用程序中的 RTMP 服务器。请给我一个小例子，如何正确初始化 AVFormatContext 并将我的编码视频 AVPackets 发送到服务器。

谢谢。

【问题讨论】：

标签： ffmpeg h.264 rtmp libavcodec libavformat

【解决方案1】：

我的问题可以通过使用 ffmpeg 源代码中的示例来解决。需要文件muxing.c。它位于 ffmpeg 源代码中的文件夹 ffmpeg\docs\examples 中。将示例流写入 rtmp 服务器或文件所需的所有源代码。我必须只了解这些来源并添加我自己的流数据而不是示例流。可能会出现意想不到的问题，但总的来说 - 有一个解决方案。

【讨论】：