mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 22:02:23 +01:00
Chandler Carruth
ce1a63509d
The old version wasn't compatible with ASan's instrumentation. Instead, use a builtin when available, and even in the fallback disable inlining and sanitizing as they can't do anything useful in this routine. This fixes a confusing internal ASan error seen by some folks since the stack detection was added.
50 lines
1.4 KiB
C++
50 lines
1.4 KiB
C++
// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
// Exceptions. See /LICENSE for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
#include "explorer/interpreter/stack_space.h"
|
|
|
|
#include "common/check.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
#include "llvm/Support/CrashRecoveryContext.h"
|
|
|
|
namespace Carbon::Internal {
|
|
|
|
static constexpr int64_t SufficientStack = 256 << 10;
|
|
static constexpr int64_t DesiredStackSpace = 8 << 20;
|
|
|
|
static LLVM_THREAD_LOCAL intptr_t bottom_of_stack = 0;
|
|
|
|
// Returns the current bottom of stack.
|
|
LLVM_NO_SANITIZE("address")
|
|
LLVM_ATTRIBUTE_NOINLINE static auto GetStackPointer() -> intptr_t {
|
|
#if __GNUC__ || __has_builtin(__builtin_frame_address)
|
|
return reinterpret_cast<intptr_t>(__builtin_frame_address(0));
|
|
#else
|
|
char char_on_stack = 0;
|
|
char* volatile ptr = &char_on_stack;
|
|
return reinterpret_cast<intptr_t>(ptr);
|
|
#endif
|
|
}
|
|
|
|
auto IsStackSpaceNearlyExhausted() -> bool {
|
|
if (bottom_of_stack == 0) {
|
|
// Not initialized on the thread; always start a new thread.
|
|
return true;
|
|
}
|
|
return std::abs(GetStackPointer() - bottom_of_stack) >
|
|
(DesiredStackSpace - SufficientStack);
|
|
}
|
|
|
|
auto RunWithExtraStackHelper(llvm::function_ref<void()> fn) -> void {
|
|
llvm::CrashRecoveryContext context;
|
|
context.RunSafelyOnThread(
|
|
[&] {
|
|
bottom_of_stack = GetStackPointer();
|
|
fn();
|
|
},
|
|
DesiredStackSpace);
|
|
}
|
|
|
|
} // namespace Carbon::Internal
|