mirror of
https://github.com/fastfetch-cli/fastfetch.git
synced 2026-11-07 21:50:08 +01:00
李通洲
64 lines
2.4 KiB
C
64 lines
2.4 KiB
C
#include "bootmgr.h"
|
|
#include "efi_helper.h"
|
|
#include "common/io.h"
|
|
#include "common/windows/nt.h"
|
|
|
|
#include <windows.h>
|
|
|
|
const char* enablePrivilege(const wchar_t* privilege)
|
|
{
|
|
FF_AUTO_CLOSE_FD HANDLE token = NULL;
|
|
if (!OpenProcessToken(NtCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &token))
|
|
return "OpenProcessToken() failed";
|
|
|
|
TOKEN_PRIVILEGES tp = {
|
|
.PrivilegeCount = 1,
|
|
.Privileges = {
|
|
(LUID_AND_ATTRIBUTES) { .Attributes = SE_PRIVILEGE_ENABLED }
|
|
},
|
|
};
|
|
if (!LookupPrivilegeValueW(NULL, privilege, &tp.Privileges[0].Luid))
|
|
return "LookupPrivilegeValue() failed";
|
|
|
|
if (!AdjustTokenPrivileges(token, false, &tp, sizeof(tp), NULL, NULL))
|
|
return "AdjustTokenPrivileges() failed";
|
|
|
|
if (GetLastError() == ERROR_NOT_ALL_ASSIGNED)
|
|
return "The token does not have the specified privilege";
|
|
|
|
return NULL;
|
|
}
|
|
|
|
const char* ffDetectBootmgr(FFBootmgrResult* result)
|
|
{
|
|
if (enablePrivilege(L"SeSystemEnvironmentPrivilege") != NULL)
|
|
return "Failed to enable SeSystemEnvironmentPrivilege";
|
|
|
|
GUID efiGlobalGuid;
|
|
if (!NT_SUCCESS(RtlGUIDFromString(&(UNICODE_STRING) RTL_CONSTANT_STRING(L"{" FF_EFI_GLOBAL_GUID L"}"), &efiGlobalGuid)))
|
|
return "RtlGUIDFromString() failed";
|
|
|
|
ULONG size = sizeof(result->order);
|
|
if (!NT_SUCCESS(NtQuerySystemEnvironmentValueEx(&(UNICODE_STRING) RTL_CONSTANT_STRING(L"BootCurrent"), &efiGlobalGuid, &result->order, &size, NULL)))
|
|
return "NtQuerySystemEnvironmentValueEx(BootCurrent) failed";
|
|
if (size != sizeof(result->order))
|
|
return "NtQuerySystemEnvironmentValueEx(BootCurrent) returned unexpected size";
|
|
|
|
uint8_t buffer[2048];
|
|
wchar_t key[9];
|
|
swprintf(key, ARRAY_SIZE(key), L"Boot%04X", result->order);
|
|
size = sizeof(buffer);
|
|
if (!NT_SUCCESS(NtQuerySystemEnvironmentValueEx(&(UNICODE_STRING) RTL_CONSTANT_STRING(key), &efiGlobalGuid, buffer, &size, NULL)))
|
|
return "NtQuerySystemEnvironmentValueEx(Boot####) failed";
|
|
if (size < sizeof(FFEfiLoadOption) || size == ARRAY_SIZE(buffer))
|
|
return "NtQuerySystemEnvironmentValueEx(Boot####) returned unexpected size";
|
|
|
|
ffEfiFillLoadOption((FFEfiLoadOption *)buffer, result);
|
|
|
|
SYSTEM_SECUREBOOT_INFORMATION ssi;
|
|
if (NT_SUCCESS(NtQuerySystemInformation(SystemSecureBootInformation, &ssi, sizeof(ssi), NULL)))
|
|
result->secureBoot = ssi.SecureBootEnabled;
|
|
|
|
return NULL;
|
|
}
|