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 = const uint64_t argb_scratch_size =
enc->use_predict_ ? tile_size * width + width : 0; enc->use_predict_ ? tile_size * width + width : 0;
const int transform_data_size = const int transform_data_size =
(enc->use_predict_ || enc->use_cross_color_) ? (enc->use_predict_ || enc->use_cross_color_)
VP8LSubSampleSize(width, enc->transform_bits_) * ? VP8LSubSampleSize(width, enc->transform_bits_) *
VP8LSubSampleSize(height, enc->transform_bits_) : 0; VP8LSubSampleSize(height, enc->transform_bits_)
: 0;
const uint64_t total_size = 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)); uint32_t* mem = (uint32_t*)WebPSafeMalloc(total_size, sizeof(*mem));
if (mem == NULL) { if (mem == NULL) {
err = VP8_ENC_ERROR_OUT_OF_MEMORY; err = VP8_ENC_ERROR_OUT_OF_MEMORY;
goto Error; goto Error;
} }
// TODO(skal): align
enc->argb_ = mem; enc->argb_ = mem;
mem += image_size; mem = (uint32_t*)WEBP_ALIGN(mem + image_size);
enc->argb_scratch_ = mem; enc->argb_scratch_ = mem;
mem += argb_scratch_size; mem = (uint32_t*)WEBP_ALIGN(mem + argb_scratch_size);
enc->transform_data_ = mem; enc->transform_data_ = mem;
enc->current_width_ = width; enc->current_width_ = width;
} }