/*

 * Copyright (c) 2000, 2018, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

#include <stdlib.h>

#include <windows.h>

#include "jni.h"

#include "jni_util.h"

#include "jvm.h"

#include "java_util_prefs_WindowsPreferences.h"

#ifdef __cplusplus

extern "C" {

#endif

/*

 * Declare library specific JNI_Onload entry if static build

 */

DEF_STATIC_JNI_OnLoad

JNIEXPORT jlongArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegOpenKey(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray lpSubKey, jint securityMask) {

    char* str = (*env)->GetByteArrayElements(env, lpSubKey, NULL);

    CHECK_NULL_RETURN(str, NULL);

    HKEY handle;

    int errorCode = RegOpenKeyEx((HKEY) hKey, str, 0, securityMask, &handle);

    (*env)->ReleaseByteArrayElements(env, lpSubKey, str, 0);

    jlong tmp[2];

    tmp[0] = (jlong) handle;

    tmp[1] = errorCode;

    jlongArray result = (*env)->NewLongArray(env, 2);

    if (result != NULL) {

        (*env)->SetLongArrayRegion(env, result, 0, 2, tmp);

    }

    return result;

}

JNIEXPORT jint JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegCloseKey(JNIEnv* env,

    jclass this_class, jlong hKey) {

    return (jint) RegCloseKey((HKEY) hKey);

};

JNIEXPORT jlongArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegCreateKeyEx(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray lpSubKey) {

    char* str = (*env)->GetByteArrayElements(env, lpSubKey, NULL);

    CHECK_NULL_RETURN(str, NULL);

    HKEY handle;

    DWORD lpdwDisposition;

    int errorCode = RegCreateKeyEx((HKEY) hKey, str, 0, NULL,

        REG_OPTION_NON_VOLATILE, KEY_READ,

        NULL, &handle, &lpdwDisposition);

    (*env)->ReleaseByteArrayElements(env, lpSubKey, str, 0);

    jlong tmp[3];

    tmp[0] = (jlong) handle;

    tmp[1] = errorCode;

    tmp[2] = lpdwDisposition;

    jlongArray result = (*env)->NewLongArray(env, 3);

    if (result != NULL) {

        (*env)->SetLongArrayRegion(env, result, 0, 3, tmp);

    }

    return result;

}

JNIEXPORT jint JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegDeleteKey(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray lpSubKey) {

    char* str = (*env)->GetByteArrayElements(env, lpSubKey, NULL);

    CHECK_NULL_RETURN(str, -1);

    int result = RegDeleteKey((HKEY) hKey, str);

    (*env)->ReleaseByteArrayElements(env, lpSubKey, str, 0);

    return result;

};

JNIEXPORT jint JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegFlushKey(JNIEnv* env,

    jclass this_class, jlong hKey) {

    return RegFlushKey((HKEY) hKey);

}

JNIEXPORT jbyteArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegQueryValueEx(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray valueName) {

    char* valueNameStr = (*env)->GetByteArrayElements(env, valueName, NULL);

    CHECK_NULL_RETURN(valueNameStr, NULL);

    DWORD valueType;

    DWORD valueSize;

    if (RegQueryValueEx((HKEY) hKey, valueNameStr, NULL, &valueType, NULL,

        &valueSize) != ERROR_SUCCESS) {

        (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

        return NULL;

    }

    char* buffer = (char*) malloc(valueSize);

    if (buffer != NULL) {

        if (RegQueryValueEx((HKEY) hKey, valueNameStr, NULL, &valueType, buffer,

            &valueSize) != ERROR_SUCCESS) {

            free(buffer);

            (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

            return NULL;

        }

    } else {

        JNU_ThrowOutOfMemoryError(env, "native memory allocation failed");

        (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

        return NULL;

    }

    jbyteArray result;

    if (valueType == REG_SZ) {

        result = (*env)->NewByteArray(env, valueSize);

        if (result != NULL) {

            (*env)->SetByteArrayRegion(env, result, 0, valueSize, buffer);

        }

    } else {

        result = NULL;

    }

    free(buffer);

    (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

    return result;

}

JNIEXPORT jint JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegSetValueEx(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray valueName, jbyteArray data) {

    if ((valueName == NULL) || (data == NULL)) {

        return -1;

    }

    int size = (*env)->GetArrayLength(env, data);

    char* dataStr = (*env)->GetByteArrayElements(env, data, NULL);

    CHECK_NULL_RETURN(dataStr, -1);

    char* valueNameStr = (*env)->GetByteArrayElements(env, valueName, NULL);

    int error_code = -1;

    if (valueNameStr != NULL) {

        error_code = RegSetValueEx((HKEY) hKey, valueNameStr, 0,

            REG_SZ, dataStr, size);

        (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

    }

    (*env)->ReleaseByteArrayElements(env, data, dataStr, 0);

    return error_code;

}

JNIEXPORT jint JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegDeleteValue(JNIEnv* env,

    jclass this_class, jlong hKey, jbyteArray valueName) {

    if (valueName == NULL) {

        return -1;

    }

    char* valueNameStr = (*env)->GetByteArrayElements(env, valueName, NULL);

    CHECK_NULL_RETURN(valueNameStr, -1);

    int error_code = RegDeleteValue((HKEY) hKey, valueNameStr);

    (*env)->ReleaseByteArrayElements(env, valueName, valueNameStr, 0);

    return error_code;

}

JNIEXPORT jlongArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegQueryInfoKey(JNIEnv* env,

    jclass this_class, jlong hKey) {

    int subKeysNumber;

    int maxSubKeyLength;

    int valuesNumber;

    int maxValueNameLength;

    int errorCode = RegQueryInfoKey((HKEY) hKey, NULL, NULL, NULL,

        &subKeysNumber, &maxSubKeyLength, NULL,

        &valuesNumber, &maxValueNameLength,

        NULL, NULL, NULL);

    jlong tmp[5];

    tmp[0] = subKeysNumber;

    tmp[1] = errorCode;

    tmp[2] = valuesNumber;

    tmp[3] = maxSubKeyLength;

    tmp[4] = maxValueNameLength;

    jintArray result = (*env)->NewLongArray(env, 5);

    if (result != NULL) {

        (*env)->SetLongArrayRegion(env, result, 0, 5, tmp);

    }

    return result;

}

JNIEXPORT jbyteArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegEnumKeyEx(JNIEnv* env,

    jclass this_class, jlong hKey, jint subKeyIndex, jint maxKeyLength) {

    int size = maxKeyLength;

    char* buffer = (char*) malloc(maxKeyLength);

    if (buffer == NULL) {

        JNU_ThrowOutOfMemoryError(env, "native memory allocation failed");

        return NULL;

    }

    if (RegEnumKeyEx((HKEY) hKey, subKeyIndex, buffer, &size, NULL, NULL,

        NULL, NULL) != ERROR_SUCCESS) {

        free(buffer);

        return NULL;

    }

    jbyteArray result = (*env)->NewByteArray(env, size + 1);

    if (result != NULL) {

        (*env)->SetByteArrayRegion(env, result, 0, size + 1, buffer);

    }

    free(buffer);

    return result;

}

JNIEXPORT jbyteArray JNICALL

Java_java_util_prefs_WindowsPreferences_WindowsRegEnumValue(JNIEnv* env,

    jclass this_class, jlong hKey, jint valueIndex, jint maxValueNameLength) {

    int size = maxValueNameLength;

    char* buffer = (char*) malloc(maxValueNameLength);

    if (buffer == NULL) {

        JNU_ThrowOutOfMemoryError(env, "native memory allocation failed");

        return NULL;

    }

    int error_code = RegEnumValue((HKEY) hKey, valueIndex, buffer,

        &size, NULL, NULL, NULL, NULL);

    if (error_code != ERROR_SUCCESS) {

        free(buffer);

        return NULL;

    }

    jbyteArray result = (*env)->NewByteArray(env, size + 1);

    if (result != NULL) {

        (*env)->SetByteArrayRegion(env, result, 0, size + 1, buffer);

    }

    free(buffer);

    return result;

}

#ifdef __cplusplus

}

#endif