[ros-diffs] [ekohl] 74981: [SCHEDSVC] Add a loop at the end of SchedServiceMain that will run until a stop event is signaled. This loop will be used to run the schduled jobs.

Author: ekohl
Date: Sat Jun 10 10:51:49 2017
New Revision: 74981

URL: http://svn.reactos.org/svn/reactos?rev=74981&view=rev
Log:
[SCHEDSVC]
Add a loop at the end of SchedServiceMain that will run until a stop event is signaled.
This loop will be used to run the schduled jobs.

Modified:
    trunk/reactos/base/services/schedsvc/schedsvc.c

Modified: trunk/reactos/base/services/schedsvc/schedsvc.c
URL:
http://svn.reactos.org/svn/reactos/trunk/reactos/base/services/schedsvc/sch…
==============================================================================
--- trunk/reactos/base/services/schedsvc/schedsvc.c	[iso-8859-1] (original)
+++ trunk/reactos/base/services/schedsvc/schedsvc.c	[iso-8859-1] Sat Jun 10 10:51:49 2017
@@ -37,6 +37,8 @@
 static SERVICE_STATUS_HANDLE ServiceStatusHandle;
 static SERVICE_STATUS ServiceStatus;
 
+static BOOL bStopService = FALSE;
+
 /* FUNCTIONS *****************************************************************/
 
 static VOID
@@ -44,7 +46,14 @@
 {
     ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
     ServiceStatus.dwCurrentState = dwState;
-    ServiceStatus.dwControlsAccepted = 0;
+
+    if (dwState == SERVICE_PAUSED || dwState == SERVICE_RUNNING)
+        ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
+                                           SERVICE_ACCEPT_SHUTDOWN |
+                                           SERVICE_ACCEPT_PAUSE_CONTINUE;
+    else
+        ServiceStatus.dwControlsAccepted = 0;
+
     ServiceStatus.dwWin32ExitCode = 0;
     ServiceStatus.dwServiceSpecificExitCode = 0;
     ServiceStatus.dwCheckPoint = 0;
@@ -73,11 +82,12 @@
     switch (dwControl)
     {
         case SERVICE_CONTROL_STOP:
-            TRACE("  SERVICE_CONTROL_STOP received\n");
+        case SERVICE_CONTROL_SHUTDOWN:
+            TRACE("  SERVICE_CONTROL_STOP/SERVICE_CONTROL_SHUTDOWN
received\n");
             UpdateServiceStatus(SERVICE_STOP_PENDING);
             /* Stop listening to incoming RPC messages */
             RpcMgmtStopServerListening(NULL);
-            UpdateServiceStatus(SERVICE_STOPPED);
+            bStopService = TRUE;
             return ERROR_SUCCESS;
 
         case SERVICE_CONTROL_PAUSE:
@@ -96,14 +106,7 @@
                              &ServiceStatus);
             return ERROR_SUCCESS;
 
-        case SERVICE_CONTROL_SHUTDOWN:
-            TRACE("  SERVICE_CONTROL_SHUTDOWN received\n");
-            UpdateServiceStatus(SERVICE_STOP_PENDING);
-            RpcMgmtStopServerListening(NULL);
-            UpdateServiceStatus(SERVICE_STOPPED);
-            return ERROR_SUCCESS;
-
-        default :
+        default:
             TRACE("  Control %lu received\n", dwControl);
             return ERROR_CALL_NOT_IMPLEMENTED;
     }
@@ -112,7 +115,7 @@
 
 static
 DWORD
-ServiceInit(VOID)
+ServiceInit(PHANDLE phEvent)
 {
     HANDLE hThread;
     DWORD dwError;
@@ -143,11 +146,19 @@
                            NULL);
     if (!hThread)
     {
-        ERR("Can't create PortThread\n");
+        ERR("Could not create the RPC thread\n");
         return GetLastError();
     }
 
     CloseHandle(hThread);
+
+    /* Create the scheduler event */
+    *phEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
+    if (*phEvent == NULL)
+    {
+        ERR("Could not create the scheduler event\n");
+        return GetLastError();
+    }
 
     return ERROR_SUCCESS;
 }
@@ -156,6 +167,7 @@
 VOID WINAPI
 SchedServiceMain(DWORD argc, LPTSTR *argv)
 {
+    HANDLE hEvent = NULL;
     DWORD dwError;
 
     UNREFERENCED_PARAMETER(argc);
@@ -174,7 +186,7 @@
 
     UpdateServiceStatus(SERVICE_START_PENDING);
 
-    dwError = ServiceInit();
+    dwError = ServiceInit(&hEvent);
     if (dwError != ERROR_SUCCESS)
     {
         ERR("Service stopped (dwError: %lu\n", dwError);
@@ -183,6 +195,23 @@
     }
 
     UpdateServiceStatus(SERVICE_RUNNING);
+
+    for (;;)
+    {
+        /* Leave the loop, if the service has to be stopped */
+        if (bStopService)
+            break;
+
+        /* Wait for the next timeout */
+        WaitForSingleObject(hEvent, 5000);
+        TRACE("Service running!\n");
+    }
+
+    /* Close the scheduler event handle */
+    CloseHandle(hEvent);
+
+    /* Stop the service */
+    UpdateServiceStatus(SERVICE_STOPPED);
 }