/*

 * Copyright (c) 2000, 2015, 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"

#ifdef __cplusplus

extern "C" {

#endif

/*

 * Declare library specific JNI_Onload entry if static build

 */

DEF_STATIC_JNI_OnLoad

    JNIEXPORT jintArray JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegOpenKey

               (JNIEnv* env, jclass this_class, jint hKey, jbyteArray lpSubKey, jint securityMask) {

        HKEY handle;

        char* str;

        int tmp[2];

        int errorCode=-1;

        jintArray result;

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

        CHECK_NULL_RETURN(str, NULL);

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

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

        tmp[0]= (int) handle;

        tmp[1]= errorCode;

        result = (*env)->NewIntArray(env,2);

        if (result != NULL) {

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

        }

        return result;

    }

    JNIEXPORT jint JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegCloseKey

               (JNIEnv* env, jclass this_class, jint hKey) {

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

    };

    JNIEXPORT jintArray JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegCreateKeyEx

               (JNIEnv* env, jclass this_class, jint hKey, jbyteArray lpSubKey) {

        HKEY handle;

        char* str;

        int tmp[3];

        DWORD lpdwDisposition;

        int errorCode;

        jintArray result = NULL;

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

        CHECK_NULL_RETURN(str, NULL);

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

                      REG_OPTION_NON_VOLATILE, KEY_READ,

                      NULL, &handle, &lpdwDisposition);

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

        tmp[0]= (int) handle;

        tmp[1]= errorCode;

        tmp[2]= lpdwDisposition;

        result = (*env)->NewIntArray(env,3);

        if (result != NULL) {

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

        }

        return result;

    }

    JNIEXPORT jint JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegDeleteKey

              (JNIEnv* env, jclass this_class, jint hKey, jbyteArray lpSubKey) {

        char* str;

        int result;

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

        CHECK_NULL_RETURN(str, -1);

        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, jint hKey) {

        return RegFlushKey ((HKEY)hKey);

        }

    JNIEXPORT jbyteArray JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegQueryValueEx

         (JNIEnv* env, jclass this_class, jint hKey, jbyteArray valueName) {

        char* valueNameStr;

        char* buffer;

        jbyteArray result;

        DWORD valueType;

        DWORD valueSize;

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

        CHECK_NULL_RETURN(valueNameStr, NULL);

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

                                                 &valueSize) != ERROR_SUCCESS) {

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

        return NULL;

        }

        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;

        }

        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, jint hKey, jbyteArray valueName, jbyteArray data) {

        char* valueNameStr;

        char* dataStr;

        int size = -1;

        int nameSize = -1;

        int error_code = -1;

        if ((valueName == NULL)||(data == NULL)) {return -1;}

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

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

        CHECK_NULL_RETURN(dataStr, -1);

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

        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, jint hKey, jbyteArray valueName) {

        char* valueNameStr;

        int error_code = -1;

        if (valueName == NULL) {return -1;}

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

        CHECK_NULL_RETURN(valueNameStr, -1);

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

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

        return error_code;

     }

    JNIEXPORT jintArray JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegQueryInfoKey

                                  (JNIEnv* env, jclass this_class, jint hKey) {

        jintArray result = NULL;

        int tmp[5];

        int valuesNumber = -1;

        int maxValueNameLength = -1;

        int maxSubKeyLength = -1;

        int subKeysNumber = -1;

        int errorCode = -1;

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

                 &subKeysNumber, &maxSubKeyLength, NULL,

                 &valuesNumber, &maxValueNameLength,

                 NULL, NULL, NULL);

        tmp[0]= subKeysNumber;

        tmp[1]= (int)errorCode;

        tmp[2]= valuesNumber;

        tmp[3]= maxSubKeyLength;

        tmp[4]= maxValueNameLength;

        result = (*env)->NewIntArray(env,5);

        if (result != NULL) {

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

        }

        return result;

    }

     JNIEXPORT jbyteArray JNICALL Java_java_util_prefs_WindowsPreferences_WindowsRegEnumKeyEx

     (JNIEnv* env, jclass this_class, jint hKey , jint subKeyIndex, jint maxKeyLength) {

        int size = maxKeyLength;

        jbyteArray result;

        char* buffer = NULL;

        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;

        }

        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, jint hKey , jint valueIndex, jint maxValueNameLength){

          int size = maxValueNameLength;

          jbyteArray result;

          char* buffer = NULL;

          int error_code;

          buffer = (char*)malloc(maxValueNameLength);

          if (buffer == NULL) {

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

              return NULL;

          }

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

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

          if (error_code!= ERROR_SUCCESS){

            free(buffer);

            return NULL;

          }

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

          if (result != NULL) {

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

          }

          free(buffer);

          return result;

     }

#ifdef __cplusplus

}

#endif