dwt: check malloc calls

Signed-off-by: Diego Biurrun <diego@biurrun.de>
This commit is contained in:
Jordi Ortiz 2012-05-22 13:18:17 +02:00 committed by Diego Biurrun
parent 3ea5429489
commit c89e428ed8

View File

@ -33,10 +33,24 @@ void ff_slice_buffer_init(slice_buffer *buf, int line_count,
buf->line_width = line_width;
buf->data_count = max_allocated_lines;
buf->line = av_mallocz(sizeof(IDWTELEM *) * line_count);
if (!buf->line)
return AVERROR(ENOMEM);
buf->data_stack = av_malloc(sizeof(IDWTELEM *) * max_allocated_lines);
if (!buf->data_stack) {
av_free(buf->line);
return AVERROR(ENOMEM);
}
for (i = 0; i < max_allocated_lines; i++)
for (i = 0; i < max_allocated_lines; i++) {
buf->data_stack[i] = av_malloc(sizeof(IDWTELEM) * line_width);
if (!buf->data_stack[i]) {
for (i--; i >=0; i--)
av_free(buf->data_stack[i]);
av_free(buf->data_stack);
av_free(buf->line);
return AVERROR(ENOMEM);
}
}
buf->data_stack_top = max_allocated_lines - 1;
}