修改FFMpeg源码—捕获丢包-优快云博客

概述

最近我们项目有一个需求就是解决客户端播放RTSP视频流花屏的问题，一般来说丢包就会引起花屏，导致客户端花屏的因素又有很多，比如说：

相机到服务器丢包
服务器到客户端丢包
等等。。。

其中服务器到客户端的丢包问题我们已经解决了，那么相机到服务器的丢包问题怎么解决呢？这个问题解决不了的，可以解决的问题就是即使相机到服务器丢包后，也让客户端知道，然后不解码丢包的那一帧数据直到下一个关键帧的到来，这样客户端播放视频就不会
花屏了，但是这样做就会让视频播放卡顿一下（以50帧一个关键帧来算的话会卡顿2秒），但是卡顿总比花屏强吧，因为花屏后我们的AI服务器就采集不到人脸了，但是FFMpeg并没有对外提供接口标志该AVPacket不完整，内部也不会将AVPacket丢弃，这样服务端调用av_read_frame读取的AVPacket时并不知道该包是否完整，一小段调用例子如下：


    while (1) {
        AVPacket pkt;
        // 不知道pkt是否完整
        ret = av_read_frame(f->ctx, &pkt);

        if (ret == AVERROR(EAGAIN)) {
            av_usleep(10000);
            continue;
        }
        if (ret < 0) {
            av_thread_message_queue_set_err_recv(f->in_thread_queue, ret);
            break;
        }
    }

FFMpeg不提供接口，那么就只有修改FFMpeg源码，浏览FFMpeg源码一天后，对外的接口只需要在AVPacket结构体里面增加一个判断包完整性的标志变量，修改源码后的接口调用如下：


    while (1) {
        AVPacket pkt;

        ret = av_read_frame(f->ctx, &pkt);
        if (pkt.nLostPackets) {
            // Do something.
        } else {
            // Do something
        }
    }

下面将介绍修改FFMpeg源码的细节。

修改FFMpeg源码(ffmpeg2.8.6)

一、avformat.h里面增加int av_read_frame_aozhen(AVFormatContext s, AVPacket pkt)函数：

并且在对应实现文件utils.c里面对其实现：

二、avcodec.h里面的AVPacket结构体增加成员变量int nIsLostPackets：

并且在avpacket.c里面的av_init_packets函数里面对其初始化：

三、utils.c里面read_frame_internal函数增加临时变量int nIsLostPackets = 0，read_frame_internal函数调用ff_read_packet的后一句增加nIsLostPackets = cur_pkt.nIsLostPackets：

并且在函数末尾将nIsLostPackets赋值给pkt->nIsLostPackets：