tests/asu.c

   1 /*
   2  * Copyright (c) 2001 - 2002 Kungliga Tekniska Högskolan
   3  * (Royal Institute of Technology, Stockholm, Sweden).
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the Institute nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  */
  33
  34 #ifdef HAVE_CONFIG_H
  35 #include <config.h>
  36 RCSID("$Id$");
  37 #endif
  38
  39 #include <sys/types.h>
  40 #include <sys/param.h>
  41 #include <sys/ioctl.h>
  42 #include <stdio.h>
  43 #include <pwd.h>
  44 #include <unistd.h>
  45
  46 #include <roken.h>
  47
  48 #include <atypes.h>
  49
  50 #include <kafs.h>
  51
  52 #include <err.h>
  53
  54 static void
  55 usage(int exit_val)
  56 {
  57     fprintf(stderr, "%s user program [arguments ...]\n", getprogname());
  58     exit(exit_val);
  59 }
  60
  61 #define NNPFS_PAG1_LLIM 33536
  62 #define NNPFS_PAG1_ULIM 34560
  63 #define NNPFS_PAG2_LLIM 32512
  64 #define NNPFS_PAG2_ULIM 48896
  65
  66 static int
  67 is_pag(void)
  68 {
  69 #ifdef NGROUPS_MAX
  70     gid_t groups[NGROUPS_MAX];
  71     int num;
  72     int base = 0;
  73     gid_t egid = getegid();
  74
  75     num = getgroups(NGROUPS_MAX, groups);
  76     if (num < 0)
  77         err(1, "getgroups failed");
  78
  79     if (num >= 1 && groups[0] == egid)
  80         base = 1;
  81
  82     if (num >= 2 + base &&
  83         groups[base] >= NNPFS_PAG1_LLIM &&
  84         groups[base] <= NNPFS_PAG1_ULIM &&
  85         groups[base + 1] >= NNPFS_PAG2_LLIM &&
  86         groups[base + 1] <= NNPFS_PAG2_ULIM)
  87         return 1;
  88 #endif /* NGROUPS_MAX */
  89
  90     return 0;
  91 }
  92
  93 int
  94 main(int argc, char **argv)
  95 {
  96     const char *user, *prog;
  97
  98     setprogname(argv[0]);
  99
 100     if (argc < 3)
 101         usage(1);
 102
 103     user = argv[1];
 104     prog = argv[2];
 105
 106     if (getuid() == 0 && is_pag()) {
 107         struct passwd *pw;
 108         gid_t groups[1];
 109         uid_t uid;
 110         gid_t gid;
 111
 112         pw = getpwnam(user);
 113         if(pw == NULL)
 114             errx(1, "no such user %s", user);
 115
 116         uid = pw->pw_uid;
 117         gid = pw->pw_gid;
 118         groups[0] = gid;
 119
 120         if (setgroups(1, groups))
 121             errx(1, "setgroups failed");
 122
 123         setgid(gid);
 124         setuid(uid);
 125         setegid(gid);
 126         seteuid(uid);
 127     }
 128
 129 #if 0
 130     if (k_hasafs()) {
 131         int ret = k_setpag();
 132         if (ret < 0)
 133             warn("k_setpag");
 134     }
 135 #endif
 136
 137     execvp(prog, &argv[2]);
 138
 139     err(1, "failed to execute %s", prog);
 140 }