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

#include <stdlib.h>
#include <errno.h>
#include <sys/types.h>
#include <libnvpair.h>
#include <sys/fcntl.h>
#include <sys/devfm.h>
#include <fmd_agent_impl.h>

static int
cleanup_set_errno(fmd_agent_hdl_t *hdl, nvlist_t *innvl, nvlist_t *outnvl,
    int err)
{
	nvlist_free(innvl);
	nvlist_free(outnvl);
	return (fmd_agent_seterrno(hdl, err));
}

static int
fmd_agent_physcpu_info_v1(fmd_agent_hdl_t *hdl, nvlist_t ***cpusp,
    uint_t *ncpup)
{
	int err;
	nvlist_t *nvl, **nvl_array, **cpus;
	uint_t i, n;

	if ((err = fmd_agent_nvl_ioctl(hdl, FM_IOC_PHYSCPU_INFO, 1,
	    NULL, &nvl)) != 0)
		return (cleanup_set_errno(hdl, NULL, NULL, err));
	if ((err = nvlist_lookup_nvlist_array(nvl, FM_PHYSCPU_INFO_CPUS,
	    &cpus, &n)) != 0)
		return (cleanup_set_errno(hdl, NULL, nvl, err));

	if ((nvl_array = umem_alloc(sizeof (nvlist_t *) * n, UMEM_DEFAULT))
	    == NULL)
		return (cleanup_set_errno(hdl, NULL, nvl, errno));
	for (i = 0; i < n; i++) {
		if ((err = nvlist_dup(cpus[i], nvl_array + i, 0)) != 0) {
			while (i > 0)
				nvlist_free(nvl_array[--i]);
			umem_free(nvl_array, sizeof (nvlist_t *) * n);
			return (cleanup_set_errno(hdl, NULL, nvl, err));
		}
	}

	nvlist_free(nvl);
	*cpusp = nvl_array;
	*ncpup = n;
	return (0);
}

int
fmd_agent_physcpu_info(fmd_agent_hdl_t *hdl, nvlist_t ***cpusp, uint_t *ncpu)
{
	uint32_t ver;

	if (fmd_agent_version(hdl, FM_CPU_INFO_VERSION, &ver) == -1)
		return (fmd_agent_seterrno(hdl, errno));

	switch (ver) {
	case 1:
		return (fmd_agent_physcpu_info_v1(hdl, cpusp, ncpu));

	default:
		return (fmd_agent_seterrno(hdl, ENOTSUP));
	}
}

static int
fmd_agent_cpuop_v1(fmd_agent_hdl_t *hdl, int cmd, int chipid, int coreid,
    int strandid, int *old_status)
{
	int err;
	nvlist_t *nvl = NULL, *outnvl = NULL;
	int32_t status;

	if ((err = nvlist_alloc(&nvl, NV_UNIQUE_NAME_TYPE, 0)) != 0 ||
	    (err = nvlist_add_int32(nvl, FM_CPU_RETIRE_CHIP_ID, chipid)) != 0 ||
	    (err = nvlist_add_int32(nvl, FM_CPU_RETIRE_CORE_ID, coreid)) != 0 ||
	    (err = nvlist_add_int32(nvl, FM_CPU_RETIRE_STRAND_ID, strandid))
	    != 0 || (err = fmd_agent_nvl_ioctl(hdl, cmd, 1, nvl, &outnvl)) != 0)
		return (cleanup_set_errno(hdl, nvl, NULL, err));

	nvlist_free(nvl);
	if (outnvl != NULL) {
		if (old_status != NULL) {
			(void) nvlist_lookup_int32(outnvl,
			    FM_CPU_RETIRE_OLDSTATUS, &status);
			*old_status = status;
		}
		nvlist_free(outnvl);
	}

	return (0);
}

static int
fmd_agent_cpuop(fmd_agent_hdl_t *hdl, int cmd, int chipid, int coreid,
    int strandid, int *old_status)
{
	uint32_t ver;

	if (fmd_agent_version(hdl, FM_CPU_OP_VERSION, &ver) == -1)
		return (cleanup_set_errno(hdl, NULL, NULL, errno));

	switch (ver) {
	case 1:
		return (fmd_agent_cpuop_v1(hdl, cmd, chipid, coreid, strandid,
		    old_status));

	default:
		return (fmd_agent_seterrno(hdl, ENOTSUP));
	}
}

int
fmd_agent_cpu_retire(fmd_agent_hdl_t *hdl, int chipid, int coreid, int strandid)
{
	int ret;

	ret = fmd_agent_cpuop(hdl, FM_IOC_CPU_RETIRE, chipid, coreid, strandid,
	    NULL);

	return (ret == 0 ? FMD_AGENT_RETIRE_DONE : FMD_AGENT_RETIRE_FAIL);
}

int
fmd_agent_cpu_isretired(fmd_agent_hdl_t *hdl, int chipid, int coreid,
    int strandid)
{
	int ret, status;

	ret = fmd_agent_cpuop(hdl, FM_IOC_CPU_STATUS, chipid, coreid, strandid,
	    &status);

	return (ret == 0 && status != P_ONLINE ?
	    FMD_AGENT_RETIRE_DONE : FMD_AGENT_RETIRE_FAIL);
}

int
fmd_agent_cpu_unretire(fmd_agent_hdl_t *hdl, int chipid, int coreid,
    int strandid)
{
	int ret;

	ret = fmd_agent_cpuop(hdl, FM_IOC_CPU_UNRETIRE, chipid, coreid,
	    strandid, NULL);

	return (ret == 0 ? FMD_AGENT_RETIRE_DONE : FMD_AGENT_RETIRE_FAIL);
}