- add libbb function str_tolower to convert a string to lowercase.
[oweals/busybox.git] / coreutils / chown.c
index 8d6cffa142a524796240e44ef9517036aacf6541..e64a39c3e8ce6f9dc1c8a8610cd817fbfc14ecc2 100644 (file)
+/* vi: set sw=4 ts=4: */
 /*
- * Mini chown/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 tarball for details.
  */
 
-#include <stdio.h>
-#include <grp.h>
-#include <pwd.h>
-#include "internal.h"
+/* BB_AUDIT SUSv3 defects - none? */
+/* BB_AUDIT GNU defects - unsupported long options. */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
 
+#include "busybox.h"
 
-static int uid=-1;
-static int gid=0;
-static int chownApp;
-static char* invocationName=NULL;
+#define OPT_STR     ("Rh" USE_DESKTOP("vcfLHP"))
+#define BIT_RECURSE 1
+#define OPT_RECURSE (option_mask32 & 1)
+#define OPT_NODEREF (option_mask32 & 2)
+#define OPT_VERBOSE (USE_DESKTOP(option_mask32 & 0x04) SKIP_DESKTOP(0))
+#define OPT_CHANGED (USE_DESKTOP(option_mask32 & 0x08) SKIP_DESKTOP(0))
+#define OPT_QUIET   (USE_DESKTOP(option_mask32 & 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(option_mask32 & BIT_TRAVERSE) SKIP_DESKTOP(0))
+/* -H or -L */
+#define BIT_TRAVERSE_TOP (0x20|0x40)
+#define OPT_TRAVERSE_TOP (USE_DESKTOP(option_mask32 & BIT_TRAVERSE_TOP) SKIP_DESKTOP(0))
 
+typedef int (*chown_fptr)(const char *, uid_t, gid_t);
 
-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 struct bb_uidgid_t ugid = { -1, -1 };
 
-
-
-static int fileAction(const char *fileName)
+static int fileAction(const char *fileName, struct stat *statbuf,
+               void *cf, int depth)
 {
-    struct stat statBuf;
-    if ((stat(fileName, &statBuf) < 0) || 
-           (chown(fileName, 
-                  ((chownApp==TRUE)? uid: statBuf.st_uid), 
-                  gid) < 0)) { 
-       perror(fileName);
-       return( FALSE);
-    }
-    return( TRUE);
+       uid_t u = (ugid.uid == (uid_t)-1) ? statbuf->st_uid : ugid.uid;
+       gid_t g = (ugid.gid == (gid_t)-1) ? statbuf->st_gid : ugid.gid;
+
+       if (!((chown_fptr)cf)(fileName, u, g)) {
+               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_perror_msg("%s", fileName);  /* A filename can have % in it... */
+       return FALSE;
 }
 
+int chown_main(int argc, char **argv);
 int chown_main(int argc, char **argv)
 {
-    struct group *grp;
-    struct passwd *pwd;
-    int recursiveFlag=FALSE;
-    char *groupName;
-
-
-    chownApp = (strcmp(*argv, "chown")==0)? TRUE : FALSE;
-
-    if (argc < 2) {
-       fprintf(stderr, "Usage: %s %s", *argv, 
-               (chownApp==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++;
-    }
-    
-    /* Find the selected group */
-    groupName = strchr(*argv, '.');
-    if ( chownApp==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 (chownApp==TRUE) {
-       pwd = getpwnam(*argv);
-       if (pwd == NULL) {
-           fprintf(stderr, "%s: Unknown user name: %s\n", invocationName, *argv);
-           exit( FALSE);
+       int retval = EXIT_SUCCESS;
+       chown_fptr chown_func;
+
+       opt_complementary = "-2";
+       getopt32(argc, argv, OPT_STR);
+       argv += optind;
+
+       /* This matches coreutils behavior (almost - see below) */
+       chown_func = chown;
+       if (OPT_NODEREF
+           /* || (OPT_RECURSE && !OPT_TRAVERSE_TOP): */
+           USE_DESKTOP( || (option_mask32 & (BIT_RECURSE|BIT_TRAVERSE_TOP)) == BIT_RECURSE)
+       ) {
+               chown_func = lchown;
        }
-       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) {
+
+       parse_chown_usergroup_or_die(&ugid, argv[0]);
+
+       /* Ok, ready to do the deed now */
        argv++;
-       recursiveAction( *argv, recursiveFlag, TRUE, fileAction, fileAction);
-    }
-    exit(TRUE);
+       do {
+               char *arg = *argv;
+
+               if (OPT_TRAVERSE_TOP) {
+                       /* resolves symlink (even recursive) */
+                       arg = xmalloc_realpath(arg);
+                       if (!arg)
+                               continue;
+               }
+
+               if (!recursive_action(arg,
+                               (OPT_RECURSE ? ACTION_RECURSE : 0) | /* recurse */
+                               (OPT_TRAVERSE ? ACTION_FOLLOWLINKS : 0),/* follow links if -L */
+                               fileAction,     /* file action */
+                               fileAction,     /* dir action */
+                               chown_func,     /* user data */
+                               0)              /* depth */
+               ) {
+                       retval = EXIT_FAILURE;
+               }
+
+               if (OPT_TRAVERSE_TOP)
+                       free(arg);
+       } 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
+    (cd test1; $t1 $1)
+    (cd test2; $t2 $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"
+# These match coreutils 6.8:
+tst_for_each ""
+tst_for_each "-R"
+tst_for_each "-RP"
+tst_for_each "-RL"
+tst_for_each "-RH"
+tst_for_each "-h"
+tst_for_each "-hR"
+tst_for_each "-hRP"
+# Below: with "chown linkdir" coreutils 6.8 will chown linkdir _target_,
+# we lchown _the link_. I believe we are "more correct".
+#tst_for_each "-hRL"
+#tst_for_each "-hRH"
+
+*/