[lttng-tools.git] / tests / utils / testapp / signal-helper.h

/*
 * Copyright (C) 2018 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 */

#ifndef LTTNG_TESTAPP_SIGNAL_HELPER_H
#define LTTNG_TESTAPP_SIGNAL_HELPER_H

#include <signal.h>

static volatile int should_quit;

static
void sighandler(int sig)
{
	if (sig == SIGTERM) {
		should_quit = 1;
	}
}

static
int set_signal_handler(void)
{
	int ret;
	struct sigaction sa = {
		.sa_flags = 0,
		.sa_handler = sighandler,
	};

	ret = sigemptyset(&sa.sa_mask);
	if (ret) {
		perror("sigemptyset");
		goto end;
	}

	ret = sigaction(SIGTERM, &sa, NULL);
	if (ret) {
		perror("sigaction");
		goto end;
	}
end:
	return ret;
}

#endif /* LTTNG_TESTAPP_SIGNAL_HELPER_H */
Commit	Line	Data
95983a02	1	/*
9d16b343	2	* Copyright (C) 2018 Jérémie Galarneau <jeremie.galarneau@efficios.com>
95983a02	3	*
9d16b343	4	* SPDX-License-Identifier: LGPL-2.1-only
95983a02	5	*
95983a02 JG	6	*/
	7
	8	#ifndef LTTNG_TESTAPP_SIGNAL_HELPER_H
	9	#define LTTNG_TESTAPP_SIGNAL_HELPER_H
	10
	11	#include <signal.h>
	12
	13	static volatile int should_quit;
	14
	15	static
	16	void sighandler(int sig)
	17	{
	18	if (sig == SIGTERM) {
	19	should_quit = 1;
	20	}
	21	}
	22
	23	static
	24	int set_signal_handler(void)
	25	{
	26	int ret;
	27	struct sigaction sa = {
	28	.sa_flags = 0,
	29	.sa_handler = sighandler,
	30	};
	31
	32	ret = sigemptyset(&sa.sa_mask);
	33	if (ret) {
	34	perror("sigemptyset");
	35	goto end;
	36	}
	37
	38	ret = sigaction(SIGTERM, &sa, NULL);
	39	if (ret) {
	40	perror("sigaction");
	41	goto end;
	42	}
	43	end:
	44	return ret;
	45	}
	46
	47	#endif /* LTTNG_TESTAPP_SIGNAL_HELPER_H */