336 lines
8.2 KiB
C++
336 lines
8.2 KiB
C++
/*
|
|
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include "StdAfx.h"
|
|
#include "DialogPackage.h"
|
|
#include "DialogInstall.h"
|
|
#include "resource.h"
|
|
#include "SkinInstaller.h"
|
|
|
|
EXTERN_C IMAGE_DOS_HEADER __ImageBase;
|
|
|
|
GlobalData g_Data;
|
|
|
|
OsNameVersion g_OsNameVersions[] =
|
|
{
|
|
{ L"XP", L"5.1" },
|
|
{ L"Vista", L"6.0" },
|
|
{ L"7", L"6.1" },
|
|
// { L"8", L"6.2" }
|
|
};
|
|
|
|
/*
|
|
** Entry point
|
|
**
|
|
*/
|
|
int SkinInstallerMain(LPWSTR lpCmdLine)
|
|
{
|
|
// Avoid loading a dll from current directory
|
|
SetDllDirectory(L"");
|
|
|
|
CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE);
|
|
InitCommonControls();
|
|
|
|
if (lpCmdLine[0] == L'"')
|
|
{
|
|
// Strip quotes
|
|
++lpCmdLine;
|
|
WCHAR* pos = wcsrchr(lpCmdLine, L'"');
|
|
if (pos)
|
|
{
|
|
*pos = L'\0';
|
|
}
|
|
}
|
|
|
|
WCHAR buffer[MAX_PATH];
|
|
GetModuleFileName(GetInstanceHandle(), buffer, MAX_PATH);
|
|
|
|
// Remove the module's name from the path
|
|
WCHAR* pos = wcsrchr(buffer, L'\\');
|
|
if (pos)
|
|
{
|
|
*(pos + 1) = L'\0';
|
|
}
|
|
|
|
g_Data.programPath = g_Data.settingsPath = buffer;
|
|
wcscat(buffer, L"Rainmeter.ini");
|
|
|
|
// Find the settings file and read skins path off it
|
|
if (_waccess(buffer, 0) == 0)
|
|
{
|
|
g_Data.iniFile = buffer;
|
|
if (GetPrivateProfileString(L"Rainmeter", L"SkinPath", L"", buffer, MAX_LINE_LENGTH, buffer) > 0)
|
|
{
|
|
g_Data.skinsPath = buffer;
|
|
if (g_Data.skinsPath.back() != L'\\' && g_Data.skinsPath.back() != L'/')
|
|
{
|
|
g_Data.skinsPath += L'\\';
|
|
}
|
|
}
|
|
else
|
|
{
|
|
g_Data.skinsPath = g_Data.programPath;
|
|
g_Data.skinsPath += L"Skins\\";
|
|
}
|
|
}
|
|
else
|
|
{
|
|
HRESULT hr = SHGetFolderPath(nullptr, CSIDL_APPDATA, nullptr, SHGFP_TYPE_CURRENT, buffer);
|
|
wcscat(buffer, L"\\Rainmeter\\");
|
|
g_Data.settingsPath = buffer;
|
|
wcscat(buffer, L"Rainmeter.ini");
|
|
g_Data.iniFile = buffer;
|
|
if (SUCCEEDED(hr) && _waccess(buffer, 0) == 0)
|
|
{
|
|
if (GetPrivateProfileString(L"Rainmeter", L"SkinPath", L"", buffer, MAX_LINE_LENGTH, buffer) > 0)
|
|
{
|
|
g_Data.skinsPath = buffer;
|
|
if (g_Data.skinsPath.back() != L'\\' && g_Data.skinsPath.back() != L'/')
|
|
{
|
|
g_Data.skinsPath += L'\\';
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::wstring error = L"SkinPath not found.\nMake sure that Rainmeter has been run at least once.";
|
|
MessageBox(nullptr, error.c_str(), L"Rainmeter Skin Installer", MB_ERROR);
|
|
return 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::wstring error = L"Rainmeter.ini not found.\nMake sure that Rainmeter has been run at least once.";
|
|
MessageBox(nullptr, error.c_str(), L"Rainmeter Skin Installer", MB_ERROR);
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
std::wstring layoutsPath = g_Data.settingsPath + L"Layouts\\";
|
|
if (_waccess(layoutsPath.c_str(), 0) == -1)
|
|
{
|
|
// Migrate Themes into Layouts for backwards compatibility and rename
|
|
// Rainmeter.thm to Rainmeter.ini and RainThemes.bmp to Wallpaper.bmp.
|
|
std::wstring themesPath = g_Data.settingsPath + L"Themes";
|
|
if (_waccess(themesPath.c_str(), 0) != -1)
|
|
{
|
|
// Migrate Themes into Layouts for backwards compatibility and rename
|
|
// Rainmeter.thm to Rainmeter.ini and RainThemes.bmp to Wallpaper.bmp.
|
|
MoveFile(themesPath.c_str(), layoutsPath.c_str());
|
|
|
|
layoutsPath += L'*'; // For FindFirstFile.
|
|
WIN32_FIND_DATA fd;
|
|
HANDLE hFind = FindFirstFile(layoutsPath.c_str(), &fd);
|
|
layoutsPath.pop_back(); // Remove '*'.
|
|
|
|
if (hFind != INVALID_HANDLE_VALUE)
|
|
{
|
|
do
|
|
{
|
|
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY &&
|
|
wcscmp(L".", fd.cFileName) != 0 &&
|
|
wcscmp(L"..", fd.cFileName) != 0)
|
|
{
|
|
std::wstring layoutFolder = layoutsPath + fd.cFileName;
|
|
layoutFolder += L'\\';
|
|
|
|
std::wstring file = layoutFolder + L"Rainmeter.thm";
|
|
if (_waccess(file.c_str(), 0) != -1)
|
|
{
|
|
std::wstring newFile = layoutFolder + L"Rainmeter.ini";
|
|
MoveFile(file.c_str(), newFile.c_str());
|
|
}
|
|
|
|
file = layoutFolder + L"RainThemes.bmp";
|
|
if (_waccess(file.c_str(), 0) != -1)
|
|
{
|
|
std::wstring newFile = layoutFolder + L"Wallpaper.bmp";
|
|
MoveFile(file.c_str(), newFile.c_str());
|
|
}
|
|
}
|
|
}
|
|
while (FindNextFile(hFind, &fd));
|
|
|
|
FindClose(hFind);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (_wcsnicmp(lpCmdLine, L"/LoadTheme ", 11) == 0)
|
|
{
|
|
// For backwards compatibility.
|
|
std::wstring args = L"!LoadLayout \"";
|
|
args += &lpCmdLine[11]; // Skip "/LoadTheme ".
|
|
args += L'"';
|
|
|
|
std::wstring file = g_Data.programPath + L"Rainmeter.exe";
|
|
SHELLEXECUTEINFO sei = {0};
|
|
sei.cbSize = sizeof(SHELLEXECUTEINFO);
|
|
sei.fMask = SEE_MASK_UNICODE;
|
|
sei.lpFile = file.c_str();
|
|
sei.lpParameters = args.c_str();
|
|
sei.lpDirectory = g_Data.programPath.c_str();
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
ShellExecuteEx(&sei);
|
|
|
|
return 0;
|
|
}
|
|
else if (wcscmp(lpCmdLine, L"/Packager") == 0)
|
|
{
|
|
DialogPackage::Create(GetInstanceHandle(), lpCmdLine);
|
|
}
|
|
else
|
|
{
|
|
DialogInstall::Create(GetInstanceHandle(), lpCmdLine);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
bool CloseRainmeterIfActive()
|
|
{
|
|
// Close Rainmeter.exe
|
|
HWND hwnd = FindWindow(L"DummyRainWClass", L"Rainmeter control window");
|
|
if (hwnd)
|
|
{
|
|
DWORD pID, exitCode;
|
|
GetWindowThreadProcessId(hwnd, &pID);
|
|
HANDLE hProcess = OpenProcess(PROCESS_TERMINATE | SYNCHRONIZE, FALSE, pID);
|
|
PostMessage(hwnd, WM_DESTROY, 0, 0);
|
|
|
|
// Wait up to 5 seconds for Rainmeter to close
|
|
WaitForSingleObject(hProcess, 5000);
|
|
GetExitCodeProcess(hProcess, &exitCode);
|
|
CloseHandle(hProcess);
|
|
|
|
if (exitCode == STILL_ACTIVE)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
HINSTANCE GetInstanceHandle()
|
|
{
|
|
return (HINSTANCE)&__ImageBase;
|
|
}
|
|
|
|
// -----------------------------------------------------------------------------------------------
|
|
// Stolen functions from Rainmeter Litestep.cpp, System.cpp, and Application.cpp
|
|
// -----------------------------------------------------------------------------------------------
|
|
|
|
bool IsRunning(const WCHAR* name, HANDLE* hMutex)
|
|
{
|
|
// Create mutex
|
|
HANDLE hMutexTmp = CreateMutex(nullptr, FALSE, name);
|
|
if (GetLastError() == ERROR_ALREADY_EXISTS)
|
|
{
|
|
*hMutex = nullptr;
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
*hMutex = hMutexTmp;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool CopyFiles(const std::wstring& strFrom, const std::wstring& strTo, bool bMove)
|
|
{
|
|
std::wstring tmpFrom(strFrom), tmpTo(strTo);
|
|
|
|
// The strings must end with double nul
|
|
tmpFrom.append(1, L'\0');
|
|
tmpTo.append(1, L'\0');
|
|
|
|
SHFILEOPSTRUCT fo =
|
|
{
|
|
nullptr,
|
|
bMove ? FO_MOVE : FO_COPY,
|
|
tmpFrom.c_str(),
|
|
tmpTo.c_str(),
|
|
FOF_NO_UI | FOF_NOCONFIRMATION | FOF_ALLOWUNDO
|
|
};
|
|
|
|
return SHFileOperation(&fo) == 0;
|
|
}
|
|
|
|
OSPLATFORM GetOSPlatform()
|
|
{
|
|
OSVERSIONINFOEX osvi = {sizeof(OSVERSIONINFOEX)};
|
|
if (GetVersionEx((OSVERSIONINFO*)&osvi))
|
|
{
|
|
if (osvi.dwMajorVersion == 5)
|
|
{
|
|
// Not checking for osvi.dwMinorVersion >= 1 because Rainmeter won't run on pre-XP
|
|
return OSPLATFORM_XP;
|
|
}
|
|
else if (osvi.dwMajorVersion == 6)
|
|
{
|
|
if (osvi.dwMinorVersion == 0)
|
|
{
|
|
return OSPLATFORM_VISTA; // Vista, Server 2008
|
|
}
|
|
else
|
|
{
|
|
return OSPLATFORM_7; // 7, Server 2008R2
|
|
}
|
|
}
|
|
else // newer OS
|
|
{
|
|
return OSPLATFORM_7;
|
|
}
|
|
}
|
|
|
|
return OSPLATFORM_UNKNOWN;
|
|
}
|
|
|
|
std::string ConvertToAscii(LPCTSTR str)
|
|
{
|
|
std::string szAscii;
|
|
|
|
if (str && *str)
|
|
{
|
|
int strLen = (int)wcslen(str);
|
|
int bufLen = WideCharToMultiByte(CP_ACP, 0, str, strLen, nullptr, 0, nullptr, nullptr);
|
|
if (bufLen > 0)
|
|
{
|
|
szAscii.resize(bufLen);
|
|
WideCharToMultiByte(CP_ACP, 0, str, strLen, &szAscii[0], bufLen, nullptr, nullptr);
|
|
}
|
|
}
|
|
return szAscii;
|
|
}
|
|
|
|
std::wstring ConvertToWide(LPCSTR str)
|
|
{
|
|
std::wstring szWide;
|
|
|
|
if (str && *str)
|
|
{
|
|
int strLen = (int)strlen(str);
|
|
int bufLen = MultiByteToWideChar(CP_ACP, 0, str, strLen, nullptr, 0);
|
|
if (bufLen > 0)
|
|
{
|
|
szWide.resize(bufLen);
|
|
MultiByteToWideChar(CP_ACP, 0, str, strLen, &szWide[0], bufLen);
|
|
}
|
|
}
|
|
return szWide;
|
|
} |