From 4abab5ad6c8465a7528ccdd5f49367da05f78bbd Mon Sep 17 00:00:00 2001 From: Vladimir Azarov Date: Tue, 1 Oct 2024 15:47:05 +0200 Subject: Initial version --- src/misc/setrlimit.c | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 src/misc/setrlimit.c (limited to 'src/misc/setrlimit.c') diff --git a/src/misc/setrlimit.c b/src/misc/setrlimit.c new file mode 100644 index 0000000..edb413f --- /dev/null +++ b/src/misc/setrlimit.c @@ -0,0 +1,51 @@ +#include +#include +#include "syscall.h" +#include "libc.h" + +#define MIN(a, b) ((a)<(b) ? (a) : (b)) +#define FIX(x) do{ if ((x)>=SYSCALL_RLIM_INFINITY) (x)=RLIM_INFINITY; }while(0) + +struct ctx { + unsigned long lim[2]; + int res; + int err; +}; + +#ifdef SYS_setrlimit +static void do_setrlimit(void *p) +{ + struct ctx *c = p; + if (c->err>0) return; + c->err = -__syscall(SYS_setrlimit, c->res, c->lim); +} +#endif + +int setrlimit(int resource, const struct rlimit *rlim) +{ + struct rlimit tmp; + if (SYSCALL_RLIM_INFINITY != RLIM_INFINITY) { + tmp = *rlim; + FIX(tmp.rlim_cur); + FIX(tmp.rlim_max); + rlim = &tmp; + } + int ret = __syscall(SYS_prlimit64, 0, resource, rlim, 0); +#ifdef SYS_setrlimit + if (ret != -ENOSYS) return __syscall_ret(ret); + + struct ctx c = { + .lim[0] = MIN(rlim->rlim_cur, MIN(-1UL, SYSCALL_RLIM_INFINITY)), + .lim[1] = MIN(rlim->rlim_max, MIN(-1UL, SYSCALL_RLIM_INFINITY)), + .res = resource, .err = -1 + }; + __synccall(do_setrlimit, &c); + if (c.err) { + if (c.err>0) errno = c.err; + return -1; + } + return 0; +#else + return __syscall_ret(ret); +#endif +} -- cgit v1.2.3