mirror of https://git.ffmpeg.org/ffmpeg.git
avcodec/nvdec: Check av_buffer_ref()
It (unfortunately) involves an allocation and can therefore fail. Reviewed-by: Timo Rothenpieler <timo@rothenpieler.org> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
parent
eb608fd0b0
commit
d3730acca3
|
@ -532,8 +532,11 @@ static int nvdec_retrieve_data(void *logctx, AVFrame *frame)
|
||||||
}
|
}
|
||||||
|
|
||||||
unmap_data->idx = cf->idx;
|
unmap_data->idx = cf->idx;
|
||||||
unmap_data->idx_ref = av_buffer_ref(cf->idx_ref);
|
if (!(unmap_data->idx_ref = av_buffer_ref(cf->idx_ref)) ||
|
||||||
unmap_data->decoder_ref = av_buffer_ref(cf->decoder_ref);
|
!(unmap_data->decoder_ref = av_buffer_ref(cf->decoder_ref))) {
|
||||||
|
ret = AVERROR(ENOMEM);
|
||||||
|
goto copy_fail;
|
||||||
|
}
|
||||||
|
|
||||||
av_pix_fmt_get_chroma_sub_sample(hwctx->sw_format, &shift_h, &shift_v);
|
av_pix_fmt_get_chroma_sub_sample(hwctx->sw_format, &shift_h, &shift_v);
|
||||||
for (i = 0; frame->linesize[i]; i++) {
|
for (i = 0; frame->linesize[i]; i++) {
|
||||||
|
|
Loading…
Reference in New Issue