为什么将音频流添加到ffmpeg的libavcodec输出容器会导致崩溃?

就目前而言,我的项目正确使用 libavcodec 解码视频,其中每一帧都被操纵(无论如何)并输出到新视频。我从网上找到的例子拼凑起来,它的工作原理。结果是处理过的帧的完美 .mp4,减去音频。

我的问题是,当我尝试将音频流添加到输出容器时,我在 mux.c 中遇到了无法解释的崩溃。它在static int compute_muxer_pkt_fields(AVFormatContext *s, AVStream *st, AVPacket *pkt). 当st->internal->priv_pts->val = pkt->dts;尝试,priv_pts是nullptr。

我不记得版本号,但这是 2020 年 11 月 4 日 ffmpeg 从 git 构建的。

我的MediaContentMgr比我在这里的要大得多。我正在剥离与帧操作有关的所有内容,因此如果我遗漏了任何内容,请告诉我,我会进行编辑。

添加时触发 nullptr 异常的代码被内联调用

.h:

#ifndef _API_EXAMPLE_H
#define _API_EXAMPLE_H

#include <glad/glad.h>
#include <GLFW/glfw3.h>
#include "glm/glm.hpp"

extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/avutil.h>
#include <libavutil/opt.h>
#include <libswscale/swscale.h>
}

#include "shader_s.h"

class MediaContainerMgr {
public:
    MediaContainerMgr(const std::string& infile, const std::string& vert, const std::string& frag, 
                      const glm::vec3* extents);
    ~MediaContainerMgr();
    void render();
    bool recording() { return m_recording; }

    // Major thanks to "shi-yan" who helped make this possible:
    // https://github.com/shi-yan/videosamples/blob/master/libavmp4encoding/main.cpp
    bool init_video_output(const std::string& video_file_name, unsigned int width, unsigned int height);
    bool output_video_frame(uint8_t* buf);
    bool finalize_output();

private:
    AVFormatContext*   m_format_context;
    AVCodec*           m_video_codec;
    AVCodec*           m_audio_codec;
    AVCodecParameters* m_video_codec_parameters;
    AVCodecParameters* m_audio_codec_parameters;
    AVCodecContext*    m_codec_context;
    AVFrame*           m_frame;
    AVPacket*          m_packet;
    uint32_t           m_video_stream_index;
    uint32_t           m_audio_stream_index;
    
    void init_rendering(const glm::vec3* extents);
    int decode_packet();

    // For writing the output video:
    void free_output_assets();
    bool                   m_recording;
    AVOutputFormat*        m_output_format;
    AVFormatContext*       m_output_format_context;
    AVCodec*               m_output_video_codec;
    AVCodecContext*        m_output_video_codec_context;
    AVFrame*               m_output_video_frame;
    SwsContext*            m_output_scale_context;
    AVStream*              m_output_video_stream;
    
    AVCodec*               m_output_audio_codec;
    AVStream*              m_output_audio_stream;
    AVCodecContext*        m_output_audio_codec_context;
};

#endif

而且,地狱般的.cpp:

#include <stdio.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <inttypes.h>

#include "media_container_manager.h"

MediaContainerMgr::MediaContainerMgr(const std::string& infile, const std::string& vert, const std::string& frag,
    const glm::vec3* extents) :
    m_video_stream_index(-1),
    m_audio_stream_index(-1),
    m_recording(false),
    m_output_format(nullptr),
    m_output_format_context(nullptr),
    m_output_video_codec(nullptr),
    m_output_video_codec_context(nullptr),
    m_output_video_frame(nullptr),
    m_output_scale_context(nullptr),
    m_output_video_stream(nullptr)
{
    // AVFormatContext holds header info from the format specified in the container:
    m_format_context = avformat_alloc_context();
    if (!m_format_context) {
        throw "ERROR could not allocate memory for Format Context";
    }
    
    // open the file and read its header. Codecs are not opened here.
    if (avformat_open_input(&m_format_context, infile.c_str(), NULL, NULL) != 0) {
        throw "ERROR could not open input file for reading";
    }

    printf("format %s, duration %lldus, bit_rate %lldn", m_format_context->iformat->name, m_format_context->duration, m_format_context->bit_rate);
    //read avPackets (?) from the avFormat (?) to get stream info. This populates format_context->streams.
    if (avformat_find_stream_info(m_format_context, NULL) < 0) {
        throw "ERROR could not get stream info";
    }

    for (unsigned int i = 0; i < m_format_context->nb_streams; i++) {
        AVCodecParameters* local_codec_parameters = NULL;
        local_codec_parameters = m_format_context->streams[i]->codecpar;
        printf("AVStream->time base before open coded %d/%dn", m_format_context->streams[i]->time_base.num, m_format_context->streams[i]->time_base.den);
        printf("AVStream->r_frame_rate before open coded %d/%dn", m_format_context->streams[i]->r_frame_rate.num, m_format_context->streams[i]->r_frame_rate.den);
        printf("AVStream->start_time %" PRId64 "n", m_format_context->streams[i]->start_time);
        printf("AVStream->duration %" PRId64 "n", m_format_context->streams[i]->duration);
        printf("duration(s): %lfn", (float)m_format_context->streams[i]->duration / m_format_context->streams[i]->time_base.den * m_format_context->streams[i]->time_base.num);
        AVCodec* local_codec = NULL;
        local_codec = avcodec_find_decoder(local_codec_parameters->codec_id);
        if (local_codec == NULL) {
            throw "ERROR unsupported codec!";
        }

        if (local_codec_parameters->codec_type == AVMEDIA_TYPE_VIDEO) {
            if (m_video_stream_index == -1) {
                m_video_stream_index = i;
                m_video_codec = local_codec;
                m_video_codec_parameters = local_codec_parameters;
            }
            m_height = local_codec_parameters->height;
            m_width = local_codec_parameters->width;
            printf("Video Codec: resolution %dx%dn", m_width, m_height);
        }
        else if (local_codec_parameters->codec_type == AVMEDIA_TYPE_AUDIO) {
            if (m_audio_stream_index == -1) {
                m_audio_stream_index = i;
                m_audio_codec = local_codec;
                m_audio_codec_parameters = local_codec_parameters;
            }
            printf("Audio Codec: %d channels, sample rate %dn", local_codec_parameters->channels, local_codec_parameters->sample_rate);
        }

        printf("tCodec %s ID %d bit_rate %lldn", local_codec->name, local_codec->id, local_codec_parameters->bit_rate);
    }

    m_codec_context = avcodec_alloc_context3(m_video_codec);
    if (!m_codec_context) {
        throw "ERROR failed to allocate memory for AVCodecContext";
    }

    if (avcodec_parameters_to_context(m_codec_context, m_video_codec_parameters) < 0) {
        throw "ERROR failed to copy codec params to codec context";
    }

    if (avcodec_open2(m_codec_context, m_video_codec, NULL) < 0) {
        throw "ERROR avcodec_open2 failed to open codec";
    }

    m_frame = av_frame_alloc();
    if (!m_frame) {
        throw "ERROR failed to allocate AVFrame memory";
    }

    m_packet = av_packet_alloc();
    if (!m_packet) {
        throw "ERROR failed to allocate AVPacket memory";
    }
}

MediaContainerMgr::~MediaContainerMgr() {
    avformat_close_input(&m_format_context);
    av_packet_free(&m_packet);
    av_frame_free(&m_frame);
    avcodec_free_context(&m_codec_context);


    glDeleteVertexArrays(1, &m_VAO);
    glDeleteBuffers(1, &m_VBO);
}


bool MediaContainerMgr::advance_frame() {
    while (true) {
        if (av_read_frame(m_format_context, m_packet) < 0) {
            // Do we actually need to unref the packet if it failed?
            av_packet_unref(m_packet);
            continue;
            //return false;
        }
        else {
            if (m_packet->stream_index == m_video_stream_index) {
                //printf("AVPacket->pts %" PRId64 "n", m_packet->pts);
                int response = decode_packet();
                av_packet_unref(m_packet);
                if (response != 0) {
                    continue;
                    //return false;
                }
                return true;
            }
            else {
                printf("m_packet->stream_index: %dn", m_packet->stream_index);
                printf("  m_packet->pts: %lldn", m_packet->pts);
                printf("  mpacket->size: %dn", m_packet->size);
                if (m_recording) {
                    int err = 0;
                    //err = avcodec_send_packet(m_output_video_codec_context, m_packet);
                    printf("  encoding error: %dn", err);
                }
            }
        }

        // We're done with the packet (it's been unpacked to a frame), so deallocate & reset to defaults:
/*
        if (m_frame == NULL)
            return false;

        if (m_frame->data[0] == NULL || m_frame->data[1] == NULL || m_frame->data[2] == NULL) {
            printf("WARNING: null frame data");
            continue;
        }
*/
    }
}

int MediaContainerMgr::decode_packet() {
    // Supply raw packet data as input to a decoder
    // https://ffmpeg.org/doxygen/trunk/group__lavc__decoding.html#ga58bc4bf1e0ac59e27362597e467efff3
    int response = avcodec_send_packet(m_codec_context, m_packet);

    if (response < 0) {
        char buf[256];
        av_strerror(response, buf, 256);
        printf("Error while receiving a frame from the decoder: %sn", buf);
        return response;
    }

    // Return decoded output data (into a frame) from a decoder
    // https://ffmpeg.org/doxygen/trunk/group__lavc__decoding.html#ga11e6542c4e66d3028668788a1a74217c
    response = avcodec_receive_frame(m_codec_context, m_frame);
    if (response == AVERROR(EAGAIN) || response == AVERROR_EOF) {
        return response;
    } else if (response < 0) {
        char buf[256];
        av_strerror(response, buf, 256);
        printf("Error while receiving a frame from the decoder: %sn", buf);
        return response;
    } else {
        printf(
            "Frame %d (type=%c, size=%d bytes) pts %lld key_frame %d [DTS %d]n",
            m_codec_context->frame_number,
            av_get_picture_type_char(m_frame->pict_type),
            m_frame->pkt_size,
            m_frame->pts,
            m_frame->key_frame,
            m_frame->coded_picture_number
        );
    }
    return 0;
}


bool MediaContainerMgr::init_video_output(const std::string& video_file_name, unsigned int width, unsigned int height) {
    if (m_recording)
        return true;
    m_recording = true;

    advance_to(0L); // I've deleted the implmentation. Just seeks to beginning of vid. Works fine.

    if (!(m_output_format = av_guess_format(nullptr, video_file_name.c_str(), nullptr))) {
        printf("Cannot guess output format.n");
        return false;
    }

    int err = avformat_alloc_output_context2(&m_output_format_context, m_output_format, nullptr, video_file_name.c_str());
    if (err < 0) {
        printf("Failed to allocate output context.n");
        return false;
    }

    //TODO(P0): Break out the video and audio inits into their own methods.
    m_output_video_codec = avcodec_find_encoder(m_output_format->video_codec);
    if (!m_output_video_codec) {
        printf("Failed to create video codec.n");
        return false;
    }
    m_output_video_stream = avformat_new_stream(m_output_format_context, m_output_video_codec);
    if (!m_output_video_stream) {
        printf("Failed to find video format.n");
        return false;
    } 
    m_output_video_codec_context = avcodec_alloc_context3(m_output_video_codec);
    if (!m_output_video_codec_context) {
        printf("Failed to create video codec context.n");
        return(false);
    }
    m_output_video_stream->codecpar->codec_id = m_output_format->video_codec;
    m_output_video_stream->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
    m_output_video_stream->codecpar->width = width;
    m_output_video_stream->codecpar->height = height;
    m_output_video_stream->codecpar->format = AV_PIX_FMT_YUV420P;
    // Use the same bit rate as the input stream.
    m_output_video_stream->codecpar->bit_rate = m_format_context->streams[m_video_stream_index]->codecpar->bit_rate;
    m_output_video_stream->avg_frame_rate = m_format_context->streams[m_video_stream_index]->avg_frame_rate;
    avcodec_parameters_to_context(m_output_video_codec_context, m_output_video_stream->codecpar);
    m_output_video_codec_context->time_base = m_format_context->streams[m_video_stream_index]->time_base;
    
    //TODO(P1): Set these to match the input stream?
    m_output_video_codec_context->max_b_frames = 2;
    m_output_video_codec_context->gop_size = 12;
    m_output_video_codec_context->framerate = m_format_context->streams[m_video_stream_index]->r_frame_rate;
    //m_output_codec_context->refcounted_frames = 0;
    if (m_output_video_stream->codecpar->codec_id == AV_CODEC_ID_H264) {
        av_opt_set(m_output_video_codec_context, "preset", "ultrafast", 0);
    } else if (m_output_video_stream->codecpar->codec_id == AV_CODEC_ID_H265) {
        av_opt_set(m_output_video_codec_context, "preset", "ultrafast", 0);
    } else {
        av_opt_set_int(m_output_video_codec_context, "lossless", 1, 0);
    }
    avcodec_parameters_from_context(m_output_video_stream->codecpar, m_output_video_codec_context);

    m_output_audio_codec = avcodec_find_encoder(m_output_format->audio_codec);
    if (!m_output_audio_codec) {
        printf("Failed to create audio codec.n");
        return false;
    }

我已经注释掉了下一行之外的所有音频流初始化,因为这是问题开始的地方。创建此输出流会导致我提到的空引用。如果我在这里取消注释下面的所有内容,我仍然会得到空 deref。如果我注释掉这一行,则 deref 异常就会消失。(IOW,我注释掉了越来越多的代码,直到我发现这是导致问题的触发器。)

我认为在其余注释掉的代码中我做错了一些事情,当修复时,将修复 nullptr 并给我一个有效的音频流。

    m_output_audio_stream = avformat_new_stream(m_output_format_context, m_output_audio_codec);
    if (!m_output_audio_stream) {
        printf("Failed to find audio format.n");
        return false;
    }
    /*
    m_output_audio_codec_context = avcodec_alloc_context3(m_output_audio_codec);
    if (!m_output_audio_codec_context) {
        printf("Failed to create audio codec context.n");
        return(false);
    }
    m_output_audio_stream->codecpar->codec_id = m_output_format->audio_codec;
    m_output_audio_stream->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
    m_output_audio_stream->codecpar->format = m_format_context->streams[m_audio_stream_index]->codecpar->format;
    m_output_audio_stream->codecpar->bit_rate = m_format_context->streams[m_audio_stream_index]->codecpar->bit_rate;
    m_output_audio_stream->avg_frame_rate = m_format_context->streams[m_audio_stream_index]->avg_frame_rate;
    avcodec_parameters_to_context(m_output_audio_codec_context, m_output_audio_stream->codecpar);
    m_output_audio_codec_context->time_base = m_format_context->streams[m_audio_stream_index]->time_base;
    */

    //TODO(P2): Free assets that have been allocated.
    err = avcodec_open2(m_output_video_codec_context, m_output_video_codec, nullptr);
    if (err < 0) {
        printf("Failed to open codec.n");
        return false;
    }

    if (!(m_output_format->flags & AVFMT_NOFILE)) {
        err = avio_open(&m_output_format_context->pb, video_file_name.c_str(), AVIO_FLAG_WRITE);
        if (err < 0) {
            printf("Failed to open output file.");
            return false;
        }
    }

    err = avformat_write_header(m_output_format_context, NULL);
    if (err < 0) {
        printf("Failed to write header.n");
        return false;
    }

    av_dump_format(m_output_format_context, 0, video_file_name.c_str(), 1);

    return true;
}


//TODO(P2): make this a member. (Thanks to https://emvlo.wordpress.com/2016/03/10/sws_scale/)
void PrepareFlipFrameJ420(AVFrame* pFrame) {
    for (int i = 0; i < 4; i++) {
        if (i)
            pFrame->data[i] += pFrame->linesize[i] * ((pFrame->height >> 1) - 1);
        else
            pFrame->data[i] += pFrame->linesize[i] * (pFrame->height - 1);
        pFrame->linesize[i] = -pFrame->linesize[i];
    }
}

这是我们获取更改后的帧并将其写入输出容器的地方。只要我们没有在输出容器中设置音频流,这就可以正常工作。

bool MediaContainerMgr::output_video_frame(uint8_t* buf) {
int err;

if (!m_output_video_frame) {
m_output_video_frame = av_frame_alloc();
m_output_video_frame->format = AV_PIX_FMT_YUV420P;
m_output_video_frame->width = m_output_video_codec_context->width;
m_output_video_frame->height = m_output_video_codec_context->height;
err = av_frame_get_buffer(m_output_video_frame, 32);
if (err < 0) {
printf("Failed to allocate output frame.n");
return false;
}
}

if (!m_output_scale_context) {
m_output_scale_context = sws_getContext(m_output_video_codec_context->width, m_output_video_codec_context->height,
AV_PIX_FMT_RGB24,
m_output_video_codec_context->width, m_output_video_codec_context->height,
AV_PIX_FMT_YUV420P, SWS_BICUBIC, nullptr, nullptr, nullptr);
}

int inLinesize[1] = { 3 * m_output_video_codec_context->width };
sws_scale(m_output_scale_context, (const uint8_t* const*)&buf, inLinesize, 0, m_output_video_codec_context->height,
m_output_video_frame->data, m_output_video_frame->linesize);
PrepareFlipFrameJ420(m_output_video_frame);
//TODO(P0): Switch m_frame to be m_input_video_frame so I don't end up using the presentation timestamp from
// an audio frame if I threadify the frame reading.
m_output_video_frame->pts = m_frame->pts;
printf("Output PTS: %d, time_base: %d/%dn", m_output_video_frame->pts,
m_output_video_codec_context->time_base.num, m_output_video_codec_context->time_base.den);
err = avcodec_send_frame(m_output_video_codec_context, m_output_video_frame);
if (err < 0) {
printf(" ERROR sending new video frame output: ");
switch (err) {
case AVERROR(EAGAIN):
printf("AVERROR(EAGAIN): %dn", err);
break;
case AVERROR_EOF:
printf("AVERROR_EOF: %dn", err);
break;
case AVERROR(EINVAL):
printf("AVERROR(EINVAL): %dn", err);
break;
case AVERROR(ENOMEM):
printf("AVERROR(ENOMEM): %dn", err);
break;
}

return false;
}

AVPacket pkt;
av_init_packet(&pkt);
pkt.data = nullptr;
pkt.size = 0;
pkt.flags |= AV_PKT_FLAG_KEY;
int ret = 0;
if ((ret = avcodec_receive_packet(m_output_video_codec_context, &pkt)) == 0) {
static int counter = 0;
printf("pkt.key: 0x%08x, pkt.size: %d, counter:n", pkt.flags & AV_PKT_FLAG_KEY, pkt.size, counter++);
uint8_t* size = ((uint8_t*)pkt.data);
printf("sizes: %d %d %d %d %d %d %d %d %dn", size[0], size[1], size[2], size[2], size[3], size[4], size[5], size[6], size[7]);
av_interleaved_write_frame(m_output_format_context, &pkt);
}
printf("push: %dn", ret);
av_packet_unref(&pkt);

return true;
}

bool MediaContainerMgr::finalize_output() {
if (!m_recording)
return true;

AVPacket pkt;
av_init_packet(&pkt);
pkt.data = nullptr;
pkt.size = 0;

for (;;) {
avcodec_send_frame(m_output_video_codec_context, nullptr);
if (avcodec_receive_packet(m_output_video_codec_context, &pkt) == 0) {
av_inter

以上是为什么将音频流添加到ffmpeg的libavcodec输出容器会导致崩溃?的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>