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 static void do_setxid(void *p)
  15 {
  16         struct ctx *c = p;
  17         if (c->err) return;
  18         if (c->rlim && c->id >= 0 && c->id != getuid()) {
  19                 struct rlimit inf = { RLIM_INFINITY, RLIM_INFINITY }, old;
  20                 getrlimit(RLIMIT_NPROC, &old);
  21                 if (setrlimit(RLIMIT_NPROC, &inf) && libc.threads_minus_1) {
  22                         c->err = errno;
  23                         return;
  24                 }
  25                 if (__syscall(c->nr, c->id, c->eid, c->sid))
  26                         c->err = errno;
  27                 setrlimit(RLIMIT_NPROC, &old);
  28                 return;
  29         }
  30         if (__syscall(c->nr, c->id, c->eid, c->sid))
  31                 c->err = errno;
  32 }
  33
  34 int __setxid(int nr, int id, int eid, int sid)
  35 {
  36         struct ctx c = { .nr = nr, .id = id, .eid = eid, .sid = sid };
  37         switch (nr) {
  38         case SYS_setuid:
  39         case SYS_setreuid:
  40         case SYS_setresuid:
  41                 c.rlim = 1;
  42         }
  43         __synccall(do_setxid, &c);
  44         if (c.err) {
  45                 errno = c.err;
  46                 return -1;
  47         }
  48         return 0;
  49 }