27c02fe5430c0814913c295ef8d5567721a21583
[oweals/busybox.git] / chmod_chown_chgrp.c
1 /*
2  * Mini chown/chmod/chgrp implementation for busybox
3  *
4  * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19  *
20  */
21
22 #include <stdio.h>
23 #include <grp.h>
24 #include <pwd.h>
25 #include "internal.h"
26
27
28 static int uid=-1;
29 static int gid=0;
30 static int whichApp;
31 static char* invocationName=NULL;
32 static mode_t mode=0644;
33
34
35 #define CHGRP_APP   1
36 #define CHOWN_APP   2
37 #define CHMOD_APP   3
38
39 static const char chgrp_usage[] = "[OPTION]... GROUP FILE...\n"
40     "Change the group membership of each FILE to GROUP.\n"
41     "\n\tOptions:\n" "\t-R\tchange files and directories recursively\n";
42 static const char chown_usage[] = "[OPTION]...  OWNER[.[GROUP] FILE...\n"
43     "Change the owner and/or group of each FILE to OWNER and/or GROUP.\n"
44     "\n\tOptions:\n" "\t-R\tchange files and directories recursively\n";
45 static const char chmod_usage[] = "[-R] MODE[,MODE]... FILE...\n"
46 "Each MODE is one or more of the letters ugoa, one of the symbols +-= and\n"
47 "one or more of the letters rwxst.\n\n"
48  "\t-R\tchange files and directories recursively.\n";
49
50
51
52 static int fileAction(const char *fileName, struct stat* statbuf)
53 {
54     switch (whichApp) {
55         case CHGRP_APP:
56         case CHOWN_APP:
57             if (chown(fileName, ((whichApp==CHOWN_APP)? uid: statbuf->st_uid), gid) < 0)
58                 return( TRUE);
59         case CHMOD_APP:
60             fprintf(stderr, "%s, %d\n", fileName, mode);
61             if (chmod(fileName, mode))
62                 return( TRUE);
63     }
64     perror(fileName);
65     return( FALSE);
66 }
67
68 int chmod_chown_chgrp_main(int argc, char **argv)
69 {
70     struct group *grp;
71     struct passwd *pwd;
72     int recursiveFlag=FALSE;
73     char *groupName;
74     mode_t andWithMode= S_ISVTX|S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
75
76     whichApp = (strcmp(*argv, "chown")==0)? CHOWN_APP : (strcmp(*argv, "chmod")==0)? CHMOD_APP : CHGRP_APP; 
77
78     if (argc < 2) {
79         fprintf(stderr, "Usage: %s %s", *argv, 
80                 (whichApp==TRUE)? chown_usage : chgrp_usage);
81         exit( FALSE);
82     }
83     invocationName=*argv;
84     argc--;
85     argv++;
86
87     /* Parse options */
88     while (**argv == '-') {
89         while (*++(*argv)) switch (**argv) {
90             case 'R':
91                 recursiveFlag = TRUE;
92                 break;
93             default:
94                 fprintf(stderr, "Unknown option: %c\n", **argv);
95                 exit( FALSE);
96         }
97         argc--;
98         argv++;
99     }
100     
101     if ( whichApp == CHMOD_APP ) {
102         /* Find the specified modes */
103         if ( parse_mode(*argv, &mode) == FALSE ) {
104             fprintf(stderr, "%s: Unknown mode: %s\n", invocationName, *argv);
105             exit( FALSE);
106         }
107         //mode &= andWithMode;
108         fprintf(stderr, "mode %d\n", mode);
109     } else {
110
111         /* Find the selected group */
112         groupName = strchr(*argv, '.');
113         if ( whichApp==TRUE && groupName )
114             *groupName++ = '\0';
115         else
116             groupName = *argv;
117         grp = getgrnam(groupName);
118         if (grp == NULL) {
119             fprintf(stderr, "%s: Unknown group name: %s\n", invocationName, groupName);
120             exit( FALSE);
121         }
122         gid = grp->gr_gid;
123
124         /* Find the selected user (if appropriate)  */
125         if (whichApp==TRUE) {
126             pwd = getpwnam(*argv);
127             if (pwd == NULL) {
128                 fprintf(stderr, "%s: Unknown user name: %s\n", invocationName, *argv);
129                 exit( FALSE);
130             }
131             uid = pwd->pw_uid;
132         }
133     }
134     
135     /* Ok, ready to do the deed now */
136     if (argc <= 1) {
137         fprintf(stderr, "%s: too few arguments", invocationName);
138         exit( FALSE);
139     }
140     while (argc-- > 1) {
141         if (recursiveAction( *(++argv), recursiveFlag, TRUE, FALSE, fileAction, fileAction)==FALSE)
142             exit( FALSE);
143     }
144     exit(TRUE);
145 }
146