X-Git-Url: http://nsz.repo.hu/git/?p=musl;a=blobdiff_plain;f=src%2Fmisc%2Fsetrlimit.c;h=ddc13e98bf9e8272c17fe4b074ac0a86cc7cc8d5;hp=7fdfc4e596d808eac7eadb8b908c712fd5b19d84;hb=b6218764ebca59ff5dae0e87b696188c8de0119e;hpb=0b44a0315b47dd8eced9f3b7f31580cf14bbfc01 diff --git a/src/misc/setrlimit.c b/src/misc/setrlimit.c index 7fdfc4e5..ddc13e98 100644 --- a/src/misc/setrlimit.c +++ b/src/misc/setrlimit.c @@ -1,11 +1,42 @@ #include +#include #include "syscall.h" #include "libc.h" +#define MIN(a, b) ((a)<(b) ? (a) : (b)) + +int __setrlimit(int resource, const struct rlimit *rlim) +{ + unsigned long k_rlim[2]; + int ret = __syscall(SYS_prlimit64, 0, resource, rlim, 0); + if (ret != -ENOSYS) return ret; + k_rlim[0] = MIN(rlim->rlim_cur, -1UL); + k_rlim[1] = MIN(rlim->rlim_max, -1UL); + return __syscall(SYS_setrlimit, resource, k_rlim); +} + +struct ctx { + const struct rlimit *rlim; + int res; + int err; +}; + +static void do_setrlimit(void *p) +{ + struct ctx *c = p; + if (c->err) return; + c->err = -__setrlimit(c->res, c->rlim); +} + int setrlimit(int resource, const struct rlimit *rlim) { - long k_rlim[2] = { rlim->rlim_cur, rlim->rlim_max }; - return syscall2(__NR_setrlimit, resource, (long)k_rlim); + struct ctx c = { .res = resource, .rlim = rlim }; + __synccall(do_setrlimit, &c); + if (c.err) { + errno = c.err; + return -1; + } + return 0; } LFS64(setrlimit);