har0ke/FFmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

avcodec/libutvideoenc: Check avpicture_get_size() return code

Browse Source 
Fixes CID1257656

Reviewed-by: Derek Buitenhuis <derek.buitenhuis@gmail.com>
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2014-12-11 13:55:26 +01:00
parent 086e29a011
commit f96fcba1e3
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libavcodec/libutvideoenc.cpp
View File

@ -39,6 +39,7 @@ static av_cold int utvideo_encode_init(AVCodecContext *avctx)
UtVideoContext *utv = (UtVideoContext *)avctx->priv_data;
UtVideoExtra *info;
uint32_t flags, in_format;
int ret;
switch (avctx->pix_fmt) {
case AV_PIX_FMT_YUV420P:
@ -94,8 +95,11 @@ static av_cold int utvideo_encode_init(AVCodecContext *avctx)
* We use this buffer to hold the data that Ut Video returns,
* since we cannot decode planes separately with it.
*/
utv->buf_size = avpicture_get_size(avctx->pix_fmt,
avctx->width, avctx->height);
ret = avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height);
if (ret < 0)
return ret;
utv->buf_size = ret;
utv->buffer = (uint8_t *)av_malloc(utv->buf_size);
if (utv->buffer == NULL) {