[lttng-tools.git] / src / bin / lttng-sessiond / fd-limit.cpp

/*
 * Copyright (C) 2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 *
 * SPDX-License-Identifier: GPL-2.0-only
 *
 */

#define _LGPL_SOURCE
#include "fd-limit.hpp"

#include <common/compat/errno.hpp>
#include <common/error.hpp>

#include <stdio.h>
#include <sys/resource.h>
#include <sys/time.h>
#include <urcu/uatomic.h>

/* total count of fd. */
static long fd_count;

/*
 * threshold in % of number of fd allowed.
 */
static long fd_threshold[LTTNG_FD_NR_TYPES] = {
	75, /* LTTNG_FD_APPS */
};

static rlim_t max_nr_fd;

int lttng_fd_get(enum lttng_fd_type type, unsigned int nr)
{
	long newval;

	if (type >= LTTNG_FD_NR_TYPES) {
		return -EINVAL;
	}

	newval = uatomic_add_return(&fd_count, (long) nr);
	if ((long) (newval * 100) - (long) (max_nr_fd * fd_threshold[type]) > 0) {
		uatomic_sub(&fd_count, (long) nr);
		return -EPERM;
	}
	return 0;
}

void lttng_fd_put(enum lttng_fd_type type __attribute__((unused)), unsigned int nr)
{
	uatomic_sub(&fd_count, (long) nr);
}

void lttng_fd_init()
{
	struct rlimit rlim;
	int ret;

	ret = getrlimit(RLIMIT_NOFILE, &rlim);
	if (ret < 0) {
		PERROR("getrlimit");
	}
	max_nr_fd = rlim.rlim_cur;
}
Commit	Line	Data
	1	/*
	2	* Copyright (C) 2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
	3	*
	4	* SPDX-License-Identifier: GPL-2.0-only
	5	*
	6	*/
	7
	8	#define _LGPL_SOURCE
	9	#include "fd-limit.hpp"
	10
	11	#include <common/compat/errno.hpp>
	12	#include <common/error.hpp>
	13
	14	#include <stdio.h>
	15	#include <sys/resource.h>
	16	#include <sys/time.h>
	17	#include <urcu/uatomic.h>
	18
	19	/* total count of fd. */
	20	static long fd_count;
	21
	22	/*
	23	* threshold in % of number of fd allowed.
	24	*/
	25	static long fd_threshold[LTTNG_FD_NR_TYPES] = {
	26	75, /* LTTNG_FD_APPS */
	27	};
	28
	29	static rlim_t max_nr_fd;
	30
	31	int lttng_fd_get(enum lttng_fd_type type, unsigned int nr)
	32	{
	33	long newval;
	34
	35	if (type >= LTTNG_FD_NR_TYPES) {
	36	return -EINVAL;
	37	}
	38
	39	newval = uatomic_add_return(&fd_count, (long) nr);
	40	if ((long) (newval * 100) - (long) (max_nr_fd * fd_threshold[type]) > 0) {
	41	uatomic_sub(&fd_count, (long) nr);
	42	return -EPERM;
	43	}
	44	return 0;
	45	}
	46
	47	void lttng_fd_put(enum lttng_fd_type type __attribute__((unused)), unsigned int nr)
	48	{
	49	uatomic_sub(&fd_count, (long) nr);
	50	}
	51
	52	void lttng_fd_init()
	53	{
	54	struct rlimit rlim;
	55	int ret;
	56
	57	ret = getrlimit(RLIMIT_NOFILE, &rlim);
	58	if (ret < 0) {
	59	PERROR("getrlimit");
	60	}
	61	max_nr_fd = rlim.rlim_cur;
	62	}