rewritten from manual page; add Berkeley specific header
[unix-history] / usr / src / usr.sbin / chown / chgrp.c
index ea53806..f33d495 100644 (file)
@@ -1,9 +1,15 @@
+/*
+ * Copyright (c) 1980 Regents of the University of California.
+ * All rights reserved.  The Berkeley software License Agreement
+ * specifies the terms and conditions for redistribution.
+ */
+
 #ifndef lint
 #ifndef lint
-static char *sccsid = "@(#)chgrp.c     4.4 83/01/20";
-#endif
+static char sccsid[] = "@(#)chgrp.c    5.9 (Berkeley) %G%";
+#endif not lint
 
 /*
 
 /*
- * chgrp gid file ...
+ * chgrp -fR gid file ...
  */
 
 #include <stdio.h>
  */
 
 #include <stdio.h>
@@ -12,12 +18,14 @@ static      char *sccsid = "@(#)chgrp.c     4.4 83/01/20";
 #include <sys/stat.h>
 #include <grp.h>
 #include <pwd.h>
 #include <sys/stat.h>
 #include <grp.h>
 #include <pwd.h>
+#include <sys/dir.h>
 
 struct group *gr, *getgrnam(), *getgrgid();
 struct passwd *getpwuid(), *pwd;
 struct stat stbuf;
 int    gid, uid;
 int    status;
 
 struct group *gr, *getgrnam(), *getgrgid();
 struct passwd *getpwuid(), *pwd;
 struct stat stbuf;
 int    gid, uid;
 int    status;
+int    fflag, rflag;
 /* VARARGS */
 int    fprintf();
 
 /* VARARGS */
 int    fprintf();
 
@@ -26,55 +34,72 @@ main(argc, argv)
        char *argv[];
 {
        register c, i;
        char *argv[];
 {
        register c, i;
+       register char *cp;
 
        argc--, argv++;
 
        argc--, argv++;
+       while (argc > 0 && argv[0][0] == '-') {
+               for (cp = &argv[0][1]; *cp; cp++) switch (*cp) {
+
+               case 'f':
+                       fflag++;
+                       break;
+
+               case 'R':
+                       rflag++;
+                       break;
+
+               default:
+                       fatal(255, "unknown option: %c", *cp);
+                       /*NOTREACHED*/
+               }
+               argv++, argc--;
+       }
        if (argc < 2) {
        if (argc < 2) {
-               printf("usage: chgrp gid file ...\n");
-               exit(2);
+               fprintf(stderr, "usage: chgrp [-fR] gid file ...\n");
+               exit(255);
        }
        uid = getuid();
        if (isnumber(argv[0])) {
                gid = atoi(argv[0]);
                gr = getgrgid(gid);
        }
        uid = getuid();
        if (isnumber(argv[0])) {
                gid = atoi(argv[0]);
                gr = getgrgid(gid);
-               if (uid && gr == NULL) {
-                       printf("%s: unknown group\n", argv[0]);
-                       exit(2);
-               }
+               if (uid && gr == NULL)
+                       fatal(255, "%s: unknown group", argv[0]);
        } else {
                gr = getgrnam(argv[0]);
        } else {
                gr = getgrnam(argv[0]);
-               if (gr == NULL) {
-                       printf("%s: unknown group\n", argv[0]);
-                       exit(2);
-               }
+               if (gr == NULL)
+                       fatal(255, "%s: unknown group", argv[0]);
                gid = gr->gr_gid;
        }
        pwd = getpwuid(uid);
                gid = gr->gr_gid;
        }
        pwd = getpwuid(uid);
-       if (pwd == NULL) {
-               fprintf(stderr, "Who are you?\n");
-               exit(2);
-       }
+       if (pwd == NULL)
+               fatal(255, "Who are you?");
        if (uid && pwd->pw_gid != gid) {
                for (i=0; gr->gr_mem[i]; i++)
                        if (!(strcmp(pwd->pw_name, gr->gr_mem[i])))
                                goto ok;
        if (uid && pwd->pw_gid != gid) {
                for (i=0; gr->gr_mem[i]; i++)
                        if (!(strcmp(pwd->pw_name, gr->gr_mem[i])))
                                goto ok;
-               fprintf(stderr, "You are not a member of the %s group.\n",
-                   argv[0]);
-               exit(2);
+               if (fflag)
+                       exit(0);
+               fatal(255, "You are not a member of the %s group", argv[0]);
        }
 ok:
        for (c = 1; c < argc; c++) {
        }
 ok:
        for (c = 1; c < argc; c++) {
-               if (stat(argv[c], &stbuf)) {
-                       perror(argv[c]);
+               /* do stat for directory arguments */
+               if (lstat(argv[c], &stbuf)) {
+                       status += Perror(argv[c]);
                        continue;
                }
                if (uid && uid != stbuf.st_uid) {
                        continue;
                }
                if (uid && uid != stbuf.st_uid) {
-                       fprintf(stderr, "You are not the owner of %s\n",
-                           argv[c]);
-                       status = 1;
+                       status += error("You are not the owner of %s", argv[c]);
+                       continue;
+               }
+               if (rflag && ((stbuf.st_mode & S_IFMT) == S_IFDIR)) {
+                       status += chownr(argv[c]);
+                       continue;
+               }
+               if (chown(argv[c], -1, gid)) {
+                       status += Perror(argv[c]);
                        continue;
                }
                        continue;
                }
-               if (chown(argv[c], stbuf.st_uid, gid))
-                       perror(argv[c]);
        }
        exit(status);
 }
        }
        exit(status);
 }
@@ -89,3 +114,91 @@ isnumber(s)
                        return (0);
        return (1);
 }
                        return (0);
        return (1);
 }
+
+chownr(dir)
+       char *dir;
+{
+       register DIR *dirp;
+       register struct direct *dp;
+       struct stat st;
+       char savedir[1024];
+       int ecode;
+
+       if (getwd(savedir) == 0)
+               fatal(255, "%s", savedir);
+       /*
+        * Change what we are given before doing its contents.
+        */
+       if (chown(dir, -1, gid) < 0 && Perror(dir))
+               return (1);
+       if (chdir(dir) < 0) {
+               Perror(dir);
+               return (1);
+       }
+       if ((dirp = opendir(".")) == NULL) {
+               Perror(dir);
+               return (1);
+       }
+       dp = readdir(dirp);
+       dp = readdir(dirp); /* read "." and ".." */
+       ecode = 0;
+       for (dp = readdir(dirp); dp != NULL; dp = readdir(dirp)) {
+               if (lstat(dp->d_name, &st) < 0) {
+                       ecode = Perror(dp->d_name);
+                       if (ecode)
+                               break;
+                       continue;
+               }
+               if (uid && uid != st.st_uid) {
+                       ecode = error("You are not the owner of %s",
+                               dp->d_name);
+                       continue;
+               }
+               if ((st.st_mode & S_IFMT) == S_IFDIR) {
+                       ecode = chownr(dp->d_name);
+                       if (ecode)
+                               break;
+                       continue;
+               }
+               if (chown(dp->d_name, -1, gid) < 0 &&
+                   (ecode = Perror(dp->d_name)))
+                       break;
+       }
+       closedir(dirp);
+       if (chdir(savedir) < 0)
+               fatal(255, "can't change back to %s", savedir);
+       return (ecode);
+}
+
+error(fmt, a)
+       char *fmt, *a;
+{
+
+       if (!fflag) {
+               fprintf(stderr, "chgrp: ");
+               fprintf(stderr, fmt, a);
+               putc('\n', stderr);
+       }
+       return (!fflag);
+}
+
+fatal(status, fmt, a)
+       int status;
+       char *fmt, *a;
+{
+
+       fflag = 0;
+       (void) error(fmt, a);
+       exit(status);
+}
+
+Perror(s)
+       char *s;
+{
+
+       if (!fflag) {
+               fprintf(stderr, "chgrp: ");
+               perror(s);
+       }
+       return (!fflag);
+}