llvm-project/clang-tools-extra/clangd/unittests/ThreadCrashReporterTests.cpp
Emma Blink 045695f85c [clangd] Print current request context along with the stack trace
Motivation:

At the moment it is hard to attribute a clangd crash to a specific request out of all in-flight requests that might be processed concurrently. So before we can act on production clangd crashes, we have to do quite some digging through the log tables populated by our in-house VSCode extension or sometimes even directly reach out to the affected developer. Having all the details needed to reproduce a crash printed alongside its stack trace has a potential to save us quite some time, that could better be spent on fixing the actual problems.

Implementation approach:

* introduce `ThreadCrashReporter` class that allows to set a temporary signal handler for the current thread
* follow RAII pattern to simplify printing context for crashes occurring within a particular scope
* hold `std::function` as a handler to allow capturing context to print
* set local `ThreadCrashReporter` within `JSONTransport::loop()` to print request JSON for main thread crashes, and in `ASTWorker::run()` to print the file paths, arguments and contents for worker thread crashes

`ThreadCrashReporter` currently allows only one active handler per thread, but the approach can be extended to support stacked handlers printing context incrementally.

Example output for main thread crashes:

  ```
  ...
  #15 0x00007f7ddc819493 __libc_start_main (/lib64/libc.so.6+0x23493)
  #16 0x000000000249775e _start (/home/emmablink/local/llvm-project/build/bin/clangd+0x249775e)
  Signalled while processing message:
  {"jsonrpc": "2.0", "method": "textDocument/didOpen", "params": {"textDocument": {"uri": "file:///home/emmablink/test.cpp", "languageId": "cpp", "version": 1, "text": "template <typename>\nclass Bar {\n  Bar<int> *variables_to_modify;\n  foo() {\n    for (auto *c : *variables_to_modify)\n      delete c;\n  }\n};\n"}}}
  ```

Example output for AST worker crashes:

  ```
  ...
  #41 0x00007fb18304c14a start_thread pthread_create.c:0:0
  #42 0x00007fb181bfcdc3 clone (/lib64/libc.so.6+0xfcdc3)
  Signalled during AST action:
  Filename: test.cpp
  Directory: /home/emmablink
  Command Line: /usr/bin/clang -resource-dir=/data/users/emmablink/llvm-project/build/lib/clang/14.0.0 -- /home/emmablink/test.cpp
  Version: 1
  Contents:
  template <typename>
  class Bar {
    Bar<int> *variables_to_modify;
    foo() {
      for (auto *c : *variables_to_modify)
        delete c;
    }
  };
  ```

Testing:

The unit test covers the thread-localitity and nesting aspects of `ThreadCrashReporter`. There might be way to set up a lit-based integration test that would spawn clangd, send a message to it, signal it immediately and check the standard output, but this might be prone to raceconditions.

Reviewed By: sammccall

Differential Revision: https://reviews.llvm.org/D109506
2021-10-26 00:58:09 +02:00

79 lines
2.2 KiB
C++

///===- 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