/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 *      setlabel - sets a file label.
 */

#include <errno.h>
#include <fcntl.h>
#include <locale.h>
#include <pwd.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <utmp.h>

#include <tsol/label.h>

static int      set_label(char *, char *);
static int      setlabel(char *, bslabel_t *);
static void     usage(void);
static void     m_label_err(const char *, const int);

static char *prog = NULL;

int
main(int argc, char **argv)
{
        int     rc = 0;
        char    *label;

        (void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN)       /* Should be defined by cc -D */
#define TEXT_DOMAIN     "SYS_TEST"      /* Use this only if it weren't */
#endif
        (void) textdomain(TEXT_DOMAIN);

        if ((prog = strrchr(argv[0], '/')) == NULL)
                prog = argv[0];
        else
                prog++;

        if (argc < 3) {
                usage();
                return (2);
        }

        argv++;
        argc--;

        label = *argv;
        argv++;
        argc--;
        while (argc-- > 0) {
                if (set_label(*argv++, label) != 0)
                        rc = 1;
        }

        return (rc);
}

static int
set_label(char *filename, char *label)
{
        int rval = 0;
        int err;
        m_label_t *blabel;

        if ((blabel = m_label_alloc(MAC_LABEL)) == NULL) {
                (void) fprintf(stderr, "setlabel: ");
                perror(filename);
                return (2);
        }
        rval = getlabel(filename, blabel);
        if (rval) {
                (void) fprintf(stderr, "setlabel: ");
                perror(filename);
                return (rval);
        }
        if (!bslvalid(blabel)) {
                (void) fprintf(stderr,
                    gettext("%s: Current label is invalid\n"),
                    filename);
                blabel = NULL;
        }
        if (str_to_label(label, &blabel, MAC_LABEL, L_DEFAULT, &err) == -1) {
                m_label_err(label, err);
        }

        rval = setlabel(filename, blabel);
        if (rval == 0)
                m_label_free(blabel);
        return (rval);
}

static int
setlabel(char *filename, bslabel_t *label)
{
        int     rval;

        rval = setflabel(filename, label);

        if (rval) {
                (void) fprintf(stderr, "setlabel: ");
                perror(filename);
        }
        return (rval);
}

static void
m_label_err(const char *ascii, const int err)
{
        if (errno == EINVAL) {
                switch (err) {
                case M_BAD_STRING:
                        (void) fprintf(stderr,
                            gettext("setlabel: bad string %s\n"), ascii);
                break;
                case M_BAD_LABEL:
                        (void) fprintf(stderr,
                            gettext("setlabel: bad previous label\n"));
                break;
                default:
                        (void) fprintf(stderr,
                            gettext("setlabel: parsing error found in "
                            "\"%s\" at position %d\n"), ascii, err);
                break;
                }
        } else {
                perror("setlabel");
        }
        exit(1);
}
/*
 * usage()
 *
 * This routine is called whenever there is a usage type of error has
 * occured.  For example, when a invalid option has has been specified.
 *
 */
static void
usage(void)
{

        (void) fprintf(stderr, gettext("Usage: \n"));
        (void) fprintf(stderr, gettext(
            "   %s newlabel filename [...] \n"), prog);

}