ffmpeg 把RTSP转成RTMP

最新推荐文章于 2023-09-22 10:46:55 发布

llljw

最新推荐文章于 2023-09-22 10:46:55 发布

阅读量2.5k

点赞数

分类专栏： ffmpeg

ffmpeg 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种将RTSP流转换为RTMP流的方法。通过使用FFmpeg库，该过程涉及打开并读取RTSP流，然后将这些数据重新封装为RTMP流进行推送。文章提供了详细的代码实现，包括错误处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

可以通过SRS搭建流媒体服务器，并在其提供的SRS播放器播放，或者通过VLC播放器播放

#include <iostream>
using namespace std;

extern "C"
{
#include "libavformat/avformat.h"
#include "libavutil/time.h"
#include "libavutil/opt.h"
}
#pragma comment(lib, "avformat.lib")
#pragma comment(lib, "avutil.lib")
#pragma comment(lib, "avcodec.lib")

//AVDictionary *options = NULL;
char av_error[AV_ERROR_MAX_STRING_SIZE] = { 0 };
#define av_err2str(errnum) \
av_make_error_string(av_error, AV_ERROR_MAX_STRING_SIZE, errnum)

#define INFILE  "t.flv"
#define RTMP    "rtmp://..."
#define RTSP    "rtsp://..."
int rtsp2rtmp(){
    int ret = 0;
    //封装上下文
    AVFormatContext* ictx = nullptr;
    AVFormatContext* octx = nullptr;
    const char* iurl = RTSP;
    const char* ourl = RTMP;
    int64_t starttime;
    av_register_all();
    ret = avformat_network_init();
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    
    
    //打开文件，解封文件头
    ret = avformat_open_input(&ictx, iurl, nullptr, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cerr << "open file " << iurl << " success." << endl;
    
    //获取音视频流信息,h264 flv
    ret = avformat_find_stream_info(ictx, nullptr);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    
    //打印媒体信息
    av_dump_format(ictx, 0, iurl, 0);
    
    //////////////////////////////
    
    //输出流
    ret = avformat_alloc_output_context2(&octx, av_guess_format(nullptr, INFILE, nullptr), nullptr, ourl);
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    cerr << "octx create success." << endl;
    
    //配置输出流
    for (int i = 0; i < ictx->nb_streams; ++i)
    {
        //创建流
        AVStream* ostream = avformat_new_stream(octx, avcodec_find_encoder(ictx->streams[i]->codecpar->codec_id));
        if (ostream == nullptr)
            return -1;
        //复制配置信息
        ret = avcodec_parameters_copy(ostream->codecpar, ictx->streams[i]->codecpar);
        if (ret != 0)
        {
            cout << av_err2str(ret) << endl;
            goto END;
        }
        ostream->codecpar->codec_tag = 0;//标记不需要重新编解码
    }
    
    av_dump_format(octx, 0, ourl, 1);
    
    //////////////////////////////
    
    //推流
    if (ret != 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    
    ret = avio_open(&octx->pb, ourl, AVIO_FLAG_WRITE);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    
    AVCodecContext* c;
    
    
    //写入头信息
    ret = avformat_write_header(octx, nullptr);
    if (ret < 0)
    {
        cout << av_err2str(ret) << endl;
        goto END;
    }
    
    //推流每一帧数据
    AVPacket pkt;
    starttime = av_gettime();
    while (av_read_frame(ictx, &pkt) == 0)
    {
        //计算转换pts dts
        AVRational itime = ictx->streams[pkt.stream_index]->time_base;
        AVRational otime = octx->streams[pkt.stream_index]->time_base;
        pkt.pts = av_rescale_q_rnd(pkt.pts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.dts = av_rescale_q_rnd(pkt.dts, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.duration = av_rescale_q_rnd(pkt.duration, itime, otime, (AVRounding)(AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX));
        pkt.pos = -1;
        
        ret = av_interleaved_write_frame(octx, &pkt);
        av_packet_unref(&pkt);
        if (ret < 0)
            cout << av_err2str(ret) << endl;//不用退出
    }
    
END:
    if (ictx != nullptr)
        avformat_close_input(&ictx);
    if (octx != nullptr)
    {
        avio_close(octx->pb);
        avformat_free_context(octx);
    }
    ret = avformat_network_deinit();
    if (ret != 0)
        cout << av_err2str(ret) << endl;
    return 0;
}

int main()
{
//    httRtmp();
    rtsp2rtmp();
    system("pause");
    return 0;
}