speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix memory leak of FfmpegLoad

Browse Source
This commit is contained in:
雾聪 2023-12-19 11:45:24 +08:00
parent 3d4398a362
commit 6823fe0c82
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
runtime/onnxruntime/src/audio.cpp
View File

@ -462,6 +462,10 @@ bool Audio::FfmpegLoad(const char* buf, int n_file_len){
nullptr, // write callback (not used here) nullptr, // write callback (not used here)
nullptr // seek callback (not used here) nullptr // seek callback (not used here)
); );
if (!avio_ctx) {
av_free(buf_copy);
return false;
}
AVFormatContext* formatContext = avformat_alloc_context(); AVFormatContext* formatContext = avformat_alloc_context();
formatContext->pb = avio_ctx; formatContext->pb = avio_ctx;
if (avformat_open_input(&formatContext, "", NULL, NULL) != 0) { if (avformat_open_input(&formatContext, "", NULL, NULL) != 0) {
@ -576,7 +580,9 @@ bool Audio::FfmpegLoad(const char* buf, int n_file_len){
av_packet_unref(packet); av_packet_unref(packet);
} }
avio_context_free(&avio_ctx); //avio_context_free(&avio_ctx);
av_freep(&avio_ctx ->buffer);
av_freep(&avio_ctx);
avformat_close_input(&formatContext); avformat_close_input(&formatContext);
avformat_free_context(formatContext); avformat_free_context(formatContext);
avcodec_free_context(&codecContext); avcodec_free_context(&codecContext);