1 /*
2 * Copyright 2011 Tresys Technology, LLC. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *
7 * 1. Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 *
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY TRESYS TECHNOLOGY, LLC ``AS IS'' AND ANY EXPRESS
15 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
16 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
17 * EVENT SHALL TRESYS TECHNOLOGY, LLC OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
18 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
19 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
21 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
22 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
23 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 *
25 * The views and conclusions contained in the software and documentation are those
26 * of the authors and should not be interpreted as representing official policies,
27 * either expressed or implied, of Tresys Technology, LLC.
28 */
29
30 #include <stdlib.h>
31 #include <stdio.h>
32 #include <string.h>
33
34 #include "cil_log.h"
35
cil_default_mem_error_handler(void)36 __attribute__((noreturn)) void cil_default_mem_error_handler(void)
37 {
38 cil_log(CIL_ERR, "Failed to allocate memory\n");
39 exit(1);
40 }
41
42 void (*cil_mem_error_handler)(void) = &cil_default_mem_error_handler;
43
cil_set_mem_error_handler(void (* handler)(void))44 void cil_set_mem_error_handler(void (*handler)(void))
45 {
46 cil_mem_error_handler = handler;
47 }
48
cil_malloc(size_t size)49 void *cil_malloc(size_t size)
50 {
51 void *mem = malloc(size);
52 if (mem == NULL){
53 if (size == 0) {
54 return NULL;
55 }
56 (*cil_mem_error_handler)();
57 }
58
59 return mem;
60 }
61
cil_calloc(size_t num_elements,size_t element_size)62 void *cil_calloc(size_t num_elements, size_t element_size)
63 {
64 void *mem = calloc(num_elements, element_size);
65 if (mem == NULL){
66 (*cil_mem_error_handler)();
67 }
68
69 return mem;
70 }
71
cil_realloc(void * ptr,size_t size)72 void *cil_realloc(void *ptr, size_t size)
73 {
74 void *mem = realloc(ptr, size);
75 if (mem == NULL){
76 if (size == 0) {
77 return NULL;
78 }
79 (*cil_mem_error_handler)();
80 }
81
82 return mem;
83 }
84
85
cil_strdup(const char * str)86 char *cil_strdup(const char *str)
87 {
88 char *mem = NULL;
89
90 if (str == NULL) {
91 return NULL;
92 }
93
94 mem = strdup(str);
95 if (mem == NULL) {
96 (*cil_mem_error_handler)();
97 }
98
99 return mem;
100 }
101