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)
safe_malloc_(size_t n)23 static 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
safe_calloc_(size_t n1,size_t n2)30 static 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)
safe_malloc_add_2op_(size_t n1,size_t n2)37 static 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)
safe_malloc_mul_2op_(size_t n1,size_t n2)46 static 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))
safe_malloc_muladd2_(size_t n1,size_t n2,size_t n3)55 static 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)
safe_realloc_mul_2op_(void * ptr,size_t n1,size_t n2)64 static 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