| Szabolcs Nagy | 5717444 | 2013-12-12 05:09:18 +0000 | [diff] [blame] | 1 | #define _GNU_SOURCE |
| Rich Felker | 26f3832 | 2012-01-20 22:10:47 -0500 | [diff] [blame] | 2 | #include <sys/resource.h> |
| 3 | #include "syscall.h" |
| 4 | |
| Szabolcs Nagy | 8258014 | 2014-05-30 08:47:35 +0200 | [diff] [blame] | 5 | #define FIX(x) do{ if ((x)>=SYSCALL_RLIM_INFINITY) (x)=RLIM_INFINITY; }while(0) |
| 6 | |
| Rich Felker | 26f3832 | 2012-01-20 22:10:47 -0500 | [diff] [blame] | 7 | int prlimit(pid_t pid, int resource, const struct rlimit *new_limit, struct rlimit *old_limit) |
| 8 | { |
| Szabolcs Nagy | 8258014 | 2014-05-30 08:47:35 +0200 | [diff] [blame] | 9 | struct rlimit tmp; |
| 10 | int r; |
| 11 | if (new_limit && SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { |
| 12 | tmp = *new_limit; |
| 13 | FIX(tmp.rlim_cur); |
| 14 | FIX(tmp.rlim_max); |
| 15 | new_limit = &tmp; |
| 16 | } |
| 17 | r = syscall(SYS_prlimit64, pid, resource, new_limit, old_limit); |
| 18 | if (!r && old_limit && SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { |
| 19 | FIX(old_limit->rlim_cur); |
| 20 | FIX(old_limit->rlim_max); |
| 21 | } |
| 22 | return r; |
| Rich Felker | 26f3832 | 2012-01-20 22:10:47 -0500 | [diff] [blame] | 23 | } |
| Rich Felker | f600105 | 2012-09-21 04:05:01 -0400 | [diff] [blame] | 24 | |
| Szabolcs Nagy | 5717444 | 2013-12-12 05:09:18 +0000 | [diff] [blame] | 25 | #undef prlimit64 |
| Rich Felker | 63a4c9a | 2018-09-12 00:28:34 -0400 | [diff] [blame] | 26 | weak_alias(prlimit, prlimit64); |