mirror of
https://github.com/chibicitiberiu/rainmeter-studio.git
synced 2024-02-24 04:33:31 +00:00
Modified the behavior of starting multiple processes.
This commit is contained in:
parent
10257297bc
commit
cce8a5ead4
@ -21,6 +21,8 @@
|
|||||||
#define _CRTDBG_MAP_ALLOC
|
#define _CRTDBG_MAP_ALLOC
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <crtdbg.h>
|
#include <crtdbg.h>
|
||||||
|
#include <string>
|
||||||
|
#include <algorithm>
|
||||||
#include "resource.h"
|
#include "resource.h"
|
||||||
#include "..\Library\Rainmeter.h"
|
#include "..\Library\Rainmeter.h"
|
||||||
|
|
||||||
@ -41,6 +43,7 @@ BOOL InitApplication(HINSTANCE hInstance, const WCHAR* WinClass);
|
|||||||
HWND InitInstance(HINSTANCE hInstance, const WCHAR* WinClass, const WCHAR* WinName);
|
HWND InitInstance(HINSTANCE hInstance, const WCHAR* WinClass, const WCHAR* WinName);
|
||||||
LRESULT CALLBACK MainWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
|
LRESULT CALLBACK MainWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
|
||||||
void Bang(const WCHAR* command);
|
void Bang(const WCHAR* command);
|
||||||
|
BOOL IsRunning(HANDLE* hMutex);
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Stuff from the DLL
|
** Stuff from the DLL
|
||||||
@ -61,6 +64,7 @@ const WCHAR* WinName = L"Rainmeter control window";
|
|||||||
*/
|
*/
|
||||||
int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow)
|
int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nCmdShow)
|
||||||
{
|
{
|
||||||
|
HANDLE hMutex = NULL;
|
||||||
MSG msg;
|
MSG msg;
|
||||||
HWND hWnd;
|
HWND hWnd;
|
||||||
|
|
||||||
@ -74,6 +78,13 @@ int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmd
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Check whether Rainmeter.exe is already running
|
||||||
|
if (IsRunning(&hMutex))
|
||||||
|
{
|
||||||
|
//MessageBox(NULL, L"Rainmeter.exe is already running.", L"Rainmeter", MB_ICONWARNING | MB_TOPMOST | MB_OK);
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
if(!hPrevInstance)
|
if(!hPrevInstance)
|
||||||
{
|
{
|
||||||
if (!InitApplication(hInstance, WinClass)) return FALSE;
|
if (!InitApplication(hInstance, WinClass)) return FALSE;
|
||||||
@ -115,6 +126,7 @@ int APIENTRY wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmd
|
|||||||
DispatchMessage(&msg);
|
DispatchMessage(&msg);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (hMutex) ReleaseMutex(hMutex);
|
||||||
return (int)msg.wParam;
|
return (int)msg.wParam;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -197,6 +209,82 @@ void Bang(const WCHAR* command)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
** IsRunning
|
||||||
|
**
|
||||||
|
** Checks whether Rainmeter.exe is running.
|
||||||
|
**
|
||||||
|
*/
|
||||||
|
BOOL IsRunning(HANDLE* hMutex)
|
||||||
|
{
|
||||||
|
typedef struct
|
||||||
|
{
|
||||||
|
ULONG i[2];
|
||||||
|
ULONG buf[4];
|
||||||
|
unsigned char in[64];
|
||||||
|
unsigned char digest[16];
|
||||||
|
} MD5_CTX;
|
||||||
|
|
||||||
|
typedef void (WINAPI *FPMD5INIT)(MD5_CTX *context);
|
||||||
|
typedef void (WINAPI *FPMD5UPDATE)(MD5_CTX *context, const unsigned char *input, unsigned int inlen);
|
||||||
|
typedef void (WINAPI *FPMD5FINAL)(MD5_CTX *context);
|
||||||
|
|
||||||
|
// Create MD5 digest from command line
|
||||||
|
HMODULE hCryptDll = LoadLibrary(L"cryptdll.dll");
|
||||||
|
if (!hCryptDll) // Unable to check the mutex
|
||||||
|
{
|
||||||
|
*hMutex = NULL;
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
FPMD5INIT MD5Init = (FPMD5INIT)GetProcAddress(hCryptDll, "MD5Init");
|
||||||
|
FPMD5UPDATE MD5Update = (FPMD5UPDATE)GetProcAddress(hCryptDll, "MD5Update");
|
||||||
|
FPMD5FINAL MD5Final = (FPMD5FINAL)GetProcAddress(hCryptDll, "MD5Final");
|
||||||
|
if (!MD5Init || !MD5Update || !MD5Final) // Unable to check the mutex
|
||||||
|
{
|
||||||
|
FreeLibrary(hCryptDll);
|
||||||
|
*hMutex = NULL;
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::wstring cmdLine = GetCommandLine();
|
||||||
|
std::transform(cmdLine.begin(), cmdLine.end(), cmdLine.begin(), ::towlower);
|
||||||
|
|
||||||
|
MD5_CTX ctx = {0};
|
||||||
|
|
||||||
|
MD5Init(&ctx);
|
||||||
|
MD5Update(&ctx, (LPBYTE)cmdLine.c_str(), cmdLine.length() * sizeof(WCHAR));
|
||||||
|
MD5Final(&ctx);
|
||||||
|
|
||||||
|
FreeLibrary(hCryptDll);
|
||||||
|
|
||||||
|
// Convert MD5 digest to mutex string (e.g. "Rainmeter@0123456789abcdef0123456789abcdef")
|
||||||
|
const WCHAR szHexTable[] = L"0123456789abcdef";
|
||||||
|
WCHAR szMutex[64] = {0};
|
||||||
|
wcscpy(szMutex, L"Rainmeter@");
|
||||||
|
|
||||||
|
WCHAR* pos = szMutex + wcslen(szMutex);
|
||||||
|
|
||||||
|
for (size_t i = 0; i < 16; ++i)
|
||||||
|
{
|
||||||
|
*(pos++) = *(szHexTable + ((ctx.digest[i] >> 4) & 0xF));
|
||||||
|
*(pos++) = *(szHexTable + (ctx.digest[i] & 0xF));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create mutex
|
||||||
|
HANDLE hMutexTemp = CreateMutex(NULL, FALSE, szMutex);
|
||||||
|
if (GetLastError() == ERROR_ALREADY_EXISTS)
|
||||||
|
{
|
||||||
|
// Rainmeter.exe is already running
|
||||||
|
*hMutex = NULL;
|
||||||
|
return TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Rainmeter.exe is not running
|
||||||
|
*hMutex = hMutexTemp;
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** MainWndProc
|
** MainWndProc
|
||||||
**
|
**
|
||||||
|
Loading…
x
Reference in New Issue
Block a user