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
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
42void (*cil_mem_error_handler)(void) = &cil_default_mem_error_handler;
43
44void cil_set_mem_error_handler(void (*handler)(void))
45{
46	cil_mem_error_handler = handler;
47}
48
49void *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
62void *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
72void *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
86char *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