/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (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 2004 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>
#include <pool.h>
#include "pool_internal.h"

/*
 * libpool Value Manipulation Routines
 *
 * pool_value.c implements the value (pool_value_t) functionality for
 * libpool. The datatypes supported are: uint64_t, int64_t, double,
 * uchar_t (boolean), const char * (string). Values are used to
 * represent data stored to and retrieved from the datastore in a
 * simple discriminated union.
 *
 * Values are dynamically allocated using pool_value_alloc() and
 * destroyed using pool_value_free().
 *
 * Values may be allocated statically for internal use in
 * libpool. Statically allocated pool_value_t variables must be
 * initialised with the POOL_VALUE_INITIALIZER macro, otherwise the
 * results are unpredictable.
 *
 * A pool_value_t variable can be used to store values in any of the
 * supported datatypes.
 *
 * A pool_value_t's name and string value are limited in size to
 * PV_NAME_MAX_LEN and PV_VALUE_MAX_LEN respectively. Attempting to
 * store values which are greater than this in length will fail with a
 * POE_BADPARAM error.
 */

/*
 * Get the uint64_t data held by the value. If the data type isn't
 * uint64_t return PO_FAIL and set pool_error to be POE_BAD_PROP_TYPE.
 */
int
pool_value_get_uint64(const pool_value_t *pv, uint64_t *result)
{
	if (pv->pv_class != POC_UINT) {
		pool_seterror(POE_BAD_PROP_TYPE);
		return (PO_FAIL);
	}
	*result = pv->pv_u.u;
	return (PO_SUCCESS);
}

/*
 * Get the int64_t data held by the value. If the data type isn't
 * int64_t return PO_FAIL and set pool_error to be POE_BAD_PROP_TYPE.
 */
int
pool_value_get_int64(const pool_value_t *pv, int64_t *result)
{
	if (pv->pv_class != POC_INT) {
		pool_seterror(POE_BAD_PROP_TYPE);
		return (PO_FAIL);
	}
	*result = pv->pv_u.i;
	return (PO_SUCCESS);
}

/*
 * Get the double data held by the value. If the data type isn't
 * double return PO_FAIL and set pool_error to be POE_BAD_PROP_TYPE.
 */

int
pool_value_get_double(const pool_value_t *pv, double *result)
{
	if (pv->pv_class != POC_DOUBLE) {
		pool_seterror(POE_BAD_PROP_TYPE);
		return (PO_FAIL);
	}
	*result = pv->pv_u.d;
	return (PO_SUCCESS);
}

/*
 * Get the boolean data held by the value. If the data type isn't
 * boolean return PO_FAIL and set pool_error to be POE_BAD_PROP_TYPE.
 */
int
pool_value_get_bool(const pool_value_t *pv, uchar_t *result)
{
	if (pv->pv_class != POC_BOOL) {
		pool_seterror(POE_BAD_PROP_TYPE);
		return (PO_FAIL);
	}
	*result = pv->pv_u.b;
	return (PO_SUCCESS);
}

/*
 * Get the string data held by the value. If the data type isn't
 * string return PO_FAIL and set pool_error to be POE_BAD_PROP_TYPE.
 */
int
pool_value_get_string(const pool_value_t *pv, const char **result)
{
	if (pv->pv_class != POC_STRING) {
		pool_seterror(POE_BAD_PROP_TYPE);
		return (PO_FAIL);
	}
	*result = pv->pv_u.s;
	return (PO_SUCCESS);
}

/*
 * Get the type of the data held by the value. If the value has never
 * been used to store data, then the type is POC_INVAL.
 */
pool_value_class_t
pool_value_get_type(const pool_value_t *pv)
{
	return (pv->pv_class);
}

/*
 * Set the value's data to the supplied uint64_t data. Update the type
 * of the value data to POC_UINT.
 */
void
pool_value_set_uint64(pool_value_t *pv, uint64_t val)
{
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	pv->pv_class = POC_UINT;
	pv->pv_u.u = val;
}

/*
 * Set the value's data to the supplied int64_t data. Update the type
 * of the value data to POC_INT.
 */
void
pool_value_set_int64(pool_value_t *pv, int64_t val)
{
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	pv->pv_class = POC_INT;
	pv->pv_u.i = val;
}

/*
 * Set the value's data to the supplied double data. Update the type
 * of the value data to POC_DOUBLE.
 */

void
pool_value_set_double(pool_value_t *pv, double val)
{
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	pv->pv_class = POC_DOUBLE;
	pv->pv_u.d = val;
}

/*
 * Set the value's data to the supplied uchar_t data. Update the type
 * of the value data to POC_BOOL.
 */
void
pool_value_set_bool(pool_value_t *pv, uchar_t val)
{
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	pv->pv_class = POC_BOOL;
	pv->pv_u.b = !!val;	/* Lock value at 0 or 1 */
}

/*
 * Try to make an internal copy of the val, returning PO_SUCCESS or
 * PO_FAIL if the copy works or fails.
 */
int
pool_value_set_string(pool_value_t *pv, const char *val)
{
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	pv->pv_class = POC_STRING;
	if (val == NULL || strlen(val) >= PV_VALUE_MAX_LEN) {
		pool_seterror(POE_BADPARAM);
		return (PO_FAIL);
	} else {
		if ((pv->pv_u.s = atom_string(val)) == NULL)
			return (PO_FAIL);
	}
	return (PO_SUCCESS);
}

/*
 * Allocate a pool_value_t structure and initialise it to 0. Set the
 * type to POC_INVAL and return a pointer to the new pool_value_t. If
 * memory allocation fails, set POE_SYSTEM and return NULL.
 */
pool_value_t *
pool_value_alloc(void)
{
	pool_value_t *val;

	if ((val = malloc(sizeof (pool_value_t))) == NULL) {
		pool_seterror(POE_SYSTEM);
		return (NULL);
	}
	(void) memset(val, 0, sizeof (pool_value_t));
	val->pv_class = POC_INVAL;
	return (val);
}

/*
 * Free any atoms associated with the value and then free the value
 * itself.
 */
void
pool_value_free(pool_value_t *pv)
{
	if (pv->pv_name)
		atom_free(pv->pv_name);
	if (pv->pv_class == POC_STRING)
		atom_free(pv->pv_u.s);
	free(pv);
}

/*
 * Return a pointer to the name of the value. This may be NULL if the
 * name has never been set.
 */
const char *
pool_value_get_name(const pool_value_t *pv)
{
	return (pv->pv_name);
}

/*
 * Set the name of the value to the supplied name.
 */
int
pool_value_set_name(pool_value_t *pv, const char *name)
{
	if (name == NULL || strlen(name) >= PV_NAME_MAX_LEN) {
		pool_seterror(POE_BADPARAM);
		return (PO_FAIL);
	} else {
		if (pv->pv_name)
			atom_free(pv->pv_name);
		if ((pv->pv_name = atom_string(name)) == NULL)
			return (PO_FAIL);
	}
	return (PO_SUCCESS);
}

/*
 * Use the supplied nvpair_t to set the name, type and value of the
 * supplied pool_value_t.
 *
 * Return: PO_SUCCESS/PO_FAIL
 */
int
pool_value_from_nvpair(pool_value_t *pv, nvpair_t *pn)
{
	uchar_t bval;
	uint64_t uval;
	int64_t ival;
	double dval;
	uint_t nelem;
	uchar_t *dval_b;
	char *sval;

	if (pool_value_set_name(pv, nvpair_name(pn)) != PO_SUCCESS)
		return (PO_FAIL);
	switch (nvpair_type(pn)) {
	case DATA_TYPE_BYTE:
		if (nvpair_value_byte(pn, &bval) != 0) {
			pool_seterror(POE_SYSTEM);
			return (PO_FAIL);
		}
		pool_value_set_bool(pv, bval);
		break;
	case DATA_TYPE_BYTE_ARRAY:
		if (nvpair_value_byte_array(pn, &dval_b, &nelem) != 0) {
			pool_seterror(POE_SYSTEM);
			return (PO_FAIL);
		}
		(void) memcpy(&dval, dval_b, sizeof (double));
		pool_value_set_double(pv, dval);
		break;
	case DATA_TYPE_INT64:
		if (nvpair_value_int64(pn, &ival) != 0) {
			pool_seterror(POE_SYSTEM);
			return (PO_FAIL);
		}
		pool_value_set_int64(pv, ival);
		break;
	case DATA_TYPE_UINT64:
		if (nvpair_value_uint64(pn, &uval) != 0) {
			pool_seterror(POE_SYSTEM);
			return (PO_FAIL);
		}
		pool_value_set_uint64(pv, uval);
		break;
	case DATA_TYPE_STRING:
		if (nvpair_value_string(pn, &sval) != 0) {
			pool_seterror(POE_SYSTEM);
			return (PO_FAIL);
		}
		if (pool_value_set_string(pv, sval) != PO_SUCCESS)
			return (PO_FAIL);
		break;
	default:
		pool_seterror(POE_SYSTEM);
		return (PO_FAIL);
	}
	return (PO_SUCCESS);
}

/*
 * Check to see if the values held by two supplied values are
 * equal. First compare the pointers to see if we are comparing to
 * ourselves, if we are return PO_TRUE. If not, get the types and
 * ensure they match, if they don't return PO_FALSE. Then do a type
 * specific comparison returning PO_TRUE or PO_FALSE accordingly.
 */
int
pool_value_equal(pool_value_t *pv1, pool_value_t *pv2)
{
	uint64_t uval1, uval2;
	int64_t ival1, ival2;
	double dval1, dval2;
	uchar_t bval1, bval2;
	const char *sval1, *sval2;
	pool_value_class_t type;

	if (pv1 == pv2) /* optimisation */
		return (PO_TRUE);

	type = pool_value_get_type(pv1);
	if (type != pool_value_get_type(pv2))
	    return (PO_FALSE);

	switch (type) {
		case POC_UINT:
			(void) pool_value_get_uint64(pv1, &uval1);
			(void) pool_value_get_uint64(pv2, &uval2);
			if (uval1 == uval2)
				return (PO_TRUE);
			break;
		case POC_INT:
			(void) pool_value_get_int64(pv1, &ival1);
			(void) pool_value_get_int64(pv2, &ival2);
			if (ival1 == ival2)
				return (PO_TRUE);
			break;
		case POC_DOUBLE:
			(void) pool_value_get_double(pv1, &dval1);
			(void) pool_value_get_double(pv2, &dval2);
			if (dval1 == dval2)
				return (PO_TRUE);
			break;
		case POC_BOOL:
			(void) pool_value_get_bool(pv1, &bval1);
			(void) pool_value_get_bool(pv2, &bval2);
			if (bval1 == bval2)
				return (PO_TRUE);
			break;
		case POC_STRING:
			(void) pool_value_get_string(pv1, &sval1);
			(void) pool_value_get_string(pv2, &sval2);
			if (strcmp(sval1, sval2) == 0)
				return (PO_TRUE);
			break;
	}
	return (PO_FALSE);
}

#ifdef DEBUG
/*
 * Trace pool_value_t details using dprintf
 */
void
pool_value_dprintf(const pool_value_t *pv)
{
	const char *class_name[] = {
		"POC_UINT",
		"POC_INT",
		"POC_DOUBLE",
		"POC_BOOL",
		"POC_STRING"
	};

	dprintf("name: %s\n", pv->pv_name ? pv->pv_name : "NULL");
	if (pv->pv_class >= POC_UINT && pv->pv_class <= POC_STRING)
		dprintf("type: %s\n", class_name[pv->pv_class]);
	else
		dprintf("type: POC_INVAL\n");
	switch (pv->pv_class) {
	case POC_UINT:
		dprintf("value: %llu\n", pv->pv_u.u);
		break;
	case POC_INT:
		dprintf("value: %lld\n", pv->pv_u.i);
		break;
	case POC_DOUBLE:
		dprintf("value: %f\n", pv->pv_u.d);
		break;
	case POC_BOOL:
		dprintf("value: %s\n", pv->pv_u.b ? "true" : "false");
		break;
	case POC_STRING:
		dprintf("value: %s\n", pv->pv_u.s);
		break;
	default:
		dprintf("value: invalid\n");
		break;
	}
}
#endif	/* DEBUG */