mirror of
https://github.com/Open-Shell/Open-Shell-Menu.git
synced 2026-04-12 18:07:24 +10:00
Initial commit
This commit is contained in:
@@ -0,0 +1,90 @@
|
||||
// Classic Shell (c) 2009-2016, Ivo Beltchev
|
||||
// Confidential information of Ivo Beltchev. Not for disclosure or distribution without prior written consent from the author
|
||||
|
||||
#define STRICT_TYPED_ITEMIDS
|
||||
#define _ATL_CSTRING_EXPLICIT_CONSTRUCTORS // some CString constructors will be explicit
|
||||
#include <windows.h>
|
||||
#include <atlstr.h>
|
||||
#include "ResourceHelper.h"
|
||||
|
||||
HINSTANCE g_hInstance;
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
int ExitStartMenu( void )
|
||||
{
|
||||
HKEY hKey=NULL;
|
||||
if (RegCreateKeyEx(HKEY_LOCAL_MACHINE,L"SOFTWARE\\IvoSoft\\ClassicShell",0,NULL,REG_OPTION_NON_VOLATILE,KEY_READ|KEY_QUERY_VALUE|KEY_WOW64_64KEY,NULL,&hKey,NULL)==ERROR_SUCCESS)
|
||||
{
|
||||
DWORD type=0;
|
||||
wchar_t path[_MAX_PATH];
|
||||
DWORD size=sizeof(path);
|
||||
if (RegQueryValueEx(hKey,L"Path",0,&type,(BYTE*)path,&size)==ERROR_SUCCESS && type==REG_SZ)
|
||||
{
|
||||
STARTUPINFO startupInfo={sizeof(startupInfo)};
|
||||
PROCESS_INFORMATION processInfo;
|
||||
memset(&processInfo,0,sizeof(processInfo));
|
||||
wcscat_s(path,L"ClassicStartMenu.exe");
|
||||
HANDLE h=CreateFile(path,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
|
||||
if (h!=INVALID_HANDLE_VALUE)
|
||||
{
|
||||
CloseHandle(h);
|
||||
wcscat_s(path,L" -exit");
|
||||
if (CreateProcess(NULL,path,NULL,NULL,TRUE,0,NULL,NULL,&startupInfo,&processInfo))
|
||||
{
|
||||
CloseHandle(processInfo.hThread);
|
||||
WaitForSingleObject(processInfo.hProcess,5000);
|
||||
CloseHandle(processInfo.hProcess);
|
||||
}
|
||||
}
|
||||
}
|
||||
RegCloseKey(hKey);
|
||||
}
|
||||
HWND updateOwner=FindWindow(L"ClassicShellUpdate.COwnerWindow",NULL);
|
||||
if (updateOwner)
|
||||
PostMessage(updateOwner,WM_CLEAR,0,0);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int FixVersion( void )
|
||||
{
|
||||
HKEY hKey=NULL;
|
||||
if (RegCreateKeyEx(HKEY_LOCAL_MACHINE,L"SOFTWARE\\IvoSoft\\ClassicShell",0,NULL,REG_OPTION_NON_VOLATILE,KEY_READ|KEY_WRITE|KEY_WOW64_64KEY,NULL,&hKey,NULL)==ERROR_SUCCESS)
|
||||
{
|
||||
DWORD winVer=GetVersionEx(GetModuleHandle(L"user32.dll"));
|
||||
RegSetValueEx(hKey,L"WinVersion",NULL,REG_DWORD,(BYTE*)&winVer,sizeof(DWORD));
|
||||
RegCloseKey(hKey);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
// Setup Helper - performs custom actions during Classic Shell install/uninstall
|
||||
// Usage:
|
||||
// exitSM // exits the start menu if it is running
|
||||
// fixVersion // sets the correct OS version
|
||||
|
||||
int WINAPI wWinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpstrCmdLine, int nCmdShow )
|
||||
{
|
||||
// MessageBox(NULL,lpstrCmdLine,L"Command Line",MB_OK|MB_SYSTEMMODAL);
|
||||
|
||||
int count;
|
||||
wchar_t *const *params=CommandLineToArgvW(lpstrCmdLine,&count);
|
||||
if (!params) return 1;
|
||||
|
||||
g_hInstance=hInstance;
|
||||
|
||||
for (;count>0;count--,params++)
|
||||
{
|
||||
if (_wcsicmp(params[0],L"exitSM")==0)
|
||||
{
|
||||
return ExitStartMenu();
|
||||
}
|
||||
if (_wcsicmp(params[0],L"fixVersion")==0)
|
||||
{
|
||||
return FixVersion();
|
||||
}
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
Reference in New Issue
Block a user