//===- unittest/Support/ProgramStackTest.cpp ------------------------------===// // // 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 // //===----------------------------------------------------------------------===// #include "llvm/Support/ProgramStack.h" #include "llvm/Support/Process.h" #include "gtest/gtest.h" using namespace llvm; // One way of running on a new stack is to start a new thread. When threading // is disabled a "new thread" is implemented as simply calling the function. #if LLVM_ENABLE_THREADS static uintptr_t func(int &A) { A = 7; return getStackPointer(); } static void func2(int &A) { A = 5; } TEST(ProgramStackTest, runOnNewStack) { int A = 0; uintptr_t Stack = runOnNewStack(0, function_ref(func), A); EXPECT_EQ(A, 7); intptr_t StackDiff = (intptr_t)llvm::getStackPointer() - (intptr_t)Stack; size_t StackDistance = (size_t)std::abs(StackDiff); // Page size is used as it's large enough to guarantee were not on the same // stack but not too large to cause spurious failures. EXPECT_GT(StackDistance, llvm::sys::Process::getPageSizeEstimate()); runOnNewStack(0, function_ref(func2), A); EXPECT_EQ(A, 5); } #endif /*if LLVM_ENABLE_THREADS */