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 #include "Config/config.h" // Get the signal handler return type 15 16 static std::vector<std::string> FilesToRemove; 17 18 // IntSigs - Signals that may interrupt the program at any time. 19 static const int IntSigs[] = { 20 SIGHUP, SIGINT, SIGQUIT, SIGKILL, SIGPIPE, SIGTERM, SIGUSR1, SIGUSR2 21 }; 22 static const int *IntSigsEnd = IntSigs + sizeof(IntSigs)/sizeof(IntSigs[0]); 23 24 // KillSigs - Signals that are synchronous with the program that will cause it 25 // to die. 26 static const int KillSigs[] = { 27 SIGILL, SIGTRAP, SIGABRT, SIGFPE, SIGBUS, SIGSEGV, SIGSYS, SIGXCPU, SIGXFSZ 28 #ifdef SIGEMT 29 , SIGEMT 30 #endif 31 }; 32 static const int *KillSigsEnd = KillSigs + sizeof(KillSigs)/sizeof(KillSigs[0]); 33 34 35 // SignalHandler - The signal handler that runs... 36 static RETSIGTYPE SignalHandler(int Sig) { 37 while (!FilesToRemove.empty()) { 38 std::remove(FilesToRemove.back().c_str()); 39 FilesToRemove.pop_back(); 40 } 41 42 if (std::find(IntSigs, IntSigsEnd, Sig) != IntSigsEnd) 43 exit(1); // If this is an interrupt signal, exit the program 44 45 // Otherwise if it is a fault (like SEGV) reissue the signal to die... 46 signal(Sig, SIG_DFL); 47 } 48 49 static void RegisterHandler(int Signal) { signal(Signal, SignalHandler); } 50 51 // RemoveFileOnSignal - The public API 52 void RemoveFileOnSignal(const std::string &Filename) { 53 FilesToRemove.push_back(Filename); 54 55 std::for_each(IntSigs, IntSigsEnd, RegisterHandler); 56 std::for_each(KillSigs, KillSigsEnd, RegisterHandler); 57 } 58