cmd/ldmad/ldma_system.c

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