Another update from Larry:
[oweals/busybox.git] / chmod.c
diff --git a/chmod.c b/chmod.c
index 9f84d0d37e290b14af7e3988e58b5d7e35f98e5a..9139b3f4dfb3832780b2319d82585672d2811387 100644 (file)
--- a/chmod.c
+++ b/chmod.c
@@ -1,7 +1,10 @@
+/* vi: set sw=4 ts=4: */
 /*
- * Mini chmod implementation for busybox
+ * Mini chown/chmod/chgrp implementation for busybox
  *
- * Copyright (C) 1998 by Erik Andersen <andersee@debian.org>
+ *
+ * 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
  * it under the terms of the GNU General Public License as published by
  */
 
 #include <stdio.h>
-#include <grp.h>
-#include <pwd.h>
-#include "internal.h"
-
-
-static mode_t mode=7777;
-
-
-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";
-
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <getopt.h>
+#include "busybox.h"
 
-
-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) || 
-           (chmod(fileName, mode)) < 0) { 
+       if (!parse_mode((char *)junk, &(statbuf->st_mode)))
+               error_msg_and_die("internal error");
+       if (chmod(fileName, statbuf->st_mode) == 0)
+               return (TRUE);
        perror(fileName);
-       return( FALSE);
-    }
-    return( TRUE);
+       return (FALSE);
 }
 
-/* [ugoa]{+|-|=}[rwxstl] */
-int parse_mode( const char* s, mode_t* or, mode_t* and, int* group_execute) 
+int chmod_main(int argc, 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;
+       int i;
+       int opt;
+       int recursiveFlag = FALSE;
+
+       /* do normal option parsing */
+       while ((opt = getopt(argc, argv, "R")) > 0) {
+               switch (opt) {
+                       case 'R':
+                               recursiveFlag = TRUE;
                                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;
+                               show_usage();
                }
+       }
 
-               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;
+       if (argc > optind && argc > 2 && argv[optind]) {
+               /* Parse the specified mode */
+               mode_t mode;
+               if (parse_mode(argv[optind], &mode) == FALSE) {
+                       error_msg_and_die( "unknown mode: %s", argv[optind]);
                }
-               switch ( type ) {
-               case '=':
-                       *and &= ~(groups);
-                       /* fall through */
-               case '+':
-                       *or |= mode & groups;
-                       break;
-               case '-':
-                       *and &= ~(mode & groups);
-                       *or &= *and;
-                       break;
-               }
-       } while ( c == ',' );
-       return (TRUE);
-}
-
-
-int chmod_main(int argc, char **argv)
-{
-    int recursiveFlag=FALSE;
-    mode_t andWithMode = S_ISVTX|S_ISUID|S_ISGID|S_IRWXU|S_IRWXG|S_IRWXO;
-    mode_t orWithMode = 0;
-    char *invocationName=*argv;
-
-    if (argc < 3) {
-       fprintf(stderr, "Usage: %s %s", invocationName, chmod_usage);
-       exit( FALSE);
-    }
-    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);
+       } else {
+               error_msg_and_die(too_few_args);
        }
-       argc--;
-       argv++;
-    }
-    
-    /* Find the selected group */
-    if ( parse_mode(*argv, &orWithMode, &andWithMode, 0) == FALSE ) {
-       fprintf(stderr, "%s: Unknown mode: %s\n", invocationName, *argv);
-       exit( FALSE);
-    }
-    mode &= andWithMode;
-    mode |= orWithMode;
 
-    /* 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);
+       /* Ok, ready to do the deed now */
+       for (i = optind + 1; i < argc; i++) {
+               if (recursive_action (argv[i], recursiveFlag, FALSE, FALSE, fileAction,
+                                       fileAction, argv[optind]) == FALSE) {
+                       return EXIT_FAILURE;
+               }
+       }
+       return EXIT_SUCCESS;
 }
+
+/*
+Local Variables:
+c-file-style: "linux"
+c-basic-offset: 4
+tab-width: 4
+End:
+*/