libzfs_jni/common/libzfs_jni_main.c

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

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

#include "libzfs_jni_main.h"
#include "libzfs_jni_util.h"
#include "libzfs_jni_dataset.h"
#include "libzfs_jni_property.h"
#include "libzfs_jni_pool.h"
#include "libzfs_jni_diskmgt.h"
#include "libzfs_jni_disk.h"

/*
 * Function prototypes
 */

static void handle_error(const char *, va_list);
static void init();

/*
 * Static functions
 */

char libzfs_err[1024];
static void
handle_error(const char *fmt, va_list ap)
{
    /* Save the error message in case it's needed */
    (void) vsnprintf(libzfs_err, sizeof (libzfs_err), fmt, ap);
#ifdef  DEBUG
    (void) fprintf(stderr, "caught error: %s\n", libzfs_err);
#endif
}

/*
 * Initialize the library.  Sets the error handler.
 */
#pragma init(init)
static void
init()
{
    libzfs_err[0] = '\0';

    /* libzfs error handler */
    zfs_set_error_handler(handle_error);

    /* diskmgt.o error handler */
    dmgt_set_error_handler(handle_error);
}

/*
 * JNI functions
 */

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getPools
 * Signature: ()[Lcom/sun/zfs/common/model/Pool;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getPools(JNIEnv *env, jobject obj)
{
    zjni_DatasetArrayCallbackData_t data = {0};
    int result;

    /* Create an array list */
    zjni_ArrayList_t list_obj = {0};
    zjni_ArrayList_t *list = &list_obj;
    zjni_new_ArrayList(env, list);

    data.data.env = env;
    data.data.list = (zjni_Collection_t *)list;
    data.typemask = ZFS_TYPE_FILESYSTEM;

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

    return (zjni_Collection_to_array(env, (zjni_Collection_t *)list,
        ZFSJNI_PACKAGE_DATA "Pool"));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getPool
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/Pool;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getPool(JNIEnv *env,
    jobject obj, jstring poolUTF)
{
    return (zjni_get_Dataset(env, poolUTF, ZFS_TYPE_FILESYSTEM));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getFileSystems
 * Signature: (Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/FileSystem;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getFileSystems(JNIEnv *env,
    jobject obj, jstring containerUTF)
{
    return (zjni_get_Datasets_below(env, containerUTF,
        ZFS_TYPE_FILESYSTEM, ZFS_TYPE_FILESYSTEM,
        ZFSJNI_PACKAGE_DATA "FileSystem"));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getFileSystem
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/FileSystem;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getFileSystem(JNIEnv *env,
    jobject obj, jstring nameUTF)
{
    return (zjni_get_Dataset(env, nameUTF, ZFS_TYPE_FILESYSTEM));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getVolumes
 * Signature: (Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/Volume;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getVolumes(JNIEnv *env,
    jobject obj, jstring containerUTF)
{
    return (zjni_get_Datasets_below(env, containerUTF,
        ZFS_TYPE_FILESYSTEM, ZFS_TYPE_VOLUME,
        ZFSJNI_PACKAGE_DATA "Volume"));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getVolume
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/Volume;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getVolume(JNIEnv *env,
    jobject obj, jstring nameUTF)
{
    return (zjni_get_Dataset(env, nameUTF, ZFS_TYPE_VOLUME));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getSnapshots
 * Signature: (Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/Snapshot;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getSnapshots(JNIEnv *env,
    jobject obj, jstring datasetUTF)
{
    return (zjni_get_Datasets_below(env, datasetUTF,
        ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME, ZFS_TYPE_SNAPSHOT,
        ZFSJNI_PACKAGE_DATA "Snapshot"));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getSnapshot
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/Snapshot;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getSnapshot(JNIEnv *env,
    jobject obj, jstring nameUTF)
{
    return (zjni_get_Dataset(env, nameUTF, ZFS_TYPE_SNAPSHOT));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getDatasets
 * Signature: (Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/Dataset;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getDatasets(JNIEnv *env,
    jobject obj, jstring containerUTF)
{
    if (containerUTF == NULL) {
        return (Java_com_sun_zfs_common_model_SystemDataModel_getPools(
            env, obj));
    }

    return (zjni_get_Datasets_below(env, containerUTF,
        ZFS_TYPE_FILESYSTEM, ZFS_TYPE_ANY,
        ZFSJNI_PACKAGE_DATA "Dataset"));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getDataset
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/Dataset;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getDataset(JNIEnv *env,
    jobject obj, jstring nameUTF)
{
    return (zjni_get_Dataset(env, nameUTF, ZFS_TYPE_ANY));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getVirtualDevice
 * Signature: (Ljava/lang/String;J)Lcom/sun/zfs/common/model/VirtualDevice;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getVirtualDevice(JNIEnv *env,
    jobject obj, jstring poolUTF, jlong index)
{
    jobject vdev = NULL;

    if (poolUTF != NULL) {
        const char *pool = (*env)->GetStringUTFChars(env, poolUTF,
            NULL);
        zpool_handle_t *zhp = zpool_open(pool);
        (*env)->ReleaseStringUTFChars(env, poolUTF, pool);

        if (zhp != NULL) {
            nvlist_t *vdev_cfg = zjni_get_vdev(zhp, NULL, index);
            if (vdev_cfg != NULL) {
                vdev = zjni_get_VirtualDevice_from_vdev(env,
                    zhp, vdev_cfg);
            }
            zpool_close(zhp);
        }
    }

    return (vdev);
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getVirtualDevices
 * Signature: (Ljava/lang/String;J)
 *            [Lcom/sun/zfs/common/model/VirtualDevice;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
/* CSTYLED */
Java_com_sun_zfs_common_model_SystemDataModel_getVirtualDevices__Ljava_lang_String_2J(
    JNIEnv *env, jobject obj, jstring poolUTF, jlong index)
{
    jobjectArray vdevs = NULL;

    if (poolUTF != NULL) {
        const char *pool = (*env)->GetStringUTFChars(env, poolUTF,
            NULL);
        zpool_handle_t *zhp = zpool_open(pool);
        (*env)->ReleaseStringUTFChars(env, poolUTF, pool);

        /* Is the pool valid? */
        if (zhp != NULL) {
            nvlist_t *vdev_cfg = zjni_get_vdev(zhp, NULL, index);
            if (vdev_cfg != NULL) {
                vdevs = zjni_get_VirtualDevices_from_vdev(
                    env, zhp, vdev_cfg);
            }
            zpool_close(zhp);
        }
    }

    return (vdevs);
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getVirtualDevices
 * Signature: (Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/VirtualDevice;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
/* CSTYLED */
Java_com_sun_zfs_common_model_SystemDataModel_getVirtualDevices__Ljava_lang_String_2(
    JNIEnv *env, jobject obj, jstring poolUTF)
{
    jobjectArray vdevs = NULL;

    if (poolUTF != NULL) {
        const char *pool = (*env)->GetStringUTFChars(env,
            poolUTF, NULL);
        zpool_handle_t *zhp = zpool_open(pool);
        (*env)->ReleaseStringUTFChars(env, poolUTF, pool);

        /* Is the pool valid? */
        if (zhp != NULL) {
            vdevs = zjni_get_VirtualDevices_from_vdev(env,
                zhp, NULL);
            zpool_close(zhp);
        }
    }

    return (vdevs);
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getAvailableDisks
 * Signature: ()[Lcom/sun/zfs/common/model/DiskDevice;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getAvailableDisks(JNIEnv *env,
    jobject obj)
{
    int error;
    zjni_ArrayCallbackData_t data = {0};
    jobjectArray array = NULL;

    /* Create an array list */
    zjni_ArrayList_t list_obj = {0};
    zjni_ArrayList_t *list = &list_obj;
    zjni_new_ArrayList(env, list);

    data.env = env;
    data.list = (zjni_Collection_t *)list;
    error = dmgt_avail_disk_iter(zjni_create_add_DiskDevice, &data);

    if (error) {
        zjni_throw_exception(env, "%s", libzfs_err);
    } else {
        array = zjni_Collection_to_array(
            env, (zjni_Collection_t *)list,
            ZFSJNI_PACKAGE_DATA "DiskDevice");
    }

    return (array);
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getDependents
 * Signature: ([Ljava/lang/String;)
 *            [Lcom/sun/zfs/common/model/Dataset;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getDependents(JNIEnv *env,
    jobject obj, jobjectArray paths)
{
    return (zjni_get_Datasets_dependents(env, paths));
}

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getPropertyDefault
 * Signature: (Ljava/lang/String;)
 *            Lcom/sun/zfs/common/model/Property;
 */
/* ARGSUSED */
JNIEXPORT jobject JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getPropertyDefault(JNIEnv *env,
    jobject obj, jstring nameUTF)
{
    jobject defProperty = NULL;

    const char *name = (*env)->GetStringUTFChars(env, nameUTF, NULL);
    zfs_prop_t prop = zjni_get_property_from_name(name);
    (*env)->ReleaseStringUTFChars(env, nameUTF, name);

    if (prop != ZFS_PROP_INVAL) {
        defProperty = zjni_get_default_property(env, prop);
    }

    return (defProperty);
}

typedef struct zjni_class_type_map {
    char *class;
    zfs_type_t type;
} zjni_class_type_map_t;

/*
 * Class:     com_sun_zfs_common_model_SystemDataModel
 * Method:    getValidPropertyNames
 * Signature: (Ljava/lang/Class;)
 *            [Ljava/lang/String;
 */
/* ARGSUSED */
JNIEXPORT jobjectArray JNICALL
Java_com_sun_zfs_common_model_SystemDataModel_getValidPropertyNames(JNIEnv *env,
    jobject obj, jclass class)
{
    int i;

    /* Mappings of class names to zfs_type_t */
    static zjni_class_type_map_t mappings[] = {
        { ZFSJNI_PACKAGE_DATA "FileSystem", ZFS_TYPE_FILESYSTEM },
        { ZFSJNI_PACKAGE_DATA "Volume", ZFS_TYPE_VOLUME },
        { ZFSJNI_PACKAGE_DATA "Snapshot", ZFS_TYPE_SNAPSHOT },
    };
    int nmappings = sizeof (mappings) / sizeof (zjni_class_type_map_t);

    jclass class_Class = (*env)->FindClass(env, "java/lang/Class");

    jmethodID isAssignableFrom = (*env)->GetMethodID(
        env, class_Class, "isAssignableFrom", "(Ljava/lang/Class;)Z");

    /* Create an array list for the property names */
    zjni_ArrayList_t list_obj = {0};
    zjni_ArrayList_t *list = &list_obj;
    zjni_new_ArrayList(env, list);

    /* For each mapping... */
    for (i = 0; i < nmappings; i++) {
        /*
         * Is the given class an instance of the class in the mapping?
         */
        jclass typeClass = (*env)->FindClass(env, mappings[i].class);

        jboolean isInstance = (*env)->CallBooleanMethod(
            env, typeClass, isAssignableFrom, class);

        if (isInstance == JNI_TRUE) {
            zfs_prop_t prop;
            for (prop = 0; prop < ZFS_NPROP_VISIBLE; prop++) {
                if (zfs_prop_valid_for_type(prop,
                    mappings[i].type)) {
                    /* Add name of property to list */
                    jstring propName =
                        (*env)->NewStringUTF(env,
                        zfs_prop_to_name(prop));
                    (*env)->CallBooleanMethod(
                        env,
                        ((zjni_Object_t *)list)->object,
                        ((zjni_Collection_t *)list)->
                        method_add, propName);
                }
            }
            break;
        }
    }

    return (zjni_Collection_to_array(
        env, (zjni_Collection_t *)list, "java/lang/String"));
}