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);
|
||
|
|
}
|
||
|
|
}
|