mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
3fa1d1fe3f
This adds a new copy of LLVMSupport into the runtime. This is the final step before changing the inline namespace for the runtime support. This will allow us to avoid the ODR violations from the header definitions of LLVMSupport. LLVMSupport forked at: 22492eead218ec91d349c8c50439880fbeacf2b7 Changes made to LLVMSupport from that revision: process.inc forward declares `_beginthreadex` due to compilation issues due to custom flag handling API changes required that we alter the `Deallocate` routine to account for the alignment. This is a temporary state, meant to simplify the process. We do not use the entire LLVMSupport library and there is no value in keeping the entire library. Subsequent commits will prune the library to the needs for the runtime.
88 lines
3.2 KiB
C++
88 lines
3.2 KiB
C++
//===- MemAlloc.h - Memory allocation functions -----------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
/// \file
|
|
///
|
|
/// This file defines counterparts of C library allocation functions defined in
|
|
/// the namespace 'std'. The new allocation functions crash on allocation
|
|
/// failure instead of returning null pointer.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_SUPPORT_MEMALLOC_H
|
|
#define LLVM_SUPPORT_MEMALLOC_H
|
|
|
|
#include "llvm/Support/Compiler.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include <cstdlib>
|
|
|
|
namespace llvm {
|
|
|
|
LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_malloc(size_t Sz) {
|
|
void *Result = std::malloc(Sz);
|
|
if (Result == nullptr) {
|
|
// It is implementation-defined whether allocation occurs if the space
|
|
// requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting
|
|
// non-zero, if the space requested was zero.
|
|
if (Sz == 0)
|
|
return safe_malloc(1);
|
|
report_bad_alloc_error("Allocation failed");
|
|
}
|
|
return Result;
|
|
}
|
|
|
|
LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_calloc(size_t Count,
|
|
size_t Sz) {
|
|
void *Result = std::calloc(Count, Sz);
|
|
if (Result == nullptr) {
|
|
// It is implementation-defined whether allocation occurs if the space
|
|
// requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting
|
|
// non-zero, if the space requested was zero.
|
|
if (Count == 0 || Sz == 0)
|
|
return safe_malloc(1);
|
|
report_bad_alloc_error("Allocation failed");
|
|
}
|
|
return Result;
|
|
}
|
|
|
|
LLVM_ATTRIBUTE_RETURNS_NONNULL inline void *safe_realloc(void *Ptr, size_t Sz) {
|
|
void *Result = std::realloc(Ptr, Sz);
|
|
if (Result == nullptr) {
|
|
// It is implementation-defined whether allocation occurs if the space
|
|
// requested is zero (ISO/IEC 9899:2018 7.22.3). Retry, requesting
|
|
// non-zero, if the space requested was zero.
|
|
if (Sz == 0)
|
|
return safe_malloc(1);
|
|
report_bad_alloc_error("Allocation failed");
|
|
}
|
|
return Result;
|
|
}
|
|
|
|
/// Allocate a buffer of memory with the given size and alignment.
|
|
///
|
|
/// When the compiler supports aligned operator new, this will use it to to
|
|
/// handle even over-aligned allocations.
|
|
///
|
|
/// However, this doesn't make any attempt to leverage the fancier techniques
|
|
/// like posix_memalign due to portability. It is mostly intended to allow
|
|
/// compatibility with platforms that, after aligned allocation was added, use
|
|
/// reduced default alignment.
|
|
LLVM_ATTRIBUTE_RETURNS_NONNULL LLVM_ATTRIBUTE_RETURNS_NOALIAS void *
|
|
allocate_buffer(size_t Size, size_t Alignment);
|
|
|
|
/// Deallocate a buffer of memory with the given size and alignment.
|
|
///
|
|
/// If supported, this will used the sized delete operator. Also if supported,
|
|
/// this will pass the alignment to the delete operator.
|
|
///
|
|
/// The pointer must have been allocated with the corresponding new operator,
|
|
/// most likely using the above helper.
|
|
void deallocate_buffer(void *Ptr, size_t Size, size_t Alignment);
|
|
|
|
} // namespace llvm
|
|
#endif
|