ffmpeg c++ rtsp拉流 并推流

使用 FFmpeg C++ API 实现 RTSP 拉流并推流需要以下步骤：

1. 初始化 FFmpeg 库和 AVFormatContext。

av_register_all();
avformat_network_init();

AVFormatContext *inputContext = avformat_alloc_context();

2. 打开 RTSP 流并读取媒体信息。

if (avformat_open_input(&inputContext, "rtsp://example.com/stream", nullptr, nullptr) != 0) {
    // 处理打开 RTSP 流失败的情况
}

if (avformat_find_stream_info(inputContext, nullptr) < 0) {
    // 处理读取媒体信息失败的情况
}

3. 查找视频流和音频流，并为它们分配解码器。

int videoStreamIndex = -1;
int audioStreamIndex = -1;

for (int i = 0; i < inputContext->nb_streams; i++) {
    AVStream *stream = inputContext->streams[i];
    AVCodecParameters *codecParameters = stream->codecpar;
    AVCodec *codec = avcodec_find_decoder(codecParameters->codec_id);

    if (!codec) {
        continue;
    }

    if (codecParameters->codec_type == AVMEDIA_TYPE_VIDEO && videoStreamIndex < 0) {
        videoStreamIndex = i;
        AVCodecContext *codecContext = avcodec_alloc_context3(codec);
        avcodec_parameters_to_context(codecContext, codecParameters);
        avcodec_open2(codecContext, codec, nullptr);
        // 处理视频流
    } else if (codecParameters->codec_type == AVMEDIA_TYPE_AUDIO && audioStreamIndex < 0) {
        audioStreamIndex = i;
        AVCodecContext *codecContext = avcodec_alloc_context3(codec);
        avcodec_parameters_to_context(codecContext, codecParameters);
        avcodec_open2(codecContext, codec, nullptr);
        // 处理音频流
    }
}

if (videoStreamIndex < 0 || audioStreamIndex < 0) {
    // 处理找不到视频流或音频流的情况
}

4. 创建输出 AVFormatContext，并为视频流和音频流添加编码器。

AVFormatContext *outputContext = avformat_alloc_context();
avformat_alloc_output_context2(&outputContext, nullptr, "flv", "rtmp://example.com/live");

if (!outputContext) {
    // 处理创建输出 AVFormatContext 失败的情况
}

AVStream *videoStream = avformat_new_stream(outputContext, nullptr);
AVStream *audioStream = avformat_new_stream(outputContext, nullptr);

if (!videoStream || !audioStream) {
    // 处理创建输出流失败的情况
}

AVCodecContext *videoCodecContext = avcodec_alloc_context3(nullptr);
AVCodecContext *audioCodecContext = avcodec_alloc_context3(nullptr);

if (!videoCodecContext || !audioCodecContext) {
    // 处理创建编码器上下文失败的情况
}

videoCodecContext->codec_id = AV_CODEC_ID_H264;
videoCodecContext->codec_type = AVMEDIA_TYPE_VIDEO;
videoCodecContext->pix_fmt = AV_PIX_FMT_YUV420P;
videoCodecContext->width = 1280;
videoCodecContext->height = 720;
videoCodecContext->time_base = {1, 25};

audioCodecContext->codec_id = AV_CODEC_ID_AAC;
audioCodecContext->codec_type = AVMEDIA_TYPE_AUDIO;
audioCodecContext->sample_rate = 44100;
audioCodecContext->channels = 2;
audioCodecContext->channel_layout = AV_CH_LAYOUT_STEREO;
audioCodecContext->time_base = {1, 44100};

if (avcodec_open2(videoCodecContext, avcodec_find_encoder(videoCodecContext->codec_id), nullptr) < 0 ||
    avcodec_open2(audioCodecContext, avcodec_find_encoder(audioCodecContext->codec_id), nullptr) < 0) {
    // 处理打开编码器失败的情况
}

avcodec_parameters_from_context(videoStream->codecpar, videoCodecContext);
avcodec_parameters_from_context(audioStream->codecpar, audioCodecContext);

5. 打开输出流并写入媒体头。

if (!(outputContext->oformat->flags & AVFMT_NOFILE)) {
    if (avio_open(&outputContext->pb, "rtmp://example.com/live", AVIO_FLAG_WRITE) < 0) {
        // 处理打开输出流失败的情况
    }
}

if (avformat_write_header(outputContext, nullptr) < 0) {
    // 处理写入媒体头失败的情况
}

6. 读取 RTSP 流中的帧并写入输出流。

AVPacket packet;
av_init_packet(&packet);

while (av_read_frame(inputContext, &packet) == 0) {
    AVStream *inputStream = inputContext->streams[packet.stream_index];
    AVStream *outputStream = outputContext->streams[packet.stream_index];

    if (packet.stream_index == videoStreamIndex) {
        packet.pts = av_rescale_q(packet.pts, inputStream->time_base, videoStream->time_base);
        packet.dts = av_rescale_q(packet.dts, inputStream->time_base, videoStream->time_base);
        packet.duration = av_rescale_q(packet.duration, inputStream->time_base, videoStream->time_base);
        packet.pos = -1;
        av_interleaved_write_frame(outputContext, &packet);
    } else if (packet.stream_index == audioStreamIndex) {
        packet.pts = av_rescale_q(packet.pts, inputStream->time_base, audioStream->time_base);
        packet.dts = av_rescale_q(packet.dts, inputStream->time_base, audioStream->time_base);
        packet.duration = av_rescale_q(packet.duration, inputStream->time_base, audioStream->time_base);
        packet.pos = -1;
        av_interleaved_write_frame(outputContext, &packet);
    }

    av_packet_unref(&packet);
}

7. 写入媒体尾并释放资源。

av_write_trailer(outputContext);
avcodec_free_context(&videoCodecContext);
avcodec_free_context(&audioCodecContext);
avformat_close_input(&inputContext);
avformat_free_context(inputContext);
avformat_free_context(outputContext);

以上就是使用 FFmpeg C++ API 实现 RTSP 拉流并推流的流程。需要注意的是，这只是一个简单的示例，实际的情况可能会更加复杂。