tsol/plabel/plabel.c

	plabel.c revision f875b4ebb1dd9fdbeb043557cab38ab3bf7f6e01
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * CDDL HEADER START
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * The contents of this file are subject to the terms of the
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Common Development and Distribution License (the "License").
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * You may not use this file except in compliance with the License.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * or http://www.opensolaris.org/os/licensing.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * See the License for the specific language governing permissions
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * and limitations under the License.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * When distributing Covered Code, include this CDDL HEADER in each
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * If applicable, add the following below this CDDL HEADER, with the
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * fields enclosed by brackets "[]" replaced with your own identifying
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * information: Portions Copyright [yyyy] [name of copyright owner]
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * CDDL HEADER END
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross * Use is subject to license terms.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#pragma ident   "%Z%%M% %I% %E% SMI"
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross/*
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross *  plabel - gets process label.
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <stdio.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <stdlib.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <errno.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <unistd.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <fcntl.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <string.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <locale.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <procfs.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <sys/proc.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <zone.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <sys/tsol/label_macro.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#include <tsol/label.h>
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#define s_flag  0x04
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#define S_flag  0x08
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#define INIT_ALLOC_LEN  1024
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#define MAX_ALLOC_NUM   11
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic int look(char *);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic int perr(char *);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic void usage(void);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic char procname[64];
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic unsigned int opt_flag = 0;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossstatic char *cmd = NULL;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossint
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Rossmain(int argc, char **argv)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross{
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    int err, rc = 0;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    int opt;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    (void) setlocale(LC_ALL, "");
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#if !defined(TEXT_DOMAIN)   /* Should be defined by cc -D */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#define TEXT_DOMAIN "SYS_TEST"  /* Use this only if it weren't */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross#endif
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    (void) textdomain(TEXT_DOMAIN);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    if ((cmd = strrchr(argv[0], '/')) == NULL)
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        cmd = argv[0];
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    else
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        cmd++;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    /* Error if labeling is not active. */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    if (!is_system_labeled()) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        (void) fprintf(stderr,
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            gettext("%s: Trusted Extensions must be enabled\n"), cmd);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        return (1);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    while ((opt = getopt(argc, argv, "sS")) != EOF) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        switch (opt) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        case 's':
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            if (opt_flag & (s_flag | S_flag)) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross                usage();
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross                return (1);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            opt_flag |= s_flag;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            break;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        case 'S':
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            if (opt_flag & (s_flag | S_flag)) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross                usage();
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross                return (1);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            opt_flag |= S_flag;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            break;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        default:
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            usage();
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross            return (1);
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    }
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    argc -= optind;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    argv += optind;
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross    if (argc == 0) {
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross        char pid[11]; /* 32 bit pids go to 4294967295 plus a NUL */
a90cf9f29973990687fa61de9f1f6ea22e924e40Gordon Ross
        (void) sprintf(pid, "%d", (int)getpid());
        rc = look(pid);
    } else {
        while (argc-- > 0) {
            err = look(*argv++);
            if (rc == 0)
                rc = err;
        }
    }
    return (rc);
}

static int
look(char *arg)
{
    int fd;
    m_label_t *plabel;
    psinfo_t info;      /* process information from /proc */
    char *str;
    int wordlen = DEF_NAMES;

    if (opt_flag == S_flag)
        wordlen = LONG_NAMES;
    else if (opt_flag == s_flag)
        wordlen = SHORT_NAMES;

    if (strchr(arg, '/') != NULL)
        (void) strncpy(procname, arg, sizeof (procname));
    else {
        (void) strcpy(procname, "/proc/");
        (void) strncat(procname, arg,
            sizeof (procname) - strlen(procname));
    }
    (void) strlcat(procname, "/psinfo", sizeof (procname)
        - strlen(procname));

    /*
     * Open the process to be examined.
     */
retry:
    if ((fd = open(procname, O_RDONLY)) < 0) {
        /*
         * Make clean message for non-existent process.
         */
        if (errno == ENOENT) {
            errno = ESRCH;
            perror(arg);
            return (1);
        }
        return (perr(NULL));
    }


    /*
     * Get the info structure for the process and close quickly.
     */
    if (read(fd, &info, sizeof (info)) < 0) {
        int saverr = errno;

        (void) close(fd);
        if (saverr == EAGAIN)
            goto retry;
        if (saverr != ENOENT)
            perror(arg);
        return (1);
    }
    (void) close(fd);

    if (info.pr_lwp.pr_state == 0)  /* can't happen? */
        return (1);

    if ((plabel = getzonelabelbyid(info.pr_zoneid)) == NULL) {
        return (1);
    }

    /*
     * The process label for global zone is admin_high
     */
    if (info.pr_zoneid == GLOBAL_ZONEID) {
        _BSLHIGH(plabel);
    }

    if (label_to_str(plabel, &str, M_LABEL, wordlen) != 0) {
        perror(arg);
        return (2);
    }
    (void) printf("%s\n", str);
    m_label_free(plabel);
    free(str);
    return (0);
}


/*
 * 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, "Usage: \n");
    (void) fprintf(stderr,
        gettext("   %s [pid ...]    \n"), cmd);
    (void) fprintf(stderr,
        gettext("   %s -s  [pid ...] \n"), cmd);
    (void) fprintf(stderr,
        gettext("   %s -S  [pid ...] \n"), cmd);
}

static int
perr(char *s) {

    if (s)
        (void) fprintf(stderr, "%s: ", procname);
    else
        s = procname;
    perror(s);
    return (1);
}