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

/*
 * Logical Domains System Agent
 */

#include <errno.h>
#include <fcntl.h>
#include <libds.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <synch.h>
#include <thread.h>
#include <unistd.h>
#include <sys/utsname.h>
#include <sys/mdesc.h>
#include <sys/mdesc_impl.h>

#include "ldma.h"
#include "pri.h"

#define	LDMA_MODULE	LDMA_NAME_SYSTEM

#define	LDMA_NVERSIONS	(sizeof (ldma_versions) / sizeof (ds_ver_t))
#define	LDMA_NHANDLERS	(sizeof (ldma_handlers) / sizeof (ldma_msg_handler_t))

static ldm_msg_func_t ldma_sys_get_sysinfo;
static ldm_msg_func_t ldma_sys_get_chassisno;

/* ptr to cached value of chassisno */
static char *ldma_sys_chassisno = NULL;
mutex_t ldma_chassisno_lock = DEFAULTMUTEX;

static ds_ver_t ldma_versions[] = { { 1, 0 } };

static ldma_msg_handler_t ldma_handlers[] = {
	{ LDMA_MSGSYS_GET_SYSINFO,   LDMA_MSGFLG_ACCESS_ANY,
	    ldma_sys_get_sysinfo },
	{ LDMA_MSGSYS_GET_CHASSISNO, LDMA_MSGFLG_ACCESS_ANY,
	    ldma_sys_get_chassisno }
};

ldma_agent_info_t ldma_system_info = {
	LDMA_NAME_SYSTEM,
	ldma_versions, LDMA_NVERSIONS,
	ldma_handlers, LDMA_NHANDLERS
};

/*ARGSUSED*/
static ldma_request_status_t
ldma_sys_get_sysinfo(ds_ver_t *ver, ldma_message_header_t *request,
    size_t request_dlen, ldma_message_header_t **replyp, size_t *reply_dlenp)
{
	ldma_message_header_t *reply;
	struct utsname name;
	size_t syslen, nodlen, rellen, maclen, verlen;
	size_t rlen;
	char *data;
	int status;

	LDMA_DBG("GET_SYSINFO");

	if (request->msg_info != 0 || request_dlen != 0) {
		status = LDMA_REQ_INVALID;
		goto done;
	}

	if (uname(&name) == -1) {
		LDMA_DBG("GET_SYSINFO: uname failed with error %d", errno);
		status = LDMA_REQ_FAILED;
		goto done;
	}

	syslen = strlen(name.sysname) + 1;
	nodlen = strlen(name.nodename) + 1;
	rellen = strlen(name.release) + 1;
	verlen = strlen(name.version) + 1;
	maclen = strlen(name.machine) + 1;

	rlen = syslen + nodlen + rellen + verlen + maclen;

	reply = ldma_alloc_result_msg(request, rlen);

	if (reply == NULL) {
		status = LDMA_REQ_FAILED;
		goto done;
	}

	reply->msg_info = rlen;

	data = LDMA_HDR2DATA(reply);

	(void) strcpy(data, name.sysname);
	data += syslen;

	(void) strcpy(data, name.nodename);
	data += nodlen;

	(void) strcpy(data, name.release);
	data += rellen;

	(void) strcpy(data, name.version);
	data += verlen;

	(void) strcpy(data, name.machine);

	LDMA_DBG("GET_SYSINFO: return info=%u, {%s, %s, %s, %s, %s}", rlen,
	    name.sysname, name.nodename, name.release, name.version,
	    name.machine);

	*replyp = reply;
	*reply_dlenp = rlen;

	return (LDMA_REQ_COMPLETED);

done:
	LDMA_DBG("GET_SYSINFO: return error %d", status);
	return (status);
}

/*
 * Wrapper for MD free: need unused size argument.
 */
/* ARGSUSED */
static void
ldma_md_free(void *buf, size_t n)
{
	free(buf);
}

/*
 * Wrapper for MD init: read PRI MD and invoke md_init_intern.
 */
static md_t *
ldma_md_init()
{
	md_t *mdp;
	uint64_t *buf = NULL;
	uint64_t token;
	ssize_t status;

	if (pri_init() == -1)
		return (NULL);

	status = pri_get(PRI_GET, &token, &buf, malloc, ldma_md_free);
	pri_fini();

	if (status == (ssize_t)(-1))
		return (NULL);

	mdp = md_init_intern(buf, malloc, ldma_md_free);

	return (mdp);
}

/*
 * Wrapper for md_fini.  Allow NULL md ptr and free MD buffer.
 */
static void
ldma_md_fini(void *md)
{
	md_impl_t *mdp = (md_impl_t *)md;

	if (mdp) {
		free(mdp->caddr);
		(void) md_fini(md);
	}
}

static int
ldma_get_chassis_serialno(char **strp)
{
	md_t *mdp;
	mde_cookie_t *component_nodes, rootnode;
	int list_size, ncomponents, num_nodes, i;
	char *component_type, *serialno;
	int rv = 0;

	(void) mutex_lock(&ldma_chassisno_lock);
	if (ldma_sys_chassisno != NULL) {
		*strp = ldma_sys_chassisno;
		(void) mutex_unlock(&ldma_chassisno_lock);
		return (1);
	}

	mdp = ldma_md_init();
	if (mdp == NULL) {
		(void) mutex_unlock(&ldma_chassisno_lock);
		return (0);
	}

	num_nodes = md_node_count(mdp);
	list_size = num_nodes * sizeof (mde_cookie_t);
	component_nodes = malloc(list_size);
	if (component_nodes == NULL) {
		(void) mutex_unlock(&ldma_chassisno_lock);
		ldma_md_fini(mdp);
		return (0);
	}

	rootnode = md_root_node(mdp);

	ncomponents = md_scan_dag(mdp, rootnode, md_find_name(mdp, "component"),
	    md_find_name(mdp, "fwd"), component_nodes);

	for (i = 0; i < ncomponents; i++) {
		if (md_get_prop_str(mdp, component_nodes[i], "type",
		    &component_type))
			continue;
		if (strcmp(component_type, "chassis") != 0)
			continue;
		if (md_get_prop_str(mdp, component_nodes[i],
		    "serial_number", &serialno) == 0) {
			ldma_sys_chassisno = strdup(serialno);
			*strp = ldma_sys_chassisno;
			rv = 1;
			break;
		}
	}
	(void) mutex_unlock(&ldma_chassisno_lock);
	free(component_nodes);
	ldma_md_fini(mdp);
	return (rv);
}

/*ARGSUSED*/
static ldma_request_status_t
ldma_sys_get_chassisno(ds_ver_t *ver, ldma_message_header_t *request,
    size_t request_dlen, ldma_message_header_t **replyp, size_t *reply_dlenp)
{
	ldma_message_header_t *reply;
	char *str;
	size_t rlen;
	char *data;
	int status;

	LDMA_DBG("GET_CHASSISNO");

	if (request->msg_info != 0 || request_dlen != 0) {
		status = LDMA_REQ_INVALID;
		goto done;
	}

	if (ldma_get_chassis_serialno(&str) == 0) {
		LDMA_DBG("GET_CHASSISNO: ldma_get_chassisno failed "
		    "with error %d", errno);
		status = LDMA_REQ_FAILED;
		goto done;
	}

	rlen = strlen(str) + 1;

	reply = ldma_alloc_result_msg(request, rlen);

	if (reply == NULL) {
		status = LDMA_REQ_FAILED;
		goto done;
	}

	reply->msg_info = rlen;

	data = LDMA_HDR2DATA(reply);

	(void) strcpy(data, str);

	LDMA_DBG("GET_CHASSISNO: return info=%u, {%s}", rlen, str);

	*replyp = reply;
	*reply_dlenp = rlen;

	return (LDMA_REQ_COMPLETED);

done:
	LDMA_DBG("GET_CHASSISNO: return error %d", status);
	return (status);
}