Home | History | Annotate | Line # | Download | only in chmod
chmod.c revision 1.36.8.1
      1 /* $NetBSD: chmod.c,v 1.36.8.1 2012/11/20 02:57:28 tls Exp $ */
      2 
      3 /*
      4  * Copyright (c) 1989, 1993, 1994
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 #ifndef lint
     34 __COPYRIGHT(
     35 "@(#) Copyright (c) 1989, 1993, 1994\
     36  The Regents of the University of California.  All rights reserved.");
     37 #endif /* not lint */
     38 
     39 #ifndef lint
     40 #if 0
     41 static char sccsid[] = "@(#)chmod.c	8.8 (Berkeley) 4/1/94";
     42 #else
     43 __RCSID("$NetBSD: chmod.c,v 1.36.8.1 2012/11/20 02:57:28 tls Exp $");
     44 #endif
     45 #endif /* not lint */
     46 
     47 #include <sys/param.h>
     48 #include <sys/stat.h>
     49 #include <sys/types.h>
     50 
     51 #include <err.h>
     52 #include <errno.h>
     53 #include <fts.h>
     54 #include <limits.h>
     55 #include <locale.h>
     56 #include <stdio.h>
     57 #include <stdlib.h>
     58 #include <string.h>
     59 #include <unistd.h>
     60 #include <getopt.h>
     61 
     62 __dead static void	usage(void);
     63 
     64 struct option chmod_longopts[] = {
     65 	{ "reference",		required_argument,	0,
     66 						1 },
     67 	{ NULL,			0,			0,
     68 						0 },
     69 };
     70 
     71 int
     72 main(int argc, char *argv[])
     73 {
     74 	FTS *ftsp;
     75 	FTSENT *p;
     76 	void *set;
     77 	mode_t mval;
     78 	int Hflag, Lflag, Rflag, ch, fflag, fts_options, hflag, rval;
     79 	char *mode, *reference;
     80 	int (*change_mode)(const char *, mode_t);
     81 
     82 	setprogname(argv[0]);
     83 	(void)setlocale(LC_ALL, "");
     84 
     85 	Hflag = Lflag = Rflag = fflag = hflag = 0;
     86 	reference = NULL;
     87 	while ((ch = getopt_long(argc, argv, "HLPRXfghorstuwx",
     88 	    chmod_longopts, NULL)) != -1)
     89 		switch (ch) {
     90 		case 1:
     91 			reference = optarg;
     92 			break;
     93 		case 'H':
     94 			Hflag = 1;
     95 			Lflag = 0;
     96 			break;
     97 		case 'L':
     98 			Lflag = 1;
     99 			Hflag = 0;
    100 			break;
    101 		case 'P':
    102 			Hflag = Lflag = 0;
    103 			break;
    104 		case 'R':
    105 			Rflag = 1;
    106 			break;
    107 		case 'f':
    108 			fflag = 1;
    109 			break;
    110 		case 'h':
    111 			/*
    112 			 * In System V the -h option causes chmod to
    113 			 * change the mode of the symbolic link.
    114 			 * 4.4BSD's symbolic links didn't have modes,
    115 			 * so it was an undocumented noop.  In NetBSD
    116 			 * 1.3, lchmod(2) is introduced and this
    117 			 * option does real work.
    118 			 */
    119 			hflag = 1;
    120 			break;
    121 		/*
    122 		 * XXX
    123 		 * "-[rwx]" are valid mode commands.  If they are the entire
    124 		 * argument, getopt has moved past them, so decrement optind.
    125 		 * Regardless, we're done argument processing.
    126 		 */
    127 		case 'g': case 'o': case 'r': case 's':
    128 		case 't': case 'u': case 'w': case 'X': case 'x':
    129 			if (argv[optind - 1][0] == '-' &&
    130 			    argv[optind - 1][1] == ch &&
    131 			    argv[optind - 1][2] == '\0')
    132 				--optind;
    133 			goto done;
    134 		case '?':
    135 		default:
    136 			usage();
    137 		}
    138 done:	argv += optind;
    139 	argc -= optind;
    140 
    141 	if (argc == 0 || (argc == 1 && reference == NULL))
    142 		usage();
    143 
    144 	fts_options = FTS_PHYSICAL;
    145 	if (Rflag) {
    146 		if (hflag) {
    147 			errx(EXIT_FAILURE,
    148 		"the -R and -h options may not be specified together.");
    149 			/* NOTREACHED */
    150 		}
    151 		if (Hflag)
    152 			fts_options |= FTS_COMFOLLOW;
    153 		if (Lflag) {
    154 			fts_options &= ~FTS_PHYSICAL;
    155 			fts_options |= FTS_LOGICAL;
    156 		}
    157 	} else if (!hflag)
    158 		fts_options |= FTS_COMFOLLOW;
    159 	if (hflag)
    160 		change_mode = lchmod;
    161 	else
    162 		change_mode = chmod;
    163 
    164 	if (reference == NULL) {
    165 		mode = *argv++;
    166 		if ((set = setmode(mode)) == NULL) {
    167 			err(EXIT_FAILURE, "Cannot set file mode `%s'", mode);
    168 			/* NOTREACHED */
    169 		}
    170 		mval = 0;
    171 	} else {
    172 		struct stat st;
    173 
    174 		if (stat(reference, &st) == -1)
    175 			err(EXIT_FAILURE, "Cannot stat `%s'", reference);
    176 		mval = st.st_mode;
    177 		set = NULL;
    178 	}
    179 
    180 	if ((ftsp = fts_open(argv, fts_options, 0)) == NULL) {
    181 		err(EXIT_FAILURE, "fts_open");
    182 		/* NOTREACHED */
    183 	}
    184 	for (rval = 0; (p = fts_read(ftsp)) != NULL;) {
    185 		switch (p->fts_info) {
    186 		case FTS_D:
    187 			if (!Rflag)
    188 				(void)fts_set(ftsp, p, FTS_SKIP);
    189 			break;
    190 		case FTS_DNR:			/* Warn, chmod, continue. */
    191 			warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
    192 			rval = 1;
    193 			break;
    194 		case FTS_DP:			/* Already changed at FTS_D. */
    195 			continue;
    196 		case FTS_ERR:			/* Warn, continue. */
    197 		case FTS_NS:
    198 			warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
    199 			rval = 1;
    200 			continue;
    201 		case FTS_SL:			/* Ignore. */
    202 		case FTS_SLNONE:
    203 			/*
    204 			 * The only symlinks that end up here are ones that
    205 			 * don't point to anything and ones that we found
    206 			 * doing a physical walk.
    207 			 */
    208 			if (!hflag)
    209 				continue;
    210 			/* else */
    211 			/* FALLTHROUGH */
    212 		default:
    213 			break;
    214 		}
    215 		if ((*change_mode)(p->fts_accpath,
    216 		    set ? getmode(set, p->fts_statp->st_mode) : mval)
    217 		    && !fflag) {
    218 			warn("%s", p->fts_path);
    219 			rval = 1;
    220 		}
    221 	}
    222 	if (errno) {
    223 		err(EXIT_FAILURE, "fts_read");
    224 		/* NOTREACHED */
    225 	}
    226 	exit(rval);
    227 	/* NOTREACHED */
    228 }
    229 
    230 static void
    231 usage(void)
    232 {
    233 	(void)fprintf(stderr,
    234 	    "Usage: %s [-R [-H | -L | -P]] [-fh] mode file ...\n"
    235 	    "\t%s [-R [-H | -L | -P]] [-fh] --reference=rfile file ...\n",
    236 	    getprogname(), getprogname());
    237 	exit(1);
    238 	/* NOTREACHED */
    239 }
    240