source: MondoRescue/branches/3.2/mindi-busybox/coreutils/chown.c@ 3186

/* 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 source tree.
 */

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

#include "libbb.h"

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


#define OPT_STR     ("Rh" IF_DESKTOP("vcfLHP"))
#define BIT_RECURSE 1
#define OPT_RECURSE (opt & 1)
#define OPT_NODEREF (opt & 2)
#define OPT_VERBOSE (IF_DESKTOP(opt & 0x04) IF_NOT_DESKTOP(0))
#define OPT_CHANGED (IF_DESKTOP(opt & 0x08) IF_NOT_DESKTOP(0))
#define OPT_QUIET   (IF_DESKTOP(opt & 0x10) IF_NOT_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 (IF_DESKTOP(opt & BIT_TRAVERSE) IF_NOT_DESKTOP(0))
/* -H or -L */
#define BIT_TRAVERSE_TOP (0x20|0x40)
#define OPT_TRAVERSE_TOP (IF_DESKTOP(opt & BIT_TRAVERSE_TOP) IF_NOT_DESKTOP(0))

#if ENABLE_FEATURE_CHOWN_LONG_OPTIONS
static const char chown_longopts[] ALIGN1 =
    "recursive\0"        No_argument   "R"
    "dereference\0"      No_argument   "\xff"
    "no-dereference\0"   No_argument   "h"
# if ENABLE_DESKTOP
    "changes\0"          No_argument   "c"
    "silent\0"           No_argument   "f"
    "quiet\0"            No_argument   "f"
    "verbose\0"          No_argument   "v"
# endif
    ;
#endif

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)-1L) ? statbuf->st_uid : param.ugid.uid;
    gid_t g = (param.ugid.gid == (gid_t)-1L) ? 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);
    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;

    /* Just -1 might not work: uid_t may be unsigned long */
    param.ugid.uid = -1L;
    param.ugid.gid = -1L;

#if ENABLE_FEATURE_CHOWN_LONG_OPTIONS
    applet_long_options = chown_longopts;
#endif
    opt_complementary = "-2";
    opt = getopt32(argv, OPT_STR);
    argv += optind;

    /* This matches coreutils behavior (almost - see below) */
    param.chown_func = chown;
    if (OPT_NODEREF
        /* || (OPT_RECURSE && !OPT_TRAVERSE_TOP): */
        IF_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 */
    while (*++argv) {
        if (!recursive_action(*argv,
                flags,          /* flags */
                fileAction,     /* file action */
                fileAction,     /* dir action */
                &param,         /* user data */
                0)              /* depth */
        ) {
            retval = EXIT_FAILURE;
        }
    }

    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

*/