Mirrors/PowerToys
1
0
Fork 0
mirror of https://github.com/microsoft/PowerToys.git synced 2026-04-10 05:06:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

PowerLauncher: fix disabling by adjusting OpenProcess params (#4577)

Browse Source
This commit is contained in:
Andrey Nekrasov
2020-07-01 12:36:26 +03:00
committed by GitHub
parent 17343210c0
commit 738b5f5707
1 changed files with 255 additions and 233 deletions
Download Patch File Download Diff File Expand all files Collapse all files

488
src/modules/launcher/Microsoft.Launcher/dllmain.cpp
View File

@@ -10,271 +10,293 @@
extern "C" IMAGE_DOS_HEADER __ImageBase; extern "C" IMAGE_DOS_HEADER __ImageBase;
BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved) { BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
switch (ul_reason_for_call) { {
case DLL_PROCESS_ATTACH: switch (ul_reason_for_call)
Trace::RegisterProvider(); {
break; case DLL_PROCESS_ATTACH:
case DLL_THREAD_ATTACH: Trace::RegisterProvider();
case DLL_THREAD_DETACH: break;
break; case DLL_THREAD_ATTACH:
case DLL_PROCESS_DETACH: case DLL_THREAD_DETACH:
Trace::UnregisterProvider(); break;
break; case DLL_PROCESS_DETACH:
} Trace::UnregisterProvider();
return TRUE; break;
}
return TRUE;
} }
// These are the properties shown in the Settings page. // These are the properties shown in the Settings page.
struct ModuleSettings { struct ModuleSettings
{
} g_settings; } g_settings;
// Implement the PowerToy Module Interface and all the required methods. // Implement the PowerToy Module Interface and all the required methods.
class Microsoft_Launcher : public PowertoyModuleIface { class Microsoft_Launcher : public PowertoyModuleIface
{
private: private:
// The PowerToy state. // The PowerToy state.
bool m_enabled = false; bool m_enabled = false;
// Load initial settings from the persisted values. // Load initial settings from the persisted values.
void init_settings(); void init_settings();
// Handle to launch and terminate the launcher // Handle to launch and terminate the launcher
HANDLE m_hProcess; HANDLE m_hProcess;
//contains the name of the powerToys //contains the name of the powerToys
std::wstring app_name; std::wstring app_name;
// Time to wait for process to close after sending WM_CLOSE signal
static const int MAX_WAIT_MILLISEC = 10000;
// Time to wait for process to close after sending WM_CLOSE signal
static const int MAX_WAIT_MILLISEC = 10000;
public: public:
// Constructor // Constructor
Microsoft_Launcher() { Microsoft_Launcher()
app_name = GET_RESOURCE_STRING(IDS_LAUNCHER_NAME); {
init_settings(); app_name = GET_RESOURCE_STRING(IDS_LAUNCHER_NAME);
}; init_settings();
};
~Microsoft_Launcher() { ~Microsoft_Launcher()
if (m_enabled) {
{ if (m_enabled)
terminateProcess(); {
} terminateProcess();
m_enabled = false; }
} m_enabled = false;
// Destroy the powertoy and free memory
virtual void destroy() override {
delete this;
}
// Return the display name of the powertoy, this will be cached by the runner
virtual const wchar_t* get_name() override {
return app_name.c_str();
}
// Return array of the names of all events that this powertoy listens for, with
// nullptr as the last element of the array. Nullptr can also be returned for empty
// list.
virtual const wchar_t** get_events() override {
static const wchar_t* events[] = { nullptr };
// Available events:
// - ll_keyboard
// - win_hook_event
//
// static const wchar_t* events[] = { ll_keyboard,
// win_hook_event,
// nullptr };
return events;
}
// Return JSON with the configuration options.
virtual bool get_config(wchar_t* buffer, int* buffer_size) override {
HINSTANCE hinstance = reinterpret_cast<HINSTANCE>(&__ImageBase);
// Create a Settings object.
PowerToysSettings::Settings settings(hinstance, get_name());
settings.set_description(GET_RESOURCE_STRING(IDS_LAUNCHER_SETTINGS_DESC));
settings.set_overview_link(L"https://aka.ms/PowerToysOverview_PowerToysRun");
return settings.serialize_to_buffer(buffer, buffer_size);
}
// Signal from the Settings editor to call a custom action.
// This can be used to spawn more complex editors.
virtual void call_custom_action(const wchar_t* action) override {
static UINT custom_action_num_calls = 0;
try {
// Parse the action values, including name.
PowerToysSettings::CustomActionObject action_object =
PowerToysSettings::CustomActionObject::from_json_string(action);
} }
catch (std::exception ex) {
// Improper JSON. // Destroy the powertoy and free memory
virtual void destroy() override
{
delete this;
} }
}
// Called by the runner to pass the updated settings values as a serialized JSON. // Return the display name of the powertoy, this will be cached by the runner
virtual void set_config(const wchar_t* config) override { virtual const wchar_t* get_name() override
try { {
// Parse the input JSON string. return app_name.c_str();
PowerToysSettings::PowerToyValues values =
PowerToysSettings::PowerToyValues::from_json_string(config);
// If you don't need to do any custom processing of the settings, proceed
// to persists the values calling:
values.save_to_settings_file();
// Otherwise call a custom function to process the settings before saving them to disk:
// save_settings();
} }
catch (std::exception ex) {
// Improper JSON. // Return array of the names of all events that this powertoy listens for, with
// nullptr as the last element of the array. Nullptr can also be returned for empty
// list.
virtual const wchar_t** get_events() override
{
static const wchar_t* events[] = { nullptr };
// Available events:
// - ll_keyboard
// - win_hook_event
//
// static const wchar_t* events[] = { ll_keyboard,
// win_hook_event,
// nullptr };
return events;
} }
}
// Enable the powertoy // Return JSON with the configuration options.
virtual void enable() virtual bool get_config(wchar_t* buffer, int* buffer_size) override
{ {
// Start PowerLauncher.exe only if the OS is 19H1 or higher HINSTANCE hinstance = reinterpret_cast<HINSTANCE>(&__ImageBase);
if (UseNewSettings())
{
unsigned long powertoys_pid = GetCurrentProcessId();
if (!is_process_elevated(false)) // Create a Settings object.
{ PowerToysSettings::Settings settings(hinstance, get_name());
std::wstring executable_args = L""; settings.set_description(GET_RESOURCE_STRING(IDS_LAUNCHER_SETTINGS_DESC));
executable_args.append(std::to_wstring(powertoys_pid)); settings.set_overview_link(L"https://aka.ms/PowerToysOverview_PowerToysRun");
SHELLEXECUTEINFOW sei{ sizeof(sei) }; return settings.serialize_to_buffer(buffer, buffer_size);
sei.fMask = { SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_NO_UI };
sei.lpFile = L"modules\\launcher\\PowerLauncher.exe";
sei.nShow = SW_SHOWNORMAL;
sei.lpParameters = executable_args.data();
ShellExecuteExW(&sei);
m_hProcess = sei.hProcess;
}
else
{
std::wstring action_runner_path = get_module_folderpath();
std::wstring params;
params += L"-run-non-elevated ";
params += L"-target modules\\launcher\\PowerLauncher.exe ";
params += L"-pidFile ";
params += POWER_LAUNCHER_PID_SHARED_FILE;
params += L" " + std::to_wstring(powertoys_pid) + L" ";
action_runner_path += L"\\action_runner.exe";
// Set up the shared file from which to retrieve the PID of PowerLauncher
HANDLE hMapFile = CreateFileMappingW(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, sizeof(DWORD), POWER_LAUNCHER_PID_SHARED_FILE);
if (hMapFile)
{
PDWORD pidBuffer = reinterpret_cast<PDWORD>(MapViewOfFile(hMapFile, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(DWORD)));
if (pidBuffer)
{
*pidBuffer = 0;
m_hProcess = NULL;
if (run_non_elevated(action_runner_path, params, pidBuffer))
{
const int maxRetries = 80;
for (int retry = 0; retry < maxRetries; ++retry)
{
Sleep(50);
DWORD pid = *pidBuffer;
if (pid)
{
m_hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, pid);
break;
}
}
}
}
CloseHandle(hMapFile);
}
}
}
m_enabled = true;
}
// Disable the powertoy
virtual void disable()
{
if (m_enabled)
{
terminateProcess();
}
m_enabled = false;
}
// Returns if the powertoys is enabled
virtual bool is_enabled() override {
return m_enabled;
}
// Handle incoming event, data is event-specific
virtual intptr_t signal_event(const wchar_t* name, intptr_t data) override {
if (wcscmp(name, ll_keyboard) == 0) {
auto& event = *(reinterpret_cast<LowlevelKeyboardEvent*>(data));
// Return 1 if the keypress is to be suppressed (not forwarded to Windows),
// otherwise return 0.
return 0;
} }
else if (wcscmp(name, win_hook_event) == 0) {
auto& event = *(reinterpret_cast<WinHookEvent*>(data)); // Signal from the Settings editor to call a custom action.
// Return value is ignored // This can be used to spawn more complex editors.
return 0; virtual void call_custom_action(const wchar_t* action) override
{
static UINT custom_action_num_calls = 0;
try
{
// Parse the action values, including name.
PowerToysSettings::CustomActionObject action_object =
PowerToysSettings::CustomActionObject::from_json_string(action);
}
catch (std::exception ex)
{
// Improper JSON.
}
} }
return 0;
}
// Callback to send WM_CLOSE signal to each top level window. // Called by the runner to pass the updated settings values as a serialized JSON.
static BOOL CALLBACK requestMainWindowClose(HWND nextWindow, LPARAM closePid) { virtual void set_config(const wchar_t* config) override
DWORD windowPid; {
GetWindowThreadProcessId(nextWindow, &windowPid); try
{
// Parse the input JSON string.
PowerToysSettings::PowerToyValues values =
PowerToysSettings::PowerToyValues::from_json_string(config);
if (windowPid == (DWORD)closePid) // If you don't need to do any custom processing of the settings, proceed
::PostMessage(nextWindow, WM_CLOSE, 0, 0); // to persists the values calling:
values.save_to_settings_file();
// Otherwise call a custom function to process the settings before saving them to disk:
// save_settings();
}
catch (std::exception ex)
{
// Improper JSON.
}
}
return true; // Enable the powertoy
} virtual void enable()
{
// Start PowerLauncher.exe only if the OS is 19H1 or higher
if (UseNewSettings())
{
unsigned long powertoys_pid = GetCurrentProcessId();
// Terminate process by sending WM_CLOSE signal and if it fails, force terminate. if (!is_process_elevated(false))
void terminateProcess() { {
DWORD processID = GetProcessId(m_hProcess); std::wstring executable_args = L"";
EnumWindows(&requestMainWindowClose, processID); executable_args.append(std::to_wstring(powertoys_pid));
const DWORD result = WaitForSingleObject(m_hProcess, MAX_WAIT_MILLISEC);
if (result == WAIT_TIMEOUT)
{
TerminateProcess(m_hProcess, 1);
}
}
/* Register helper class to handle system menu items related actions. */ SHELLEXECUTEINFOW sei{ sizeof(sei) };
virtual void register_system_menu_helper(PowertoySystemMenuIface* helper) {} sei.fMask = { SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_NO_UI };
/* Handle action on system menu item. */ sei.lpFile = L"modules\\launcher\\PowerLauncher.exe";
virtual void signal_system_menu_action(const wchar_t* name) {} sei.nShow = SW_SHOWNORMAL;
sei.lpParameters = executable_args.data();
ShellExecuteExW(&sei);
m_hProcess = sei.hProcess;
}
else
{
std::wstring action_runner_path = get_module_folderpath();
std::wstring params;
params += L"-run-non-elevated ";
params += L"-target modules\\launcher\\PowerLauncher.exe ";
params += L"-pidFile ";
params += POWER_LAUNCHER_PID_SHARED_FILE;
params += L" " + std::to_wstring(powertoys_pid) + L" ";
action_runner_path += L"\\action_runner.exe";
// Set up the shared file from which to retrieve the PID of PowerLauncher
HANDLE hMapFile = CreateFileMappingW(INVALID_HANDLE_VALUE, NULL, PAGE_READWRITE, 0, sizeof(DWORD), POWER_LAUNCHER_PID_SHARED_FILE);
if (hMapFile)
{
PDWORD pidBuffer = reinterpret_cast<PDWORD>(MapViewOfFile(hMapFile, FILE_MAP_ALL_ACCESS, 0, 0, sizeof(DWORD)));
if (pidBuffer)
{
*pidBuffer = 0;
m_hProcess = NULL;
if (run_non_elevated(action_runner_path, params, pidBuffer))
{
const int maxRetries = 80;
for (int retry = 0; retry < maxRetries; ++retry)
{
Sleep(50);
DWORD pid = *pidBuffer;
if (pid)
{
m_hProcess = OpenProcess(PROCESS_TERMINATE | PROCESS_QUERY_INFORMATION | SYNCHRONIZE, FALSE, pid);
break;
}
}
}
}
CloseHandle(hMapFile);
}
}
}
m_enabled = true;
}
// Disable the powertoy
virtual void disable()
{
if (m_enabled)
{
terminateProcess();
}
m_enabled = false;
}
// Returns if the powertoys is enabled
virtual bool is_enabled() override
{
return m_enabled;
}
// Handle incoming event, data is event-specific
virtual intptr_t signal_event(const wchar_t* name, intptr_t data) override
{
if (wcscmp(name, ll_keyboard) == 0)
{
auto& event = *(reinterpret_cast<LowlevelKeyboardEvent*>(data));
// Return 1 if the keypress is to be suppressed (not forwarded to Windows),
// otherwise return 0.
return 0;
}
else if (wcscmp(name, win_hook_event) == 0)
{
auto& event = *(reinterpret_cast<WinHookEvent*>(data));
// Return value is ignored
return 0;
}
return 0;
}
// Callback to send WM_CLOSE signal to each top level window.
static BOOL CALLBACK requestMainWindowClose(HWND nextWindow, LPARAM closePid)
{
DWORD windowPid;
GetWindowThreadProcessId(nextWindow, &windowPid);
if (windowPid == (DWORD)closePid)
::PostMessage(nextWindow, WM_CLOSE, 0, 0);
return true;
}
// Terminate process by sending WM_CLOSE signal and if it fails, force terminate.
void terminateProcess()
{
DWORD processID = GetProcessId(m_hProcess);
EnumWindows(&requestMainWindowClose, processID);
const DWORD result = WaitForSingleObject(m_hProcess, MAX_WAIT_MILLISEC);
if (result == WAIT_TIMEOUT || result == WAIT_FAILED)
{
TerminateProcess(m_hProcess, 1);
}
}
/* Register helper class to handle system menu items related actions. */
virtual void register_system_menu_helper(PowertoySystemMenuIface* helper) {}
/* Handle action on system menu item. */
virtual void signal_system_menu_action(const wchar_t* name) {}
}; };
// Load the settings file. // Load the settings file.
void Microsoft_Launcher::init_settings() { void Microsoft_Launcher::init_settings()
try { {
// Load and parse the settings file for this PowerToy. try
PowerToysSettings::PowerToyValues settings = {
PowerToysSettings::PowerToyValues::load_from_settings_file(get_name()); // Load and parse the settings file for this PowerToy.
PowerToysSettings::PowerToyValues settings =
} PowerToysSettings::PowerToyValues::load_from_settings_file(get_name());
catch (std::exception ex) { }
// Error while loading from the settings file. Let default values stay as they are. catch (std::exception ex)
} {
// Error while loading from the settings file. Let default values stay as they are.
}
} }
extern "C" __declspec(dllexport) PowertoyModuleIface* __cdecl powertoy_create()
extern "C" __declspec(dllexport) PowertoyModuleIface* __cdecl powertoy_create() { {
return new Microsoft_Launcher(); return new Microsoft_Launcher();
} }