/*
 * 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.
 * Copyright (c) 2015 by Delphix. All rights reserved.
 */

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

/*
 * Types
 */

/* Signature for function to convert string to a specific Java object */
typedef jobject (*str_to_obj_f)(JNIEnv *, char *);

/* Signature for function to convert uint64_t to a specific Java object */
typedef jobject (*uint64_to_obj_f)(JNIEnv *, uint64_t);

/*
 * Describes a property and the parameters needed to create a Java
 * Property object for it
 */
typedef struct custom_prop_desct {
	zfs_prop_t prop;
	str_to_obj_f convert_str;
	uint64_to_obj_f convert_uint64;
	char *propClass;
	char *valueClass;
} custom_prop_desct_t;

/*
 * Function prototypes
 */

static jobject create_BasicProperty(JNIEnv *, zfs_handle_t *,
    zfs_prop_t, str_to_obj_f, uint64_to_obj_f, char *, char *);
static jobject create_BooleanProperty(JNIEnv *, zfs_handle_t *, zfs_prop_t);
static jobject create_LongProperty(JNIEnv *, zfs_handle_t *, zfs_prop_t);
static jobject create_StringProperty(JNIEnv *, zfs_handle_t *, zfs_prop_t);
static jobject create_ObjectProperty(JNIEnv *, zfs_handle_t *,
    zfs_prop_t, str_to_obj_f, uint64_to_obj_f, char *, char *);
static jobject create_default_BasicProperty(JNIEnv *, zfs_prop_t,
    str_to_obj_f, uint64_to_obj_f, char *, char *);
static jobject create_default_BooleanProperty(JNIEnv *, zfs_prop_t);
static jobject create_default_LongProperty(JNIEnv *, zfs_prop_t);
static jobject create_default_StringProperty(JNIEnv *, zfs_prop_t);
static jobject create_default_ObjectProperty(
    JNIEnv *, zfs_prop_t, str_to_obj_f, uint64_to_obj_f, char *, char *);
static jobject str_to_enum_element(JNIEnv *, char *, char *);
static jobject str_to_aclinherit(JNIEnv *, char *);
static jobject str_to_aclmode(JNIEnv *, char *);
static jobject str_to_checksum(JNIEnv *, char *);
static jobject str_to_compression(JNIEnv *, char *);
static jobject str_to_snapdir(JNIEnv *, char *);
static jobject str_to_string(JNIEnv *, char *);

/*
 * Static data
 */

zfs_prop_t props_boolean[] = {
	ZFS_PROP_ATIME,
	ZFS_PROP_DEVICES,
	ZFS_PROP_EXEC,
	ZFS_PROP_MOUNTED,
	ZFS_PROP_READONLY,
	ZFS_PROP_SETUID,
	ZFS_PROP_ZONED,
	ZFS_PROP_DEFER_DESTROY,
	ZPROP_INVAL
};

zfs_prop_t props_long[] = {
	ZFS_PROP_AVAILABLE,
	ZFS_PROP_CREATETXG,
	ZFS_PROP_QUOTA,
	ZFS_PROP_REFERENCED,
	ZFS_PROP_RESERVATION,
	ZFS_PROP_USED,
	ZFS_PROP_VOLSIZE,
	ZFS_PROP_REFQUOTA,
	ZFS_PROP_REFRESERVATION,
	ZFS_PROP_USERREFS,
	ZPROP_INVAL
};

zfs_prop_t props_string[] = {
	ZFS_PROP_ORIGIN,
	/* ZFS_PROP_TYPE, */
	ZPROP_INVAL
};

custom_prop_desct_t props_custom[] = {
	{ ZFS_PROP_ACLINHERIT, str_to_aclinherit, NULL,
	    ZFSJNI_PACKAGE_DATA "AclInheritProperty",
	    ZFSJNI_PACKAGE_DATA "AclInheritProperty$AclInherit" },

	{ ZFS_PROP_ACLMODE, str_to_aclmode, NULL,
	    ZFSJNI_PACKAGE_DATA "AclModeProperty",
	    ZFSJNI_PACKAGE_DATA "AclModeProperty$AclMode" },

	{ ZFS_PROP_CHECKSUM, str_to_checksum, NULL,
	    ZFSJNI_PACKAGE_DATA "ChecksumProperty",
	    ZFSJNI_PACKAGE_DATA "ChecksumProperty$Checksum" },

	{ ZFS_PROP_COMPRESSION, str_to_compression, NULL,
	    ZFSJNI_PACKAGE_DATA "CompressionProperty",
	    ZFSJNI_PACKAGE_DATA "CompressionProperty$Compression" },

	{ ZFS_PROP_COMPRESSRATIO, NULL, zjni_long_to_Long,
	    ZFSJNI_PACKAGE_DATA "CompressRatioProperty",
	    "java/lang/Long" },

	{ ZFS_PROP_CREATION, zjni_str_to_date, NULL,
	    ZFSJNI_PACKAGE_DATA "CreationProperty",
	    "java/util/Date" },

	{ ZFS_PROP_MOUNTPOINT, str_to_string, NULL,
	    ZFSJNI_PACKAGE_DATA "MountPointProperty",
	    "java/lang/String" },

	{ ZFS_PROP_RECORDSIZE, NULL, zjni_long_to_Long,
	    ZFSJNI_PACKAGE_DATA "RecordSizeProperty",
	    "java/lang/Long" },

	{ ZFS_PROP_SHARENFS, str_to_string, NULL,
	    ZFSJNI_PACKAGE_DATA "ShareNFSProperty",
	    "java/lang/String" },

	{ ZFS_PROP_SNAPDIR, str_to_snapdir, NULL,
	    ZFSJNI_PACKAGE_DATA "SnapDirProperty",
	    ZFSJNI_PACKAGE_DATA "SnapDirProperty$SnapDir" },

	{ ZFS_PROP_VOLBLOCKSIZE, NULL, zjni_long_to_Long,
	    ZFSJNI_PACKAGE_DATA "VolBlockSizeProperty",
	    "java/lang/Long" },

	{ ZPROP_INVAL, NULL, NULL, NULL, NULL },
};

/*
 * Static functions
 */

static jobject
create_BasicProperty(JNIEnv *env, zfs_handle_t *zhp, zfs_prop_t prop,
    str_to_obj_f convert_str, uint64_to_obj_f convert_uint64,
    char *propClass, char *valueClass)
{
	jobject propertyObject = NULL;
	char source[ZFS_MAX_DATASET_NAME_LEN];
	zprop_source_t srctype;
	jobject propValue = NULL;

	if (convert_str != NULL) {
		char propbuf[ZFS_MAXPROPLEN];
		int result = zfs_prop_get(zhp, prop, propbuf,
		    sizeof (propbuf), &srctype, source, sizeof (source), 1);

		if (result == 0)
			propValue = convert_str(env, propbuf);
	} else {
		uint64_t value;
		int result = zfs_prop_get_numeric(
		    zhp, prop, &value, &srctype, source, sizeof (source));

		if (result == 0)
			propValue = convert_uint64(env, value);
	}

	if (propValue != NULL) {

		jmethodID constructor;
		char signature[1024];
		jclass class = (*env)->FindClass(env, propClass);

		jstring propName = (*env)->NewStringUTF(
		    env, zfs_prop_to_name(prop));

		jboolean readOnly = zfs_prop_readonly(prop) ?
		    JNI_TRUE : JNI_FALSE;

		if (srctype == ZPROP_SRC_INHERITED) {

			jstring propSource = (*env)->NewStringUTF(env, source);

			(void) snprintf(signature, sizeof (signature),
			    "(Ljava/lang/String;L%s;ZLjava/lang/String;)V",
			    valueClass);

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

			propertyObject = (*env)->NewObject(
			    env, class, constructor, propName, propValue,
			    readOnly, propSource);
		} else {
			jobject lineage = zjni_int_to_Lineage(env, srctype);

			(void) snprintf(signature, sizeof (signature),
			    "(Ljava/lang/String;L%s;ZL"
			    ZFSJNI_PACKAGE_DATA "Property$Lineage;)V",
			    valueClass);

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

			propertyObject = (*env)->NewObject(
			    env, class, constructor, propName, propValue,
			    readOnly, lineage);
		}
	}

	return (propertyObject);
}

static jobject
create_BooleanProperty(JNIEnv *env, zfs_handle_t *zhp, zfs_prop_t prop)
{
	return (create_BasicProperty(env, zhp, prop, NULL, zjni_int_to_boolean,
	    ZFSJNI_PACKAGE_DATA "BooleanProperty", "java/lang/Boolean"));
}

static jobject
create_LongProperty(JNIEnv *env, zfs_handle_t *zhp, zfs_prop_t prop)
{
	return (create_BasicProperty(env, zhp, prop, NULL, zjni_long_to_Long,
	    ZFSJNI_PACKAGE_DATA "LongProperty", "java/lang/Long"));
}

static jobject
create_StringProperty(JNIEnv *env, zfs_handle_t *zhp, zfs_prop_t prop)
{
	return (create_BasicProperty(env, zhp, prop, str_to_string, NULL,
	    ZFSJNI_PACKAGE_DATA "StringProperty", "java/lang/String"));
}

static jobject
create_ObjectProperty(JNIEnv *env, zfs_handle_t *zhp, zfs_prop_t prop,
    str_to_obj_f convert_str, uint64_to_obj_f convert_uint64,
    char *propClass, char *valueClass)
{
	jobject propertyObject = NULL;
	char source[ZFS_MAX_DATASET_NAME_LEN];
	zprop_source_t srctype;
	jobject propValue = NULL;

	if (convert_str != NULL) {
		char propbuf[ZFS_MAXPROPLEN];
		int result = zfs_prop_get(zhp, prop, propbuf,
		    sizeof (propbuf), &srctype, source, sizeof (source), 1);

		if (result == 0)
			propValue = convert_str(env, propbuf);
	} else {
		uint64_t value;
		int result = zfs_prop_get_numeric(
		    zhp, prop, &value, &srctype, source, sizeof (source));

		if (result == 0)
			propValue = convert_uint64(env, value);
	}

	if (propValue != NULL) {

		jmethodID constructor;
		char signature[1024];
		jclass class = (*env)->FindClass(env, propClass);

		if (srctype == ZPROP_SRC_INHERITED) {

			jstring propSource = (*env)->NewStringUTF(env, source);

			(void) snprintf(signature, sizeof (signature),
			    "(L%s;Ljava/lang/String;)V", valueClass);

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

			propertyObject = (*env)->NewObject(env,
			    class, constructor, propValue, propSource);

		} else {
			jobject lineage = zjni_int_to_Lineage(env, srctype);

			(void) snprintf(signature, sizeof (signature),
			    "(L%s;L" ZFSJNI_PACKAGE_DATA "Property$Lineage;)V",
			    valueClass);

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

			propertyObject = (*env)->NewObject(env,
			    class, constructor, propValue, lineage);
		}
	}

	return (propertyObject);
}

static jobject
create_default_BasicProperty(JNIEnv *env, zfs_prop_t prop,
    str_to_obj_f convert_str, uint64_to_obj_f convert_uint64,
    char *propClass, char *valueClass)
{
	jobject propertyObject = NULL;

	if (!zfs_prop_readonly(prop)) {
		jobject propValue;

		if (convert_str != NULL) {
			char *propbuf = (char *)zfs_prop_default_string(prop);
			propValue = convert_str(env, propbuf);
		} else {
			uint64_t value = zfs_prop_default_numeric(prop);
			propValue = convert_uint64(env, value);
		}

		if (propValue != NULL) {
			char signature[1024];
			jmethodID constructor;

			jstring propName =
			    (*env)->NewStringUTF(env, zfs_prop_to_name(prop));

			jboolean readOnly = zfs_prop_readonly(prop) ?
			    JNI_TRUE : JNI_FALSE;

			jclass class = (*env)->FindClass(env, propClass);
			jobject lineage =
			    zjni_int_to_Lineage(env, ZPROP_SRC_DEFAULT);

			(void) snprintf(signature, sizeof (signature),
			    "(Ljava/lang/String;L%s;ZL" ZFSJNI_PACKAGE_DATA
			    "Property$Lineage;)V", valueClass);

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

			propertyObject = (*env)->NewObject(
			    env, class, constructor,
			    propName, propValue, readOnly, lineage);
		}
	}

	return (propertyObject);
}

static jobject
create_default_BooleanProperty(JNIEnv *env, zfs_prop_t prop)
{
	return (create_default_BasicProperty(env, prop, NULL,
	    zjni_int_to_boolean, ZFSJNI_PACKAGE_DATA "BooleanProperty",
	    "java/lang/Boolean"));
}

static jobject
create_default_LongProperty(JNIEnv *env, zfs_prop_t prop)
{
	return (create_default_BasicProperty(env, prop, NULL,
	    zjni_long_to_Long, ZFSJNI_PACKAGE_DATA "LongProperty",
	    "java/lang/Long"));
}

static jobject
create_default_StringProperty(JNIEnv *env, zfs_prop_t prop)
{
	return (create_default_BasicProperty(env, prop, str_to_string, NULL,
	    ZFSJNI_PACKAGE_DATA "StringProperty", "java/lang/String"));
}

static jobject
create_default_ObjectProperty(JNIEnv *env, zfs_prop_t prop,
    str_to_obj_f convert_str, uint64_to_obj_f convert_uint64,
    char *propClass, char *valueClass)
{
	jobject propertyObject = NULL;

	if (!zfs_prop_readonly(prop)) {
		jobject propValue;

		if (convert_str != NULL) {
			char *propbuf = (char *)zfs_prop_default_string(prop);
			propValue = convert_str(env, propbuf);
		} else {
			uint64_t value = zfs_prop_default_numeric(prop);
			propValue = convert_uint64(env, value);
		}

		if (propValue != NULL) {
			char signature[1024];
			jmethodID constructor;

			jclass class = (*env)->FindClass(env, propClass);
			jobject lineage =
			    zjni_int_to_Lineage(env, ZPROP_SRC_DEFAULT);

			(void) snprintf(signature, sizeof (signature),
			    "(L%s;L" ZFSJNI_PACKAGE_DATA "Property$Lineage;)V",
			    valueClass);

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

			propertyObject = (*env)->NewObject(
			    env, class, constructor, propValue, lineage);
		}
	}

	return (propertyObject);
}

static jobject
str_to_enum_element(JNIEnv *env, char *str, char *valueClass)
{
	char signature[1024];
	jmethodID method_valueOf;

	jstring utf = (*env)->NewStringUTF(env, str);
	jclass class = (*env)->FindClass(env, valueClass);

	(void) snprintf(signature, sizeof (signature),
	    "(Ljava/lang/String;)L%s;", valueClass);

	method_valueOf = (*env)->GetStaticMethodID(
	    env, class, "valueOf", signature);

	return (*env)->CallStaticObjectMethod(env, class, method_valueOf, utf);
}

static jobject
str_to_aclinherit(JNIEnv *env, char *str)
{
	return (str_to_enum_element(env, str,
	    ZFSJNI_PACKAGE_DATA "AclInheritProperty$AclInherit"));
}

static jobject
str_to_aclmode(JNIEnv *env, char *str)
{
	return (str_to_enum_element(env, str,
	    ZFSJNI_PACKAGE_DATA "AclModeProperty$AclMode"));
}

static jobject
str_to_checksum(JNIEnv *env, char *str)
{
	return (str_to_enum_element(env, str,
	    ZFSJNI_PACKAGE_DATA "ChecksumProperty$Checksum"));
}

static jobject
str_to_compression(JNIEnv *env, char *str)
{
	return (str_to_enum_element(env, str,
	    ZFSJNI_PACKAGE_DATA "CompressionProperty$Compression"));
}

static jobject
str_to_snapdir(JNIEnv *env, char *str)
{
	return (str_to_enum_element(env, str,
	    ZFSJNI_PACKAGE_DATA "SnapDirProperty$SnapDir"));
}

static jobject
str_to_string(JNIEnv *env, char *str)
{
	return (*env)->NewStringUTF(env, str);
}

/*
 * Package-private functions
 */

jobject
zjni_get_default_property(JNIEnv *env, zfs_prop_t prop)
{
	int i;
	for (i = 0; props_boolean[i] != ZPROP_INVAL; i++) {
		if (prop == props_boolean[i]) {
			return (create_default_BooleanProperty(env, prop));
		}
	}

	for (i = 0; props_long[i] != ZPROP_INVAL; i++) {
		if (prop == props_long[i]) {
			return (create_default_LongProperty(env, prop));
		}
	}

	for (i = 0; props_string[i] != ZPROP_INVAL; i++) {
		if (prop == props_string[i]) {
			return (create_default_StringProperty(env, prop));
		}
	}

	for (i = 0; props_custom[i].prop != ZPROP_INVAL; i++) {
		if (prop == props_custom[i].prop) {
			return create_default_ObjectProperty(env,
			    props_custom[i].prop,
			    props_custom[i].convert_str,
			    props_custom[i].convert_uint64,
			    props_custom[i].propClass,
			    props_custom[i].valueClass);
		}
	}

	return (NULL);
}

static int
zjni_get_property_from_name_cb(int prop, void *cb)
{
	const char *name = cb;

	if (strcasecmp(name, zfs_prop_to_name(prop)) == 0)
		return (prop);

	return (ZPROP_CONT);
}

zfs_prop_t
zjni_get_property_from_name(const char *name)
{
	zfs_prop_t prop;

	prop = zprop_iter(zjni_get_property_from_name_cb, (void *)name,
	    B_FALSE, B_FALSE, ZFS_TYPE_DATASET);
	return (prop == ZPROP_CONT ? ZPROP_INVAL : prop);
}

jobject
zjni_int_to_Lineage(JNIEnv *env, zprop_source_t srctype)
{
	/* zprop_source_t to Property$Lineage map */
	static zjni_field_mapping_t lineage_map[] = {
		{ ZPROP_SRC_NONE, "ZFS_PROP_LINEAGE_NOTINHERITABLE" },
		{ ZPROP_SRC_DEFAULT, "ZFS_PROP_LINEAGE_DEFAULT" },
		{ ZPROP_SRC_LOCAL, "ZFS_PROP_LINEAGE_LOCAL" },
		{ ZPROP_SRC_TEMPORARY, "ZFS_PROP_LINEAGE_TEMPORARY" },
		{ ZPROP_SRC_INHERITED, "ZFS_PROP_LINEAGE_INHERITED" }
	};

	return (zjni_int_to_enum(env, srctype,
	    ZFSJNI_PACKAGE_DATA "Property$Lineage",
	    "ZFS_PROP_LINEAGE_INHERITED", lineage_map));
}

jobjectArray
zjni_get_Dataset_properties(JNIEnv *env, zfs_handle_t *zhp)
{
	jobject prop;
	int i;

	/* Create an array list for the properties */
	zjni_ArrayList_t proplist_obj = {0};
	zjni_ArrayList_t *proplist = &proplist_obj;
	zjni_new_ArrayList(env, proplist);

	for (i = 0; props_boolean[i] != ZPROP_INVAL; i++) {
		/* Create property and add to list */
		prop = create_BooleanProperty(env, zhp, props_boolean[i]);

		/* Does this property apply to this object? */
		if (prop != NULL) {

			(*env)->CallBooleanMethod(
			    env, ((zjni_Object_t *)proplist)->object,
			    ((zjni_Collection_t *)proplist)->method_add, prop);
		} else {

			if ((*env)->ExceptionOccurred(env) != NULL) {
				return (NULL);
			}
#ifdef	DEBUG
			(void) fprintf(stderr, "Property %s is not appropriate "
			    "for %s\n", zfs_prop_to_name(props_boolean[i]),
			    zfs_get_name(zhp));
#endif
		}
	}

	for (i = 0; props_long[i] != ZPROP_INVAL; i++) {
		/* Create property and add to list */
		prop = create_LongProperty(env, zhp, props_long[i]);

		/* Does this property apply to this object? */
		if (prop != NULL) {

			(*env)->CallBooleanMethod(
			    env, ((zjni_Object_t *)proplist)->object,
			    ((zjni_Collection_t *)proplist)->method_add, prop);
		} else {
			if ((*env)->ExceptionOccurred(env) != NULL) {
				return (NULL);
			}
#ifdef	DEBUG
			(void) fprintf(stderr, "Property %s is not appropriate "
			    "for %s\n", zfs_prop_to_name(props_long[i]),
			    zfs_get_name(zhp));
#endif
		}
	}

	for (i = 0; props_string[i] != ZPROP_INVAL; i++) {
		/* Create property and add to list */
		prop = create_StringProperty(env, zhp, props_string[i]);

		/* Does this property apply to this object? */
		if (prop != NULL) {

			(*env)->CallBooleanMethod(
			    env, ((zjni_Object_t *)proplist)->object,
			    ((zjni_Collection_t *)proplist)->method_add, prop);
		} else {
			if ((*env)->ExceptionOccurred(env) != NULL) {
				return (NULL);
			}
#ifdef	DEBUG
			(void) fprintf(stderr, "Property %s is not appropriate "
			    "for %s\n", zfs_prop_to_name(props_string[i]),
			    zfs_get_name(zhp));
#endif
		}
	}

	for (i = 0; props_custom[i].prop != ZPROP_INVAL; i++) {
		/* Create property and add to list */
		prop = create_ObjectProperty(env, zhp, props_custom[i].prop,
		    props_custom[i].convert_str, props_custom[i].convert_uint64,
		    props_custom[i].propClass, props_custom[i].valueClass);

		/* Does this property apply to this object? */
		if (prop != NULL) {

			(*env)->CallBooleanMethod(
			    env, ((zjni_Object_t *)proplist)->object,
			    ((zjni_Collection_t *)proplist)->method_add, prop);
		} else {
			if ((*env)->ExceptionOccurred(env) != NULL) {
				return (NULL);
			}
#ifdef	DEBUG
			(void) fprintf(stderr, "Property %s is not appropriate "
			    "for %s\n", zfs_prop_to_name(props_custom[i].prop),
			    zfs_get_name(zhp));
#endif
		}
	}

	return (zjni_Collection_to_array(env,
	    (zjni_Collection_t *)proplist, ZFSJNI_PACKAGE_DATA "Property"));
}