blob: 2321751c539fef1b438dfd26b3247a9ce0a3fa46 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
//===- 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<uintptr_t(int &)>(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<void(int &)>(func2), A);
EXPECT_EQ(A, 5);
}
#endif /*if LLVM_ENABLE_THREADS */
|