使用线程局部存储解决ffmpeg中多实例调用下自定义日志回调问题

1 问题描述

最近在封装一个库,调用方传入一个URL及对应的回调后就开始执行ffmpeg拉流+硬解码+硬件格式转换,并将得到的数据帧通过回调传递给调用方;除了数据帧回调外,还有日志回调用来传递一些调试信息。

因为该封装库可能被一个进程调用多次(如存在多个小窗口同时播放多个视频),只要在日志回调中包含调用方唯一标识即可进行区分,对于我们自己的日志输出可以这样做,但是发现ffmpeg内部的一些日志如av_dump_format打印却不行,因为没有额外的信息来传递调用方标识,最后发现可以使用线程局部存储来解决这个问题,在此进行记录。

2 问题解决

为了获得av_dump_format的日志输出,我们首先使用av_log_set_callback来对ffmpeg里面的日志进行接管,但是av_dump_format里面打印日志时无法传递任何标识。

这里我们发现av_dump_format是在process函数中调用的,他们一定属于同一个线程,所以可以通过定义一个全局的线程局部变量,并在process函数开始设置为当前调用的上下文,之后就可以在av_dump_format的日志回调中使用这个线程局部变量来获取当前调用的上下文,这样就实现了区分多实例的目的。

其中线程局部变量使用c++11的thread_local来实现,相关代码如下。

cpp 复制代码
static thread_local DecData *thread_local_data = NULL; /*这样可以区分多个实例的使用*/

// 接收ffmpeg内部的日志,当前仅仅显示小于info级别的日志
static void ffmpeg_log_callback(void *ptr, int level, const char *fmt, va_list vl) {
    if (level > AV_LOG_INFO) {
        return;
    }

    int log_size = vsnprintf(NULL, 0, fmt, vl) + 1;
    char * buf = (char*)malloc(log_size);
    vsnprintf(buf, log_size, fmt, vl);
    
    if (thread_local_data) {
        thread_local_data->config.log_cb(buf);
    }
    free(buf);
}

int process(DecData*  data) {
    // ffmpeg
    AVFormatContext *fmt_ctx = NULL;
    int video_stream_index = 0;
    int audio_stream_index = 0;
    AVStream* video_stream = NULL;
    AVStream* audio_stream = NULL;
    AVDictionary *opt = NULL;
    int ret = 0;
    AVPacket *pkt = NULL;
    AVStream* out_video_stream = NULL;
    AVStream* out_audio_stream = NULL;
    AVBSFContext * bst_ctx = NULL;
    AVPacket *bst_pkt = NULL;
    const char *name;

    ConfigInfo *config = &data->config;
    
    // 自定义ffmpeg日志回调函数
    thread_local_data = data;
    av_log_set_callback(ffmpeg_log_callback);

    // 如果是rtsp或rtsps则使用rtsp over tcp
    if (strncmp(config->pull_url, "rtsp://", 7) == 0 || strncmp(config->pull_url, "rtsps://", 8) == 0) {
        av_dict_set(&opt, "rtsp_transport", "tcp", 0);
    }
    // 如果是网络流,则设置i/o超时时间
    av_dict_set(&opt, "timeout", "5000000", 0);

    // 打开输入文件
    ret = avformat_open_input(&fmt_ctx, config->pull_url, NULL, &opt);
    if (ret != 0) {
        logging(data, "avformat_open_input error %d\n", ret);
        goto end;
    }
    av_dict_free(&opt);
    fmt_ctx->opaque = data;

    // 获取输入文件信息
    ret = avformat_find_stream_info(fmt_ctx, NULL);
    if (ret != 0) {
        logging(data, "avformat_find_stream_info error %d\n", ret);
        goto end;
    }

    // 打印输入文件信息
    av_dump_format(fmt_ctx, 0, config->pull_url, 0);
    .........
}
相关推荐
简鹿办公1 天前
FFmpeg vs 去水印软件:哪种方式更适合你?
ffmpeg·怎样去除视频水印·如何去视频logo视频水印
小狮子安度因2 天前
ffplay数据结构分析
数据结构·ffmpeg
小狮子安度因2 天前
ffplay音频重采样
ffmpeg·音视频
小狮子安度因3 天前
AAC ADTS格式分析
网络·ffmpeg·aac
勘察加熊人3 天前
ffmpeg切割音频
ffmpeg·音视频
xiaohouzi1122334 天前
Python读取视频-硬解和软解
python·opencv·ffmpeg·视频编解码·gstreamer
kimble_xia@oracle5 天前
性能优化笔记
ffmpeg
wang_chao1185 天前
RK3399平台ffmpeg-VPU硬编码录制USB摄像头视频、H264或MJPEG编码
ffmpeg·音视频
鹅毛在路上了6 天前
C++, ffmpeg, libavcodec-RTSP拉流,opencv实时预览
c++·opencv·ffmpeg
Hi202402177 天前
Orin-Apollo园区版本:订阅多个摄像头画面拼接与硬编码RTMP推流
ffmpeg·apollo·orin·图像拼接·图传