| OLD | NEW |
| 1 // Copyright 2010 Google Inc. | 1 // Copyright 2010 Google Inc. All Rights Reserved. |
| 2 // | 2 // |
| 3 // This code is licensed under the same terms as WebM: | 3 // This code is licensed under the same terms as WebM: |
| 4 // Software License Agreement: http://www.webmproject.org/license/software/ | 4 // Software License Agreement: http://www.webmproject.org/license/software/ |
| 5 // Additional IP Rights Grant: http://www.webmproject.org/license/additional/ | 5 // Additional IP Rights Grant: http://www.webmproject.org/license/additional/ |
| 6 // ----------------------------------------------------------------------------- | 6 // ----------------------------------------------------------------------------- |
| 7 // | 7 // |
| 8 // Main decoding functions for WEBP images. | 8 // Main decoding functions for WEBP images. |
| 9 // | 9 // |
| 10 // Author: Skal (pascal.massimino@gmail.com) | 10 // Author: Skal (pascal.massimino@gmail.com) |
| 11 | 11 |
| 12 #include <stdlib.h> | 12 #include <stdlib.h> |
| 13 #include "vp8i.h" | 13 |
| 14 #include "webpi.h" | 14 #include "./vp8i.h" |
| 15 #include "./vp8li.h" |
| 16 #include "./webpi.h" |
| 17 #include "../webp/format_constants.h" |
| 15 | 18 |
| 16 #if defined(__cplusplus) || defined(c_plusplus) | 19 #if defined(__cplusplus) || defined(c_plusplus) |
| 17 extern "C" { | 20 extern "C" { |
| 18 #endif | 21 #endif |
| 19 | 22 |
| 20 //------------------------------------------------------------------------------ | 23 //------------------------------------------------------------------------------ |
| 21 // RIFF layout is: | 24 // RIFF layout is: |
| 22 // Offset tag | 25 // Offset tag |
| 23 // 0...3 "RIFF" 4-byte tag | 26 // 0...3 "RIFF" 4-byte tag |
| 24 // 4...7 size of image data (including metadata) starting at offset 8 | 27 // 4...7 size of image data (including metadata) starting at offset 8 |
| 25 // 8...11 "WEBP" our form-type signature | 28 // 8...11 "WEBP" our form-type signature |
| 26 // The RIFF container (12 bytes) is followed by appropriate chunks: | 29 // The RIFF container (12 bytes) is followed by appropriate chunks: |
| 27 // 12..15 "VP8 ": 4-bytes tags, describing the raw video format used | 30 // 12..15 "VP8 ": 4-bytes tags, signaling the use of VP8 video format |
| 28 // 16..19 size of the raw VP8 image data, starting at offset 20 | 31 // 16..19 size of the raw VP8 image data, starting at offset 20 |
| 29 // 20.... the VP8 bytes | 32 // 20.... the VP8 bytes |
| 30 // Or, | 33 // Or, |
| 34 // 12..15 "VP8L": 4-bytes tags, signaling the use of VP8L lossless format |
| 35 // 16..19 size of the raw VP8L image data, starting at offset 20 |
| 36 // 20.... the VP8L bytes |
| 37 // Or, |
| 31 // 12..15 "VP8X": 4-bytes tags, describing the extended-VP8 chunk. | 38 // 12..15 "VP8X": 4-bytes tags, describing the extended-VP8 chunk. |
| 32 // 16..19 size of the VP8X chunk starting at offset 20. | 39 // 16..19 size of the VP8X chunk starting at offset 20. |
| 33 // 20..23 VP8X flags bit-map corresponding to the chunk-types present. | 40 // 20..23 VP8X flags bit-map corresponding to the chunk-types present. |
| 34 // 24..27 Width of the Canvas Image. | 41 // 24..26 Width of the Canvas Image. |
| 35 // 28..31 Height of the Canvas Image. | 42 // 27..29 Height of the Canvas Image. |
| 36 // There can be extra chunks after the "VP8X" chunk (ICCP, TILE, FRM, VP8, | 43 // There can be extra chunks after the "VP8X" chunk (ICCP, TILE, FRM, VP8, |
| 37 // META ...) | 44 // META ...) |
| 38 // All 32-bits sizes are in little-endian order. | 45 // All sizes are in little-endian order. |
| 39 // Note: chunk data must be padded to multiple of 2 in size | 46 // Note: chunk data size must be padded to multiple of 2 when written. |
| 40 | 47 |
| 41 static inline uint32_t get_le32(const uint8_t* const data) { | 48 static WEBP_INLINE uint32_t get_le24(const uint8_t* const data) { |
| 42 return data[0] | (data[1] << 8) | (data[2] << 16) | (data[3] << 24); | 49 return data[0] | (data[1] << 8) | (data[2] << 16); |
| 43 } | 50 } |
| 44 | 51 |
| 45 VP8StatusCode WebPParseRIFF(const uint8_t** data, uint32_t* data_size, | 52 static WEBP_INLINE uint32_t get_le32(const uint8_t* const data) { |
| 46 uint32_t* riff_size) { | 53 return (uint32_t)get_le24(data) | (data[3] << 24); |
| 47 assert(data); | 54 } |
| 48 assert(data_size); | |
| 49 assert(riff_size); | |
| 50 | 55 |
| 51 if (*data_size >= RIFF_HEADER_SIZE && | 56 // Validates the RIFF container (if detected) and skips over it. |
| 52 !memcmp(*data, "RIFF", TAG_SIZE)) { | 57 // If a RIFF container is detected, |
| 58 // Returns VP8_STATUS_BITSTREAM_ERROR for invalid header, and |
| 59 // VP8_STATUS_OK otherwise. |
| 60 // In case there are not enough bytes (partial RIFF container), return 0 for |
| 61 // *riff_size. Else return the RIFF size extracted from the header. |
| 62 static VP8StatusCode ParseRIFF(const uint8_t** const data, |
| 63 size_t* const data_size, |
| 64 size_t* const riff_size) { |
| 65 assert(data != NULL); |
| 66 assert(data_size != NULL); |
| 67 assert(riff_size != NULL); |
| 68 |
| 69 *riff_size = 0; // Default: no RIFF present. |
| 70 if (*data_size >= RIFF_HEADER_SIZE && !memcmp(*data, "RIFF", TAG_SIZE)) { |
| 53 if (memcmp(*data + 8, "WEBP", TAG_SIZE)) { | 71 if (memcmp(*data + 8, "WEBP", TAG_SIZE)) { |
| 54 return VP8_STATUS_BITSTREAM_ERROR; // Wrong image file signature. | 72 return VP8_STATUS_BITSTREAM_ERROR; // Wrong image file signature. |
| 55 } else { | 73 } else { |
| 56 *riff_size = get_le32(*data + TAG_SIZE); | 74 const uint32_t size = get_le32(*data + TAG_SIZE); |
| 57 // Check that we have at least one chunk (i.e "WEBP" + "VP8?nnnn"). | 75 // Check that we have at least one chunk (i.e "WEBP" + "VP8?nnnn"). |
| 58 if (*riff_size < TAG_SIZE + CHUNK_HEADER_SIZE) { | 76 if (size < TAG_SIZE + CHUNK_HEADER_SIZE) { |
| 59 return VP8_STATUS_BITSTREAM_ERROR; | 77 return VP8_STATUS_BITSTREAM_ERROR; |
| 60 } | 78 } |
| 61 // We have a RIFF container. Skip it. | 79 // We have a RIFF container. Skip it. |
| 80 *riff_size = size; |
| 62 *data += RIFF_HEADER_SIZE; | 81 *data += RIFF_HEADER_SIZE; |
| 63 *data_size -= RIFF_HEADER_SIZE; | 82 *data_size -= RIFF_HEADER_SIZE; |
| 64 } | 83 } |
| 65 } else { | |
| 66 *riff_size = 0; // Did not get full RIFF Header. | |
| 67 } | 84 } |
| 68 return VP8_STATUS_OK; | 85 return VP8_STATUS_OK; |
| 69 } | 86 } |
| 70 | 87 |
| 71 VP8StatusCode WebPParseVP8X(const uint8_t** data, uint32_t* data_size, | 88 // Validates the VP8X header and skips over it. |
| 72 uint32_t* bytes_skipped, | 89 // Returns VP8_STATUS_BITSTREAM_ERROR for invalid VP8X header, |
| 73 int* width, int* height, uint32_t* flags) { | 90 // VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and |
| 74 assert(data); | 91 // VP8_STATUS_OK otherwise. |
| 75 assert(data_size); | 92 // If a VP8X chunk is found, found_vp8x is set to true and *width_ptr, |
| 76 assert(bytes_skipped); | 93 // *height_ptr and *flags_ptr are set to the corresponding values extracted |
| 94 // from the VP8X chunk. |
| 95 static VP8StatusCode ParseVP8X(const uint8_t** const data, |
| 96 size_t* const data_size, |
| 97 int* const found_vp8x, |
| 98 int* const width_ptr, int* const height_ptr, |
| 99 uint32_t* const flags_ptr) { |
| 100 const uint32_t vp8x_size = CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE; |
| 101 assert(data != NULL); |
| 102 assert(data_size != NULL); |
| 103 assert(found_vp8x != NULL); |
| 77 | 104 |
| 78 *bytes_skipped = 0; | 105 *found_vp8x = 0; |
| 79 | 106 |
| 80 if (*data_size < CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE) { | 107 if (*data_size < CHUNK_HEADER_SIZE) { |
| 81 return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data. | 108 return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data. |
| 82 } | 109 } |
| 83 | 110 |
| 84 if (!memcmp(*data, "VP8X", TAG_SIZE)) { | 111 if (!memcmp(*data, "VP8X", TAG_SIZE)) { |
| 112 int width, height; |
| 113 uint32_t flags; |
| 85 const uint32_t chunk_size = get_le32(*data + TAG_SIZE); | 114 const uint32_t chunk_size = get_le32(*data + TAG_SIZE); |
| 86 if (chunk_size != VP8X_CHUNK_SIZE) { | 115 if (chunk_size != VP8X_CHUNK_SIZE) { |
| 87 return VP8_STATUS_BITSTREAM_ERROR; // Wrong chunk size. | 116 return VP8_STATUS_BITSTREAM_ERROR; // Wrong chunk size. |
| 88 } | 117 } |
| 89 if (flags) { | 118 |
| 90 *flags = get_le32(*data + 8); | 119 // Verify if enough data is available to validate the VP8X chunk. |
| 120 if (*data_size < vp8x_size) { |
| 121 return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data. |
| 91 } | 122 } |
| 92 if (width) { | 123 flags = get_le32(*data + 8); |
| 93 *width = get_le32(*data + 12); | 124 width = 1 + get_le24(*data + 12); |
| 125 height = 1 + get_le24(*data + 15); |
| 126 if (width * (uint64_t)height >= MAX_IMAGE_AREA) { |
| 127 return VP8_STATUS_BITSTREAM_ERROR; // image is too large |
| 94 } | 128 } |
| 95 if (height) { | 129 |
| 96 *height = get_le32(*data + 16); | 130 if (flags_ptr != NULL) *flags_ptr = flags; |
| 97 } | 131 if (width_ptr != NULL) *width_ptr = width; |
| 98 // We have consumed 20 bytes from VP8X. Skip them. | 132 if (height_ptr != NULL) *height_ptr = height; |
| 99 *bytes_skipped = CHUNK_HEADER_SIZE + VP8X_CHUNK_SIZE; | 133 // Skip over VP8X header bytes. |
| 100 *data += *bytes_skipped; | 134 *data += vp8x_size; |
| 101 *data_size -= *bytes_skipped; | 135 *data_size -= vp8x_size; |
| 136 *found_vp8x = 1; |
| 102 } | 137 } |
| 103 return VP8_STATUS_OK; | 138 return VP8_STATUS_OK; |
| 104 } | 139 } |
| 105 | 140 |
| 106 VP8StatusCode WebPParseOptionalChunks(const uint8_t** data, uint32_t* data_size, | 141 // Skips to the next VP8/VP8L chunk header in the data given the size of the |
| 107 uint32_t riff_size, | 142 // RIFF chunk 'riff_size'. |
| 108 uint32_t* bytes_skipped) { | 143 // Returns VP8_STATUS_BITSTREAM_ERROR if any invalid chunk size is encountered, |
| 144 // VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and |
| 145 // VP8_STATUS_OK otherwise. |
| 146 // If an alpha chunk is found, *alpha_data and *alpha_size are set |
| 147 // appropriately. |
| 148 static VP8StatusCode ParseOptionalChunks(const uint8_t** const data, |
| 149 size_t* const data_size, |
| 150 size_t const riff_size, |
| 151 const uint8_t** const alpha_data, |
| 152 size_t* const alpha_size) { |
| 109 const uint8_t* buf; | 153 const uint8_t* buf; |
| 110 uint32_t buf_size; | 154 size_t buf_size; |
| 111 | 155 uint32_t total_size = TAG_SIZE + // "WEBP". |
| 112 assert(data); | 156 CHUNK_HEADER_SIZE + // "VP8Xnnnn". |
| 113 assert(data_size); | 157 VP8X_CHUNK_SIZE; // data. |
| 114 assert(bytes_skipped); | 158 assert(data != NULL); |
| 115 | 159 assert(data_size != NULL); |
| 116 buf = *data; | 160 buf = *data; |
| 117 buf_size = *data_size; | 161 buf_size = *data_size; |
| 118 *bytes_skipped = 0; | 162 |
| 163 assert(alpha_data != NULL); |
| 164 assert(alpha_size != NULL); |
| 165 *alpha_data = NULL; |
| 166 *alpha_size = 0; |
| 119 | 167 |
| 120 while (1) { | 168 while (1) { |
| 121 uint32_t chunk_size; | 169 uint32_t chunk_size; |
| 122 uint32_t cur_skip_size; | 170 uint32_t disk_chunk_size; // chunk_size with padding |
| 123 const uint32_t bytes_skipped_header = TAG_SIZE + // "WEBP". | 171 |
| 124 CHUNK_HEADER_SIZE + // "VP8Xnnnn". | |
| 125 VP8X_CHUNK_SIZE; // Data. | |
| 126 *data = buf; | 172 *data = buf; |
| 127 *data_size = buf_size; | 173 *data_size = buf_size; |
| 128 | 174 |
| 129 if (buf_size < CHUNK_HEADER_SIZE) { // Insufficient data. | 175 if (buf_size < CHUNK_HEADER_SIZE) { // Insufficient data. |
| 130 return VP8_STATUS_NOT_ENOUGH_DATA; | 176 return VP8_STATUS_NOT_ENOUGH_DATA; |
| 131 } | 177 } |
| 132 | 178 |
| 133 chunk_size = get_le32(buf + TAG_SIZE); | 179 chunk_size = get_le32(buf + TAG_SIZE); |
| 134 cur_skip_size = CHUNK_HEADER_SIZE + chunk_size; | 180 // For odd-sized chunk-payload, there's one byte padding at the end. |
| 181 disk_chunk_size = (CHUNK_HEADER_SIZE + chunk_size + 1) & ~1; |
| 182 total_size += disk_chunk_size; |
| 135 | 183 |
| 136 // Check that total bytes skipped along with current chunk size | 184 // Check that total bytes skipped so far does not exceed riff_size. |
| 137 // does not exceed riff_size. | 185 if (riff_size > 0 && (total_size > riff_size)) { |
| 138 if (riff_size > 0 && | 186 return VP8_STATUS_BITSTREAM_ERROR; // Not a valid chunk size. |
| 139 (bytes_skipped_header + *bytes_skipped + cur_skip_size > riff_size)) { | |
| 140 return VP8_STATUS_BITSTREAM_ERROR; // Not a valid chunk size. | |
| 141 } | 187 } |
| 142 | 188 |
| 143 if (buf_size < cur_skip_size) { // Insufficient data. | 189 if (buf_size < disk_chunk_size) { // Insufficient data. |
| 144 return VP8_STATUS_NOT_ENOUGH_DATA; | 190 return VP8_STATUS_NOT_ENOUGH_DATA; |
| 145 } | 191 } |
| 146 | 192 |
| 147 if (!memcmp(buf, "VP8 ", TAG_SIZE)) { // A valid VP8 header. | 193 if (!memcmp(buf, "ALPH", TAG_SIZE)) { // A valid ALPH header. |
| 194 *alpha_data = buf + CHUNK_HEADER_SIZE; |
| 195 *alpha_size = chunk_size; |
| 196 } else if (!memcmp(buf, "VP8 ", TAG_SIZE) || |
| 197 !memcmp(buf, "VP8L", TAG_SIZE)) { // A valid VP8/VP8L header. |
| 148 return VP8_STATUS_OK; // Found. | 198 return VP8_STATUS_OK; // Found. |
| 149 } | 199 } |
| 150 | 200 |
| 151 // We have a full & valid chunk; skip it. | 201 // We have a full and valid chunk; skip it. |
| 152 buf += cur_skip_size; | 202 buf += disk_chunk_size; |
| 153 buf_size -= cur_skip_size; | 203 buf_size -= disk_chunk_size; |
| 154 *bytes_skipped += cur_skip_size; | |
| 155 } | 204 } |
| 156 } | 205 } |
| 157 | 206 |
| 158 VP8StatusCode WebPParseVP8Header(const uint8_t** data, uint32_t* data_size, | 207 // Validates the VP8/VP8L Header ("VP8 nnnn" or "VP8L nnnn") and skips over it. |
| 159 uint32_t riff_size, uint32_t* bytes_skipped, | 208 // Returns VP8_STATUS_BITSTREAM_ERROR for invalid (chunk larger than |
| 160 uint32_t* vp8_chunk_size) { | 209 // riff_size) VP8/VP8L header, |
| 161 assert(data); | 210 // VP8_STATUS_NOT_ENOUGH_DATA in case of insufficient data, and |
| 162 assert(data_size); | 211 // VP8_STATUS_OK otherwise. |
| 163 assert(bytes_skipped); | 212 // If a VP8/VP8L chunk is found, *chunk_size is set to the total number of bytes |
| 164 assert(vp8_chunk_size); | 213 // extracted from the VP8/VP8L chunk header. |
| 165 | 214 // The flag '*is_lossless' is set to 1 in case of VP8L chunk / raw VP8L data. |
| 166 *bytes_skipped = 0; | 215 static VP8StatusCode ParseVP8Header(const uint8_t** const data_ptr, |
| 167 *vp8_chunk_size = 0; | 216 size_t* const data_size, |
| 217 size_t riff_size, |
| 218 size_t* const chunk_size, |
| 219 int* const is_lossless) { |
| 220 const uint8_t* const data = *data_ptr; |
| 221 const int is_vp8 = !memcmp(data, "VP8 ", TAG_SIZE); |
| 222 const int is_vp8l = !memcmp(data, "VP8L", TAG_SIZE); |
| 223 const uint32_t minimal_size = |
| 224 TAG_SIZE + CHUNK_HEADER_SIZE; // "WEBP" + "VP8 nnnn" OR |
| 225 // "WEBP" + "VP8Lnnnn" |
| 226 assert(data != NULL); |
| 227 assert(data_size != NULL); |
| 228 assert(chunk_size != NULL); |
| 229 assert(is_lossless != NULL); |
| 168 | 230 |
| 169 if (*data_size < CHUNK_HEADER_SIZE) { | 231 if (*data_size < CHUNK_HEADER_SIZE) { |
| 170 return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data. | 232 return VP8_STATUS_NOT_ENOUGH_DATA; // Insufficient data. |
| 171 } | 233 } |
| 172 | 234 |
| 173 if (!memcmp(*data, "VP8 ", TAG_SIZE)) { | 235 if (is_vp8 || is_vp8l) { |
| 174 *vp8_chunk_size = get_le32(*data + TAG_SIZE); | 236 // Bitstream contains VP8/VP8L header. |
| 175 if (riff_size >= TAG_SIZE + CHUNK_HEADER_SIZE && // "WEBP" + "VP8 nnnn". | 237 const uint32_t size = get_le32(data + TAG_SIZE); |
| 176 (*vp8_chunk_size > riff_size - (TAG_SIZE + CHUNK_HEADER_SIZE))) { | 238 if ((riff_size >= minimal_size) && (size > riff_size - minimal_size)) { |
| 177 return VP8_STATUS_BITSTREAM_ERROR; // Inconsistent size information. | 239 return VP8_STATUS_BITSTREAM_ERROR; // Inconsistent size information. |
| 178 } | 240 } |
| 179 // We have consumed CHUNK_HEADER_SIZE bytes from VP8 Header. Skip them. | 241 // Skip over CHUNK_HEADER_SIZE bytes from VP8/VP8L Header. |
| 180 *bytes_skipped = CHUNK_HEADER_SIZE; | 242 *chunk_size = size; |
| 181 *data += *bytes_skipped; | 243 *data_ptr += CHUNK_HEADER_SIZE; |
| 182 *data_size -= *bytes_skipped; | 244 *data_size -= CHUNK_HEADER_SIZE; |
| 183 } | 245 *is_lossless = is_vp8l; |
| 184 return VP8_STATUS_OK; | 246 } else { |
| 185 } | 247 // Raw VP8/VP8L bitstream (no header). |
| 186 | 248 *is_lossless = VP8LCheckSignature(data, *data_size); |
| 187 VP8StatusCode WebPParseHeaders(const uint8_t** data, uint32_t* data_size, | 249 *chunk_size = *data_size; |
| 188 uint32_t* vp8_size, uint32_t* bytes_skipped) { | |
| 189 const uint8_t* buf; | |
| 190 uint32_t buf_size; | |
| 191 uint32_t riff_size; | |
| 192 uint32_t vp8_size_tmp; | |
| 193 uint32_t optional_data_size; | |
| 194 uint32_t vp8x_skip_size; | |
| 195 uint32_t vp8_skip_size; | |
| 196 VP8StatusCode status; | |
| 197 | |
| 198 assert(data); | |
| 199 assert(data_size); | |
| 200 assert(vp8_size); | |
| 201 assert(bytes_skipped); | |
| 202 | |
| 203 buf = *data; | |
| 204 buf_size = *data_size; | |
| 205 | |
| 206 *vp8_size = 0; | |
| 207 *bytes_skipped = 0; | |
| 208 | |
| 209 if (buf == NULL || buf_size < RIFF_HEADER_SIZE) { | |
| 210 return VP8_STATUS_NOT_ENOUGH_DATA; | |
| 211 } | 250 } |
| 212 | 251 |
| 213 // Skip over RIFF header. | |
| 214 if (WebPParseRIFF(&buf, &buf_size, &riff_size) != VP8_STATUS_OK) { | |
| 215 return VP8_STATUS_BITSTREAM_ERROR; // Wrong RIFF Header. | |
| 216 } | |
| 217 | |
| 218 // Skip over VP8X header. | |
| 219 status = WebPParseVP8X(&buf, &buf_size, &vp8x_skip_size, NULL, NULL, NULL); | |
| 220 if (status != VP8_STATUS_OK) { | |
| 221 return status; // Wrong VP8X Chunk / Insufficient data. | |
| 222 } | |
| 223 if (vp8x_skip_size > 0) { | |
| 224 // Skip over optional chunks. | |
| 225 status = WebPParseOptionalChunks(&buf, &buf_size, riff_size, | |
| 226 &optional_data_size); | |
| 227 if (status != VP8_STATUS_OK) { | |
| 228 return status; // Found an invalid chunk size / Insufficient data. | |
| 229 } | |
| 230 } | |
| 231 | |
| 232 // Skip over VP8 chunk header. | |
| 233 status = WebPParseVP8Header(&buf, &buf_size, riff_size, &vp8_skip_size, | |
| 234 &vp8_size_tmp); | |
| 235 if (status != VP8_STATUS_OK) { | |
| 236 return status; // Invalid VP8 header / Insufficient data. | |
| 237 } | |
| 238 if (vp8_skip_size > 0) { | |
| 239 *vp8_size = vp8_size_tmp; | |
| 240 } | |
| 241 | |
| 242 *bytes_skipped = buf - *data; | |
| 243 assert(*bytes_skipped == *data_size - buf_size); | |
| 244 *data = buf; | |
| 245 *data_size = buf_size; | |
| 246 return VP8_STATUS_OK; | 252 return VP8_STATUS_OK; |
| 247 } | 253 } |
| 248 | 254 |
| 249 //------------------------------------------------------------------------------ | 255 //------------------------------------------------------------------------------ |
| 256 |
| 257 // Fetch '*width', '*height', '*has_alpha' and fill out 'headers' based on |
| 258 // 'data'. All the output parameters may be NULL. If 'headers' is NULL only the |
| 259 // minimal amount will be read to fetch the remaining parameters. |
| 260 // If 'headers' is non-NULL this function will attempt to locate both alpha |
| 261 // data (with or without a VP8X chunk) and the bitstream chunk (VP8/VP8L). |
| 262 // Note: The following chunk sequences (before the raw VP8/VP8L data) are |
| 263 // considered valid by this function: |
| 264 // RIFF + VP8(L) |
| 265 // RIFF + VP8X + (optional chunks) + VP8(L) |
| 266 // ALPH + VP8 <-- Not a valid WebP format: only allowed for internal purpose. |
| 267 // VP8(L) <-- Not a valid WebP format: only allowed for internal purpose. |
| 268 static VP8StatusCode ParseHeadersInternal(const uint8_t* data, |
| 269 size_t data_size, |
| 270 int* const width, |
| 271 int* const height, |
| 272 int* const has_alpha, |
| 273 WebPHeaderStructure* const headers) { |
| 274 int found_riff = 0; |
| 275 int found_vp8x = 0; |
| 276 VP8StatusCode status; |
| 277 WebPHeaderStructure hdrs; |
| 278 |
| 279 if (data == NULL || data_size < RIFF_HEADER_SIZE) { |
| 280 return VP8_STATUS_NOT_ENOUGH_DATA; |
| 281 } |
| 282 memset(&hdrs, 0, sizeof(hdrs)); |
| 283 hdrs.data = data; |
| 284 hdrs.data_size = data_size; |
| 285 |
| 286 // Skip over RIFF header. |
| 287 status = ParseRIFF(&data, &data_size, &hdrs.riff_size); |
| 288 if (status != VP8_STATUS_OK) { |
| 289 return status; // Wrong RIFF header / insufficient data. |
| 290 } |
| 291 found_riff = (hdrs.riff_size > 0); |
| 292 |
| 293 // Skip over VP8X. |
| 294 { |
| 295 uint32_t flags = 0; |
| 296 status = ParseVP8X(&data, &data_size, &found_vp8x, width, height, &flags); |
| 297 if (status != VP8_STATUS_OK) { |
| 298 return status; // Wrong VP8X / insufficient data. |
| 299 } |
| 300 if (!found_riff && found_vp8x) { |
| 301 // Note: This restriction may be removed in the future, if it becomes |
| 302 // necessary to send VP8X chunk to the decoder. |
| 303 return VP8_STATUS_BITSTREAM_ERROR; |
| 304 } |
| 305 if (has_alpha != NULL) *has_alpha = !!(flags & ALPHA_FLAG_BIT); |
| 306 if (found_vp8x && headers == NULL) { |
| 307 return VP8_STATUS_OK; // Return features from VP8X header. |
| 308 } |
| 309 } |
| 310 |
| 311 if (data_size < TAG_SIZE) return VP8_STATUS_NOT_ENOUGH_DATA; |
| 312 |
| 313 // Skip over optional chunks if data started with "RIFF + VP8X" or "ALPH". |
| 314 if ((found_riff && found_vp8x) || |
| 315 (!found_riff && !found_vp8x && !memcmp(data, "ALPH", TAG_SIZE))) { |
| 316 status = ParseOptionalChunks(&data, &data_size, hdrs.riff_size, |
| 317 &hdrs.alpha_data, &hdrs.alpha_data_size); |
| 318 if (status != VP8_STATUS_OK) { |
| 319 return status; // Found an invalid chunk size / insufficient data. |
| 320 } |
| 321 } |
| 322 |
| 323 // Skip over VP8/VP8L header. |
| 324 status = ParseVP8Header(&data, &data_size, hdrs.riff_size, |
| 325 &hdrs.compressed_size, &hdrs.is_lossless); |
| 326 if (status != VP8_STATUS_OK) { |
| 327 return status; // Wrong VP8/VP8L chunk-header / insufficient data. |
| 328 } |
| 329 if (hdrs.compressed_size > MAX_CHUNK_PAYLOAD) { |
| 330 return VP8_STATUS_BITSTREAM_ERROR; |
| 331 } |
| 332 |
| 333 if (!hdrs.is_lossless) { |
| 334 if (data_size < VP8_FRAME_HEADER_SIZE) { |
| 335 return VP8_STATUS_NOT_ENOUGH_DATA; |
| 336 } |
| 337 // Validates raw VP8 data. |
| 338 if (!VP8GetInfo(data, data_size, |
| 339 (uint32_t)hdrs.compressed_size, width, height)) { |
| 340 return VP8_STATUS_BITSTREAM_ERROR; |
| 341 } |
| 342 } else { |
| 343 if (data_size < VP8L_FRAME_HEADER_SIZE) { |
| 344 return VP8_STATUS_NOT_ENOUGH_DATA; |
| 345 } |
| 346 // Validates raw VP8L data. |
| 347 if (!VP8LGetInfo(data, data_size, width, height, has_alpha)) { |
| 348 return VP8_STATUS_BITSTREAM_ERROR; |
| 349 } |
| 350 } |
| 351 |
| 352 if (has_alpha != NULL) { |
| 353 // If the data did not contain a VP8X/VP8L chunk the only definitive way |
| 354 // to set this is by looking for alpha data (from an ALPH chunk). |
| 355 *has_alpha |= (hdrs.alpha_data != NULL); |
| 356 } |
| 357 if (headers != NULL) { |
| 358 *headers = hdrs; |
| 359 headers->offset = data - headers->data; |
| 360 assert((uint64_t)(data - headers->data) < MAX_CHUNK_PAYLOAD); |
| 361 assert(headers->offset == headers->data_size - data_size); |
| 362 } |
| 363 return VP8_STATUS_OK; // Return features from VP8 header. |
| 364 } |
| 365 |
| 366 VP8StatusCode WebPParseHeaders(WebPHeaderStructure* const headers) { |
| 367 assert(headers != NULL); |
| 368 // fill out headers, ignore width/height/has_alpha. |
| 369 return ParseHeadersInternal(headers->data, headers->data_size, |
| 370 NULL, NULL, NULL, headers); |
| 371 } |
| 372 |
| 373 //------------------------------------------------------------------------------ |
| 250 // WebPDecParams | 374 // WebPDecParams |
| 251 | 375 |
| 252 void WebPResetDecParams(WebPDecParams* const params) { | 376 void WebPResetDecParams(WebPDecParams* const params) { |
| 253 if (params) { | 377 if (params) { |
| 254 memset(params, 0, sizeof(*params)); | 378 memset(params, 0, sizeof(*params)); |
| 255 } | 379 } |
| 256 } | 380 } |
| 257 | 381 |
| 258 //------------------------------------------------------------------------------ | 382 //------------------------------------------------------------------------------ |
| 259 // "Into" decoding variants | 383 // "Into" decoding variants |
| 260 | 384 |
| 261 // Main flow | 385 // Main flow |
| 262 static VP8StatusCode DecodeInto(const uint8_t* data, uint32_t data_size, | 386 static VP8StatusCode DecodeInto(const uint8_t* const data, size_t data_size, |
| 263 WebPDecParams* const params) { | 387 WebPDecParams* const params) { |
| 264 VP8Decoder* dec = VP8New(); | 388 VP8StatusCode status; |
| 265 VP8StatusCode status = VP8_STATUS_OK; | |
| 266 VP8Io io; | 389 VP8Io io; |
| 390 WebPHeaderStructure headers; |
| 267 | 391 |
| 268 assert(params); | 392 headers.data = data; |
| 269 if (dec == NULL) { | 393 headers.data_size = data_size; |
| 270 return VP8_STATUS_INVALID_PARAM; | 394 status = WebPParseHeaders(&headers); // Process Pre-VP8 chunks. |
| 395 if (status != VP8_STATUS_OK) { |
| 396 return status; |
| 271 } | 397 } |
| 272 | 398 |
| 399 assert(params != NULL); |
| 273 VP8InitIo(&io); | 400 VP8InitIo(&io); |
| 274 io.data = data; | 401 io.data = headers.data + headers.offset; |
| 275 io.data_size = data_size; | 402 io.data_size = headers.data_size - headers.offset; |
| 276 WebPInitCustomIo(params, &io); // Plug the I/O functions. | 403 WebPInitCustomIo(params, &io); // Plug the I/O functions. |
| 277 | 404 |
| 405 if (!headers.is_lossless) { |
| 406 VP8Decoder* const dec = VP8New(); |
| 407 if (dec == NULL) { |
| 408 return VP8_STATUS_OUT_OF_MEMORY; |
| 409 } |
| 278 #ifdef WEBP_USE_THREAD | 410 #ifdef WEBP_USE_THREAD |
| 279 dec->use_threads_ = params->options && (params->options->use_threads > 0); | 411 dec->use_threads_ = params->options && (params->options->use_threads > 0); |
| 280 #else | 412 #else |
| 281 dec->use_threads_ = 0; | 413 dec->use_threads_ = 0; |
| 282 #endif | 414 #endif |
| 415 dec->alpha_data_ = headers.alpha_data; |
| 416 dec->alpha_data_size_ = headers.alpha_data_size; |
| 283 | 417 |
| 284 // Decode bitstream header, update io->width/io->height. | 418 // Decode bitstream header, update io->width/io->height. |
| 285 if (!VP8GetHeaders(dec, &io)) { | 419 if (!VP8GetHeaders(dec, &io)) { |
| 286 status = VP8_STATUS_BITSTREAM_ERROR; | 420 status = dec->status_; // An error occurred. Grab error status. |
| 287 } else { | 421 } else { |
| 288 // Allocate/check output buffers. | 422 // Allocate/check output buffers. |
| 289 status = WebPAllocateDecBuffer(io.width, io.height, params->options, | 423 status = WebPAllocateDecBuffer(io.width, io.height, params->options, |
| 290 params->output); | 424 params->output); |
| 291 if (status == VP8_STATUS_OK) { | 425 if (status == VP8_STATUS_OK) { // Decode |
| 292 // Decode | 426 if (!VP8Decode(dec, &io)) { |
| 293 if (!VP8Decode(dec, &io)) { | 427 status = dec->status_; |
| 294 status = dec->status_; | 428 } |
| 295 } | 429 } |
| 296 } | 430 } |
| 431 VP8Delete(dec); |
| 432 } else { |
| 433 VP8LDecoder* const dec = VP8LNew(); |
| 434 if (dec == NULL) { |
| 435 return VP8_STATUS_OUT_OF_MEMORY; |
| 436 } |
| 437 if (!VP8LDecodeHeader(dec, &io)) { |
| 438 status = dec->status_; // An error occurred. Grab error status. |
| 439 } else { |
| 440 // Allocate/check output buffers. |
| 441 status = WebPAllocateDecBuffer(io.width, io.height, params->options, |
| 442 params->output); |
| 443 if (status == VP8_STATUS_OK) { // Decode |
| 444 if (!VP8LDecodeImage(dec)) { |
| 445 status = dec->status_; |
| 446 } |
| 447 } |
| 448 } |
| 449 VP8LDelete(dec); |
| 297 } | 450 } |
| 298 VP8Delete(dec); | 451 |
| 299 if (status != VP8_STATUS_OK) { | 452 if (status != VP8_STATUS_OK) { |
| 300 WebPFreeDecBuffer(params->output); | 453 WebPFreeDecBuffer(params->output); |
| 301 } | 454 } |
| 302 return status; | 455 return status; |
| 303 } | 456 } |
| 304 | 457 |
| 305 // Helpers | 458 // Helpers |
| 306 static uint8_t* DecodeIntoRGBABuffer(WEBP_CSP_MODE colorspace, | 459 static uint8_t* DecodeIntoRGBABuffer(WEBP_CSP_MODE colorspace, |
| 307 const uint8_t* data, uint32_t data_size, | 460 const uint8_t* const data, |
| 308 uint8_t* rgba, int stride, int size) { | 461 size_t data_size, |
| 462 uint8_t* const rgba, |
| 463 int stride, size_t size) { |
| 309 WebPDecParams params; | 464 WebPDecParams params; |
| 310 WebPDecBuffer buf; | 465 WebPDecBuffer buf; |
| 311 if (rgba == NULL) { | 466 if (rgba == NULL) { |
| 312 return NULL; | 467 return NULL; |
| 313 } | 468 } |
| 314 WebPInitDecBuffer(&buf); | 469 WebPInitDecBuffer(&buf); |
| 315 WebPResetDecParams(¶ms); | 470 WebPResetDecParams(¶ms); |
| 316 params.output = &buf; | 471 params.output = &buf; |
| 317 buf.colorspace = colorspace; | 472 buf.colorspace = colorspace; |
| 318 buf.u.RGBA.rgba = rgba; | 473 buf.u.RGBA.rgba = rgba; |
| 319 buf.u.RGBA.stride = stride; | 474 buf.u.RGBA.stride = stride; |
| 320 buf.u.RGBA.size = size; | 475 buf.u.RGBA.size = size; |
| 321 buf.is_external_memory = 1; | 476 buf.is_external_memory = 1; |
| 322 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { | 477 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { |
| 323 return NULL; | 478 return NULL; |
| 324 } | 479 } |
| 325 return rgba; | 480 return rgba; |
| 326 } | 481 } |
| 327 | 482 |
| 328 uint8_t* WebPDecodeRGBInto(const uint8_t* data, uint32_t data_size, | 483 uint8_t* WebPDecodeRGBInto(const uint8_t* data, size_t data_size, |
| 329 uint8_t* output, int size, int stride) { | 484 uint8_t* output, size_t size, int stride) { |
| 330 return DecodeIntoRGBABuffer(MODE_RGB, data, data_size, output, stride, size); | 485 return DecodeIntoRGBABuffer(MODE_RGB, data, data_size, output, stride, size); |
| 331 } | 486 } |
| 332 | 487 |
| 333 uint8_t* WebPDecodeRGBAInto(const uint8_t* data, uint32_t data_size, | 488 uint8_t* WebPDecodeRGBAInto(const uint8_t* data, size_t data_size, |
| 334 uint8_t* output, int size, int stride) { | 489 uint8_t* output, size_t size, int stride) { |
| 335 return DecodeIntoRGBABuffer(MODE_RGBA, data, data_size, output, stride, size); | 490 return DecodeIntoRGBABuffer(MODE_RGBA, data, data_size, output, stride, size); |
| 336 } | 491 } |
| 337 | 492 |
| 338 uint8_t* WebPDecodeARGBInto(const uint8_t* data, uint32_t data_size, | 493 uint8_t* WebPDecodeARGBInto(const uint8_t* data, size_t data_size, |
| 339 uint8_t* output, int size, int stride) { | 494 uint8_t* output, size_t size, int stride) { |
| 340 return DecodeIntoRGBABuffer(MODE_ARGB, data, data_size, output, stride, size); | 495 return DecodeIntoRGBABuffer(MODE_ARGB, data, data_size, output, stride, size); |
| 341 } | 496 } |
| 342 | 497 |
| 343 uint8_t* WebPDecodeBGRInto(const uint8_t* data, uint32_t data_size, | 498 uint8_t* WebPDecodeBGRInto(const uint8_t* data, size_t data_size, |
| 344 uint8_t* output, int size, int stride) { | 499 uint8_t* output, size_t size, int stride) { |
| 345 return DecodeIntoRGBABuffer(MODE_BGR, data, data_size, output, stride, size); | 500 return DecodeIntoRGBABuffer(MODE_BGR, data, data_size, output, stride, size); |
| 346 } | 501 } |
| 347 | 502 |
| 348 uint8_t* WebPDecodeBGRAInto(const uint8_t* data, uint32_t data_size, | 503 uint8_t* WebPDecodeBGRAInto(const uint8_t* data, size_t data_size, |
| 349 uint8_t* output, int size, int stride) { | 504 uint8_t* output, size_t size, int stride) { |
| 350 return DecodeIntoRGBABuffer(MODE_BGRA, data, data_size, output, stride, size); | 505 return DecodeIntoRGBABuffer(MODE_BGRA, data, data_size, output, stride, size); |
| 351 } | 506 } |
| 352 | 507 |
| 353 uint8_t* WebPDecodeYUVInto(const uint8_t* data, uint32_t data_size, | 508 uint8_t* WebPDecodeYUVInto(const uint8_t* data, size_t data_size, |
| 354 uint8_t* luma, int luma_size, int luma_stride, | 509 uint8_t* luma, size_t luma_size, int luma_stride, |
| 355 uint8_t* u, int u_size, int u_stride, | 510 uint8_t* u, size_t u_size, int u_stride, |
| 356 uint8_t* v, int v_size, int v_stride) { | 511 uint8_t* v, size_t v_size, int v_stride) { |
| 357 WebPDecParams params; | 512 WebPDecParams params; |
| 358 WebPDecBuffer output; | 513 WebPDecBuffer output; |
| 359 if (luma == NULL) return NULL; | 514 if (luma == NULL) return NULL; |
| 360 WebPInitDecBuffer(&output); | 515 WebPInitDecBuffer(&output); |
| 361 WebPResetDecParams(¶ms); | 516 WebPResetDecParams(¶ms); |
| 362 params.output = &output; | 517 params.output = &output; |
| 363 output.colorspace = MODE_YUV; | 518 output.colorspace = MODE_YUV; |
| 364 output.u.YUVA.y = luma; | 519 output.u.YUVA.y = luma; |
| 365 output.u.YUVA.y_stride = luma_stride; | 520 output.u.YUVA.y_stride = luma_stride; |
| 366 output.u.YUVA.y_size = luma_size; | 521 output.u.YUVA.y_size = luma_size; |
| 367 output.u.YUVA.u = u; | 522 output.u.YUVA.u = u; |
| 368 output.u.YUVA.u_stride = u_stride; | 523 output.u.YUVA.u_stride = u_stride; |
| 369 output.u.YUVA.u_size = u_size; | 524 output.u.YUVA.u_size = u_size; |
| 370 output.u.YUVA.v = v; | 525 output.u.YUVA.v = v; |
| 371 output.u.YUVA.v_stride = v_stride; | 526 output.u.YUVA.v_stride = v_stride; |
| 372 output.u.YUVA.v_size = v_size; | 527 output.u.YUVA.v_size = v_size; |
| 373 output.is_external_memory = 1; | 528 output.is_external_memory = 1; |
| 374 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { | 529 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { |
| 375 return NULL; | 530 return NULL; |
| 376 } | 531 } |
| 377 return luma; | 532 return luma; |
| 378 } | 533 } |
| 379 | 534 |
| 380 //------------------------------------------------------------------------------ | 535 //------------------------------------------------------------------------------ |
| 381 | 536 |
| 382 static uint8_t* Decode(WEBP_CSP_MODE mode, const uint8_t* data, | 537 static uint8_t* Decode(WEBP_CSP_MODE mode, const uint8_t* const data, |
| 383 uint32_t data_size, int* width, int* height, | 538 size_t data_size, int* const width, int* const height, |
| 384 WebPDecBuffer* keep_info) { | 539 WebPDecBuffer* const keep_info) { |
| 385 WebPDecParams params; | 540 WebPDecParams params; |
| 386 WebPDecBuffer output; | 541 WebPDecBuffer output; |
| 387 | 542 |
| 388 WebPInitDecBuffer(&output); | 543 WebPInitDecBuffer(&output); |
| 389 WebPResetDecParams(¶ms); | 544 WebPResetDecParams(¶ms); |
| 390 params.output = &output; | 545 params.output = &output; |
| 391 output.colorspace = mode; | 546 output.colorspace = mode; |
| 392 | 547 |
| 393 // Retrieve (and report back) the required dimensions from bitstream. | 548 // Retrieve (and report back) the required dimensions from bitstream. |
| 394 if (!WebPGetInfo(data, data_size, &output.width, &output.height)) { | 549 if (!WebPGetInfo(data, data_size, &output.width, &output.height)) { |
| 395 return NULL; | 550 return NULL; |
| 396 } | 551 } |
| 397 if (width) *width = output.width; | 552 if (width != NULL) *width = output.width; |
| 398 if (height) *height = output.height; | 553 if (height != NULL) *height = output.height; |
| 399 | 554 |
| 400 // Decode | 555 // Decode |
| 401 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { | 556 if (DecodeInto(data, data_size, ¶ms) != VP8_STATUS_OK) { |
| 402 return NULL; | 557 return NULL; |
| 403 } | 558 } |
| 404 if (keep_info) { // keep track of the side-info | 559 if (keep_info != NULL) { // keep track of the side-info |
| 405 WebPCopyDecBuffer(&output, keep_info); | 560 WebPCopyDecBuffer(&output, keep_info); |
| 406 } | 561 } |
| 407 // return decoded samples (don't clear 'output'!) | 562 // return decoded samples (don't clear 'output'!) |
| 408 return (mode >= MODE_YUV) ? output.u.YUVA.y : output.u.RGBA.rgba; | 563 return WebPIsRGBMode(mode) ? output.u.RGBA.rgba : output.u.YUVA.y; |
| 409 } | 564 } |
| 410 | 565 |
| 411 uint8_t* WebPDecodeRGB(const uint8_t* data, uint32_t data_size, | 566 uint8_t* WebPDecodeRGB(const uint8_t* data, size_t data_size, |
| 412 int* width, int* height) { | 567 int* width, int* height) { |
| 413 return Decode(MODE_RGB, data, data_size, width, height, NULL); | 568 return Decode(MODE_RGB, data, data_size, width, height, NULL); |
| 414 } | 569 } |
| 415 | 570 |
| 416 uint8_t* WebPDecodeRGBA(const uint8_t* data, uint32_t data_size, | 571 uint8_t* WebPDecodeRGBA(const uint8_t* data, size_t data_size, |
| 417 int* width, int* height) { | 572 int* width, int* height) { |
| 418 return Decode(MODE_RGBA, data, data_size, width, height, NULL); | 573 return Decode(MODE_RGBA, data, data_size, width, height, NULL); |
| 419 } | 574 } |
| 420 | 575 |
| 421 uint8_t* WebPDecodeARGB(const uint8_t* data, uint32_t data_size, | 576 uint8_t* WebPDecodeARGB(const uint8_t* data, size_t data_size, |
| 422 int* width, int* height) { | 577 int* width, int* height) { |
| 423 return Decode(MODE_ARGB, data, data_size, width, height, NULL); | 578 return Decode(MODE_ARGB, data, data_size, width, height, NULL); |
| 424 } | 579 } |
| 425 | 580 |
| 426 uint8_t* WebPDecodeBGR(const uint8_t* data, uint32_t data_size, | 581 uint8_t* WebPDecodeBGR(const uint8_t* data, size_t data_size, |
| 427 int* width, int* height) { | 582 int* width, int* height) { |
| 428 return Decode(MODE_BGR, data, data_size, width, height, NULL); | 583 return Decode(MODE_BGR, data, data_size, width, height, NULL); |
| 429 } | 584 } |
| 430 | 585 |
| 431 uint8_t* WebPDecodeBGRA(const uint8_t* data, uint32_t data_size, | 586 uint8_t* WebPDecodeBGRA(const uint8_t* data, size_t data_size, |
| 432 int* width, int* height) { | 587 int* width, int* height) { |
| 433 return Decode(MODE_BGRA, data, data_size, width, height, NULL); | 588 return Decode(MODE_BGRA, data, data_size, width, height, NULL); |
| 434 } | 589 } |
| 435 | 590 |
| 436 uint8_t* WebPDecodeYUV(const uint8_t* data, uint32_t data_size, | 591 uint8_t* WebPDecodeYUV(const uint8_t* data, size_t data_size, |
| 437 int* width, int* height, uint8_t** u, uint8_t** v, | 592 int* width, int* height, uint8_t** u, uint8_t** v, |
| 438 int* stride, int* uv_stride) { | 593 int* stride, int* uv_stride) { |
| 439 WebPDecBuffer output; // only to preserve the side-infos | 594 WebPDecBuffer output; // only to preserve the side-infos |
| 440 uint8_t* const out = Decode(MODE_YUV, data, data_size, | 595 uint8_t* const out = Decode(MODE_YUV, data, data_size, |
| 441 width, height, &output); | 596 width, height, &output); |
| 442 | 597 |
| 443 if (out) { | 598 if (out != NULL) { |
| 444 const WebPYUVABuffer* const buf = &output.u.YUVA; | 599 const WebPYUVABuffer* const buf = &output.u.YUVA; |
| 445 *u = buf->u; | 600 *u = buf->u; |
| 446 *v = buf->v; | 601 *v = buf->v; |
| 447 *stride = buf->y_stride; | 602 *stride = buf->y_stride; |
| 448 *uv_stride = buf->u_stride; | 603 *uv_stride = buf->u_stride; |
| 449 assert(buf->u_stride == buf->v_stride); | 604 assert(buf->u_stride == buf->v_stride); |
| 450 } | 605 } |
| 451 return out; | 606 return out; |
| 452 } | 607 } |
| 453 | 608 |
| 454 static void DefaultFeatures(WebPBitstreamFeatures* const features) { | 609 static void DefaultFeatures(WebPBitstreamFeatures* const features) { |
| 455 assert(features); | 610 assert(features != NULL); |
| 456 memset(features, 0, sizeof(*features)); | 611 memset(features, 0, sizeof(*features)); |
| 457 features->bitstream_version = 0; | 612 features->bitstream_version = 0; |
| 458 } | 613 } |
| 459 | 614 |
| 460 static VP8StatusCode GetFeatures(const uint8_t* data, uint32_t data_size, | 615 static VP8StatusCode GetFeatures(const uint8_t* const data, size_t data_size, |
| 461 WebPBitstreamFeatures* const features) { | 616 WebPBitstreamFeatures* const features) { |
| 462 uint32_t vp8_chunk_size = 0; | 617 if (features == NULL || data == NULL) { |
| 463 uint32_t riff_size = 0; | |
| 464 uint32_t flags = 0; | |
| 465 uint32_t vp8x_skip_size = 0; | |
| 466 uint32_t vp8_skip_size = 0; | |
| 467 VP8StatusCode status; | |
| 468 | |
| 469 if (features == NULL) { | |
| 470 return VP8_STATUS_INVALID_PARAM; | 618 return VP8_STATUS_INVALID_PARAM; |
| 471 } | 619 } |
| 472 DefaultFeatures(features); | 620 DefaultFeatures(features); |
| 473 | 621 |
| 474 if (data == NULL) { | 622 // Only parse enough of the data to retrieve width/height/has_alpha. |
| 475 return VP8_STATUS_INVALID_PARAM; | 623 return ParseHeadersInternal(data, data_size, |
| 476 } | 624 &features->width, &features->height, |
| 477 | 625 &features->has_alpha, NULL); |
| 478 // Skip over RIFF header. | |
| 479 status = WebPParseRIFF(&data, &data_size, &riff_size); | |
| 480 if (status != VP8_STATUS_OK) { | |
| 481 return status; // Wrong RIFF Header / Insufficient data. | |
| 482 } | |
| 483 | |
| 484 // Skip over VP8X. | |
| 485 status = WebPParseVP8X(&data, &data_size, &vp8x_skip_size, &features->width, | |
| 486 &features->height, &flags); | |
| 487 if (status != VP8_STATUS_OK) { | |
| 488 return status; // Wrong VP8X / insufficient data. | |
| 489 | |
| 490 } | |
| 491 if (vp8x_skip_size > 0) { | |
| 492 return VP8_STATUS_OK; // Return features from VP8X header. | |
| 493 } | |
| 494 | |
| 495 // Skip over VP8 header. | |
| 496 status = WebPParseVP8Header(&data, &data_size, riff_size, &vp8_skip_size, | |
| 497 &vp8_chunk_size); | |
| 498 if (status != VP8_STATUS_OK) { | |
| 499 return status; // Wrong VP8 Chunk-header / insufficient data. | |
| 500 } | |
| 501 if (vp8_skip_size == 0) { | |
| 502 vp8_chunk_size = data_size; // No VP8 chunk wrapper over raw VP8 data. | |
| 503 } | |
| 504 | |
| 505 // Validates raw VP8 data. | |
| 506 if (!VP8GetInfo(data, data_size, vp8_chunk_size, | |
| 507 &features->width, &features->height, &features->has_alpha)) { | |
| 508 return VP8_STATUS_BITSTREAM_ERROR; | |
| 509 } | |
| 510 | |
| 511 return VP8_STATUS_OK; // Return features from VP8 header. | |
| 512 } | 626 } |
| 513 | 627 |
| 514 //------------------------------------------------------------------------------ | 628 //------------------------------------------------------------------------------ |
| 515 // WebPGetInfo() | 629 // WebPGetInfo() |
| 516 | 630 |
| 517 int WebPGetInfo(const uint8_t* data, uint32_t data_size, | 631 int WebPGetInfo(const uint8_t* data, size_t data_size, |
| 518 int* width, int* height) { | 632 int* width, int* height) { |
| 519 WebPBitstreamFeatures features; | 633 WebPBitstreamFeatures features; |
| 520 | 634 |
| 521 if (GetFeatures(data, data_size, &features) != VP8_STATUS_OK) { | 635 if (GetFeatures(data, data_size, &features) != VP8_STATUS_OK) { |
| 522 return 0; | 636 return 0; |
| 523 } | 637 } |
| 524 | 638 |
| 525 if (width) { | 639 if (width != NULL) { |
| 526 *width = features.width; | 640 *width = features.width; |
| 527 } | 641 } |
| 528 if (height) { | 642 if (height != NULL) { |
| 529 *height = features.height; | 643 *height = features.height; |
| 530 } | 644 } |
| 531 | 645 |
| 532 return 1; | 646 return 1; |
| 533 } | 647 } |
| 534 | 648 |
| 535 //------------------------------------------------------------------------------ | 649 //------------------------------------------------------------------------------ |
| 536 // Advance decoding API | 650 // Advance decoding API |
| 537 | 651 |
| 538 int WebPInitDecoderConfigInternal(WebPDecoderConfig* const config, | 652 int WebPInitDecoderConfigInternal(WebPDecoderConfig* config, |
| 539 int version) { | 653 int version) { |
| 540 if (version != WEBP_DECODER_ABI_VERSION) { | 654 if (WEBP_ABI_IS_INCOMPATIBLE(version, WEBP_DECODER_ABI_VERSION)) { |
| 541 return 0; // version mismatch | 655 return 0; // version mismatch |
| 542 } | 656 } |
| 543 if (config == NULL) { | 657 if (config == NULL) { |
| 544 return 0; | 658 return 0; |
| 545 } | 659 } |
| 546 memset(config, 0, sizeof(*config)); | 660 memset(config, 0, sizeof(*config)); |
| 547 DefaultFeatures(&config->input); | 661 DefaultFeatures(&config->input); |
| 548 WebPInitDecBuffer(&config->output); | 662 WebPInitDecBuffer(&config->output); |
| 549 return 1; | 663 return 1; |
| 550 } | 664 } |
| 551 | 665 |
| 552 VP8StatusCode WebPGetFeaturesInternal(const uint8_t* data, uint32_t data_size, | 666 VP8StatusCode WebPGetFeaturesInternal(const uint8_t* data, size_t data_size, |
| 553 WebPBitstreamFeatures* const features, | 667 WebPBitstreamFeatures* features, |
| 554 int version) { | 668 int version) { |
| 555 VP8StatusCode status; | 669 VP8StatusCode status; |
| 556 if (version != WEBP_DECODER_ABI_VERSION) { | 670 if (WEBP_ABI_IS_INCOMPATIBLE(version, WEBP_DECODER_ABI_VERSION)) { |
| 557 return VP8_STATUS_INVALID_PARAM; // version mismatch | 671 return VP8_STATUS_INVALID_PARAM; // version mismatch |
| 558 } | 672 } |
| 559 if (features == NULL) { | 673 if (features == NULL) { |
| 560 return VP8_STATUS_INVALID_PARAM; | 674 return VP8_STATUS_INVALID_PARAM; |
| 561 } | 675 } |
| 562 | 676 |
| 563 status = GetFeatures(data, data_size, features); | 677 status = GetFeatures(data, data_size, features); |
| 564 if (status == VP8_STATUS_NOT_ENOUGH_DATA) { | 678 if (status == VP8_STATUS_NOT_ENOUGH_DATA) { |
| 565 return VP8_STATUS_BITSTREAM_ERROR; // Not enough data treated as error. | 679 return VP8_STATUS_BITSTREAM_ERROR; // Not-enough-data treated as error. |
| 566 } | 680 } |
| 567 return status; | 681 return status; |
| 568 } | 682 } |
| 569 | 683 |
| 570 VP8StatusCode WebPDecode(const uint8_t* data, uint32_t data_size, | 684 VP8StatusCode WebPDecode(const uint8_t* data, size_t data_size, |
| 571 WebPDecoderConfig* const config) { | 685 WebPDecoderConfig* config) { |
| 572 WebPDecParams params; | 686 WebPDecParams params; |
| 573 VP8StatusCode status; | 687 VP8StatusCode status; |
| 574 | 688 |
| 575 if (!config) { | 689 if (config == NULL) { |
| 576 return VP8_STATUS_INVALID_PARAM; | 690 return VP8_STATUS_INVALID_PARAM; |
| 577 } | 691 } |
| 578 | 692 |
| 579 status = GetFeatures(data, data_size, &config->input); | 693 status = GetFeatures(data, data_size, &config->input); |
| 580 if (status != VP8_STATUS_OK) { | 694 if (status != VP8_STATUS_OK) { |
| 581 if (status == VP8_STATUS_NOT_ENOUGH_DATA) { | 695 if (status == VP8_STATUS_NOT_ENOUGH_DATA) { |
| 582 return VP8_STATUS_BITSTREAM_ERROR; // Not enough data treated as error. | 696 return VP8_STATUS_BITSTREAM_ERROR; // Not-enough-data treated as error. |
| 583 } | 697 } |
| 584 return status; | 698 return status; |
| 585 } | 699 } |
| 586 | 700 |
| 587 WebPResetDecParams(¶ms); | 701 WebPResetDecParams(¶ms); |
| 588 params.output = &config->output; | 702 params.output = &config->output; |
| 589 params.options = &config->options; | 703 params.options = &config->options; |
| 590 status = DecodeInto(data, data_size, ¶ms); | 704 status = DecodeInto(data, data_size, ¶ms); |
| 591 | 705 |
| 592 return status; | 706 return status; |
| 593 } | 707 } |
| 594 | 708 |
| 709 //------------------------------------------------------------------------------ |
| 710 // Cropping and rescaling. |
| 711 |
| 712 int WebPIoInitFromOptions(const WebPDecoderOptions* const options, |
| 713 VP8Io* const io, WEBP_CSP_MODE src_colorspace) { |
| 714 const int W = io->width; |
| 715 const int H = io->height; |
| 716 int x = 0, y = 0, w = W, h = H; |
| 717 |
| 718 // Cropping |
| 719 io->use_cropping = (options != NULL) && (options->use_cropping > 0); |
| 720 if (io->use_cropping) { |
| 721 w = options->crop_width; |
| 722 h = options->crop_height; |
| 723 x = options->crop_left; |
| 724 y = options->crop_top; |
| 725 if (!WebPIsRGBMode(src_colorspace)) { // only snap for YUV420 or YUV422 |
| 726 x &= ~1; |
| 727 y &= ~1; // TODO(later): only for YUV420, not YUV422. |
| 728 } |
| 729 if (x < 0 || y < 0 || w <= 0 || h <= 0 || x + w > W || y + h > H) { |
| 730 return 0; // out of frame boundary error |
| 731 } |
| 732 } |
| 733 io->crop_left = x; |
| 734 io->crop_top = y; |
| 735 io->crop_right = x + w; |
| 736 io->crop_bottom = y + h; |
| 737 io->mb_w = w; |
| 738 io->mb_h = h; |
| 739 |
| 740 // Scaling |
| 741 io->use_scaling = (options != NULL) && (options->use_scaling > 0); |
| 742 if (io->use_scaling) { |
| 743 if (options->scaled_width <= 0 || options->scaled_height <= 0) { |
| 744 return 0; |
| 745 } |
| 746 io->scaled_width = options->scaled_width; |
| 747 io->scaled_height = options->scaled_height; |
| 748 } |
| 749 |
| 750 // Filter |
| 751 io->bypass_filtering = options && options->bypass_filtering; |
| 752 |
| 753 // Fancy upsampler |
| 754 #ifdef FANCY_UPSAMPLING |
| 755 io->fancy_upsampling = (options == NULL) || (!options->no_fancy_upsampling); |
| 756 #endif |
| 757 |
| 758 if (io->use_scaling) { |
| 759 // disable filter (only for large downscaling ratio). |
| 760 io->bypass_filtering = (io->scaled_width < W * 3 / 4) && |
| 761 (io->scaled_height < H * 3 / 4); |
| 762 io->fancy_upsampling = 0; |
| 763 } |
| 764 return 1; |
| 765 } |
| 766 |
| 767 //------------------------------------------------------------------------------ |
| 768 |
| 595 #if defined(__cplusplus) || defined(c_plusplus) | 769 #if defined(__cplusplus) || defined(c_plusplus) |
| 596 } // extern "C" | 770 } // extern "C" |
| 597 #endif | 771 #endif |
| OLD | NEW |