/*****************************************************************************
#                                                                            #
#    uStreamer - Lightweight and fast MJPEG-HTTP streamer.                   #
#                                                                            #
#    Copyright (C) 2018-2024  Maxim Devaev <mdevaev@gmail.com>               #
#                                                                            #
#    This program is free software: you can redistribute it and/or modify    #
#    it under the terms of the GNU General Public License as published by    #
#    the Free Software Foundation, either version 3 of the License, or       #
#    (at your option) any later version.                                     #
#                                                                            #
#    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, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
*****************************************************************************/


#pragma once

#include <stdio.h>
#include <unistd.h>
#include <signal.h>
#include <assert.h>

#include <sys/syscall.h>

#include <pthread.h>
#ifdef WITH_PTHREAD_NP
#	if defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__DragonFly__)
#		include <pthread_np.h>
#		include <sys/param.h>
#	endif
#endif

#include "types.h"
#include "tools.h"


#ifdef PTHREAD_MAX_NAMELEN_NP
#	define US_THREAD_NAME_SIZE ((uz)(PTHREAD_MAX_NAMELEN_NP))
#else
#	define US_THREAD_NAME_SIZE ((uz)16)
#endif

#define US_THREAD_CREATE(x_tid, x_func, x_arg)	assert(!pthread_create(&(x_tid), NULL, (x_func), (x_arg)))
#define US_THREAD_JOIN(x_tid)					assert(!pthread_join((x_tid), NULL))

#ifdef WITH_PTHREAD_NP
#	define US_THREAD_RENAME(x_fmt, ...) { \
			char m_new_tname_buf[US_THREAD_NAME_SIZE] = {0}; \
			US_SNPRINTF(m_new_tname_buf, (US_THREAD_NAME_SIZE - 1), (x_fmt), ##__VA_ARGS__); \
			us_thread_set_name(m_new_tname_buf); \
		}
#else
#	define US_THREAD_RENAME(x_fmt, ...)
#endif

#define US_THREAD_SETTLE(x_fmt, ...) { \
		US_THREAD_RENAME((x_fmt), ##__VA_ARGS__); \
		us_thread_block_signals(); \
	}

#define US_MUTEX_INIT(x_mutex)		assert(!pthread_mutex_init(&(x_mutex), NULL))
#define US_MUTEX_DESTROY(x_mutex)	assert(!pthread_mutex_destroy(&(x_mutex)))
#define US_MUTEX_LOCK(x_mutex)		assert(!pthread_mutex_lock(&(x_mutex)))
#define US_MUTEX_UNLOCK(x_mutex)	assert(!pthread_mutex_unlock(&(x_mutex)))

#define US_COND_INIT(x_cond)		assert(!pthread_cond_init(&(x_cond), NULL))
#define US_COND_DESTROY(x_cond)		assert(!pthread_cond_destroy(&(x_cond)))
#define US_COND_SIGNAL(x_cond)		assert(!pthread_cond_signal(&(x_cond)))
#define US_COND_BROADCAST(x_cond)	assert(!pthread_cond_broadcast(&(x_cond)))
#define US_COND_WAIT_FOR(x_var, x_cond, x_mutex) { while(!(x_var)) assert(!pthread_cond_wait(&(x_cond), &(x_mutex))); }


#ifdef WITH_PTHREAD_NP
INLINE void us_thread_set_name(const char *name) {
#	if defined(__linux__)
	pthread_setname_np(pthread_self(), name);
#	elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__DragonFly__)
	pthread_set_name_np(pthread_self(), name);
#	elif defined(__NetBSD__)
	pthread_setname_np(pthread_self(), "%s", (void*)name);
#	else
#		error us_thread_set_name() not implemented, you can disable it using WITH_PTHREAD_NP=0
#	endif
}
#endif

INLINE void us_thread_get_name(char *name) { // Always required for logging
#ifdef WITH_PTHREAD_NP
	int retval = -1;
#	if defined(__linux__) || defined (__NetBSD__)
	retval = pthread_getname_np(pthread_self(), name, US_THREAD_NAME_SIZE - 1);
#	elif \
		(defined(__FreeBSD__) && defined(__FreeBSD_version) && __FreeBSD_version >= 1103500) \
		|| (defined(__OpenBSD__) && defined(OpenBSD) && OpenBSD >= 201905) \
		|| defined(__DragonFly__)
	pthread_get_name_np(pthread_self(), name, US_THREAD_NAME_SIZE - 1);
	if (name[0] != '\0') {
		retval = 0;
	}
#	else
#		error us_thread_get_name() not implemented, you can disable it using WITH_PTHREAD_NP=0
#	endif
	if (retval < 0) {
#endif

#if defined(__linux__)
		const pid_t tid = syscall(SYS_gettid);
#elif defined(__FreeBSD__)
		long id;
		assert(!syscall(SYS_thr_self, &id));
		const pid_t tid = id;
#elif defined(__OpenBSD__)
		const pid_t tid = syscall(SYS_getthrid);
#elif defined(__NetBSD__)
		const pid_t tid = syscall(SYS__lwp_self);
#elif defined(__DragonFly__)
		const pid_t tid = syscall(SYS_lwp_gettid);
#else
		const pid_t tid = 0; // Makes cppcheck happy
#	warning gettid() not implemented
#endif
		US_SNPRINTF(name, (US_THREAD_NAME_SIZE - 1), "tid=%d", tid);

#ifdef WITH_PTHREAD_NP
	}
#endif
}

INLINE void us_thread_block_signals(void) {
	sigset_t mask;
	assert(!sigemptyset(&mask));
	assert(!sigaddset(&mask, SIGINT));
	assert(!sigaddset(&mask, SIGTERM));
	assert(!pthread_sigmask(SIG_BLOCK, &mask, NULL));
}