Quellcode-Bibliothek rlimit.c   Sprache: C

/* SPDX-License-Identifier: LGPL-2.1 */

#include <errno.h>
#include "util/debug.h"
#include "util/rlimit.h"
#include <sys/time.h>
#include <sys/resource.h>

/*
 * Bump the memlock so that we can get bpf maps of a reasonable size,
 * like the ones used with 'perf trace' and with 'perf test bpf',
 * improve this to some specific request if needed.
 */
void rlimit__bump_memlock(void)
{
 struct rlimit rlim;

 if (getrlimit(RLIMIT_MEMLOCK, &rlim) == 0) {
  rlim.rlim_cur *= 4;
  rlim.rlim_max *= 4;

  if (setrlimit(RLIMIT_MEMLOCK, &rlim) < 0) {
   rlim.rlim_cur /= 2;
   rlim.rlim_max /= 2;

   if (setrlimit(RLIMIT_MEMLOCK, &rlim) < 0)
    pr_debug("Couldn't bump rlimit(MEMLOCK), failures may take place when creating BPF maps, etc\n");
  }
 }
}

bool rlimit__increase_nofile(enum rlimit_action *set_rlimit)
{
 int old_errno;
 struct rlimit l;

 if (*set_rlimit < INCREASED_MAX) {
  old_errno = errno;

  if (getrlimit(RLIMIT_NOFILE, &l) == 0) {
   if (*set_rlimit == NO_CHANGE) {
    l.rlim_cur = l.rlim_max;
   } else {
    l.rlim_cur = l.rlim_max + 1000;
    l.rlim_max = l.rlim_cur;
   }
   if (setrlimit(RLIMIT_NOFILE, &l) == 0) {
    (*set_rlimit) += 1;
    errno = old_errno;
    return true;
   }
  }
  errno = old_errno;
 }

 return false;
}