OLD | NEW |
(Empty) | |
| 1 /* Copyright 2013 Google Inc. All Rights Reserved. |
| 2 |
| 3 Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 you may not use this file except in compliance with the License. |
| 5 You may obtain a copy of the License at |
| 6 |
| 7 http://www.apache.org/licenses/LICENSE-2.0 |
| 8 |
| 9 Unless required by applicable law or agreed to in writing, software |
| 10 distributed under the License is distributed on an "AS IS" BASIS, |
| 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 See the License for the specific language governing permissions and |
| 13 limitations under the License. |
| 14 |
| 15 Size-checked memory allocation. |
| 16 */ |
| 17 |
| 18 #include <stdlib.h> |
| 19 #include "./safe_malloc.h" |
| 20 |
| 21 #if defined(__cplusplus) || defined(c_plusplus) |
| 22 extern "C" { |
| 23 #endif |
| 24 |
| 25 /* Returns 0 in case of overflow of nmemb * size. */ |
| 26 static int CheckSizeArgumentsOverflow(uint64_t nmemb, size_t size) { |
| 27 const uint64_t total_size = nmemb * size; |
| 28 if (nmemb == 0) return 1; |
| 29 if ((uint64_t)size > BROTLI_MAX_ALLOCABLE_MEMORY / nmemb) return 0; |
| 30 if (total_size != (size_t)total_size) return 0; |
| 31 return 1; |
| 32 } |
| 33 |
| 34 void* BrotliSafeMalloc(uint64_t nmemb, size_t size) { |
| 35 if (!CheckSizeArgumentsOverflow(nmemb, size)) return NULL; |
| 36 assert(nmemb * size > 0); |
| 37 return malloc((size_t)(nmemb * size)); |
| 38 } |
| 39 |
| 40 #if defined(__cplusplus) || defined(c_plusplus) |
| 41 } /* extern "C" */ |
| 42 #endif |
OLD | NEW |