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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
//===- Signals.cpp - Signal Handling support ------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file was developed by the LLVM research group and is distributed under
// the University of Illinois Open Source License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file defines some helpful functions for dealing with the possibility of
// Unix signals occuring while your program is running.
//
//===----------------------------------------------------------------------===//
#include "llvm/System/Signals.h"
#include <vector>
#include <algorithm>
#include <cstdlib>
#include <cstdio>
#include "Config/config.h" // Get the signal handler return type
#ifdef HAVE_EXECINFO_H
# include <execinfo.h> // For backtrace().
#endif
#include <signal.h>
#include <unistd.h>
#include <sys/wait.h>
#include <cerrno>
using namespace llvm;
static std::vector<std::string> FilesToRemove;
// IntSigs - Signals that may interrupt the program at any time.
static const int IntSigs[] = {
SIGHUP, SIGINT, SIGQUIT, SIGKILL, SIGPIPE, SIGTERM, SIGUSR1, SIGUSR2
};
static const int *IntSigsEnd = IntSigs + sizeof(IntSigs)/sizeof(IntSigs[0]);
// KillSigs - Signals that are synchronous with the program that will cause it
// to die.
static const int KillSigs[] = {
SIGILL, SIGTRAP, SIGABRT, SIGFPE, SIGBUS, SIGSEGV, SIGSYS, SIGXCPU, SIGXFSZ
#ifdef SIGEMT
, SIGEMT
#endif
};
static const int *KillSigsEnd = KillSigs + sizeof(KillSigs)/sizeof(KillSigs[0]);
#ifdef HAVE_BACKTRACE
static void* StackTrace[256];
#endif
// PrintStackTrace - In the case of a program crash or fault, print out a stack
// trace so that the user has an indication of why and where we died.
//
// On glibc systems we have the 'backtrace' function, which works nicely, but
// doesn't demangle symbols. In order to backtrace symbols, we fork and exec a
// 'c++filt' process to do the demangling. This seems like the simplest and
// most robust solution when we can't allocate memory (such as in a signal
// handler). If we can't find 'c++filt', we fallback to printing mangled names.
//
static void PrintStackTrace() {
#ifdef HAVE_BACKTRACE
// Use backtrace() to output a backtrace on Linux systems with glibc.
int depth = backtrace(StackTrace, sizeof(StackTrace)/sizeof(StackTrace[0]));
// Create a one-way unix pipe. The backtracing process writes to PipeFDs[1],
// the c++filt process reads from PipeFDs[0].
int PipeFDs[2];
if (pipe(PipeFDs)) {
backtrace_symbols_fd(StackTrace, depth, STDERR_FILENO);
return;
}
switch (pid_t ChildPID = fork()) {
case -1: // Error forking, print mangled stack trace
close(PipeFDs[0]);
close(PipeFDs[1]);
backtrace_symbols_fd(StackTrace, depth, STDERR_FILENO);
return;
default: // backtracing process
close(PipeFDs[0]); // Close the reader side.
// Print the mangled backtrace into the pipe.
backtrace_symbols_fd(StackTrace, depth, PipeFDs[1]);
close(PipeFDs[1]); // We are done writing.
while (waitpid(ChildPID, 0, 0) == -1)
if (errno != EINTR) break;
return;
case 0: // c++filt process
close(PipeFDs[1]); // Close the writer side.
dup2(PipeFDs[0], 0); // Read from standard input
close(PipeFDs[0]); // Close the old descriptor
dup2(2, 1); // Revector stdout -> stderr
// Try to run c++filt or gc++filt. If neither is found, call back on 'cat'
// to print the mangled stack trace. If we can't find cat, just exit.
execlp("c++filt", "c++filt", 0);
execlp("gc++filt", "gc++filt", 0);
execlp("cat", "cat", 0);
execlp("/bin/cat", "cat", 0);
exit(0);
}
#endif
}
// SignalHandler - The signal handler that runs...
static RETSIGTYPE SignalHandler(int Sig) {
while (!FilesToRemove.empty()) {
std::remove(FilesToRemove.back().c_str());
FilesToRemove.pop_back();
}
if (std::find(IntSigs, IntSigsEnd, Sig) != IntSigsEnd)
exit(1); // If this is an interrupt signal, exit the program
// Otherwise if it is a fault (like SEGV) output the stacktrace to
// STDERR (if we can) and reissue the signal to die...
PrintStackTrace();
signal(Sig, SIG_DFL);
}
static void RegisterHandler(int Signal) { signal(Signal, SignalHandler); }
// RemoveFileOnSignal - The public API
void llvm::RemoveFileOnSignal(const std::string &Filename) {
FilesToRemove.push_back(Filename);
std::for_each(IntSigs, IntSigsEnd, RegisterHandler);
std::for_each(KillSigs, KillSigsEnd, RegisterHandler);
}
/// PrintStackTraceOnErrorSignal - When an error signal (such as SIBABRT or
/// SIGSEGV) is delivered to the process, print a stack trace and then exit.
void llvm::PrintStackTraceOnErrorSignal() {
std::for_each(KillSigs, KillSigsEnd, RegisterHandler);
}
|
