HMCLauncher: try bundled JRE

2019-02-08 22:05:16 +08:00
parent f0c7f54cc6
commit 426ea607e8
12 changed files with 375 additions and 5 deletions
--- a/HMCLauncher/HMCL/java.cpp
+++ b/HMCLauncher/HMCL/java.cpp
@@ -0,0 +1,88 @@
+#include "stdafx.h"
+#include "java.h"
+#include "os.h"
+
+const Version JAVA_8(L"1.8"), JAVA_11(L"11");
+
+const LPCWSTR JDK_OLD = L"SOFTWARE\\JavaSoft\\Java Development Kit";
+const LPCWSTR JRE_OLD = L"SOFTWARE\\JavaSoft\\Java Runtime Environment";
+const LPCWSTR JDK_NEW = L"SOFTWARE\\JavaSoft\\JDK";
+const LPCWSTR JRE_NEW = L"SOFTWARE\\JavaSoft\\JRE";
+
+bool oldJavaFound = false, newJavaFound = false;
+
+bool FindJavaByRegistryKey(HKEY rootKey, LPCWSTR subKey, std::wstring & path)
+{
+	WCHAR javaVer[MAX_KEY_LENGTH];  // buffer for subkey name, special for JavaVersion
+	DWORD cbName;                   // size of name string
+	DWORD cSubKeys = 0;             // number of subkeys
+	DWORD cbMaxSubKey;              // longest subkey size
+	DWORD cValues;                  // number of values for key
+	DWORD cchMaxValue;              // longest value name
+	DWORD cbMaxValueData;           // longest value data
+	LSTATUS result;
+
+	HKEY hKey;
+	if (ERROR_SUCCESS != (result = RegOpenKeyEx(rootKey, subKey, 0, KEY_WOW64_64KEY | KEY_READ, &hKey)))
+		return false;
+
+	RegQueryInfoKey(
+		hKey,                    // key handle
+		NULL,                    // buffer for class name
+		NULL,                    // size of class string
+		NULL,                    // reserved
+		&cSubKeys,               // number of subkeys
+		&cbMaxSubKey,            // longest subkey size
+		NULL,                    // longest class string
+		&cValues,                // number of values for this key
+		&cchMaxValue,            // longest value name
+		&cbMaxValueData,         // longest value data
+		NULL,                    // security descriptor
+		NULL);                   // last write time
+
+	if (!cSubKeys)
+		return false;
+
+	bool flag = false;
+	for (DWORD i = 0; i < cSubKeys; ++i)
+	{
+		cbName = MAX_KEY_LENGTH;
+		if (ERROR_SUCCESS != (result = RegEnumKeyEx(hKey, i, javaVer, &cbName, NULL, NULL, NULL, NULL)))
+			continue;
+
+		HKEY javaKey;
+		if (ERROR_SUCCESS != RegOpenKeyEx(hKey, javaVer, 0, KEY_READ, &javaKey))
+			continue;
+
+		if (ERROR_SUCCESS == MyRegQueryValue(javaKey, L"JavaHome", REG_SZ, path))
+		{
+			if (Version(javaVer) < JAVA_8)
+				oldJavaFound = true;
+			else if (!(Version(javaVer) < JAVA_11))
+				newJavaFound = true;
+			else
+				flag = true;
+		}
+
+		if (flag)
+			break;
+	}
+
+	RegCloseKey(hKey);
+
+	return flag;
+}
+
+bool FindJavaInRegistry(std::wstring & path)
+{
+	return FindJavaByRegistryKey(HKEY_LOCAL_MACHINE, JDK_OLD, path) ||
+		FindJavaByRegistryKey(HKEY_LOCAL_MACHINE, JRE_OLD, path) ||
+		FindJavaByRegistryKey(HKEY_LOCAL_MACHINE, JDK_NEW, path) ||
+		FindJavaByRegistryKey(HKEY_LOCAL_MACHINE, JRE_NEW, path);
+}
+
+bool FindJava(std::wstring & path)
+{
+	return FindJavaInRegistry(path) ||
+		ERROR_SUCCESS == MyGetEnvironmentVariable(L"JAVA_HOME", path);
+}