1/* 2 * Copyright (C) 2011 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#ifndef __SHARE_ALLOC_H_ 18#define __SHARE_ALLOC_H_ 19 20#include <limits.h> 21 22// malloc(n floor 1) 23static inline void *safe_malloc_(size_t n) 24{ 25 // dlmalloc is already safe 26 return malloc(n); 27} 28 29// malloc(n1 * n2) then memset to zero 30static inline void *safe_calloc_(size_t n1, size_t n2) 31{ 32 // dlcalloc is already safe 33 return calloc(n1, n2); 34} 35 36// malloc(n1 + n2) 37static inline void *safe_malloc_add_2op_(size_t n1, size_t n2) 38{ 39 unsigned long long n = n1 + n2; 40 size_t ns = n; 41 // check for overflow 42 return n == ns ? malloc(ns) : NULL; 43} 44 45// malloc(n1 * n2) 46static inline void *safe_malloc_mul_2op_(size_t n1, size_t n2) 47{ 48 unsigned long long n = n1 * n2; 49 size_t ns = n; 50 // check for overflow 51 return n == ns ? malloc(ns) : NULL; 52} 53 54// malloc(n1 * (n2 + n3)) 55static inline void *safe_malloc_muladd2_(size_t n1, size_t n2, size_t n3) 56{ 57 unsigned long long n = n1 * (n2 + n3); 58 size_t ns = n; 59 // check for overflow 60 return n == ns ? malloc(ns) : NULL; 61} 62 63// realloc(ptr, n1 * n2) 64static inline void *safe_realloc_mul_2op_(void *ptr, size_t n1, size_t n2) 65{ 66 unsigned long long n = n1 * n2; 67 size_t ns = n; 68 // check for overflow 69 return n == ns ? realloc(ptr, ns) : NULL; 70} 71 72#endif // __SHARE_ALLOC_H_ 73