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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
///===- ThreadCrashReporterTests.cpp - Thread local signal handling tests -===//
//
// 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 "support/ThreadCrashReporter.h"
#include "support/Threading.h"
#include "llvm/Support/Signals.h"
#include "gtest/gtest.h"
#include <csignal>
#include <string>
namespace clang {
namespace clangd {
namespace {
static void infoSignalHandler() { ThreadCrashReporter::runCrashHandlers(); }
TEST(ThreadCrashReporterTest, All) {
#if defined(_WIN32)
// Simulate signals on Windows for unit testing purposes.
// The `crash.test` lit test checks the end-to-end integration.
auto SignalCurrentThread = []() { infoSignalHandler(); };
#else
llvm::sys::SetInfoSignalFunction(&infoSignalHandler);
auto SignalCurrentThread = []() { raise(SIGUSR1); };
#endif
AsyncTaskRunner Runner;
auto SignalAnotherThread = [&]() {
Runner.runAsync("signal another thread", SignalCurrentThread);
Runner.wait();
};
bool Called;
{
ThreadCrashReporter ScopedReporter([&Called]() { Called = true; });
// Check handler gets called when a signal gets delivered to the current
// thread.
Called = false;
SignalCurrentThread();
EXPECT_TRUE(Called);
// Check handler does not get called when another thread gets signalled.
Called = false;
SignalAnotherThread();
EXPECT_FALSE(Called);
}
// Check handler does not get called when the reporter object goes out of
// scope.
Called = false;
SignalCurrentThread();
EXPECT_FALSE(Called);
std::string Order = "";
{
ThreadCrashReporter ScopedReporter([&Order] { Order.push_back('a'); });
{
ThreadCrashReporter ScopedReporter([&Order] { Order.push_back('b'); });
SignalCurrentThread();
}
// Check that handlers are called in LIFO order.
EXPECT_EQ(Order, "ba");
// Check that current handler is the only one after the nested scope is
// over.
SignalCurrentThread();
EXPECT_EQ(Order, "baa");
}
}
} // namespace
} // namespace clangd
} // namespace clang
|
