/**

 * 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>

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;

	WLog_INFO(TAG, "Caught signal '%s' [%d]", strsignal(signum), signum);

	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