src/bin/lttng-sessiond/fd-limit.cpp

   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 }