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