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