/*
 * 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 (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
 */

#include "libzfs_jni_util.h"
#include "libzfs_jni_pool.h"
#include <strings.h>

/*
 * Types
 */

typedef struct ImportablePoolBean {
	zjni_Object_t super;
	PoolStatsBean_t interface_PoolStats;

	jmethodID method_setName;
	jmethodID method_setId;
} ImportablePoolBean_t;

typedef struct VirtualDeviceBean {
	zjni_Object_t super;
	DeviceStatsBean_t interface_DeviceStats;

	jmethodID method_setPoolName;
	jmethodID method_setParentIndex;
	jmethodID method_setIndex;
} VirtualDeviceBean_t;

typedef struct LeafVirtualDeviceBean {
	VirtualDeviceBean_t super;

	jmethodID method_setName;
} LeafVirtualDeviceBean_t;

typedef struct DiskVirtualDeviceBean {
	LeafVirtualDeviceBean_t super;
} DiskVirtualDeviceBean_t;

typedef struct SliceVirtualDeviceBean {
	LeafVirtualDeviceBean_t super;
} SliceVirtualDeviceBean_t;

typedef struct FileVirtualDeviceBean {
	LeafVirtualDeviceBean_t super;
} FileVirtualDeviceBean_t;

typedef struct RAIDVirtualDeviceBean {
	VirtualDeviceBean_t super;

	jmethodID method_setParity;
} RAIDVirtualDeviceBean_t;

typedef struct MirrorVirtualDeviceBean {
	VirtualDeviceBean_t super;
} MirrorVirtualDeviceBean_t;

/*
 * Data
 */

/* vdev_state_t to DeviceStats$DeviceState map */
static zjni_field_mapping_t vdev_state_map[] = {
	{ VDEV_STATE_CANT_OPEN, "VDEV_STATE_CANT_OPEN" },
	{ VDEV_STATE_CLOSED, "VDEV_STATE_CLOSED" },
	{ VDEV_STATE_DEGRADED, "VDEV_STATE_DEGRADED" },
	{ VDEV_STATE_HEALTHY, "VDEV_STATE_HEALTHY" },
	{ VDEV_STATE_OFFLINE, "VDEV_STATE_OFFLINE" },
	{ VDEV_STATE_UNKNOWN, "VDEV_STATE_UNKNOWN" },
	{ -1, NULL },
};

/* vdev_aux_t to DeviceStats$DeviceStatus map */
static zjni_field_mapping_t vdev_aux_map[] = {
	{ VDEV_AUX_NONE, "VDEV_AUX_NONE" },
	{ VDEV_AUX_OPEN_FAILED, "VDEV_AUX_OPEN_FAILED" },
	{ VDEV_AUX_CORRUPT_DATA, "VDEV_AUX_CORRUPT_DATA" },
	{ VDEV_AUX_NO_REPLICAS, "VDEV_AUX_NO_REPLICAS" },
	{ VDEV_AUX_BAD_GUID_SUM, "VDEV_AUX_BAD_GUID_SUM" },
	{ VDEV_AUX_TOO_SMALL, "VDEV_AUX_TOO_SMALL" },
	{ VDEV_AUX_BAD_LABEL, "VDEV_AUX_BAD_LABEL" },
	{ -1, NULL },
};

/* zpool_state_t to PoolStats$PoolState map */
static zjni_field_mapping_t pool_state_map[] = {
	{ POOL_STATE_ACTIVE, "POOL_STATE_ACTIVE" },
	{ POOL_STATE_EXPORTED, "POOL_STATE_EXPORTED" },
	{ POOL_STATE_DESTROYED, "POOL_STATE_DESTROYED" },
	{ POOL_STATE_SPARE, "POOL_STATE_SPARE" },
	{ POOL_STATE_UNINITIALIZED, "POOL_STATE_UNINITIALIZED" },
	{ POOL_STATE_UNAVAIL, "POOL_STATE_UNAVAIL" },
	{ POOL_STATE_POTENTIALLY_ACTIVE, "POOL_STATE_POTENTIALLY_ACTIVE" },
	{ -1, NULL },
};

/* zpool_status_t to PoolStats$PoolStatus map */
static zjni_field_mapping_t zpool_status_map[] = {
	{ ZPOOL_STATUS_CORRUPT_CACHE, "ZPOOL_STATUS_CORRUPT_CACHE" },
	{ ZPOOL_STATUS_MISSING_DEV_R, "ZPOOL_STATUS_MISSING_DEV_R" },
	{ ZPOOL_STATUS_MISSING_DEV_NR, "ZPOOL_STATUS_MISSING_DEV_NR" },
	{ ZPOOL_STATUS_CORRUPT_LABEL_R, "ZPOOL_STATUS_CORRUPT_LABEL_R" },
	{ ZPOOL_STATUS_CORRUPT_LABEL_NR, "ZPOOL_STATUS_CORRUPT_LABEL_NR" },
	{ ZPOOL_STATUS_BAD_GUID_SUM, "ZPOOL_STATUS_BAD_GUID_SUM" },
	{ ZPOOL_STATUS_CORRUPT_POOL, "ZPOOL_STATUS_CORRUPT_POOL" },
	{ ZPOOL_STATUS_CORRUPT_DATA, "ZPOOL_STATUS_CORRUPT_DATA" },
	{ ZPOOL_STATUS_FAILING_DEV, "ZPOOL_STATUS_FAILING_DEV" },
	{ ZPOOL_STATUS_VERSION_NEWER, "ZPOOL_STATUS_VERSION_NEWER" },
	{ ZPOOL_STATUS_HOSTID_MISMATCH, "ZPOOL_STATUS_HOSTID_MISMATCH" },
	{ ZPOOL_STATUS_FAULTED_DEV_R, "ZPOOL_STATUS_FAULTED_DEV_R" },
	{ ZPOOL_STATUS_FAULTED_DEV_NR, "ZPOOL_STATUS_FAULTED_DEV_NR" },
	{ ZPOOL_STATUS_BAD_LOG, "ZPOOL_STATUS_BAD_LOG" },
	{ ZPOOL_STATUS_VERSION_OLDER, "ZPOOL_STATUS_VERSION_OLDER" },
	{ ZPOOL_STATUS_RESILVERING, "ZPOOL_STATUS_RESILVERING" },
	{ ZPOOL_STATUS_OFFLINE_DEV, "ZPOOL_STATUS_OFFLINE_DEV" },
	{ ZPOOL_STATUS_REMOVED_DEV, "ZPOOL_STATUS_REMOVED_DEV" },
	{ ZPOOL_STATUS_OK, "ZPOOL_STATUS_OK" },
	{ -1, NULL }
};

/*
 * Function prototypes
 */

static void new_ImportablePoolBean(JNIEnv *, ImportablePoolBean_t *);
static void new_VirtualDevice(JNIEnv *, VirtualDeviceBean_t *);
static void new_LeafVirtualDevice(JNIEnv *, LeafVirtualDeviceBean_t *);
static void new_DiskVirtualDeviceBean(JNIEnv *, DiskVirtualDeviceBean_t *);
static void new_SliceVirtualDeviceBean(JNIEnv *, SliceVirtualDeviceBean_t *);
static void new_FileVirtualDeviceBean(JNIEnv *, FileVirtualDeviceBean_t *);
static void new_RAIDVirtualDeviceBean(JNIEnv *, RAIDVirtualDeviceBean_t *);
static void new_MirrorVirtualDeviceBean(JNIEnv *, MirrorVirtualDeviceBean_t *);
static int populate_ImportablePoolBean(
    JNIEnv *, ImportablePoolBean_t *, nvlist_t *);
static int populate_VirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, VirtualDeviceBean_t *);
static int populate_LeafVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, LeafVirtualDeviceBean_t *);
static int populate_DiskVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, DiskVirtualDeviceBean_t *);
static int populate_SliceVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, SliceVirtualDeviceBean_t *);
static int populate_FileVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, FileVirtualDeviceBean_t *);
static int populate_RAIDVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, RAIDVirtualDeviceBean_t *);
static int populate_MirrorVirtualDeviceBean(JNIEnv *, zpool_handle_t *,
    nvlist_t *, uint64_t *p_vdev_id, MirrorVirtualDeviceBean_t *);
static jobject create_ImportablePoolBean(JNIEnv *, nvlist_t *);
static jobject create_DiskVirtualDeviceBean(
    JNIEnv *, zpool_handle_t *, nvlist_t *, uint64_t *p_vdev_id);
static jobject create_SliceVirtualDeviceBean(
    JNIEnv *, zpool_handle_t *, nvlist_t *, uint64_t *p_vdev_id);
static jobject create_FileVirtualDeviceBean(
    JNIEnv *, zpool_handle_t *, nvlist_t *, uint64_t *p_vdev_id);
static jobject create_RAIDVirtualDeviceBean(
    JNIEnv *, zpool_handle_t *, nvlist_t *, uint64_t *p_vdev_id);
static jobject create_MirrorVirtualDeviceBean(
    JNIEnv *, zpool_handle_t *, nvlist_t *, uint64_t *p_vdev_id);
static char *find_field(const zjni_field_mapping_t *, int);
static jobject zjni_vdev_state_to_obj(JNIEnv *, vdev_state_t);
static jobject zjni_vdev_aux_to_obj(JNIEnv *, vdev_aux_t);

/*
 * Static functions
 */

/* Create a ImportablePoolBean */
static void
new_ImportablePoolBean(JNIEnv *env, ImportablePoolBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class =
		    (*env)->FindClass(env,
		    ZFSJNI_PACKAGE_DATA "ImportablePoolBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_PoolStats(env, &(bean->interface_PoolStats), object);

	bean->method_setName = (*env)->GetMethodID(
	    env, object->class, "setName", "(Ljava/lang/String;)V");

	bean->method_setId = (*env)->GetMethodID(
	    env, object->class, "setId", "(J)V");
}

/* Create a VirtualDeviceBean */
static void
new_VirtualDevice(JNIEnv *env, VirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class =
		    (*env)->FindClass(env,
		    ZFSJNI_PACKAGE_DATA "VirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_DeviceStats(env, &(bean->interface_DeviceStats), object);

	bean->method_setPoolName = (*env)->GetMethodID(
	    env, object->class, "setPoolName", "(Ljava/lang/String;)V");

	bean->method_setParentIndex = (*env)->GetMethodID(
	    env, object->class, "setParentIndex", "(Ljava/lang/Long;)V");

	bean->method_setIndex = (*env)->GetMethodID(
	    env, object->class, "setIndex", "(J)V");
}

/* Create a LeafVirtualDeviceBean */
static void
new_LeafVirtualDevice(JNIEnv *env, LeafVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class =
		    (*env)->FindClass(env,
		    ZFSJNI_PACKAGE_DATA "LeafVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_VirtualDevice(env, (VirtualDeviceBean_t *)bean);

	bean->method_setName = (*env)->GetMethodID(
	    env, object->class, "setName", "(Ljava/lang/String;)V");
}

/* Create a DiskVirtualDeviceBean */
static void
new_DiskVirtualDeviceBean(JNIEnv *env, DiskVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class = (*env)->FindClass(
		    env, ZFSJNI_PACKAGE_DATA "DiskVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_LeafVirtualDevice(env, (LeafVirtualDeviceBean_t *)bean);
}

/* Create a SliceVirtualDeviceBean */
static void
new_SliceVirtualDeviceBean(JNIEnv *env, SliceVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class = (*env)->FindClass(
		    env, ZFSJNI_PACKAGE_DATA "SliceVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_LeafVirtualDevice(env, (LeafVirtualDeviceBean_t *)bean);
}

/* Create a FileVirtualDeviceBean */
static void
new_FileVirtualDeviceBean(JNIEnv *env, FileVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class = (*env)->FindClass(
		    env, ZFSJNI_PACKAGE_DATA "FileVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_LeafVirtualDevice(env, (LeafVirtualDeviceBean_t *)bean);
}

/* Create a RAIDVirtualDeviceBean */
static void
new_RAIDVirtualDeviceBean(JNIEnv *env, RAIDVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {

		object->class = (*env)->FindClass(
		    env, ZFSJNI_PACKAGE_DATA "RAIDVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_VirtualDevice(env, (VirtualDeviceBean_t *)bean);

	bean->method_setParity = (*env)->GetMethodID(
	    env, object->class, "setParity", "(J)V");
}

/* Create a MirrorVirtualDeviceBean */
static void
new_MirrorVirtualDeviceBean(JNIEnv *env, MirrorVirtualDeviceBean_t *bean)
{
	zjni_Object_t *object = (zjni_Object_t *)bean;

	if (object->object == NULL) {
		object->class = (*env)->FindClass(
		    env, ZFSJNI_PACKAGE_DATA "MirrorVirtualDeviceBean");

		object->constructor =
		    (*env)->GetMethodID(env, object->class, "<init>", "()V");

		object->object =
		    (*env)->NewObject(env, object->class, object->constructor);
	}

	new_VirtualDevice(env, (VirtualDeviceBean_t *)bean);
}

static int
populate_ImportablePoolBean(JNIEnv *env, ImportablePoolBean_t *bean,
    nvlist_t *config)
{
	char *c;
	char *name;
	uint64_t guid;
	uint64_t state;
	uint64_t version;
	nvlist_t *devices;

	zjni_Object_t *object = (zjni_Object_t *)bean;
	PoolStatsBean_t *pool_stats = &(bean->interface_PoolStats);
	DeviceStatsBean_t *dev_stats = (DeviceStatsBean_t *)pool_stats;

	if (nvlist_lookup_string(config, ZPOOL_CONFIG_POOL_NAME, &name) ||
	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_POOL_GUID, &guid) ||
	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_POOL_STATE, &state) ||
	    nvlist_lookup_uint64(config, ZPOOL_CONFIG_VERSION, &version) ||
	    nvlist_lookup_nvlist(config, ZPOOL_CONFIG_VDEV_TREE, &devices) ||
	    populate_DeviceStatsBean(env, devices, dev_stats, object)) {
		return (-1);
	}

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setName, (*env)->NewStringUTF(env, name));

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setId, (jlong)guid);

	(*env)->CallVoidMethod(env, object->object,
	    pool_stats->method_setPoolState,
	    zjni_pool_state_to_obj(env, (pool_state_t)state));

	(*env)->CallVoidMethod(env, object->object,
	    pool_stats->method_setPoolStatus,
	    zjni_pool_status_to_obj(env, zpool_import_status(config, &c)));

	(*env)->CallVoidMethod(env, object->object,
	    pool_stats->method_setPoolVersion, (jlong)version);

	return (0);
}

static int
populate_VirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, VirtualDeviceBean_t *bean)
{
	int result;
	uint64_t vdev_id;
	jstring poolUTF;

	zjni_Object_t *object = (zjni_Object_t *)bean;
	DeviceStatsBean_t *stats = &(bean->interface_DeviceStats);

	result = populate_DeviceStatsBean(env, vdev, stats, object);
	if (result != 0) {
		return (1);
	}

	/* Set pool name */
	poolUTF = (*env)->NewStringUTF(env, zpool_get_name(zhp));
	(*env)->CallVoidMethod(
	    env, object->object, bean->method_setPoolName, poolUTF);

	/* Set parent vdev index */
	(*env)->CallVoidMethod(
	    env, object->object, bean->method_setParentIndex,
	    p_vdev_id == NULL ? NULL :
	    zjni_long_to_Long(env, *p_vdev_id));

	/* Get index */
	result = nvlist_lookup_uint64(vdev, ZPOOL_CONFIG_GUID, &vdev_id);
	if (result != 0) {
		zjni_throw_exception(env,
		    "could not retrieve virtual device ID (pool %s)",
		    zpool_get_name(zhp));
		return (1);
	}

	(*env)->CallVoidMethod(
	    env, object->object, bean->method_setIndex, (jlong)vdev_id);

	return (0);
}

static int
populate_LeafVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, LeafVirtualDeviceBean_t *bean)
{
	return (populate_VirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, (VirtualDeviceBean_t *)bean));
}

static int
populate_DiskVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, DiskVirtualDeviceBean_t *bean)
{
	char *path;
	int result = populate_LeafVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, (LeafVirtualDeviceBean_t *)bean);

	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (-1);
	}

	/* Set path */
	result = nvlist_lookup_string(vdev, ZPOOL_CONFIG_PATH, &path);
	if (result != 0) {
		zjni_throw_exception(env,
		    "could not retrieve path from disk virtual device "
		    "(pool %s)", zpool_get_name(zhp));
	} else {

		regex_t re;
		regmatch_t matches[2];
		jstring pathUTF = NULL;

		/* Strip off slice portion of name, if applicable */
		if (regcomp(&re, "^(/dev/dsk/.*)s[0-9]+$", REG_EXTENDED) == 0) {
			if (regexec(&re, path, 2, matches, 0) == 0) {
				regmatch_t *match = matches + 1;
				if (match->rm_so != -1 && match->rm_eo != -1) {
					char *tmp = strdup(path);
					if (tmp != NULL) {
						char *end = tmp + match->rm_eo;
						*end = '\0';
						pathUTF = (*env)->NewStringUTF(
						    env, tmp);
						free(tmp);
					}
				}
			}
			regfree(&re);
		}
		if (regcomp(&re, "^(/dev/dsk/.*)s[0-9]+/old$", REG_EXTENDED) ==
		    0) {
			if (regexec(&re, path, 2, matches, 0) == 0) {
				regmatch_t *match = matches + 1;
				if (match->rm_so != -1 && match->rm_eo != -1) {
					char *tmp = strdup(path);
					if (tmp != NULL) {
						(void) strcpy(tmp +
						    match->rm_eo, "/old");
						pathUTF = (*env)->NewStringUTF(
						    env, tmp);
						free(tmp);
					}
				}
			}
			regfree(&re);
		}

		if (pathUTF == NULL) {
			pathUTF = (*env)->NewStringUTF(env, path);
		}

		(*env)->CallVoidMethod(env, ((zjni_Object_t *)bean)->object,
		    ((LeafVirtualDeviceBean_t *)bean)->method_setName, pathUTF);
	}

	return (result != 0);
}

static int
populate_SliceVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, SliceVirtualDeviceBean_t *bean)
{
	char *path;
	int result = populate_LeafVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, (LeafVirtualDeviceBean_t *)bean);

	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (-1);
	}

	/* Set path */
	result = nvlist_lookup_string(vdev, ZPOOL_CONFIG_PATH, &path);
	if (result != 0) {
		zjni_throw_exception(env,
		    "could not retrieve path from slice virtual device (pool "
		    "%s)", zpool_get_name(zhp));
	} else {

		jstring pathUTF = (*env)->NewStringUTF(env, path);
		(*env)->CallVoidMethod(env, ((zjni_Object_t *)bean)->object,
		    ((LeafVirtualDeviceBean_t *)bean)->method_setName,
		    pathUTF);
	}

	return (result != 0);
}

static int
populate_FileVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, FileVirtualDeviceBean_t *bean)
{
	char *path;
	int result = populate_LeafVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, (LeafVirtualDeviceBean_t *)bean);

	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (-1);
	}

	/* Set path */
	result = nvlist_lookup_string(vdev, ZPOOL_CONFIG_PATH, &path);
	if (result != 0) {
		zjni_throw_exception(env,
		    "could not retrieve path from disk virtual device "
		    "(pool %s)", zpool_get_name(zhp));
	} else {

		jstring pathUTF = (*env)->NewStringUTF(env, path);
		(*env)->CallVoidMethod(env, ((zjni_Object_t *)bean)->object,
		    ((LeafVirtualDeviceBean_t *)bean)->method_setName, pathUTF);
	}

	return (result != 0);
}

static int
populate_RAIDVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, RAIDVirtualDeviceBean_t *bean)
{
	return (populate_VirtualDeviceBean(env, zhp, vdev, p_vdev_id,
	    (VirtualDeviceBean_t *)bean));
}

static int
populate_MirrorVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id, MirrorVirtualDeviceBean_t *bean)
{
	return (populate_VirtualDeviceBean(env, zhp, vdev, p_vdev_id,
	    (VirtualDeviceBean_t *)bean));
}

static jobject
create_ImportablePoolBean(JNIEnv *env, nvlist_t *config)
{
	int result;
	ImportablePoolBean_t bean_obj = {0};
	ImportablePoolBean_t *bean = &bean_obj;

	/* Construct ImportablePoolBean */
	new_ImportablePoolBean(env, bean);

	result = populate_ImportablePoolBean(env, bean, config);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static jobject
create_DiskVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	int result;
	DiskVirtualDeviceBean_t bean_obj = {0};
	DiskVirtualDeviceBean_t *bean = &bean_obj;

	/* Construct DiskVirtualDeviceBean */
	new_DiskVirtualDeviceBean(env, bean);

	result = populate_DiskVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, bean);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static jobject
create_SliceVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	int result;
	SliceVirtualDeviceBean_t bean_obj = {0};
	SliceVirtualDeviceBean_t *bean = &bean_obj;

	/* Construct SliceVirtualDeviceBean */
	new_SliceVirtualDeviceBean(env, bean);

	result = populate_SliceVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, bean);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static jobject
create_FileVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	int result;
	FileVirtualDeviceBean_t bean_obj = {0};
	FileVirtualDeviceBean_t *bean = &bean_obj;

	/* Construct FileVirtualDeviceBean */
	new_FileVirtualDeviceBean(env, bean);

	result = populate_FileVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, bean);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static jobject
create_RAIDVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	int result;
	uint64_t parity;
	RAIDVirtualDeviceBean_t bean_obj = {0};
	RAIDVirtualDeviceBean_t *bean = &bean_obj;

	((zjni_Object_t *)bean)->object = NULL;

	/* Construct RAIDVirtualDeviceBean */
	new_RAIDVirtualDeviceBean(env, bean);

	/* Set parity bit */
	result = nvlist_lookup_uint64(vdev, ZPOOL_CONFIG_NPARITY,
	    &parity);
	if (result) {
		/* Default to RAID-Z1 in case of error */
		parity = 1;
	}

	(*env)->CallVoidMethod(
	    env, ((zjni_Object_t *)bean)->object, bean->method_setParity,
	    (jlong)parity);


	result = populate_RAIDVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, bean);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static jobject
create_MirrorVirtualDeviceBean(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	int result;
	MirrorVirtualDeviceBean_t bean_obj = {0};
	MirrorVirtualDeviceBean_t *bean = &bean_obj;

	/* Construct MirrorVirtualDeviceBean */
	new_MirrorVirtualDeviceBean(env, bean);

	result = populate_MirrorVirtualDeviceBean(
	    env, zhp, vdev, p_vdev_id, bean);
	if (result) {
		/* Must not call any more Java methods to preserve exception */
		return (NULL);
	}

	return (((zjni_Object_t *)bean)->object);
}

static char *
find_field(const zjni_field_mapping_t *mapping, int value) {
	int i;
	for (i = 0; mapping[i].name != NULL; i++) {
		if (value == mapping[i].value) {
			return (mapping[i].name);
		}
	}
	return (NULL);
}

/*
 * Converts a vdev_state_t to a Java DeviceStats$DeviceState object.
 */
static jobject
zjni_vdev_state_to_obj(JNIEnv *env, vdev_state_t state)
{
	return (zjni_int_to_enum(env, state,
	    ZFSJNI_PACKAGE_DATA "DeviceStats$DeviceState",
	    "VDEV_STATE_UNKNOWN", vdev_state_map));
}

/*
 * Converts a vdev_aux_t to a Java DeviceStats$DeviceStatus object.
 */
static jobject
zjni_vdev_aux_to_obj(JNIEnv *env, vdev_aux_t aux)
{
	return (zjni_int_to_enum(env, aux,
	    ZFSJNI_PACKAGE_DATA "DeviceStats$DeviceStatus",
	    "VDEV_AUX_NONE", vdev_aux_map));
}

/*
 * Package-private functions
 */

/* Create a DeviceStatsBean */
void
new_DeviceStats(JNIEnv *env, DeviceStatsBean_t *bean, zjni_Object_t *object)
{
	bean->method_setSize = (*env)->GetMethodID(
	    env, object->class, "setSize", "(J)V");

	bean->method_setReplacementSize = (*env)->GetMethodID(
	    env, object->class, "setReplacementSize", "(J)V");

	bean->method_setUsed = (*env)->GetMethodID(
	    env, object->class, "setUsed", "(J)V");

	bean->method_setReadBytes = (*env)->GetMethodID(
	    env, object->class, "setReadBytes", "(J)V");

	bean->method_setWriteBytes = (*env)->GetMethodID(
	    env, object->class, "setWriteBytes", "(J)V");

	bean->method_setReadOperations = (*env)->GetMethodID(
	    env, object->class, "setReadOperations", "(J)V");

	bean->method_setWriteOperations = (*env)->GetMethodID(
	    env, object->class, "setWriteOperations", "(J)V");

	bean->method_setReadErrors = (*env)->GetMethodID(
	    env, object->class, "setReadErrors", "(J)V");

	bean->method_setWriteErrors = (*env)->GetMethodID(
	    env, object->class, "setWriteErrors", "(J)V");

	bean->method_setChecksumErrors = (*env)->GetMethodID(
	    env, object->class, "setChecksumErrors", "(J)V");

	bean->method_setDeviceState = (*env)->GetMethodID(
	    env, object->class, "setDeviceState",
	    "(L" ZFSJNI_PACKAGE_DATA "DeviceStats$DeviceState;)V");

	bean->method_setDeviceStatus = (*env)->GetMethodID(
	    env, object->class, "setDeviceStatus",
	    "(L" ZFSJNI_PACKAGE_DATA "DeviceStats$DeviceStatus;)V");
}

/* Create a PoolStatsBean */
void
new_PoolStats(JNIEnv *env, PoolStatsBean_t *bean, zjni_Object_t *object)
{
	new_DeviceStats(env, (DeviceStatsBean_t *)bean, object);

	bean->method_setPoolState = (*env)->GetMethodID(
	    env, object->class, "setPoolState",
	    "(L" ZFSJNI_PACKAGE_DATA "PoolStats$PoolState;)V");

	bean->method_setPoolStatus = (*env)->GetMethodID(
	    env, object->class, "setPoolStatus",
	    "(L" ZFSJNI_PACKAGE_DATA "PoolStats$PoolStatus;)V");

	bean->method_setPoolVersion = (*env)->GetMethodID(
	    env, object->class, "setPoolVersion", "(J)V");
}

/*
 * Gets the root vdev (an nvlist_t *) for the given pool.
 */
nvlist_t *
zjni_get_root_vdev(zpool_handle_t *zhp)
{
	nvlist_t *root = NULL;

	if (zhp != NULL) {
		nvlist_t *attrs = zpool_get_config(zhp, NULL);

		if (attrs != NULL) {
			int result = nvlist_lookup_nvlist(
			    attrs, ZPOOL_CONFIG_VDEV_TREE, &root);
			if (result != 0) {
				root = NULL;
			}
		}
	}

	return (root);
}

/*
 * Gets the vdev (an nvlist_t *) with the given vdev_id, below the
 * given vdev.  If the given vdev is NULL, all vdevs within the given
 * pool are searched.
 *
 * If p_vdev_id is not NULL, it will be set to the ID of the parent
 * vdev, if any, or to vdev_id_to_find if the searched-for vdev is a
 * toplevel vdev.
 */
nvlist_t *
zjni_get_vdev(zpool_handle_t *zhp, nvlist_t *vdev_parent,
    uint64_t vdev_id_to_find, uint64_t *p_vdev_id)
{
	int result;
	uint64_t id = vdev_id_to_find;

	/* Was a vdev specified? */
	if (vdev_parent == NULL) {
		/* No -- retrieve the top-level pool vdev */
		vdev_parent = zjni_get_root_vdev(zhp);
	} else {
		/* Get index of this vdev and compare with vdev_id_to_find */
		result = nvlist_lookup_uint64(
		    vdev_parent, ZPOOL_CONFIG_GUID, &id);
		if (result == 0 && id == vdev_id_to_find) {
			return (vdev_parent);
		}
	}

	if (vdev_parent != NULL) {

		nvlist_t **children;
		uint_t nelem = 0;

		/* Get the vdevs under this vdev */
		result = nvlist_lookup_nvlist_array(
		    vdev_parent, ZPOOL_CONFIG_CHILDREN, &children, &nelem);

		if (result == 0) {

			int i;
			nvlist_t *child;

			/* For each vdev child... */
			for (i = 0; i < nelem; i++) {
				if (p_vdev_id != NULL) {
					/* Save parent vdev id */
					*p_vdev_id = id;
				}

				child = zjni_get_vdev(zhp, children[i],
				    vdev_id_to_find, p_vdev_id);
				if (child != NULL) {
					return (child);
				}
			}
		}
	}

	return (NULL);
}

jobject
zjni_get_VirtualDevice_from_vdev(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev, uint64_t *p_vdev_id)
{
	jobject obj = NULL;
	char *type = NULL;
	int result = nvlist_lookup_string(vdev, ZPOOL_CONFIG_TYPE, &type);

	if (result == 0) {
		if (strcmp(type, VDEV_TYPE_DISK) == 0) {
			uint64_t wholedisk;
			if (nvlist_lookup_uint64(vdev, ZPOOL_CONFIG_WHOLE_DISK,
			    &wholedisk) == 0 && wholedisk) {
				obj = create_DiskVirtualDeviceBean(
				    env, zhp, vdev, p_vdev_id);
			} else {
				obj = create_SliceVirtualDeviceBean(
				    env, zhp, vdev, p_vdev_id);
			}
		} else if (strcmp(type, VDEV_TYPE_FILE) == 0) {
			obj = create_FileVirtualDeviceBean(
			    env, zhp, vdev, p_vdev_id);
		} else if (strcmp(type, VDEV_TYPE_RAIDZ) == 0) {
			obj = create_RAIDVirtualDeviceBean(
			    env, zhp, vdev, p_vdev_id);
		} else if (strcmp(type, VDEV_TYPE_MIRROR) == 0) {
			obj = create_MirrorVirtualDeviceBean(
			    env, zhp, vdev, p_vdev_id);
		} else if (strcmp(type, VDEV_TYPE_REPLACING) == 0) {

			/* Get the vdevs under this vdev */
			nvlist_t **children;
			uint_t nelem = 0;
			int result = nvlist_lookup_nvlist_array(
			    vdev, ZPOOL_CONFIG_CHILDREN, &children, &nelem);

			if (result == 0 && nelem > 0) {

				/* Get last vdev child (replacement device) */
				nvlist_t *child = children[nelem - 1];

				obj = zjni_get_VirtualDevice_from_vdev(env,
				    zhp, child, p_vdev_id);
			}
		}
	}

	return (obj);
}

jobject
zjni_get_VirtualDevices_from_vdev(JNIEnv *env, zpool_handle_t *zhp,
    nvlist_t *vdev_parent, uint64_t *p_vdev_id)
{
	/* Create an array list for the vdevs */
	zjni_ArrayList_t list_class = {0};
	zjni_ArrayList_t *list_class_p = &list_class;
	zjni_new_ArrayList(env, list_class_p);

	/* Was a vdev specified? */
	if (vdev_parent == NULL) {
		/* No -- retrieve the top-level pool vdev */
		vdev_parent = zjni_get_root_vdev(zhp);
	}

	if (vdev_parent != NULL) {

		/* Get the vdevs under this vdev */
		nvlist_t **children;
		uint_t nelem = 0;
		int result = nvlist_lookup_nvlist_array(
		    vdev_parent, ZPOOL_CONFIG_CHILDREN, &children, &nelem);

		if (result == 0) {

			/* For each vdev child... */
			int i;
			for (i = 0; i < nelem; i++) {
				nvlist_t *child = children[i];

				/* Create a Java object from this vdev */
				jobject obj =
				    zjni_get_VirtualDevice_from_vdev(env,
				    zhp, child, p_vdev_id);

				if ((*env)->ExceptionOccurred(env) != NULL) {
					/*
					 * Must not call any more Java methods
					 * to preserve exception
					 */
					return (NULL);
				}

				if (obj != NULL) {
				    /* Add child to child vdev list */
					(*env)->CallBooleanMethod(env,
					    ((zjni_Object_t *)
					    list_class_p)->object,
					    ((zjni_Collection_t *)
					    list_class_p)->method_add, obj);
				}
			}
		}
	}

	return (zjni_Collection_to_array(
	    env, (zjni_Collection_t *)list_class_p,
	    ZFSJNI_PACKAGE_DATA "VirtualDevice"));
}

int
zjni_create_add_ImportablePool(nvlist_t *config, void *data) {

	JNIEnv *env = ((zjni_ArrayCallbackData_t *)data)->env;
	zjni_Collection_t *list = ((zjni_ArrayCallbackData_t *)data)->list;

	/* Construct ImportablePool object */
	jobject bean = create_ImportablePoolBean(env, config);
	if (bean == NULL) {
		return (-1);
	}

	/* Add bean to list */
	(*env)->CallBooleanMethod(env, ((zjni_Object_t *)list)->object,
	    ((zjni_Collection_t *)list)->method_add, bean);

	return (0);
}

int
populate_DeviceStatsBean(JNIEnv *env, nvlist_t *vdev,
    DeviceStatsBean_t *bean, zjni_Object_t *object)
{
	uint_t c;
	vdev_stat_t *vs;

	int result = nvlist_lookup_uint64_array(
	    vdev, ZPOOL_CONFIG_VDEV_STATS, (uint64_t **)&vs, &c);
	if (result != 0) {
		zjni_throw_exception(env,
		    "could not retrieve virtual device statistics");
		return (1);
	}

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setUsed, (jlong)vs->vs_alloc);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setSize, (jlong)vs->vs_space);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setReplacementSize, (jlong)vs->vs_rsize);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setReadBytes, (jlong)vs->vs_bytes[ZIO_TYPE_READ]);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setWriteBytes, (jlong)vs->vs_bytes[ZIO_TYPE_WRITE]);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setReadOperations, (jlong)vs->vs_ops[ZIO_TYPE_READ]);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setWriteOperations, (jlong)vs->vs_ops[ZIO_TYPE_WRITE]);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setReadErrors, (jlong)vs->vs_read_errors);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setWriteErrors, (jlong)vs->vs_write_errors);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setChecksumErrors, (jlong)vs->vs_checksum_errors);

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setDeviceState,
	    zjni_vdev_state_to_obj(env, vs->vs_state));

	(*env)->CallVoidMethod(env, object->object,
	    bean->method_setDeviceStatus,
	    zjni_vdev_aux_to_obj(env, vs->vs_aux));

	return (0);
}

/*
 * Converts a pool_state_t to a Java PoolStats$PoolState object.
 */
jobject
zjni_pool_state_to_obj(JNIEnv *env, pool_state_t state)
{
	return (zjni_int_to_enum(env, state,
	    ZFSJNI_PACKAGE_DATA "PoolStats$PoolState",
	    "POOL_STATE_ACTIVE", pool_state_map));
}

/*
 * Converts a zpool_status_t to a Java PoolStats$PoolStatus object.
 */
jobject
zjni_pool_status_to_obj(JNIEnv *env, zpool_status_t status)
{
	return (zjni_int_to_enum(env, status,
	    ZFSJNI_PACKAGE_DATA "PoolStats$PoolStatus",
	    "ZPOOL_STATUS_OK", zpool_status_map));
}

/*
 * Extern functions
 */

/*
 * Iterates through each importable pool on the system.  For each
 * importable pool, runs the given function with the given void as the
 * last arg.
 */
int
zjni_ipool_iter(int argc, char **argv, zjni_ipool_iter_f func, void *data)
{
	nvlist_t *pools = zpool_find_import(g_zfs, argc, argv);

	if (pools != NULL) {
		nvpair_t *elem = NULL;

		while ((elem = nvlist_next_nvpair(pools, elem)) != NULL) {
			nvlist_t *config;

			if (nvpair_value_nvlist(elem, &config) != 0 ||
			    func(config, data)) {
				return (-1);
			}
		}
	}

	return (0);
}

char *
zjni_vdev_state_to_str(vdev_state_t state) {
	return (find_field(vdev_state_map, state));
}

char *
zjni_vdev_aux_to_str(vdev_aux_t aux) {
	return (find_field(vdev_aux_map, aux));
}

char *
zjni_pool_state_to_str(pool_state_t state) {
	return (find_field(pool_state_map, state));
}

char *
zjni_pool_status_to_str(zpool_status_t status) {
	return (find_field(zpool_status_map, status));
}