mirror of
https://github.com/reactos/reactos.git
synced 2025-06-13 13:58:30 +00:00
[CTFMON][SDK] Add ctfmon.exe (#6149)
ctfmon.exe will be a replacement of our kbswitch.exe in the future. That is the front-end of Language Bar. It is needed to support TIPs. - Add ctfmon.exe at base/applications/ctfmon. - Add <cicero/cicbase.h>, <cicero/CModulePath.h>, and <cicero/osinfo.h> headers and use them. CORE-19362
This commit is contained in:
Katayama Hirofumi MZ
GitHub
parent
a19ba4760e
commit
079b36542c
14 changed files with 1156 additions and 0 deletions
378
base/applications/ctfmon/ctfmon.cpp
Normal file
378
base/applications/ctfmon/ctfmon.cpp
Normal file
|
|
@ -0,0 +1,378 @@
|
|||
/*
|
||||
* PROJECT: ReactOS CTF Monitor
|
||||
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
|
||||
* PURPOSE: Providing Language Bar front-end
|
||||
* COPYRIGHT: Copyright 2023 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
|
||||
*/
|
||||
|
||||
#include "precomp.h"
|
||||
#include "CRegWatcher.h"
|
||||
#include "CLoaderWnd.h"
|
||||
|
||||
// ntdll!NtQueryInformationProcess
|
||||
typedef NTSTATUS (WINAPI *FN_NtQueryInformationProcess)(HANDLE, PROCESSINFOCLASS, PVOID, ULONG, PULONG);
|
||||
FN_NtQueryInformationProcess g_fnNtQueryInformationProcess = NULL;
|
||||
|
||||
// kernel32!SetProcessShutdownParameters
|
||||
typedef BOOL (WINAPI *FN_SetProcessShutdownParameters)(DWORD, DWORD);
|
||||
FN_SetProcessShutdownParameters g_fnSetProcessShutdownParameters = NULL;
|
||||
|
||||
// kernel32!GetSystemWow64DirectoryW
|
||||
typedef UINT (WINAPI *FN_GetSystemWow64DirectoryW)(LPWSTR, UINT);
|
||||
FN_GetSystemWow64DirectoryW g_fnGetSystemWow64DirectoryW = NULL;
|
||||
|
||||
HINSTANCE g_hInst = NULL; // The application instance
|
||||
HINSTANCE g_hKernel32 = NULL; // The "kernel32.dll" instance
|
||||
UINT g_uACP = CP_ACP; // The active codepage
|
||||
BOOL g_fWinLogon = FALSE; // Is it a log-on process?
|
||||
HANDLE g_hCicMutex = NULL; // The Cicero mutex
|
||||
BOOL g_bOnWow64 = FALSE; // Is the app running on WoW64?
|
||||
BOOL g_fNoRunKey = FALSE; // Don't write registry key "Run"?
|
||||
BOOL g_fJustRunKey = FALSE; // Just write registry key "Run"?
|
||||
DWORD g_dwOsInfo = 0; // The OS version info. See GetOSInfo
|
||||
CLoaderWnd* g_pLoaderWnd = NULL; // TIP Bar loader window
|
||||
|
||||
// Is the current process on WoW64?
|
||||
static BOOL
|
||||
IsWow64(VOID)
|
||||
{
|
||||
HMODULE hNTDLL = GetSystemModuleHandle(L"ntdll.dll", FALSE);
|
||||
if (!hNTDLL)
|
||||
return FALSE;
|
||||
|
||||
g_fnNtQueryInformationProcess =
|
||||
(FN_NtQueryInformationProcess)::GetProcAddress(hNTDLL, "NtQueryInformationProcess");
|
||||
if (!g_fnNtQueryInformationProcess)
|
||||
return FALSE;
|
||||
|
||||
ULONG_PTR Value = 0;
|
||||
NTSTATUS Status = g_fnNtQueryInformationProcess(::GetCurrentProcess(), ProcessWow64Information,
|
||||
&Value, sizeof(Value), NULL);
|
||||
if (!NT_SUCCESS(Status))
|
||||
return FALSE;
|
||||
|
||||
return !!Value;
|
||||
}
|
||||
|
||||
static VOID
|
||||
ParseCommandLine(
|
||||
_In_ LPCWSTR pszCmdLine)
|
||||
{
|
||||
g_fNoRunKey = g_fJustRunKey = FALSE;
|
||||
|
||||
for (LPCWSTR pch = pszCmdLine; *pch; ++pch)
|
||||
{
|
||||
// Skip space
|
||||
while (*pch == L' ')
|
||||
++pch;
|
||||
|
||||
if (*pch == UNICODE_NULL)
|
||||
return;
|
||||
|
||||
if ((*pch == L'-') || (*pch == L'/'))
|
||||
{
|
||||
++pch;
|
||||
switch (*pch)
|
||||
{
|
||||
case L'N': case L'n': // Found "/N" option
|
||||
g_fNoRunKey = TRUE;
|
||||
break;
|
||||
|
||||
case L'R': case L'r': // Found "/R" option
|
||||
g_fJustRunKey = TRUE;
|
||||
break;
|
||||
|
||||
case UNICODE_NULL:
|
||||
return;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static VOID
|
||||
WriteRegRun(VOID)
|
||||
{
|
||||
if (g_fNoRunKey) // If "/N" option is specified
|
||||
return; // Don't write
|
||||
|
||||
// Open "Run" key
|
||||
HKEY hKey;
|
||||
LSTATUS error = ::RegCreateKeyW(HKEY_CURRENT_USER,
|
||||
L"Software\\Microsoft\\Windows\\CurrentVersion\\Run",
|
||||
&hKey);
|
||||
if (error != ERROR_SUCCESS)
|
||||
return;
|
||||
|
||||
// Write the module path
|
||||
CModulePath ModPath;
|
||||
if (ModPath.Init(L"ctfmon.exe", FALSE))
|
||||
{
|
||||
DWORD cbData = (ModPath.m_cchPath + 1) * sizeof(WCHAR);
|
||||
::RegSetValueExW(hKey, L"ctfmon.exe", 0, REG_SZ, (BYTE*)ModPath.m_szPath, cbData);
|
||||
}
|
||||
|
||||
::RegCloseKey(hKey);
|
||||
}
|
||||
|
||||
static HRESULT
|
||||
GetGlobalCompartment(
|
||||
_In_ REFGUID guid,
|
||||
_Inout_ ITfCompartment **ppComp)
|
||||
{
|
||||
*ppComp = NULL;
|
||||
|
||||
ITfCompartmentMgr *pCompMgr = NULL;
|
||||
HRESULT hr = TF_GetGlobalCompartment(&pCompMgr);
|
||||
if (FAILED(hr))
|
||||
return hr;
|
||||
|
||||
if (!pCompMgr)
|
||||
return E_FAIL;
|
||||
|
||||
hr = pCompMgr->GetCompartment(guid, ppComp);
|
||||
pCompMgr->Release();
|
||||
return hr;
|
||||
}
|
||||
|
||||
static HRESULT
|
||||
SetGlobalCompartmentDWORD(
|
||||
_In_ REFGUID guid,
|
||||
_In_ DWORD dwValue)
|
||||
{
|
||||
HRESULT hr;
|
||||
VARIANT vari;
|
||||
ITfCompartment *pComp;
|
||||
|
||||
hr = GetGlobalCompartment(guid, &pComp);
|
||||
if (FAILED(hr))
|
||||
return hr;
|
||||
|
||||
V_VT(&vari) = VT_I4;
|
||||
V_I4(&vari) = dwValue;
|
||||
hr = pComp->SetValue(0, &vari);
|
||||
|
||||
pComp->Release();
|
||||
return hr;
|
||||
}
|
||||
|
||||
static BOOL
|
||||
CheckX64System(
|
||||
_In_ LPWSTR lpCmdLine)
|
||||
{
|
||||
// Is the system x64?
|
||||
SYSTEM_INFO SystemInfo;
|
||||
::GetSystemInfo(&SystemInfo);
|
||||
if (SystemInfo.wProcessorArchitecture != PROCESSOR_ARCHITECTURE_IA64 ||
|
||||
SystemInfo.wProcessorArchitecture != PROCESSOR_ARCHITECTURE_AMD64)
|
||||
{
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
// Get GetSystemWow64DirectoryW function
|
||||
g_hKernel32 = GetSystemModuleHandle(L"kernel32.dll", FALSE);
|
||||
g_fnGetSystemWow64DirectoryW =
|
||||
(FN_GetSystemWow64DirectoryW)::GetProcAddress(g_hKernel32, "GetSystemWow64DirectoryW");
|
||||
if (!g_fnGetSystemWow64DirectoryW)
|
||||
return FALSE;
|
||||
|
||||
// Build WoW64 ctfmon.exe pathname
|
||||
WCHAR szPath[MAX_PATH];
|
||||
UINT cchPath = g_fnGetSystemWow64DirectoryW(szPath, _countof(szPath));
|
||||
if (!cchPath && FAILED(StringCchCatW(szPath, _countof(szPath), L"\\ctfmon.exe")))
|
||||
return FALSE;
|
||||
|
||||
// Create a WoW64 ctfmon.exe process
|
||||
PROCESS_INFORMATION pi;
|
||||
STARTUPINFOW si = { sizeof(si) };
|
||||
si.wShowWindow = SW_SHOWMINNOACTIVE;
|
||||
if (!::CreateProcessW(szPath, lpCmdLine, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi))
|
||||
return FALSE;
|
||||
|
||||
::CloseHandle(pi.hThread);
|
||||
::CloseHandle(pi.hProcess);
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
static BOOL
|
||||
InitApp(
|
||||
_In_ HINSTANCE hInstance,
|
||||
_In_ LPWSTR lpCmdLine)
|
||||
{
|
||||
g_hInst = hInstance; // Save the instance handle
|
||||
|
||||
g_uACP = ::GetACP(); // Save the active codepage
|
||||
g_bOnWow64 = IsWow64(); // Is the current process on WoW64?
|
||||
g_dwOsInfo = GetOSInfo(); // Get OS info
|
||||
|
||||
// Create a mutex for Cicero
|
||||
g_hCicMutex = TF_CreateCicLoadMutex(&g_fWinLogon);
|
||||
if (!g_hCicMutex)
|
||||
return FALSE;
|
||||
|
||||
// Write to "Run" registry key for starting up
|
||||
WriteRegRun();
|
||||
|
||||
// Call SetProcessShutdownParameters if possible
|
||||
if (g_dwOsInfo & OSINFO_NT)
|
||||
{
|
||||
g_hKernel32 = GetSystemModuleHandle(L"kernel32.dll", FALSE);
|
||||
g_fnSetProcessShutdownParameters =
|
||||
(FN_SetProcessShutdownParameters)
|
||||
::GetProcAddress(g_hKernel32, "SetProcessShutdownParameters");
|
||||
if (g_fnSetProcessShutdownParameters)
|
||||
g_fnSetProcessShutdownParameters(0xF0, SHUTDOWN_NORETRY);
|
||||
}
|
||||
|
||||
// Start text framework
|
||||
TF_InitSystem();
|
||||
|
||||
// Start watching registry if x86/x64 native
|
||||
if (!g_bOnWow64)
|
||||
CRegWatcher::Init();
|
||||
|
||||
// Create TIP Bar loader window
|
||||
g_pLoaderWnd = new CLoaderWnd();
|
||||
if (!g_pLoaderWnd || !g_pLoaderWnd->Init())
|
||||
return FALSE;
|
||||
|
||||
if (g_pLoaderWnd->CreateWnd())
|
||||
{
|
||||
// Go to the bottom of the hell
|
||||
::SetWindowPos(g_pLoaderWnd->m_hWnd, HWND_BOTTOM, 0, 0, 0, 0,
|
||||
SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
|
||||
}
|
||||
|
||||
// Display TIP Bar Popup if x86/x64 native
|
||||
if (!g_bOnWow64)
|
||||
GetPopupTipbar(g_pLoaderWnd->m_hWnd, g_fWinLogon);
|
||||
|
||||
// Do x64 stuffs
|
||||
CheckX64System(lpCmdLine);
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
VOID
|
||||
UninitApp(VOID)
|
||||
{
|
||||
// Close TIP Bar Popup
|
||||
ClosePopupTipbar();
|
||||
|
||||
// Close the mutex
|
||||
::CloseHandle(g_hCicMutex);
|
||||
g_hCicMutex = NULL;
|
||||
|
||||
// Quit watching registry if x86/x64 native
|
||||
if (!g_bOnWow64)
|
||||
CRegWatcher::Uninit();
|
||||
}
|
||||
|
||||
static INT
|
||||
DoMainLoop(VOID)
|
||||
{
|
||||
MSG msg;
|
||||
|
||||
if (g_bOnWow64) // Is the current process on WoW64?
|
||||
{
|
||||
// Just a simple message loop
|
||||
while (::GetMessageW(&msg, NULL, 0, 0))
|
||||
{
|
||||
::TranslateMessage(&msg);
|
||||
::DispatchMessageW(&msg);
|
||||
}
|
||||
return (INT)msg.wParam;
|
||||
}
|
||||
|
||||
// Open the existing event by the name
|
||||
HANDLE hSwitchEvent = ::OpenEventW(SYNCHRONIZE, FALSE, L"WinSta0_DesktopSwitch");
|
||||
|
||||
// The target events to watch
|
||||
HANDLE ahEvents[WATCHENTRY_MAX + 1];
|
||||
|
||||
// Borrow some handles from CRegWatcher
|
||||
CopyMemory(ahEvents, CRegWatcher::s_ahWatchEvents, WATCHENTRY_MAX * sizeof(HANDLE));
|
||||
|
||||
ahEvents[WI_DESKTOP_SWITCH] = hSwitchEvent; // Add it
|
||||
|
||||
// Another message loop
|
||||
for (;;)
|
||||
{
|
||||
// Wait for target signal
|
||||
DWORD dwWait = ::MsgWaitForMultipleObjects(_countof(ahEvents), ahEvents, 0, INFINITE,
|
||||
QS_ALLINPUT);
|
||||
if (dwWait == (WAIT_OBJECT_0 + _countof(ahEvents))) // Is input available?
|
||||
{
|
||||
// Do the events
|
||||
while (::PeekMessageW(&msg, NULL, 0, 0, PM_REMOVE))
|
||||
{
|
||||
if (msg.message == WM_QUIT)
|
||||
goto Quit;
|
||||
|
||||
::TranslateMessage(&msg);
|
||||
::DispatchMessageW(&msg);
|
||||
}
|
||||
}
|
||||
else if (dwWait == (WAIT_OBJECT_0 + WI_DESKTOP_SWITCH)) // Desktop switch?
|
||||
{
|
||||
SetGlobalCompartmentDWORD(GUID_COMPARTMENT_SPEECH_OPENCLOSE, 0);
|
||||
::ResetEvent(hSwitchEvent);
|
||||
}
|
||||
else // Do the other events
|
||||
{
|
||||
CRegWatcher::OnEvent(dwWait - WAIT_OBJECT_0);
|
||||
}
|
||||
}
|
||||
|
||||
Quit:
|
||||
::CloseHandle(hSwitchEvent);
|
||||
|
||||
return (INT)msg.wParam;
|
||||
}
|
||||
|
||||
// The main function for Unicode Win32
|
||||
EXTERN_C INT WINAPI
|
||||
wWinMain(
|
||||
HINSTANCE hInstance,
|
||||
HINSTANCE hPrevInst,
|
||||
LPWSTR lpCmdLine,
|
||||
INT nCmdShow)
|
||||
{
|
||||
UNREFERENCED_PARAMETER(hPrevInst);
|
||||
UNREFERENCED_PARAMETER(nCmdShow);
|
||||
|
||||
// Parse command line
|
||||
ParseCommandLine(lpCmdLine);
|
||||
|
||||
if (g_fJustRunKey) // If "/R" option is specified
|
||||
{
|
||||
// Just write registry and exit
|
||||
WriteRegRun();
|
||||
return 1;
|
||||
}
|
||||
|
||||
// Initialize the application
|
||||
if (!InitApp(hInstance, lpCmdLine))
|
||||
return 0;
|
||||
|
||||
// The main loop
|
||||
INT ret = DoMainLoop();
|
||||
|
||||
// Clean up the loader
|
||||
if (g_pLoaderWnd)
|
||||
{
|
||||
delete g_pLoaderWnd;
|
||||
g_pLoaderWnd = NULL;
|
||||
}
|
||||
|
||||
// Un-initialize app and text framework
|
||||
if (!CLoaderWnd::s_bUninitedSystem)
|
||||
{
|
||||
UninitApp();
|
||||
TF_UninitSystem();
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue