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