/*

 * Copyright (c) 1997, 2014, 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 <windows.h>

#include <io.h>

#include <process.h>

#include <stdlib.h>

#include <stdio.h>

#include <stdarg.h>

#include <string.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <wtypes.h>

#include <commctrl.h>

#include <jni.h>

#include "java.h"

#define JVM_DLL "jvm.dll"

#define JAVA_DLL "java.dll"

/*

 * Prototypes.

 */

static jboolean GetPublicJREHome(char *path, jint pathsize);

static jboolean GetJVMPath(const char *jrepath, const char *jvmtype,

                           char *jvmpath, jint jvmpathsize);

static jboolean GetJREPath(char *path, jint pathsize);

/* We supports warmup for UI stack that is performed in parallel

 * to VM initialization.

 * This helps to improve startup of UI application as warmup phase

 * might be long due to initialization of OS or hardware resources.

 * It is not CPU bound and therefore it does not interfere with VM init.

 * Obviously such warmup only has sense for UI apps and therefore it needs

 * to be explicitly requested by passing -Dsun.awt.warmup=true property

 * (this is always the case for plugin/javaws).

 *

 * Implementation launches new thread after VM starts and use it to perform

 * warmup code (platform dependent).

 * This thread is later reused as AWT toolkit thread as graphics toolkit

 * often assume that they are used from the same thread they were launched on.

 *

 * At the moment we only support warmup for D3D. It only possible on windows

 * and only if other flags do not prohibit this (e.g. OpenGL support requested).

 */

#undef ENABLE_AWT_PRELOAD

#ifndef JAVA_ARGS /* turn off AWT preloading for javac, jar, etc */

    /* CR6999872: fastdebug crashes if awt library is loaded before JVM is

     * initialized*/

    #if !defined(DEBUG)

        #define ENABLE_AWT_PRELOAD

    #endif

#endif

#ifdef ENABLE_AWT_PRELOAD

/* "AWT was preloaded" flag;

 * turned on by AWTPreload().

 */

int awtPreloaded = 0;

/* Calls a function with the name specified

 * the function must be int(*fn)(void).

 */

int AWTPreload(const char *funcName);

/* stops AWT preloading */

void AWTPreloadStop();

/* D3D preloading */

/* -1: not initialized; 0: OFF, 1: ON */

int awtPreloadD3D = -1;

/* command line parameter to swith D3D preloading on */

#define PARAM_PRELOAD_D3D "-Dsun.awt.warmup"

/* D3D/OpenGL management parameters */

#define PARAM_NODDRAW "-Dsun.java2d.noddraw"

#define PARAM_D3D "-Dsun.java2d.d3d"

#define PARAM_OPENGL "-Dsun.java2d.opengl"

/* funtion in awt.dll (src/windows/native/sun/java2d/d3d/D3DPipelineManager.cpp) */

#define D3D_PRELOAD_FUNC "preloadD3D"

/* Extracts value of a parameter with the specified name

 * from command line argument (returns pointer in the argument).

 * Returns NULL if the argument does not contains the parameter.

 * e.g.:

 * GetParamValue("theParam", "theParam=value") returns pointer to "value".

 */

const char * GetParamValue(const char *paramName, const char *arg) {

    int nameLen = JLI_StrLen(paramName);

    if (JLI_StrNCmp(paramName, arg, nameLen) == 0) {

        /* arg[nameLen] is valid (may contain final NULL) */

        if (arg[nameLen] == '=') {

            return arg + nameLen + 1;

        }

    }

    return NULL;

}

/* Checks if commandline argument contains property specified

 * and analyze it as boolean property (true/false).

 * Returns -1 if the argument does not contain the parameter;

 * Returns 1 if the argument contains the parameter and its value is "true";

 * Returns 0 if the argument contains the parameter and its value is "false".

 */

int GetBoolParamValue(const char *paramName, const char *arg) {

    const char * paramValue = GetParamValue(paramName, arg);

    if (paramValue != NULL) {

        if (JLI_StrCaseCmp(paramValue, "true") == 0) {

            return 1;

        }

        if (JLI_StrCaseCmp(paramValue, "false") == 0) {

            return 0;

        }

    }

    return -1;

}

#endif /* ENABLE_AWT_PRELOAD */

static jboolean _isjavaw = JNI_FALSE;

jboolean

IsJavaw()

{

    return _isjavaw;

}

/*

 * Returns the arch path, to get the current arch use the

 * macro GetArch, nbits here is ignored for now.

 */

const char *

GetArchPath(int nbits)

{

#ifdef _M_AMD64

    return "amd64";

#elif defined(_M_IA64)

    return "ia64";

#else

    return "i386";

#endif

}

/*

 *

 */

void

CreateExecutionEnvironment(int *pargc, char ***pargv,

                           char *jrepath, jint so_jrepath,

                           char *jvmpath, jint so_jvmpath,

                           char *jvmcfg,  jint so_jvmcfg) {

    char * jvmtype;

    int i = 0;

    int running = CURRENT_DATA_MODEL;

    int wanted = running;

    char** argv = *pargv;

    for (i = 1; i < *pargc ; i++) {

        if (JLI_StrCmp(argv[i], "-J-d64") == 0 || JLI_StrCmp(argv[i], "-d64") == 0) {

            wanted = 64;

            continue;

        }

        if (JLI_StrCmp(argv[i], "-J-d32") == 0 || JLI_StrCmp(argv[i], "-d32") == 0) {

            wanted = 32;

            continue;

        }

        if (IsJavaArgs() && argv[i][0] != '-')

            continue;

        if (argv[i][0] != '-')

            break;

    }

    if (running != wanted) {

        JLI_ReportErrorMessage(JRE_ERROR2, wanted);

        exit(1);

    }

    /* Find out where the JRE is that we will be using. */

    if (!GetJREPath(jrepath, so_jrepath)) {

        JLI_ReportErrorMessage(JRE_ERROR1);

        exit(2);

    }

    JLI_Snprintf(jvmcfg, so_jvmcfg, "%s%slib%s%s%sjvm.cfg",

        jrepath, FILESEP, FILESEP, (char*)GetArch(), FILESEP);

    /* Find the specified JVM type */

    if (ReadKnownVMs(jvmcfg, JNI_FALSE) < 1) {

        JLI_ReportErrorMessage(CFG_ERROR7);

        exit(1);

    }

    jvmtype = CheckJvmType(pargc, pargv, JNI_FALSE);

    if (JLI_StrCmp(jvmtype, "ERROR") == 0) {

        JLI_ReportErrorMessage(CFG_ERROR9);

        exit(4);

    }

    jvmpath[0] = '\0';

    if (!GetJVMPath(jrepath, jvmtype, jvmpath, so_jvmpath)) {

        JLI_ReportErrorMessage(CFG_ERROR8, jvmtype, jvmpath);

        exit(4);

    }

    /* If we got here, jvmpath has been correctly initialized. */

    /* Check if we need preload AWT */

#ifdef ENABLE_AWT_PRELOAD

    argv = *pargv;

    for (i = 0; i < *pargc ; i++) {

        /* Tests the "turn on" parameter only if not set yet. */

        if (awtPreloadD3D < 0) {

            if (GetBoolParamValue(PARAM_PRELOAD_D3D, argv[i]) == 1) {

                awtPreloadD3D = 1;

            }

        }

        /* Test parameters which can disable preloading if not already disabled. */

        if (awtPreloadD3D != 0) {

            if (GetBoolParamValue(PARAM_NODDRAW, argv[i]) == 1

                || GetBoolParamValue(PARAM_D3D, argv[i]) == 0

                || GetBoolParamValue(PARAM_OPENGL, argv[i]) == 1)

            {

                awtPreloadD3D = 0;

                /* no need to test the rest of the parameters */

                break;

            }

        }

    }

#endif /* ENABLE_AWT_PRELOAD */

}

static jboolean

LoadMSVCRT()

{

    // Only do this once

    static int loaded = 0;

    char crtpath[MAXPATHLEN];

    if (!loaded) {

        /*

         * The Microsoft C Runtime Library needs to be loaded first.  A copy is

         * assumed to be present in the "JRE path" directory.  If it is not found

         * there (or "JRE path" fails to resolve), skip the explicit load and let

         * nature take its course, which is likely to be a failure to execute.

         */

        if (GetJREPath(crtpath, MAXPATHLEN)) {

            if (JLI_StrLen(crtpath) + JLI_StrLen("\\bin\\") +

                JLI_ReportErrorMessage(JRE_ERROR11);

                return JNI_FALSE;

            }

            JLI_TraceLauncher("CRT path is %s\n", crtpath);

            if (_access(crtpath, 0) == 0) {

                if (LoadLibrary(crtpath) == 0) {

                    JLI_ReportErrorMessage(DLL_ERROR4, crtpath);

                    return JNI_FALSE;

                }

            }

        }

        loaded = 1;

    }

    return JNI_TRUE;

}

/*

 * Find path to JRE based on .exe's location or registry settings.

 */

jboolean

GetJREPath(char *path, jint pathsize)

{

    char javadll[MAXPATHLEN];

    struct stat s;

    if (GetApplicationHome(path, pathsize)) {

        /* Is JRE co-located with the application? */

        JLI_Snprintf(javadll, sizeof(javadll), "%s\\bin\\" JAVA_DLL, path);

        if (stat(javadll, &s) == 0) {

            JLI_TraceLauncher("JRE path is %s\n", path);

            return JNI_TRUE;

        }

        /* Does this app ship a private JRE in <apphome>\jre directory? */

        JLI_Snprintf(javadll, sizeof (javadll), "%s\\jre\\bin\\" JAVA_DLL, path);

        if (stat(javadll, &s) == 0) {

            JLI_StrCat(path, "\\jre");

            JLI_TraceLauncher("JRE path is %s\n", path);

            return JNI_TRUE;

        }

    }

    /* Look for a public JRE on this machine. */

    if (GetPublicJREHome(path, pathsize)) {

        JLI_TraceLauncher("JRE path is %s\n", path);

        return JNI_TRUE;

    }

    JLI_ReportErrorMessage(JRE_ERROR8 JAVA_DLL);

    return JNI_FALSE;

}

/*

 * Given a JRE location and a JVM type, construct what the name the

 * JVM shared library will be.  Return true, if such a library

 * exists, false otherwise.

 */

static jboolean

GetJVMPath(const char *jrepath, const char *jvmtype,

           char *jvmpath, jint jvmpathsize)

{

    struct stat s;

    if (JLI_StrChr(jvmtype, '/') || JLI_StrChr(jvmtype, '\\')) {

        JLI_Snprintf(jvmpath, jvmpathsize, "%s\\" JVM_DLL, jvmtype);

    } else {

        JLI_Snprintf(jvmpath, jvmpathsize, "%s\\bin\\%s\\" JVM_DLL,

                     jrepath, jvmtype);

    }

    if (stat(jvmpath, &s) == 0) {

        return JNI_TRUE;

    } else {

        return JNI_FALSE;

    }

}

/*

 * Load a jvm from "jvmpath" and initialize the invocation functions.

 */

jboolean

LoadJavaVM(const char *jvmpath, InvocationFunctions *ifn)

{

    HINSTANCE handle;

    JLI_TraceLauncher("JVM path is %s\n", jvmpath);

    /*

     * The Microsoft C Runtime Library needs to be loaded first.  A copy is

     * assumed to be present in the "JRE path" directory.  If it is not found

     * there (or "JRE path" fails to resolve), skip the explicit load and let

     * nature take its course, which is likely to be a failure to execute.

     *

     */

    LoadMSVCRT();

    /* Load the Java VM DLL */

    if ((handle = LoadLibrary(jvmpath)) == 0) {

        JLI_ReportErrorMessage(DLL_ERROR4, (char *)jvmpath);

        return JNI_FALSE;

    }

    /* Now get the function addresses */

    ifn->CreateJavaVM =

        (void *)GetProcAddress(handle, "JNI_CreateJavaVM");

    ifn->GetDefaultJavaVMInitArgs =

        (void *)GetProcAddress(handle, "JNI_GetDefaultJavaVMInitArgs");

    if (ifn->CreateJavaVM == 0 || ifn->GetDefaultJavaVMInitArgs == 0) {

        JLI_ReportErrorMessage(JNI_ERROR1, (char *)jvmpath);

        return JNI_FALSE;

    }

    return JNI_TRUE;

}

/*

 * If app is "c:\foo\bin\javac", then put "c:\foo" into buf.

 */

jboolean

GetApplicationHome(char *buf, jint bufsize)

{

    char *cp;

    GetModuleFileName(0, buf, bufsize);

    *JLI_StrRChr(buf, '\\') = '\0'; /* remove .exe file name */

    if ((cp = JLI_StrRChr(buf, '\\')) == 0) {

        /* This happens if the application is in a drive root, and

         * there is no bin directory. */

        buf[0] = '\0';

        return JNI_FALSE;

    }

    *cp = '\0';  /* remove the bin\ part */

    return JNI_TRUE;

}

/*

 * Helpers to look in the registry for a public JRE.

 */

                    /* Same for 1.5.0, 1.5.1, 1.5.2 etc. */

#define JRE_KEY     "Software\\JavaSoft\\Java Runtime Environment"

static jboolean

GetStringFromRegistry(HKEY key, const char *name, char *buf, jint bufsize)

{

    DWORD type, size;

    if (RegQueryValueEx(key, name, 0, &type, 0, &size) == 0

        && type == REG_SZ

        && (size < (unsigned int)bufsize)) {

        if (RegQueryValueEx(key, name, 0, 0, buf, &size) == 0) {

            return JNI_TRUE;

        }

    }

    return JNI_FALSE;

}

static jboolean

GetPublicJREHome(char *buf, jint bufsize)

{

    HKEY key, subkey;

    char version[MAXPATHLEN];

    /*

     * Note: There is a very similar implementation of the following

     * registry reading code in the Windows java control panel (javacp.cpl).

     * If there are bugs here, a similar bug probably exists there.  Hence,

     * changes here require inspection there.

     */

    /* Find the current version of the JRE */

    if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, JRE_KEY, 0, KEY_READ, &key) != 0) {

        JLI_ReportErrorMessage(REG_ERROR1, JRE_KEY);

        return JNI_FALSE;

    }

    if (!GetStringFromRegistry(key, "CurrentVersion",

                               version, sizeof(version))) {

        JLI_ReportErrorMessage(REG_ERROR2, JRE_KEY);

        RegCloseKey(key);

        return JNI_FALSE;

    }

    if (JLI_StrCmp(version, GetDotVersion()) != 0) {

        JLI_ReportErrorMessage(REG_ERROR3, JRE_KEY, version, GetDotVersion()

        );

        RegCloseKey(key);

        return JNI_FALSE;

    }

    /* Find directory where the current version is installed. */

    if (RegOpenKeyEx(key, version, 0, KEY_READ, &subkey) != 0) {

        JLI_ReportErrorMessage(REG_ERROR1, JRE_KEY, version);

        RegCloseKey(key);

        return JNI_FALSE;

    }

    if (!GetStringFromRegistry(subkey, "JavaHome", buf, bufsize)) {

        JLI_ReportErrorMessage(REG_ERROR4, JRE_KEY, version);

        RegCloseKey(key);

        RegCloseKey(subkey);

        return JNI_FALSE;

    }

    if (JLI_IsTraceLauncher()) {

        char micro[MAXPATHLEN];

        if (!GetStringFromRegistry(subkey, "MicroVersion", micro,

                                   sizeof(micro))) {

            printf("Warning: Can't read MicroVersion\n");

            micro[0] = '\0';

        }

        printf("Version major.minor.micro = %s.%s\n", version, micro);

    }

    RegCloseKey(key);

    RegCloseKey(subkey);

    return JNI_TRUE;