Disable any buffering to stdout

[oweals/busybox.git] / coreutils / chown.c

diff --git a/coreutils/chown.c b/coreutils/chown.c
index fc0c2424fd0dc369a141a4ea0870c4987eb49c8c..07d673f28d12668d785e0a0929a775d64cca0019 100644 (file)
--- a/coreutils/chown.c
+++ b/coreutils/chown.c
@@ -1,7 +1,8 @@
+/* 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>
+ * 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
@@ -19,107 +20,86 @@
  *
  */
 
-#include <stdio.h>
-#include <grp.h>
-#include <pwd.h>
-#include "internal.h"
+/* 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 */
 
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include "busybox.h"
 
-static int uid=-1;
-static int gid=0;
-static int chownApp;
-static char* invocationName=NULL;
+/* Don't use lchown for glibc older then 2.1.x */
+#if (__GLIBC__ <= 2) && (__GLIBC_MINOR__ < 1)
+#define lchown chown
+#endif
 
+static long uid;
+static long gid;
 
-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";
-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 int (*chown_func)(const char *, uid_t, gid_t) = chown;
 
-
-
-static int fileAction(const char *fileName)
+static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
 {
-    struct stat statBuf;
-    if ((stat(fileName, &statBuf) < 0) || 
-           (chown(fileName, 
-                  ((chownApp==TRUE)? uid: statBuf.st_uid), 
-                  gid) < 0)) { 
-       perror(fileName);
-       return( TRUE);
-    }
-    return( FALSE);
+       if (chown_func(fileName, uid, (gid == -1) ? statbuf->st_gid : gid) == 0) {
+               chmod(fileName, statbuf->st_mode);
+               return (TRUE);
+       }
+       bb_perror_msg("%s", fileName);  /* Avoid multibyte problems. */
+       return (FALSE);
 }
 
+#define FLAG_R 1
+#define FLAG_h 2
+
 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);
-       return( 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);
-               return( FALSE);
+       int flags;
+       int retval = EXIT_SUCCESS;
+       char *groupName;
+
+       flags = bb_getopt_ulflags(argc, argv, "Rh");
+
+       if (flags & FLAG_h) chown_func = lchown;
+
+       if (argc - optind < 2) {
+               bb_show_usage();
+       }
+
+       argv += optind;
+
+       /* First, check if there is a group name here */
+       if ((groupName = strchr(*argv, '.')) == NULL) {
+               groupName = strchr(*argv, ':');
        }
-       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);
-       return( 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);
-           return( FALSE);
+
+       gid = -1;
+       if (groupName) {
+               *groupName++ = '\0';
+               gid = get_ug_id(groupName, my_getgrnam);
        }
-       uid = pwd->pw_uid;
-    }
-
-    /* Ok, ready to do the deed now */
-    if (argc <= 1) {
-       fprintf(stderr, "%s: too few arguments", invocationName);
-       return( FALSE);
-    }
-    while (argc-- > 1) {
-       argv++;
-       if (recursiveFlag==TRUE) 
-           recursiveAction( *argv, TRUE, fileAction, fileAction);
-       else
-           fileAction( *argv);
-    }
-    return(TRUE);
+
+       /* Now check for the username */
+       uid = get_ug_id(*argv, my_getpwnam);
+
+       ++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:
+*/