[lttng-tools.git] / src / common / compat / getenv.hpp

#ifndef _COMPAT_GETENV_H
#define _COMPAT_GETENV_H

/*
 * Copyright (C) 2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 *
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 */

#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <common/error.hpp>

static inline
int lttng_is_setuid_setgid()
{
	return geteuid() != getuid() || getegid() != getgid();
}

static inline
char *lttng_secure_getenv(const char *name)
{
	if (lttng_is_setuid_setgid()) {
		WARN("Getting environment variable '%s' from setuid/setgid binary refused for security reasons.",
			name);
		return nullptr;
	}
	return getenv(name);
}

#endif /* _COMPAT_GETENV_H */
Commit	Line	Data
e8fa9fb0 MD	1	#ifndef _COMPAT_GETENV_H
	2	#define _COMPAT_GETENV_H
	3
	4	/*
ab5be9fa	5	* Copyright (C) 2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
e8fa9fb0	6	*
c922647d	7	* SPDX-License-Identifier: LGPL-2.1-only
e8fa9fb0	8	*
e8fa9fb0 MD	9	*/
	10
	11	#include <stdlib.h>
	12	#include <unistd.h>
	13	#include <sys/types.h>
c9e313bc	14	#include <common/error.hpp>
e8fa9fb0 MD	15
e8fa9fb0 MD	16	static inline
cd9adb8b	17	int lttng_is_setuid_setgid()
e8fa9fb0 MD	18	{
	19	return geteuid() != getuid() \|\| getegid() != getgid();
	20	}
	21
	22	static inline
	23	char lttng_secure_getenv(const char name)
	24	{
	25	if (lttng_is_setuid_setgid()) {
	26	WARN("Getting environment variable '%s' from setuid/setgid binary refused for security reasons.",
	27	name);
cd9adb8b	28	return nullptr;
e8fa9fb0 MD	29	}
	30	return getenv(name);
	31	}
	32
	33	#endif /* _COMPAT_GETENV_H */