nsz Git - musl/blob - src/unistd/setxid.c

   1 #include <unistd.h>
   2 #include <errno.h>
   3 #include <sys/resource.h>
   4 #include "syscall.h"
   5 #include "libc.h"
   6
   7 struct ctx {
   8         int id, eid, sid;
   9         int nr, rlim, err;
  10 };
  11
  12 /* We jump through hoops to eliminate the possibility of partial failures. */
  13
  14 int __setrlimit(int, const struct rlimit *);
  15
  16 static void do_setxid(void *p)
  17 {
  18         struct ctx *c = p;
  19         if (c->err) return;
  20         if (c->rlim && c->id >= 0 && c->id != getuid()) {
  21                 struct rlimit inf = { RLIM_INFINITY, RLIM_INFINITY }, old;
  22                 getrlimit(RLIMIT_NPROC, &old);
  23                 if ((c->err = -__setrlimit(RLIMIT_NPROC, &inf)) && libc.threads_minus_1)
  24                         return;
  25                 c->err = -__syscall(c->nr, c->id, c->eid, c->sid);
  26                 __setrlimit(RLIMIT_NPROC, &old);
  27                 return;
  28         }
  29         c->err = -__syscall(c->nr, c->id, c->eid, c->sid);
  30 }
  31
  32 int __setxid(int nr, int id, int eid, int sid)
  33 {
  34         struct ctx c = { .nr = nr, .id = id, .eid = eid, .sid = sid };
  35         switch (nr) {
  36         case SYS_setuid:
  37         case SYS_setreuid:
  38         case SYS_setresuid:
  39                 c.rlim = 1;
  40         }
  41         __synccall(do_setxid, &c);
  42         if (c.err) {
  43                 errno = c.err;
  44                 return -1;
  45         }
  46         return 0;
  47 }