Disable any buffering to stdout
[oweals/busybox.git] / coreutils / chown.c
index a611f92f1863961a61ec017311465954e59c3ed5..07d673f28d12668d785e0a0929a775d64cca0019 100644 (file)
-#include "internal.h"
-#include <pwd.h>
-#include <grp.h>
-#include <string.h>
-#include <stdio.h>
+/* vi: set sw=4 ts=4: */
+/*
+ * Mini chown implementation for busybox
+ *
+ * Copyright (C) 1999-2003 by Erik Andersen <andersen@codepoet.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
+ *
+ */
 
-const char     chown_usage[] = "chown [-R] user-name file [file ...]\n"
-"\n\tThe group list is kept in the file /etc/groups.\n\n"
-"\t-R:\tRecursively change the mode of all files and directories\n"
-"\t\tunder the argument directory.";
+/* BB_AUDIT SUSv3 defects - unsupported options -h, -H, -L, and -P. */
+/* BB_AUDIT GNU defects - unsupported options -h, -c, -f, -v, and long options. */
+/* BB_AUDIT Note: gnu chown does not support -H, -L, or -P. */
+/* http://www.opengroup.org/onlinepubs/007904975/utilities/chown.html */
 
-int
-parse_user_name(const char * s, struct FileInfo * i)
-{
-       struct  passwd *        p;
-       char *                          dot = strchr(s, '.');
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include "busybox.h"
 
-       if (! dot )
-               dot = strchr(s, ':');
+/* Don't use lchown for glibc older then 2.1.x */
+#if (__GLIBC__ <= 2) && (__GLIBC_MINOR__ < 1)
+#define lchown chown
+#endif
 
-       if ( dot )
-               *dot = '\0';
+static long uid;
+static long gid;
 
-       if ( (p = getpwnam(s)) == 0 ) {
-               fprintf(stderr, "%s: no such user.\n", s);
-               return 1;
-       }
-       i->userID = p->pw_uid;
+static int (*chown_func)(const char *, uid_t, gid_t) = chown;
 
-       if ( dot ) {
-               struct group *  g = getgrnam(++dot);
-               if ( g == 0 ) {
-                       fprintf(stderr, "%s: no such group.\n", dot);
-                       return 1;
-               }
-               i->groupID = g->gr_gid;
-               i->changeGroupID = 1;
+static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
+{
+       if (chown_func(fileName, uid, (gid == -1) ? statbuf->st_gid : gid) == 0) {
+               chmod(fileName, statbuf->st_mode);
+               return (TRUE);
        }
-       return 0;
+       bb_perror_msg("%s", fileName);  /* Avoid multibyte problems. */
+       return (FALSE);
 }
 
-extern int
-chown_main(struct FileInfo * i, int argc, char * * argv)
+#define FLAG_R 1
+#define FLAG_h 2
+
+int chown_main(int argc, char **argv)
 {
-       int                                     status;
+       int flags;
+       int retval = EXIT_SUCCESS;
+       char *groupName;
+
+       flags = bb_getopt_ulflags(argc, argv, "Rh");
 
-       while ( argc >= 3 && strcmp("-R", argv[1]) == 0 ) {
-               i->recursive = 1;
-               argc--;
-               argv++;
+       if (flags & FLAG_h) chown_func = lchown;
+
+       if (argc - optind < 2) {
+               bb_show_usage();
        }
 
-       if ( (status = parse_user_name(argv[1], i)) != 0 )
-               return status;
+       argv += optind;
+
+       /* First, check if there is a group name here */
+       if ((groupName = strchr(*argv, '.')) == NULL) {
+               groupName = strchr(*argv, ':');
+       }
 
-       argv++;
-       argc--;
+       gid = -1;
+       if (groupName) {
+               *groupName++ = '\0';
+               gid = get_ug_id(groupName, my_getgrnam);
+       }
 
-       i->changeUserID = 1;
-       i->complainInPostProcess = 1;
+       /* Now check for the username */
+       uid = get_ug_id(*argv, my_getpwnam);
 
-       return monadic_main(i, argc, argv);
+       ++argv;
+       
+       /* Ok, ready to do the deed now */
+       do {
+               if (! recursive_action (*argv, (flags & FLAG_R), FALSE, FALSE, 
+                                                               fileAction, fileAction, NULL)) {
+                       retval = EXIT_FAILURE;
+               }
+       } while (*++argv);
+
+       return retval;
 }
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/