LXR busybox/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
  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
  65static 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
  78typedef int (*chown_fptr)(const char *, uid_t, gid_t);
  79
  80struct param_t {
  81        struct bb_uidgid_t ugid;
  82        chown_fptr chown_func;
  83};
  84
  85static 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
 109int 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(&param.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                                &param,         /* user data */
 152                                0)              /* depth */
 153                ) {
 154                        retval = EXIT_FAILURE;
 155                }
 156        }
 157
 158        return retval;
 159}
 160
 161/*
 162Testcase. Run in empty directory.
 163
 164#!/bin/sh
 165t1="/tmp/busybox chown"
 166t2="/usr/bin/chown"
 167create() {
 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}
 183tst() {
 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}
 196tst_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}
 202echo "If script produced 'out.diff' file, then at least one testcase failed"
 203>test1.out
 204>test2.out
 205# These match coreutils 6.8:
 206tst_for_each "-v"
 207tst_for_each "-vR"
 208tst_for_each "-vRP"
 209tst_for_each "-vRL"
 210tst_for_each "-vRH"
 211tst_for_each "-vh"
 212tst_for_each "-vhR"
 213tst_for_each "-vhRP"
 214tst_for_each "-vhRL"
 215tst_for_each "-vhRH"
 216# Fix `name' in coreutils output
 217sed 's/`/'"'"'/g' -i test2.out
 218# Compare us with coreutils output
 219diff -u test1.out test2.out
 220
 221*/
 222