gzip: speed up send_bits()
[oweals/busybox.git] / coreutils / chown.c
index 76f3333e3b24aae40b7da34bf879f2d1f7ae88e9..6429fd030067452af8d3dded05f4cb6350f5a365 100644 (file)
+/* vi: set sw=4 ts=4: */
 /*
- * Mini chown/chmod/chgrp implementation for busybox
+ * Mini chown implementation for busybox
  *
- * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
  *
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
-
-#include <stdio.h>
-#include <grp.h>
-#include <pwd.h>
-#include "internal.h"
-
-
-static int uid=-1;
-static int gid=0;
-static int whichApp;
-static char* invocationName=NULL;
-static mode_t mode=7777;
-
-
-#define CHGRP_APP   1
-#define CHOWN_APP   2
-#define CHMOD_APP   3
-
-static const char chgrp_usage[] = "[OPTION]... GROUP FILE...\n"
-    "Change the group membership of each FILE to GROUP.\n"
-    "\n\tOptions:\n" "\t-R\tchange files and directories recursively\n";
-static const char chown_usage[] = "[OPTION]...  OWNER[.[GROUP] FILE...\n"
-    "Change the owner and/or group of each FILE to OWNER and/or GROUP.\n"
-    "\n\tOptions:\n" "\t-R\tchange files and directories recursively\n";
-static const char chmod_usage[] = "[-R] MODE[,MODE]... FILE...\n"
-"Each MODE is one or more of the letters ugoa, one of the symbols +-= and\n"
-"one or more of the letters rwxst.\n\n"
- "\t-R\tchange files and directories recursively.\n";
-
-
-
-static int fileAction(const char *fileName)
+//config:config CHOWN
+//config:      bool "chown (7.2 kb)"
+//config:      default y
+//config:      help
+//config:      chown is used to change the user and/or group ownership
+//config:      of files.
+//config:
+//config:config FEATURE_CHOWN_LONG_OPTIONS
+//config:      bool "Enable long options"
+//config:      default y
+//config:      depends on CHOWN && LONG_OPTS
+
+//applet:IF_CHOWN(APPLET_NOEXEC(chown, chown, BB_DIR_BIN, BB_SUID_DROP, chown))
+
+//kbuild:lib-$(CONFIG_CHOWN) += chown.o
+
+/* BB_AUDIT SUSv3 defects - none? */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
+
+//usage:#define chown_trivial_usage
+//usage:       "[-Rh"IF_DESKTOP("LHPcvf")"]... USER[:[GRP]] FILE..."
+//usage:#define chown_full_usage "\n\n"
+//usage:       "Change the owner and/or group of each FILE to USER and/or GRP\n"
+//usage:     "\n       -R      Recurse"
+//usage:     "\n       -h      Affect symlinks instead of symlink targets"
+//usage:       IF_DESKTOP(
+//usage:     "\n       -L      Traverse all symlinks to directories"
+//usage:     "\n       -H      Traverse symlinks on command line only"
+//usage:     "\n       -P      Don't traverse symlinks (default)"
+//usage:     "\n       -c      List changed files"
+//usage:     "\n       -v      List all files"
+//usage:     "\n       -f      Hide errors"
+//usage:       )
+//usage:
+//usage:#define chown_example_usage
+//usage:       "$ ls -l /tmp/foo\n"
+//usage:       "-r--r--r--    1 andersen andersen        0 Apr 12 18:25 /tmp/foo\n"
+//usage:       "$ chown root /tmp/foo\n"
+//usage:       "$ ls -l /tmp/foo\n"
+//usage:       "-r--r--r--    1 root     andersen        0 Apr 12 18:25 /tmp/foo\n"
+//usage:       "$ chown root.root /tmp/foo\n"
+//usage:       "ls -l /tmp/foo\n"
+//usage:       "-r--r--r--    1 root     root            0 Apr 12 18:25 /tmp/foo\n"
+
+#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)
 {
-    struct stat statBuf;
-    if (stat(fileName, &statBuf) < 0) {
-       switch (whichApp) {
-           case CHGRP_APP:
-           case CHOWN_APP:
-               if (chown(fileName, ((whichApp==CHOWN_APP)? uid: statBuf.st_uid), gid) < 0)
-                   return( TRUE);
-           case CHMOD_APP:
-               if (chmod(fileName, mode))
-                   return( TRUE);
+#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;
        }
-    }
-    perror(fileName);
-    return( FALSE);
+       if (!OPT_QUIET)
+               bb_simple_perror_msg(fileName);
+       return FALSE;
+#undef opt
+#undef param
 }
 
-/* [ugoa]{+|-|=}[rwxstl] */
-int parse_mode( const char* s, mode_t* or, mode_t* and, int* group_execute) 
+int chown_main(int argc UNUSED_PARAM, char **argv)
 {
-       mode_t  mode = 0;
-       mode_t  groups = S_ISVTX;
-       char    type;
-       char    c;
-
-       do {
-               for ( ; ; ) {
-                       switch ( c = *s++ ) {
-                       case '\0':
-                               return (FALSE);
-                       case 'u':
-                               groups |= S_ISUID|S_IRWXU;
-                               continue;
-                       case 'g':
-                               groups |= S_ISGID|S_IRWXG;
-                               continue;
-                       case 'o':
-                               groups |= S_IRWXO;
-                               continue;
-                       case 'a':
-                               groups |= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
-                               continue;
-                       case '+':
-                       case '=':
-                       case '-':
-                               type = c;
-                               if ( groups == S_ISVTX ) /* The default is "all" */
-                                       groups |= S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
-                               break;
-                       default:
-                               if ( c >= '0' && c <= '7' && mode == 0 && groups == S_ISVTX ) {
-                                       *and = 0;
-                                       *or = strtol(--s, 0, 010);
-                                       return (TRUE);
-                               }
-                               else
-                                       return (FALSE);
-                       }
-                       break;
-               }
+       int retval = EXIT_SUCCESS;
+       int opt, flags;
+       struct param_t param;
+
+#if ENABLE_FEATURE_CHOWN_LONG_OPTIONS
+       opt = getopt32long(argv, "^" OPT_STR "\0" "-2", chown_longopts);
+#else
+       opt = getopt32(argv, "^" OPT_STR "\0" "-2");
+#endif
+       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;
+       }
 
-               while ( (c = *s++) != '\0' ) {
-                       switch ( c ) {
-                       case ',':
-                               break;
-                       case 'r':
-                               mode |= S_IRUSR|S_IRGRP|S_IROTH;
-                               continue;
-                       case 'w':
-                               mode |= S_IWUSR|S_IWGRP|S_IWOTH;
-                               continue;
-                       case 'x':
-                               mode |= S_IXUSR|S_IXGRP|S_IXOTH;
-                               continue;
-                       case 's':
-                               if ( group_execute != 0 && (groups & S_IRWXG) ) {
-                                       if ( *group_execute < 0 )
-                                               return (FALSE);
-                                       if ( type != '-' ) {
-                                               mode |= S_IXGRP;
-                                               *group_execute = 1;
-                                       }
-                               }
-                               mode |= S_ISUID|S_ISGID;
-                               continue;
-                       case 'l':
-                               if ( *group_execute > 0 )
-                                       return (FALSE);
-                               if ( type != '-' ) {
-                                       *and &= ~S_IXGRP;
-                                       *group_execute = -1;
-                               }
-                               mode |= S_ISGID;
-                               groups |= S_ISGID;
-                               continue;
-                       case 't':
-                               mode |= S_ISVTX;
-                               continue;
-                       default:
-                               return (FALSE);
-                       }
-                       break;
-               }
-               switch ( type ) {
-               case '=':
-                       *and &= ~(groups);
-                       /* fall through */
-               case '+':
-                       *or |= mode & groups;
-                       break;
-               case '-':
-                       *and &= ~(mode & groups);
-                       *or &= *and;
-                       break;
+       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;
                }
-       } while ( c == ',' );
-       return (TRUE);
-}
-
-
-int chmod_chown_chgrp_main(int argc, char **argv)
-{
-    struct group *grp;
-    struct passwd *pwd;
-    int recursiveFlag=FALSE;
-    char *groupName;
-
-
-    whichApp = (strcmp(*argv, "chown")==0)? CHOWN_APP : (strcmp(*argv, "chmod")==0)? CHMOD_APP : CHGRP_APP; 
-
-    if (argc < 2) {
-       fprintf(stderr, "Usage: %s %s", *argv, 
-               (whichApp==TRUE)? chown_usage : chgrp_usage);
-       exit( FALSE);
-    }
-    invocationName=*argv;
-    argc--;
-    argv++;
-
-    /* Parse options */
-    while (**argv == '-') {
-       while (*++(*argv)) switch (**argv) {
-           case 'R':
-               recursiveFlag = TRUE;
-               break;
-           default:
-               fprintf(stderr, "Unknown option: %c\n", **argv);
-               exit( FALSE);
-       }
-       argc--;
-       argv++;
-    }
-    
-    if ( whichApp == CHMOD_APP ) {
-       /* Find the specified modes */
-       if ( parse_mode(*argv, &orWithMode, &andWithMode, 0) == FALSE ) {
-           fprintf(stderr, "%s: Unknown mode: %s\n", invocationName, *argv);
-           exit( FALSE);
        }
-       mode &= andWithMode;
-       mode |= orWithMode;
-    } else {
-
-       /* Find the selected group */
-       groupName = strchr(*argv, '.');
-       if ( whichApp==TRUE && groupName )
-           *groupName++ = '\0';
-       else
-           groupName = *argv;
-       grp = getgrnam(groupName);
-       if (grp == NULL) {
-           fprintf(stderr, "%s: Unknown group name: %s\n", invocationName, groupName);
-           exit( FALSE);
-       }
-       gid = grp->gr_gid;
-
-       /* Find the selected user (if appropriate)  */
-       if (whichApp==TRUE) {
-           pwd = getpwnam(*argv);
-           if (pwd == NULL) {
-               fprintf(stderr, "%s: Unknown user name: %s\n", invocationName, *argv);
-               exit( FALSE);
-           }
-           uid = pwd->pw_uid;
-       }
-    }
-    
-    /* Ok, ready to do the deed now */
-    if (argc <= 1) {
-       fprintf(stderr, "%s: too few arguments", invocationName);
-       exit( FALSE);
-    }
-    while (argc-- > 1) {
-       argv++;
-       recursiveAction( *argv, recursiveFlag, TRUE, fileAction, fileAction);
-    }
-    exit(TRUE);
+
+       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
+
+*/