128 lines
3.2 KiB
C
128 lines
3.2 KiB
C
|
/**
|
||
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
||
|
|
* Signal handling
|
||
|
|
*
|
||
|
|
* Copyright 2011 Shea Levy <shea@shealevy.com>
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agreed to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifdef HAVE_CONFIG_H
|
||
|
|
#include "config.h"
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <stddef.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#include <winpr/crt.h>
|
||
|
|
|
||
|
|
#include <freerdp/utils/signal.h>
|
||
|
|
#include <freerdp/log.h>
|
||
|
|
|
||
|
|
#define TAG FREERDP_TAG("utils")
|
||
|
|
|
||
|
|
#ifdef _WIN32
|
||
|
|
|
||
|
|
int freerdp_handle_signals(void)
|
||
|
|
{
|
||
|
|
errno = ENOSYS;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
#else
|
||
|
|
|
||
|
|
#include <pthread.h>
|
||
|
|
#include <winpr/debug.h>
|
||
|
|
|
||
|
|
volatile sig_atomic_t terminal_needs_reset = 0;
|
||
|
|
int terminal_fildes = 0;
|
||
|
|
struct termios orig_flags;
|
||
|
|
struct termios new_flags;
|
||
|
|
|
||
|
|
static void fatal_handler(int signum)
|
||
|
|
{
|
||
|
|
struct sigaction default_sigaction;
|
||
|
|
sigset_t this_mask;
|
||
|
|
static BOOL recursive = FALSE;
|
||
|
|
|
||
|
|
if (!recursive)
|
||
|
|
{
|
||
|
|
recursive = TRUE;
|
||
|
|
WLog_ERR(TAG, "Caught signal '%s' [%d]", strsignal(signum), signum);
|
||
|
|
|
||
|
|
winpr_log_backtrace(TAG, WLOG_ERROR, 20);
|
||
|
|
}
|
||
|
|
if (terminal_needs_reset)
|
||
|
|
tcsetattr(terminal_fildes, TCSAFLUSH, &orig_flags);
|
||
|
|
|
||
|
|
default_sigaction.sa_handler = SIG_DFL;
|
||
|
|
sigfillset(&(default_sigaction.sa_mask));
|
||
|
|
default_sigaction.sa_flags = 0;
|
||
|
|
sigaction(signum, &default_sigaction, NULL);
|
||
|
|
sigemptyset(&this_mask);
|
||
|
|
sigaddset(&this_mask, signum);
|
||
|
|
pthread_sigmask(SIG_UNBLOCK, &this_mask, NULL);
|
||
|
|
raise(signum);
|
||
|
|
}
|
||
|
|
|
||
|
|
const int fatal_signals[] = { SIGABRT, SIGALRM, SIGBUS, SIGFPE, SIGHUP, SIGILL,
|
||
|
|
SIGINT, SIGKILL, SIGQUIT, SIGSEGV, SIGSTOP, SIGTERM,
|
||
|
|
SIGTSTP, SIGTTIN, SIGTTOU, SIGUSR1, SIGUSR2,
|
||
|
|
#ifdef SIGPOLL
|
||
|
|
SIGPOLL,
|
||
|
|
#endif
|
||
|
|
#ifdef SIGPROF
|
||
|
|
SIGPROF,
|
||
|
|
#endif
|
||
|
|
#ifdef SIGSYS
|
||
|
|
SIGSYS,
|
||
|
|
#endif
|
||
|
|
SIGTRAP,
|
||
|
|
#ifdef SIGVTALRM
|
||
|
|
SIGVTALRM,
|
||
|
|
#endif
|
||
|
|
SIGXCPU, SIGXFSZ };
|
||
|
|
|
||
|
|
int freerdp_handle_signals(void)
|
||
|
|
{
|
||
|
|
size_t signal_index;
|
||
|
|
sigset_t orig_set;
|
||
|
|
struct sigaction orig_sigaction;
|
||
|
|
struct sigaction fatal_sigaction;
|
||
|
|
WLog_DBG(TAG, "Registering signal hook...");
|
||
|
|
sigfillset(&(fatal_sigaction.sa_mask));
|
||
|
|
sigdelset(&(fatal_sigaction.sa_mask), SIGCONT);
|
||
|
|
pthread_sigmask(SIG_BLOCK, &(fatal_sigaction.sa_mask), &orig_set);
|
||
|
|
fatal_sigaction.sa_handler = fatal_handler;
|
||
|
|
fatal_sigaction.sa_flags = 0;
|
||
|
|
|
||
|
|
for (signal_index = 0; signal_index < ARRAYSIZE(fatal_signals); signal_index++)
|
||
|
|
{
|
||
|
|
if (sigaction(fatal_signals[signal_index], NULL, &orig_sigaction) == 0)
|
||
|
|
{
|
||
|
|
if (orig_sigaction.sa_handler != SIG_IGN)
|
||
|
|
{
|
||
|
|
sigaction(fatal_signals[signal_index], &fatal_sigaction, NULL);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pthread_sigmask(SIG_SETMASK, &orig_set, NULL);
|
||
|
|
/* Ignore SIGPIPE signal. */
|
||
|
|
signal(SIGPIPE, SIG_IGN);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|