tsol/plabel/plabel.c

/*
 * 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.
 */

#pragma ident   "%Z%%M% %I% %E% SMI"

/*
 *  plabel - gets process label.
 */
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <locale.h>
#include <procfs.h>
#include <sys/proc.h>
#include <zone.h>

#include <sys/tsol/label_macro.h>

#include <tsol/label.h>

#define s_flag  0x04
#define S_flag  0x08

#define INIT_ALLOC_LEN  1024
#define MAX_ALLOC_NUM   11

static int look(char *);
static int perr(char *);
static void usage(void);

static char procname[64];

static unsigned int opt_flag = 0;
static char *cmd = NULL;

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

    (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 ((cmd = strrchr(argv[0], '/')) == NULL)
        cmd = argv[0];
    else
        cmd++;

    /* Error if labeling is not active. */
    if (!is_system_labeled()) {
        (void) fprintf(stderr,
            gettext("%s: Trusted Extensions must be enabled\n"), cmd);
        return (1);
    }

    while ((opt = getopt(argc, argv, "sS")) != EOF) {
        switch (opt) {
        case 's':
            if (opt_flag & (s_flag | S_flag)) {
                usage();
                return (1);
            }
            opt_flag |= s_flag;
            break;

        case 'S':
            if (opt_flag & (s_flag | S_flag)) {
                usage();
                return (1);
            }
            opt_flag |= S_flag;
            break;
        default:
            usage();
            return (1);
        }
    }

    argc -= optind;
    argv += optind;
    if (argc == 0) {
        char pid[11]; /* 32 bit pids go to 4294967295 plus a NUL */

        (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);
}