tdlight/tdutils/td/utils/port/FromApp.h
2020-12-23 18:08:34 +03:00

91 lines
3.3 KiB
C++

#pragma once
#include "td/utils/common.h"
#ifdef TD_PORT_WINDOWS
namespace td {
#if !WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP | WINAPI_PARTITION_SYSTEM)
inline HMODULE GetKernelModule() {
static const auto kernel_module = []() -> HMODULE {
MEMORY_BASIC_INFORMATION mbi;
if (VirtualQuery(VirtualQuery, &mbi, sizeof(MEMORY_BASIC_INFORMATION))) {
return reinterpret_cast<HMODULE>(mbi.AllocationBase);
}
return nullptr;
}();
return kernel_module;
}
inline HMODULE LoadLibrary(LPCTSTR lpFileName) {
using pLoadLibrary = HMODULE(WINAPI *)(_In_ LPCTSTR);
static const auto proc_load_library =
reinterpret_cast<pLoadLibrary>(GetProcAddress(GetKernelModule(), "LoadLibraryW"));
return proc_load_library(lpFileName);
}
inline HMODULE GetFromAppModule() {
static const HMODULE from_app_module = LoadLibrary(L"api-ms-win-core-file-fromapp-l1-1-0.dll");
return from_app_module;
}
#endif
template <int num, class T>
T *get_from_app_function(const char *name, T *original_func) {
#if WINAPI_FAMILY_PARTITION(WINAPI_PARTITION_DESKTOP | WINAPI_PARTITION_SYSTEM)
return original_func;
#else
static T *func = [name, original_func]() -> T * {
auto func_pointer = GetProcAddress(GetFromAppModule(), name);
if (func_pointer == nullptr) {
return original_func;
}
return reinterpret_cast<T *>(func_pointer);
}();
return func;
#endif
}
inline HANDLE CreateFile2FromAppW(_In_ LPCWSTR lpFileName, _In_ DWORD dwDesiredAccess, _In_ DWORD dwShareMode,
_In_ DWORD dwCreationDisposition,
_In_opt_ LPCREATEFILE2_EXTENDED_PARAMETERS pCreateExParams) {
auto func = get_from_app_function<0>("CreateFile2FromAppW", &CreateFile2);
return func(lpFileName, dwDesiredAccess, dwShareMode, dwCreationDisposition, pCreateExParams);
}
inline BOOL CreateDirectoryFromAppW(_In_ LPCWSTR lpPathName, _In_opt_ LPSECURITY_ATTRIBUTES lpSecurityAttributes) {
auto func = get_from_app_function<1>("CreateDirectoryFromAppW", &CreateDirectory);
return func(lpPathName, lpSecurityAttributes);
}
inline BOOL RemoveDirectoryFromAppW(_In_ LPCWSTR lpPathName) {
auto func = get_from_app_function<2>("RemoveDirectoryFromAppW", &RemoveDirectory);
return func(lpPathName);
}
inline BOOL DeleteFileFromAppW(_In_ LPCWSTR lpFileName) {
auto func = get_from_app_function<3>("DeleteFileFromAppW", &DeleteFile);
return func(lpFileName);
}
inline BOOL MoveFileExFromAppW(_In_ LPCWSTR lpExistingFileName, _In_ LPCWSTR lpNewFileName, _In_ DWORD dwFlags) {
auto func = get_from_app_function<4>("MoveFileFromAppW", &MoveFileEx);
if (func != &MoveFileEx && (dwFlags & MOVEFILE_REPLACE_EXISTING) != 0) {
td::DeleteFileFromAppW(lpNewFileName);
}
return func(lpExistingFileName, lpNewFileName, dwFlags);
}
inline HANDLE FindFirstFileExFromAppW(_In_ LPCWSTR lpFileName, _In_ FINDEX_INFO_LEVELS fInfoLevelId,
_Out_writes_bytes_(sizeof(WIN32_FIND_DATAW)) LPVOID lpFindFileData,
_In_ FINDEX_SEARCH_OPS fSearchOp, _Reserved_ LPVOID lpSearchFilter,
_In_ DWORD dwAdditionalFlags) {
auto func = get_from_app_function<5>("FindFirstFileExFromAppW", &FindFirstFileEx);
return func(lpFileName, fInfoLevelId, lpFindFileData, fSearchOp, lpSearchFilter, dwAdditionalFlags);
}
} // namespace td
#endif