libzfs_jni/common/libzfs_jni_pool.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 (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));
}