1//===- MemAlloc.h - Memory allocation functions -----------------*- C++ -*-===// 2// 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4// See https://llvm.org/LICENSE.txt for license information. 5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6// 7//===----------------------------------------------------------------------===// 8/// \file 9/// 10/// This file defines counterparts of C library allocation functions defined in 11/// the namespace 'std'. The new allocation functions crash on allocation 12/// failure instead of returning null pointer. 13/// 14//===----------------------------------------------------------------------===// 15 16#ifndef LLVM_SUPPORT_MEMALLOC_H 17#define LLVM_SUPPORT_MEMALLOC_H 18 19#include "llvm/Support/Compiler.h" 20#include "llvm/Support/ErrorHandling.h" 21#include <cstdlib> 22 23namespace llvm { 24 25LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_malloc(size_t Sz) { 26 void *Result = std::malloc(Sz); 27 if (Result == nullptr) { 28 // It is implementation-defined whether allocation occurs if the space 29 // requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting 30 // non-zero, if the space requested was zero. 31 if (Sz == 0) 32 return safe_malloc(1); 33 report_bad_alloc_error("Allocation failed"); 34 } 35 return Result; 36} 37 38LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_calloc(size_t Count, 39 size_t Sz) { 40 void *Result = std::calloc(Count, Sz); 41 if (Result == nullptr) { 42 // It is implementation-defined whether allocation occurs if the space 43 // requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting 44 // non-zero, if the space requested was zero. 45 if (Count == 0 || Sz == 0) 46 return safe_malloc(1); 47 report_bad_alloc_error("Allocation failed"); 48 } 49 return Result; 50} 51 52LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_realloc(void *Ptr, size_t Sz) { 53 void *Result = std::realloc(Ptr, Sz); 54 if (Result == nullptr) { 55 // It is implementation-defined whether allocation occurs if the space 56 // requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting 57 // non-zero, if the space requested was zero. 58 if (Sz == 0) 59 return safe_malloc(1); 60 report_bad_alloc_error("Allocation failed"); 61 } 62 return Result; 63} 64 65} 66#endif 67