1 /* SPDX-License-Identifier: LGPL-2.1 */
2 
3 #include "util/debug.h"
4 #include "util/rlimit.h"
5 #include <sys/time.h>
6 #include <sys/resource.h>
7 
8 /*
9  * Bump the memlock so that we can get bpf maps of a reasonable size,
10  * like the ones used with 'perf trace' and with 'perf test bpf',
11  * improve this to some specific request if needed.
12  */
rlimit__bump_memlock(void)13 void rlimit__bump_memlock(void)
14 {
15 	struct rlimit rlim;
16 
17 	if (getrlimit(RLIMIT_MEMLOCK, &rlim) == 0) {
18 		rlim.rlim_cur *= 4;
19 		rlim.rlim_max *= 4;
20 
21 		if (setrlimit(RLIMIT_MEMLOCK, &rlim) < 0) {
22 			rlim.rlim_cur /= 2;
23 			rlim.rlim_max /= 2;
24 
25 			if (setrlimit(RLIMIT_MEMLOCK, &rlim) < 0)
26 				pr_debug("Couldn't bump rlimit(MEMLOCK), failures may take place when creating BPF maps, etc\n");
27 		}
28 	}
29 }
30