thirdpartymirrors/UnleashedRecomp
1
0
Fork 0
mirror of https://github.com/hedge-dev/UnleashedRecomp.git synced 2026-04-27 21:01:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement reading path and unicode string from windows registry

Browse source
This commit is contained in:
Sajid 2025-01-28 15:37:38 +06:00
parent 2b6013ad26
commit 4094690282
3 changed files with 52 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
UnleashedRecomp/main.cpp
View file

@ -176,7 +176,7 @@ int main(int argc, char *argv[])
if (!Registry::RootDirectoryPath.empty())
{
if (!os::process::SetWorkingDirectory(std::filesystem::path(Registry::RootDirectoryPath)))
LOGFN_ERROR("Failed to set working directory: \"{}\"", Registry::RootDirectoryPath);
LOGFN_ERROR("Failed to set working directory: \"{}\"", Registry::RootDirectoryPath.string());
}
HostStartup();

67
UnleashedRecomp/os/win32/registry_win32.inl
View file

@ -16,44 +16,77 @@ bool os::registry::ReadValue(const std::string& name, T& data)
if (RegOpenKeyExW(HKEY_CURRENT_USER, g_registryRoot, 0, KEY_READ, &hKey) != ERROR_SUCCESS)
return false;
BYTE buffer[512];
DWORD bufferSize = sizeof(buffer);
wchar_t wideName[128];
int wideNameSize = MultiByteToWideChar(CP_UTF8, 0, name.c_str(), name.size(), wideName, sizeof(wideName));
if (wideNameSize == 0)
{
return false;
}
wideName[wideNameSize] = 0;
DWORD bufferSize = 0;
DWORD dataType = 0;
auto result = RegQueryValueExA(hKey, name.c_str(), NULL, &dataType, buffer, &bufferSize);
RegCloseKey(hKey);
auto result = RegQueryValueExW(hKey, wideName, NULL, &dataType, nullptr, &bufferSize);
if (result != ERROR_SUCCESS)
{
RegCloseKey(hKey);
return false;
}
result = ERROR_INVALID_FUNCTION;
if constexpr (std::is_same_v<T, std::string>)
{
if (dataType != REG_SZ)
return false;
if (dataType == REG_SZ)
{
std::vector<uint8_t> buffer{};
buffer.reserve(bufferSize);
result = RegQueryValueExW(hKey, wideName, nullptr, nullptr, buffer.data(), &bufferSize);
data = std::string((char*)buffer, bufferSize - 1);
if (result == ERROR_SUCCESS)
{
int valueSize = WideCharToMultiByte(CP_UTF8, 0, (wchar_t*)buffer.data(), (bufferSize / sizeof(wchar_t)) - 1, nullptr, 0, nullptr, nullptr);
data.resize(valueSize);
WideCharToMultiByte(CP_UTF8, 0, (wchar_t*)buffer.data(), (bufferSize / sizeof(wchar_t)) - 1, data.data(), valueSize, nullptr, nullptr);
}
}
}
else if constexpr (std::is_same_v<T, std::filesystem::path>)
{
if (dataType == REG_SZ)
{
std::vector<uint8_t> buffer{};
buffer.reserve(bufferSize);
result = RegQueryValueExW(hKey, wideName, nullptr, nullptr, buffer.data(), &bufferSize);
if (result == ERROR_SUCCESS)
{
data = reinterpret_cast<wchar_t*>(buffer.data());
}
}
}
else if constexpr (std::is_same_v<T, uint32_t>)
{
if (dataType != REG_DWORD)
return false;
data = *(uint32_t*)buffer;
if (dataType == REG_DWORD)
{
result = RegQueryValueExW(hKey, wideName, nullptr, nullptr, (BYTE*)&data, &bufferSize);
}
}
else if constexpr (std::is_same_v<T, uint64_t>)
{
if (dataType != REG_QWORD)
return false;
data = *(uint32_t*)buffer;
if (dataType == REG_QWORD)
{
result = RegQueryValueExW(hKey, wideName, nullptr, nullptr, (BYTE*)&data, &bufferSize);
}
}
else
{
static_assert(false, "Unsupported data type.");
}
return true;
RegCloseKey(hKey);
return result == ERROR_SUCCESS;
}
template<typename T>

2
UnleashedRecomp/user/registry.h
View file

@ -3,7 +3,7 @@
class Registry
{
public:
inline static std::string RootDirectoryPath;
inline static std::filesystem::path RootDirectoryPath;
static void Load();
static void Save();