Signals.cpp revision 39602b2002cf8b640226df767ee924a9d42a5c0f 
1//===- Signals.cpp - Signal Handling support ------------------------------===//
2//
3// This file defines some helpful functions for dealing with the possibility of
4// unix signals occuring while your program is running.
5//
6//===----------------------------------------------------------------------===//
7
8#include "Support/Signals.h"
9#include <vector>
10#include <algorithm>
11#include <cstdlib>
12#include <cstdio>
13#include <signal.h>
14
15static std::vector<std::string> FilesToRemove;
16
17// IntSigs - Signals that may interrupt the program at any time.
18static const int IntSigs[] = {
19  SIGHUP, SIGINT, SIGQUIT, SIGKILL, SIGPIPE, SIGTERM, SIGUSR1, SIGUSR2
20};
21static const int *IntSigsEnd = IntSigs + sizeof(IntSigs)/sizeof(IntSigs[0]);
22
23// KillSigs - Signals that are synchronous with the program that will cause it
24// to die.
25static const int KillSigs[] = {
26  SIGILL, SIGTRAP, SIGABRT, SIGFPE, SIGBUS, SIGSEGV, SIGSYS, SIGXCPU, SIGXFSZ
27#ifdef SIGEMT
28  , SIGEMT
29#endif
30};
31static const int *KillSigsEnd = KillSigs + sizeof(KillSigs)/sizeof(KillSigs[0]);
32
33
34// SignalHandler - The signal handler that runs...
35static void SignalHandler(int Sig) {
36  while (!FilesToRemove.empty()) {
37    std::remove(FilesToRemove.back().c_str());
38    FilesToRemove.pop_back();
39  }
40
41  if (std::find(IntSigs, IntSigsEnd, Sig) != IntSigsEnd)
42    exit(1);   // If this is an interrupt signal, exit the program
43
44  // Otherwise if it is a fault (like SEGV) reissue the signal to die...
45  signal(Sig, SIG_DFL);
46}
47
48static void RegisterHandler(int Signal) { signal(Signal, SignalHandler); }
49
50// RemoveFileOnSignal - The public API
51void RemoveFileOnSignal(const std::string &Filename) {
52  FilesToRemove.push_back(Filename);
53
54  std::for_each(IntSigs, IntSigsEnd, RegisterHandler);
55  std::for_each(KillSigs, KillSigsEnd, RegisterHandler);
56}
57