chibicitiberiu/rainmeter-studio
1
0
Fork 0
mirror of https://github.com/chibicitiberiu/rainmeter-studio.git synced 2024-02-24 04:33:31 +00:00
Code Issues Releases Wiki Activity
c3ed2e5fa3
rainmeter-studio/SkinInstaller/Rainstaller.cpp

1749 lines
55 KiB
C++
Raw Blame History

/*
Copyright (C) 2011 Birunthan Mohanathas
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#include "StdAfx.h"
#include "resource.h"
#include "unzip.h"
#include "Application.h"
#include "Rainstaller.h"
RMSKIN_DATA data;
HBITMAP hBitmap;
HANDLE hMapFile;
LPCTSTR pBuffer;
std::wstring exeFile;
#define APP_NAME L"Rainstaller"
/*
** Entry point
*/
int Rainstaller(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow)
{
if (wcscmp(lpCmdLine, L"/ELEVATE") == 0)
{
if (!IsCurrentProcessAdmin())
{
std::wstring error = L"Elevated process does not have administrative rights.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
// If the main (or dialog) process of Rainstaller isn't running as admin, it calls CreateProcessElevated()
// to create create an elevated process of Rainstaller with the /ELEVATE switch.
hMapFile = OpenFileMapping(FILE_MAP_READ, FALSE, OBJECT_NAME);
if (hMapFile == NULL)
{
std::wstring error = L"Error in elevated process: unable to open file mapping.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
pBuffer = (LPCTSTR)MapViewOfFile(hMapFile, FILE_MAP_READ, 0, 0, 2048);
if (pBuffer == NULL)
{
CloseHandle(hMapFile);
std::wstring error = L"Error in elevated process: unable to view map of file.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
// Split the shared buffer into tokens
std::vector<std::wstring> vec = Tokenize(pBuffer, L"\"");
UnmapViewOfFile(pBuffer);
CloseHandle(hMapFile);
// Check that the .rmskin file is acessible from elevated account
data.rmskinFile = vec[0];
if (_waccess(data.rmskinFile.c_str(), 0) == -1)
{
std::wstring error = L"Error in elevated process: Unable to access .rmskin file.\n\nMake sure that the .rmskin file saved at a location accessible to all users and try again.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
// Assign the tokens to variables
data.instType = INSTTYPE_ADMIN;
data.rootLen = _wtoi(vec[1].c_str());
data.rainmeterPath = vec[2];
data.backupFolder = vec[3];
data.addonsList = vec[4];
data.pluginsList = vec[5];
data.fontsList = vec[6];
if (vec[7] == L"1") data.rainmeterFonts = true;
if (!InstallComponents(&data))
{
std::wstring error = L"Error in elevated process: Install failed.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
return 0;
}
// Check whether Rainstaller.exe is already running and bring it to front if so
HANDLE hMutex;
if (IsRunning(L"RainstallerMutex_6D42B76464DA", &hMutex))
{
HWND hwnd = FindWindow(L"#32770", L"Rainstaller");
SetForegroundWindow(hwnd);
return 0;
}
// Avoid loading a dll from current directory
SetDllDirectory(L"");
WCHAR buffer[MAX_PATH];
GetModuleFileName(hInstance, buffer, MAX_PATH);
exeFile = buffer;
// Remove the module's name from the path
WCHAR* pos = wcsrchr(buffer, L'\\');
if (pos)
{
*(pos + 1) = L'\0';
}
std::wstring str = buffer;
str += L"Rainmeter.exe";
if (_waccess(str.c_str(), 0) != 0)
{
std::wstring error = L"Unable to locate Rainmeter.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
data.rainmeterPath = buffer;
str.replace(str.length() - 3, 3, L"ini");
// Find the settings file and read skins path off it
if (_waccess(str.c_str(), 0) == 0)
{
data.iniPath = buffer;
if (GetPrivateProfileString(L"Rainmeter", L"SkinPath", L"", buffer, MAX_LINE_LENGTH, str.c_str()) > 0)
{
data.skinsPath = buffer;
}
else
{
data.skinsPath = data.rainmeterPath;
data.skinsPath += L"Skins\\";
}
}
else
{
HRESULT hr = SHGetFolderPath(NULL, CSIDL_APPDATA, NULL, SHGFP_TYPE_CURRENT, buffer);
wcscat(buffer, L"\\Rainmeter\\");
data.iniPath = buffer;
wcscat(buffer, L"Rainmeter.ini");
if (SUCCEEDED(hr) && _waccess(buffer, 0) == 0)
{
if (GetPrivateProfileString(L"Rainmeter", L"SkinPath", L"", buffer, MAX_LINE_LENGTH, buffer) > 0)
{
data.skinsPath = buffer;
}
else
{
std::wstring error = L"SkinPath not found.\nMake sure that Rainmeter has been run at least once.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
}
else
{
std::wstring error = L"Rainmeter.ini not found.\nMake sure that Rainmeter has been run at least once.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return 1;
}
}
data.rmskinFile = lpCmdLine;
if (data.rmskinFile.empty())
{
// Show the Open File dialog if no arguments were given
OPENFILENAME ofn = {0};
ofn.lStructSize = sizeof(OPENFILENAME);
ofn.lpstrFilter = L"Rainmeter skin file (.rmskin)\0*.rmskin;*.zip";
ofn.nFilterIndex = 1;
ofn.lpstrFile = buffer;
ofn.nMaxFile = MAX_PATH;
ofn.lpstrTitle = L"Select Rainmeter skin file";
ofn.lpstrDefExt = L"rmskin";
ofn.Flags = OFN_FILEMUSTEXIST;
buffer[0] = L'\0';
if (!GetOpenFileName((OPENFILENAME*)&ofn)) return 0; // Abort without warning on cancel
data.rmskinFile = buffer;
}
else if (data.rmskinFile[0] == L'\"')
{
// Remove first and last quote if needed
data.rmskinFile.erase(0, 1);
data.rmskinFile.resize(data.rmskinFile.length() - 1);
}
return ReadArchive() ? DialogBox(hInstance, MAKEINTRESOURCE(IDD_INSTALLER_DIALOG), NULL, (DLGPROC)DlgProc) : 1;
}
/*
** Main dialog window procedure
*/
BOOL CALLBACK DlgProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
switch (msg)
{
case WM_INITDIALOG:
InitDialog(hwnd);
SetForegroundWindow(hwnd);
break;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDC_CANCEL_BUTTON:
EndDialog(hwnd, 0);
break;
case IDC_INSTALL_BUTTON:
InitInstall(hwnd);
break;
case IDC_SKINS_LABEL:
EnableWindow((HWND)lParam, 0);
data.skinsList.clear();
break;
case IDC_THEMES_LABEL:
EnableWindow((HWND)lParam, 0);
data.themesList.clear();
break;
case IDC_ADDONS_LABEL:
EnableWindow((HWND)lParam, 0);
data.addonsList.clear();
break;
case IDC_PLUGINS_LABEL:
EnableWindow((HWND)lParam, 0);
data.pluginsList.clear();
break;
case IDC_FONTS_LABEL:
EnableWindow((HWND)lParam, 0);
data.fontsList.clear();
break;
}
break;
case WM_TIMER:
switch (wParam)
{
case TIMER_THREAD:
{
DWORD exitCode;
GetExitCodeThread(data.instHandle, &exitCode);
if (exitCode != STILL_ACTIVE)
{
KillTimer(hwnd, TIMER_THREAD);
CloseHandle(data.instHandle);
if (exitCode != 0)
{
std::wstring error = L"Install thead failed.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
}
else if (data.launchRainmeter)
{
LaunchRainmeter();
}
EndDialog(hwnd, 0);
}
}
break;
case TIMER_PROCESS:
{
DWORD exitCode;
GetExitCodeProcess(data.instHandle, &exitCode);
if (exitCode != STILL_ACTIVE)
{
KillTimer(hwnd, TIMER_PROCESS);
UnmapViewOfFile(pBuffer);
CloseHandle(hMapFile);
CloseHandle(data.instHandle);
if (exitCode == 0)
{
if (!data.skinsList.empty() || !data.themesList.empty())
{
data.instType = INSTTYPE_NOADMIN;
data.instHandle = (HANDLE)_beginthreadex(NULL, 0, CreateInstallThread, &data, 0, NULL);
SetTimer(hwnd, TIMER_THREAD, 100, NULL);
break;
}
else if (data.launchRainmeter)
{
LaunchRainmeter();
}
}
EndDialog(hwnd, 0);
}
}
break;
}
break;
case WM_CLOSE:
// Don't close dialog if install is running
if (!data.instHandle) EndDialog(hwnd, 0);
break;
default:
return FALSE;
}
return TRUE;
}
void InitDialog(HWND hwnd)
{
SendMessage(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDI_INSTALLER)));
SendMessage(GetDlgItem(hwnd, IDC_LAUNCH_CHECKBOX), BM_SETCHECK, BST_CHECKED, 0);
ShowWindow(GetDlgItem(hwnd, IDC_PROGRESSBAR), SW_HIDE);
if (data.loadTheme.empty())
{
ShowWindow(GetDlgItem(hwnd, IDC_THEME_CHECKBOX), SW_HIDE);
}
else
{
std::wstring text = L"Apply theme (";
text += data.loadTheme;
text += L")";
SendMessage(GetDlgItem(hwnd, IDC_THEME_CHECKBOX), BM_SETCHECK, BST_CHECKED, 0);
SetDlgItemText(hwnd, IDC_THEME_CHECKBOX, text.c_str());
}
if (data.backupFolder.empty()) // Hide the backup footnote if not needed
{
RECT dlgRect, sepRect;
GetWindowRect(hwnd, &dlgRect);
GetWindowRect(GetDlgItem(hwnd, IDC_SEPERATOR), &sepRect);
// Width is kept same, the height is cut to the relative y-pos of the sepeator
SetWindowPos(hwnd, NULL, NULL, NULL, dlgRect.right - dlgRect.left,
dlgRect.bottom - dlgRect.top - (dlgRect.bottom - sepRect.bottom) + 1,
SWP_NOMOVE | SWP_NOREPOSITION | SWP_NOZORDER);
}
else
{
HICON hIcon = (HICON)LoadIcon(NULL, IDI_WARNING);
SendMessage(GetDlgItem(hwnd, IDC_WARNING_ICON), STM_SETICON, (WPARAM)hIcon, 0);
}
// Show UAC sheild if needed
if (!IsCurrentProcessAdmin() && (!data.addonsList.empty() || !data.fontsList.empty() || !data.pluginsList.empty()))
{
SendMessage(GetDlgItem(hwnd, IDC_INSTALL_BUTTON), BCM_SETSHIELD, 0, (LPARAM)TRUE);
}
// Set custom header
if (hBitmap)
{
SendMessage(GetDlgItem(hwnd, IDC_BITMAP), STM_SETIMAGE, (WPARAM)IMAGE_BITMAP, (LPARAM)hBitmap);
}
// Get the system font and size (Tahoma/8 on XP, Segoe UI/9 on Vista+)
NONCLIENTMETRICS ncm;
ncm.cbSize = sizeof(NONCLIENTMETRICS) - sizeof(ncm.iPaddedBorderWidth);
SystemParametersInfo(SPI_GETNONCLIENTMETRICS, sizeof(NONCLIENTMETRICS), &ncm, 0);
HFONT hFont = CreateFontIndirect(&ncm.lfMenuFont);
ncm.lfMenuFont.lfWeight = FW_BOLD;
HFONT hFontBold = CreateFontIndirect(&ncm.lfMenuFont);
SendMessage(GetDlgItem(hwnd, IDC_NAME_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_AUTHOR_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_VERSION_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_SKINS_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_THEMES_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_ADDONS_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_PLUGINS_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_FONTS_LABEL), WM_SETFONT, (WPARAM)hFontBold, 0);
SendMessage(GetDlgItem(hwnd, IDC_NAME_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_AUTHOR_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_VERSION_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_SKINS_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_THEMES_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_ADDONS_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_PLUGINS_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_FONTS_VALUE), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_THEME_CHECKBOX), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_LAUNCH_CHECKBOX), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_INSTALL_BUTTON), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_CANCEL_BUTTON), WM_SETFONT, (WPARAM)hFont, 0);
SendMessage(GetDlgItem(hwnd, IDC_BACKUP_LABEL), WM_SETFONT, (WPARAM)hFont, 0);
SetDlgItemText(hwnd, IDC_NAME_VALUE, data.packageName.c_str());
SetDlgItemText(hwnd, IDC_AUTHOR_VALUE, data.packageAuthor.c_str());
SetDlgItemText(hwnd, IDC_VERSION_VALUE, data.packageVersion.c_str());
data.skinsList.empty() ? EnableWindow(GetDlgItem(hwnd, IDC_SKINS_LABEL), 0) : SetDlgItemText(hwnd, IDC_SKINS_VALUE, data.skinsList.c_str());
data.themesList.empty() ? EnableWindow(GetDlgItem(hwnd, IDC_THEMES_LABEL), 0) : SetDlgItemText(hwnd, IDC_THEMES_VALUE, data.themesList.c_str());
data.addonsList.empty() ? EnableWindow(GetDlgItem(hwnd, IDC_ADDONS_LABEL), 0) : SetDlgItemText(hwnd, IDC_ADDONS_VALUE, data.addonsList.c_str());
data.pluginsList.empty() ? EnableWindow(GetDlgItem(hwnd, IDC_PLUGINS_LABEL), 0) : SetDlgItemText(hwnd, IDC_PLUGINS_VALUE, data.pluginsList.c_str());
data.fontsList.empty() ? EnableWindow(GetDlgItem(hwnd, IDC_FONTS_LABEL), 0) : SetDlgItemText(hwnd, IDC_FONTS_VALUE, data.fontsList.c_str());
// Enable tooltips for the component lists
HWND hwndTT = CreateWindow(TOOLTIPS_CLASS, NULL, WS_POPUP, 0, 0, 0, 0, hwnd, NULL, NULL, 0);
if (hwndTT)
{
SendMessage(hwndTT, TTM_ACTIVATE, TRUE, 0);
TOOLINFO toolinfo = {0};
toolinfo.cbSize = sizeof(TOOLINFO);
toolinfo.uFlags = TTF_SUBCLASS | TTF_IDISHWND;
toolinfo.hwnd = hwnd;
toolinfo.hinst = NULL;
toolinfo.uId = (LPARAM)GetDlgItem(hwnd, IDC_SKINS_VALUE);
toolinfo.lpszText = (LPTSTR)data.skinsList.c_str();
SendMessage(hwndTT, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
toolinfo.uId = (LPARAM)GetDlgItem(hwnd, IDC_THEMES_VALUE);
toolinfo.lpszText = (LPTSTR)data.themesList.c_str();
SendMessage(hwndTT, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
toolinfo.uId = (LPARAM)GetDlgItem(hwnd, IDC_ADDONS_VALUE);
toolinfo.lpszText = (LPTSTR)data.addonsList.c_str();
SendMessage(hwndTT, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
toolinfo.uId = (LPARAM)GetDlgItem(hwnd, IDC_PLUGINS_VALUE);
toolinfo.lpszText = (LPTSTR)data.pluginsList.c_str();
SendMessage(hwndTT, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
toolinfo.uId = (LPARAM)GetDlgItem(hwnd, IDC_FONTS_VALUE);
toolinfo.lpszText = (LPTSTR)data.fontsList.c_str();
SendMessage(hwndTT, TTM_ADDTOOL, 0, (LPARAM)&toolinfo);
}
}
void InitInstall(HWND hwnd)
{
// Disable the X button
HMENU hMenu = GetSystemMenu(hwnd, false);
RemoveMenu(hMenu, GetMenuItemCount(hMenu) - 1, MF_DISABLED | MF_BYPOSITION);
EnableWindow(GetDlgItem(hwnd, IDC_CANCEL_BUTTON), 0);
EnableWindow(GetDlgItem(hwnd, IDC_INSTALL_BUTTON), 0);
EnableWindow(GetDlgItem(hwnd, IDC_THEME_CHECKBOX), 0);
EnableWindow(GetDlgItem(hwnd, IDC_LAUNCH_CHECKBOX), 0);
if (SendMessage(GetDlgItem(hwnd, IDC_LAUNCH_CHECKBOX), BM_GETCHECK, 0, 0) == BST_CHECKED) data.launchRainmeter = true;
if (IsCurrentProcessAdmin())
{
data.instType = INSTTYPE_FULL;
data.instHandle = (HANDLE)_beginthreadex(NULL, 0, CreateInstallThread, &data, 0, NULL);
SetTimer(hwnd, TIMER_THREAD, 250, NULL);
}
else
{
if (!data.addonsList.empty() || !data.pluginsList.empty() || !data.fontsList.empty())
{
data.instHandle = CreateProcessElevated(hwnd);
if (data.instHandle == NULL)
{
std::wstring error = L"Failed to create elevated process.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
UnmapViewOfFile(pBuffer);
CloseHandle(hMapFile);
EndDialog(hwnd, 0);
}
SetTimer(hwnd, TIMER_PROCESS, 250, NULL);
}
else if (!data.skinsList.empty() || !data.themesList.empty())
{
data.instType = INSTTYPE_NOADMIN;
data.instHandle = (HANDLE)_beginthreadex(NULL, 0, CreateInstallThread, &data, 0, NULL);
SetTimer(hwnd, TIMER_THREAD, 250, NULL);
}
}
// Hide the backup text if necessary and show the progress bar
ShowWindow(GetDlgItem(hwnd, IDC_PROGRESSBAR), SW_SHOWNORMAL);
SendMessage(GetDlgItem(hwnd, IDC_PROGRESSBAR), PBM_SETMARQUEE, (WPARAM)true, IsAboveVista() ? 0 : 50);
SendMessage(GetDlgItem(hwnd, IDC_INSTALL_BUTTON), BCM_SETSHIELD, 0, (LPARAM)FALSE);
SetDlgItemText(hwnd, IDC_INSTALL_BUTTON, L"Installing..");
// Clear loadTheme if checkbox unchecked
if (SendMessage(GetDlgItem(hwnd, IDC_THEME_CHECKBOX), BM_GETCHECK, 0, 0) != BST_CHECKED) data.loadTheme.clear();
}
/*
** Run As dialog window prodcedure(for Vista/7 with UAC disabled)
*/
BOOL CALLBACK RunAsProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
// Based on the work of Anders Kjersem
switch (msg)
{
case WM_INITDIALOG:
{
HMODULE hDLL = LoadLibrary(L"shell32.dll");
HICON hIcon = (HICON)LoadImage(hDLL, MAKEINTRESOURCE(220), IMAGE_ICON, 16, 16, LR_SHARED);
SendMessage(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)hIcon);
hIcon = (HICON)LoadImage(hDLL, MAKEINTRESOURCE(194), IMAGE_ICON, 32, 32, LR_SHARED);
SendMessage(GetDlgItem(hwnd, IDC_KEY_ICON), STM_SETICON, (WPARAM)hIcon, 0);
FreeLibrary(hDLL);
SendMessage(GetDlgItem(hwnd, IDC_USERNAME), EM_LIMITTEXT, (WPARAM)256, 0);
SendMessage(GetDlgItem(hwnd, IDC_PASSWORD), EM_LIMITTEXT, (WPARAM)256, 0);
SendMessage(GetDlgItem(hwnd, IDC_SPECIFIED_BUTTON), BM_SETCHECK, BST_CHECKED, 0);
}
return TRUE;
case WM_CLOSE:
return EndDialog(hwnd, 1);
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDOK:
{
WCHAR wszPwd[256];
WCHAR wszUser[256];
PROCESS_INFORMATION pi = {0};
STARTUPINFO si = {sizeof(STARTUPINFO)};
SendMessage(GetDlgItem(hwnd,IDC_USERNAME), WM_GETTEXT, (WPARAM)256, (LPARAM)wszUser);
SendMessage(GetDlgItem(hwnd,IDC_PASSWORD), WM_GETTEXT, (WPARAM)256, (LPARAM)wszPwd);
WCHAR arg[MAX_PATH];
_snwprintf_s(arg, _TRUNCATE, L"\"%s\" /ELEVATE", exeFile.c_str());
if (!CreateProcessWithLogonW(wszUser, 0, wszPwd, LOGON_WITH_PROFILE, NULL, arg, 0, 0, 0, &si, &pi))
{
std::wstring error = L"Unable to logon with given creditials.\nPlease ensure that both the username and password are correct.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
}
else
{
if (pi.hProcess)
{
data.instHandle = pi.hProcess;
EndDialog(hwnd, 0);
}
else
{
EndDialog(hwnd, 1);
}
}
}
break;
case IDCANCEL:
EndDialog(hwnd, 1);
break;
}
break;
}
return FALSE;
}
/*
** Go through archive, list contents, and read settings
*/
bool ReadArchive()
{
unzFile ufile = unzOpen(ConvertToAscii(data.rmskinFile.c_str()).c_str());
if (!ufile)
{
std::wstring error = L"The specified file is not a valid archive:\n";
error += data.rmskinFile;
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
// Get temp file path for the configuration file/bitmap
WCHAR tempFile[MAX_PATH];
GetTempPath(MAX_PATH, tempFile);
GetTempFileName(tempFile, L"dat", 0, tempFile);
char cBuffer[MAX_PATH * 3];
WCHAR buffer[MAX_PATH];
unz_file_info ufi;
// Loop through the contents of the archive until the settings file is found
do
{
if (unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0) != UNZ_OK) break;
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
WCHAR* fileName = wcsrchr(buffer, L'/');
if (!fileName)
{
fileName = buffer;
}
else
{
++fileName; // Get rid of leading slash
}
if (_wcsicmp(fileName, L"RMSKIN.ini") == 0 || _wcsicmp(fileName, L"Rainstaller.cfg") == 0)
{
if (!ExtractCurrentFile(ufile, tempFile)) return false;
// The number of characters in path before the configuration file
data.rootLen = wcslen(buffer) - wcslen(fileName);
break;
}
else if (_wcsicmp(fileName, L"RMSKIN.bmp") == 0 || _wcsicmp(fileName, L"Rainstaller.bmp") == 0)
{
if (!ExtractCurrentFile(ufile, tempFile)) return false;
hBitmap = (HBITMAP)LoadImage(NULL, tempFile, IMAGE_BITMAP, 0, 0, LR_LOADFROMFILE);
}
} while (unzGoToNextFile(ufile) == UNZ_OK);
if (!ReadSettings(tempFile))
{
unzClose(ufile);
std::wstring error = L"Invalid settings file in:\n";
error += data.rmskinFile;
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
bool backupRequired = false;
WCHAR* filePath = buffer;
filePath += data.rootLen;
WCHAR* filePath6(filePath), * filePath7(filePath), * filePath14(filePath);
filePath6 += 6; // Relative to path to the beginning of the skin/font root
filePath7 += 7; // Relative to path to the beginning of the theme/addon root
filePath14 += 14; // Relative to path to the beginning of the plugin root
std::wstring themesPath(data.iniPath), addonsPath(data.rainmeterPath), pluginsPath(data.rainmeterPath);
themesPath += L"Themes\\";
addonsPath += L"Addons\\";
pluginsPath += L"Plugins\\";
// Loop through the archive a second time now that we know the where the contents are. Ugly code ahead!
unzGoToFirstFile(ufile);
do
{
if (unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0) != UNZ_OK) break;
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
if (_wcsnicmp(filePath, L"Skins/", 6) == 0)
{
WCHAR* pos = wcschr(filePath6, L'/');
if (pos)
{
pos[0] = L'\0';
std::wstring name(filePath6), root(filePath);
root += L"/";
int len = root.length();
// Loop until we get to another skin (or another component altogether)
while (unzGoToNextFile(ufile) == UNZ_OK)
{
unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
if (_wcsnicmp(filePath, root.c_str(), len) != 0) break;
}
if (_wcsicmp(name.c_str(), L"Backup") != 0)
{
// Add the folder to list
if (!data.skinsList.empty()) data.skinsList += L" | ";
data.skinsList += name;
name.insert(0, data.skinsPath);
if (_waccess(name.c_str(), 0) == 0)
{
data.skinsList += L"*";
backupRequired = true;
}
}
continue;
}
}
else if (_wcsnicmp(filePath, L"Themes/", 7) == 0)
{
WCHAR* pos = wcschr(filePath7, L'/');
if (pos)
{
pos[0] = L'\0';
std::wstring name(filePath7), root(filePath);
root += L"/";
int len = root.length();
while (unzGoToNextFile(ufile) == UNZ_OK)
{
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
if (_wcsnicmp(filePath, root.c_str(), len) != 0) break;
}
if (_wcsicmp(name.c_str(), L"Backup") != 0)
{
if (!data.themesList.empty()) data.themesList += L" | ";
data.themesList += name;
name.insert(0, themesPath);
if (_waccess(name.c_str(), 0) == 0)
{
data.themesList += L"*";
backupRequired = true;
}
}
continue;
}
}
else if (_wcsnicmp(filePath, L"Addons/", 7) == 0)
{
WCHAR* pos = wcschr(filePath7, L'/');
if (pos)
{
pos[0] = L'\0';
std::wstring name(filePath7), root(filePath);
root += L"/";
int len = root.length();
while (unzGoToNextFile(ufile) == UNZ_OK)
{
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
if (_wcsnicmp(filePath, root.c_str(), len) != 0) break;
}
if (_wcsicmp(name.c_str(), L"Backup") != 0 && !IsDefaultAddon(name.c_str()))
{
if (!data.addonsList.empty()) data.addonsList += L" | ";
data.addonsList += name;
name.insert(0, addonsPath);
if (_waccess(name.c_str(), 0) == 0)
{
data.addonsList += L"*";
backupRequired = true;
}
}
continue;
}
}
else if (_wcsnicmp(filePath, L"Fonts/", 6) == 0 && filePath6[0] != L'\0' &&
!wcschr(filePath14, L'/') &&
_wcsicmp(wcschr(filePath6, L'.'), L".ttf") == 0)
{
if (!data.fontsList.empty()) data.fontsList += L" | ";
data.fontsList += filePath6;
}
else if (_wcsnicmp(filePath, PLUGINS_ROOT, 14) == 0 &&
filePath14[0] != L'\0' &&
!wcschr(filePath14, L'/') &&
_wcsicmp(wcschr(filePath14, L'.'), L".dll") == 0
&& !IsDefaultPlugin(filePath14))
{
if (!data.pluginsList.empty()) data.pluginsList += L" | ";
std::wstring name = filePath14;
data.pluginsList += name;
name.insert(0, pluginsPath);
if (_waccess(name.c_str(), 0) == 0)
{
backupRequired = true;
data.pluginsList += L"*";
}
}
} while (unzGoToNextFile(ufile) == UNZ_OK);
unzClose(ufile);
DeleteFile(tempFile);
if (!data.skinsList.empty() || !data.themesList.empty() || !data.fontsList.empty() || !data.addonsList.empty() || !data.pluginsList.empty())
{
if (backupRequired) // Get current time as the name for backup folders
{
SYSTEMTIME lt;
GetLocalTime(&lt);
_snwprintf_s(buffer, _TRUNCATE, L"%02d.%02d.%02d %02d.%02d-%02d", lt.wYear, lt.wMonth, lt.wDay, lt.wHour, lt.wMinute, lt.wSecond);
data.backupFolder = buffer;
}
return true;
}
else
{
std::wstring error = L"The specified file did not contain any components to install:\n";
error += data.rmskinFile;
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
}
bool ExtractCurrentFile(unzFile& ufile, LPCTSTR fileName)
{
if (fileName[wcslen(fileName) - 1] == L'/')
{
CreateDirectory(fileName, 0);
return true;
}
HANDLE hFile = CreateFile(fileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
// Some zip files don't list directories, so try creating all directories first
std::wstring path = fileName;
std::wstring::size_type pos = 0;
while (true)
{
pos = path.find_first_of(L"/", pos);
if (pos == std::wstring::npos) break;
++pos;
std::wstring dir = path.substr(0, pos);
if (_waccess(dir.c_str(), 0) == -1) CreateDirectory(dir.c_str(), 0);
}
hFile = CreateFile(fileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
// Creating the directories didn't help..
std::wstring error = L"Unable to create file: ";
error += fileName;
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK);
return false;
}
}
bool ret = true;
int res = unzOpenCurrentFile(ufile);
if (res != UNZ_OK)
{
std::wstring error = L"Unable to open file for extraction: ";
error += fileName;
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK);
ret = false;
}
else
{
do
{
char buffer[16384];
DWORD dwWritten;
res = unzReadCurrentFile(ufile, buffer, 16384);
if (res < 0 || (res > 0 && !WriteFile(hFile, (LPCVOID)buffer, res, &dwWritten, NULL)))
{
std::wstring error = L"Unable to write file: ";
error += fileName;
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK);
ret = false;
break;
}
} while (res > 0);
if (unzCloseCurrentFile(ufile) == UNZ_CRCERROR)
{
std::wstring error = L"Checksum error (archive is damaged) with file: ";
error += fileName;
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK);
ret = false;
}
}
if (!ret) unzClose(ufile);
CloseHandle(hFile);
return ret;
}
bool ReadSettings(LPCTSTR filePath)
{
std::wstring rainmeterFile = data.rainmeterPath;
rainmeterFile += L"Rainmeter.exe";
std::wstring rainmeterVersion = GetFileVersion(rainmeterFile);
WCHAR buffer[MAX_LINE_LENGTH];
if (GetPrivateProfileString(L"Rainstaller", L"Name", L"", buffer, 64, filePath) > 0)
{
// Old, pre-2.0 format
data.packageName = buffer;
GetPrivateProfileString(L"Rainstaller", L"Author", L"", buffer, 64, filePath);
data.packageAuthor = buffer;
GetPrivateProfileString(L"Rainstaller", L"Version", L"", buffer, 64, filePath);
data.packageVersion = buffer;
if ((GetPrivateProfileString(L"Rainstaller", L"LaunchType", L"", buffer, MAX_LINE_LENGTH, filePath) > 0))
{
if (_wcsicmp(buffer, L"load") == 0)
{
GetPrivateProfileString(L"Rainstaller", L"LaunchCommand", L"", buffer, MAX_LINE_LENGTH, filePath);
data.loadSkins = buffer;
}
else
{
GetPrivateProfileString(L"Rainstaller", L"LaunchCommand", L"", buffer, MAX_LINE_LENGTH, filePath);
data.loadTheme = buffer;
}
}
if (GetPrivateProfileString(L"Rainstaller", L"RainmeterFonts", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.rainmeterFonts = (buffer[0] == L'1') ? true : false;
}
if (GetPrivateProfileString(L"Rainstaller", L"Merge", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.mergeSkins = (buffer[0] == L'1' || buffer[0] == L'2') ? true : false;
}
if (GetPrivateProfileString(L"Rainstaller", L"KeepVar", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.keepVariables = buffer;
}
if ((GetPrivateProfileString(L"Rainstaller", L"MinRainmeterVer", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
&& (CompareVersions(buffer, rainmeterVersion) == 1))
{
std::wstring error = L"This package requires Rainmeter version ";
error += buffer;
error += L" or higher.\nDownload the latest version at rainmeter.net and try again.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK | MB_ICONERROR);
return false;
}
return true;
}
else if (GetPrivateProfileString(L"RMSKIN", L"PackageName", L"", buffer, 64, filePath) > 0)
{
// New format
data.packageName = buffer;
GetPrivateProfileString(L"RMSKIN", L"PackageAuthor", L"", buffer, 64, filePath);
data.packageAuthor = buffer;
GetPrivateProfileString(L"RMSKIN", L"PackageVersion", L"", buffer, 64, filePath);
data.packageVersion = buffer;
if (GetPrivateProfileString(L"RMSKIN", L"LoadTheme", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.loadTheme = buffer;
}
if (GetPrivateProfileString(L"RMSKIN", L"LoadSkins", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.loadSkins = buffer;
}
if (GetPrivateProfileString(L"RMSKIN", L"FontDirectory", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.rainmeterFonts = (_wcsicmp(buffer, L"rainmeter") == 0) ? true : false;
}
if (GetPrivateProfileString(L"RMSKIN", L"MergeSkins", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.mergeSkins = (buffer[0] == L'1') ? true : false;
}
if (GetPrivateProfileString(L"RMSKIN", L"KeepVariables", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
{
data.keepVariables = buffer;
}
if ((GetPrivateProfileString(L"RMSKIN", L"MinimumRainmeter", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
&& (CompareVersions(buffer, rainmeterVersion) == 1))
{
std::wstring error = L"This package requires Rainmeter version ";
error += buffer;
error += L" or higher.\nDownload the latest version at rainmeter.net and try again.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK | MB_ICONERROR);
return false;
}
if ((GetPrivateProfileString(L"RMSKIN", L"MinimumDotNET", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
&& (CompareVersions(buffer, GetDotNETVersion()) == 1))
{
std::wstring error = L"This packages requires .NET framework version ";
error += buffer;
error += L" or higher.\nEnsure that you have the required .NET framework and try again.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_OK | MB_ICONERROR);
return false;
}
if ((GetPrivateProfileString(L"RMSKIN", L"MinimumWindows", L"", buffer, MAX_LINE_LENGTH, filePath) > 0)
&& (CompareVersions(buffer, GetWindowsVersion()) == 1))
{
std::wstring error = L"This package requires Windows ";
error += buffer;
error += L" or higher.\nContact the package author for more information.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
return true;
}
else
{
return false;
}
}
bool IsDefaultPlugin(LPCTSTR plugin)
{
return (_wcsicmp(plugin, L"AdvancedCPU.dll") == 0 ||
_wcsicmp(plugin, L"CoreTemp.dll") == 0 ||
_wcsicmp(plugin, L"FolderInfo.dll") == 0 ||
_wcsicmp(plugin, L"InputText.dll") == 0 ||
_wcsicmp(plugin, L"iTunesPlugin.dll") == 0 ||
_wcsicmp(plugin, L"MediaKey.dll") == 0 ||
_wcsicmp(plugin, L"NowPlaying.dll") == 0 ||
_wcsicmp(plugin, L"PerfMon.dll") == 0 ||
_wcsicmp(plugin, L"PingPlugin.dll") == 0 ||
_wcsicmp(plugin, L"PowerPlugin.dll") == 0 ||
_wcsicmp(plugin, L"QuotePlugin.dll") == 0 ||
_wcsicmp(plugin, L"RecycleManager.dll") == 0 ||
_wcsicmp(plugin, L"ResMon.dll") == 0 ||
_wcsicmp(plugin, L"SpeedFanPlugin.dll") == 0 ||
_wcsicmp(plugin, L"SysInfo.dll") == 0 ||
_wcsicmp(plugin, L"VirtualDesktops.dll") == 0 ||
_wcsicmp(plugin, L"WebParser.dll") == 0 ||
_wcsicmp(plugin, L"WifiStatus.dll") == 0 ||
_wcsicmp(plugin, L"Win7AudioPlugin.dll") == 0 ||
_wcsicmp(plugin, L"WindowMessagePlugin.dll") == 0) ? true : false;
}
bool IsDefaultAddon(LPCTSTR addon)
{
return (_wcsnicmp(addon, L"RainBackup", 10) == 0 ||
_wcsnicmp(addon, L"Rainstaller", 11) == 0) ? true : false;
}
/*
** Backup and install components
*/
bool InstallComponents(RMSKIN_DATA* data)
{
if (!CloseRainmeterIfActive())
{
std::wstring error = L"Failed to close Rainmeter.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
int result;
char cBuffer[MAX_PATH * 3];
WCHAR buffer[MAX_PATH];
unzFile ufile = unzOpen(ConvertToAscii(data->rmskinFile.c_str()).c_str());
unz_file_info ufi;
WCHAR* filePath = buffer;
filePath += data->rootLen;
// Loop through the archive and install stuff. Get ready to scratch your head.
do
{
// Assume no errors as we've already looped through archive
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
if (data->instType & INSTTYPE_NOADMIN)
{
if (_wcsnicmp(filePath, L"Skins/", 6) == 0 && wcslen(filePath) > 6 && !data->skinsList.empty())
{
WCHAR* fileName = filePath;
fileName += 6;
SetCurrentDirectory(data->skinsPath.c_str());
if (!data->mergeSkins && !BackupComponent(data->backupFolder, data->skinsList, data->skinsPath)) break;
// Loop and extract everything in Skins/
do
{
if (!ExtractCurrentFile(ufile, fileName)) return false;
result = unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
} while (result == UNZ_OK && _wcsnicmp(filePath, L"Skins/", 6) == 0);
if (!data->keepVariables.empty()) KeepVariables(data->backupFolder, data->skinsPath, data->keepVariables);
continue; // To skip the unzGotoNextFile at end of loop
}
else if (_wcsnicmp(filePath, L"Themes/", 7) == 0 && wcslen(filePath) > 7 && !data->themesList.empty())
{
std::wstring themesPath = data->iniPath;
themesPath += L"Themes\\";
SetCurrentDirectory(data->iniPath.c_str());
if (!BackupComponent(data->backupFolder, data->themesList, themesPath)) break;
do
{
if (!ExtractCurrentFile(ufile, filePath)) return false;
if (wcsstr(filePath, L"Rainmeter.thm")) // Remove per-user values from Rainmeter.thm
{
WritePrivateProfileString(L"Rainmeter", L"SkinPath", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"ConfigEditor", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"DisableDragging", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"DisableRDP", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"DisableVersionCheck", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"Logging", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"Language", NULL, filePath);
WritePrivateProfileString(L"Rainmeter", L"Debug", NULL, filePath);
}
result = unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
} while (result == UNZ_OK && _wcsnicmp(filePath, L"Themes/", 7) == 0);
continue;
}
}
if (data->instType & INSTTYPE_ADMIN)
{
if (_wcsnicmp(filePath, L"Addons/", 7) == 0 && wcslen(filePath) > 7 && !data->addonsList.empty())
{
WCHAR* fileName = filePath;
fileName += 7;
std::wstring addonsPath = data->rainmeterPath;
addonsPath += L"Addons\\";
SetCurrentDirectory(data->rainmeterPath.c_str());
if (!BackupComponent(data->backupFolder, data->addonsList, addonsPath)) break;
do
{
if (!IsDefaultAddon(fileName))
{
if (!ExtractCurrentFile(ufile, filePath)) return false;
}
result = unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
} while (result == UNZ_OK && _wcsnicmp(filePath, L"Addons/", 7) == 0);
continue;
}
else if (_wcsnicmp(filePath, L"Fonts/", 6) == 0 && wcslen(filePath) > 6 && !data->fontsList.empty())
{
std::wstring fontsPath;
HMODULE hDLL;
GETFONTRESOURCEINFO GetFontResourceInfo;
HKEY hKey;
if (data->rainmeterFonts)
{
fontsPath = data->rainmeterPath;
fontsPath += L"Fonts\\";
CreateDirectory(fontsPath.c_str(), NULL);
}
else
{
WCHAR buffer[MAX_PATH];
if (!SHGetSpecialFolderPath(NULL, buffer, CSIDL_FONTS, FALSE)) break;
fontsPath = buffer;
fontsPath += L"\\";
// Undocumented API
hDLL = LoadLibrary(L"gdi32.dll");
GetFontResourceInfo = (GETFONTRESOURCEINFO)GetProcAddress(hDLL, "GetFontResourceInfoW");
if (!GetFontResourceInfo ||
RegOpenKeyEx(HKEY_LOCAL_MACHINE,
L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Fonts",
0,
KEY_READ | KEY_WRITE,
&hKey) != ERROR_SUCCESS)
{
std::wstring error = L"Failed to access GetFontResourceInfo.";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
FreeLibrary(hDLL);
break;
}
}
SetCurrentDirectory(fontsPath.c_str());
WCHAR* fileName = filePath;
fileName += 6;
do
{
if (!wcschr(fileName, L'/') && _wcsicmp(wcschr(fileName, L'.'), L".ttf") == 0 && _waccess(fileName, 0) == -1)
{
if (!ExtractCurrentFile(ufile, fileName)) return false;
if (!data->rainmeterFonts)
{
// Write the font information to registry so that it will be enumerated by Rainmeter
DWORD dwSize = MAX_PATH;
WCHAR buffer[MAX_PATH];
AddFontResource(fileName);
if (GetFontResourceInfo(fileName, &dwSize, &buffer, 1) != 0)
{
wcscat(buffer, L" (TrueType)");
if (RegQueryValueEx(hKey, buffer, NULL, NULL, NULL, NULL) == ERROR_FILE_NOT_FOUND)
{
RegSetValueEx(hKey, buffer, 0, REG_SZ, (PBYTE)fileName, ((DWORD)wcslen(fileName) + 1) * sizeof(WCHAR));
}
else
{
// Font is already installed with a different filename
RemoveFontResource(fileName);
DeleteFile(fileName);
}
}
}
}
result = unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
while (WCHAR* pos = wcschr(buffer, L'\\')) *pos = L'/';
} while (result == UNZ_OK && _wcsnicmp(filePath, L"Fonts/", 6) == 0);
FreeLibrary(hDLL);
if (!data->rainmeterFonts) PostMessage(HWND_BROADCAST, WM_FONTCHANGE, 0, 0);
continue;
}
else if (_wcsnicmp(filePath, PLUGINS_ROOT, 14) == 0 && wcslen(filePath) > 14 && !data->pluginsList.empty())
{
std::wstring pluginsPath = data->rainmeterPath;
pluginsPath += L"Plugins\\";
SetCurrentDirectory(pluginsPath.c_str());
WCHAR* fileName = filePath;
fileName += 14;
if (!BackupComponent(data->backupFolder, data->pluginsList, pluginsPath)) break;
do
{
if (!wcschr(fileName, L'/') && _wcsicmp(wcschr(fileName, L'.'), L".dll") == 0 && !IsDefaultPlugin(fileName))
{
if (!ExtractCurrentFile(ufile, fileName)) return false;
}
result = unzGoToNextFile(ufile);
unzGetCurrentFileInfo(ufile, &ufi, cBuffer, MAX_PATH * 3, NULL, 0, NULL, 0);
MultiByteToWideChar(CP_ACP, 0, cBuffer, strlen(cBuffer) + 1, buffer, MAX_PATH);
} while (result == UNZ_OK && _wcsnicmp(filePath, PLUGINS_ROOT, 14) == 0);
continue;
}
}
result = unzGoToNextFile(ufile);
} while (result == UNZ_OK);
unzClose(ufile);
return (result == UNZ_END_OF_LIST_OF_FILE) ? true : false;
}
bool BackupComponent(const std::wstring& backupFolder, const std::wstring& list, const std::wstring& path)
{
if (!list.empty() && (list.find(L"*") != std::wstring::npos))
{
std::vector<std::wstring> vec = Tokenize(list, L"|");
std::vector<std::wstring>::size_type vecSize = vec.size();
std::wstring tmpFrom(path), tmpTo(path);
tmpTo += L"Backup\\";
if (_waccess(tmpTo.c_str(), 0) == -1) CreateDirectory(tmpTo.c_str(), NULL);
tmpTo += backupFolder;
CreateDirectory(tmpTo.c_str(), NULL);
for (unsigned int i = 0; i < vecSize; ++i)
{
if (vec[i][vec[i].length() - 1] == L'*')
{
vec[i].resize(vec[i].length() - 1); // Get rid of trailing asterisk
tmpFrom += vec[i];
if (!CopyFiles(tmpFrom.c_str(), tmpTo.c_str(), true))
{
std::wstring error = L"Unable to backup from:\n";
error += tmpFrom;
error += L"\n\nTo:\n";
error += tmpTo;
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return false;
}
tmpFrom.resize(tmpFrom.length() - vec[i].length());
}
}
}
return true;
}
void KeepVariables(const std::wstring& backupFolder, const std::wstring& skinsPath, const std::wstring& fileList)
{
WCHAR keyname[32767]; // Max size returned by GetPrivateProfileSection
WCHAR buffer[4];
std::wstring currKey, currValue;
std::vector<std::wstring> vec = Tokenize(fileList, L"|");
for (unsigned int i = 0, max = vec.size(); i < max; ++i)
{
std::wstring fromPath = skinsPath;
fromPath += L"Backup\\";
fromPath += backupFolder;
fromPath += L"\\";
fromPath += vec[i];
std::wstring toPath = skinsPath;
toPath += vec[i];
unsigned int count = GetPrivateProfileSection(L"Variables", keyname, 32767, fromPath.c_str());
if ((_waccess(fromPath.c_str(), 0) == 0) && (_waccess(toPath.c_str(), 0) == 0)
&& (count > 0))
{
for (unsigned int j = 0; j < count; ++j)
{
if (keyname[j] == L'=')
{
if (GetPrivateProfileString(L"Variables", currKey.c_str(), NULL, buffer, 4, toPath.c_str()) > 0)
{
while (keyname[++j] != L'\0') currValue += keyname[j];
WritePrivateProfileString(L"Variables", currKey.c_str(), currValue.c_str(), toPath.c_str());
currValue.clear();
}
else
{
while (keyname[j] != L'\0') ++j;
}
currKey.clear();
}
else
{
currKey += keyname[j];
}
}
}
}
}
void PreserveSetting(std::wstring& iniFile, std::wstring& backupFile, LPCTSTR section, LPCTSTR key)
{
WCHAR buffer[MAX_LINE_LENGTH];
if (GetPrivateProfileString(section, key, L"", buffer, MAX_LINE_LENGTH, iniFile.c_str()) == 0 &&
GetPrivateProfileString(section, key, L"", buffer, MAX_LINE_LENGTH, backupFile.c_str()) > 0)
{
WritePrivateProfileString(section, key, buffer, iniFile.c_str());
}
}
void LaunchRainmeter()
{
SetCurrentDirectory(data.rainmeterPath.c_str());
// Take a copy of current Rainmeter.ini before doing anything
std::wstring iniFile = data.iniPath;
iniFile += L"Rainmeter.ini";
std::wstring backupFile = data.iniPath;
backupFile += L"Themes\\Backup\\";
CreateDirectory(backupFile.c_str(), NULL);
backupFile += L"Rainmeter.thm";
CopyFiles(iniFile, backupFile, false);
if (!data.loadTheme.empty())
{
std::wstring themeFile = data.iniPath;
themeFile += L"Themes\\";
themeFile += data.loadTheme;
themeFile += L"\\Rainmeter.thm";
if (_waccess(themeFile.c_str(), 0) == 0)
{
CopyFiles(themeFile, iniFile, false);
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"SkinPath");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"ConfigEditor");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"LogViewer");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"Logging");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"DisableVersionCheck");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"Language");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteL");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteM");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteR");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteDL");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteDM");
PreserveSetting(iniFile, backupFile, L"Rainmeter", L"TrayExecuteDR");
}
}
// Execute and wait up to a minute for Rainmeter to process all messages
SHELLEXECUTEINFO sei = {0};
sei.cbSize = sizeof(SHELLEXECUTEINFO);
sei.fMask = SEE_MASK_WAITFORINPUTIDLE | SEE_MASK_UNICODE;
sei.hwnd = NULL;
sei.lpVerb = NULL;
sei.lpFile = L"Rainmeter.exe";
sei.lpParameters = NULL;
sei.nShow = SW_SHOWNORMAL;
ShellExecuteEx(&sei);
if (!data.loadSkins.empty())
{
std::wstring::size_type pos;
std::wstring bang;
std::vector<std::wstring> vec = Tokenize(data.loadSkins, L"|");
for (unsigned int i = 0, max = vec.size(); i < max; ++i)
{
pos = vec[i].find_last_of(L"\\");
if (pos != std::wstring::npos)
{
// Append with [!RainmeterActivateConfig "Config" "File.ini"]
bang += L"[!RainmeterActivateConfig \"";
bang += vec[i].substr(0, pos);
bang += L"\" \"";
bang += vec[i].substr(pos + 1);
bang += L"\"]";
}
}
if (!bang.empty())
{
bang.insert(0, L"!Execute ");
sei.fMask = SEE_MASK_UNICODE;
sei.lpParameters = (LPCTSTR)bang.c_str();
ShellExecuteEx(&sei);
}
}
}
/*
** Create elevated process (to install addons, fonts, and plugins)
*/
HANDLE CreateProcessElevated(HWND hwnd)
{
WCHAR secDesc[SECURITY_DESCRIPTOR_MIN_LENGTH];
SECURITY_ATTRIBUTES sa;
sa.nLength = sizeof(SECURITY_ATTRIBUTES);
sa.bInheritHandle = FALSE;
sa.lpSecurityDescriptor = &secDesc;
InitializeSecurityDescriptor(sa.lpSecurityDescriptor, SECURITY_DESCRIPTOR_REVISION);
SetSecurityDescriptorDacl(sa.lpSecurityDescriptor, TRUE, 0, FALSE);
// Create shared memory
hMapFile = CreateFileMapping(INVALID_HANDLE_VALUE, &sa, PAGE_READWRITE, 0, 2048, OBJECT_NAME);
if (hMapFile == NULL)
{
std::wstring error = L"Failed to create elevated process (unable to create file mapping).";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return NULL;
}
pBuffer = (LPCTSTR)MapViewOfFile(hMapFile, FILE_MAP_ALL_ACCESS, 0, 0, 2048);
if (pBuffer == NULL)
{
CloseHandle(hMapFile);
std::wstring error = L"Failed to create elevated process (unable to view file mapping).";
MessageBox(NULL, error.c_str(), APP_NAME, MB_ERROR);
return NULL;
}
WCHAR buffer[2048];
// Copy values that are to be passed to the elevated process into buffer
_snwprintf_s(buffer, _TRUNCATE, L"%s\"%i\"%s\"%s\"%s\"%s\"%s\"%i",
data.rmskinFile.c_str(),
data.rootLen,
data.rainmeterPath.c_str(),
data.backupFolder.empty() ? L" " : data.backupFolder.c_str(),
data.addonsList.empty() ? L" " : data.addonsList.c_str(),
data.pluginsList.empty() ? L" " : data.pluginsList.c_str(),
data.fontsList.empty() ? L" " : data.fontsList.c_str(),
data.rainmeterFonts ? 1 : 0);
CopyMemory((PVOID)pBuffer, buffer, sizeof(buffer));
// If UAC is not active, show custom Run As dialog. There is a bug in Vista/7, which makes elevation
// impossible when UAC is disabled
if (IsAboveVista() && !IsActiveUAC())
{
return (DialogBox(GetModuleHandle(NULL), MAKEINTRESOURCE(IDD_RUNAS), hwnd, (DLGPROC)RunAsProc) == 0) ? data.instHandle : NULL;
}
SHELLEXECUTEINFO sei = {0};
sei.cbSize = sizeof(SHELLEXECUTEINFO);
sei.fMask = SEE_MASK_NOCLOSEPROCESS;
sei.hwnd = NULL;
sei.lpVerb = L"runas";
sei.lpFile = exeFile.c_str();
sei.lpParameters = L"/ELEVATE";
sei.nShow = SW_SHOWNORMAL;
ShellExecuteEx(&sei);
if (!IsAboveVista()) _beginthreadex(NULL, 0, SetRunAsThread, NULL, 0, NULL);
return (sei.hProcess) ? sei.hProcess : NULL;
}
unsigned __stdcall SetRunAsThread(void*)
{
const UINT IDC_USRSAFER = 0x106, IDC_OTHERUSER = 0x104;
// Wait for up to 5 seconds for the Run As dialog to appear
for (int i = 0; i < 50; ++i)
{
Sleep(100);
HWND hwnd = FindWindow(L"#32770", L"Run As...");
if (hwnd)
{
SendMessage(GetDlgItem(hwnd, IDC_USRSAFER), BM_SETCHECK, BST_UNCHECKED, 0);
SendMessage(GetDlgItem(hwnd, IDC_OTHERUSER), BM_CLICK, 0, 0);
break;
}
}
return 0;
}
unsigned __stdcall CreateInstallThread(void* pParam)
{
return InstallComponents((RMSKIN_DATA*)pParam) ? 0 : 1;
}
/*
** Splits the string from the delimiters and trims whitespace
*/
std::vector<std::wstring> Tokenize(const std::wstring& str, const std::wstring& delimiters)
{
// Modified from http://www.digitalpeer.com/id/simple
std::vector<std::wstring> tokens;
std::wstring::size_type lastPos = str.find_first_not_of(delimiters, 0); // Skip delimiters at beginning
std::wstring::size_type pos = str.find_first_of(delimiters, lastPos); // Find first "non-delimiter"
while (std::wstring::npos != pos || std::wstring::npos != lastPos)
{
std::wstring tmpStr = str.substr(lastPos, pos - lastPos);
std::wstring::size_type tmpPos = tmpStr.find_first_not_of(L" \t");
if (tmpPos != std::wstring::npos)
{
tmpStr.erase(0, tmpPos);
tmpPos = tmpStr.find_last_not_of(L" \t");
if (tmpPos != std::wstring::npos)
{
tmpStr.resize(tmpPos + 1);
}
tokens.push_back(tmpStr);
}
else
{
tokens.push_back(L""); // Add empty string
}
lastPos = str.find_first_not_of(delimiters, pos); // Skip delimiters. Note the "not_of"
pos = str.find_first_of(delimiters, lastPos); // Find next "non-delimiter"
}
return tokens;
}
/*
** Compares two version strings. Returns 0 if equal, 1 if A > B and -1 if A < B.
*/
int CompareVersions(const std::wstring& strA, const std::wstring& strB)
{
if (strA.empty() && strB.empty()) return 0;
if (strA.empty()) return -1;
if (strB.empty()) return 1;
std::vector<std::wstring> arrayA = Tokenize(strA, L".");
std::vector<std::wstring> arrayB = Tokenize(strB, L".");
size_t len = max(arrayA.size(), arrayB.size());
for (size_t i = 0; i < len; ++i)
{
int a = 0;
int b = 0;
if (i < arrayA.size())
{
a = _wtoi(arrayA[i].c_str());
}
if (i < arrayB.size())
{
b = _wtoi(arrayB[i].c_str());
}
if (a > b) return 1;
if (a < b) return -1;
}
return 0;
}
std::wstring GetFileVersion(const std::wstring& file)
{
DWORD bufSize = GetFileVersionInfoSize(file.c_str(), 0);
void* versionInfo = new WCHAR[bufSize];
void* fileVersion = 0;
UINT valueSize;
std::wstring result;
if (GetFileVersionInfo(file.c_str(), 0, bufSize, versionInfo))
{
struct LANGANDCODEPAGE
{
WORD wLanguage;
WORD wCodePage;
} *languageInfo;
VerQueryValue(versionInfo, L"\\VarFileInfo\\Translation", (LPVOID*)&languageInfo, &valueSize);
WCHAR blockName[64];
_snwprintf_s(blockName, _TRUNCATE, L"\\StringFileInfo\\%04x%04x\\FileVersion", languageInfo[0].wLanguage, languageInfo[0].wCodePage);
VerQueryValue(versionInfo, blockName, &fileVersion, &valueSize);
if (valueSize)
{
result = (WCHAR*)fileVersion;
}
}
delete [] (WCHAR*)versionInfo;
return result;
}
std::wstring GetDotNETVersion()
{
WCHAR buffer[255];
HKEY hKey;
LONG lRet = RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"SOFTWARE\\Microsoft\\NET Framework Setup\\NDP", 0L, KEY_READ, &hKey);
std::wstring currVer(L"v0"), prevVer;
int i = 0;
while (lRet == ERROR_SUCCESS)
{
lRet = RegEnumKey(hKey, i, buffer, 255);
if (buffer[0] == L'v')
{
currVer = buffer;
}
++i;
}
RegCloseKey(hKey);
currVer.erase(0, 1); // Get rid of the 'v'
return currVer;
}
std::wstring GetWindowsVersion()
{
WCHAR buffer[16];
OSVERSIONINFOEX osvi = {sizeof(OSVERSIONINFOEX)};
GetVersionEx((OSVERSIONINFO*)&osvi);
_snwprintf_s(buffer, _TRUNCATE, L"%d.%d.%d", osvi.dwMajorVersion, osvi.dwMinorVersion, osvi.dwBuildNumber);
return buffer;
}
BOOL IsActiveUAC()
{
// First check if user has a split token (that implies UAC)
HANDLE hToken;
if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
{
TOKEN_ELEVATION_TYPE tet;
DWORD dwReturnLength = 0;
if (GetTokenInformation(hToken, TokenElevationType, &tet, sizeof(TOKEN_ELEVATION_TYPE), &dwReturnLength) &&
tet != TokenElevationTypeDefault)
{
return TRUE;
}
CloseHandle(hToken);
}
// Check from registry
DWORD dwValue = 0;
DWORD dwSize = sizeof(DWORD);
HKEY hKey;
RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Policies\\System", 0L, KEY_QUERY_VALUE, &hKey);
RegQueryValueEx(hKey, L"EnableLUA", NULL, NULL, (LPBYTE)&dwValue, (LPDWORD)&dwSize);
RegCloseKey(hKey);
return (dwValue != 0) ? TRUE : FALSE;
}
BOOL IsAboveVista()
{
static BOOL isAbove = -1;
if (isAbove == -1)
{
OSVERSIONINFOEX osvi = {sizeof(OSVERSIONINFOEX)};
GetVersionEx((OSVERSIONINFO*)&osvi);
isAbove = (osvi.dwMajorVersion >= 6) ? TRUE : FALSE;
}
return isAbove;
}
BOOL IsCurrentProcessAdmin()
{
static BOOL isAdmin = -1;
// Based on the work of Anders Kjersem: http://nsis.sourceforge.net/UAC_plug-in
if (isAdmin == -1)
{
isAdmin = FALSE;
HANDLE hToken;
if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken))
{
SID_IDENTIFIER_AUTHORITY SystemSidAuthority = SECURITY_NT_AUTHORITY;
PSID psid = 0;
if (AllocateAndInitializeSid(&SystemSidAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &psid))
{
if (!CheckTokenMembership(0, psid, &isAdmin))
{
isAdmin = FALSE;
DWORD cbTokenGrps;
if (!GetTokenInformation(hToken, TokenGroups, 0, 0, &cbTokenGrps) && GetLastError() == ERROR_INSUFFICIENT_BUFFER)
{
TOKEN_GROUPS* ptg = (TOKEN_GROUPS*)GlobalAlloc(LPTR, cbTokenGrps);
if (ptg)
{
if (GetTokenInformation(hToken, TokenGroups, ptg, cbTokenGrps, &cbTokenGrps))
{
for (DWORD i = 0; i < ptg->GroupCount; ++i)
{
if (EqualSid(ptg->Groups[i].Sid, psid))
{
isAdmin = TRUE;
break;
}
}
}
GlobalFree(ptg);
}
}
}
FreeSid(psid);
}
// Check if UAC admin with split token check
if (isAdmin && IsAboveVista())
{
TOKEN_ELEVATION_TYPE tet;
DWORD dwReturnLength = 0;
if (GetTokenInformation(hToken, TokenElevationType, &tet, sizeof(TOKEN_ELEVATION_TYPE), &dwReturnLength) &&
tet == TokenElevationTypeLimited)
{
isAdmin = FALSE;
}
}
CloseHandle(hToken);
}
}
return isAdmin;
}
Reference in New Issue View Git Blame Copy Permalink