lywsvip/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2026-05-23 07:40:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

[DCOMLAUNCH][WMISVC] Stop busy waiting shutdown

Browse Source
This commit is contained in:
Pierre Schweitzer
2019-03-03 23:14:07 +01:00
parent bd0a28d1e4
commit e5b0087c8e
2 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
base/services/dcomlaunch/dcomlaunch.c
View File

@@ -25,6 +25,7 @@ static WCHAR ServiceName[] = L"dcomlaunch";
static SERVICE_STATUS_HANDLE ServiceStatusHandle;
static SERVICE_STATUS ServiceStatus;
static HANDLE ShutdownEvent;
/* FUNCTIONS *****************************************************************/
@@ -62,7 +63,8 @@ ServiceControlHandler(DWORD dwControl,
{
case SERVICE_CONTROL_STOP:
DPRINT1(" SERVICE_CONTROL_STOP received\n");
UpdateServiceStatus(SERVICE_STOPPED);
SetEvent(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
case SERVICE_CONTROL_PAUSE:
@@ -83,7 +85,8 @@ ServiceControlHandler(DWORD dwControl,
case SERVICE_CONTROL_SHUTDOWN:
DPRINT1(" SERVICE_CONTROL_SHUTDOWN received\n");
UpdateServiceStatus(SERVICE_STOPPED);
SetEvent(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
default :
@@ -111,14 +114,14 @@ ServiceMain(DWORD argc, LPTSTR *argv)
return;
}
ShutdownEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
DealWithDeviceEvent();
UpdateServiceStatus(SERVICE_RUNNING);
do
{
Sleep(1);
} while (1);
WaitForSingleObject(ShutdownEvent, INFINITE);
CloseHandle(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOPPED);
}

15
base/services/wmisvc/wmisvc.c
View File

@@ -44,6 +44,7 @@ static WCHAR ServiceName[] = L"winmgmt";
static SERVICE_STATUS_HANDLE ServiceStatusHandle;
static SERVICE_STATUS ServiceStatus;
static HANDLE ShutdownEvent;
/* FUNCTIONS *****************************************************************/
@@ -81,7 +82,8 @@ ServiceControlHandler(DWORD dwControl,
{
case SERVICE_CONTROL_STOP:
DPRINT1(" SERVICE_CONTROL_STOP received\n");
UpdateServiceStatus(SERVICE_STOPPED);
SetEvent(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
case SERVICE_CONTROL_PAUSE:
@@ -102,7 +104,8 @@ ServiceControlHandler(DWORD dwControl,
case SERVICE_CONTROL_SHUTDOWN:
DPRINT1(" SERVICE_CONTROL_SHUTDOWN received\n");
UpdateServiceStatus(SERVICE_STOPPED);
SetEvent(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
default :
@@ -128,12 +131,12 @@ ServiceMain(DWORD argc, LPTSTR *argv)
return;
}
ShutdownEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
UpdateServiceStatus(SERVICE_RUNNING);
do
{
Sleep(1);
} while (1);
WaitForSingleObject(ShutdownEvent, INFINITE);
CloseHandle(ShutdownEvent);
UpdateServiceStatus(SERVICE_STOPPED);
}