reactos/base/applications/rapps/misc.cpp
Whindmar Saksit 57b775ef6e
[RAPPS] Automatically generate installer/uninstaller for downloaded zip/cab files (#6652)
With a single database line added to applications distributed as zip/cab allows rapps.exe to act as an installer that automatically extracts the files and creates a startmenu shortcut. It can also uninstall the extracted files (and optionally other files and registry entries created by the application).
2024-05-08 23:58:54 +02:00

491 lines
12 KiB
C++

/*
* PROJECT: ReactOS Applications Manager
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
* PURPOSE: Misc functions
* COPYRIGHT: Copyright 2009 Dmitry Chapyshev (dmitry@reactos.org)
* Copyright 2015 Ismael Ferreras Morezuelas (swyterzone+ros@gmail.com)
* Copyright 2017 Alexander Shaposhnikov (sanchaez@reactos.org)
*/
#include "rapps.h"
#include "misc.h"
static HANDLE hLog = NULL;
VOID
CopyTextToClipboard(LPCWSTR lpszText)
{
if (!OpenClipboard(NULL))
{
return;
}
HRESULT hr;
HGLOBAL ClipBuffer;
LPWSTR Buffer;
DWORD cchBuffer;
EmptyClipboard();
cchBuffer = wcslen(lpszText) + 1;
ClipBuffer = GlobalAlloc(GMEM_DDESHARE, cchBuffer * sizeof(WCHAR));
Buffer = (PWCHAR)GlobalLock(ClipBuffer);
hr = StringCchCopyW(Buffer, cchBuffer, lpszText);
GlobalUnlock(ClipBuffer);
if (SUCCEEDED(hr))
SetClipboardData(CF_UNICODETEXT, ClipBuffer);
CloseClipboard();
}
VOID
ShowPopupMenuEx(HWND hwnd, HWND hwndOwner, UINT MenuID, UINT DefaultItem)
{
HMENU hMenu = NULL;
HMENU hPopupMenu;
MENUITEMINFO ItemInfo;
POINT pt;
if (MenuID)
{
hMenu = LoadMenuW(hInst, MAKEINTRESOURCEW(MenuID));
hPopupMenu = GetSubMenu(hMenu, 0);
}
else
{
hPopupMenu = GetMenu(hwnd);
}
ZeroMemory(&ItemInfo, sizeof(ItemInfo));
ItemInfo.cbSize = sizeof(ItemInfo);
ItemInfo.fMask = MIIM_STATE;
GetMenuItemInfoW(hPopupMenu, DefaultItem, FALSE, &ItemInfo);
if (!(ItemInfo.fState & MFS_GRAYED))
{
SetMenuDefaultItem(hPopupMenu, DefaultItem, FALSE);
}
GetCursorPos(&pt);
SetForegroundWindow(hwnd);
TrackPopupMenu(hPopupMenu, 0, pt.x, pt.y, 0, hwndOwner, NULL);
if (hMenu)
{
DestroyMenu(hMenu);
}
}
BOOL
StartProcess(const CStringW &Path, BOOL Wait)
{
PROCESS_INFORMATION pi;
STARTUPINFOW si;
DWORD dwRet;
MSG msg;
ZeroMemory(&si, sizeof(si));
si.cb = sizeof(si);
si.dwFlags = STARTF_USESHOWWINDOW;
si.wShowWindow = SW_SHOW;
// The Unicode version of CreateProcess can modify the contents of this string.
CStringW Tmp = Path;
BOOL fSuccess = CreateProcessW(NULL, Tmp.GetBuffer(), NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
Tmp.ReleaseBuffer();
if (!fSuccess)
{
return FALSE;
}
CloseHandle(pi.hThread);
if (Wait)
{
EnableWindow(hMainWnd, FALSE);
}
while (Wait)
{
dwRet = MsgWaitForMultipleObjects(1, &pi.hProcess, FALSE, INFINITE, QS_ALLEVENTS);
if (dwRet == WAIT_OBJECT_0 + 1)
{
while (PeekMessageW(&msg, NULL, 0, 0, PM_REMOVE))
{
TranslateMessage(&msg);
DispatchMessageW(&msg);
}
}
else
{
if (dwRet == WAIT_OBJECT_0 || dwRet == WAIT_FAILED)
break;
}
}
CloseHandle(pi.hProcess);
if (Wait)
{
EnableWindow(hMainWnd, TRUE);
SetForegroundWindow(hMainWnd);
SetFocus(hMainWnd);
}
return TRUE;
}
BOOL
GetStorageDirectory(CStringW &Directory)
{
static CStringW CachedDirectory;
static BOOL CachedDirectoryInitialized = FALSE;
if (!CachedDirectoryInitialized)
{
LPWSTR DirectoryStr = CachedDirectory.GetBuffer(MAX_PATH);
BOOL bHasPath = SHGetSpecialFolderPathW(NULL, DirectoryStr, CSIDL_LOCAL_APPDATA, TRUE);
if (bHasPath)
{
PathAppendW(DirectoryStr, RAPPS_NAME);
}
CachedDirectory.ReleaseBuffer();
if (bHasPath)
{
if (!CreateDirectoryW(CachedDirectory, NULL) && GetLastError() != ERROR_ALREADY_EXISTS)
{
CachedDirectory.Empty();
}
}
else
{
CachedDirectory.Empty();
}
CachedDirectoryInitialized = TRUE;
}
Directory = CachedDirectory;
return !Directory.IsEmpty();
}
VOID
InitLogs()
{
if (!SettingsInfo.bLogEnabled)
{
return;
}
WCHAR szPath[MAX_PATH];
DWORD dwCategoryNum = 1;
DWORD dwDisp, dwData;
ATL::CRegKey key;
if (key.Create(
HKEY_LOCAL_MACHINE,
L"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\ReactOS Application Manager", REG_NONE,
REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &dwDisp) != ERROR_SUCCESS)
{
return;
}
if (!GetModuleFileNameW(NULL, szPath, _countof(szPath)))
{
return;
}
dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE | EVENTLOG_INFORMATION_TYPE;
if ((key.SetStringValue(L"EventMessageFile", szPath, REG_EXPAND_SZ) == ERROR_SUCCESS) &&
(key.SetStringValue(L"CategoryMessageFile", szPath, REG_EXPAND_SZ) == ERROR_SUCCESS) &&
(key.SetDWORDValue(L"TypesSupported", dwData) == ERROR_SUCCESS) &&
(key.SetDWORDValue(L"CategoryCount", dwCategoryNum) == ERROR_SUCCESS))
{
hLog = RegisterEventSourceW(NULL, L"ReactOS Application Manager");
}
}
VOID
FreeLogs()
{
if (hLog)
{
DeregisterEventSource(hLog);
}
}
BOOL
WriteLogMessage(WORD wType, DWORD dwEventID, LPCWSTR lpMsg)
{
if (!SettingsInfo.bLogEnabled)
{
return TRUE;
}
if (!ReportEventW(hLog, wType, 0, dwEventID, NULL, 1, 0, &lpMsg, NULL))
{
return FALSE;
}
return TRUE;
}
BOOL
GetInstalledVersion_WowUser(CStringW *szVersionResult, const CStringW &szRegName, BOOL IsUserKey, REGSAM keyWow)
{
BOOL bHasSucceded = FALSE;
ATL::CRegKey key;
CStringW szVersion;
CStringW szPath = CStringW(L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\") + szRegName;
if (key.Open(IsUserKey ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE, szPath.GetString(), keyWow | KEY_READ) !=
ERROR_SUCCESS)
{
return FALSE;
}
if (szVersionResult != NULL)
{
ULONG dwSize = MAX_PATH * sizeof(WCHAR);
if (key.QueryStringValue(L"DisplayVersion", szVersion.GetBuffer(MAX_PATH), &dwSize) == ERROR_SUCCESS)
{
szVersion.ReleaseBuffer();
*szVersionResult = szVersion;
bHasSucceded = TRUE;
}
else
{
szVersion.ReleaseBuffer();
}
}
else
{
bHasSucceded = TRUE;
}
return bHasSucceded;
}
BOOL
GetInstalledVersion(CStringW *pszVersion, const CStringW &szRegName)
{
return (
!szRegName.IsEmpty() && (GetInstalledVersion_WowUser(pszVersion, szRegName, TRUE, KEY_WOW64_32KEY) ||
GetInstalledVersion_WowUser(pszVersion, szRegName, FALSE, KEY_WOW64_32KEY) ||
GetInstalledVersion_WowUser(pszVersion, szRegName, TRUE, KEY_WOW64_64KEY) ||
GetInstalledVersion_WowUser(pszVersion, szRegName, FALSE, KEY_WOW64_64KEY)));
}
BOOL
IsSystem64Bit()
{
#ifdef _WIN64
return TRUE;
#else
static UINT cache = 0;
if (!cache)
cache = 1 + (IsOS(OS_WOW6432) != FALSE);
return cache - 1;
#endif
}
INT
GetSystemColorDepth()
{
DEVMODEW pDevMode;
INT ColorDepth;
pDevMode.dmSize = sizeof(pDevMode);
pDevMode.dmDriverExtra = 0;
if (!EnumDisplaySettingsW(NULL, ENUM_CURRENT_SETTINGS, &pDevMode))
{
/* TODO: Error message */
return ILC_COLOR;
}
switch (pDevMode.dmBitsPerPel)
{
case 32:
ColorDepth = ILC_COLOR32;
break;
case 24:
ColorDepth = ILC_COLOR24;
break;
case 16:
ColorDepth = ILC_COLOR16;
break;
case 8:
ColorDepth = ILC_COLOR8;
break;
case 4:
ColorDepth = ILC_COLOR4;
break;
default:
ColorDepth = ILC_COLOR;
break;
}
return ColorDepth;
}
void
UnixTimeToFileTime(DWORD dwUnixTime, LPFILETIME pFileTime)
{
// Note that LONGLONG is a 64-bit value
LONGLONG ll;
ll = Int32x32To64(dwUnixTime, 10000000) + 116444736000000000;
pFileTime->dwLowDateTime = (DWORD)ll;
pFileTime->dwHighDateTime = ll >> 32;
}
HRESULT
RegKeyHasValues(HKEY hKey, LPCWSTR Path, REGSAM wowsam)
{
CRegKey key;
LONG err = key.Open(hKey, Path, KEY_QUERY_VALUE | wowsam);
if (err == ERROR_SUCCESS)
{
WCHAR name[1];
DWORD cchname = _countof(name), cbsize = 0;
err = RegEnumValueW(key, 0, name, &cchname, NULL, NULL, NULL, &cbsize);
if (err == ERROR_NO_MORE_ITEMS)
return S_FALSE;
if (err == ERROR_MORE_DATA)
err = ERROR_SUCCESS;
}
return HRESULT_FROM_WIN32(err);
}
LPCWSTR
GetRegString(CRegKey &Key, LPCWSTR Name, CStringW &Value)
{
for (;;)
{
ULONG cb = 0, cch;
ULONG err = Key.QueryValue(Name, NULL, NULL, &cb);
if (err)
break;
cch = cb / sizeof(WCHAR);
LPWSTR p = Value.GetBuffer(cch + 1);
p[cch] = UNICODE_NULL;
err = Key.QueryValue(Name, NULL, (BYTE*)p, &cb);
if (err == ERROR_MORE_DATA)
continue;
if (err)
break;
Value.ReleaseBuffer();
return Value.GetString();
}
return NULL;
}
bool
ExpandEnvStrings(CStringW &Str)
{
CStringW buf;
DWORD cch = ExpandEnvironmentStringsW(Str, NULL, 0);
if (cch)
{
if (ExpandEnvironmentStringsW(Str, buf.GetBuffer(cch), cch) == cch)
{
buf.ReleaseBuffer(cch - 1);
Str = buf;
return true;
}
}
return false;
}
BOOL
SearchPatternMatch(LPCWSTR szHaystack, LPCWSTR szNeedle)
{
if (!*szNeedle)
return TRUE;
/* TODO: Improve pattern search beyond a simple case-insensitive substring search. */
return StrStrIW(szHaystack, szNeedle) != NULL;
}
BOOL
DeleteDirectoryTree(LPCWSTR Dir, HWND hwnd)
{
CStringW from(Dir);
UINT cch = from.GetLength();
from.Append(L"00");
LPWSTR p = from.GetBuffer();
p[cch] = p[cch + 1] = L'\0'; // Double null-terminate
UINT fof = FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_SILENT;
SHFILEOPSTRUCT shfos = { hwnd, FO_DELETE, p, NULL, (FILEOP_FLAGS)fof };
return SHFileOperationW(&shfos);
}
UINT
CreateDirectoryTree(LPCWSTR Dir)
{
UINT err = SHCreateDirectory(NULL, Dir);
return err == ERROR_ALREADY_EXISTS ? 0 : err;
}
CStringW
SplitFileAndDirectory(LPCWSTR FullPath, CStringW *pDir)
{
CPathW dir = FullPath;
//int win = dir.ReverseFind(L'\\'), nix = dir.ReverseFind(L'/'), sep = max(win, nix);
int sep = dir.FindFileName();
CStringW file = dir.m_strPath.Mid(sep);
if (pDir)
*pDir = sep == -1 ? L"" : dir.m_strPath.Left(sep - 1);
return file;
}
HRESULT
GetSpecialPath(UINT csidl, CStringW &Path, HWND hwnd)
{
if (!SHGetSpecialFolderPathW(hwnd, Path.GetBuffer(MAX_PATH), csidl, TRUE))
return E_FAIL;
Path.ReleaseBuffer();
return S_OK;
}
HRESULT
GetKnownPath(REFKNOWNFOLDERID kfid, CStringW &Path, DWORD Flags)
{
PWSTR p;
FARPROC f = GetProcAddress(LoadLibraryW(L"SHELL32"), "SHGetKnownFolderPath");
if (!f)
return HRESULT_FROM_WIN32(ERROR_OLD_WIN_VERSION);
HRESULT hr = ((HRESULT(WINAPI*)(REFKNOWNFOLDERID,UINT,HANDLE,PWSTR*))f)(kfid, Flags, NULL, &p);
if (FAILED(hr))
return hr;
Path = p;
CoTaskMemFree(p);
return hr;
}
HRESULT
GetProgramFilesPath(CStringW &Path, BOOL PerUser, HWND hwnd)
{
if (!PerUser)
return GetSpecialPath(CSIDL_PROGRAM_FILES, Path, hwnd);
HRESULT hr = GetKnownPath(FOLDERID_UserProgramFiles, Path);
if (FAILED(hr))
{
hr = GetSpecialPath(CSIDL_LOCAL_APPDATA, Path, hwnd);
// Use the correct path on NT6 (on NT5 the path becomes a bit long)
if (SUCCEEDED(hr) && LOBYTE(GetVersion()) >= 6)
{
Path = BuildPath(Path, L"Programs"); // Should not be localized
}
}
return hr;
}