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

Question

祝你们有美好的一天， friend 们!

我正在为 Windows 编写一个应用程序，它将捕获屏幕并通过 rtmp(用于广播)将流发送到 Wowza 服务器。我的应用程序使用 ffmpeg 和 Qt。我使用 WinApi 捕获屏幕，将缓冲区转换为 YUV444(因为它最简单)并按照文件 decoding_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 发送到服务器的小例子。

谢谢。

Answer 1

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