Commit e29153f4 authored by Lukasz Marek's avatar Lukasz Marek

lavc/avuienc: fix mem leak in case of init failure

Signed-off-by: 's avatarLukasz Marek <lukasz.m.luki2@gmail.com>
parent 23e91a1b
...@@ -25,16 +25,10 @@ ...@@ -25,16 +25,10 @@
static av_cold int avui_encode_init(AVCodecContext *avctx) static av_cold int avui_encode_init(AVCodecContext *avctx)
{ {
avctx->coded_frame = av_frame_alloc();
if (avctx->width != 720 || avctx->height != 486 && avctx->height != 576) { if (avctx->width != 720 || avctx->height != 486 && avctx->height != 576) {
av_log(avctx, AV_LOG_ERROR, "Only 720x486 and 720x576 are supported.\n"); av_log(avctx, AV_LOG_ERROR, "Only 720x486 and 720x576 are supported.\n");
return AVERROR(EINVAL); return AVERROR(EINVAL);
} }
if (!avctx->coded_frame) {
av_log(avctx, AV_LOG_ERROR, "Could not allocate frame.\n");
return AVERROR(ENOMEM);
}
if (!(avctx->extradata = av_mallocz(24 + FF_INPUT_BUFFER_PADDING_SIZE))) if (!(avctx->extradata = av_mallocz(24 + FF_INPUT_BUFFER_PADDING_SIZE)))
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
avctx->extradata_size = 24; avctx->extradata_size = 24;
...@@ -45,6 +39,11 @@ static av_cold int avui_encode_init(AVCodecContext *avctx) ...@@ -45,6 +39,11 @@ static av_cold int avui_encode_init(AVCodecContext *avctx)
avctx->extradata[19] = 1; avctx->extradata[19] = 1;
} }
avctx->coded_frame = av_frame_alloc();
if (!avctx->coded_frame) {
av_log(avctx, AV_LOG_ERROR, "Could not allocate frame.\n");
return AVERROR(ENOMEM);
}
return 0; return 0;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment