| 1 | /* SPDX-License-Identifier: LGPL-2.1 */ |
|---|---|
| 2 | |
| 3 | #include <errno.h> |
| 4 | #include "util/debug.h" |
| 5 | #include "util/rlimit.h" |
| 6 | #include <sys/time.h> |
| 7 | #include <sys/resource.h> |
| 8 | |
| 9 | /* |
| 10 | * Bump the memlock so that we can get bpf maps of a reasonable size, |
| 11 | * like the ones used with 'perf trace' and with 'perf test bpf', |
| 12 | * improve this to some specific request if needed. |
| 13 | */ |
| 14 | void rlimit__bump_memlock(void) |
| 15 | { |
| 16 | struct rlimit rlim; |
| 17 | |
| 18 | if (getrlimit(RLIMIT_MEMLOCK, rlimits: &rlim) == 0) { |
| 19 | rlim.rlim_cur *= 4; |
| 20 | rlim.rlim_max *= 4; |
| 21 | |
| 22 | if (setrlimit(RLIMIT_MEMLOCK, rlimits: &rlim) < 0) { |
| 23 | rlim.rlim_cur /= 2; |
| 24 | rlim.rlim_max /= 2; |
| 25 | |
| 26 | if (setrlimit(RLIMIT_MEMLOCK, rlimits: &rlim) < 0) |
| 27 | pr_debug("Couldn't bump rlimit(MEMLOCK), failures may take place when creating BPF maps, etc\n"); |
| 28 | } |
| 29 | } |
| 30 | } |
| 31 | |
| 32 | bool rlimit__increase_nofile(enum rlimit_action *set_rlimit) |
| 33 | { |
| 34 | int old_errno; |
| 35 | struct rlimit l; |
| 36 | |
| 37 | if (*set_rlimit < INCREASED_MAX) { |
| 38 | old_errno = errno; |
| 39 | |
| 40 | if (getrlimit(RLIMIT_NOFILE, rlimits: &l) == 0) { |
| 41 | if (*set_rlimit == NO_CHANGE) { |
| 42 | l.rlim_cur = l.rlim_max; |
| 43 | } else { |
| 44 | l.rlim_cur = l.rlim_max + 1000; |
| 45 | l.rlim_max = l.rlim_cur; |
| 46 | } |
| 47 | if (setrlimit(RLIMIT_NOFILE, rlimits: &l) == 0) { |
| 48 | (*set_rlimit) += 1; |
| 49 | errno = old_errno; |
| 50 | return true; |
| 51 | } |
| 52 | } |
| 53 | errno = old_errno; |
| 54 | } |
| 55 | |
| 56 | return false; |
| 57 | } |
| 58 |