Mercurial > hg > CbC > CbC_llvm
comparison include/llvm/Support/MemAlloc.h @ 147:c2174574ed3a
LLVM 10
author | Shinji KONO <kono@ie.u-ryukyu.ac.jp> |
---|---|
date | Wed, 14 Aug 2019 16:55:33 +0900 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
134:3a76565eade5 | 147:c2174574ed3a |
---|---|
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 | |
23 namespace llvm { | |
24 | |
25 LLVM_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 | |
38 LLVM_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 | |
52 LLVM_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 |