InfoMagic Source Code 1993 July/THE_SOURCE_CODE_CD_ROM.iso/bsd_srcs/usr.bin/chflags/chflags.c

C/C++ Source or Header | 1992-05-20 | 5.1 KB | 222 lines

/*
 * Copyright (c) 1992 The Regents of the University of California.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *    This product includes software developed by the University of
 *    California, Berkeley and its contributors.
 * 4. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

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

#ifndef lint
static char sccsid[] = "@(#)chflags.c    5.2 (Berkeley) 3/10/92";
#endif /* not lint */

#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <fts.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct cmdset {
    long clrbits;
    long setbits;
} CMDS;
CMDS cmds;
int retval;

void     err __P((const char *, ...));
void     error __P((char *));
int     getflags __P((CMDS *, int));
void    *setflags __P((char *));
void     usage __P((void));

int
main(argc, argv)
    int argc;
    char *argv[];
{
    register FTS *fts;
    register FTSENT *p;
    register long oflags;
    register int oct;
    register char *flags;
    struct stat sb;
    void *set;
    int ch, rflag;
    char *ep;

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

    if (argc < 2)
        usage();

    flags = *argv;
    if (*flags >= '0' && *flags <= '7') {
        oflags = (int)strtol(flags, &ep, 8);
                if (oflags < 0 || *ep)
                        err("invalid flags: %s", flags);
                oct = 1;
    } else {
        if ((set = setflags(flags)) == NULL)
                        err("invalid flags: %s", flags);
        oct = 0;
    }

    retval = 0;
    if (rflag) {
        if (!(fts = fts_open(++argv,
            oct ? FTS_NOSTAT|FTS_PHYSICAL : FTS_PHYSICAL, 0)))
            err("%s", strerror(errno));
        while (p = fts_read(fts))
            switch(p->fts_info) {
            case FTS_D:
                break;
            case FTS_DNR:
            case FTS_ERR:
            case FTS_NS:
                err("%s: %s", p->fts_path, strerror(errno));
            default:
                if (chflags(p->fts_accpath, oct ? oflags :
                    getflags(set, p->fts_statp->st_flags)))
                    error(p->fts_path);
                break;
            }
        exit(retval);
    }
    if (oct) {
        while (*++argv)
            if (chflags(*argv, oflags))
                error(*argv);
    } else
        while (*++argv)
            if (lstat(*argv, &sb) ||
                chflags(*argv, getflags(set, sb.st_flags)))
                error(*argv);
    exit(retval);
}

/*
 * These are analogous to the setmode/getmode routines in the C library.
 */
void *
setflags(cp)
    char *cp;
{
    register CMDS *fset;
    register char *arg;

    fset = &cmds;
    fset->clrbits = 0;
    fset->setbits = 0;
    while (cp) {
        while ((arg = strsep(&cp, ",")) != NULL && *arg == '\0')
            /* void */;
        if (!strcasecmp(arg, "dump"))
            fset->clrbits |= NODUMP;
        else if (!strcasecmp(arg, "nodump"))
            fset->setbits |= NODUMP;
        else
            return (NULL);
    }
    return (fset);
}

int
getflags(fset, oflags)
    register CMDS *fset;
    register int oflags;
{

    oflags &= ~fset->clrbits;
    oflags |= fset->setbits;
    return (oflags);
}

#if __STDC__
#include <stdarg.h>
#else
#include <varargs.h>
#endif

void
#if __STDC__
err(const char *fmt, ...)
#else
err(fmt, va_alist)
    char *fmt;
        va_dcl
#endif
{
    va_list ap;
#if __STDC__
    va_start(ap, fmt);
#else
    va_start(ap);
#endif
    (void)fprintf(stderr, "chflags: ");
    (void)vfprintf(stderr, fmt, ap);
    va_end(ap);
    (void)fprintf(stderr, "\n");
    exit(1);
    /* NOTREACHED */
}

void
error(name)
    char *name;
{
    (void)fprintf(stderr, "chflags: %s: %s.\n", name, strerror(errno));
    retval = 1;
}

void
usage()
{
    (void)fprintf(stderr, "usage: chflags [-R] flags file ...\n");
    exit(1);
}