FFmpeg rtsp转rtmp推流 c++代码

当然可以！以下是一个简单的示例代码，用于将RTSP流转发到RTMP服务器：

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <unistd.h>
#include <libavformat/avformat.h>
#include <libavutil/opt.h>

int main(int argc, char* argv[]) {
    // 注册所有的FFmpeg组件
    av_register_all();

    // 创建输入上下文
    AVFormatContext* inputContext = avformat_alloc_context();

    // 打开RTSP流
    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 = avformat_alloc_context();

    // 设置输出格式为RTMP
    AVOutputFormat* outputFormat = av_guess_format("flv", nullptr, nullptr);
    outputContext->oformat = outputFormat;

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

    // 遍历输入流
    for (unsigned int i = 0; i < inputContext->nb_streams; i++) {
        AVStream* inputStream = inputContext->streams[i];
        AVStream* outputStream = avformat_new_stream(outputContext, inputStream->codec->codec);

        // 复制流参数
        if (avcodec_copy_context(outputStream->codec, inputStream->codec) < 0) {
            std::cerr << "无法复制流参数" << std::endl;
            return -1;
        }

        outputStream->codec->codec_tag = 0;
        if (outputContext->oformat->flags & AVFMT_GLOBALHEADER) {
            outputStream->codec->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;
        }
    }

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

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

        // 设置时间基
        av_packet_rescale_ts(&packet, inputStream->time_base, outputStream->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);
    avio_close(outputContext->pb);
    avformat_free_context(outputContext);

    return 0;
}

请注意，这只是一个简单的示例代码，用于演示如何使用FFmpeg将RTSP流转发到RTMP服务器。你需要根据自己的需求进行适当的修改和调整。