2022-10-06 20:01:42 +08:00
|
|
|
extern "C" {
|
|
|
|
|
#include "os.h"
|
2022-11-22 01:33:22 +08:00
|
|
|
#include "util/windows/unicode.h"
|
2022-10-06 20:01:42 +08:00
|
|
|
}
|
|
|
|
|
#include "util/windows/wmi.hpp"
|
|
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
static const char* getOsNameByWmi(FFstrbuf* osName)
|
|
|
|
|
{
|
|
|
|
|
FFWmiQuery query(L"SELECT Caption FROM Win32_OperatingSystem");
|
|
|
|
|
if(!query)
|
|
|
|
|
return "Query WMI service failed";
|
|
|
|
|
|
|
|
|
|
if(FFWmiRecord record = query.next())
|
|
|
|
|
{
|
|
|
|
|
record.getString(L"Caption", osName);
|
|
|
|
|
ffStrbufTrimRight(osName, ' ');
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return "No WMI result returned";
|
|
|
|
|
}
|
|
|
|
|
|
2023-06-25 14:01:50 +08:00
|
|
|
PWSTR WINAPI BrandingFormatString(PCWSTR format);
|
2022-11-21 18:51:46 +08:00
|
|
|
|
|
|
|
|
static const char* getOsNameByWinbrand(FFstrbuf* osName)
|
|
|
|
|
{
|
|
|
|
|
//https://dennisbabkin.com/blog/?t=how-to-tell-the-real-version-of-windows-your-app-is-running-on#ver_string
|
2023-06-25 14:01:50 +08:00
|
|
|
FF_LIBRARY_LOAD(winbrand, nullptr, "dlopen winbrand" FF_LIBRARY_EXTENSION " failed", "winbrand" FF_LIBRARY_EXTENSION, 1);
|
|
|
|
|
FF_LIBRARY_LOAD_SYMBOL_MESSAGE(winbrand, BrandingFormatString);
|
2022-11-21 18:51:46 +08:00
|
|
|
|
2023-06-25 14:01:50 +08:00
|
|
|
const wchar_t* rawName = ffBrandingFormatString(L"%WINDOWS_LONG%");
|
2022-11-26 12:52:04 +08:00
|
|
|
ffStrbufSetWS(osName, rawName);
|
2022-11-21 18:51:46 +08:00
|
|
|
GlobalFree((HGLOBAL)rawName);
|
|
|
|
|
return NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2022-10-06 20:01:42 +08:00
|
|
|
extern "C"
|
2023-06-24 10:59:53 +08:00
|
|
|
void ffDetectOSImpl(FFOSResult* os)
|
2022-10-06 20:01:42 +08:00
|
|
|
{
|
|
|
|
|
ffStrbufInit(&os->name);
|
|
|
|
|
ffStrbufInit(&os->prettyName);
|
|
|
|
|
ffStrbufInit(&os->id);
|
|
|
|
|
ffStrbufInit(&os->idLike);
|
|
|
|
|
ffStrbufInit(&os->variant);
|
|
|
|
|
ffStrbufInit(&os->variantID);
|
|
|
|
|
ffStrbufInit(&os->version);
|
|
|
|
|
ffStrbufInit(&os->versionID);
|
|
|
|
|
ffStrbufInit(&os->codename);
|
|
|
|
|
ffStrbufInit(&os->buildID);
|
|
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
if(getOsNameByWinbrand(&os->variant) && getOsNameByWmi(&os->variant))
|
2022-10-06 20:01:42 +08:00
|
|
|
return;
|
|
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
ffStrbufTrimRight(&os->variant, ' ');
|
|
|
|
|
|
|
|
|
|
//WMI returns the "Microsoft" prefix while BrandingFormatString doesn't. Make them consistant.
|
|
|
|
|
if(ffStrbufStartsWithS(&os->variant, "Microsoft "))
|
|
|
|
|
ffStrbufSubstrAfter(&os->variant, strlen("Microsoft ") - 1);
|
|
|
|
|
|
|
|
|
|
if(ffStrbufStartsWithS(&os->variant, "Windows "))
|
2022-10-06 20:01:42 +08:00
|
|
|
{
|
2022-11-21 18:51:46 +08:00
|
|
|
ffStrbufAppendS(&os->name, "Windows");
|
|
|
|
|
ffStrbufAppendS(&os->prettyName, "Windows");
|
2022-10-06 20:01:42 +08:00
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
ffStrbufSubstrAfter(&os->variant, strlen("Windows ") - 1);
|
2022-10-12 00:36:01 +08:00
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
if(ffStrbufStartsWithS(&os->variant, "Server "))
|
|
|
|
|
{
|
|
|
|
|
ffStrbufAppendS(&os->name, " Server");
|
|
|
|
|
ffStrbufAppendS(&os->prettyName, " Server");
|
|
|
|
|
ffStrbufSubstrAfter(&os->variant, strlen(" Server") - 1);
|
|
|
|
|
}
|
2022-10-12 00:36:01 +08:00
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
uint32_t index = ffStrbufFirstIndexC(&os->variant, ' ');
|
|
|
|
|
ffStrbufAppendNS(&os->version, index, os->variant.chars);
|
|
|
|
|
ffStrbufSubstrAfter(&os->variant, index);
|
2022-10-15 23:46:00 +08:00
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
// Windows Server 20xx Rx
|
|
|
|
|
if(ffStrbufEndsWithC(&os->prettyName, 'r'))
|
|
|
|
|
{
|
|
|
|
|
if(os->variant.chars[0] == 'R' &&
|
|
|
|
|
isdigit(os->variant.chars[1]) &&
|
|
|
|
|
(os->variant.chars[2] == '\0' || os->variant.chars[2] == ' '))
|
2022-10-15 23:46:00 +08:00
|
|
|
{
|
2022-11-21 18:51:46 +08:00
|
|
|
ffStrbufAppendF(&os->version, " R%c", os->variant.chars[1]);
|
|
|
|
|
ffStrbufSubstrAfter(&os->variant, strlen("Rx ") - 1);
|
2022-10-15 23:46:00 +08:00
|
|
|
}
|
|
|
|
|
}
|
2022-11-21 18:51:46 +08:00
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
// Unknown Windows name, please report this
|
|
|
|
|
ffStrbufAppend(&os->name, &os->variant);
|
|
|
|
|
ffStrbufClear(&os->variant);
|
|
|
|
|
}
|
2022-10-06 20:01:42 +08:00
|
|
|
|
2022-11-21 18:51:46 +08:00
|
|
|
ffStrbufAppendF(&os->id, "%*s %*s", os->prettyName.length, os->prettyName.chars, os->version.length, os->version.chars);
|
2022-10-06 20:01:42 +08:00
|
|
|
}
|