blob: 1a91276221a39afb8c18947a31b93f333ade7f12
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: "\n -L Traverse all symlinks to directories" |
20 | //usage: "\n -H Traverse symlinks on command line only" |
21 | //usage: "\n -P Don't traverse symlinks (default)" |
22 | //usage: IF_DESKTOP( |
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(¶m.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 | ¶m, /* 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 | */ |
222 |