3380677bc8ac908874cd93c0fc8cdcf8a4001a83
[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 tarball for details.
8  */
9
10 /* BB_AUDIT SUSv3 defects - unsupported options -H, -L, and -P. */
11 /* BB_AUDIT GNU defects - unsupported long options. */
12 /* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
13
14 #include "busybox.h"
15
16 static struct bb_uidgid_t ugid = { -1, -1 };
17
18 static int (*chown_func)(const char *, uid_t, gid_t) = chown;
19
20 #define OPT_RECURSE (option_mask32 & 1)
21 #define OPT_NODEREF (option_mask32 & 2)
22 #define OPT_VERBOSE (USE_DESKTOP(option_mask32 & 4) SKIP_DESKTOP(0))
23 #define OPT_CHANGED (USE_DESKTOP(option_mask32 & 8) SKIP_DESKTOP(0))
24 #define OPT_QUIET   (USE_DESKTOP(option_mask32 & 0x10) SKIP_DESKTOP(0))
25 #define OPT_STR     ("Rh" USE_DESKTOP("vcf"))
26
27 /* TODO:
28  * -H if a command line argument is a symbolic link to a directory, traverse it
29  * -L traverse every symbolic link to a directory encountered
30  * -P do not traverse any symbolic links (default)
31  */
32
33 static int fileAction(const char *fileName, struct stat *statbuf,
34                 void ATTRIBUTE_UNUSED *junk, int depth)
35 {
36         // TODO: -H/-L/-P
37         // if (depth ... && S_ISLNK(statbuf->st_mode)) ....
38
39         if (!chown_func(fileName,
40                         (ugid.uid == (uid_t)-1) ? statbuf->st_uid : ugid.uid,
41                         (ugid.gid == (gid_t)-1) ? statbuf->st_gid : ugid.gid)
42         ) {
43                 if (OPT_VERBOSE
44                  || (OPT_CHANGED && (statbuf->st_uid != ugid.uid || statbuf->st_gid != ugid.gid))
45                 ) {
46                         printf("changed ownership of '%s' to %u:%u\n",
47                                         fileName, ugid.uid, ugid.gid);
48                 }
49                 return TRUE;
50         }
51         if (!OPT_QUIET)
52                 bb_perror_msg("%s", fileName);  /* A filename can have % in it... */
53         return FALSE;
54 }
55
56 int chown_main(int argc, char **argv);
57 int chown_main(int argc, char **argv)
58 {
59         int retval = EXIT_SUCCESS;
60
61         opt_complementary = "-2";
62         getopt32(argc, argv, OPT_STR);
63         argv += optind;
64
65         if (OPT_NODEREF) chown_func = lchown;
66
67         parse_chown_usergroup_or_die(&ugid, argv[0]);
68
69         /* Ok, ready to do the deed now */
70         argv++;
71         do {
72                 if (!recursive_action(*argv,
73                                 OPT_RECURSE,    // recurse
74                                 FALSE,          // follow links: TODO: -H/-L/-P
75                                 FALSE,          // depth first
76                                 fileAction,     // file action
77                                 fileAction,     // dir action
78                                 NULL,           // user data
79                                 0)              // depth
80                 ) {
81                         retval = EXIT_FAILURE;
82                 }
83         } while (*++argv);
84
85         return retval;
86 }