inetd: fix order of array index check and array access
[oweals/busybox.git] / coreutils / chown.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini chown implementation for busybox
4  *
5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6  *
7  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
8  */
9
10 /* BB_AUDIT SUSv3 defects - none? */
11 /* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
12
13 //usage:#define chown_trivial_usage
14 //usage:       "[-RhLHP"IF_DESKTOP("cvf")"]... OWNER[<.|:>[GROUP]] FILE..."
15 //usage:#define chown_full_usage "\n\n"
16 //usage:       "Change the owner and/or group of each FILE to OWNER and/or GROUP\n"
17 //usage:     "\n        -R      Recurse"
18 //usage:     "\n        -h      Affect symlinks instead of symlink targets"
19 //usage:        IF_DESKTOP(
20 //usage:     "\n        -L      Traverse all symlinks to directories"
21 //usage:     "\n        -H      Traverse symlinks on command line only"
22 //usage:     "\n        -P      Don't traverse symlinks (default)"
23 //usage:     "\n        -c      List changed files"
24 //usage:     "\n        -v      List all files"
25 //usage:     "\n        -f      Hide errors"
26 //usage:        )
27 //usage:
28 //usage:#define chown_example_usage
29 //usage:       "$ ls -l /tmp/foo\n"
30 //usage:       "-r--r--r--    1 andersen andersen        0 Apr 12 18:25 /tmp/foo\n"
31 //usage:       "$ chown root /tmp/foo\n"
32 //usage:       "$ ls -l /tmp/foo\n"
33 //usage:       "-r--r--r--    1 root     andersen        0 Apr 12 18:25 /tmp/foo\n"
34 //usage:       "$ chown root.root /tmp/foo\n"
35 //usage:       "ls -l /tmp/foo\n"
36 //usage:       "-r--r--r--    1 root     root            0 Apr 12 18:25 /tmp/foo\n"
37
38 #include "libbb.h"
39
40 /* This is a NOEXEC applet. Be very careful! */
41
42
43 #define OPT_STR     ("Rh" IF_DESKTOP("vcfLHP"))
44 #define BIT_RECURSE 1
45 #define OPT_RECURSE (opt & 1)
46 #define OPT_NODEREF (opt & 2)
47 #define OPT_VERBOSE (IF_DESKTOP(opt & 0x04) IF_NOT_DESKTOP(0))
48 #define OPT_CHANGED (IF_DESKTOP(opt & 0x08) IF_NOT_DESKTOP(0))
49 #define OPT_QUIET   (IF_DESKTOP(opt & 0x10) IF_NOT_DESKTOP(0))
50 /* POSIX options
51  * -L traverse every symbolic link to a directory encountered
52  * -H if a command line argument is a symbolic link to a directory, traverse it
53  * -P do not traverse any symbolic links (default)
54  * We do not conform to the following:
55  * "Specifying more than one of -H, -L, and -P is not an error.
56  * The last option specified shall determine the behavior of the utility." */
57 /* -L */
58 #define BIT_TRAVERSE 0x20
59 #define OPT_TRAVERSE (IF_DESKTOP(opt & BIT_TRAVERSE) IF_NOT_DESKTOP(0))
60 /* -H or -L */
61 #define BIT_TRAVERSE_TOP (0x20|0x40)
62 #define OPT_TRAVERSE_TOP (IF_DESKTOP(opt & BIT_TRAVERSE_TOP) IF_NOT_DESKTOP(0))
63
64 #if ENABLE_FEATURE_CHOWN_LONG_OPTIONS
65 static const char chown_longopts[] ALIGN1 =
66         "recursive\0"        No_argument   "R"
67         "dereference\0"      No_argument   "\xff"
68         "no-dereference\0"   No_argument   "h"
69 # if ENABLE_DESKTOP
70         "changes\0"          No_argument   "c"
71         "silent\0"           No_argument   "f"
72         "quiet\0"            No_argument   "f"
73         "verbose\0"          No_argument   "v"
74 # endif
75         ;
76 #endif
77
78 typedef int (*chown_fptr)(const char *, uid_t, gid_t);
79
80 struct param_t {
81         struct bb_uidgid_t ugid;
82         chown_fptr chown_func;
83 };
84
85 static int FAST_FUNC fileAction(const char *fileName, struct stat *statbuf,
86                 void *vparam, int depth UNUSED_PARAM)
87 {
88 #define param  (*(struct param_t*)vparam)
89 #define opt option_mask32
90         uid_t u = (param.ugid.uid == (uid_t)-1L) ? statbuf->st_uid : param.ugid.uid;
91         gid_t g = (param.ugid.gid == (gid_t)-1L) ? statbuf->st_gid : param.ugid.gid;
92
93         if (param.chown_func(fileName, u, g) == 0) {
94                 if (OPT_VERBOSE
95                  || (OPT_CHANGED && (statbuf->st_uid != u || statbuf->st_gid != g))
96                 ) {
97                         printf("changed ownership of '%s' to %u:%u\n",
98                                         fileName, (unsigned)u, (unsigned)g);
99                 }
100                 return TRUE;
101         }
102         if (!OPT_QUIET)
103                 bb_simple_perror_msg(fileName);
104         return FALSE;
105 #undef opt
106 #undef param
107 }
108
109 int chown_main(int argc UNUSED_PARAM, char **argv)
110 {
111         int retval = EXIT_SUCCESS;
112         int opt, flags;
113         struct param_t param;
114
115         /* Just -1 might not work: uid_t may be unsigned long */
116         param.ugid.uid = -1L;
117         param.ugid.gid = -1L;
118
119 #if ENABLE_FEATURE_CHOWN_LONG_OPTIONS
120         applet_long_options = chown_longopts;
121 #endif
122         opt_complementary = "-2";
123         opt = getopt32(argv, OPT_STR);
124         argv += optind;
125
126         /* This matches coreutils behavior (almost - see below) */
127         param.chown_func = chown;
128         if (OPT_NODEREF
129         /* || (OPT_RECURSE && !OPT_TRAVERSE_TOP): */
130         IF_DESKTOP( || (opt & (BIT_RECURSE|BIT_TRAVERSE_TOP)) == BIT_RECURSE)
131         ) {
132                 param.chown_func = lchown;
133         }
134
135         flags = ACTION_DEPTHFIRST; /* match coreutils order */
136         if (OPT_RECURSE)
137                 flags |= ACTION_RECURSE;
138         if (OPT_TRAVERSE_TOP)
139                 flags |= ACTION_FOLLOWLINKS_L0; /* -H/-L: follow links on depth 0 */
140         if (OPT_TRAVERSE)
141                 flags |= ACTION_FOLLOWLINKS; /* follow links if -L */
142
143         parse_chown_usergroup_or_die(&param.ugid, argv[0]);
144
145         /* Ok, ready to do the deed now */
146         while (*++argv) {
147                 if (!recursive_action(*argv,
148                                 flags,          /* flags */
149                                 fileAction,     /* file action */
150                                 fileAction,     /* dir action */
151                                 &param,         /* user data */
152                                 0)              /* depth */
153                 ) {
154                         retval = EXIT_FAILURE;
155                 }
156         }
157
158         return retval;
159 }
160
161 /*
162 Testcase. Run in empty directory.
163
164 #!/bin/sh
165 t1="/tmp/busybox chown"
166 t2="/usr/bin/chown"
167 create() {
168     rm -rf $1; mkdir $1
169     (
170     cd $1 || exit 1
171     mkdir dir dir2
172     >up
173     >file
174     >dir/file
175     >dir2/file
176     ln -s dir linkdir
177     ln -s file linkfile
178     ln -s ../up dir/linkup
179     ln -s ../dir2 dir/linkupdir2
180     )
181     chown -R 0:0 $1
182 }
183 tst() {
184     create test1
185     create test2
186     echo "[$1]" >>test1.out
187     echo "[$1]" >>test2.out
188     (cd test1; $t1 $1) >>test1.out 2>&1
189     (cd test2; $t2 $1) >>test2.out 2>&1
190     (cd test1; ls -lnR) >out1
191     (cd test2; ls -lnR) >out2
192     echo "chown $1" >out.diff
193     if ! diff -u out1 out2 >>out.diff; then exit 1; fi
194     rm out.diff
195 }
196 tst_for_each() {
197     tst "$1 1:1 file"
198     tst "$1 1:1 dir"
199     tst "$1 1:1 linkdir"
200     tst "$1 1:1 linkfile"
201 }
202 echo "If script produced 'out.diff' file, then at least one testcase failed"
203 >test1.out
204 >test2.out
205 # These match coreutils 6.8:
206 tst_for_each "-v"
207 tst_for_each "-vR"
208 tst_for_each "-vRP"
209 tst_for_each "-vRL"
210 tst_for_each "-vRH"
211 tst_for_each "-vh"
212 tst_for_each "-vhR"
213 tst_for_each "-vhRP"
214 tst_for_each "-vhRL"
215 tst_for_each "-vhRH"
216 # Fix `name' in coreutils output
217 sed 's/`/'"'"'/g' -i test2.out
218 # Compare us with coreutils output
219 diff -u test1.out test2.out
220
221 */