.Xr botch fix

[unix-history] / usr / src / usr.sbin / chown / chown.c

/*
* Copyright (c) 1988 Regents of the University of California.
* All rights reserved.
*
* %sccs.include.redist.c%
*/

#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1988 Regents of the University of California.\n\
All rights reserved.\n";
#endif /* not lint */

#ifndef lint
static char sccsid[] = "@(#)chown.c     5.15 (Berkeley) %G%";
#endif /* not lint */

#include <sys/param.h>
#include <sys/stat.h>
#include <sys/errno.h>
#include <dirent.h>
#include <pwd.h>
#include <grp.h>
#include <stdio.h>
#include <ctype.h>

int ischown, uid, gid, fflag, rflag, retval;
char *gname, *myname;

main(argc, argv)
       int argc;
       char **argv;
{
       extern char *optarg;
       extern int optind;
       register char *cp;
       int ch;
       char curpath[MAXPATHLEN], *reset, *index(), *rindex();

       myname = (cp = rindex(*argv, '/')) ? cp + 1 : *argv;
       ischown = myname[2] == 'o';

       while ((ch = getopt(argc, argv, "Rf")) != EOF)
               switch((char)ch) {
               case 'R':
                       rflag++;
                       break;
               case 'f':
                       fflag++;
                       break;
               case '?':
               default:
                       usage();
               }
       argv += optind;
       argc -= optind;

       if (argc < 2)
               usage();

       if (ischown) {
               if (cp = index(*argv, '.')) {
                       *cp++ = '\0';
                       setgid(cp);
               }
               else
                       gid = -1;
               setuid(*argv);
       }
       else {
               uid = -1;
               setgid(*argv);
       }

       while (*++argv) {
               if (reset = index(*argv, '/'))
                       (void)getwd(curpath);
               change(*argv);
               if (reset && chdir(curpath)) {
                       if (fflag)
                               exit(0);
                       err(curpath);
                       exit(-1);
               }
       }
       exit(retval);
}

setgid(s)
       register char *s;
{
       struct group *gr, *getgrnam();

       if (!*s) {
               gid = -1;                       /* argument was "uid." */
               return;
       }
       for (gname = s; *s && isdigit(*s); ++s);
       if (!*s)
               gid = atoi(gname);
       else {
               if (!(gr = getgrnam(gname))) {
                       if (fflag)
                               exit(0);
                       (void)fprintf(stderr, "%s: unknown group id: %s\n",
                           myname, gname);
                       exit(-1);
               }
               gid = gr->gr_gid;
       }
}

setuid(s)
       register char *s;
{
       struct passwd *pwd, *getpwnam();
       char *beg;

       if (!*s) {
               uid = -1;                       /* argument was ".gid" */
               return;
       }
       for (beg = s; *s && isdigit(*s); ++s);
       if (!*s)
               uid = atoi(beg);
       else {
               if (!(pwd = getpwnam(beg))) {
                       if (fflag)
                               exit(0);
                       (void)fprintf(stderr,
                           "chown: unknown user id: %s\n", beg);
                       exit(-1);
               }
               uid = pwd->pw_uid;
       }
}

change(file)
       char *file;
{
       register DIR *dirp;
       register struct dirent *dp;
       struct stat buf;

       if (chown(file, uid, gid)) {
               chownerr(file);
               return;
       }
       if (!rflag)
               return;
       if (lstat(file, &buf)) {
               err(file);
               return;
       }
       if ((buf.st_mode & S_IFMT) == S_IFDIR) {
               if (chdir(file) < 0 || !(dirp = opendir("."))) {
                       err(file);
                       return;
               }
               for (dp = readdir(dirp); dp; dp = readdir(dirp)) {
                       if (dp->d_name[0] == '.' && (!dp->d_name[1] ||
                           dp->d_name[1] == '.' && !dp->d_name[2]))
                               continue;
                       change(dp->d_name);
               }
               closedir(dirp);
               if (chdir("..")) {
                       err("..");
                       exit(fflag ? 0 : -1);
               }
       }
}

chownerr(file)
       char *file;
{
       extern int errno;
       static int euid = -1, ngroups = -1;

       /* check for chown without being root */
       if (errno != EPERM || uid != -1 && euid == -1 && (euid = geteuid())) {
               if (fflag)
                       exit(0);
               err(file);
               exit(-1);
       }
       /* check group membership; kernel just returns EPERM */
       if (gid != -1 && ngroups == -1) {
               int groups[NGROUPS];

               ngroups = getgroups(NGROUPS, groups);
               while (--ngroups >= 0 && gid != groups[ngroups]);
               if (ngroups < 0) {
                       if (fflag)
                               exit(0);
                       (void)fprintf(stderr,
                           "%s: you are not a member of group %s.\n",
                           myname, gname);
                       exit(-1);
               }
       }
       err(file);
}

err(s)
       char *s;
{
       extern int errno;
       char *strerror();

       if (fflag)
               return;
       (void)fprintf(stderr, "%s: %s: %s\n", myname, s, strerror(errno));
       retval = -1;
}

usage()
{
       (void)fprintf(stderr, "usage: %s [-Rf] %s file ...\n", myname,
           ischown ? "owner[.group]" : "group");
       exit(-1);
}