Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-06-29 01:12:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[WMISERV][WUAUSERV] Stop the service if the stop event creation fails.

Browse source
This commit is contained in:
Eric Kohl 2019-07-21 18:51:33 +02:00
parent 85917e266a
commit 1f082ecc68
2 changed files with 24 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
base/services/wmisvc/wmisvc.c
View file

@ -44,7 +44,7 @@ static WCHAR ServiceName[] = L"winmgmt";
static SERVICE_STATUS_HANDLE ServiceStatusHandle;
static SERVICE_STATUS ServiceStatus;
static HANDLE ShutdownEvent;
static HANDLE hStopEvent = NULL;
/* FUNCTIONS *****************************************************************/
@ -85,7 +85,7 @@ ServiceControlHandler(DWORD dwControl,
{
case SERVICE_CONTROL_STOP:
DPRINT1(" SERVICE_CONTROL_STOP received\n");
SetEvent(ShutdownEvent);
SetEvent(hStopEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
@ -107,7 +107,7 @@ ServiceControlHandler(DWORD dwControl,
case SERVICE_CONTROL_SHUTDOWN:
DPRINT1(" SERVICE_CONTROL_SHUTDOWN received\n");
SetEvent(ShutdownEvent);
SetEvent(hStopEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
@ -134,13 +134,19 @@ ServiceMain(DWORD argc, LPTSTR *argv)
return;
}
ShutdownEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
hStopEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
if (hStopEvent == NULL)
{
DPRINT1("CreateEvent() failed! (Error %lu)\n", GetLastError());
goto done;
}
UpdateServiceStatus(SERVICE_RUNNING);
WaitForSingleObject(ShutdownEvent, INFINITE);
CloseHandle(ShutdownEvent);
WaitForSingleObject(hStopEvent, INFINITE);
CloseHandle(hStopEvent);
done:
UpdateServiceStatus(SERVICE_STOPPED);
}

18
base/services/wuauserv/wuauserv.c
View file

@ -15,7 +15,7 @@ static WCHAR ServiceName[] = L"wuauserv";
static SERVICE_STATUS_HANDLE ServiceStatusHandle;
static SERVICE_STATUS ServiceStatus;
static HANDLE exitEvent = NULL;
static HANDLE hStopEvent = NULL;
/* FUNCTIONS *****************************************************************/
@ -55,7 +55,7 @@ ServiceControlHandler(DWORD dwControl,
{
case SERVICE_CONTROL_STOP:
DPRINT1("WU ServiceControlHandler() SERVICE_CONTROL_STOP received\n");
SetEvent(exitEvent);
SetEvent(hStopEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
@ -77,7 +77,7 @@ ServiceControlHandler(DWORD dwControl,
case SERVICE_CONTROL_SHUTDOWN:
DPRINT1("WU ServiceControlHandler() SERVICE_CONTROL_SHUTDOWN received\n");
SetEvent(exitEvent);
SetEvent(hStopEvent);
UpdateServiceStatus(SERVICE_STOP_PENDING);
return ERROR_SUCCESS;
@ -104,13 +104,19 @@ ServiceMain(DWORD argc, LPTSTR *argv)
return;
}
exitEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
hStopEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
if (hStopEvent == NULL)
{
DPRINT1("CreateEvent() failed! (Error %lu)\n", GetLastError());
goto done;
}
UpdateServiceStatus(SERVICE_RUNNING);
WaitForSingleObject(exitEvent, INFINITE);
CloseHandle(exitEvent);
WaitForSingleObject(hStopEvent, INFINITE);
CloseHandle(hStopEvent);
done:
UpdateServiceStatus(SERVICE_STOPPED);
}