diff options
| author | Vladimir Azarov <avm@intermediate-node.net> | 2024-10-01 15:47:05 +0200 |
|---|---|---|
| committer | Vladimir Azarov <avm@intermediate-node.net> | 2024-10-01 15:47:05 +0200 |
| commit | 4abab5ad6c8465a7528ccdd5f49367da05f78bbd (patch) | |
| tree | ebf009bf1376a5a223a915bc27cbbd791a1316bc /src/linux/prlimit.c | |
Initial version
Diffstat (limited to 'src/linux/prlimit.c')
| -rw-r--r-- | src/linux/prlimit.c | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/linux/prlimit.c b/src/linux/prlimit.c new file mode 100644 index 0000000..fcf45aa --- /dev/null +++ b/src/linux/prlimit.c @@ -0,0 +1,23 @@ +#define _GNU_SOURCE +#include <sys/resource.h> +#include "syscall.h" + +#define FIX(x) do{ if ((x)>=SYSCALL_RLIM_INFINITY) (x)=RLIM_INFINITY; }while(0) + +int prlimit(pid_t pid, int resource, const struct rlimit *new_limit, struct rlimit *old_limit) +{ + struct rlimit tmp; + int r; + if (new_limit && SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { + tmp = *new_limit; + FIX(tmp.rlim_cur); + FIX(tmp.rlim_max); + new_limit = &tmp; + } + r = syscall(SYS_prlimit64, pid, resource, new_limit, old_limit); + if (!r && old_limit && SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { + FIX(old_limit->rlim_cur); + FIX(old_limit->rlim_max); + } + return r; +} |