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

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <unistd.h>
#include <libgen.h>
#include <libintl.h>
#include <libv12n.h>
#include <locale.h>
#include <zone.h>
#include <sys/types.h>
#include <sys/param.h>
#include <uuid/uuid.h>

#if !defined(TEXT_DOMAIN)	/* Should be defined by cc -D */
#define	TEXT_DOMAIN "SYS_TEST"	/* Use this only if it wasn't */
#endif

static char *cmdname;

char *options = "acdpstu";

static void
virtinfo_usage()
{
	(void) fprintf(stderr, gettext("usage: %s [-%s]\n"), cmdname, options);
	exit(1);
}

static char *
virtinfo_cap_to_impl(int cap)
{
	if (cap & V12N_CAP_IMPL_LDOMS)
		return ("LDoms");
	return ("Unknown");
}


int
main(int argc, char *argv[])
{
	int cap;
	int roles;
	size_t rv;
	int opt;
	int errflg = 0;
	int aflg = 0, cflg = 0, dflg = 0, pflg = 0, sflg = 0, tflg = 0,
	    uflg = 0;

	/* Set locale environment variables local definitions */
	(void) setlocale(LC_ALL, "");
	(void) textdomain(TEXT_DOMAIN);

	cmdname = basename(argv[0]);

	/* disable getopt error messages */
	opterr = 0;

	while ((opt = getopt(argc, argv, options)) != EOF) {

		switch (opt) {
		case 'a':
			aflg = 1;
			break;
		case 'c':
			cflg = 1;
			break;
		case 'd':
			dflg = 1;
			break;
		case 'p':
			pflg = 1;
			break;
		case 's':
			sflg = 1;
			break;
		case 't':
			tflg = 1;
			break;
		case 'u':
			uflg = 1;
			break;
		case '?':
		default:
			errflg = 1;
			break;
		}
	}

	if (errflg || optind != argc)
		virtinfo_usage();

	if (aflg) {
		/* aflg -> set all flags except -p */
		cflg = dflg = sflg = tflg = uflg = 1;
	} else if (cflg == 0 && dflg == 0 && sflg == 0 && tflg == 0 &&
	    uflg == 0) {
		/* no flag set, default to '-t' */
		tflg = 1;
	}

	if (getzoneid() != GLOBAL_ZONEID) {
		(void) printf(gettext(
		    "%s can only be run from the global zone\n"), cmdname);
		exit(0);
	}

	cap = v12n_capabilities();
	if ((cap & V12N_CAP_SUPPORTED) == 0) {
		(void) printf(gettext("Virtual machines are not supported\n"));
		exit(0);
	} else if ((cap & V12N_CAP_ENABLED) == 0) {
		(void) printf(gettext(
		    "Virtual machines (%s) are supported but not enabled\n"),
		    virtinfo_cap_to_impl(cap));
		exit(0);
	}

	if (pflg) {
		(void) printf("VERSION 1.0\n");
	}

	if (tflg) {
		char *impl = "", *role = "", *io = "", *service = "",
		    *root = "";

		roles = v12n_domain_roles();

		if (roles == -1 || (cap & V12N_CAP_IMPL_LDOMS) == 0) {
			if (pflg)
				impl = "impl=Unknown";
			else
				impl = "Unknown";
		} else if (pflg) {
			impl = "impl=LDoms";
			role = (roles & V12N_ROLE_CONTROL) ?
			    "|control=true" : "|control=false";
			io = (roles & V12N_ROLE_IO) ?
			    "|io=true" : "|io=false";
			service = (roles & V12N_ROLE_SERVICE) ?
			    "|service=true" : "|service=false";
			root = (roles & V12N_ROLE_ROOT) ?
			    "|root=true" : "|root=false";
		} else {
			impl = "LDoms";
			role = (roles & V12N_ROLE_CONTROL) ?
			    " control" : " guest";
			io = (roles & V12N_ROLE_IO) ?
			    " I/O" : "";
			service = (roles & V12N_ROLE_SERVICE) ?
			    " service" : "";
			root = (roles & V12N_ROLE_ROOT) ?
			    " root" : "";
		}
		(void) printf("%s%s%s%s%s%s\n", pflg ? "DOMAINROLE|" :
		    gettext("Domain role: "), impl, role, io, service, root);
	}

	if (dflg) {
		char domain_name[V12N_NAME_MAX];

		rv = v12n_domain_name(domain_name, sizeof (domain_name));
		if (rv == (size_t)(-1)) {
			(void) strcpy(domain_name, "Unknown");
		}
		(void) printf("%s%s\n", pflg ? "DOMAINNAME|name=" :
		    gettext("Domain name: "), domain_name);
	}

	if (uflg) {
		uuid_t uuid;
		char uuid_str[UUID_PRINTABLE_STRING_LENGTH];

		rv = v12n_domain_uuid(uuid);

		if (rv == (size_t)(-1)) {
			(void) strcpy(uuid_str, "Unknown");
		} else {
			uuid_unparse(uuid, uuid_str);
		}
		(void) printf("%s%s\n", pflg ? "DOMAINUUID|uuid=" :
		    gettext("Domain UUID: "), uuid_str);
	}

	if (cflg) {
		char ctrl_name[V12N_NAME_MAX];

		rv = v12n_ctrl_domain(ctrl_name, sizeof (ctrl_name));

		if (rv == (size_t)(-1)) {
			(void) strcpy(ctrl_name, "Unknown");
		}
		(void) printf("%s%s\n", pflg ? "DOMAINCONTROL|name=" :
		    gettext("Control domain: "), ctrl_name);
	}

	if (sflg) {
		char serial_no[V12N_NAME_MAX];

		rv = v12n_chassis_serialno(serial_no, sizeof (serial_no));

		if (rv == (size_t)(-1)) {
			(void) strcpy(serial_no, "Unknown");
		}
		(void) printf("%s%s\n", pflg ? "DOMAINCHASSIS|serialno=" :
		    gettext("Chassis serial#: "), serial_no);
	}
	return (0);
}