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

#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <strings.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <libintl.h>
#include <locale.h>
#include <libdscp.h>

#if     !defined(TEXT_DOMAIN)
#define TEXT_DOMAIN "SYS_TEST"
#endif

#define OPT_SP          1
#define OPT_DOMAIN      2

static void     usage(void);
static void     parse_options(int, char **, int *);
static int      get_address(int, char *);
static void     trace(char *, ...);
static void     err(char *, ...);
static char     *dscp_strerror(int);

static int      verbose = 0;

int
main(int argc, char **argv)
{
        int     options;
        char    saddr[INET_ADDRSTRLEN];
        char    daddr[INET_ADDRSTRLEN];

        (void) setlocale(LC_ALL, "");
        (void) textdomain(TEXT_DOMAIN);

        parse_options(argc, argv, &options);

        /*
         * Get the desired IP addresses.
         */
        if ((options & OPT_SP) != 0) {
                trace(gettext("Looking up SP address...\n"));
                if (get_address(DSCP_ADDR_REMOTE, saddr) < 0) {
                        err(gettext("SP Address lookup failed. Aborting.\n"));
                        exit(-1);
                }
        }
        if ((options & OPT_DOMAIN) != 0) {
                trace(gettext("Looking up domain address...\n"));
                if (get_address(DSCP_ADDR_LOCAL, daddr) < 0) {
                        err(gettext("Domain Address lookup failed. "
                            "Aborting.\n"));
                        exit(-1);
                }
        }

        /*
         * Print the IP addresses.
         */
        if (options == OPT_SP) {
                (void) printf("%s\n", saddr);
        } else if (options == OPT_DOMAIN) {
                (void) printf("%s\n", daddr);
        } else {
                (void) printf(gettext("Domain Address: %s\n"), daddr);
                (void) printf(gettext("SP Address: %s\n"), saddr);
        }

        return (0);
}

/*
 * parse_options()
 *
 *      Parse the commandline options.
 */
static void
parse_options(int argc, char **argv, int *options)
{
        int             i;
        int             c;
        extern int      opterr;
        extern int      optopt;

        /*
         * Unless told otherwise, print everything.
         */
        *options = (OPT_SP | OPT_DOMAIN);

        /*
         * Skip this routine if no options exist.
         */
        if (argc == 1) {
                return;
        }

        /*
         * Scan for the -h option separately, so that
         * other commandline options are ignored.
         */
        for (i = 1; i < argc; i++) {
                if (strcmp(argv[i], "-h") == 0) {
                        usage();
                        exit(0);
                }
        }

        /*
         * Disable the built-in error reporting, so that
         * error messages can be properly internationalized.
         */
        opterr = 0;

        /*
         * The main loop for parsing options.
         */
        while ((c = getopt(argc, argv, "vsd")) != -1) {
                switch (c) {
                case 'v':
                        verbose = 1;
                        break;
                case 's':
                        if (*options == OPT_DOMAIN) {
                                err(gettext("cannot use -s and -d together"));
                                usage();
                                exit(-1);
                        }
                        *options = OPT_SP;
                        break;
                case 'd':
                        if (*options == OPT_SP) {
                                err(gettext("cannot use -s and -d together"));
                                usage();
                                exit(-1);
                        }
                        *options = OPT_DOMAIN;
                        break;
                default:
                        err(gettext("invalid option -%c"), optopt);
                        usage();
                        exit(-1);
                }
        }
}

/*
 * usage()
 *
 *      Print a brief synopsis of the program's usage.
 */
static void
usage(void)
{
        (void) printf(gettext("Usage:  prtdscp -h \n"));
        (void) printf(gettext("        prtdscp [-v] [-s|-d]\n"));
}

/*
 * get_address()
 *
 *      Retrieve a DSCP IP address using libdscp.
 */
static int
get_address(int which, char *addr)
{
        int                     len;
        int                     error;
        struct sockaddr_in      *sin;
        struct sockaddr         saddr;

        error = dscpAddr(0, which, &saddr, &len);
        if (error != DSCP_OK) {
                trace(gettext("dscpAddr() failed: %s"), dscp_strerror(error));
                return (-1);
        }

        /* LINTED pointer cast may result in improper alignment */
        sin = (struct sockaddr_in *)&saddr;
        if (inet_ntop(AF_INET, &(sin->sin_addr), addr, sizeof (*sin)) == NULL) {
                trace(gettext("address string conversion failed."));
                return (-1);
        }

        return (0);
}

/*
 * trace()
 *
 *      Print tracing statements to stderr when in verbose mode.
 */
/*PRINTFLIKE1*/
static void
trace(char *fmt, ...)
{
        va_list args;

        if (verbose != 0) {
                va_start(args, fmt);
                (void) vfprintf(stderr, fmt, args);
                va_end(args);
        }
}

/*
 * err()
 *
 *      Print error messages to stderr.
 */
/*PRINTFLIKE1*/
static void
err(char *fmt, ...)
{
        va_list args;

        va_start(args, fmt);

        (void) fprintf(stderr, gettext("ERROR: "));
        (void) vfprintf(stderr, fmt, args);
        (void) fprintf(stderr, "\n");

        va_end(args);
}

/*
 * dscp_strerror()
 *
 *      Convert a DSCP error value into a localized string.
 */
static char *
dscp_strerror(int error)
{
        switch (error) {
        case DSCP_OK:
                return (gettext("Success."));
        case DSCP_ERROR:
                return (gettext("General error."));
        case DSCP_ERROR_ALREADY:
                return (gettext("Socket already bound."));
        case DSCP_ERROR_INVALID:
                return (gettext("Invalid arguments."));
        case DSCP_ERROR_NOENT:
                return (gettext("No entry found."));
        case DSCP_ERROR_DB:
                return (gettext("Error reading database."));
        case DSCP_ERROR_REJECT:
                return (gettext("Connection rejected."));
        default:
                return (gettext("Unknown failure."));
        }
}