/*
 * 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 2010 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

/*
 * tninfo.c - Trusted network reporting utility
 */
#include <sys/types.h>
#include <errno.h>
#include <stdio.h>
#include <locale.h>
#include <string.h>
#include <stdlib.h>
#include <libtsnet.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <tsol/label.h>
#include <zone.h>

static void usage(void);
static int print_rhtp(const char *);
static int print_rh(const char *);
static int print_mlp(const char *);

int
main(int argc, char *argv[])
{
        int chr;
        int ret = 0; /* return code */

        /* set the locale for only the messages system (all else is clean) */
        (void) setlocale(LC_ALL, "");
#ifndef 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 (argc <= 1)
                usage();

        while ((chr = getopt(argc, argv, "h:m:t:")) != EOF) {
                switch (chr) {
                case 'h':
                        ret |= print_rh(optarg);
                        break;
                case 'm':
                        ret |= print_mlp(optarg);
                        break;
                case 't':
                        ret |= print_rhtp(optarg);
                        break;
                default:
                        usage();
                }
        }

        return (ret);
}

static void
usage(void)
{
        (void) fprintf(stderr, gettext("usage: tninfo [-h host_name] "
            "[-m zone_name] [-t template_name]\n"));
        exit(1);
}

static void
l_to_str(const m_label_t *l, char **str, int ltype)
{
        if (label_to_str(l, str, ltype, DEF_NAMES) != 0)
                *str = strdup(gettext("translation failed"));
}

static int
print_rhtp(const char *rhtp_name)
{
        tsol_tpent_t tp;
        char *str, *str2;
        const m_label_t *l1, *l2;
        int i;

        (void) strlcpy(tp.name, rhtp_name, sizeof (tp.name));

        if (tnrhtp(TNDB_GET, &tp) != 0) {
                if (errno == ENOENT)
                        (void) fprintf(stderr, gettext("tninfo: tnrhtp entry "
                            "%1$s does not exist\n"), tp.name);
                else
                        (void) fprintf(stderr,
                            gettext("tninfo: tnrhtp TNDB_GET(%1$s) failed: "
                            "%2$s\n"), tp.name, strerror(errno));
                return (1);
        }

        (void) printf("=====================================\n");
        (void) printf(gettext("Remote Host Template Table Entries:\n"));

        (void) printf("__________________________\n");
        (void) printf(gettext("template: %s\n"), tp.name);

        switch (tp.host_type) {
        case UNLABELED:
                (void) printf(gettext("host_type: UNLABELED\n"));
                (void) printf(gettext("doi: %d\n"), tp.tp_doi);

                if (tp.tp_mask_unl & TSOL_MSK_DEF_LABEL) {
                        l_to_str(&tp.tp_def_label, &str, M_LABEL);
                        l_to_str(&tp.tp_def_label, &str2, M_INTERNAL);
                        (void) printf(gettext("def_label: %s\nhex: %s\n"),
                            str, str2);
                        free(str);
                        free(str2);
                }

                if (tp.tp_mask_unl & TSOL_MSK_SL_RANGE_TSOL) {
                        (void) printf(gettext("For routing only:\n"));
                        l_to_str(&tp.tp_gw_sl_range.lower_bound,
                            &str, M_LABEL);
                        l_to_str(&tp.tp_gw_sl_range.lower_bound,
                            &str2, M_INTERNAL);
                        (void) printf(gettext("min_sl: %s\nhex: %s\n"),
                            str, str2);
                        free(str);
                        free(str2);

                        l_to_str(&tp.tp_gw_sl_range.upper_bound,
                            &str, M_LABEL);
                        l_to_str(&tp.tp_gw_sl_range.upper_bound,
                            &str2, M_INTERNAL);
                        (void) printf(gettext("max_sl: %s\nhex: %s\n"),
                            str, str2);
                        free(str);
                        free(str2);

                        l1 = (const m_label_t *)&tp.tp_gw_sl_set[0];
                        l2 = (const m_label_t *)&tp.tp_gw_sl_set[NSLS_MAX];
                        for (i = 0; l1 < l2; l1++, i++) {
                                if (label_to_str(l1, &str2, M_INTERNAL,
                                    DEF_NAMES) != 0)
                                        break;
                                l_to_str(l1, &str, M_LABEL);
                                (void) printf(gettext("sl_set[%1$d]: %2$s\n"
                                    "hex: %3$s\n"), i, str, str2);
                                free(str);
                                free(str2);
                        }
                }
                break;

        case SUN_CIPSO:
                (void) printf(gettext("host_type: CIPSO\n"));
                (void) printf(gettext("doi: %d\n"), tp.tp_doi);
                if (tp.tp_mask_cipso & TSOL_MSK_SL_RANGE_TSOL) {
                        l_to_str(&tp.tp_sl_range_cipso.lower_bound,
                            &str, M_LABEL);
                        l_to_str(&tp.tp_sl_range_cipso.lower_bound,
                            &str2, M_INTERNAL);

                        (void) printf(gettext("min_sl: %s\nhex: %s\n"),
                            str, str2);
                        free(str);
                        free(str2);

                        l_to_str(&tp.tp_sl_range_cipso.upper_bound,
                            &str, M_LABEL);
                        l_to_str(&tp.tp_sl_range_cipso.upper_bound,
                            &str2, M_INTERNAL);

                        (void) printf(gettext("max_sl: %s\nhex: %s\n"),
                            str, str2);
                        free(str);
                        free(str2);

                        l1 = (const m_label_t *)&tp.tp_sl_set_cipso[0];
                        l2 = (const m_label_t *)&tp.tp_sl_set_cipso[NSLS_MAX];
                        for (i = 0; l1 < l2; l1++, i++) {
                                if (label_to_str(l1, &str2, M_INTERNAL,
                                    DEF_NAMES) != 0)
                                        break;
                                l_to_str(l1, &str, M_LABEL);

                                (void) printf(gettext("sl_set[%1$d]: %2$s\n"
                                    "hex: %3$s\n"), i, str, str2);
                                free(str);
                                free(str2);
                        }
                }
                break;

        default:
                (void) printf(gettext("unsupported host type: %ld\n"),
                    tp.host_type);
        }
        return (0);
}

static int
print_rh(const char *rh_name)
{
        int herr;
        struct hostent *hp;
        in6_addr_t in6;
        char abuf[INET6_ADDRSTRLEN];
        tsol_rhent_t rhent;

        if ((hp = getipnodebyname(rh_name, AF_INET6,
            AI_ALL | AI_ADDRCONFIG | AI_V4MAPPED, &herr)) == NULL) {
                (void) fprintf(stderr, gettext("tninfo: unknown host or "
                    "invalid literal address: %s\n"), rh_name);
                if (herr == TRY_AGAIN)
                        (void) fprintf(stderr,
                            gettext("\t(try again later)\n"));
                return (1);
        }

        (void) memset(&rhent, 0, sizeof (rhent));
        (void) memcpy(&in6, hp->h_addr, hp->h_length);

        if (IN6_IS_ADDR_V4MAPPED(&in6)) {
                rhent.rh_address.ta_family = AF_INET;
                IN6_V4MAPPED_TO_INADDR(&in6, &rhent.rh_address.ta_addr_v4);
                (void) inet_ntop(AF_INET, &rhent.rh_address.ta_addr_v4, abuf,
                    sizeof (abuf));
        } else {
                rhent.rh_address.ta_family = AF_INET6;
                rhent.rh_address.ta_addr_v6 = in6;
                (void) inet_ntop(AF_INET6, &in6, abuf, sizeof (abuf));
        }

        (void) printf(gettext("IP address= %s\n"), abuf);

        if (tnrh(TNDB_GET, &rhent) != 0) {
                if (errno == ENOENT)
                        (void) fprintf(stderr, gettext("tninfo: tnrhdb entry "
                            "%1$s does not exist\n"), abuf);
                else
                        (void) fprintf(stderr, gettext("tninfo: TNDB_GET(%1$s) "
                            "failed: %2$s\n"), abuf, strerror(errno));
                return (1);
        }

        if (rhent.rh_template[0] != '\0')
                (void) printf(gettext("Template = %.*s\n"), TNTNAMSIZ,
                    rhent.rh_template);
        else
                (void) printf(gettext("No template exists.\n"));

        return (0);
}

static int
iterate_mlps(tsol_mlpent_t *tsme, const char *type)
{
        struct protoent *pe;

        /* get the first entry */
        tsme->tsme_mlp.mlp_ipp = 0;
        tsme->tsme_mlp.mlp_port = 0;
        tsme->tsme_mlp.mlp_port_upper = 0;
        if (tnmlp(TNDB_GET, tsme) == -1) {
                if (errno == ENOENT) {
                        (void) printf(gettext("%s: no entries\n"), type);
                        return (0);
                } else {
                        perror("tnmlp TNDB_GET");
                        return (-1);
                }
        }
        (void) printf("%s: ", type);
        for (;;) {
                (void) printf("%u", tsme->tsme_mlp.mlp_port);
                if (tsme->tsme_mlp.mlp_port != tsme->tsme_mlp.mlp_port_upper)
                        (void) printf("-%u", tsme->tsme_mlp.mlp_port_upper);
                if ((pe = getprotobynumber(tsme->tsme_mlp.mlp_ipp)) == NULL)
                        (void) printf("/%u", tsme->tsme_mlp.mlp_ipp);
                else
                        (void) printf("/%s", pe->p_name);
                if (tsme->tsme_mlp.mlp_ipp == 255) {
                        tsme->tsme_mlp.mlp_port++;
                        tsme->tsme_mlp.mlp_ipp = 0;
                } else {
                        tsme->tsme_mlp.mlp_ipp++;
                }
                if (tnmlp(TNDB_GET, tsme) == -1)
                        break;
                (void) putchar(';');
        }
        (void) putchar('\n');
        return (0);
}

/*
 * Print all of the MLPs for the given zone.
 */
static int
print_mlp(const char *zonename)
{
        tsol_mlpent_t tsme;

        if ((tsme.tsme_zoneid = getzoneidbyname(zonename)) == -1) {
                (void) fprintf(stderr, gettext("tninfo: zone '%s' unknown\n"),
                    zonename);
                return (1);
        }
        tsme.tsme_flags = 0;
        if (iterate_mlps(&tsme, gettext("private")) == -1)
                return (1);
        tsme.tsme_flags = TSOL_MEF_SHARED;
        if (iterate_mlps(&tsme, gettext("shared")) == -1)
                return (1);
        return (0);
}