Commit 4332b01c authored by Michael Niedermayer's avatar Michael Niedermayer

avcodec/huffyuv: simplify allocation of temporaries

This also fixes a null pointer dereference with rgb and plane prediction
Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent a88dee8e
......@@ -58,17 +58,11 @@ av_cold int ff_huffyuv_alloc_temp(HYuvContext *s)
{
int i;
if (s->bitstream_bpp<24 || s->version > 2) {
for (i=0; i<3; i++) {
s->temp[i]= av_malloc(2*s->width + 16);
if (!s->temp[i])
return AVERROR(ENOMEM);
s->temp16[i] = (uint16_t*)s->temp[i];
}
} else {
s->temp[0]= av_mallocz(4*s->width + 16);
if (!s->temp[0])
for (i=0; i<3; i++) {
s->temp[i]= av_malloc(4*s->width + 16);
if (!s->temp[i])
return AVERROR(ENOMEM);
s->temp16[i] = (uint16_t*)s->temp[i];
}
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