mirror of
https://github.com/google/brotli.git
synced 2024-11-22 11:40:06 +00:00
83aa24dc86
* Change order of members of bit reader state structure. * Remove unused includes for assert. Add BROTLI_DCHECK macros and use it instead of assert. * Do not calculate nbits in common case of ReadSymbol. * Introduce and use PREDICT_TRUE / PREDICT_FALSE macros. * Allocate less memory in the brotli decoder if it knows the result size beforehand. Before this, the decoder would always allocate 16MB if the encoder annotated the window size as 22 bit (which is the default), even if the file is only a few KB uncompressed. Now, it'll only allocate a ringbuffer as large as needed for the result file. But only if it can know the filesize, it's not possible to know that if there are multiple metablocks or too large uncompressed metablock.
44 lines
1.3 KiB
C
44 lines
1.3 KiB
C
/* Copyright 2013 Google Inc. All Rights Reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
/* Size-checked memory allocation. */
|
|
|
|
#include <stdlib.h>
|
|
#include "./port.h"
|
|
#include "./safe_malloc.h"
|
|
|
|
#if defined(__cplusplus) || defined(c_plusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* Returns 0 in case of overflow of nmemb * size. */
|
|
static int CheckSizeArgumentsOverflow(uint64_t nmemb, size_t size) {
|
|
const uint64_t total_size = nmemb * size;
|
|
if (nmemb == 0) return 1;
|
|
if ((uint64_t)size > BROTLI_MAX_ALLOCABLE_MEMORY / nmemb) return 0;
|
|
if (total_size != (size_t)total_size) return 0;
|
|
return 1;
|
|
}
|
|
|
|
void* BrotliSafeMalloc(uint64_t nmemb, size_t size) {
|
|
if (!CheckSizeArgumentsOverflow(nmemb, size)) return NULL;
|
|
BROTLI_DCHECK(nmemb * size > 0);
|
|
return malloc((size_t)(nmemb * size));
|
|
}
|
|
|
|
#if defined(__cplusplus) || defined(c_plusplus)
|
|
} /* extern "C" */
|
|
#endif
|