/*

 * 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.

 */

/* Access APIs for Windows Vista and above */

#ifndef _WIN32_WINNT

#define _WIN32_WINNT 0x0601

#endif

#include "jni.h"

#include "jni_util.h"

#include <windows.h>

#include <shlobj.h>

#include <objidl.h>

#include <locale.h>

#include <sys/types.h>

#include <sys/timeb.h>

#include <tchar.h>

#include <stdlib.h>

#include <Wincon.h>

#include "locale_str.h"

#include "java_props.h"

#ifndef VER_PLATFORM_WIN32_WINDOWS

#define VER_PLATFORM_WIN32_WINDOWS 1

#endif

#ifndef PROCESSOR_ARCHITECTURE_AMD64

#define PROCESSOR_ARCHITECTURE_AMD64 9

#endif

typedef void (WINAPI *PGNSI)(LPSYSTEM_INFO);

static boolean SetupI18nProps(LCID lcid, char** language, char** script, char** country,

               char** variant, char** encoding);

#define PROPSIZE 9      // eight-letter + null terminator

#define SNAMESIZE 86    // max number of chars for LOCALE_SNAME is 85

static char *

getEncodingInternal(LCID lcid)

{

    int codepage;

    char * ret = malloc(16);

    if (ret == NULL) {

        return NULL;

    }

    if (GetLocaleInfo(lcid,

                      LOCALE_IDEFAULTANSICODEPAGE,

                      ret+2, 14) == 0) {

        codepage = 1252;

    } else {

        codepage = atoi(ret+2);

    }

    switch (codepage) {

    case 0:

        strcpy(ret, "UTF-8");

        break;

    case 874:     /*  9:Thai     */

    case 932:     /* 10:Japanese */

    case 949:     /* 12:Korean Extended Wansung */

    case 950:     /* 13:Chinese (Taiwan, Hongkong, Macau) */

    case 1361:    /* 15:Korean Johab */

        ret[0] = 'M';

        ret[1] = 'S';

        break;

    case 936:

        strcpy(ret, "GBK");

        break;

    case 54936:

        strcpy(ret, "GB18030");

        break;

    default:

        ret[0] = 'C';

        ret[1] = 'p';

        break;

    }

    //Traditional Chinese Windows should use MS950_HKSCS_XP as the

    //default encoding, if HKSCS patch has been installed.

    // "old" MS950 0xfa41 -> u+e001

    // "new" MS950 0xfa41 -> u+92db

    if (strcmp(ret, "MS950") == 0) {

        TCHAR  mbChar[2] = {(char)0xfa, (char)0x41};

        WCHAR  unicodeChar;

        MultiByteToWideChar(CP_ACP, 0, mbChar, 2, &unicodeChar, 1);

        if (unicodeChar == 0x92db) {

            strcpy(ret, "MS950_HKSCS_XP");

        }

    } else {

        //SimpChinese Windows should use GB18030 as the default

        //encoding, if gb18030 patch has been installed (on windows

        //2000/XP, (1)Codepage 54936 will be available

        //(2)simsun18030.ttc will exist under system fonts dir )

        if (strcmp(ret, "GBK") == 0 && IsValidCodePage(54936)) {

            char systemPath[MAX_PATH + 1];

            char* gb18030Font = "\\FONTS\\SimSun18030.ttc";

            FILE *f = NULL;

            if (GetWindowsDirectory(systemPath, MAX_PATH + 1) != 0 &&

                strlen(systemPath) + strlen(gb18030Font) < MAX_PATH + 1) {

                strcat(systemPath, "\\FONTS\\SimSun18030.ttc");

                if ((f = fopen(systemPath, "r")) != NULL) {

                    fclose(f);

                    strcpy(ret, "GB18030");

                }

            }

        }

    }

    return ret;

}

static char* getConsoleEncoding()

{

    char* buf = malloc(16);

    int cp;

    if (buf == NULL) {

        return NULL;

    }

    cp = GetConsoleCP();

    if (cp >= 874 && cp <= 950)

        sprintf(buf, "ms%d", cp);

    else

        sprintf(buf, "cp%d", cp);

    return buf;

}

// Exported entries for AWT

DllExport const char *

getEncodingFromLangID(LANGID langID)

{

    return getEncodingInternal(MAKELCID(langID, SORT_DEFAULT));

}

// Returns BCP47 Language Tag

DllExport const char *

getJavaIDFromLangID(LANGID langID)

{

    char * elems[5]; // lang, script, ctry, variant, encoding

    char * ret;

    int index;

    ret = malloc(SNAMESIZE);

    if (ret == NULL) {

        return NULL;

    }

    for (index = 0; index < 5; index++) {

        elems[index] = NULL;

    }

    if (SetupI18nProps(MAKELCID(langID, SORT_DEFAULT),

                   &(elems[0]), &(elems[1]), &(elems[2]), &(elems[3]), &(elems[4]))) {

        // there always is the "language" tag

        strcpy(ret, elems[0]);

        // append other elements, if any

        for (index = 1; index < 4; index++) {

            if ((elems[index])[0] != '\0') {

                strcat(ret, "-");

                strcat(ret, elems[index]);

            }

        }

    } else {

        free(ret);

        ret = NULL;

    }

    for (index = 0; index < 5; index++) {

        if (elems[index] != NULL) {

            free(elems[index]);

        }

    }

    return ret;

}

/*

 * Code to figure out the user's home directory using shell32.dll

 */

WCHAR*

getHomeFromShell32()

{

    /*

     * Note that we don't free the memory allocated

     * by getHomeFromShell32.

     */

    static WCHAR *u_path = NULL;

    if (u_path == NULL) {

        HRESULT hr;

        /*

         * SHELL32 DLL is delay load DLL and we can use the trick with

         * __try/__except block.

         */

        __try {

            /*

             * For Windows Vista and later (or patched MS OS) we need to use

             * [SHGetKnownFolderPath] call to avoid MAX_PATH length limitation.

             * Shell32.dll (version 6.0.6000 or later)

             */

            hr = SHGetKnownFolderPath(&FOLDERID_Profile, KF_FLAG_DONT_VERIFY, NULL, &u_path);

        } __except(EXCEPTION_EXECUTE_HANDLER) {

            /* Exception: no [SHGetKnownFolderPath] entry */

            hr = E_FAIL;

        }

        if (FAILED(hr)) {

            WCHAR path[MAX_PATH+1];

            /* fallback solution for WinXP and Windows 2000 */

            hr = SHGetFolderPathW(NULL, CSIDL_FLAG_DONT_VERIFY | CSIDL_PROFILE, NULL, SHGFP_TYPE_CURRENT, path);

            if (FAILED(hr)) {

                /* we can't find the shell folder. */

                u_path = NULL;

            } else {

                /* Just to be sure about the path length until Windows Vista approach.

                 * [S_FALSE] could not be returned due to [CSIDL_FLAG_DONT_VERIFY] flag and UNICODE version.

                 */

                path[MAX_PATH] = 0;

                u_path = _wcsdup(path);

            }

        }

    }

    return u_path;

}

static boolean

haveMMX(void)

{

    return IsProcessorFeaturePresent(PF_MMX_INSTRUCTIONS_AVAILABLE);

}

static const char *

cpu_isalist(void)

{

    SYSTEM_INFO info;

    GetSystemInfo(&info);

    switch (info.wProcessorArchitecture) {

#ifdef PROCESSOR_ARCHITECTURE_IA64

    case PROCESSOR_ARCHITECTURE_IA64: return "ia64";

#endif

#ifdef PROCESSOR_ARCHITECTURE_AMD64

    case PROCESSOR_ARCHITECTURE_AMD64: return "amd64";

#endif

    case PROCESSOR_ARCHITECTURE_INTEL:

        switch (info.wProcessorLevel) {

        case 6: return haveMMX()

            ? "pentium_pro+mmx pentium_pro pentium+mmx pentium i486 i386 i86"

            : "pentium_pro pentium i486 i386 i86";

        case 5: return haveMMX()

            ? "pentium+mmx pentium i486 i386 i86"

            : "pentium i486 i386 i86";

        case 4: return "i486 i386 i86";

        case 3: return "i386 i86";

        }

    }

    return NULL;

}

static boolean

SetupI18nProps(LCID lcid, char** language, char** script, char** country,

               char** variant, char** encoding) {

    /* script */

    char tmp[SNAMESIZE];

    *script = malloc(PROPSIZE);

    if (*script == NULL) {

        return FALSE;

    }

    if (GetLocaleInfo(lcid,

                      LOCALE_SNAME, tmp, SNAMESIZE) == 0 ||

        sscanf(tmp, "%*[a-z\\-]%1[A-Z]%[a-z]", *script, &((*script)[1])) == 0 ||

        strlen(*script) != 4) {

        (*script)[0] = '\0';

    }

    /* country */

    *country = malloc(PROPSIZE);

    if (*country == NULL) {

        return FALSE;

    }

    if (GetLocaleInfo(lcid,

                      LOCALE_SISO3166CTRYNAME, *country, PROPSIZE) == 0 &&

        GetLocaleInfo(lcid,

                      LOCALE_SISO3166CTRYNAME2, *country, PROPSIZE) == 0) {

        (*country)[0] = '\0';

    }

    /* language */

    *language = malloc(PROPSIZE);

    if (*language == NULL) {

        return FALSE;

    }

    if (GetLocaleInfo(lcid,

                      LOCALE_SISO639LANGNAME, *language, PROPSIZE) == 0 &&

        GetLocaleInfo(lcid,

                      LOCALE_SISO639LANGNAME2, *language, PROPSIZE) == 0) {

            /* defaults to en_US */

            strcpy(*language, "en");

            strcpy(*country, "US");

        }

    /* variant */

    *variant = malloc(PROPSIZE);

    if (*variant == NULL) {

        return FALSE;

    }

    (*variant)[0] = '\0';

    /* handling for Norwegian */

    if (strcmp(*language, "nb") == 0) {

        strcpy(*language, "no");

        strcpy(*country , "NO");

    } else if (strcmp(*language, "nn") == 0) {

        strcpy(*language, "no");

        strcpy(*country , "NO");

        strcpy(*variant, "NY");

    }

    /* encoding */

    *encoding = getEncodingInternal(lcid);

    if (*encoding == NULL) {

        return FALSE;

    }

    return TRUE;

}

// GetVersionEx is deprecated; disable the warning until a replacement is found

#pragma warning(disable : 4996)

java_props_t *

GetJavaProperties(JNIEnv* env)

{

    static java_props_t sprops = {0};

    int majorVersion;

    int minorVersion;

    int buildNumber = 0;

    if (sprops.line_separator) {

        return &sprops;

    }

    /* tmp dir */

    {

        WCHAR tmpdir[MAX_PATH + 1];

        /* we might want to check that this succeed */

        GetTempPathW(MAX_PATH + 1, tmpdir);

        sprops.tmp_dir = _wcsdup(tmpdir);

    }

    /* OS properties */

    {

        char buf[100];

        boolean is_workstation;

        boolean is_64bit;

        DWORD platformId;

        {

            OSVERSIONINFOEX ver;

            ver.dwOSVersionInfoSize = sizeof(ver);

            GetVersionEx((OSVERSIONINFO *) &ver);

            majorVersion = ver.dwMajorVersion;

            minorVersion = ver.dwMinorVersion;

            /* distinguish Windows Server 2016 and 2019 by build number */

            buildNumber = ver.dwBuildNumber;

            is_workstation = (ver.wProductType == VER_NT_WORKSTATION);

            platformId = ver.dwPlatformId;

            sprops.patch_level = _strdup(ver.szCSDVersion);

        }

        {

            SYSTEM_INFO si;

            ZeroMemory(&si, sizeof(SYSTEM_INFO));

            GetNativeSystemInfo(&si);

            is_64bit = (si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64);

        }

        do {

            // Read the major and minor version number from kernel32.dll

            VS_FIXEDFILEINFO *file_info;

            WCHAR kernel32_path[MAX_PATH];

            DWORD version_size;

            LPTSTR version_info;

            UINT len, ret;

            // Get the full path to \Windows\System32\kernel32.dll and use that for

            // determining what version of Windows we're running on.

            len = MAX_PATH - (UINT)strlen("\\kernel32.dll") - 1;

            ret = GetSystemDirectoryW(kernel32_path, len);

            if (ret == 0 || ret > len) {

                break;

            }

            wcsncat(kernel32_path, L"\\kernel32.dll", MAX_PATH - ret);

            version_size = GetFileVersionInfoSizeW(kernel32_path, NULL);

            if (version_size == 0) {

                break;

            }

            version_info = (LPTSTR)malloc(version_size);

            if (version_info == NULL) {

                break;

            }

            if (!GetFileVersionInfoW(kernel32_path, 0, version_size, version_info)) {

                free(version_info);

                break;

            }

            if (!VerQueryValueW(version_info, L"\\", (LPVOID*)&file_info, &len)) {

                free(version_info);

                break;

            }

            majorVersion = HIWORD(file_info->dwProductVersionMS);

            minorVersion = LOWORD(file_info->dwProductVersionMS);

            buildNumber  = HIWORD(file_info->dwProductVersionLS);

            free(version_info);

        } while (0);

        /*

         * From msdn page on OSVERSIONINFOEX, current as of this

         * writing, decoding of dwMajorVersion and dwMinorVersion.

         *

         *  Operating system            dwMajorVersion  dwMinorVersion

         * ==================           ==============  ==============

         *

         * Windows 95                   4               0

         * Windows 98                   4               10

         * Windows ME                   4               90

         * Windows 3.51                 3               51

         * Windows NT 4.0               4               0

         * Windows 2000                 5               0

         * Windows XP 32 bit            5               1

         * Windows Server 2003 family   5               2

         * Windows XP 64 bit            5               2

         *       where ((&ver.wServicePackMinor) + 2) = 1

         *       and  si.wProcessorArchitecture = 9

         * Windows Vista family         6               0  (VER_NT_WORKSTATION)

         * Windows Server 2008          6               0  (!VER_NT_WORKSTATION)

         * Windows 7                    6               1  (VER_NT_WORKSTATION)

         * Windows Server 2008 R2       6               1  (!VER_NT_WORKSTATION)

         * Windows 8                    6               2  (VER_NT_WORKSTATION)

         * Windows Server 2012          6               2  (!VER_NT_WORKSTATION)

         * Windows Server 2012 R2       6               3  (!VER_NT_WORKSTATION)

         * Windows 10                   10              0  (VER_NT_WORKSTATION)

         * Windows Server 2016          10              0  (!VER_NT_WORKSTATION)

         * Windows Server 2019          10              0  (!VER_NT_WORKSTATION)

         *       where (buildNumber > 17762)

         *

         * This mapping will presumably be augmented as new Windows

         * versions are released.

         */

        switch (platformId) {

        case VER_PLATFORM_WIN32_WINDOWS:

           if (majorVersion == 4) {

                switch (minorVersion) {

                case  0: sprops.os_name = "Windows 95";           break;

                case 10: sprops.os_name = "Windows 98";           break;

                case 90: sprops.os_name = "Windows Me";           break;

                default: sprops.os_name = "Windows 9X (unknown)"; break;

                }

            } else {

                sprops.os_name = "Windows 9X (unknown)";

            }

            break;

        case VER_PLATFORM_WIN32_NT:

            if (majorVersion <= 4) {

                sprops.os_name = "Windows NT";

            } else if (majorVersion == 5) {

                switch (minorVersion) {

                case  0: sprops.os_name = "Windows 2000";         break;

                case  1: sprops.os_name = "Windows XP";           break;

                case  2:

                   /*