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