[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 <common/error.hpp>

#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>

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	*/
e8fa9fb0 MD	10
28f23191 JG	11	#include <common/error.hpp>
28f23191 JG	12
e8fa9fb0	13	#include <stdlib.h>
e8fa9fb0	14	#include <sys/types.h>
28f23191	15	#include <unistd.h>
e8fa9fb0	16
28f23191	17	static inline int lttng_is_setuid_setgid()
e8fa9fb0 MD	18	{
	19	return geteuid() != getuid() \|\| getegid() != getgid();
	20	}
	21
28f23191	22	static inline char lttng_secure_getenv(const char name)
e8fa9fb0 MD	23	{
	24	if (lttng_is_setuid_setgid()) {
	25	WARN("Getting environment variable '%s' from setuid/setgid binary refused for security reasons.",
28f23191	26	name);
cd9adb8b	27	return nullptr;
e8fa9fb0 MD	28	}
	29	return getenv(name);
	30	}
	31
	32	#endif /* _COMPAT_GETENV_H */