Commit fb6dd770 authored by Michael Niedermayer's avatar Michael Niedermayer

Merge commit '73dacabf'

* commit '73dacabf':
  ffv1: Check memory allocations

Conflicts:
	libavcodec/ffv1.c
Merged-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parents f4f3065c 73dacabf
......@@ -111,22 +111,22 @@ av_cold int ffv1_init_slices_state(FFV1Context *f)
av_cold int ffv1_init_slice_contexts(FFV1Context *f)
{
int i;
int i, j;
f->slice_count = f->num_h_slices * f->num_v_slices;
av_assert0(f->slice_count > 0);
for (i = 0; i < f->slice_count; i++) {
FFV1Context *fs = av_mallocz(sizeof(*fs));
int sx = i % f->num_h_slices;
int sy = i / f->num_h_slices;
int sxs = f->avctx->width * sx / f->num_h_slices;
int sxe = f->avctx->width * (sx + 1) / f->num_h_slices;
int sys = f->avctx->height * sy / f->num_v_slices;
int sye = f->avctx->height * (sy + 1) / f->num_v_slices;
FFV1Context *fs = av_mallocz(sizeof(*fs));
if (!fs)
return AVERROR(ENOMEM);
goto memfail;
f->slice_context[i] = fs;
memcpy(fs, f, sizeof(*fs));
......@@ -139,10 +139,19 @@ av_cold int ffv1_init_slice_contexts(FFV1Context *f)
fs->sample_buffer = av_malloc_array((fs->width + 6), 3 * MAX_PLANES *
sizeof(*fs->sample_buffer));
if (!fs->sample_buffer)
return AVERROR(ENOMEM);
if (!fs->sample_buffer) {
av_free(fs);
goto memfail;
}
}
return 0;
memfail:
for (j = 0; j < i; j++) {
av_free(&f->slice_context[j]->sample_buffer);
av_free(&f->slice_context[j]);
}
return AVERROR(ENOMEM);
}
int ffv1_allocate_initial_states(FFV1Context *f)
......
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