既能够用ffmpeg命令做RTSP流转RTMP流，又可以像调用avcodec/avfilter库一样逻辑编程

xiejiashu2023-09-05 8:30

又有一个需求：我们现在想做一款多路RTSP拉流转RTMP推流到CDN进行直播的功能，注意啊，是多路，原来我们有两种方式，一种是用ffmpeg.exe进行：

ffmpeg -i "rtsp://192.168.0.99:8554/1" -c:v libx264 -c:a aac -f flv "rtmp://127.0.0.1:1935/live/test"

同样，这种方式会有一个问题，那就是断线重连的问题，我们不能很好地知道拉流是不是断线了，或者当前的推流状态是什么样子的，推流中还是重连中，同时，我们也只能默认重编码成H.264和AAC，这种机器性能消耗会很高，我们完全可以在源流是H.264或者AAC编码的时候，直接-c:v copy或者-c:a copy，但是我们不好判断。

每一路流我们先ffprobe一下，看看视频编码格式是H.264还是H.265，再重新启动ffmpeg.exe拉流转推流？

很显然，上面的方法搞个几路是没问题的，搞多了就不靠谱了，维护一堆ffmpeg既不稳定又没成长！~

另一种方式，就是用ffmpeg的SDK进行开发，avfilter、avcodec一整套，获取avformat，再把avpacket导给RTMP推流，整套下来，开发难度较高，要工程化的比较稳定，还需要老师傅！看看大模型咋写的？

cpp 复制代码 
#include <libavformat/avformat.h>
#include <libavutil/time.h>

int main(int argc, char **argv) {
    AVFormatContext *pFormatCtx = NULL;
    AVOutputFormat *oFormat = NULL;
    AVStream *in_stream = NULL;
    AVStream *out_stream = NULL;
    AVCodecContext *pCodecCtxIn = NULL;
    AVCodecContext *pCodecCtxOut = NULL;
    AVPacket pkt;
    int ret, i;

    if (argc < 3) {
        printf("Usage: %s <input RTSP URL> <output RTMP URL>
", argv[0]);
        return -1;
    }

    // Register all formats and codecs
    av_register_all();

    // Open input RTSP URL
    if ((ret = avformat_open_input(&pFormatCtx, argv[1], NULL, NULL)) != 0) {
        printf("Cannot open input stream: %s
", av_err2str(ret));
        return -1;
    }

    // Retrieve stream information
    if ((ret = avformat_find_stream_info(pFormatCtx, NULL)) < 0) {
        printf("Cannot find stream information: %s
", av_err2str(ret));
        return -1;
    }

    // Find the first audio stream
    for (i = 0; i < pFormatCtx->nb_streams; i++) {
        if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
            in_stream = pFormatCtx->streams[i];
            break;
        }
    }

    // If no audio stream is found, exit
    if (!in_stream) {
        printf("Didn't find a audio stream!
");
        return -1;
    }

    // Get a pointer to the codec context for the audio stream
    pCodecCtxIn = in_stream->codec;

    // Find the first video stream
    for (i = 0; i < pFormatCtx->nb_streams; i++) {
        if (pFormatCtx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
            out_stream = pFormatCtx->streams[i];
            break;
        }
    }

    // If no video stream is found, exit
    if (!out_stream) {
        printf("Didn't find a video stream!
");
        return -1;
    }

    // Get a pointer to the codec context for the video stream
    pCodecCtxOut = out_stream->codec;

    // Open output RTMP URL
    if ((ret = avformat_alloc_output_context2(&pFormatCtx, NULL, "flv", argv[2])) < 0) {
        printf("Cannot create output context: %s
", av_err2str(ret));
        return -1;
    }

    // Add stream to the output context
    if (avformat_new_stream(pFormatCtx, out_stream->codec->codec) < 0) {
        printf("Error adding new stream to output context
");
        return -1;
    }

    // Write header
    if (avformat_write_header(pFormatCtx, NULL) < 0) {
        printf("Error occurred when opening output file.
");
        return -1;
    }

    // Process each packet from input stream
    while (av_read_frame(pFormatCtx, &pkt) >= 0) {
        // Is this a packet from the audio stream?
        if (pkt.stream_index == in_stream->index) {
            // Decode audio frame
            AVCodecContext *pCodecOut = pCodecCtxOut;
            ret = avcodec_send_packet(pCodecOut, &pkt);
            if (ret < 0) {
                printf("Error sending a packet for decoding
");
                return -1;
            }
            while (ret >= 0) {
                ret = avcodec_receive_frame(pCodecOut, NULL);
                if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
                    break;
                } else if (ret < 0) {
                    printf("Error during decoding
");
                    return -1;
                }
                // Write the decoded frame to the output file
                av_interleaved_write_frame(pFormatCtx, &pkt);
            }
        } else if (pkt.stream_index == out_stream->index) {
            // Is this a packet for the video stream?
            if (pkt.pts == in_stream->start_time) {
                // Write video frame
                ret = av_interleaved_write_frame(pFormatCtx, &pkt);
                if (ret < 0) {
                    printf("Error during writing
");
                    return -1;
                }
            }
        }
    }

    // Close the output context
    avformat_free_context(pFormatCtx);

    return 0;
}

注意啊，以上代码没有经过验证，太麻烦了，罗里吧嗦一大堆；

那么，有没有一种办法，能简化ffmpeg的SDK调用过程呢，既能解决重连的问题，又能解决SDK纷繁复杂的调用过程问题，而且还有整个过程中对音视频编码判断和重编码跳过的流程？

有，看看EasyAVFilter怎么完成这项工作：

cpp 复制代码 
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "EasyAVFilterAPI.h"

#ifdef _WIN32
#pragma comment(lib,"EasyAVFilter.lib")
#endif
int Easy_APICALL __AVFilterCallBack(void* userPtr, EASY_AV_FILTER_STATE_T status, int progress, int errCode, const char *errMsg)
{
	return 0;
}

int main(int argc, char** argv)
{
	//创建ffmpeg实例
	Easy_Handle avFilterHandle = NULL;
	EasyAVFilter_Create(&avFilterHandle);
	//设置回调函数，获取回调信息
	EasyAVFilter_SetCallback(avFilterHandle,__AVFilterCallBack,0);
	//将RTSP流转成RTMP流
	EasyAVFilter_AddInput(avFilterHandle, "rtsp://admin:admin@112.112.112.212:554/ch1/main/av_stream", 1);
	EasyAVFilter_AddFilter(avFilterHandle, "-vcodec copy -acodec aac -ac 2 -strict -2");
	EasyAVFilter_AddFilter(avFilterHandle, "-f flv");
	EasyAVFilter_SetOutput(avFilterHandle, "rtmp://172.81.216.155:13519/live/IbMkUXeVR?sign=SxMk8X6VRz", 0);
	//验证参数设置是否正确
	char filterCommand[256] = { 0 };
	EasyAVFilter_GetFilters(avFilterHandle, filterCommand);
	printf("command: %s\n", filterCommand);
	//开始RTSP转RTMP工作
	EasyAVFilter_Start(avFilterHandle, 0, 8, 10);//注意，文件转码不需要循环读取，第二个参数从1改成0
	getchar();
	EasyAVFilter_Stop(avFilterHandle);
	EasyAVFilter_Release(&avFilterHandle);
	return 0;
}

就上面八九个方法，还包括了创建实例和停止/销毁实例，核心方法就五六个，就搞定了全部ffmpeg.exe所有的功能，还能支持重连！！！

方法名称 说明
EasyAVFilter_Create 创建句柄，相当于创建了一个ffmpeg.exe
EasyAVFilter_Release 释放句柄
EasyAVFilter_SetCallback 设置回调函数和自定义指针，回调过程中的各种媒体信息/连接信息/转码进度
EasyAVFilter_AddInput 添加输入参数(源地址)
EasyAVFilter_AddFilter 添加中间参数,如:转码，兼容ffmpeg命令所有参数（例如-vcodec copy -acodec aac）
EasyAVFilter_SetOutput 设置输出参数(目标地址) ，支持默认转码H.264和自动根据源编码进行转码
EasyAVFilter_GetFilters 获取所有参数（review参数输入是否正确）
EasyAVFilter_Start 开始工作，支持0次重连和N次重连
EasyAVFilter_Stop 停止工作

详细信息可以直接看https://www.easydarwin.org/tools/153.html，具体用法和场景，看视频介绍；

相关推荐
岁月小龙3 小时前
如何让ffmpeg运行时从当前目录加载库，而不是从/lib64
ffmpeg·origin·ffprobe·rpath
行者记1 天前
ffmpeg命令——从wireshark包中的rtp包中分离h264
测试工具·ffmpeg·wireshark
EasyCVR1 天前
国标GB28181视频平台EasyCVR私有化视频平台工地防盗视频监控系统方案
运维·科技·ffmpeg·音视频·1024程序员节·监控视频接入
hypoqqq1 天前
使用ffmpeg播放rtsp视频流
ffmpeg
cuijiecheng20181 天前
音视频入门基础：FLV专题（24）——FFmpeg源码中，获取FLV文件视频信息的实现
ffmpeg·音视频
QMCY_jason1 天前
黑豹X2 armbian 编译rkmpp ffmpeg 实现CPU视频转码
ffmpeg
苍天饶过谁？1 天前
SDL基本使用
ffmpeg
HZ355721 天前
ffmpeg视频解码
ffmpeg·音视频
runing_an_min2 天前
windows运行ffmpeg的脚本报错：av_ts2str、av_ts2timestr、av_err2str =＞ E0029 C4576
c++·windows·ffmpeg·e0029
EelBarb2 天前
ffmpeg：视频字幕嵌入（GPU加速）
ffmpeg·音视频
热门推荐
01【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总02(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明03组基轨迹建模 GBTM的介绍与实现（Stata 或 R）04【AIGC】重塑未来的科技巨轮05全面解析：构建基于深度学习的安全帽检测系统（UI界面+YOLO代码+数据集）06【经验分享】Ubuntu22.04安装微信(linux官方版)07基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测08Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO09RAG 实践- Ollama+RagFlow 部署本地知识库10红米手机使用google play