busybox/coreutils/chown.c

/* vi: set sw=4 ts=4: */
/*
 * Mini chown implementation for busybox
 *
 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */

/* BB_AUDIT SUSv3 defects - none? */
/* BB_AUDIT GNU defects - unsupported long options. */
/* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */

#include "libbb.h"

/* This is a NOEXEC applet. Be very careful! */


#define OPT_STR     ("Rh" USE_DESKTOP("vcfLHP"))
#define BIT_RECURSE 1
#define OPT_RECURSE (opt & 1)
#define OPT_NODEREF (opt & 2)
#define OPT_VERBOSE (USE_DESKTOP(opt & 0x04) SKIP_DESKTOP(0))
#define OPT_CHANGED (USE_DESKTOP(opt & 0x08) SKIP_DESKTOP(0))
#define OPT_QUIET   (USE_DESKTOP(opt & 0x10) SKIP_DESKTOP(0))
/* POSIX options
 * -L traverse every symbolic link to a directory encountered
 * -H if a command line argument is a symbolic link to a directory, traverse it
 * -P do not traverse any symbolic links (default)
 * We do not conform to the following:
 * "Specifying more than one of -H, -L, and -P is not an error.
 * The last option specified shall determine the behavior of the utility." */
/* -L */
#define BIT_TRAVERSE 0x20
#define OPT_TRAVERSE (USE_DESKTOP(opt & BIT_TRAVERSE) SKIP_DESKTOP(0))
/* -H or -L */
#define BIT_TRAVERSE_TOP (0x20|0x40)
#define OPT_TRAVERSE_TOP (USE_DESKTOP(opt & BIT_TRAVERSE_TOP) SKIP_DESKTOP(0))

typedef int (*chown_fptr)(const char *, uid_t, gid_t);

struct param_t {
	struct bb_uidgid_t ugid;
	chown_fptr chown_func;
};

static int FAST_FUNC fileAction(const char *fileName, struct stat *statbuf,
		void *vparam, int depth UNUSED_PARAM)
{
#define param  (*(struct param_t*)vparam)
#define opt option_mask32
	uid_t u = (param.ugid.uid == (uid_t)-1) ? statbuf->st_uid : param.ugid.uid;
	gid_t g = (param.ugid.gid == (gid_t)-1) ? statbuf->st_gid : param.ugid.gid;

	if (param.chown_func(fileName, u, g) == 0) {
		if (OPT_VERBOSE
		 || (OPT_CHANGED && (statbuf->st_uid != u || statbuf->st_gid != g))
		) {
			printf("changed ownership of '%s' to %u:%u\n",
					fileName, (unsigned)u, (unsigned)g);
		}
		return TRUE;
	}
	if (!OPT_QUIET)
		bb_simple_perror_msg(fileName);	/* A filename can have % in it... */
	return FALSE;
#undef opt
#undef param
}

int chown_main(int argc UNUSED_PARAM, char **argv)
{
	int retval = EXIT_SUCCESS;
	int opt, flags;
	struct param_t param;

	param.ugid.uid = -1;
	param.ugid.gid = -1;
	param.chown_func = chown;

	opt_complementary = "-2";
	opt = getopt32(argv, OPT_STR);
	argv += optind;

	/* This matches coreutils behavior (almost - see below) */
	if (OPT_NODEREF
	    /* || (OPT_RECURSE && !OPT_TRAVERSE_TOP): */
	    USE_DESKTOP( || (opt & (BIT_RECURSE|BIT_TRAVERSE_TOP)) == BIT_RECURSE)
	) {
		param.chown_func = lchown;
	}

	flags = ACTION_DEPTHFIRST; /* match coreutils order */
	if (OPT_RECURSE)
		flags |= ACTION_RECURSE;
	if (OPT_TRAVERSE_TOP)
		flags |= ACTION_FOLLOWLINKS_L0; /* -H/-L: follow links on depth 0 */
	if (OPT_TRAVERSE)
		flags |= ACTION_FOLLOWLINKS; /* follow links if -L */

	parse_chown_usergroup_or_die(&param.ugid, argv[0]);

	/* Ok, ready to do the deed now */
	argv++;
	do {
		if (!recursive_action(*argv,
				flags,          /* flags */
				fileAction,     /* file action */
				fileAction,     /* dir action */
				&param,         /* user data */
				0)              /* depth */
		) {
			retval = EXIT_FAILURE;
		}
	} while (*++argv);

	return retval;
}

/*
Testcase. Run in empty directory.

#!/bin/sh
t1="/tmp/busybox chown"
t2="/usr/bin/chown"
create() {
    rm -rf $1; mkdir $1
    (
    cd $1 || exit 1
    mkdir dir dir2
    >up
    >file
    >dir/file
    >dir2/file
    ln -s dir linkdir
    ln -s file linkfile
    ln -s ../up dir/linkup
    ln -s ../dir2 dir/linkupdir2
    )
    chown -R 0:0 $1
}
tst() {
    create test1
    create test2
    echo "[$1]" >>test1.out
    echo "[$1]" >>test2.out
    (cd test1; $t1 $1) >>test1.out 2>&1
    (cd test2; $t2 $1) >>test2.out 2>&1
    (cd test1; ls -lnR) >out1
    (cd test2; ls -lnR) >out2
    echo "chown $1" >out.diff
    if ! diff -u out1 out2 >>out.diff; then exit 1; fi
    rm out.diff
}
tst_for_each() {
    tst "$1 1:1 file"
    tst "$1 1:1 dir"
    tst "$1 1:1 linkdir"
    tst "$1 1:1 linkfile"
}
echo "If script produced 'out.diff' file, then at least one testcase failed"
>test1.out
>test2.out
# These match coreutils 6.8:
tst_for_each "-v"
tst_for_each "-vR"
tst_for_each "-vRP"
tst_for_each "-vRL"
tst_for_each "-vRH"
tst_for_each "-vh"
tst_for_each "-vhR"
tst_for_each "-vhRP"
tst_for_each "-vhRL"
tst_for_each "-vhRH"
# Fix `name' in coreutils output
sed 's/`/'"'"'/g' -i test2.out
# Compare us with coreutils output
diff -u test1.out test2.out

*/
1	niro	532	/* 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	niro	816	/* BB_AUDIT SUSv3 defects - none? */
11	niro	532	/* BB_AUDIT GNU defects - unsupported long options. */
12			/* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
13
14	niro	816	#include "libbb.h"
15	niro	532
16	niro	816	/* This is a NOEXEC applet. Be very careful! */
17	niro	532
18
19	niro	816	#define OPT_STR ("Rh" USE_DESKTOP("vcfLHP"))
20			#define BIT_RECURSE 1
21			#define OPT_RECURSE (opt & 1)
22			#define OPT_NODEREF (opt & 2)
23			#define OPT_VERBOSE (USE_DESKTOP(opt & 0x04) SKIP_DESKTOP(0))
24			#define OPT_CHANGED (USE_DESKTOP(opt & 0x08) SKIP_DESKTOP(0))
25			#define OPT_QUIET (USE_DESKTOP(opt & 0x10) SKIP_DESKTOP(0))
26			/* POSIX options
27			* -L traverse every symbolic link to a directory encountered
28	niro	532	* -H if a command line argument is a symbolic link to a directory, traverse it
29			* -P do not traverse any symbolic links (default)
30	niro	816	* We do not conform to the following:
31			* "Specifying more than one of -H, -L, and -P is not an error.
32			* The last option specified shall determine the behavior of the utility." */
33			/* -L */
34			#define BIT_TRAVERSE 0x20
35			#define OPT_TRAVERSE (USE_DESKTOP(opt & BIT_TRAVERSE) SKIP_DESKTOP(0))
36			/* -H or -L */
37			#define BIT_TRAVERSE_TOP (0x20\|0x40)
38			#define OPT_TRAVERSE_TOP (USE_DESKTOP(opt & BIT_TRAVERSE_TOP) SKIP_DESKTOP(0))
39	niro	532
40	niro	816	typedef int (chown_fptr)(const char , uid_t, gid_t);
41
42			struct param_t {
43			struct bb_uidgid_t ugid;
44			chown_fptr chown_func;
45			};
46
47			static int FAST_FUNC fileAction(const char fileName, struct stat statbuf,
48			void *vparam, int depth UNUSED_PARAM)
49	niro	532	{
50	niro	816	#define param ((struct param_t)vparam)
51			#define opt option_mask32
52			uid_t u = (param.ugid.uid == (uid_t)-1) ? statbuf->st_uid : param.ugid.uid;
53			gid_t g = (param.ugid.gid == (gid_t)-1) ? statbuf->st_gid : param.ugid.gid;
54	niro	532
55	niro	816	if (param.chown_func(fileName, u, g) == 0) {
56	niro	532	if (OPT_VERBOSE
57	niro	816	\|\| (OPT_CHANGED && (statbuf->st_uid != u \|\| statbuf->st_gid != g))
58	niro	532	) {
59			printf("changed ownership of '%s' to %u:%u\n",
60	niro	816	fileName, (unsigned)u, (unsigned)g);
61	niro	532	}
62			return TRUE;
63			}
64			if (!OPT_QUIET)
65	niro	816	bb_simple_perror_msg(fileName); /* A filename can have % in it... */
66	niro	532	return FALSE;
67	niro	816	#undef opt
68			#undef param
69	niro	532	}
70
71	niro	816	int chown_main(int argc UNUSED_PARAM, char **argv)
72	niro	532	{
73			int retval = EXIT_SUCCESS;
74	niro	816	int opt, flags;
75			struct param_t param;
76	niro	532
77	niro	816	param.ugid.uid = -1;
78			param.ugid.gid = -1;
79			param.chown_func = chown;
80
81	niro	532	opt_complementary = "-2";
82	niro	816	opt = getopt32(argv, OPT_STR);
83	niro	532	argv += optind;
84
85	niro	816	/* This matches coreutils behavior (almost - see below) */
86			if (OPT_NODEREF
87			/* \|\| (OPT_RECURSE && !OPT_TRAVERSE_TOP): */
88			USE_DESKTOP( \|\| (opt & (BIT_RECURSE\|BIT_TRAVERSE_TOP)) == BIT_RECURSE)
89			) {
90			param.chown_func = lchown;
91			}
92	niro	532
93	niro	816	flags = ACTION_DEPTHFIRST; /* match coreutils order */
94			if (OPT_RECURSE)
95			flags \|= ACTION_RECURSE;
96			if (OPT_TRAVERSE_TOP)
97			flags \|= ACTION_FOLLOWLINKS_L0; /* -H/-L: follow links on depth 0 */
98			if (OPT_TRAVERSE)
99			flags \|= ACTION_FOLLOWLINKS; /* follow links if -L */
100	niro	532
101	niro	816	parse_chown_usergroup_or_die(&param.ugid, argv[0]);
102	niro	532
103			/* Ok, ready to do the deed now */
104			argv++;
105			do {
106			if (!recursive_action(*argv,
107	niro	816	flags, /* flags */
108			fileAction, /* file action */
109			fileAction, /* dir action */
110			&param, /* user data */
111			0) /* depth */
112	niro	532	) {
113			retval = EXIT_FAILURE;
114			}
115			} while (*++argv);
116
117			return retval;
118			}
119	niro	816
120			/*
121			Testcase. Run in empty directory.
122
123			#!/bin/sh
124			t1="/tmp/busybox chown"
125			t2="/usr/bin/chown"
126			create() {
127			rm -rf $1; mkdir $1
128			(
129			cd $1 \|\| exit 1
130			mkdir dir dir2
131			>up
132			>file
133			>dir/file
134			>dir2/file
135			ln -s dir linkdir
136			ln -s file linkfile
137			ln -s ../up dir/linkup
138			ln -s ../dir2 dir/linkupdir2
139			)
140			chown -R 0:0 $1
141			}
142			tst() {
143			create test1
144			create test2
145			echo "[$1]" >>test1.out
146			echo "[$1]" >>test2.out
147			(cd test1; $t1 $1) >>test1.out 2>&1
148			(cd test2; $t2 $1) >>test2.out 2>&1
149			(cd test1; ls -lnR) >out1
150			(cd test2; ls -lnR) >out2
151			echo "chown $1" >out.diff
152			if ! diff -u out1 out2 >>out.diff; then exit 1; fi
153			rm out.diff
154			}
155			tst_for_each() {
156			tst "$1 1:1 file"
157			tst "$1 1:1 dir"
158			tst "$1 1:1 linkdir"
159			tst "$1 1:1 linkfile"
160			}
161			echo "If script produced 'out.diff' file, then at least one testcase failed"
162			>test1.out
163			>test2.out
164			# These match coreutils 6.8:
165			tst_for_each "-v"
166			tst_for_each "-vR"
167			tst_for_each "-vRP"
168			tst_for_each "-vRL"
169			tst_for_each "-vRH"
170			tst_for_each "-vh"
171			tst_for_each "-vhR"
172			tst_for_each "-vhRP"
173			tst_for_each "-vhRL"
174			tst_for_each "-vhRH"
175			# Fix `name' in coreutils output
176			sed 's/`/'"'"'/g' -i test2.out
177			# Compare us with coreutils output
178			diff -u test1.out test2.out
179
180			*/