mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-11-02 05:01:58 +01:00
李通洲
4fa3ca98a2
We can't use RegGetValueA because RegGetValueA converts wide string to non UTF-8 string, which results in crashing when printing
83 lines
2.6 KiB
C
83 lines
2.6 KiB
C
#include "register.h"
|
|
#include "unicode.h"
|
|
#include "util/mallocHelper.h"
|
|
|
|
static const char* hKey2Str(HKEY hKey)
|
|
{
|
|
#define HKEY_CASE(compareKey) if(hKey == compareKey) return #compareKey;
|
|
HKEY_CASE(HKEY_CLASSES_ROOT)
|
|
HKEY_CASE(HKEY_CURRENT_USER)
|
|
HKEY_CASE(HKEY_LOCAL_MACHINE)
|
|
HKEY_CASE(HKEY_USERS)
|
|
HKEY_CASE(HKEY_PERFORMANCE_DATA)
|
|
HKEY_CASE(HKEY_PERFORMANCE_TEXT)
|
|
HKEY_CASE(HKEY_PERFORMANCE_NLSTEXT)
|
|
HKEY_CASE(HKEY_CURRENT_CONFIG)
|
|
HKEY_CASE(HKEY_DYN_DATA)
|
|
HKEY_CASE(HKEY_CURRENT_USER_LOCAL_SETTINGS)
|
|
#undef HKEY_CASE
|
|
|
|
return "UNKNOWN";
|
|
}
|
|
|
|
bool ffRegOpenKeyForRead(HKEY hKey, const wchar_t* subKeyW, HKEY* result, FFstrbuf* error)
|
|
{
|
|
if(RegOpenKeyExW(hKey, subKeyW, 0, KEY_READ, result) != ERROR_SUCCESS)
|
|
{
|
|
if(error)
|
|
{
|
|
FF_STRBUF_AUTO_DESTROY subKeyA = ffStrbufFromWchar(subKeyW);
|
|
ffStrbufAppendF(error, "RegOpenKeyExW(%s\\%s) failed", hKey2Str(hKey), subKeyA.chars);
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool ffRegReadStrbuf(HKEY hKey, const wchar_t* valueNameW, FFstrbuf* result, FFstrbuf* error)
|
|
{
|
|
DWORD bufSize; //with tailing '\0'
|
|
if(RegGetValueW(hKey, NULL, valueNameW, RRF_RT_REG_SZ, NULL, NULL, &bufSize) != ERROR_SUCCESS)
|
|
{
|
|
if(error)
|
|
{
|
|
if(!valueNameW)
|
|
valueNameW = L"(default)";
|
|
FF_STRBUF_AUTO_DESTROY valueNameA = ffStrbufFromWchar(valueNameW);
|
|
ffStrbufAppendF(error, "RegGetValueA(%s, NULL, RRF_RT_REG_SZ) failed", valueNameA.chars);
|
|
}
|
|
return false;
|
|
}
|
|
wchar_t* FF_AUTO_FREE resultW = (wchar_t*)malloc(bufSize);
|
|
if(RegGetValueW(hKey, NULL, valueNameW, RRF_RT_REG_SZ, NULL, resultW, &bufSize) != ERROR_SUCCESS)
|
|
{
|
|
if(error)
|
|
{
|
|
if(!valueNameW)
|
|
valueNameW = L"(default)";
|
|
FF_STRBUF_AUTO_DESTROY valueNameA = ffStrbufFromWchar(valueNameW);
|
|
ffStrbufAppendF(error, "RegGetValueA(%s, result, RRF_RT_REG_SZ) failed", valueNameA.chars);
|
|
}
|
|
return false;
|
|
}
|
|
ffWcharToUtf8(resultW, result);
|
|
return true;
|
|
}
|
|
|
|
bool ffRegReadUint(HKEY hKey, const wchar_t* valueNameW, uint32_t* result, FFstrbuf* error)
|
|
{
|
|
DWORD bufSize = sizeof(*result);
|
|
if(RegGetValueW(hKey, NULL, valueNameW, RRF_RT_DWORD, NULL, result, &bufSize) != ERROR_SUCCESS)
|
|
{
|
|
if(error)
|
|
{
|
|
if(!valueNameW)
|
|
valueNameW = L"(default)";
|
|
FF_STRBUF_AUTO_DESTROY valueNameA = ffStrbufFromWchar(valueNameW);
|
|
ffStrbufAppendF(error, "RegGetValueA(%s, result, RRF_RT_DWORD) failed", valueNameA.chars);
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|