[lttng-ust.git] / liblttng-ust / compat.h

#ifndef _UST_COMPAT_H
#define _UST_COMPAT_H

/*
 * Copyright (C) 2011 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 * Copyright (C) 2016 Raphaël Beamonte <raphael.beamonte@gmail.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; only
 * version 2.1 of the License.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/*
 * lttng_ust_getprocname.
 */
#ifdef __linux__

#include <sys/prctl.h>

#define LTTNG_UST_PROCNAME_LEN 17

static inline
void lttng_ust_getprocname(char *name)
{
	(void) prctl(PR_GET_NAME, (unsigned long) name, 0, 0, 0);
}

/*
 * If pthread_setname_np is available.
 */
#ifdef HAVE_PTHREAD_SETNAME_NP
static inline
int lttng_pthread_setname_np(pthread_t thread, const char *name)
{
	return pthread_setname_np(thread, name);
}
#endif

#elif defined(__FreeBSD__)

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

/*
 * Limit imposed by Linux UST-sessiond ABI.
 */
#define LTTNG_UST_PROCNAME_LEN 17

/*
 * Acts like linux prctl, the string is not necessarily 0-terminated if
 * 16-byte long.
 */
static inline
void lttng_ust_getprocname(char *name)
{
	const char *bsd_name;

	bsd_name = getprogname();
	if (!bsd_name)
		name[0] = '\0';
	else
		strncpy(name, bsd_name, LTTNG_UST_PROCNAME_LEN - 1);
}

/*
 * If pthread_set_name_np is available.
 */
#ifdef HAVE_PTHREAD_SET_NAME_NP
static inline
int lttng_pthread_setname_np(pthread_t thread, const char *name)
{
	return pthread_set_name_np(thread, name);
}
#endif

#endif

/*
 * If a pthread setname/set_name function is available, declare
 * the setustprocname() function that will add '-ust' to the end
 * of the current process name, while truncating it if needed.
 */
#if defined(HAVE_PTHREAD_SETNAME_NP) || defined(HAVE_PTHREAD_SETNAME_NP)
#define LTTNG_UST_PROCNAME_SUFFIX "-ust"

#include <pthread.h>

static inline
int lttng_ust_setustprocname(void)
{
	int ret = 0, len;
	char name[LTTNG_UST_PROCNAME_LEN];
	int limit = LTTNG_UST_PROCNAME_LEN - strlen(LTTNG_UST_PROCNAME_SUFFIX) - 1;

	lttng_ust_getprocname(name);

	len = strlen(name);
	if (len > limit) {
		len = limit;
	}

	ret = sprintf(name + len, LTTNG_UST_PROCNAME_SUFFIX);
	if (ret != strlen(LTTNG_UST_PROCNAME_SUFFIX)) {
		goto error;
	}

	ret = lttng_pthread_setname_np(pthread_self(), name);

error:
	return ret;
}
#else
static inline
int lttng_ust_setustprocname(void)
{
	return 0;
}
#endif

#include <errno.h>

#ifndef ENODATA
#define ENODATA	ENOMSG
#endif

#endif /* _UST_COMPAT_H */
Commit	Line	Data
	1	#ifndef _UST_COMPAT_H
	2	#define _UST_COMPAT_H
	3
	4	/*
	5	* Copyright (C) 2011 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
	6	* Copyright (C) 2016 Raphaël Beamonte <raphael.beamonte@gmail.com>
	7	*
	8	* This library is free software; you can redistribute it and/or
	9	* modify it under the terms of the GNU Lesser General Public
	10	* License as published by the Free Software Foundation; only
	11	* version 2.1 of the License.
	12	*
	13	* This library is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	16	* Lesser General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU Lesser General Public
	19	* License along with this library; if not, write to the Free Software
	20	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	21	*/
	22
	23	/*
	24	* lttng_ust_getprocname.
	25	*/
	26	#ifdef __linux__
	27
	28	#include <sys/prctl.h>
	29
	30	#define LTTNG_UST_PROCNAME_LEN 17
	31
	32	static inline
	33	void lttng_ust_getprocname(char *name)
	34	{
	35	(void) prctl(PR_GET_NAME, (unsigned long) name, 0, 0, 0);
	36	}
	37
	38	/*
	39	* If pthread_setname_np is available.
	40	*/
	41	#ifdef HAVE_PTHREAD_SETNAME_NP
	42	static inline
	43	int lttng_pthread_setname_np(pthread_t thread, const char *name)
	44	{
	45	return pthread_setname_np(thread, name);
	46	}
	47	#endif
	48
	49	#elif defined(__FreeBSD__)
	50
	51	#include <stdlib.h>
	52	#include <string.h>
	53
	54	/*
	55	* Limit imposed by Linux UST-sessiond ABI.
	56	*/
	57	#define LTTNG_UST_PROCNAME_LEN 17
	58
	59	/*
	60	* Acts like linux prctl, the string is not necessarily 0-terminated if
	61	* 16-byte long.
	62	*/
	63	static inline
	64	void lttng_ust_getprocname(char *name)
	65	{
	66	const char *bsd_name;
	67
	68	bsd_name = getprogname();
	69	if (!bsd_name)
	70	name[0] = '\0';
	71	else
	72	strncpy(name, bsd_name, LTTNG_UST_PROCNAME_LEN - 1);
	73	}
	74
	75	/*
	76	* If pthread_set_name_np is available.
	77	*/
	78	#ifdef HAVE_PTHREAD_SET_NAME_NP
	79	static inline
	80	int lttng_pthread_setname_np(pthread_t thread, const char *name)
	81	{
	82	return pthread_set_name_np(thread, name);
	83	}
	84	#endif
	85
	86	#endif
	87
	88	/*
	89	* If a pthread setname/set_name function is available, declare
	90	* the setustprocname() function that will add '-ust' to the end
	91	* of the current process name, while truncating it if needed.
	92	*/
	93	#if defined(HAVE_PTHREAD_SETNAME_NP) \|\| defined(HAVE_PTHREAD_SETNAME_NP)
	94	#define LTTNG_UST_PROCNAME_SUFFIX "-ust"
	95
	96	#include <pthread.h>
	97
	98	static inline
	99	int lttng_ust_setustprocname(void)
	100	{
	101	int ret = 0, len;
	102	char name[LTTNG_UST_PROCNAME_LEN];
	103	int limit = LTTNG_UST_PROCNAME_LEN - strlen(LTTNG_UST_PROCNAME_SUFFIX) - 1;
	104
	105	lttng_ust_getprocname(name);
	106
	107	len = strlen(name);
	108	if (len > limit) {
	109	len = limit;
	110	}
	111
	112	ret = sprintf(name + len, LTTNG_UST_PROCNAME_SUFFIX);
	113	if (ret != strlen(LTTNG_UST_PROCNAME_SUFFIX)) {
	114	goto error;
	115	}
	116
	117	ret = lttng_pthread_setname_np(pthread_self(), name);
	118
	119	error:
	120	return ret;
	121	}
	122	#else
	123	static inline
	124	int lttng_ust_setustprocname(void)
	125	{
	126	return 0;
	127	}
	128	#endif
	129
	130	#include <errno.h>
	131
	132	#ifndef ENODATA
	133	#define ENODATA ENOMSG
	134	#endif
	135
	136	#endif /* _UST_COMPAT_H */