Extract usage information into a separate file.
[oweals/busybox.git] / chmod_chown_chgrp.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini chown/chmod/chgrp implementation for busybox
4  *
5  *
6  * Copyright (C) 1999,2000 by Lineo, inc.
7  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  * General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22  *
23  */
24
25 #include "internal.h"
26 #define BB_DECLARE_EXTERN
27 #define bb_need_invalid_option
28 #define bb_need_too_few_args
29 #include "messages.c"
30
31 #include <stdio.h>
32 #include <grp.h>
33 #include <pwd.h>
34
35
36 static long uid = -1;
37 static long gid = -1;
38 static int whichApp;
39 static char *theMode = NULL;
40
41
42 #define CHGRP_APP   1
43 #define CHOWN_APP   2
44 #define CHMOD_APP   3
45
46 static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
47 {
48         switch (whichApp) {
49         case CHGRP_APP:
50         case CHOWN_APP:
51 #if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
52                 if (lchown
53                         (fileName, (whichApp == CHOWN_APP) ? uid : statbuf->st_uid,
54                          (gid == -1) ? statbuf->st_gid : gid) == 0)
55 #else
56                 if (chown
57                         (fileName, (whichApp == CHOWN_APP) ? uid : statbuf->st_uid,
58                          (gid == -1) ? statbuf->st_gid : gid) == 0)
59 #endif
60                 {
61                         return (TRUE);
62                 }
63                 break;
64         case CHMOD_APP:
65                 /* Parse the specified modes */
66                 if (parse_mode(theMode, &(statbuf->st_mode)) == FALSE) {
67                         fatalError( "unknown mode: %s\n", theMode);
68                 }
69                 if (chmod(fileName, statbuf->st_mode) == 0)
70                         return (TRUE);
71                 break;
72         }
73         perror(fileName);
74         return (FALSE);
75 }
76
77 int chmod_chown_chgrp_main(int argc, char **argv)
78 {
79         int recursiveFlag = FALSE;
80         char *groupName=NULL;
81         char *p=NULL;
82         const char *appUsage;
83
84         whichApp = (strcmp(applet_name, "chown") == 0)? 
85                         CHOWN_APP : (strcmp(applet_name, "chmod") == 0)? 
86                                 CHMOD_APP : CHGRP_APP;
87
88         appUsage = (whichApp == CHOWN_APP)? 
89                         chown_usage : (whichApp == CHMOD_APP) ? chmod_usage : chgrp_usage;
90
91         if (argc < 2)
92                 usage(appUsage);
93         argv++;
94
95         /* Parse options */
96         while (--argc >= 0 && *argv && (**argv == '-')) {
97                 while (*++(*argv)) {
98                         switch (**argv) {
99                                 case 'R':
100                                         recursiveFlag = TRUE;
101                                         break;
102                                 default:
103                                         errorMsg(invalid_option, **argv);
104                                         usage(appUsage);
105                         }
106                 }
107                 argv++;
108         }
109
110         if (argc == 0 || *argv == NULL) {
111                 errorMsg(too_few_args);
112                 usage(appUsage);
113         }
114
115         if (whichApp == CHMOD_APP) {
116                 theMode = *argv;
117         } else {
118
119                 /* Find the selected group */
120                 if (whichApp == CHGRP_APP) {
121                         groupName = *argv;
122                         gid = strtoul(groupName, &p, 10);       /* maybe it's already numeric */
123                         if (groupName == p)
124                                 gid = my_getgrnam(groupName);
125                         if (gid == -1)
126                                 goto bad_group;
127                 } else {
128                         groupName = strchr(*argv, '.');
129                         if (groupName == NULL)
130                                 groupName = strchr(*argv, ':');
131                         if (groupName) {
132                                 *groupName++ = '\0';
133                                 gid = strtoul(groupName, &p, 10);
134                                 if (groupName == p)
135                                         gid = my_getgrnam(groupName);
136                                 if (gid == -1)
137                                         goto bad_group;
138                         } else
139                                 gid = -1;
140                 }
141
142
143                 /* Find the selected user (if appropriate)  */
144                 if (whichApp == CHOWN_APP) {
145                         uid = strtoul(*argv, &p, 10);   /* if numeric ... */
146                         if (*argv == p)
147                                 uid = my_getpwnam(*argv);
148                         if (uid == -1) {
149                                 fatalError( "unknown user name: %s\n", *argv);
150                         }
151                 }
152         }
153
154         /* Ok, ready to do the deed now */
155         if (argc <= 1) {
156                 fatalError(too_few_args);
157         }
158         while (argc-- > 1) {
159                 if (recursiveAction (*(++argv), recursiveFlag, FALSE, FALSE, 
160                                         fileAction, fileAction, NULL) == FALSE)
161                         exit(FALSE);
162         }
163         exit(TRUE);
164
165   bad_group:
166         fatalError( "unknown group name: %s\n", groupName);
167 }
168
169 /*
170 Local Variables:
171 c-file-style: "linux"
172 c-basic-offset: 4
173 tab-width: 4
174 End:
175 */