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