| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 | 11 |
| 12 #include "dboolhuff.h" | 12 #include "dboolhuff.h" |
| 13 | 13 |
| 14 int vp8dx_start_decode(BOOL_DECODER *br, | 14 int vp8dx_start_decode(BOOL_DECODER *br, |
| 15 const unsigned char *source, | 15 const unsigned char *source, |
| 16 unsigned int source_sz, | 16 unsigned int source_sz, |
| 17 const unsigned char *origin, | 17 vp8_decrypt_cb *decrypt_cb, |
| 18 const unsigned char *key) | 18 void *decrypt_state) |
| 19 { | 19 { |
| 20 br->user_buffer_end = source+source_sz; | 20 br->user_buffer_end = source+source_sz; |
| 21 br->user_buffer = source; | 21 br->user_buffer = source; |
| 22 br->value = 0; | 22 br->value = 0; |
| 23 br->count = -8; | 23 br->count = -8; |
| 24 br->range = 255; | 24 br->range = 255; |
| 25 br->origin = origin; | 25 br->decrypt_cb = decrypt_cb; |
| 26 br->key = key; | 26 br->decrypt_state = decrypt_state; |
| 27 | 27 |
| 28 if (source_sz && !source) | 28 if (source_sz && !source) |
| 29 return 1; | 29 return 1; |
| 30 | 30 |
| 31 /* Populate the buffer */ | 31 /* Populate the buffer */ |
| 32 vp8dx_bool_decoder_fill(br); | 32 vp8dx_bool_decoder_fill(br); |
| 33 | 33 |
| 34 return 0; | 34 return 0; |
| 35 } | 35 } |
| 36 | 36 |
| 37 void vp8dx_bool_decoder_fill(BOOL_DECODER *br) | 37 void vp8dx_bool_decoder_fill(BOOL_DECODER *br) |
| 38 { | 38 { |
| 39 const unsigned char *bufptr = br->user_buffer; | 39 const unsigned char *bufptr = br->user_buffer; |
| 40 const unsigned char *bufend = br->user_buffer_end; | |
| 41 VP8_BD_VALUE value = br->value; | 40 VP8_BD_VALUE value = br->value; |
| 42 int count = br->count; | 41 int count = br->count; |
| 43 int shift = VP8_BD_VALUE_SIZE - 8 - (count + 8); | 42 int shift = VP8_BD_VALUE_SIZE - 8 - (count + 8); |
| 44 size_t bits_left = (bufend - bufptr)*CHAR_BIT; | 43 size_t bytes_left = br->user_buffer_end - bufptr; |
| 44 size_t bits_left = bytes_left * CHAR_BIT; |
| 45 int x = (int)(shift + CHAR_BIT - bits_left); | 45 int x = (int)(shift + CHAR_BIT - bits_left); |
| 46 int loop_end = 0; | 46 int loop_end = 0; |
| 47 unsigned char decrypted[sizeof(VP8_BD_VALUE) + 1]; |
| 48 |
| 49 if (br->decrypt_cb) { |
| 50 int n = bytes_left > sizeof(decrypted) ? sizeof(decrypted) : bytes_left; |
| 51 br->decrypt_cb(br->decrypt_state, bufptr, decrypted, n); |
| 52 bufptr = decrypted; |
| 53 } |
| 47 | 54 |
| 48 if(x >= 0) | 55 if(x >= 0) |
| 49 { | 56 { |
| 50 count += VP8_LOTS_OF_BITS; | 57 count += VP8_LOTS_OF_BITS; |
| 51 loop_end = x; | 58 loop_end = x; |
| 52 } | 59 } |
| 53 | 60 |
| 54 if (x < 0 || bits_left) | 61 if (x < 0 || bits_left) |
| 55 { | 62 { |
| 56 while(shift >= loop_end) | 63 while(shift >= loop_end) |
| 57 { | 64 { |
| 58 count += CHAR_BIT; | 65 count += CHAR_BIT; |
| 59 value |= ((VP8_BD_VALUE)decrypt_byte(bufptr, br->origin, | 66 value |= (VP8_BD_VALUE)*bufptr << shift; |
| 60 br->key)) << shift; | |
| 61 ++bufptr; | 67 ++bufptr; |
| 68 ++br->user_buffer; |
| 62 shift -= CHAR_BIT; | 69 shift -= CHAR_BIT; |
| 63 } | 70 } |
| 64 } | 71 } |
| 65 | 72 |
| 66 br->user_buffer = bufptr; | |
| 67 br->value = value; | 73 br->value = value; |
| 68 br->count = count; | 74 br->count = count; |
| 69 } | 75 } |
| OLD | NEW |