mirror of
https://github.com/AuxXxilium/linux_dsm_epyc7002.git
synced 2024-12-21 09:15:42 +07:00
fe8d662aef
Unify memlock handling into bpf_rlimit.h and replace all occurences in BPF kselftests with it. Signed-off-by: Daniel Borkmann <daniel@iogearbox.net> Acked-by: Alexei Starovoitov <ast@kernel.org> Signed-off-by: Alexei Starovoitov <ast@kernel.org>
29 lines
907 B
C
29 lines
907 B
C
#include <sys/resource.h>
|
|
#include <stdio.h>
|
|
|
|
static __attribute__((constructor)) void bpf_rlimit_ctor(void)
|
|
{
|
|
struct rlimit rlim_old, rlim_new = {
|
|
.rlim_cur = RLIM_INFINITY,
|
|
.rlim_max = RLIM_INFINITY,
|
|
};
|
|
|
|
getrlimit(RLIMIT_MEMLOCK, &rlim_old);
|
|
/* For the sake of running the test cases, we temporarily
|
|
* set rlimit to infinity in order for kernel to focus on
|
|
* errors from actual test cases and not getting noise
|
|
* from hitting memlock limits. The limit is on per-process
|
|
* basis and not a global one, hence destructor not really
|
|
* needed here.
|
|
*/
|
|
if (setrlimit(RLIMIT_MEMLOCK, &rlim_new) < 0) {
|
|
perror("Unable to lift memlock rlimit");
|
|
/* Trying out lower limit, but expect potential test
|
|
* case failures from this!
|
|
*/
|
|
rlim_new.rlim_cur = rlim_old.rlim_cur + (1UL << 20);
|
|
rlim_new.rlim_max = rlim_old.rlim_max + (1UL << 20);
|
|
setrlimit(RLIMIT_MEMLOCK, &rlim_new);
|
|
}
|
|
}
|