| Index: third_party/libwebp/dec/webp.c
|
| diff --git a/third_party/libwebp/dec/webp.c b/third_party/libwebp/dec/webp.c
|
| index 91ac75f7bb88a967eb6476817b1430b46e21934a..edd348cbe70c509dc721bffd502621e8c9f08d22 100644
|
| --- a/third_party/libwebp/dec/webp.c
|
| +++ b/third_party/libwebp/dec/webp.c
|
| @@ -1,4 +1,4 @@
|
| -// Copyright 2010 Google Inc.
|
| +// Copyright 2010 Google Inc. All Rights Reserved.
|
| //
|
| // This code is licensed under the same terms as WebM:
|
| // Software License Agreement: http://www.webmproject.org/license/software/
|
| @@ -10,8 +10,11 @@
|
| // Author: Skal (pascal.massimino@gmail.com)
|
|
|
| #include <stdlib.h>
|
| -#include "vp8i.h"
|
| -#include "webpi.h"
|
| +
|
| +#include "./vp8i.h"
|
| +#include "./vp8li.h"
|
| +#include "./webpi.h"
|
| +#include "../webp/format_constants.h"
|
|
|
| #if defined(__cplusplus) || defined(c_plusplus)
|
| extern "C" {
|
| @@ -24,105 +27,148 @@ extern "C" {
|
| // 4...7 size of image data (including metadata) starting at offset 8
|
| // 8...11 "WEBP" our form-type signature
|
| // The RIFF container (12 bytes) is followed by appropriate chunks:
|
| -// 12..15 "VP8 ": 4-bytes tags, describing the raw video format used
|
| +// 12..15 "VP8 ": 4-bytes tags, signaling the use of VP8 video format
|
| // 16..19 size of the raw VP8 image data, starting at offset 20
|
| // 20.... the VP8 bytes
|
| // Or,
|
| +// 12..15 "VP8L": 4-bytes tags, signaling the use of VP8L lossless format
|
| +// 16..19 size of the raw VP8L image data, starting at offset 20
|
| +// 20.... the VP8L bytes
|
| +// Or,
|
| // 12..15 "VP8X": 4-bytes tags, describing the extended-VP8 chunk.
|
| // 16..19 size of the VP8X chunk starting at offset 20.
|
| // 20..23 VP8X flags bit-map corresponding to the chunk-types present.
|
| -// 24..27 Width of the Canvas Image.
|
| -// 28..31 Height of the Canvas Image.
|
| +// 24..26 Width of the Canvas Image.
|
| +// 27..29 Height of the Canvas Image.
|
| // There can be extra chunks after the "VP8X" chunk (ICCP, TILE, FRM, VP8,
|
| // META ...)
|
| -// All 32-bits sizes are in little-endian order.
|
| -// Note: chunk data must be padded to multiple of 2 in size
|
| +// All sizes are in little-endian order.
|
| +// Note: chunk data size must be padded to multiple of 2 when written.
|
|
|
| -static inline uint32_t get_le32(const uint8_t* const data) {
|
| - return data[0] | (data[1] << 8) | (data[2] << 16) | (data[3] << 24);
|
| +static WEBP_INLINE uint32_t get_le24(const uint8_t* const data) {
|
| + return data[0] | (data[1] << 8) | (data[2] << 16);
|
| }
|
|
|
| -VP8StatusCode WebPParseRIFF(const uint8_t** data, uint32_t* data_size,
|
| - uint32_t* riff_size) {
|
| - assert(data);
|
| - assert(data_size);
|
| - assert(riff_size);
|
| +static WEBP_INLINE uint32_t get_le32(const uint8_t* const data) {
|
| + return (uint32_t)get_le24(data) | (data[3] << 24);
|
| +}
|
|
|
| - if (*data_size >= RIFF_HEADER_SIZE &&
|
| - !memcmp(*data, "RIFF", TAG_SIZE)) {
|
| +// Validates the RIFF container (if detected) and skips over it.
|
| +// If a RIFF container is detected,
|
| +// Returns VP8_STATUS_BITSTREAM_ERROR for invalid header, and
|
| +// VP8_STATUS_OK otherwise.
|
| +// In case there are not enough bytes (partial RIFF container), return 0 for
|
| +// *riff_size. Else return the RIFF size extracted from the header.
|
| +static VP8StatusCode ParseRIFF(const uint8_t** const data,
|
| + size_t* const data_size,
|
| + size_t* const riff_size) {
|
| + assert(data != NULL);
|
| + assert(data_size != NULL);
|
| + assert(riff_size != NULL);
|
| +
|
| + *riff_size = 0; // Default: no RIFF present.
|
| + if (*data_size >= RIFF_HEADER_SIZE && !memcmp(*data, "RIFF", TAG_SIZE)) {
|
| if (memcmp(*data + 8, "WEBP", TAG_SIZE)) {
|
| return VP8_STATUS_BITSTREAM_ERROR; // Wrong image file signature.
|
| } else {
|
| - *riff_size = get_le32(*data + TAG_SIZE);
|
| + const uint32_t size = get_le32(*data + TAG_SIZE);
|
| // Check that we have at least one chunk (i.e "WEBP" + "VP8?nnnn").
|
| - if (*riff_size < TAG_SIZE + CHUNK_HEADER_SIZE) {
|
| + if (size < TAG_SIZE + CHUNK_HEADER_SIZE) {
|
| return VP8_STATUS_BITSTREAM_ERROR;
|
| }
|
| // We have a RIFF container. Skip it.
|
| + *riff_size = size;
|
| *data += RIFF_HEADER_SIZE;
|
| *data_size -= RIFF_HEADER_SIZE;
|
| }
|
| - } else {
|
| - *riff_size = 0; // Did not get full RIFF Header.
|
| }
|
| return VP8_STATUS_OK;
|
| }
|
|
|
| -VP8StatusCode WebPParseVP8X(const uint8_t** data, uint32_t* data_size,
|
| - uint32_t* bytes_skipped,
|
| - int* width, int* height, uint32_t* flags) {
|
| - assert(data);
|
| - assert(data_size);
|
| - assert(bytes_skipped);
|
| +// Validates the VP8X header and skips over it.
|
| +// Returns VP8_STATUS_BITSTREAM_ERROR for invalid VP8X header,
|
| +// VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and
|
| +// VP8_STATUS_OK otherwise.
|
| +// If a VP8X chunk is found, found_vp8x is set to true and *width_ptr,
|
| +// *height_ptr and *flags_ptr are set to the corresponding values extracted
|
| +// from the VP8X chunk.
|
| +static VP8StatusCode ParseVP8X(const uint8_t** const data,
|
| + size_t* const data_size,
|
| + int* const found_vp8x,
|
| + int* const width_ptr, int* const height_ptr,
|
| + uint32_t* const flags_ptr) {
|
| + const uint32_t vp8x_size = CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE;
|
| + assert(data != NULL);
|
| + assert(data_size != NULL);
|
| + assert(found_vp8x != NULL);
|
| +
|
| + *found_vp8x = 0;
|
|
|
| - *bytes_skipped = 0;
|
| -
|
| - if (*data_size < CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE) {
|
| + if (*data_size < CHUNK_HEADER_SIZE) {
|
| return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data.
|
| }
|
|
|
| if (!memcmp(*data, "VP8X", TAG_SIZE)) {
|
| + int width, height;
|
| + uint32_t flags;
|
| const uint32_t chunk_size = get_le32(*data + TAG_SIZE);
|
| if (chunk_size != VP8X_CHUNK_SIZE) {
|
| return VP8_STATUS_BITSTREAM_ERROR; // Wrong chunk size.
|
| }
|
| - if (flags) {
|
| - *flags = get_le32(*data + 8);
|
| - }
|
| - if (width) {
|
| - *width = get_le32(*data + 12);
|
| +
|
| + // Verify if enough data is available to validate the VP8X chunk.
|
| + if (*data_size < vp8x_size) {
|
| + return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data.
|
| }
|
| - if (height) {
|
| - *height = get_le32(*data + 16);
|
| + flags = get_le32(*data + 8);
|
| + width = 1 + get_le24(*data + 12);
|
| + height = 1 + get_le24(*data + 15);
|
| + if (width * (uint64_t)height >= MAX_IMAGE_AREA) {
|
| + return VP8_STATUS_BITSTREAM_ERROR; // image is too large
|
| }
|
| - // We have consumed 20 bytes from VP8X. Skip them.
|
| - *bytes_skipped = CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE;
|
| - *data += *bytes_skipped;
|
| - *data_size -= *bytes_skipped;
|
| +
|
| + if (flags_ptr != NULL) *flags_ptr = flags;
|
| + if (width_ptr != NULL) *width_ptr = width;
|
| + if (height_ptr != NULL) *height_ptr = height;
|
| + // Skip over VP8X header bytes.
|
| + *data += vp8x_size;
|
| + *data_size -= vp8x_size;
|
| + *found_vp8x = 1;
|
| }
|
| return VP8_STATUS_OK;
|
| }
|
|
|
| -VP8StatusCode WebPParseOptionalChunks(const uint8_t** data, uint32_t* data_size,
|
| - uint32_t riff_size,
|
| - uint32_t* bytes_skipped) {
|
| +// Skips to the next VP8/VP8L chunk header in the data given the size of the
|
| +// RIFF chunk 'riff_size'.
|
| +// Returns VP8_STATUS_BITSTREAM_ERROR if any invalid chunk size is encountered,
|
| +// VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and
|
| +// VP8_STATUS_OK otherwise.
|
| +// If an alpha chunk is found, *alpha_data and *alpha_size are set
|
| +// appropriately.
|
| +static VP8StatusCode ParseOptionalChunks(const uint8_t** const data,
|
| + size_t* const data_size,
|
| + size_t const riff_size,
|
| + const uint8_t** const alpha_data,
|
| + size_t* const alpha_size) {
|
| const uint8_t* buf;
|
| - uint32_t buf_size;
|
| -
|
| - assert(data);
|
| - assert(data_size);
|
| - assert(bytes_skipped);
|
| -
|
| + size_t buf_size;
|
| + uint32_t total_size = TAG_SIZE + // "WEBP".
|
| + CHUNK_HEADER_SIZE + // "VP8Xnnnn".
|
| + VP8X_CHUNK_SIZE; // data.
|
| + assert(data != NULL);
|
| + assert(data_size != NULL);
|
| buf = *data;
|
| buf_size = *data_size;
|
| - *bytes_skipped = 0;
|
| +
|
| + assert(alpha_data != NULL);
|
| + assert(alpha_size != NULL);
|
| + *alpha_data = NULL;
|
| + *alpha_size = 0;
|
|
|
| while (1) {
|
| uint32_t chunk_size;
|
| - uint32_t cur_skip_size;
|
| - const uint32_t bytes_skipped_header = TAG_SIZE + // "WEBP".
|
| - CHUNK_HEADER_SIZE + // "VP8Xnnnn".
|
| - VP8X_CHUNK_SIZE; // Data.
|
| + uint32_t disk_chunk_size; // chunk_size with padding
|
| +
|
| *data = buf;
|
| *data_size = buf_size;
|
|
|
| @@ -131,119 +177,197 @@ VP8StatusCode WebPParseOptionalChunks(const uint8_t** data, uint32_t* data_size,
|
| }
|
|
|
| chunk_size = get_le32(buf + TAG_SIZE);
|
| - cur_skip_size = CHUNK_HEADER_SIZE + chunk_size;
|
| + // For odd-sized chunk-payload, there's one byte padding at the end.
|
| + disk_chunk_size = (CHUNK_HEADER_SIZE + chunk_size + 1) & ~1;
|
| + total_size += disk_chunk_size;
|
|
|
| - // Check that total bytes skipped along with current chunk size
|
| - // does not exceed riff_size.
|
| - if (riff_size > 0 &&
|
| - (bytes_skipped_header + *bytes_skipped + cur_skip_size > riff_size)) {
|
| - return VP8_STATUS_BITSTREAM_ERROR; // Not a valid chunk size.
|
| + // Check that total bytes skipped so far does not exceed riff_size.
|
| + if (riff_size > 0 && (total_size > riff_size)) {
|
| + return VP8_STATUS_BITSTREAM_ERROR; // Not a valid chunk size.
|
| }
|
|
|
| - if (buf_size < cur_skip_size) { // Insufficient data.
|
| + if (buf_size < disk_chunk_size) { // Insufficient data.
|
| return VP8_STATUS_NOT_ENOUGH_DATA;
|
| }
|
|
|
| - if (!memcmp(buf, "VP8 ", TAG_SIZE)) { // A valid VP8 header.
|
| + if (!memcmp(buf, "ALPH", TAG_SIZE)) { // A valid ALPH header.
|
| + *alpha_data = buf + CHUNK_HEADER_SIZE;
|
| + *alpha_size = chunk_size;
|
| + } else if (!memcmp(buf, "VP8 ", TAG_SIZE) ||
|
| + !memcmp(buf, "VP8L", TAG_SIZE)) { // A valid VP8/VP8L header.
|
| return VP8_STATUS_OK; // Found.
|
| }
|
|
|
| - // We have a full & valid chunk; skip it.
|
| - buf += cur_skip_size;
|
| - buf_size -= cur_skip_size;
|
| - *bytes_skipped += cur_skip_size;
|
| + // We have a full and valid chunk; skip it.
|
| + buf += disk_chunk_size;
|
| + buf_size -= disk_chunk_size;
|
| }
|
| }
|
|
|
| -VP8StatusCode WebPParseVP8Header(const uint8_t** data, uint32_t* data_size,
|
| - uint32_t riff_size, uint32_t* bytes_skipped,
|
| - uint32_t* vp8_chunk_size) {
|
| - assert(data);
|
| - assert(data_size);
|
| - assert(bytes_skipped);
|
| - assert(vp8_chunk_size);
|
| -
|
| - *bytes_skipped = 0;
|
| - *vp8_chunk_size = 0;
|
| +// Validates the VP8/VP8L Header ("VP8 nnnn" or "VP8L nnnn") and skips over it.
|
| +// Returns VP8_STATUS_BITSTREAM_ERROR for invalid (chunk larger than
|
| +// riff_size) VP8/VP8L header,
|
| +// VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and
|
| +// VP8_STATUS_OK otherwise.
|
| +// If a VP8/VP8L chunk is found, *chunk_size is set to the total number of bytes
|
| +// extracted from the VP8/VP8L chunk header.
|
| +// The flag '*is_lossless' is set to 1 in case of VP8L chunk / raw VP8L data.
|
| +static VP8StatusCode ParseVP8Header(const uint8_t** const data_ptr,
|
| + size_t* const data_size,
|
| + size_t riff_size,
|
| + size_t* const chunk_size,
|
| + int* const is_lossless) {
|
| + const uint8_t* const data = *data_ptr;
|
| + const int is_vp8 = !memcmp(data, "VP8 ", TAG_SIZE);
|
| + const int is_vp8l = !memcmp(data, "VP8L", TAG_SIZE);
|
| + const uint32_t minimal_size =
|
| + TAG_SIZE + CHUNK_HEADER_SIZE; // "WEBP" + "VP8 nnnn" OR
|
| + // "WEBP" + "VP8Lnnnn"
|
| + assert(data != NULL);
|
| + assert(data_size != NULL);
|
| + assert(chunk_size != NULL);
|
| + assert(is_lossless != NULL);
|
|
|
| if (*data_size < CHUNK_HEADER_SIZE) {
|
| return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data.
|
| }
|
|
|
| - if (!memcmp(*data, "VP8 ", TAG_SIZE)) {
|
| - *vp8_chunk_size = get_le32(*data + TAG_SIZE);
|
| - if (riff_size >= TAG_SIZE + CHUNK_HEADER_SIZE && // "WEBP" + "VP8 nnnn".
|
| - (*vp8_chunk_size > riff_size - (TAG_SIZE + CHUNK_HEADER_SIZE))) {
|
| + if (is_vp8 || is_vp8l) {
|
| + // Bitstream contains VP8/VP8L header.
|
| + const uint32_t size = get_le32(data + TAG_SIZE);
|
| + if ((riff_size >= minimal_size) && (size > riff_size - minimal_size)) {
|
| return VP8_STATUS_BITSTREAM_ERROR; // Inconsistent size information.
|
| }
|
| - // We have consumed CHUNK_HEADER_SIZE bytes from VP8 Header. Skip them.
|
| - *bytes_skipped = CHUNK_HEADER_SIZE;
|
| - *data += *bytes_skipped;
|
| - *data_size -= *bytes_skipped;
|
| + // Skip over CHUNK_HEADER_SIZE bytes from VP8/VP8L Header.
|
| + *chunk_size = size;
|
| + *data_ptr += CHUNK_HEADER_SIZE;
|
| + *data_size -= CHUNK_HEADER_SIZE;
|
| + *is_lossless = is_vp8l;
|
| + } else {
|
| + // Raw VP8/VP8L bitstream (no header).
|
| + *is_lossless = VP8LCheckSignature(data, *data_size);
|
| + *chunk_size = *data_size;
|
| }
|
| +
|
| return VP8_STATUS_OK;
|
| }
|
|
|
| -VP8StatusCode WebPParseHeaders(const uint8_t** data, uint32_t* data_size,
|
| - uint32_t* vp8_size, uint32_t* bytes_skipped) {
|
| - const uint8_t* buf;
|
| - uint32_t buf_size;
|
| - uint32_t riff_size;
|
| - uint32_t vp8_size_tmp;
|
| - uint32_t optional_data_size;
|
| - uint32_t vp8x_skip_size;
|
| - uint32_t vp8_skip_size;
|
| - VP8StatusCode status;
|
| -
|
| - assert(data);
|
| - assert(data_size);
|
| - assert(vp8_size);
|
| - assert(bytes_skipped);
|
| -
|
| - buf = *data;
|
| - buf_size = *data_size;
|
| +//------------------------------------------------------------------------------
|
|
|
| - *vp8_size = 0;
|
| - *bytes_skipped = 0;
|
| +// Fetch '*width', '*height', '*has_alpha' and fill out 'headers' based on
|
| +// 'data'. All the output parameters may be NULL. If 'headers' is NULL only the
|
| +// minimal amount will be read to fetch the remaining parameters.
|
| +// If 'headers' is non-NULL this function will attempt to locate both alpha
|
| +// data (with or without a VP8X chunk) and the bitstream chunk (VP8/VP8L).
|
| +// Note: The following chunk sequences (before the raw VP8/VP8L data) are
|
| +// considered valid by this function:
|
| +// RIFF + VP8(L)
|
| +// RIFF + VP8X + (optional chunks) + VP8(L)
|
| +// ALPH + VP8 <-- Not a valid WebP format: only allowed for internal purpose.
|
| +// VP8(L) <-- Not a valid WebP format: only allowed for internal purpose.
|
| +static VP8StatusCode ParseHeadersInternal(const uint8_t* data,
|
| + size_t data_size,
|
| + int* const width,
|
| + int* const height,
|
| + int* const has_alpha,
|
| + WebPHeaderStructure* const headers) {
|
| + int found_riff = 0;
|
| + int found_vp8x = 0;
|
| + VP8StatusCode status;
|
| + WebPHeaderStructure hdrs;
|
|
|
| - if (buf == NULL || buf_size < RIFF_HEADER_SIZE) {
|
| + if (data == NULL || data_size < RIFF_HEADER_SIZE) {
|
| return VP8_STATUS_NOT_ENOUGH_DATA;
|
| }
|
| + memset(&hdrs, 0, sizeof(hdrs));
|
| + hdrs.data = data;
|
| + hdrs.data_size = data_size;
|
|
|
| // Skip over RIFF header.
|
| - if (WebPParseRIFF(&buf, &buf_size, &riff_size) != VP8_STATUS_OK) {
|
| - return VP8_STATUS_BITSTREAM_ERROR; // Wrong RIFF Header.
|
| + status = ParseRIFF(&data, &data_size, &hdrs.riff_size);
|
| + if (status != VP8_STATUS_OK) {
|
| + return status; // Wrong RIFF header / insufficient data.
|
| }
|
| + found_riff = (hdrs.riff_size > 0);
|
|
|
| - // Skip over VP8X header.
|
| - status = WebPParseVP8X(&buf, &buf_size, &vp8x_skip_size, NULL, NULL, NULL);
|
| - if (status != VP8_STATUS_OK) {
|
| - return status; // Wrong VP8X Chunk / Insufficient data.
|
| + // Skip over VP8X.
|
| + {
|
| + uint32_t flags = 0;
|
| + status = ParseVP8X(&data, &data_size, &found_vp8x, width, height, &flags);
|
| + if (status != VP8_STATUS_OK) {
|
| + return status; // Wrong VP8X / insufficient data.
|
| + }
|
| + if (!found_riff && found_vp8x) {
|
| + // Note: This restriction may be removed in the future, if it becomes
|
| + // necessary to send VP8X chunk to the decoder.
|
| + return VP8_STATUS_BITSTREAM_ERROR;
|
| + }
|
| + if (has_alpha != NULL) *has_alpha = !!(flags & ALPHA_FLAG_BIT);
|
| + if (found_vp8x && headers == NULL) {
|
| + return VP8_STATUS_OK; // Return features from VP8X header.
|
| + }
|
| }
|
| - if (vp8x_skip_size > 0) {
|
| - // Skip over optional chunks.
|
| - status = WebPParseOptionalChunks(&buf, &buf_size, riff_size,
|
| - &optional_data_size);
|
| +
|
| + if (data_size < TAG_SIZE) return VP8_STATUS_NOT_ENOUGH_DATA;
|
| +
|
| + // Skip over optional chunks if data started with "RIFF + VP8X" or "ALPH".
|
| + if ((found_riff && found_vp8x) ||
|
| + (!found_riff && !found_vp8x && !memcmp(data, "ALPH", TAG_SIZE))) {
|
| + status = ParseOptionalChunks(&data, &data_size, hdrs.riff_size,
|
| + &hdrs.alpha_data, &hdrs.alpha_data_size);
|
| if (status != VP8_STATUS_OK) {
|
| - return status; // Found an invalid chunk size / Insufficient data.
|
| + return status; // Found an invalid chunk size / insufficient data.
|
| }
|
| }
|
|
|
| - // Skip over VP8 chunk header.
|
| - status = WebPParseVP8Header(&buf, &buf_size, riff_size, &vp8_skip_size,
|
| - &vp8_size_tmp);
|
| + // Skip over VP8/VP8L header.
|
| + status = ParseVP8Header(&data, &data_size, hdrs.riff_size,
|
| + &hdrs.compressed_size, &hdrs.is_lossless);
|
| if (status != VP8_STATUS_OK) {
|
| - return status; // Invalid VP8 header / Insufficient data.
|
| + return status; // Wrong VP8/VP8L chunk-header / insufficient data.
|
| }
|
| - if (vp8_skip_size > 0) {
|
| - *vp8_size = vp8_size_tmp;
|
| + if (hdrs.compressed_size > MAX_CHUNK_PAYLOAD) {
|
| + return VP8_STATUS_BITSTREAM_ERROR;
|
| }
|
|
|
| - *bytes_skipped = buf - *data;
|
| - assert(*bytes_skipped == *data_size - buf_size);
|
| - *data = buf;
|
| - *data_size = buf_size;
|
| - return VP8_STATUS_OK;
|
| + if (!hdrs.is_lossless) {
|
| + if (data_size < VP8_FRAME_HEADER_SIZE) {
|
| + return VP8_STATUS_NOT_ENOUGH_DATA;
|
| + }
|
| + // Validates raw VP8 data.
|
| + if (!VP8GetInfo(data, data_size,
|
| + (uint32_t)hdrs.compressed_size, width, height)) {
|
| + return VP8_STATUS_BITSTREAM_ERROR;
|
| + }
|
| + } else {
|
| + if (data_size < VP8L_FRAME_HEADER_SIZE) {
|
| + return VP8_STATUS_NOT_ENOUGH_DATA;
|
| + }
|
| + // Validates raw VP8L data.
|
| + if (!VP8LGetInfo(data, data_size, width, height, has_alpha)) {
|
| + return VP8_STATUS_BITSTREAM_ERROR;
|
| + }
|
| + }
|
| +
|
| + if (has_alpha != NULL) {
|
| + // If the data did not contain a VP8X/VP8L chunk the only definitive way
|
| + // to set this is by looking for alpha data (from an ALPH chunk).
|
| + *has_alpha |= (hdrs.alpha_data != NULL);
|
| + }
|
| + if (headers != NULL) {
|
| + *headers = hdrs;
|
| + headers->offset = data - headers->data;
|
| + assert((uint64_t)(data - headers->data) < MAX_CHUNK_PAYLOAD);
|
| + assert(headers->offset == headers->data_size - data_size);
|
| + }
|
| + return VP8_STATUS_OK; // Return features from VP8 header.
|
| +}
|
| +
|
| +VP8StatusCode WebPParseHeaders(WebPHeaderStructure* const headers) {
|
| + assert(headers != NULL);
|
| + // fill out headers, ignore width/height/has_alpha.
|
| + return ParseHeadersInternal(headers->data, headers->data_size,
|
| + NULL, NULL, NULL, headers);
|
| }
|
|
|
| //------------------------------------------------------------------------------
|
| @@ -259,43 +383,72 @@ void WebPResetDecParams(WebPDecParams* const params) {
|
| // "Into" decoding variants
|
|
|
| // Main flow
|
| -static VP8StatusCode DecodeInto(const uint8_t* data, uint32_t data_size,
|
| +static VP8StatusCode DecodeInto(const uint8_t* const data, size_t data_size,
|
| WebPDecParams* const params) {
|
| - VP8Decoder* dec = VP8New();
|
| - VP8StatusCode status = VP8_STATUS_OK;
|
| + VP8StatusCode status;
|
| VP8Io io;
|
| + WebPHeaderStructure headers;
|
|
|
| - assert(params);
|
| - if (dec == NULL) {
|
| - return VP8_STATUS_INVALID_PARAM;
|
| + headers.data = data;
|
| + headers.data_size = data_size;
|
| + status = WebPParseHeaders(&headers); // Process Pre-VP8 chunks.
|
| + if (status != VP8_STATUS_OK) {
|
| + return status;
|
| }
|
|
|
| + assert(params != NULL);
|
| VP8InitIo(&io);
|
| - io.data = data;
|
| - io.data_size = data_size;
|
| + io.data = headers.data + headers.offset;
|
| + io.data_size = headers.data_size - headers.offset;
|
| WebPInitCustomIo(params, &io); // Plug the I/O functions.
|
|
|
| + if (!headers.is_lossless) {
|
| + VP8Decoder* const dec = VP8New();
|
| + if (dec == NULL) {
|
| + return VP8_STATUS_OUT_OF_MEMORY;
|
| + }
|
| #ifdef WEBP_USE_THREAD
|
| - dec->use_threads_ = params->options && (params->options->use_threads > 0);
|
| + dec->use_threads_ = params->options && (params->options->use_threads > 0);
|
| #else
|
| - dec->use_threads_ = 0;
|
| + dec->use_threads_ = 0;
|
| #endif
|
| + dec->alpha_data_ = headers.alpha_data;
|
| + dec->alpha_data_size_ = headers.alpha_data_size;
|
|
|
| - // Decode bitstream header, update io->width/io->height.
|
| - if (!VP8GetHeaders(dec, &io)) {
|
| - status = VP8_STATUS_BITSTREAM_ERROR;
|
| + // Decode bitstream header, update io->width/io->height.
|
| + if (!VP8GetHeaders(dec, &io)) {
|
| + status = dec->status_; // An error occurred. Grab error status.
|
| + } else {
|
| + // Allocate/check output buffers.
|
| + status = WebPAllocateDecBuffer(io.width, io.height, params->options,
|
| + params->output);
|
| + if (status == VP8_STATUS_OK) { // Decode
|
| + if (!VP8Decode(dec, &io)) {
|
| + status = dec->status_;
|
| + }
|
| + }
|
| + }
|
| + VP8Delete(dec);
|
| } else {
|
| - // Allocate/check output buffers.
|
| - status = WebPAllocateDecBuffer(io.width, io.height, params->options,
|
| - params->output);
|
| - if (status == VP8_STATUS_OK) {
|
| - // Decode
|
| - if (!VP8Decode(dec, &io)) {
|
| - status = dec->status_;
|
| + VP8LDecoder* const dec = VP8LNew();
|
| + if (dec == NULL) {
|
| + return VP8_STATUS_OUT_OF_MEMORY;
|
| + }
|
| + if (!VP8LDecodeHeader(dec, &io)) {
|
| + status = dec->status_; // An error occurred. Grab error status.
|
| + } else {
|
| + // Allocate/check output buffers.
|
| + status = WebPAllocateDecBuffer(io.width, io.height, params->options,
|
| + params->output);
|
| + if (status == VP8_STATUS_OK) { // Decode
|
| + if (!VP8LDecodeImage(dec)) {
|
| + status = dec->status_;
|
| + }
|
| }
|
| }
|
| + VP8LDelete(dec);
|
| }
|
| - VP8Delete(dec);
|
| +
|
| if (status != VP8_STATUS_OK) {
|
| WebPFreeDecBuffer(params->output);
|
| }
|
| @@ -304,8 +457,10 @@ static VP8StatusCode DecodeInto(const uint8_t* data, uint32_t data_size,
|
|
|
| // Helpers
|
| static uint8_t* DecodeIntoRGBABuffer(WEBP_CSP_MODE colorspace,
|
| - const uint8_t* data, uint32_t data_size,
|
| - uint8_t* rgba, int stride, int size) {
|
| + const uint8_t* const data,
|
| + size_t data_size,
|
| + uint8_t* const rgba,
|
| + int stride, size_t size) {
|
| WebPDecParams params;
|
| WebPDecBuffer buf;
|
| if (rgba == NULL) {
|
| @@ -325,35 +480,35 @@ static uint8_t* DecodeIntoRGBABuffer(WEBP_CSP_MODE colorspace,
|
| return rgba;
|
| }
|
|
|
| -uint8_t* WebPDecodeRGBInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* output, int size, int stride) {
|
| +uint8_t* WebPDecodeRGBInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* output, size_t size, int stride) {
|
| return DecodeIntoRGBABuffer(MODE_RGB, data, data_size, output, stride, size);
|
| }
|
|
|
| -uint8_t* WebPDecodeRGBAInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* output, int size, int stride) {
|
| +uint8_t* WebPDecodeRGBAInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* output, size_t size, int stride) {
|
| return DecodeIntoRGBABuffer(MODE_RGBA, data, data_size, output, stride, size);
|
| }
|
|
|
| -uint8_t* WebPDecodeARGBInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* output, int size, int stride) {
|
| +uint8_t* WebPDecodeARGBInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* output, size_t size, int stride) {
|
| return DecodeIntoRGBABuffer(MODE_ARGB, data, data_size, output, stride, size);
|
| }
|
|
|
| -uint8_t* WebPDecodeBGRInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* output, int size, int stride) {
|
| +uint8_t* WebPDecodeBGRInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* output, size_t size, int stride) {
|
| return DecodeIntoRGBABuffer(MODE_BGR, data, data_size, output, stride, size);
|
| }
|
|
|
| -uint8_t* WebPDecodeBGRAInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* output, int size, int stride) {
|
| +uint8_t* WebPDecodeBGRAInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* output, size_t size, int stride) {
|
| return DecodeIntoRGBABuffer(MODE_BGRA, data, data_size, output, stride, size);
|
| }
|
|
|
| -uint8_t* WebPDecodeYUVInto(const uint8_t* data, uint32_t data_size,
|
| - uint8_t* luma, int luma_size, int luma_stride,
|
| - uint8_t* u, int u_size, int u_stride,
|
| - uint8_t* v, int v_size, int v_stride) {
|
| +uint8_t* WebPDecodeYUVInto(const uint8_t* data, size_t data_size,
|
| + uint8_t* luma, size_t luma_size, int luma_stride,
|
| + uint8_t* u, size_t u_size, int u_stride,
|
| + uint8_t* v, size_t v_size, int v_stride) {
|
| WebPDecParams params;
|
| WebPDecBuffer output;
|
| if (luma == NULL) return NULL;
|
| @@ -379,9 +534,9 @@ uint8_t* WebPDecodeYUVInto(const uint8_t* data, uint32_t data_size,
|
|
|
| //------------------------------------------------------------------------------
|
|
|
| -static uint8_t* Decode(WEBP_CSP_MODE mode, const uint8_t* data,
|
| - uint32_t data_size, int* width, int* height,
|
| - WebPDecBuffer* keep_info) {
|
| +static uint8_t* Decode(WEBP_CSP_MODE mode, const uint8_t* const data,
|
| + size_t data_size, int* const width, int* const height,
|
| + WebPDecBuffer* const keep_info) {
|
| WebPDecParams params;
|
| WebPDecBuffer output;
|
|
|
| @@ -394,53 +549,53 @@ static uint8_t* Decode(WEBP_CSP_MODE mode, const uint8_t* data,
|
| if (!WebPGetInfo(data, data_size, &output.width, &output.height)) {
|
| return NULL;
|
| }
|
| - if (width) *width = output.width;
|
| - if (height) *height = output.height;
|
| + if (width != NULL) *width = output.width;
|
| + if (height != NULL) *height = output.height;
|
|
|
| // Decode
|
| if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) {
|
| return NULL;
|
| }
|
| - if (keep_info) { // keep track of the side-info
|
| + if (keep_info != NULL) { // keep track of the side-info
|
| WebPCopyDecBuffer(&output, keep_info);
|
| }
|
| // return decoded samples (don't clear 'output'!)
|
| - return (mode >= MODE_YUV) ? output.u.YUVA.y : output.u.RGBA.rgba;
|
| + return WebPIsRGBMode(mode) ? output.u.RGBA.rgba : output.u.YUVA.y;
|
| }
|
|
|
| -uint8_t* WebPDecodeRGB(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeRGB(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| return Decode(MODE_RGB, data, data_size, width, height, NULL);
|
| }
|
|
|
| -uint8_t* WebPDecodeRGBA(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeRGBA(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| return Decode(MODE_RGBA, data, data_size, width, height, NULL);
|
| }
|
|
|
| -uint8_t* WebPDecodeARGB(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeARGB(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| return Decode(MODE_ARGB, data, data_size, width, height, NULL);
|
| }
|
|
|
| -uint8_t* WebPDecodeBGR(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeBGR(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| return Decode(MODE_BGR, data, data_size, width, height, NULL);
|
| }
|
|
|
| -uint8_t* WebPDecodeBGRA(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeBGRA(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| return Decode(MODE_BGRA, data, data_size, width, height, NULL);
|
| }
|
|
|
| -uint8_t* WebPDecodeYUV(const uint8_t* data, uint32_t data_size,
|
| +uint8_t* WebPDecodeYUV(const uint8_t* data, size_t data_size,
|
| int* width, int* height, uint8_t** u, uint8_t** v,
|
| int* stride, int* uv_stride) {
|
| WebPDecBuffer output; // only to preserve the side-infos
|
| uint8_t* const out = Decode(MODE_YUV, data, data_size,
|
| width, height, &output);
|
|
|
| - if (out) {
|
| + if (out != NULL) {
|
| const WebPYUVABuffer* const buf = &output.u.YUVA;
|
| *u = buf->u;
|
| *v = buf->v;
|
| @@ -452,69 +607,28 @@ uint8_t* WebPDecodeYUV(const uint8_t* data, uint32_t data_size,
|
| }
|
|
|
| static void DefaultFeatures(WebPBitstreamFeatures* const features) {
|
| - assert(features);
|
| + assert(features != NULL);
|
| memset(features, 0, sizeof(*features));
|
| features->bitstream_version = 0;
|
| }
|
|
|
| -static VP8StatusCode GetFeatures(const uint8_t* data, uint32_t data_size,
|
| +static VP8StatusCode GetFeatures(const uint8_t* const data, size_t data_size,
|
| WebPBitstreamFeatures* const features) {
|
| - uint32_t vp8_chunk_size = 0;
|
| - uint32_t riff_size = 0;
|
| - uint32_t flags = 0;
|
| - uint32_t vp8x_skip_size = 0;
|
| - uint32_t vp8_skip_size = 0;
|
| - VP8StatusCode status;
|
| -
|
| - if (features == NULL) {
|
| + if (features == NULL || data == NULL) {
|
| return VP8_STATUS_INVALID_PARAM;
|
| }
|
| DefaultFeatures(features);
|
|
|
| - if (data == NULL) {
|
| - return VP8_STATUS_INVALID_PARAM;
|
| - }
|
| -
|
| - // Skip over RIFF header.
|
| - status = WebPParseRIFF(&data, &data_size, &riff_size);
|
| - if (status != VP8_STATUS_OK) {
|
| - return status; // Wrong RIFF Header / Insufficient data.
|
| - }
|
| -
|
| - // Skip over VP8X.
|
| - status = WebPParseVP8X(&data, &data_size, &vp8x_skip_size, &features->width,
|
| - &features->height, &flags);
|
| - if (status != VP8_STATUS_OK) {
|
| - return status; // Wrong VP8X / insufficient data.
|
| -
|
| - }
|
| - if (vp8x_skip_size > 0) {
|
| - return VP8_STATUS_OK; // Return features from VP8X header.
|
| - }
|
| -
|
| - // Skip over VP8 header.
|
| - status = WebPParseVP8Header(&data, &data_size, riff_size, &vp8_skip_size,
|
| - &vp8_chunk_size);
|
| - if (status != VP8_STATUS_OK) {
|
| - return status; // Wrong VP8 Chunk-header / insufficient data.
|
| - }
|
| - if (vp8_skip_size == 0) {
|
| - vp8_chunk_size = data_size; // No VP8 chunk wrapper over raw VP8 data.
|
| - }
|
| -
|
| - // Validates raw VP8 data.
|
| - if (!VP8GetInfo(data, data_size, vp8_chunk_size,
|
| - &features->width, &features->height, &features->has_alpha)) {
|
| - return VP8_STATUS_BITSTREAM_ERROR;
|
| - }
|
| -
|
| - return VP8_STATUS_OK; // Return features from VP8 header.
|
| + // Only parse enough of the data to retrieve width/height/has_alpha.
|
| + return ParseHeadersInternal(data, data_size,
|
| + &features->width, &features->height,
|
| + &features->has_alpha, NULL);
|
| }
|
|
|
| //------------------------------------------------------------------------------
|
| // WebPGetInfo()
|
|
|
| -int WebPGetInfo(const uint8_t* data, uint32_t data_size,
|
| +int WebPGetInfo(const uint8_t* data, size_t data_size,
|
| int* width, int* height) {
|
| WebPBitstreamFeatures features;
|
|
|
| @@ -522,10 +636,10 @@ int WebPGetInfo(const uint8_t* data, uint32_t data_size,
|
| return 0;
|
| }
|
|
|
| - if (width) {
|
| + if (width != NULL) {
|
| *width = features.width;
|
| }
|
| - if (height) {
|
| + if (height != NULL) {
|
| *height = features.height;
|
| }
|
|
|
| @@ -535,9 +649,9 @@ int WebPGetInfo(const uint8_t* data, uint32_t data_size,
|
| //------------------------------------------------------------------------------
|
| // Advance decoding API
|
|
|
| -int WebPInitDecoderConfigInternal(WebPDecoderConfig* const config,
|
| +int WebPInitDecoderConfigInternal(WebPDecoderConfig* config,
|
| int version) {
|
| - if (version != WEBP_DECODER_ABI_VERSION) {
|
| + if (WEBP_ABI_IS_INCOMPATIBLE(version, WEBP_DECODER_ABI_VERSION)) {
|
| return 0; // version mismatch
|
| }
|
| if (config == NULL) {
|
| @@ -549,11 +663,11 @@ int WebPInitDecoderConfigInternal(WebPDecoderConfig* const config,
|
| return 1;
|
| }
|
|
|
| -VP8StatusCode WebPGetFeaturesInternal(const uint8_t* data, uint32_t data_size,
|
| - WebPBitstreamFeatures* const features,
|
| +VP8StatusCode WebPGetFeaturesInternal(const uint8_t* data, size_t data_size,
|
| + WebPBitstreamFeatures* features,
|
| int version) {
|
| VP8StatusCode status;
|
| - if (version != WEBP_DECODER_ABI_VERSION) {
|
| + if (WEBP_ABI_IS_INCOMPATIBLE(version, WEBP_DECODER_ABI_VERSION)) {
|
| return VP8_STATUS_INVALID_PARAM; // version mismatch
|
| }
|
| if (features == NULL) {
|
| @@ -562,24 +676,24 @@ VP8StatusCode WebPGetFeaturesInternal(const uint8_t* data, uint32_t data_size,
|
|
|
| status = GetFeatures(data, data_size, features);
|
| if (status == VP8_STATUS_NOT_ENOUGH_DATA) {
|
| - return VP8_STATUS_BITSTREAM_ERROR; // Not enough data treated as error.
|
| + return VP8_STATUS_BITSTREAM_ERROR; // Not-enough-data treated as error.
|
| }
|
| return status;
|
| }
|
|
|
| -VP8StatusCode WebPDecode(const uint8_t* data, uint32_t data_size,
|
| - WebPDecoderConfig* const config) {
|
| +VP8StatusCode WebPDecode(const uint8_t* data, size_t data_size,
|
| + WebPDecoderConfig* config) {
|
| WebPDecParams params;
|
| VP8StatusCode status;
|
|
|
| - if (!config) {
|
| + if (config == NULL) {
|
| return VP8_STATUS_INVALID_PARAM;
|
| }
|
|
|
| status = GetFeatures(data, data_size, &config->input);
|
| if (status != VP8_STATUS_OK) {
|
| if (status == VP8_STATUS_NOT_ENOUGH_DATA) {
|
| - return VP8_STATUS_BITSTREAM_ERROR; // Not enough data treated as error.
|
| + return VP8_STATUS_BITSTREAM_ERROR; // Not-enough-data treated as error.
|
| }
|
| return status;
|
| }
|
| @@ -592,6 +706,66 @@ VP8StatusCode WebPDecode(const uint8_t* data, uint32_t data_size,
|
| return status;
|
| }
|
|
|
| +//------------------------------------------------------------------------------
|
| +// Cropping and rescaling.
|
| +
|
| +int WebPIoInitFromOptions(const WebPDecoderOptions* const options,
|
| + VP8Io* const io, WEBP_CSP_MODE src_colorspace) {
|
| + const int W = io->width;
|
| + const int H = io->height;
|
| + int x = 0, y = 0, w = W, h = H;
|
| +
|
| + // Cropping
|
| + io->use_cropping = (options != NULL) && (options->use_cropping > 0);
|
| + if (io->use_cropping) {
|
| + w = options->crop_width;
|
| + h = options->crop_height;
|
| + x = options->crop_left;
|
| + y = options->crop_top;
|
| + if (!WebPIsRGBMode(src_colorspace)) { // only snap for YUV420 or YUV422
|
| + x &= ~1;
|
| + y &= ~1; // TODO(later): only for YUV420, not YUV422.
|
| + }
|
| + if (x < 0 || y < 0 || w <= 0 || h <= 0 || x + w > W || y + h > H) {
|
| + return 0; // out of frame boundary error
|
| + }
|
| + }
|
| + io->crop_left = x;
|
| + io->crop_top = y;
|
| + io->crop_right = x + w;
|
| + io->crop_bottom = y + h;
|
| + io->mb_w = w;
|
| + io->mb_h = h;
|
| +
|
| + // Scaling
|
| + io->use_scaling = (options != NULL) && (options->use_scaling > 0);
|
| + if (io->use_scaling) {
|
| + if (options->scaled_width <= 0 || options->scaled_height <= 0) {
|
| + return 0;
|
| + }
|
| + io->scaled_width = options->scaled_width;
|
| + io->scaled_height = options->scaled_height;
|
| + }
|
| +
|
| + // Filter
|
| + io->bypass_filtering = options && options->bypass_filtering;
|
| +
|
| + // Fancy upsampler
|
| +#ifdef FANCY_UPSAMPLING
|
| + io->fancy_upsampling = (options == NULL) || (!options->no_fancy_upsampling);
|
| +#endif
|
| +
|
| + if (io->use_scaling) {
|
| + // disable filter (only for large downscaling ratio).
|
| + io->bypass_filtering = (io->scaled_width < W * 3 / 4) &&
|
| + (io->scaled_height < H * 3 / 4);
|
| + io->fancy_upsampling = 0;
|
| + }
|
| + return 1;
|
| +}
|
| +
|
| +//------------------------------------------------------------------------------
|
| +
|
| #if defined(__cplusplus) || defined(c_plusplus)
|
| } // extern "C"
|
| #endif
|
|
|