Merge "fix alignment of allocated memory in AllocateTransformBuffer"

This commit is contained in:
James Zern 2015-10-19 19:15:04 +00:00 committed by Gerrit Code Review
commit 927ccdc43b

View File

@ -1127,21 +1127,23 @@ static WebPEncodingError AllocateTransformBuffer(VP8LEncoder* const enc,
const uint64_t argb_scratch_size =
enc->use_predict_ ? tile_size * width + width : 0;
const int transform_data_size =
(enc->use_predict_ || enc->use_cross_color_) ?
VP8LSubSampleSize(width, enc->transform_bits_) *
VP8LSubSampleSize(height, enc->transform_bits_) : 0;
(enc->use_predict_ || enc->use_cross_color_)
? VP8LSubSampleSize(width, enc->transform_bits_) *
VP8LSubSampleSize(height, enc->transform_bits_)
: 0;
const uint64_t total_size =
image_size + argb_scratch_size + (uint64_t)transform_data_size;
image_size + WEBP_ALIGN_CST +
argb_scratch_size + WEBP_ALIGN_CST +
(uint64_t)transform_data_size;
uint32_t* mem = (uint32_t*)WebPSafeMalloc(total_size, sizeof(*mem));
if (mem == NULL) {
err = VP8_ENC_ERROR_OUT_OF_MEMORY;
goto Error;
}
// TODO(skal): align
enc->argb_ = mem;
mem += image_size;
mem = (uint32_t*)WEBP_ALIGN(mem + image_size);
enc->argb_scratch_ = mem;
mem += argb_scratch_size;
mem = (uint32_t*)WEBP_ALIGN(mem + argb_scratch_size);
enc->transform_data_ = mem;
enc->current_width_ = width;
}