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 <stdarg.h>
33 #include <string.h>
34
35 #include "cil_log.h"
36
cil_default_mem_error_handler(void)37 __attribute__((noreturn)) void cil_default_mem_error_handler(void)
38 {
39 cil_log(CIL_ERR, "Failed to allocate memory\n");
40 exit(1);
41 }
42
43 void (*cil_mem_error_handler)(void) = &cil_default_mem_error_handler;
44
cil_set_mem_error_handler(void (* handler)(void))45 void cil_set_mem_error_handler(void (*handler)(void))
46 {
47 cil_mem_error_handler = handler;
48 }
49
cil_malloc(size_t size)50 void *cil_malloc(size_t size)
51 {
52 void *mem = malloc(size);
53 if (mem == NULL){
54 if (size == 0) {
55 return NULL;
56 }
57 (*cil_mem_error_handler)();
58 }
59
60 return mem;
61 }
62
cil_calloc(size_t num_elements,size_t element_size)63 void *cil_calloc(size_t num_elements, size_t element_size)
64 {
65 void *mem = calloc(num_elements, element_size);
66 if (mem == NULL){
67 (*cil_mem_error_handler)();
68 }
69
70 return mem;
71 }
72
cil_realloc(void * ptr,size_t size)73 void *cil_realloc(void *ptr, size_t size)
74 {
75 void *mem = realloc(ptr, size);
76 if (mem == NULL){
77 if (size == 0) {
78 return NULL;
79 }
80 (*cil_mem_error_handler)();
81 }
82
83 return mem;
84 }
85
86
cil_strdup(const char * str)87 char *cil_strdup(const char *str)
88 {
89 char *mem = NULL;
90
91 if (str == NULL) {
92 return NULL;
93 }
94
95 mem = strdup(str);
96 if (mem == NULL) {
97 (*cil_mem_error_handler)();
98 }
99
100 return mem;
101 }
102
cil_asprintf(char ** strp,const char * fmt,...)103 __attribute__ ((format (printf, 2, 3))) int cil_asprintf(char **strp, const char *fmt, ...)
104 {
105 int rc;
106 va_list ap;
107
108 va_start(ap, fmt);
109 rc = vasprintf(strp, fmt, ap);
110 va_end(ap);
111
112 if (rc == -1) {
113 (*cil_mem_error_handler)();
114 }
115
116 return rc;
117 }
118