231 lines
6.9 KiB
C++
231 lines
6.9 KiB
C++
/*************************************************************************/
|
|
/* mono_reg_utils.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2021 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2021 Godot Engine contributors (cf. AUTHORS.md). */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "mono_reg_utils.h"
|
|
#include "core/io/dir_access.h"
|
|
|
|
#ifdef WINDOWS_ENABLED
|
|
|
|
#include "core/os/os.h"
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
|
|
namespace MonoRegUtils {
|
|
|
|
template <int>
|
|
REGSAM bitness_sam_impl();
|
|
|
|
template <>
|
|
REGSAM bitness_sam_impl<4>() {
|
|
return KEY_WOW64_64KEY;
|
|
}
|
|
|
|
template <>
|
|
REGSAM bitness_sam_impl<8>() {
|
|
return KEY_WOW64_32KEY;
|
|
}
|
|
|
|
REGSAM _get_bitness_sam() {
|
|
return bitness_sam_impl<sizeof(size_t)>();
|
|
}
|
|
|
|
LONG _RegOpenKey(HKEY hKey, LPCWSTR lpSubKey, PHKEY phkResult) {
|
|
LONG res = RegOpenKeyExW(hKey, lpSubKey, 0, KEY_READ, phkResult);
|
|
|
|
if (res != ERROR_SUCCESS)
|
|
res = RegOpenKeyExW(hKey, lpSubKey, 0, KEY_READ | _get_bitness_sam(), phkResult);
|
|
|
|
return res;
|
|
}
|
|
|
|
LONG _RegKeyQueryString(HKEY hKey, const String &p_value_name, String &r_value) {
|
|
Vector<WCHAR> buffer;
|
|
buffer.resize(512);
|
|
DWORD dwBufferSize = buffer.size();
|
|
|
|
LONG res = RegQueryValueExW(hKey, (LPCWSTR)(p_value_name.utf16().get_data()), 0, nullptr, (LPBYTE)buffer.ptr(), &dwBufferSize);
|
|
|
|
if (res == ERROR_MORE_DATA) {
|
|
// dwBufferSize now contains the actual size
|
|
buffer.resize(dwBufferSize);
|
|
res = RegQueryValueExW(hKey, (LPCWSTR)(p_value_name.utf16().get_data()), 0, nullptr, (LPBYTE)buffer.ptr(), &dwBufferSize);
|
|
}
|
|
|
|
if (res == ERROR_SUCCESS) {
|
|
r_value = String(buffer.ptr(), buffer.size());
|
|
} else {
|
|
r_value = String();
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
LONG _find_mono_in_reg(const String &p_subkey, MonoRegInfo &r_info, bool p_old_reg = false) {
|
|
HKEY hKey;
|
|
LONG res = _RegOpenKey(HKEY_LOCAL_MACHINE, (LPCWSTR)(p_subkey.utf16().get_data()), &hKey);
|
|
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
if (!p_old_reg) {
|
|
res = _RegKeyQueryString(hKey, "Version", r_info.version);
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
}
|
|
|
|
res = _RegKeyQueryString(hKey, "SdkInstallRoot", r_info.install_root_dir);
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
res = _RegKeyQueryString(hKey, "FrameworkAssemblyDirectory", r_info.assembly_dir);
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
res = _RegKeyQueryString(hKey, "MonoConfigDir", r_info.config_dir);
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
if (r_info.install_root_dir.ends_with("\\"))
|
|
r_info.bin_dir = r_info.install_root_dir + "bin";
|
|
else
|
|
r_info.bin_dir = r_info.install_root_dir + "\\bin";
|
|
|
|
cleanup:
|
|
RegCloseKey(hKey);
|
|
return res;
|
|
}
|
|
|
|
LONG _find_mono_in_reg_old(const String &p_subkey, MonoRegInfo &r_info) {
|
|
String default_clr;
|
|
|
|
HKEY hKey;
|
|
LONG res = _RegOpenKey(HKEY_LOCAL_MACHINE, (LPCWSTR)(p_subkey.utf16().get_data()), &hKey);
|
|
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
res = _RegKeyQueryString(hKey, "DefaultCLR", default_clr);
|
|
|
|
if (res == ERROR_SUCCESS && default_clr.length()) {
|
|
r_info.version = default_clr;
|
|
res = _find_mono_in_reg(p_subkey + "\\" + default_clr, r_info, true);
|
|
}
|
|
|
|
cleanup:
|
|
RegCloseKey(hKey);
|
|
return res;
|
|
}
|
|
|
|
MonoRegInfo find_mono() {
|
|
MonoRegInfo info;
|
|
|
|
if (_find_mono_in_reg("Software\\Mono", info) == ERROR_SUCCESS)
|
|
return info;
|
|
|
|
if (_find_mono_in_reg_old("Software\\Novell\\Mono", info) == ERROR_SUCCESS)
|
|
return info;
|
|
|
|
return MonoRegInfo();
|
|
}
|
|
|
|
String find_msbuild_tools_path() {
|
|
String msbuild_tools_path;
|
|
|
|
// Try to find 15.0 with vswhere
|
|
|
|
String vswhere_path = OS::get_singleton()->get_environment(sizeof(size_t) == 8 ? "ProgramFiles(x86)" : "ProgramFiles");
|
|
vswhere_path += "\\Microsoft Visual Studio\\Installer\\vswhere.exe";
|
|
|
|
List<String> vswhere_args;
|
|
vswhere_args.push_back("-latest");
|
|
vswhere_args.push_back("-products");
|
|
vswhere_args.push_back("*");
|
|
vswhere_args.push_back("-requires");
|
|
vswhere_args.push_back("Microsoft.Component.MSBuild");
|
|
|
|
String output;
|
|
int exit_code;
|
|
OS::get_singleton()->execute(vswhere_path, vswhere_args, &output, &exit_code);
|
|
|
|
if (exit_code == 0) {
|
|
Vector<String> lines = output.split("\n");
|
|
|
|
for (int i = 0; i < lines.size(); i++) {
|
|
const String &line = lines[i];
|
|
int sep_idx = line.find(":");
|
|
|
|
if (sep_idx > 0) {
|
|
String key = line.substr(0, sep_idx); // No need to trim
|
|
|
|
if (key == "installationPath") {
|
|
String val = line.substr(sep_idx + 1, line.length()).strip_edges();
|
|
|
|
ERR_BREAK(val.is_empty());
|
|
|
|
if (!val.ends_with("\\")) {
|
|
val += "\\";
|
|
}
|
|
|
|
// Since VS2019, the directory is simply named "Current"
|
|
String msbuild_dir = val + "MSBuild\\Current\\Bin";
|
|
if (DirAccess::exists(msbuild_dir)) {
|
|
return msbuild_dir;
|
|
}
|
|
|
|
// Directory name "15.0" is used in VS 2017
|
|
return val + "MSBuild\\15.0\\Bin";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Try to find 14.0 in the Registry
|
|
|
|
HKEY hKey;
|
|
LONG res = _RegOpenKey(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\MSBuild\\ToolsVersions\\14.0", &hKey);
|
|
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
res = _RegKeyQueryString(hKey, "MSBuildToolsPath", msbuild_tools_path);
|
|
|
|
if (res != ERROR_SUCCESS)
|
|
goto cleanup;
|
|
|
|
cleanup:
|
|
RegCloseKey(hKey);
|
|
|
|
return msbuild_tools_path;
|
|
}
|
|
} // namespace MonoRegUtils
|
|
|
|
#endif // WINDOWS_ENABLED
|