Fixed merge problems
[xfstests-dev.git] / src / runas.c
1 /*
2  * Copyright (c) 2000-2001 Silicon Graphics, Inc.  All Rights Reserved.
3  * 
4  * This program is free software; you can redistribute it and/or modify it
5  * under the terms of version 2 of the GNU General Public License as
6  * published by the Free Software Foundation.
7  * 
8  * This program is distributed in the hope that it would be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
11  * 
12  * Further, this software is distributed without any warranty that it is
13  * free of the rightful claim of any third person regarding infringement
14  * or the like.  Any license provided herein, whether implied or
15  * otherwise, applies only to this software file.  Patent licenses, if
16  * any, provided herein do not apply to combinations of this program with
17  * other software, or any other product whatsoever.
18  * 
19  * You should have received a copy of the GNU General Public License along
20  * with this program; if not, write the Free Software Foundation, Inc., 59
21  * Temple Place - Suite 330, Boston MA 02111-1307, USA.
22  * 
23  * Contact information: Silicon Graphics, Inc., 1600 Amphitheatre Pkwy,
24  * Mountain View, CA  94043, or:
25  * 
26  * http://www.sgi.com 
27  * 
28  * For further information regarding this notice, see: 
29  * 
30  * http://oss.sgi.com/projects/GenInfo/SGIGPLNoticeExplan/
31  */
32
33 /*
34  * Run a command with a particular 
35  *    - effective user id
36  *    - effective group id
37  *    - supplementary group list
38  */
39  
40 #include "global.h"
41 #include <grp.h>
42
43
44
45 char *prog;
46
47 void usage(void)
48 {
49     fprintf(stderr, "usage: %s [-u uid] [-g gid] [-s gid] cmd\n"
50            "flags:\n"
51            "    -u - effective user-id\n"
52            "    -g - effective group-id\n"
53            "    -s - supplementary group-id\n", prog);
54            
55 }
56
57 #define SUP_MAX 20
58
59 int
60 main(int argc, char **argv)
61 {
62         int c;
63         uid_t uid = -1;
64         gid_t gid = -1;
65         int pid;
66         char **cmd;
67         gid_t sgids[SUP_MAX];
68         int sup_cnt = 0;
69         int status;
70         char *p;
71
72         prog = basename(argv[0]);
73         for (p = prog; *p; p++) {
74                 if (*p == '/') {
75                         prog = p + 1;
76                 }
77         }
78
79
80         while ((c = getopt(argc, argv, "u:g:s:")) != -1) {
81                 switch (c) {
82                 case 'u':
83                         uid = atoi(optarg);
84                         break;
85                 case 'g':
86                         gid = atoi(optarg);
87                         break;
88                 case 's':
89                         if (sup_cnt+1 > SUP_MAX) {
90                             fprintf(stderr, "%s: too many sup groups\n", prog);
91                             exit(1);
92                         }
93                         sgids[sup_cnt++] = atoi(optarg);
94                         break;
95                 case '?':
96                         usage();
97                         exit(1);
98                 }
99         }
100
101         /* build up the cmd */
102         if (optind == argc) {
103             usage();
104             exit(1);
105         }
106         else {
107             char **p;
108             p = cmd = (char **)malloc(sizeof(char *) * (argc - optind + 1));
109             for ( ; optind < argc; optind++, p++) {
110                 *p = strdup(argv[optind]);
111             }
112             *p = NULL;
113         } 
114
115         if (gid != -1) {
116             if (setegid(gid) == -1) {
117                 fprintf(stderr, "%s: setegid(%d) failed: %s\n",
118                         prog, (int)gid, strerror(errno));
119                 exit(1);
120             }   
121         }
122
123         if (sup_cnt > 0) {
124             if (setgroups(sup_cnt, sgids) == -1) {
125                 fprintf(stderr, "%s: setgroups() failed: %s\n",
126                         prog, strerror(errno));
127                 exit(1);
128             }   
129         }
130
131         if (uid != -1) {
132             if (seteuid(uid) == -1) {
133                 fprintf(stderr, "%s: seteuid(%d) failed: %s\n",
134                         prog, (int)uid, strerror(errno));
135                 exit(1);
136             }   
137         }
138
139         pid = fork();
140         if (pid == -1) {
141             fprintf(stderr, "%s: fork failed: %s\n",
142                     prog, strerror(errno));
143             exit(1);
144         }
145         if (pid == 0) {
146             execv(cmd[0], cmd);
147             fprintf(stderr, "%s: %s\n", cmd[0], strerror(errno));
148             exit(errno);
149         }
150
151         wait(&status);
152         if (WIFSIGNALED(status)) {
153             fprintf(stderr, "%s: command terminated with signal %d\n", 
154                  prog, WTERMSIG(status));
155             exit(1);
156         }
157         else if (WIFEXITED(status)) {
158             exit(WEXITSTATUS(status));
159         }
160         else {
161             fprintf(stderr, "%s: command bizarre wait status 0x%x\n", 
162                prog, status);
163             exit(1);
164         }
165
166         exit(0);
167         /* NOTREACHED */
168 }