OLD | NEW |
(Empty) | |
| 1 #include <sys/resource.h> |
| 2 #include <errno.h> |
| 3 #include "syscall.h" |
| 4 #include "libc.h" |
| 5 |
| 6 #define MIN(a, b) ((a)<(b) ? (a) : (b)) |
| 7 #define FIX(x) do{ if ((x)>=SYSCALL_RLIM_INFINITY) (x)=RLIM_INFINITY; }while(0) |
| 8 |
| 9 int __setrlimit(int resource, const struct rlimit *rlim) |
| 10 { |
| 11 unsigned long k_rlim[2]; |
| 12 struct rlimit tmp; |
| 13 if (SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { |
| 14 tmp = *rlim; |
| 15 FIX(tmp.rlim_cur); |
| 16 FIX(tmp.rlim_max); |
| 17 rlim = &tmp; |
| 18 } |
| 19 int ret = __syscall(SYS_prlimit64, 0, resource, rlim, 0); |
| 20 if (ret != -ENOSYS) return ret; |
| 21 k_rlim[0] = MIN(rlim->rlim_cur, MIN(-1UL, SYSCALL_RLIM_INFINITY)); |
| 22 k_rlim[1] = MIN(rlim->rlim_max, MIN(-1UL, SYSCALL_RLIM_INFINITY)); |
| 23 return __syscall(SYS_setrlimit, resource, k_rlim); |
| 24 } |
| 25 |
| 26 struct ctx { |
| 27 const struct rlimit *rlim; |
| 28 int res; |
| 29 int err; |
| 30 }; |
| 31 |
| 32 static void do_setrlimit(void *p) |
| 33 { |
| 34 struct ctx *c = p; |
| 35 if (c->err>0) return; |
| 36 c->err = -__setrlimit(c->res, c->rlim); |
| 37 } |
| 38 |
| 39 int setrlimit(int resource, const struct rlimit *rlim) |
| 40 { |
| 41 struct ctx c = { .res = resource, .rlim = rlim, .err = -1 }; |
| 42 __synccall(do_setrlimit, &c); |
| 43 if (c.err) { |
| 44 if (c.err>0) errno = c.err; |
| 45 return -1; |
| 46 } |
| 47 return 0; |
| 48 } |
| 49 |
| 50 LFS64(setrlimit); |
OLD | NEW |