Eliminate spurious warning, convert to getopt, and eliminate redundant check.
[oweals/busybox.git] / chmod_chown_chgrp.c
index da3ba8d5020c0f81808d83f697a219fd39cd3916..9714e1ca57ed84a017dfe1294dc23e137f4c93bf 100644 (file)
@@ -1,8 +1,9 @@
+/* vi: set sw=4 ts=4: */
 /*
  * Mini chown/chmod/chgrp implementation for busybox
  *
  *
- * Copyright (C) 1999 by Lineo, inc.
+ * Copyright (C) 1999,2000,2001 by Lineo, inc.
  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
  *
  * This program is free software; you can redistribute it and/or modify
  */
 
 #include <stdio.h>
-#include <grp.h>
-#include <pwd.h>
-#include "internal.h"
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include "busybox.h"
+#define BB_DECLARE_EXTERN
+#define bb_need_invalid_option
+#define bb_need_too_few_args
+#include "messages.c"
 
 
-static uid_t uid=-1;
-static gid_t gid=-1;
+
+static long uid = -1;
+static long gid = -1;
 static int whichApp;
-static char* invocationName=NULL;
-static mode_t mode=0644;
+static char *theMode = NULL;
 
 
 #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, struct stat* statbuf)
+static int fileAction(const char *fileName, struct stat *statbuf, void* junk)
 {
-    switch (whichApp) {
+       switch (whichApp) {
        case CHGRP_APP:
        case CHOWN_APP:
-           if (chown(fileName, (whichApp==CHOWN_APP)? uid : statbuf->st_uid, 
-                       (gid==-1)? statbuf->st_gid : gid) == 0) {
-               return( TRUE);
-           }
-           break;
+       /* Don't use lchown for libc5 or glibc older then 2.1.x */
+#if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
+               if (lchown
+                       (fileName, (whichApp == CHOWN_APP) ? uid : statbuf->st_uid,
+                        (gid == -1) ? statbuf->st_gid : gid) == 0)
+#else
+               if (chown
+                       (fileName, (whichApp == CHOWN_APP) ? uid : statbuf->st_uid,
+                        (gid == -1) ? statbuf->st_gid : gid) == 0)
+#endif
+               {
+                       return (TRUE);
+               }
+               break;
        case CHMOD_APP:
-           if (chmod(fileName, mode) == 0)
-               return( TRUE);
-           break;
-    }
-    perror(fileName);
-    return( FALSE);
+               /* Parse the specified modes */
+               if (parse_mode(theMode, &(statbuf->st_mode)) == FALSE) {
+                       error_msg_and_die( "unknown mode: %s", theMode);
+               }
+               if (chmod(fileName, statbuf->st_mode) == 0)
+                       return (TRUE);
+               break;
+       }
+       perror(fileName);
+       return (FALSE);
 }
 
 int chmod_chown_chgrp_main(int argc, char **argv)
 {
-    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--;
+       int stopIt = FALSE;
+       int recursiveFlag = FALSE;
+       char *groupName=NULL;
+       char *p=NULL;
+
+       whichApp = (applet_name[2]=='o')?           /* chown */
+               CHOWN_APP : (applet_name[2]=='m')?      /* chmod */
+               CHMOD_APP : CHGRP_APP;
+
+       if (argc < 2)
+               show_usage();
        argv++;
-    }
-    
-    if ( whichApp == CHMOD_APP ) {
-       /* Find the specified modes */
-       mode = 0;
-       if ( parse_mode(*argv, &mode) == FALSE ) {
-           fprintf(stderr, "%s: Unknown mode: %s\n", invocationName, *argv);
-           exit( FALSE);
+
+       /* Parse options */
+       while (--argc >= 0 && *argv && (**argv == '-')) {
+               while (stopIt==FALSE && *++(*argv)) {
+                       switch (**argv) {
+                               case 'R':
+                                       recursiveFlag = TRUE;
+                                       break;
+                               default:
+                                       theMode=*argv-1;
+                                       stopIt = TRUE;
+                       }
+               }
+               if (stopIt==TRUE)
+                       break;
+               argv++;
        }
-    } else {
-
-       /* Find the selected group */
-       if ( whichApp==CHGRP_APP ) {
-           groupName = *argv;
-           gid = my_getgrnam(groupName);
-           if (gid == -1)
-               goto bad_group;
-       } else {
-           groupName = strchr(*argv, '.');
-           if (groupName) {
-               *groupName++ = '\0';
-               gid = my_getgrnam(groupName);
-               if (gid == -1)
-                   goto bad_group;
-           } else
-               gid = -1;
+
+       if (argc == 0 || *argv == NULL) {
+               error_msg(too_few_args);
        }
 
+       if (whichApp == CHMOD_APP) {
+               if (theMode==NULL)
+                       theMode = *argv;
+       } else {
 
-       /* Find the selected user (if appropriate)  */
-       if (whichApp==CHOWN_APP) {
-           uid = my_getpwnam(*argv);
-           if (uid == -1) {
-               fprintf(stderr, "%s: Unknown user name: %s\n", invocationName, *argv);
-               exit( FALSE);
-           }
+               /* Find the selected group */
+               if (whichApp == CHGRP_APP) {
+                       groupName = *argv;
+                       gid = strtoul(groupName, &p, 10);       /* maybe it's already numeric */
+                       if (groupName == p)
+                               gid = my_getgrnam(groupName);
+               } else {
+                       groupName = strchr(*argv, '.');
+                       if (groupName == NULL)
+                               groupName = strchr(*argv, ':');
+                       if (groupName) {
+                               *groupName++ = '\0';
+                               gid = strtoul(groupName, &p, 10);
+                               if (groupName == p)
+                                       gid = my_getgrnam(groupName);
+                       } else
+                               gid = -1;
+               }
+
+
+               /* Find the selected user (if appropriate)  */
+               if (whichApp == CHOWN_APP) {
+                       uid = strtoul(*argv, &p, 10);   /* if numeric ... */
+                       if (*argv == p)
+                               uid = my_getpwnam(*argv);
+               }
        }
-    }
-    
-    /* Ok, ready to do the deed now */
-    if (argc <= 1) {
-       fprintf(stderr, "%s: too few arguments", invocationName);
-       exit( FALSE);
-    }
-    while (argc-- > 1) {
-       if (recursiveAction( *(++argv), recursiveFlag, TRUE, FALSE, fileAction, fileAction)==FALSE)
-           exit( FALSE);
-    }
-    exit(TRUE);
-
-bad_group:
-    fprintf(stderr, "%s: Unknown group name: %s\n", invocationName, groupName);
-    exit( FALSE);
-}
 
+       /* Ok, ready to do the deed now */
+       if (argc < 1) {
+               error_msg_and_die(too_few_args);
+       }
+       while (argc-- > 1) {
+               if (recursive_action (*(++argv), recursiveFlag, FALSE, FALSE, 
+                                       fileAction, fileAction, NULL) == FALSE)
+                       return EXIT_FAILURE;
+       }
+       return EXIT_SUCCESS;
 
+}
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/