/*
 * Copyright (C) 2012 - Christian Babeux <christian.babeux@efficios.com>
 * Copyright (C) 2014 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License, version 2 only, as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <urcu.h>

/*
 * Check if the specified environment variable is set.
 * Return 1 if set, otherwise 0.
 */
static
int check_env_var(const char *env)
{
	if (env) {
		char *env_val = getenv(env);
		if (env_val && (strncmp(env_val, "1", 1) == 0)) {
			return 1;
		}
	}

	return 0;
}

/* Session daemon */

int __testpoint_sessiond_thread_manage_clients(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_MANAGE_CLIENTS_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_registration_apps(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_REG_APPS_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_manage_apps(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_MANAGE_APPS_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_manage_kernel(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_MANAGE_KERNEL_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_manage_consumer(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_MANAGE_CONSUMER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_ht_cleanup(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_HT_CLEANUP_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_app_manage_notify(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_APP_MANAGE_NOTIFY_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_sessiond_thread_app_reg_dispatch(void)
{
	const char *var = "LTTNG_SESSIOND_THREAD_APP_REG_DISPATCH_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

/* Consumer daemon */

int __testpoint_consumerd_thread_channel(void)
{
	const char *var = "LTTNG_CONSUMERD_THREAD_CHANNEL_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_consumerd_thread_metadata(void)
{
	const char *var = "LTTNG_CONSUMERD_THREAD_METADATA_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_consumerd_thread_data(void)
{
	const char *var = "LTTNG_CONSUMERD_THREAD_DATA_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_consumerd_thread_sessiond(void)
{
	const char *var = "LTTNG_CONSUMERD_THREAD_SESSIOND_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_consumerd_thread_metadata_timer(void)
{
	const char *var = "LTTNG_CONSUMERD_THREAD_METADATA_TIMER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

/* Relay daemon */

int __testpoint_relayd_thread_dispatcher(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_DISPATCHER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_relayd_thread_worker(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_WORKER_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_relayd_thread_listener(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_LISTENER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_relayd_thread_live_dispatcher(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_LIVE_DISPATCHER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_relayd_thread_live_worker(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_LIVE_WORKER_EXIT";

	if (check_env_var(var)) {
		rcu_unregister_thread();
		pthread_exit(NULL);
	}

	return 0;
}

int __testpoint_relayd_thread_live_listener(void)
{
	const char *var = "LTTNG_RELAYD_THREAD_LIVE_LISTENER_EXIT";

	if (check_env_var(var)) {
		pthread_exit(NULL);
	}

	return 0;
}
