diff --git a/clang/test/Driver/sigpipe-handling.c b/clang/test/Driver/sigpipe-handling.c new file mode 100644 index 00000000000000..852f0bfaf79812 --- /dev/null +++ b/clang/test/Driver/sigpipe-handling.c @@ -0,0 +1,9 @@ +// REQUIRES: shell +// RUN: %clang -E -fno-integrated-cc1 %s | head | FileCheck %s + +// Test that the parent clang driver process doesn't crash when the child cc1 +// process receives a SIGPIPE (Unix-only). +// +// The child should exit with IO_ERR, and the parent should exit cleanly. + +// CHECK: sigpipe-handling.c diff --git a/llvm/include/llvm/Support/InitLLVM.h b/llvm/include/llvm/Support/InitLLVM.h index 3be8d6b6d2e075..879dc1514d10f1 100644 --- a/llvm/include/llvm/Support/InitLLVM.h +++ b/llvm/include/llvm/Support/InitLLVM.h @@ -9,6 +9,7 @@ #ifndef LLVM_SUPPORT_LLVM_H #define LLVM_SUPPORT_LLVM_H +#include "llvm/ADT/Optional.h" #include "llvm/ADT/SmallVector.h" #include "llvm/Support/Allocator.h" #include "llvm/Support/PrettyStackTrace.h" @@ -44,7 +45,7 @@ class InitLLVM { private: BumpPtrAllocator Alloc; SmallVector Args; - PrettyStackTraceProgram StackPrinter; + Optional StackPrinter; }; } // namespace llvm diff --git a/llvm/lib/Support/InitLLVM.cpp b/llvm/lib/Support/InitLLVM.cpp index 5c56b773ea6982..152de6ebae0ac1 100644 --- a/llvm/lib/Support/InitLLVM.cpp +++ b/llvm/lib/Support/InitLLVM.cpp @@ -22,10 +22,17 @@ using namespace llvm; using namespace llvm::sys; InitLLVM::InitLLVM(int &Argc, const char **&Argv, - bool InstallPipeSignalExitHandler) - : StackPrinter(Argc, Argv) { + bool InstallPipeSignalExitHandler) { if (InstallPipeSignalExitHandler) + // The pipe signal handler must be installed before any other handlers are + // registered. This is because the Unix \ref RegisterHandlers function does + // not perform a sigaction() for SIGPIPE unless a one-shot handler is + // present, to allow long-lived processes (like lldb) to fully opt-out of + // llvm's SIGPIPE handling and ignore the signal safely. sys::SetOneShotPipeSignalFunction(sys::DefaultOneShotPipeSignalHandler); + // Initialize the stack printer after installing the one-shot pipe signal + // handler, so we can perform a sigaction() for SIGPIPE on Unix if requested. + StackPrinter.emplace(Argc, Argv); sys::PrintStackTraceOnErrorSignal(Argv[0]); install_out_of_memory_new_handler();