SignalHandler.cpp

Go to the documentation of this file.

#include "SignalHandler.h"
 
#include <csignal>
#include <iostream>
 
namespace storm {
namespace utility {
namespace resources {
 
// Maximal waiting time after abort signal before terminating
int maxWaitTime = 0;
 
SignalInformation::SignalInformation() : terminate(false), lastSignal(0) {}
 
SignalInformation::~SignalInformation() {
    // Intentionally left empty.
}
 
SignalInformation& SignalInformation::infos() {
    static SignalInformation signalInfo;
    return signalInfo;
}
 
void signalHandler(int signal) {
    if (!isTerminate()) {
        // First time we get an abort signal
        // We give the program a number of seconds to print results obtained so far before termination
        std::cerr << "ERROR: The program received signal " << signal << " and will be aborted in " << maxWaitTime << "s.\n";
        SignalInformation::infos().setTerminate(true);
        // Remember original signal such that the program returns the correct original signal
        SignalInformation::infos().setErrorCode(signal);
        // Trigger a new signal after waitTime
        setTimeoutAlarm(maxWaitTime);
    } else {
        // Second time we get a signal
        // We now definitely have to terminate as fast as possible
        if (SignalInformation::infos().getErrorCode() == SIGXCPU) {
            std::cerr << "TIMEOUT.\n";
        } else if (SignalInformation::infos().getErrorCode() == ENOMEM) {
            std::cerr << "OUT OF MEMORY.\n";
        } else if (SignalInformation::infos().getErrorCode() == SIGABRT || SignalInformation::infos().getErrorCode() == SIGINT) {
            std::cerr << "ABORT.\n";
        } else {
            std::cerr << "Received signal " << signal << '\n';
        }
        quickest_exit(SignalInformation::infos().getErrorCode());
    }
}
 
void installSignalHandler(int maximalWaitTime) {
    // Set the waiting time
    maxWaitTime = maximalWaitTime;
 
    // We use the newer sigaction instead of signal
    struct sigaction sa;
    sa.sa_handler = signalHandler;
    sigemptyset(&sa.sa_mask);
    sa.sa_flags = SA_RESTART;
 
    // CPU Limit
    if (sigaction(SIGXCPU, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    // Memory out
    if (sigaction(ENOMEM, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    if (sigaction(SIGSEGV, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    if (sigaction(SIGABRT, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    if (sigaction(SIGINT, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    if (sigaction(SIGTERM, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
    if (sigaction(SIGALRM, &sa, nullptr) == -1) {
        std::cerr << "FATAL: Installing a signal handler failed.\n";
    }
}
 
}  // namespace resources
}  // namespace utility
}  // namespace storm