101 lines
2.5 KiB
C++
101 lines
2.5 KiB
C++
|
/*+
|
||
|
|
ARES/HADES/BORG Package -- ./libLSS/tools/sigcatcher.cpp
|
||
|
|
Copyright (C) 2014-2020 Guilhem Lavaux <guilhem.lavaux@iap.fr>
|
||
|
|
Copyright (C) 2009-2020 Jens Jasche <jens.jasche@fysik.su.se>
|
||
|
|
|
||
|
|
Additional contributions from:
|
||
|
|
Guilhem Lavaux <guilhem.lavaux@iap.fr> (2023)
|
||
|
|
|
||
|
|
+*/
|
||
|
|
#include <algorithm>
|
||
|
|
#include <cstring>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <signal.h>
|
||
|
|
#include "libLSS/tools/static_init.hpp"
|
||
|
|
#include "libLSS/tools/console.hpp"
|
||
|
|
#include "libLSS/tools/log_traits.hpp"
|
||
|
|
#include "libLSS/tools/sigcatcher.hpp"
|
||
|
|
|
||
|
|
AUTO_REGISTRATOR_IMPL(sigCatcher);
|
||
|
|
|
||
|
|
namespace LibLSS {
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
template<typename iterator>
|
||
|
|
iterator int_to_str_safe(iterator start, iterator max_end, int value)
|
||
|
|
{
|
||
|
|
int dec_place = 0;
|
||
|
|
int save = value;
|
||
|
|
iterator i = start, end;
|
||
|
|
|
||
|
|
if (value == 0) {
|
||
|
|
if (i == max_end) return i;
|
||
|
|
*i++ = '0';
|
||
|
|
return i;
|
||
|
|
}
|
||
|
|
|
||
|
|
while (i < max_end && value > 0) {
|
||
|
|
*i = '0' + (value % 10);
|
||
|
|
value /= 10;
|
||
|
|
i++;
|
||
|
|
}
|
||
|
|
end = i;
|
||
|
|
if (i != max_end) i--;
|
||
|
|
|
||
|
|
|
||
|
|
while (start < i) {
|
||
|
|
std::swap(*start, *i);
|
||
|
|
start++;
|
||
|
|
i--;
|
||
|
|
}
|
||
|
|
|
||
|
|
return end;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void signalCatcher(int s) {
|
||
|
|
static const int Nmax=1024;
|
||
|
|
static char staticBuffer[Nmax];
|
||
|
|
|
||
|
|
Console& cons = Console::instance();
|
||
|
|
int rank = MPI_Communication::instance()->rank();
|
||
|
|
int l;
|
||
|
|
char *end;
|
||
|
|
int rc;
|
||
|
|
static const char SEGV_msg[] = "****** Handling SEGV ******\n";
|
||
|
|
static const char SEGV_base[] = "SEGV on rank=";
|
||
|
|
|
||
|
|
rc = write(2, SEGV_msg, sizeof(SEGV_msg)-1);
|
||
|
|
strncpy(staticBuffer, SEGV_base, Nmax);
|
||
|
|
l = sizeof(SEGV_base)-1;
|
||
|
|
end = int_to_str_safe(staticBuffer + l, staticBuffer + Nmax, rank);
|
||
|
|
|
||
|
|
if (end-staticBuffer < sizeof(staticBuffer))
|
||
|
|
*end++ = '\n';
|
||
|
|
|
||
|
|
rc = write(2, staticBuffer, end - staticBuffer);
|
||
|
|
close(2);
|
||
|
|
abort();
|
||
|
|
}
|
||
|
|
|
||
|
|
struct sigaction old_act;
|
||
|
|
|
||
|
|
void initializeSigCatcher() {
|
||
|
|
struct sigaction act;
|
||
|
|
|
||
|
|
act.sa_handler = &signalCatcher;
|
||
|
|
sigfillset(&act.sa_mask);
|
||
|
|
act.sa_flags = SA_RESETHAND;
|
||
|
|
sigaction(SIGSEGV, &act, &old_act);
|
||
|
|
}
|
||
|
|
|
||
|
|
void doneSigCatcher() {
|
||
|
|
sigaction(SIGSEGV, &old_act, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
LibLSS::RegisterStaticInit reg(initializeSigCatcher, doneSigCatcher, 1, "SIGCATCHER"); // Just after console is initialized
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|