Home | History | Annotate | Line # | Download | only in chflags
chflags.c revision 1.8
      1 /*	$NetBSD: chflags.c,v 1.8 2000/08/04 08:06:57 enami Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1992, 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. All advertising materials mentioning features or use of this software
     16  *    must display the following acknowledgement:
     17  *	This product includes software developed by the University of
     18  *	California, Berkeley and its contributors.
     19  * 4. Neither the name of the University nor the names of its contributors
     20  *    may be used to endorse or promote products derived from this software
     21  *    without specific prior written permission.
     22  *
     23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     33  * SUCH DAMAGE.
     34  */
     35 
     36 #include <sys/cdefs.h>
     37 #ifndef lint
     38 __COPYRIGHT("@(#) Copyright (c) 1992, 1993, 1994\n\
     39 	The Regents of the University of California.  All rights reserved.\n");
     40 #endif /* not lint */
     41 
     42 #ifndef lint
     43 #if 0
     44 static char sccsid[] = "from: @(#)chflags.c	8.5 (Berkeley) 4/1/94";
     45 #else
     46 __RCSID("$NetBSD: chflags.c,v 1.8 2000/08/04 08:06:57 enami Exp $");
     47 #endif
     48 #endif /* not lint */
     49 
     50 #include <sys/types.h>
     51 #include <sys/stat.h>
     52 
     53 #include <err.h>
     54 #include <errno.h>
     55 #include <fts.h>
     56 #include <stdio.h>
     57 #include <stdlib.h>
     58 #include <string.h>
     59 #include <unistd.h>
     60 
     61 #include "stat_flags.h"
     62 
     63 int	main __P((int, char **));
     64 void	usage __P((void));
     65 
     66 int
     67 main(argc, argv)
     68 	int argc;
     69 	char *argv[];
     70 {
     71 	FTS *ftsp;
     72 	FTSENT *p;
     73 	u_long clear, set, newflags;
     74 	long val;
     75 	int Hflag, Lflag, Rflag, ch, fts_options, hflag, oct, rval;
     76 	char *flags, *ep;
     77 	int (*change_flags) __P((const char *, u_long));
     78 
     79 	Hflag = Lflag = Rflag = hflag = 0;
     80 	while ((ch = getopt(argc, argv, "HLPRh")) != -1)
     81 		switch (ch) {
     82 		case 'H':
     83 			Hflag = 1;
     84 			Lflag = 0;
     85 			break;
     86 		case 'L':
     87 			Lflag = 1;
     88 			Hflag = 0;
     89 			break;
     90 		case 'P':
     91 			Hflag = Lflag = 0;
     92 			break;
     93 		case 'R':
     94 			Rflag = 1;
     95 			break;
     96 		case 'h':
     97 			hflag = 1;
     98 			break;
     99 		case '?':
    100 		default:
    101 			usage();
    102 		}
    103 	argv += optind;
    104 	argc -= optind;
    105 
    106 	if (argc < 2)
    107 		usage();
    108 
    109 	fts_options = FTS_PHYSICAL;
    110 	if (Rflag) {
    111 		if (Hflag)
    112 			fts_options |= FTS_COMFOLLOW;
    113 		if (Lflag) {
    114 			fts_options &= ~FTS_PHYSICAL;
    115 			fts_options |= FTS_LOGICAL;
    116 		}
    117 	}
    118 
    119 	flags = *argv;
    120 	if (*flags >= '0' && *flags <= '7') {
    121 		errno = 0;
    122 		val = strtol(flags, &ep, 8);
    123 		if (val < 0)
    124 			errno = ERANGE;
    125 		if (errno)
    126 			err(1, "invalid flags: %s", flags);
    127 		if (*ep)
    128 			errx(1, "invalid flags: %s", flags);
    129 		set = val;
    130 		oct = 1;
    131 	} else {
    132 		if (string_to_flags(&flags, &set, &clear))
    133 			errx(1, "invalid flag: %s", flags);
    134 		clear = ~clear;
    135 		oct = 0;
    136 	}
    137 
    138 	if ((ftsp = fts_open(++argv, fts_options, NULL)) == NULL)
    139 		err(1, "fts_open");
    140 
    141 	for (rval = 0; (p = fts_read(ftsp)) != NULL;) {
    142 		change_flags = chflags;
    143 		switch (p->fts_info) {
    144 		case FTS_D:
    145 			if (Rflag)		/* Change it at FTS_DP. */
    146 				continue;
    147 			fts_set(ftsp, p, FTS_SKIP);
    148 			break;
    149 		case FTS_DNR:			/* Warn, chflag, continue. */
    150 			warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
    151 			rval = 1;
    152 			break;
    153 		case FTS_ERR:			/* Warn, continue. */
    154 		case FTS_NS:
    155 			warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
    156 			rval = 1;
    157 			continue;
    158 		case FTS_SL:			/* Ignore unless -h. */
    159 			/*
    160 			 * All symlinks we found while doing a physical
    161 			 * walk end up here.
    162 			 */
    163 			if (!hflag)
    164 				continue;
    165 			/*
    166 			 * Note that if we follow a symlink, fts_info is
    167 			 * not FTS_SL but FTS_F or whatever.  And we should
    168 			 * use lchflags only for FTS_SL and should use chflags
    169 			 * for others.
    170 			 */
    171 			change_flags = lchflags;
    172 			break;
    173 		case FTS_SLNONE:		/* Ignore. */
    174 			/*
    175 			 * The only symlinks that end up here are ones that
    176 			 * don't point to anything.  Note that if we are
    177 			 * doing a phisycal walk, we never reach here unless
    178 			 * we asked to follow explicitly.
    179 			 */
    180 			continue;
    181 		default:
    182 			break;
    183 		}
    184 		if (oct)
    185 			newflags = set;
    186 		else {
    187 			newflags = p->fts_statp->st_flags;
    188 			newflags |= set;
    189 			newflags &= clear;
    190 		}
    191 		if ((*change_flags)(p->fts_accpath, newflags)) {
    192 			warn("%s", p->fts_path);
    193 			rval = 1;
    194 		}
    195 	}
    196 	if (errno)
    197 		err(1, "fts_read");
    198 	exit(rval);
    199 }
    200 
    201 void
    202 usage()
    203 {
    204 
    205 	(void)fprintf(stderr,
    206 	    "usage: chflags [-R [-H | -L | -P]] [-h] flags file ...\n");
    207 	exit(1);
    208 }
    209