mirror of
https://github.com/chibicitiberiu/rainmeter-studio.git
synced 2024-02-24 04:33:31 +00:00
144 lines
3.7 KiB
C++
144 lines
3.7 KiB
C++
/*
|
|
Copyright (C) 2001 Kimmo Pekkola
|
|
|
|
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.
|
|
*/
|
|
|
|
#define _CRTDBG_MAP_ALLOC
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <crtdbg.h>
|
|
#include <Windows.h>
|
|
#include <ShellAPI.h>
|
|
#include <delayimp.h>
|
|
|
|
EXTERN_C IMAGE_DOS_HEADER __ImageBase;
|
|
|
|
EXTERN_C __declspec(dllimport) int RainmeterMain(LPWSTR cmdLine);
|
|
|
|
WCHAR* GetCommandLineArguments()
|
|
{
|
|
WCHAR* args = GetCommandLine();
|
|
|
|
// Skip past (quoted) application path in cmdLine.
|
|
if (*args == L'"')
|
|
{
|
|
++args; // Skip leading quote.
|
|
while (*args && *args != L'"')
|
|
{
|
|
++args;
|
|
}
|
|
++args; // Skip trailing quote.
|
|
}
|
|
else
|
|
{
|
|
while (*args && *args != L' ')
|
|
{
|
|
++args;
|
|
}
|
|
}
|
|
|
|
// Skip leading whitespace (similar to CRT implementation).
|
|
while (*args && *args <= L' ')
|
|
{
|
|
++args;
|
|
}
|
|
|
|
return args;
|
|
}
|
|
|
|
/*
|
|
** Hook to exit the process gracefully if delay-loading dependencies (i.e. Rainmeter.dll) fails.
|
|
**
|
|
*/
|
|
FARPROC WINAPI DelayLoadFailureHook(unsigned int dliNotify, DelayLoadInfo* dli)
|
|
{
|
|
if (dliNotify == dliFailLoadLib)
|
|
{
|
|
WCHAR buffer[128];
|
|
int arch = 32;
|
|
#ifdef _WIN64
|
|
arch = 64;
|
|
#endif
|
|
const WCHAR* format = L"%S (%i-bit) error %ld.\n\nDo you want to view help online?";
|
|
wsprintf(buffer, format, dli->szDll, arch, dli->dwLastError);
|
|
if (MessageBox(nullptr, buffer, L"Rainmeter", MB_YESNO | MB_ICONERROR) == IDYES)
|
|
{
|
|
ShellExecute(nullptr, L"open", L"http://rainmeter.net/dllerror", nullptr, nullptr, SW_SHOWNORMAL);
|
|
}
|
|
|
|
ExitProcess(0);
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
EXTERN_C PfnDliHook __pfnDliFailureHook2 = DelayLoadFailureHook;
|
|
|
|
/*
|
|
** Entry point. In Release builds, the entry point is Main() since the CRT is not used.
|
|
**
|
|
*/
|
|
int APIENTRY wWinMain(HINSTANCE, HINSTANCE, LPWSTR, int)
|
|
{
|
|
_CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
|
|
//_CrtSetBreakAlloc(000);
|
|
|
|
// Prevent system error message boxes.
|
|
UINT oldMode = SetErrorMode(0);
|
|
SetErrorMode(oldMode | SEM_FAILCRITICALERRORS);
|
|
|
|
HINSTANCE instance = (HINSTANCE)&__ImageBase;
|
|
WCHAR* args = GetCommandLineArguments();
|
|
|
|
HRSRC iconResource = FindResource(instance, MAKEINTRESOURCE(1), RT_ICON);
|
|
if (iconResource)
|
|
{
|
|
return RainmeterMain(args);
|
|
}
|
|
else
|
|
{
|
|
// Stub prodecure. If icon resources have been removed, try to launch the actual Rainmeter.exe.
|
|
HKEY hKey;
|
|
const REGSAM desiredSam = KEY_QUERY_VALUE | KEY_WOW64_32KEY;
|
|
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Rainmeter", 0, desiredSam, &hKey) == ERROR_SUCCESS)
|
|
{
|
|
const DWORD size = MAX_PATH;
|
|
WCHAR buffer[size];
|
|
DWORD type = 0;
|
|
if (RegQueryValueEx(hKey, nullptr , nullptr, &type, (LPBYTE)buffer, (LPDWORD)&size) == ERROR_SUCCESS &&
|
|
type == REG_SZ)
|
|
{
|
|
SetCurrentDirectory(buffer);
|
|
lstrcat(buffer, L"\\Rainmeter.exe");
|
|
ShellExecute(nullptr, L"open", buffer, args, nullptr, SW_SHOWNORMAL);
|
|
}
|
|
RegCloseKey(hKey);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
#ifndef _DEBUG
|
|
EXTERN_C int WINAPI Main()
|
|
{
|
|
int result = wWinMain(nullptr, nullptr, nullptr, 0);
|
|
ExitProcess(result);
|
|
return 0; // Never reached.
|
|
}
|
|
#endif
|