ZLMediaKit中FFmpegSource录制功能线程安全问题分析与修复

问题背景

在ZLMediaKit流媒体服务器项目中，FFmpegSource模块负责通过FFmpeg进程拉取外部媒体流并转换为RTMP流。该模块支持HLS和MP4录制功能，但在实际使用中发现，当同时开启这两种录制功能时，系统会出现崩溃现象。

问题现象

开发人员在使用addFFmpegSource API接口时，通过参数enable_mp4=1启用了MP4录制功能，系统随即出现异常崩溃。通过分析崩溃日志，可以确定问题发生在媒体源录制设置过程中。

技术分析

根本原因

经过深入代码分析，发现问题出在FFmpegSource.cpp文件的onGetMediaSource回调函数中。该函数在获取到媒体源后会立即设置录制功能，但存在以下关键问题：

1. 线程安全问题：setupRecord操作直接在主线程中执行，而媒体源操作需要在特定的事件循环线程中完成
2. 竞态条件：当快速连续设置HLS和MP4录制时，可能引发资源竞争

问题代码

原始实现中，录制设置是直接同步执行的：

if (_enable_hls) {
    src->setupRecord(Recorder::type_hls, true, "", 0);
}
if (_enable_mp4) {
    src->setupRecord(Recorder::type_mp4, true, "", 0);
}

解决方案

修复方法

通过将录制设置操作异步调度到媒体源所属的事件循环线程中执行，解决了线程安全问题。具体修改如下：

if (_enable_hls) {
    src->getOwnerPoller()->async([=]() mutable {
         src->setupRecord(Recorder::type_hls, true, "", 0);
    });
}
if (_enable_mp4) {
    src->getOwnerPoller()->async([=]() mutable {
        src->setupRecord(Recorder::type_mp4, true, "", 0);
    });
}

技术要点

1. 线程模型适配：利用getOwnerPoller()获取媒体源所属的事件循环
2. 异步调度：通过async方法将任务投递到正确的线程执行
3. 资源安全：使用mutable lambda确保闭包内状态安全

影响范围

该修复主要影响以下功能场景：

1. 通过FFmpeg拉流代理时启用录制功能
2. 同时开启HLS和MP4录制的情况
3. 高并发下的流媒体录制场景

最佳实践建议

对于ZLMediaKit的使用者，在配置FFmpegSource录制功能时应注意：

1. 录制操作属于资源密集型操作，应考虑服务器性能
2. 同时开启多种格式录制时，建议监控系统资源使用情况
3. 对于关键业务场景，建议进行充分的压力测试

总结

本次修复解决了ZLMediaKit中FFmpegSource模块在多线程环境下设置录制功能时的稳定性问题，通过正确的线程调度机制确保了媒体源操作的线程安全性。这体现了流媒体服务器开发中线程模型设计的重要性，也为类似的多线程资源管理问题提供了参考解决方案。