FFmpeg RTSP拉流转RTMP推流 C++完整代码

当然，我可以为您提供一个简单的FFmpeg C++代码示例，用于将RTSP流转发到RTMP服务器。以下是一个基本的代码框架：

#include <iostream>
#include <string>
#include <cstdlib>
#include <cstdio>

extern "C" {
#include <libavformat/avformat.h>
#include <libavutil/opt.h>
}

int main() {
    // 初始化FFmpeg库
    av_register_all();
    avformat_network_init();

    // 创建输入上下文
    AVFormatContext* inputContext = nullptr;
    if (avformat_open_input(&inputContext, "rtsp://your_rtsp_url", nullptr, nullptr) != 0) {
        std::cerr << "无法打开RTSP输入流" << std::endl;
        return -1;
    }

    // 查找输入流信息
    if (avformat_find_stream_info(inputContext, nullptr) < 0) {
        std::cerr << "无法获取输入流信息" << std::endl;
        return -1;
    }

    // 创建输出上下文
    AVFormatContext* outputContext = nullptr;
    if (avformat_alloc_output_context2(&outputContext, nullptr, "flv", "rtmp://your_rtmp_url") < 0) {
        std::cerr << "无法创建输出上下文" << std::endl;
        return -1;
    }

    // 遍历输入流，复制到输出上下文
    for (unsigned int i = 0; i < inputContext->nb_streams; i++) {
        AVStream* inStream = inputContext->streams[i];
        AVStream* outStream = avformat_new_stream(outputContext, inStream->codec->codec);
        if (!outStream) {
            std::cerr << "无法创建输出流" << std::endl;
            return -1;
        }

        // 复制流信息
        if (avcodec_copy_context(outStream->codec, inStream->codec) < 0) {
            std::cerr << "无法复制流信息" << std::endl;
            return -1;
        }
    }

    // 打开输出URL
    if (!(outputContext->oformat->flags & AVFMT_NOFILE)) {
        if (avio_open(&outputContext->pb, "rtmp://your_rtmp_url", AVIO_FLAG_WRITE) < 0) {
            std::cerr << "无法打开输出URL" << std::endl;            return -1;
        }
    }

    // 写入输出文件头
    if (avformat_write_header(outputContext, nullptr) < 0) {
        std::cerr << "无法写入输出文件头" << std::endl;
        return -1;
    }

    // 转发流数据
    AVPacket packet;
    while (av_read_frame(inputContext, &packet) >= 0) {
        AVStream* inStream = inputContext->streams[packet.stream_index];
        AVStream* outStream = outputContext->streams[packet.stream_index];

        // 设置时间基
        av_packet_rescale_ts(&packet, inStream->time_base, outStream->time_base);
        packet.pos = -1;

        // 写入输出包
        if (av_interleaved_write_frame(outputContext, &packet) < 0) {
            std::cerr << "无法写入输出包" << std::endl;
            break;
        }

        av_packet_unref(&packet);
    }

    // 写入输出文件尾
    av_write_trailer(outputContext);

    // 清理资源
    avformat_close_input(&inputContext);
    if (outputContext && !(outputContext->oformat->flags & AVFMT_NOFILE))
        avio_closep(&outputContext->pb);
    avformat_free_context(outputContext);

    return 0;
}

请注意，您需要根据实际情况修改以下内容：
- `rtsp://your_rtsp_url`：替换为您的RTSP流URL。
- `rtmp://your_rtmp_url`：替换为您的RTMP服务器URL。

此代码使用FFmpeg库来处理RTSP流，并将其转发到RTMP服务器。它首先打开RTSP输入流，然后创建一个输出上下文，并复制输入流的信息到输出流中。然后，它打开RTMP输出URL，并开始转发流数据。最后，它写入输出文件尾，并清理资源。

希望这个代码示例对您有所帮助！如果您有任何进一步的问题，请随时提问。