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

/*
 *      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);
}